1 files changed, 23 insertions, 6 deletions

diff --git a/src/unstable/protocolwebsocket.cpp b/src/unstable/protocolwebsocket.cpp
index 8d4a124..08ac176 100644
--- a/src/unstable/protocolwebsocket.cpp
+++ b/src/unstable/protocolwebsocket.cpp
@@ -66,6 +66,11 @@ void Bu::ProtocolWebSocket::onNewData( Bu::Client * /*pClient*/ )
     }
 }
 
+bool onProcessHeaders( Bu::StringList & /*lHeadersOut*/ )
+{
+    return true;
+}
+
 void Bu::ProtocolWebSocket::writeMessage( const Bu::String &sData,
         Bu::ProtocolWebSocket::Operation eOp )
 {
@@ -237,12 +242,24 @@ bool Bu::ProtocolWebSocket::processHeaders()
 
     DEBUG( Bu::println("accept: %1").arg( mbOut.getString() ) );
 
-    pClient->write("HTTP/1.1 101 Switching Protocols\r\n"
-        "Upgrade: websocket\r\n"
-        "Connection: Upgrade\r\n"
-        "Sec-WebSocket-Accept: " + mbOut.getString() + "\r\n"
-        "\r\n"
-        );
+    Bu::StringList lHeadersOut;
+    lHeadersOut.append("Upgrade: websocket");
+    lHeadersOut.append("Connection: Upgrade");
+    lHeadersOut.append("Sec-WebSocket-Accept: " + mbOut.getString());
+
+    if( !onProcessHeaders( lHeadersOut ) )
+    {
+        pClient->disconnect();
+        return false;
+    }
+
+    Bu::String sHeaderBlock("HTTP/1.1 101 Switching Protocols\r\n");
+    for( Bu::StringList::iterator i = lHeadersOut.begin(); i; i++ )
+    {
+        sHeaderBlock += (*i) + "\r\n";
+    }
+    sHeaderBlock += "\r\n";
+    pClient->write( sHeaderBlock );
 
     DEBUG( Bu::println("websocket: Switching protocols.") );