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
|
/*
* Copyright (C) 2011-2018 Team Kodi
* This file is part of Kodi - https://kodi.tv
*
* SPDX-License-Identifier: GPL-2.0-or-later
* See LICENSES/README.md for more information.
*/
#include "WebSocketManager.h"
#include "WebSocket.h"
#include "WebSocketV13.h"
#include "WebSocketV8.h"
#include "utils/HttpParser.h"
#include "utils/HttpResponse.h"
#include "utils/log.h"
#include <string>
#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"
CWebSocket* CWebSocketManager::Handle(const char* data, unsigned int length, std::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);
response = httpResponse.Create();
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 {}", value);
CHttpResponse httpResponse(HTTP::Get, HTTP::UpgradeRequired, HTTP::Version1_1);
httpResponse.AddHeader(WS_HEADER_VERSION, WS_SUPPORTED_VERSIONS);
response = httpResponse.Create();
return NULL;
}
if (websocket->Handshake(data, length, response))
return websocket;
return NULL;
}
|
