|
1 // Copyright (c) 2010, Google Inc. |
|
2 // All rights reserved. |
|
3 // |
|
4 // Redistribution and use in source and binary forms, with or without |
|
5 // modification, are permitted provided that the following conditions are |
|
6 // met: |
|
7 // |
|
8 // * Redistributions of source code must retain the above copyright |
|
9 // notice, this list of conditions and the following disclaimer. |
|
10 // * Redistributions in binary form must reproduce the above |
|
11 // copyright notice, this list of conditions and the following disclaimer |
|
12 // in the documentation and/or other materials provided with the |
|
13 // distribution. |
|
14 // * Neither the name of Google Inc. nor the names of its |
|
15 // contributors may be used to endorse or promote products derived from |
|
16 // this software without specific prior written permission. |
|
17 // |
|
18 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
|
19 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
|
20 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
|
21 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
|
22 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
|
23 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
|
24 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
|
25 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
|
26 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
|
27 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
|
28 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|
29 |
|
30 #include <arpa/inet.h> |
|
31 #include <limits.h> |
|
32 |
|
33 #include <string> |
|
34 #include <vector> |
|
35 |
|
36 #include "common/using_std_string.h" |
|
37 #include "processor/binarystream.h" |
|
38 |
|
39 namespace google_breakpad { |
|
40 using std::vector; |
|
41 |
|
42 binarystream &binarystream::operator>>(string &str) { |
|
43 uint16_t length; |
|
44 *this >> length; |
|
45 if (eof()) |
|
46 return *this; |
|
47 if (length == 0) { |
|
48 str.clear(); |
|
49 return *this; |
|
50 } |
|
51 vector<char> buffer(length); |
|
52 stream_.read(&buffer[0], length); |
|
53 if (!eof()) |
|
54 str.assign(&buffer[0], length); |
|
55 return *this; |
|
56 } |
|
57 |
|
58 binarystream &binarystream::operator>>(uint8_t &u8) { |
|
59 stream_.read((char *)&u8, 1); |
|
60 return *this; |
|
61 } |
|
62 |
|
63 binarystream &binarystream::operator>>(uint16_t &u16) { |
|
64 uint16_t temp; |
|
65 stream_.read((char *)&temp, 2); |
|
66 if (!eof()) |
|
67 u16 = ntohs(temp); |
|
68 return *this; |
|
69 } |
|
70 |
|
71 binarystream &binarystream::operator>>(uint32_t &u32) { |
|
72 uint32_t temp; |
|
73 stream_.read((char *)&temp, 4); |
|
74 if (!eof()) |
|
75 u32 = ntohl(temp); |
|
76 return *this; |
|
77 } |
|
78 |
|
79 binarystream &binarystream::operator>>(uint64_t &u64) { |
|
80 uint32_t lower, upper; |
|
81 *this >> lower >> upper; |
|
82 if (!eof()) |
|
83 u64 = static_cast<uint64_t>(lower) | (static_cast<uint64_t>(upper) << 32); |
|
84 return *this; |
|
85 } |
|
86 |
|
87 binarystream &binarystream::operator<<(const string &str) { |
|
88 if (str.length() > USHRT_MAX) { |
|
89 // truncate to 16-bit length |
|
90 *this << static_cast<uint16_t>(USHRT_MAX); |
|
91 stream_.write(str.c_str(), USHRT_MAX); |
|
92 } else { |
|
93 *this << (uint16_t)(str.length() & 0xFFFF); |
|
94 stream_.write(str.c_str(), str.length()); |
|
95 } |
|
96 return *this; |
|
97 } |
|
98 |
|
99 binarystream &binarystream::operator<<(uint8_t u8) { |
|
100 stream_.write((const char*)&u8, 1); |
|
101 return *this; |
|
102 } |
|
103 |
|
104 binarystream &binarystream::operator<<(uint16_t u16) { |
|
105 u16 = htons(u16); |
|
106 stream_.write((const char*)&u16, 2); |
|
107 return *this; |
|
108 } |
|
109 |
|
110 binarystream &binarystream::operator<<(uint32_t u32) { |
|
111 u32 = htonl(u32); |
|
112 stream_.write((const char*)&u32, 4); |
|
113 return *this; |
|
114 } |
|
115 |
|
116 binarystream &binarystream::operator<<(uint64_t u64) { |
|
117 // write 64-bit ints as two 32-bit ints, so we can byte-swap them easily |
|
118 uint32_t lower = static_cast<uint32_t>(u64 & 0xFFFFFFFF); |
|
119 uint32_t upper = static_cast<uint32_t>(u64 >> 32); |
|
120 *this << lower << upper; |
|
121 return *this; |
|
122 } |
|
123 |
|
124 } // namespace google_breakpad |