diff options
Diffstat (limited to 'includes')
| -rw-r--r-- | includes/BWebSocket.h | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/includes/BWebSocket.h b/includes/BWebSocket.h index 49f8cd1..cf25684 100644 --- a/includes/BWebSocket.h +++ b/includes/BWebSocket.h @@ -30,17 +30,19 @@ class WebSocketWorker : public StacklessTask { public: virtual bool parse(Http::Request & req) { return true; } void sendFrame(WebSocketFrame * frame) { m_sendQueue.push(frame); } + void enforceServer(void) throw (GeneralException); + void enforceClient(void) throw (GeneralException); protected: WebSocketWorker(IO<Handle> socket, const String & url) : m_socket(new BStream(socket)) { m_name = String("WebSocket:") + url + ":" + m_socket->getName(); } ~WebSocketWorker(); void disconnect() { m_socket->close(); } virtual void receiveMessage(const uint8_t * msg, size_t len, bool binary) = 0; - private: + virtual void Do(); +private: void processMessage(); void processPing(); void processPong(); - const char * getName() const { return m_name.to_charp(); } - void Do(); + virtual const char * getName() const { return m_name.to_charp(); } String m_name; IO<BStream> m_socket; enum { |