coded_stream_inl.h 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990
  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. // Author: jasonh@google.com (Jason Hsueh)
  31. //
  32. // Implements methods of coded_stream.h that need to be inlined for performance
  33. // reasons, but should not be defined in a public header.
  34. #ifndef GOOGLE_PROTOBUF_IO_CODED_STREAM_INL_H__
  35. #define GOOGLE_PROTOBUF_IO_CODED_STREAM_INL_H__
  36. #include <google/protobuf/stubs/logging.h>
  37. #include <google/protobuf/stubs/common.h>
  38. #include <google/protobuf/io/coded_stream.h>
  39. #include <google/protobuf/io/zero_copy_stream_impl_lite.h>
  40. #include <string>
  41. #include <google/protobuf/stubs/stl_util.h>
  42. namespace google {
  43. namespace protobuf {
  44. namespace io {
  45. inline bool CodedInputStream::InternalReadStringInline(string* buffer,
  46. int size) {
  47. if (size < 0) return false; // security: size is often user-supplied
  48. if (BufferSize() >= size) {
  49. STLStringResizeUninitialized(buffer, size);
  50. std::pair<char*, bool> z = as_string_data(buffer);
  51. if (z.second) {
  52. // Oddly enough, memcpy() requires its first two args to be non-NULL even
  53. // if we copy 0 bytes. So, we have ensured that z.first is non-NULL here.
  54. GOOGLE_DCHECK(z.first != NULL);
  55. memcpy(z.first, buffer_, size);
  56. Advance(size);
  57. }
  58. return true;
  59. }
  60. return ReadStringFallback(buffer, size);
  61. }
  62. inline bool CodedInputStream::InternalReadRawInline(void* buffer, int size) {
  63. int current_buffer_size;
  64. while ((current_buffer_size = BufferSize()) < size) {
  65. // Reading past end of buffer. Copy what we have, then refresh.
  66. memcpy(buffer, buffer_, current_buffer_size);
  67. buffer = reinterpret_cast<uint8*>(buffer) + current_buffer_size;
  68. size -= current_buffer_size;
  69. Advance(current_buffer_size);
  70. if (!Refresh()) return false;
  71. }
  72. memcpy(buffer, buffer_, size);
  73. Advance(size);
  74. return true;
  75. }
  76. } // namespace io
  77. } // namespace protobuf
  78. } // namespace google
  79. #endif // GOOGLE_PROTOBUF_IO_CODED_STREAM_INL_H__