add WebSocket.Headers

websocket-sharp/Handshake.cs

@@ -65,6 +65,11 @@ public void AddHeader(string name, string value)
       Headers.Add(name, value);
     }
 
+    public void AddHeaders(WebHeaderCollection headers)
+    {
+      Headers.Add(headers);
+    }
+
     public string[] GetHeaderValues(string name)
     {
       return Headers.GetValues(name);

websocket-sharp/WebSocket.cs

@@ -64,22 +64,23 @@ public class WebSocket : IDisposable
 
     #region Private Fields
 
-    private string           _base64key;
-    private bool             _client;
-    private Action           _closeContext;
-    private WebSocketContext _context;
-    private string           _extensions;
-    private AutoResetEvent   _exitMessageLoop;
-    private Object           _forClose;
-    private Object           _forSend;
-    private string           _protocol;
-    private string           _protocols;
-    private volatile WsState _readyState;
-    private AutoResetEvent   _receivePong;
-    private bool             _secure;
-    private TcpClient        _tcpClient;
-    private Uri              _uri;
-    private WsStream         _wsStream;
+    private string              _base64key;
+    private bool                _client;
+    private Action              _closeContext;
+    private WebSocketContext    _context;
+    private string              _extensions;
+    private AutoResetEvent      _exitMessageLoop;
+    private Object              _forClose;
+    private Object              _forSend;
+    private string              _protocol;
+    private string              _protocols;
+    private volatile WsState    _readyState;
+    private AutoResetEvent      _receivePong;
+    private bool                _secure;
+    private TcpClient           _tcpClient;
+    private Uri                 _uri;
+    private WebHeaderCollection _headers;
+    private WsStream            _wsStream;
 
     #endregion
 
@@ -92,6 +93,7 @@ private WebSocket()
       _forSend    = new Object();
       _protocol   = String.Empty;
       _readyState = WsState.CONNECTING;
+      _headers    = new WebHeaderCollection();
     }
 
     #endregion
@@ -289,6 +291,12 @@ internal set {
       }
     }
 
+	public WebHeaderCollection Headers {
+      get {
+        return _headers;
+      }
+	}
+
     #endregion
 
     #region Events
@@ -578,6 +586,7 @@ private RequestHandshake createOpeningHandshake()
       if (!_protocols.IsNullOrEmpty())
         req.AddHeader("Sec-WebSocket-Protocol", _protocols);
       req.AddHeader("Sec-WebSocket-Version", _version);
+      req.AddHeaders(_headers);
 
       return req;
     }
@@ -587,6 +596,7 @@ private ResponseHandshake createResponseHandshake()
     {
       var res = new ResponseHandshake();
       res.AddHeader("Sec-WebSocket-Accept", createResponseKey());
+      res.AddHeaders(_headers);
 
       return res;
     }
@@ -596,6 +606,7 @@ private ResponseHandshake createResponseHandshake(HttpStatusCode code)
     {
       var res = ResponseHandshake.CreateCloseResponse(code);
       res.AddHeader("Sec-WebSocket-Version", _version);
+      res.AddHeaders(_headers);
 
       return res;
     }