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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
/*
* Copyright (C) 2011-2013 Team XBMC
* http://xbmc.org
*
* This Program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2, or (at your option)
* any later version.
*
* This Program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with XBMC; see the file COPYING. If not, see
* <http://www.gnu.org/licenses/>.
*
*/
#include <string>
#include "WebSocketManager.h"
#include "WebSocket.h"
#include "WebSocketV8.h"
#include "WebSocketV13.h"
#include "utils/HttpParser.h"
#include "utils/HttpResponse.h"
#include "utils/log.h"
#define WS_HTTP_METHOD "GET"
#define WS_HTTP_TAG "HTTP/"
#define WS_SUPPORTED_VERSIONS "8, 13"
#define WS_HEADER_VERSION "Sec-WebSocket-Version"
#define WS_HEADER_VERSION_LC "sec-websocket-version" // "Sec-WebSocket-Version"
using namespace std;
CWebSocket* CWebSocketManager::Handle(const char* data, unsigned int length, string &response)
{
if (data == NULL || length <= 0)
return NULL;
HttpParser header;
HttpParser::status_t status = header.addBytes(data, length);
switch (status)
{
case HttpParser::Error:
case HttpParser::Incomplete:
response.clear();
return NULL;
case HttpParser::Done:
default:
break;
}
// There must be a "Sec-WebSocket-Version" header
const char* value = header.getValue(WS_HEADER_VERSION_LC);
if (value == NULL)
{
CLog::Log(LOGINFO, "WebSocket: missing Sec-WebSocket-Version");
CHttpResponse httpResponse(HTTP::Get, HTTP::BadRequest, HTTP::Version1_1);
char *responseBuffer;
int responseLength = httpResponse.Create(responseBuffer);
response = std::string(responseBuffer, responseLength);
return NULL;
}
CWebSocket *websocket = NULL;
if (strncmp(value, "8", 1) == 0)
websocket = new CWebSocketV8();
else if (strncmp(value, "13", 2) == 0)
websocket = new CWebSocketV13();
if (websocket == NULL)
{
CLog::Log(LOGINFO, "WebSocket: Unsupported Sec-WebSocket-Version %s", value);
CHttpResponse httpResponse(HTTP::Get, HTTP::UpgradeRequired, HTTP::Version1_1);
httpResponse.AddHeader(WS_HEADER_VERSION, WS_SUPPORTED_VERSIONS);
char *responseBuffer;
int responseLength = httpResponse.Create(responseBuffer);
response = std::string(responseBuffer, responseLength);
return NULL;
}
if (websocket->Handshake(data, length, response))
return websocket;
return NULL;
}
|
