Home | History | Annotate | Download | only in io
      1 // Protocol Buffers - Google's data interchange format
      2 // Copyright 2008 Google Inc.  All rights reserved.
      3 // https://developers.google.com/protocol-buffers/
      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: jasonh (at) google.com (Jason Hsueh)
     32 //
     33 // Implements methods of coded_stream.h that need to be inlined for performance
     34 // reasons, but should not be defined in a public header.
     35 
     36 #ifndef GOOGLE_PROTOBUF_IO_CODED_STREAM_INL_H__
     37 #define GOOGLE_PROTOBUF_IO_CODED_STREAM_INL_H__
     38 
     39 #include <google/protobuf/stubs/logging.h>
     40 #include <google/protobuf/stubs/common.h>
     41 #include <google/protobuf/io/coded_stream.h>
     42 #include <google/protobuf/io/zero_copy_stream_impl_lite.h>
     43 #include <string>
     44 #include <google/protobuf/stubs/stl_util.h>
     45 
     46 namespace google {
     47 namespace protobuf {
     48 namespace io {
     49 
     50 inline bool CodedInputStream::InternalReadStringInline(string* buffer,
     51                                                        int size) {
     52   if (size < 0) return false;  // security: size is often user-supplied
     53 
     54   if (BufferSize() >= size) {
     55     STLStringResizeUninitialized(buffer, size);
     56     std::pair<char*, bool> z = as_string_data(buffer);
     57     if (z.second) {
     58       // Oddly enough, memcpy() requires its first two args to be non-NULL even
     59       // if we copy 0 bytes.  So, we have ensured that z.first is non-NULL here.
     60       GOOGLE_DCHECK(z.first != NULL);
     61       memcpy(z.first, buffer_, size);
     62       Advance(size);
     63     }
     64     return true;
     65   }
     66 
     67   return ReadStringFallback(buffer, size);
     68 }
     69 
     70 inline bool CodedInputStream::InternalReadRawInline(void* buffer, int size) {
     71   int current_buffer_size;
     72   while ((current_buffer_size = BufferSize()) < size) {
     73     // Reading past end of buffer.  Copy what we have, then refresh.
     74     memcpy(buffer, buffer_, current_buffer_size);
     75     buffer = reinterpret_cast<uint8*>(buffer) + current_buffer_size;
     76     size -= current_buffer_size;
     77     Advance(current_buffer_size);
     78     if (!Refresh()) return false;
     79   }
     80 
     81   memcpy(buffer, buffer_, size);
     82   Advance(size);
     83 
     84   return true;
     85 }
     86 
     87 }  // namespace io
     88 }  // namespace protobuf
     89 }  // namespace google
     90 #endif  // GOOGLE_PROTOBUF_IO_CODED_STREAM_INL_H__
     91