Index: third_party/protobuf/src/google/protobuf/stubs/stringprintf.cc |
=================================================================== |
--- third_party/protobuf/src/google/protobuf/stubs/stringprintf.cc (revision 0) |
+++ third_party/protobuf/src/google/protobuf/stubs/stringprintf.cc (revision 0) |
@@ -0,0 +1,175 @@ |
+// Protocol Buffers - Google's data interchange format |
+// Copyright 2012 Google Inc. All rights reserved. |
+// http://code.google.com/p/protobuf/ |
+// |
+// Redistribution and use in source and binary forms, with or without |
+// modification, are permitted provided that the following conditions are |
+// met: |
+// |
+// * Redistributions of source code must retain the above copyright |
+// notice, this list of conditions and the following disclaimer. |
+// * Redistributions in binary form must reproduce the above |
+// copyright notice, this list of conditions and the following disclaimer |
+// in the documentation and/or other materials provided with the |
+// distribution. |
+// * Neither the name of Google Inc. nor the names of its |
+// contributors may be used to endorse or promote products derived from |
+// this software without specific prior written permission. |
+// |
+// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
+// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
+// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
+// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
+// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
+// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
+// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
+// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
+// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
+// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
+// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
+ |
+// from google3/base/stringprintf.cc |
+ |
+#include <google/protobuf/stubs/stringprintf.h> |
+ |
+#include <errno.h> |
+#include <stdarg.h> // For va_list and related operations |
+#include <stdio.h> // MSVC requires this for _vsnprintf |
+#include <vector> |
+#include <google/protobuf/stubs/common.h> |
+#include <google/protobuf/testing/googletest.h> |
+ |
+namespace google { |
+namespace protobuf { |
+ |
+#ifdef _MSC_VER |
+enum { IS_COMPILER_MSVC = 1 }; |
+#ifndef va_copy |
+// Define va_copy for MSVC. This is a hack, assuming va_list is simply a |
+// pointer into the stack and is safe to copy. |
+#define va_copy(dest, src) ((dest) = (src)) |
+#endif |
+#else |
+enum { IS_COMPILER_MSVC = 0 }; |
+#endif |
+ |
+void StringAppendV(string* dst, const char* format, va_list ap) { |
+ // First try with a small fixed size buffer |
+ static const int kSpaceLength = 1024; |
+ char space[kSpaceLength]; |
+ |
+ // It's possible for methods that use a va_list to invalidate |
+ // the data in it upon use. The fix is to make a copy |
+ // of the structure before using it and use that copy instead. |
+ va_list backup_ap; |
+ va_copy(backup_ap, ap); |
+ int result = vsnprintf(space, kSpaceLength, format, backup_ap); |
+ va_end(backup_ap); |
+ |
+ if (result < kSpaceLength) { |
+ if (result >= 0) { |
+ // Normal case -- everything fit. |
+ dst->append(space, result); |
+ return; |
+ } |
+ |
+ if (IS_COMPILER_MSVC) { |
+ // Error or MSVC running out of space. MSVC 8.0 and higher |
+ // can be asked about space needed with the special idiom below: |
+ va_copy(backup_ap, ap); |
+ result = vsnprintf(NULL, 0, format, backup_ap); |
+ va_end(backup_ap); |
+ } |
+ |
+ if (result < 0) { |
+ // Just an error. |
+ return; |
+ } |
+ } |
+ |
+ // Increase the buffer size to the size requested by vsnprintf, |
+ // plus one for the closing \0. |
+ int length = result+1; |
+ char* buf = new char[length]; |
+ |
+ // Restore the va_list before we use it again |
+ va_copy(backup_ap, ap); |
+ result = vsnprintf(buf, length, format, backup_ap); |
+ va_end(backup_ap); |
+ |
+ if (result >= 0 && result < length) { |
+ // It fit |
+ dst->append(buf, result); |
+ } |
+ delete[] buf; |
+} |
+ |
+ |
+string StringPrintf(const char* format, ...) { |
+ va_list ap; |
+ va_start(ap, format); |
+ string result; |
+ StringAppendV(&result, format, ap); |
+ va_end(ap); |
+ return result; |
+} |
+ |
+const string& SStringPrintf(string* dst, const char* format, ...) { |
+ va_list ap; |
+ va_start(ap, format); |
+ dst->clear(); |
+ StringAppendV(dst, format, ap); |
+ va_end(ap); |
+ return *dst; |
+} |
+ |
+void StringAppendF(string* dst, const char* format, ...) { |
+ va_list ap; |
+ va_start(ap, format); |
+ StringAppendV(dst, format, ap); |
+ va_end(ap); |
+} |
+ |
+// Max arguments supported by StringPrintVector |
+const int kStringPrintfVectorMaxArgs = 32; |
+ |
+// An empty block of zero for filler arguments. This is const so that if |
+// printf tries to write to it (via %n) then the program gets a SIGSEGV |
+// and we can fix the problem or protect against an attack. |
+static const char string_printf_empty_block[256] = { '\0' }; |
+ |
+string StringPrintfVector(const char* format, const vector<string>& v) { |
+ GOOGLE_CHECK_LE(v.size(), kStringPrintfVectorMaxArgs) |
+ << "StringPrintfVector currently only supports up to " |
+ << kStringPrintfVectorMaxArgs << " arguments. " |
+ << "Feel free to add support for more if you need it."; |
+ |
+ // Add filler arguments so that bogus format+args have a harder time |
+ // crashing the program, corrupting the program (%n), |
+ // or displaying random chunks of memory to users. |
+ |
+ const char* cstr[kStringPrintfVectorMaxArgs]; |
+ for (int i = 0; i < v.size(); ++i) { |
+ cstr[i] = v[i].c_str(); |
+ } |
+ for (int i = v.size(); i < GOOGLE_ARRAYSIZE(cstr); ++i) { |
+ cstr[i] = &string_printf_empty_block[0]; |
+ } |
+ |
+ // I do not know any way to pass kStringPrintfVectorMaxArgs arguments, |
+ // or any way to build a va_list by hand, or any API for printf |
+ // that accepts an array of arguments. The best I can do is stick |
+ // this COMPILE_ASSERT right next to the actual statement. |
+ |
+ GOOGLE_COMPILE_ASSERT(kStringPrintfVectorMaxArgs == 32, arg_count_mismatch); |
+ return StringPrintf(format, |
+ cstr[0], cstr[1], cstr[2], cstr[3], cstr[4], |
+ cstr[5], cstr[6], cstr[7], cstr[8], cstr[9], |
+ cstr[10], cstr[11], cstr[12], cstr[13], cstr[14], |
+ cstr[15], cstr[16], cstr[17], cstr[18], cstr[19], |
+ cstr[20], cstr[21], cstr[22], cstr[23], cstr[24], |
+ cstr[25], cstr[26], cstr[27], cstr[28], cstr[29], |
+ cstr[30], cstr[31]); |
+} |
+} // namespace protobuf |
+} // namespace google |
Property changes on: third_party/protobuf/src/google/protobuf/stubs/stringprintf.cc |
___________________________________________________________________ |
Added: svn:eol-style |
+ LF |