Added handler to check/modify http headers.

This is an extra call that happens during the http handshake that allows
the implementation to set additional headers, modify existing headers,
and cancel the whole connection if desired.

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.") );