src/third_party/protobuf/src/google/protobuf/stubs/substitute.cc

   1 // Protocol Buffers - Google's data interchange format
   2 // Copyright 2008 Google Inc.  All rights reserved.
   3 // http://code.google.com/p/protobuf/
   4 //
   5 // Redistribution and use in source and binary forms, with or without
   6 // modification, are permitted provided that the following conditions are
   7 // met:
   8 //
   9 //     * Redistributions of source code must retain the above copyright
  10 // notice, this list of conditions and the following disclaimer.
  11 //     * Redistributions in binary form must reproduce the above
  12 // copyright notice, this list of conditions and the following disclaimer
  13 // in the documentation and/or other materials provided with the
  14 // distribution.
  15 //     * Neither the name of Google Inc. nor the names of its
  16 // contributors may be used to endorse or promote products derived from
  17 // this software without specific prior written permission.
  18 //
  19 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  20 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  21 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  22 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  23 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  24 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  25 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  26 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  27 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  28 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  29 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  30
  31 // Author: kenton@google.com (Kenton Varda)
  32
  33 #include <google/protobuf/stubs/substitute.h>
  34 #include <google/protobuf/stubs/strutil.h>
  35 #include <google/protobuf/stubs/stl_util.h>
  36
  37 namespace google {
  38 namespace protobuf {
  39 namespace strings {
  40
  41 using internal::SubstituteArg;
  42
  43 // Returns the number of args in arg_array which were passed explicitly
  44 // to Substitute().
  45 static int CountSubstituteArgs(const SubstituteArg* const* args_array) {
  46   int count = 0;
  47   while (args_array[count] != NULL && args_array[count]->size() != -1) {
  48     ++count;
  49   }
  50   return count;
  51 }
  52
  53 string Substitute(
  54     const char* format,
  55     const SubstituteArg& arg0, const SubstituteArg& arg1,
  56     const SubstituteArg& arg2, const SubstituteArg& arg3,
  57     const SubstituteArg& arg4, const SubstituteArg& arg5,
  58     const SubstituteArg& arg6, const SubstituteArg& arg7,
  59     const SubstituteArg& arg8, const SubstituteArg& arg9) {
  60   string result;
  61   SubstituteAndAppend(&result, format, arg0, arg1, arg2, arg3, arg4,
  62                                        arg5, arg6, arg7, arg8, arg9);
  63   return result;
  64 }
  65
  66 void SubstituteAndAppend(
  67     string* output, const char* format,
  68     const SubstituteArg& arg0, const SubstituteArg& arg1,
  69     const SubstituteArg& arg2, const SubstituteArg& arg3,
  70     const SubstituteArg& arg4, const SubstituteArg& arg5,
  71     const SubstituteArg& arg6, const SubstituteArg& arg7,
  72     const SubstituteArg& arg8, const SubstituteArg& arg9) {
  73   const SubstituteArg* const args_array[] = {
  74     &arg0, &arg1, &arg2, &arg3, &arg4, &arg5, &arg6, &arg7, &arg8, &arg9, NULL
  75   };
  76
  77   // Determine total size needed.
  78   int size = 0;
  79   for (int i = 0; format[i] != '\0'; i++) {
  80     if (format[i] == '$') {
  81       if (ascii_isdigit(format[i+1])) {
  82         int index = format[i+1] - '0';
  83         if (args_array[index]->size() == -1) {
  84           GOOGLE_LOG(DFATAL)
  85             << "strings::Substitute format string invalid: asked for \"$"
  86             << index << "\", but only " << CountSubstituteArgs(args_array)
  87             << " args were given.  Full format string was: \""
  88             << CEscape(format) << "\".";
  89           return;
  90         }
  91         size += args_array[index]->size();
  92         ++i;  // Skip next char.
  93       } else if (format[i+1] == '$') {
  94         ++size;
  95         ++i;  // Skip next char.
  96       } else {
  97         GOOGLE_LOG(DFATAL)
  98           << "Invalid strings::Substitute() format string: \""
  99           << CEscape(format) << "\".";
 100         return;
 101       }
 102     } else {
 103       ++size;
 104     }
 105   }
 106
 107   if (size == 0) return;
 108
 109   // Build the string.
 110   int original_size = output->size();
 111   STLStringResizeUninitialized(output, original_size + size);
 112   char* target = string_as_array(output) + original_size;
 113   for (int i = 0; format[i] != '\0'; i++) {
 114     if (format[i] == '$') {
 115       if (ascii_isdigit(format[i+1])) {
 116         const SubstituteArg* src = args_array[format[i+1] - '0'];
 117         memcpy(target, src->data(), src->size());
 118         target += src->size();
 119         ++i;  // Skip next char.
 120       } else if (format[i+1] == '$') {
 121         *target++ = '$';
 122         ++i;  // Skip next char.
 123       }
 124     } else {
 125       *target++ = format[i];
 126     }
 127   }
 128
 129   GOOGLE_DCHECK_EQ(target - output->data(), output->size());
 130 }
 131
 132 }  // namespace strings
 133 }  // namespace protobuf
 134 }  // namespace google