summaryrefslogtreecommitdiff
path: root/src/unstable/protocolwebsocket.h
blob: 4a3d0a1e691c6b6c29794c5d2f66260f11adfcad (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
/*
 * Copyright (C) 2007-2014 Xagasoft, All rights reserved.
 *
 * This file is part of the libbu++ library and is released under the
 * terms of the license contained in the file LICENSE.
 */

#ifndef BU_PROTOCOL_WEB_SOCKET_H
#define BU_PROTOCOL_WEB_SOCKET_H

#include "bu/protocol.h"
#include "bu/hash.h"

#include "bu/mutex.h"

namespace Bu
{
    class ProtocolWebSocket : public Bu::Protocol
    {
    public:
        enum Operation
        {
            Continuation = 0,
            Text         = 1,
            Binary       = 2,
            ConClose     = 8,
            Ping         = 9,
            Pong         = 10
        };

    public:
        ProtocolWebSocket();
        virtual ~ProtocolWebSocket();

        virtual void onNewConnection( Bu::Client *pClient );
        virtual void onNewData( Bu::Client *pClient );

        virtual void onHandshakeComplete()=0;
        virtual void onNewMessage( const Bu::String &sData, Operation eOp )=0;
        void writeMessage( const Bu::String &sData, Operation eOp=Text );

    private:
        bool stateProtoId();
        bool stateHandshake();
        bool processHeaders();
        bool headerMatch( const Bu::String &sKey, const Bu::String &sValue );
        bool readHttpHdrLine( Bu::String &sLine );
        bool parseMessage();

    private:
        enum Status
        {
            stProtoId       = 0,
            stHandshake     = 1,
            stReady         = 10
        };

    protected:
        Bu::Client *pClient;
        Status eStatus;
        Bu::String sPath;
        Bu::Hash<Bu::String, Bu::StringList> hHeader;
        Bu::Mutex mClient;
    };
}

#endif