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
|
#include "location.h"
#include <util/string/cast.h>
using namespace NNeh;
namespace {
bool IsHttps(TStringBuf scheme) {
return TStringBuf("https") == scheme || TStringBuf("fulls") == scheme || TStringBuf("posts") == scheme;
}
bool SplitUserInfo(TStringBuf& path, TStringBuf& userInfo, bool mayHaveSpecificUserInfo) {
const size_t pos = path.find_first_of(mayHaveSpecificUserInfo ? TStringBuf("?@") : TStringBuf("?@/"));
if (TStringBuf::npos != pos && '@' == path[pos]) {
if (mayHaveSpecificUserInfo) {
if (!(path.StartsWith("cert=") || path.StartsWith("key="))) {
return false;
}
}
path.SplitAt(pos, userInfo, path);
path.Skip(1);
}
return true;
}
}
TParsedLocation::TParsedLocation(TStringBuf path) {
path.Split(':', Scheme, path);
path.Skip(2);
// try to handle both https://cert=./path_to_cert@host:port/... (userinfo is "cert=./path_to_cert")
// and http[s]://host:port/@zzz (userinfo is empty, host is host, not zzz
// see SEARCH-14238
if (!SplitUserInfo(path, UserInfo, IsHttps(Scheme))) {
SplitUserInfo(path, UserInfo, false);
}
auto checkRange = [](size_t b, size_t e){
return b != TStringBuf::npos && e != TStringBuf::npos && b < e;
};
size_t oBracket = path.find_first_of('[');
size_t cBracket = path.find_first_of(']');
size_t endEndPointPos = path.find_first_of('/');
if (checkRange(oBracket, cBracket)) {
endEndPointPos = path.find_first_of('/', cBracket);
}
EndPoint = path.SubStr(0, endEndPointPos);
Host = EndPoint;
size_t lastColon = EndPoint.find_last_of(':');
if (checkRange(cBracket, lastColon)
|| (cBracket == TStringBuf::npos && lastColon != TStringBuf::npos))
{
Host = EndPoint.SubStr(0, lastColon);
Port = EndPoint.SubStr(lastColon + 1, EndPoint.size() - lastColon + 1);
}
if (endEndPointPos != TStringBuf::npos) {
Service = path.SubStr(endEndPointPos + 1, path.size() - endEndPointPos + 1);
}
}
ui16 TParsedLocation::GetPort() const {
if (!Port) {
return IsHttps(Scheme) ? 443 : 80;
}
return FromString<ui16>(Port);
}
|
