OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "net/websockets/websocket_frame.h" | 5 #include "net/websockets/websocket_frame.h" |
6 | 6 |
7 #include <stddef.h> | 7 #include <stddef.h> |
8 #include <stdint.h> | 8 #include <stdint.h> |
9 | 9 |
10 #include <algorithm> | 10 #include <algorithm> |
(...skipping 17 matching lines...) Expand all Loading... |
28 | 28 |
29 using PackedMaskType = uint32_t __attribute__((vector_size(16))); | 29 using PackedMaskType = uint32_t __attribute__((vector_size(16))); |
30 | 30 |
31 #else | 31 #else |
32 | 32 |
33 using PackedMaskType = size_t; | 33 using PackedMaskType = size_t; |
34 | 34 |
35 #endif // defined(COMPILER_GCC) && defined(ARCH_CPU_X86_FAMILY) && | 35 #endif // defined(COMPILER_GCC) && defined(ARCH_CPU_X86_FAMILY) && |
36 // !defined(OS_NACL) | 36 // !defined(OS_NACL) |
37 | 37 |
38 const uint8 kFinalBit = 0x80; | 38 const uint8_t kFinalBit = 0x80; |
39 const uint8 kReserved1Bit = 0x40; | 39 const uint8_t kReserved1Bit = 0x40; |
40 const uint8 kReserved2Bit = 0x20; | 40 const uint8_t kReserved2Bit = 0x20; |
41 const uint8 kReserved3Bit = 0x10; | 41 const uint8_t kReserved3Bit = 0x10; |
42 const uint8 kOpCodeMask = 0xF; | 42 const uint8_t kOpCodeMask = 0xF; |
43 const uint8 kMaskBit = 0x80; | 43 const uint8_t kMaskBit = 0x80; |
44 const uint64 kMaxPayloadLengthWithoutExtendedLengthField = 125; | 44 const uint64_t kMaxPayloadLengthWithoutExtendedLengthField = 125; |
45 const uint64 kPayloadLengthWithTwoByteExtendedLengthField = 126; | 45 const uint64_t kPayloadLengthWithTwoByteExtendedLengthField = 126; |
46 const uint64 kPayloadLengthWithEightByteExtendedLengthField = 127; | 46 const uint64_t kPayloadLengthWithEightByteExtendedLengthField = 127; |
47 | 47 |
48 inline void MaskWebSocketFramePayloadByBytes( | 48 inline void MaskWebSocketFramePayloadByBytes( |
49 const WebSocketMaskingKey& masking_key, | 49 const WebSocketMaskingKey& masking_key, |
50 size_t masking_key_offset, | 50 size_t masking_key_offset, |
51 char* const begin, | 51 char* const begin, |
52 char* const end) { | 52 char* const end) { |
53 for (char* masked = begin; masked != end; ++masked) { | 53 for (char* masked = begin; masked != end; ++masked) { |
54 *masked ^= masking_key.key[masking_key_offset++]; | 54 *masked ^= masking_key.key[masking_key_offset++]; |
55 if (masking_key_offset == WebSocketFrameHeader::kMaskingKeyLength) | 55 if (masking_key_offset == WebSocketFrameHeader::kMaskingKeyLength) |
56 masking_key_offset = 0; | 56 masking_key_offset = 0; |
(...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
96 return (WebSocketFrameHeader::kBaseHeaderSize + extended_length_size + | 96 return (WebSocketFrameHeader::kBaseHeaderSize + extended_length_size + |
97 (header.masked ? WebSocketFrameHeader::kMaskingKeyLength : 0)); | 97 (header.masked ? WebSocketFrameHeader::kMaskingKeyLength : 0)); |
98 } | 98 } |
99 | 99 |
100 int WriteWebSocketFrameHeader(const WebSocketFrameHeader& header, | 100 int WriteWebSocketFrameHeader(const WebSocketFrameHeader& header, |
101 const WebSocketMaskingKey* masking_key, | 101 const WebSocketMaskingKey* masking_key, |
102 char* buffer, | 102 char* buffer, |
103 int buffer_size) { | 103 int buffer_size) { |
104 DCHECK((header.opcode & kOpCodeMask) == header.opcode) | 104 DCHECK((header.opcode & kOpCodeMask) == header.opcode) |
105 << "header.opcode must fit to kOpCodeMask."; | 105 << "header.opcode must fit to kOpCodeMask."; |
106 DCHECK(header.payload_length <= static_cast<uint64>(kint64max)) | 106 DCHECK(header.payload_length <= static_cast<uint64_t>(kint64max)) |
107 << "WebSocket specification doesn't allow a frame longer than " | 107 << "WebSocket specification doesn't allow a frame longer than " |
108 << "kint64max (0x7FFFFFFFFFFFFFFF) bytes."; | 108 << "kint64max (0x7FFFFFFFFFFFFFFF) bytes."; |
109 DCHECK_GE(buffer_size, 0); | 109 DCHECK_GE(buffer_size, 0); |
110 | 110 |
111 // WebSocket frame format is as follows: | 111 // WebSocket frame format is as follows: |
112 // - Common header (2 bytes) | 112 // - Common header (2 bytes) |
113 // - Optional extended payload length | 113 // - Optional extended payload length |
114 // (2 or 8 bytes, present if actual payload length is more than 125 bytes) | 114 // (2 or 8 bytes, present if actual payload length is more than 125 bytes) |
115 // - Optional masking key (4 bytes, present if MASK bit is on) | 115 // - Optional masking key (4 bytes, present if MASK bit is on) |
116 // - Actual payload (XOR masked with masking key if MASK bit is on) | 116 // - Actual payload (XOR masked with masking key if MASK bit is on) |
117 // | 117 // |
118 // This function constructs frame header (the first three in the list | 118 // This function constructs frame header (the first three in the list |
119 // above). | 119 // above). |
120 | 120 |
121 int header_size = GetWebSocketFrameHeaderSize(header); | 121 int header_size = GetWebSocketFrameHeaderSize(header); |
122 if (header_size > buffer_size) | 122 if (header_size > buffer_size) |
123 return ERR_INVALID_ARGUMENT; | 123 return ERR_INVALID_ARGUMENT; |
124 | 124 |
125 int buffer_index = 0; | 125 int buffer_index = 0; |
126 | 126 |
127 uint8 first_byte = 0u; | 127 uint8_t first_byte = 0u; |
128 first_byte |= header.final ? kFinalBit : 0u; | 128 first_byte |= header.final ? kFinalBit : 0u; |
129 first_byte |= header.reserved1 ? kReserved1Bit : 0u; | 129 first_byte |= header.reserved1 ? kReserved1Bit : 0u; |
130 first_byte |= header.reserved2 ? kReserved2Bit : 0u; | 130 first_byte |= header.reserved2 ? kReserved2Bit : 0u; |
131 first_byte |= header.reserved3 ? kReserved3Bit : 0u; | 131 first_byte |= header.reserved3 ? kReserved3Bit : 0u; |
132 first_byte |= header.opcode & kOpCodeMask; | 132 first_byte |= header.opcode & kOpCodeMask; |
133 buffer[buffer_index++] = first_byte; | 133 buffer[buffer_index++] = first_byte; |
134 | 134 |
135 int extended_length_size = 0; | 135 int extended_length_size = 0; |
136 uint8 second_byte = 0u; | 136 uint8_t second_byte = 0u; |
137 second_byte |= header.masked ? kMaskBit : 0u; | 137 second_byte |= header.masked ? kMaskBit : 0u; |
138 if (header.payload_length <= kMaxPayloadLengthWithoutExtendedLengthField) { | 138 if (header.payload_length <= kMaxPayloadLengthWithoutExtendedLengthField) { |
139 second_byte |= header.payload_length; | 139 second_byte |= header.payload_length; |
140 } else if (header.payload_length <= kuint16max) { | 140 } else if (header.payload_length <= kuint16max) { |
141 second_byte |= kPayloadLengthWithTwoByteExtendedLengthField; | 141 second_byte |= kPayloadLengthWithTwoByteExtendedLengthField; |
142 extended_length_size = 2; | 142 extended_length_size = 2; |
143 } else { | 143 } else { |
144 second_byte |= kPayloadLengthWithEightByteExtendedLengthField; | 144 second_byte |= kPayloadLengthWithEightByteExtendedLengthField; |
145 extended_length_size = 8; | 145 extended_length_size = 8; |
146 } | 146 } |
147 buffer[buffer_index++] = second_byte; | 147 buffer[buffer_index++] = second_byte; |
148 | 148 |
149 // Writes "extended payload length" field. | 149 // Writes "extended payload length" field. |
150 if (extended_length_size == 2) { | 150 if (extended_length_size == 2) { |
151 uint16 payload_length_16 = static_cast<uint16>(header.payload_length); | 151 uint16_t payload_length_16 = static_cast<uint16_t>(header.payload_length); |
152 base::WriteBigEndian(buffer + buffer_index, payload_length_16); | 152 base::WriteBigEndian(buffer + buffer_index, payload_length_16); |
153 buffer_index += sizeof(payload_length_16); | 153 buffer_index += sizeof(payload_length_16); |
154 } else if (extended_length_size == 8) { | 154 } else if (extended_length_size == 8) { |
155 base::WriteBigEndian(buffer + buffer_index, header.payload_length); | 155 base::WriteBigEndian(buffer + buffer_index, header.payload_length); |
156 buffer_index += sizeof(header.payload_length); | 156 buffer_index += sizeof(header.payload_length); |
157 } | 157 } |
158 | 158 |
159 // Writes "masking key" field, if needed. | 159 // Writes "masking key" field, if needed. |
160 if (header.masked) { | 160 if (header.masked) { |
161 DCHECK(masking_key); | 161 DCHECK(masking_key); |
(...skipping 12 matching lines...) Expand all Loading... |
174 WebSocketMaskingKey GenerateWebSocketMaskingKey() { | 174 WebSocketMaskingKey GenerateWebSocketMaskingKey() { |
175 // Masking keys should be generated from a cryptographically secure random | 175 // Masking keys should be generated from a cryptographically secure random |
176 // number generator, which means web application authors should not be able | 176 // number generator, which means web application authors should not be able |
177 // to guess the next value of masking key. | 177 // to guess the next value of masking key. |
178 WebSocketMaskingKey masking_key; | 178 WebSocketMaskingKey masking_key; |
179 base::RandBytes(masking_key.key, WebSocketFrameHeader::kMaskingKeyLength); | 179 base::RandBytes(masking_key.key, WebSocketFrameHeader::kMaskingKeyLength); |
180 return masking_key; | 180 return masking_key; |
181 } | 181 } |
182 | 182 |
183 void MaskWebSocketFramePayload(const WebSocketMaskingKey& masking_key, | 183 void MaskWebSocketFramePayload(const WebSocketMaskingKey& masking_key, |
184 uint64 frame_offset, | 184 uint64_t frame_offset, |
185 char* const data, | 185 char* const data, |
186 int data_size) { | 186 int data_size) { |
187 static const size_t kMaskingKeyLength = | 187 static const size_t kMaskingKeyLength = |
188 WebSocketFrameHeader::kMaskingKeyLength; | 188 WebSocketFrameHeader::kMaskingKeyLength; |
189 | 189 |
190 DCHECK_GE(data_size, 0); | 190 DCHECK_GE(data_size, 0); |
191 | 191 |
192 // Most of the masking is done in chunks of sizeof(PackedMaskType), except for | 192 // Most of the masking is done in chunks of sizeof(PackedMaskType), except for |
193 // the beginning and the end of the buffer which may be unaligned. | 193 // the beginning and the end of the buffer which may be unaligned. |
194 // PackedMaskType must be a multiple of kMaskingKeyLength in size. | 194 // PackedMaskType must be a multiple of kMaskingKeyLength in size. |
(...skipping 51 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
246 } | 246 } |
247 | 247 |
248 MaskWebSocketFramePayloadByBytes( | 248 MaskWebSocketFramePayloadByBytes( |
249 masking_key, | 249 masking_key, |
250 (frame_offset + (aligned_end - data)) % kMaskingKeyLength, | 250 (frame_offset + (aligned_end - data)) % kMaskingKeyLength, |
251 aligned_end, | 251 aligned_end, |
252 end); | 252 end); |
253 } | 253 } |
254 | 254 |
255 } // namespace net | 255 } // namespace net |
OLD | NEW |