Add functions used for building WebSocket frame data.

net/websockets/websocket_frame.h

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef NET_WEBSOCKETS_WEBSOCKET_FRAME_H_
 #define NET_WEBSOCKETS_WEBSOCKET_FRAME_H_
 #pragma once
 
 #include <vector>
 
@@ skipping 42 matching lines @@
 // series of chunk objects of a WebSocket frame.
 //
 // Frame dissection is necessary to handle WebSocket frame stream containing
 // abritrarily large frames in the browser process. Because the server may send
 // a huge frame that doesn't fit in the memory, we cannot store the entire
 // payload data in the memory.
 //
 // Users of this struct should treat WebSocket frames as a data stream; it's
 // important to keep the frame data flowing, especially in the browser process.
 // Users should not let the data stuck somewhere in the pipeline.
+//
+// This struct is used for reading WebSocket frame data (created by
+// WebSocketFrameParser). To construct WebSocket frames, use functions below.
 struct NET_EXPORT_PRIVATE WebSocketFrameChunk {
   WebSocketFrameChunk();
   ~WebSocketFrameChunk();
 
   // Non-null |header| is provided only if this chunk is the first part of
   // a series of chunks.
   scoped_ptr<WebSocketFrameHeader> header;
 
   // Indicates this part is the last chunk of a frame.
   bool final_chunk;
 
   // |data| is always unmasked even if the frame is masked.
   std::vector<char> data;
 };
 
+// Contains four-byte data representing "masking key" of WebSocket frames.
+struct WebSocketMaskingKey {
+  char key[WebSocketFrameHeader::kMaskingKeyLength];
+};

[mmenke, 2012/05/22 17:12:41]

Think this should also be used in net/websockets/websocket_frame_parser.  Should
be a simple enough change that you can do it in this CL, though if you prefer to
leave it for later, not something I'm going to push back on.

[Yuta Kitamura, 2012/06/06 18:06:48]

Yeah I'm going to do that later.

+
+// Writes wire format of a WebSocket frame header into |output|.
+//
+// WebSocket frame format is defined at:
+// <http://tools.ietf.org/html/rfc6455#section-5.2>. This function writes
+// everything but payload data in a WebSocket frame to |output|.
+//
+// If |header->masked| is true, |masking_key| must point to a valid
+// WebSocketMaskingKey object containing the masking key for that frame
+// (possibly generated by GenerateWebSocketMaskingKey() function below).
+// Otherwise, |masking_key| must be NULL.
+NET_EXPORT_PRIVATE void WriteWebSocketFrameHeader(
+    const WebSocketFrameHeader& header,
+    const WebSocketMaskingKey* masking_key,
+    std::vector<char>* output);
+
+// Generates a masking key suitable for use in a new WebSocket frame.
+NET_EXPORT_PRIVATE WebSocketMaskingKey GenerateWebSocketMaskingKey();
+
+// Masks WebSocket frame payload.
+//
+// A client must mask every WebSocket frame by XOR'ing the frame payload
+// with four-byte random data (masking key). This function applies the masking
+// to the given payload data.
+//
+// This function masks |frame_data| with |masking_key|, assuming |frame_data|
+// is partial data starting from |frame_offset| bytes from the beginning of
+// the payload data.
+NET_EXPORT_PRIVATE void MaskWebSocketFramePayload(
+    const WebSocketMaskingKey& masking_key,
+    uint64 frame_offset,
+    std::vector<char>* frame_data);
+
 }  // namespace net
 
 #endif  // NET_WEBSOCKETS_WEBSOCKET_FRAME_H_