thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-10-01 06:00:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Check if we got a reserved char on url for domain and top level domain

Browse Source
This commit is contained in:
Guillaume Tassery 2019-04-23 12:56:24 +07:00
parent c938935766
commit 3639f03bc6
1 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
dbms/src/Functions/domain.h
View File

@ -31,7 +31,7 @@ static inline bool isUnsafeCharUrl(char c)
return false;
}
static inline bool isEndOfUrl(char c)
static inline bool isCharEndOfUrl(char c)
{
switch (c)
{
@ -44,6 +44,22 @@ static inline bool isEndOfUrl(char c)
return false;
}
static inline bool isReservedCharUrl(char c)
{
switch (c)
{
case ';':
case '/':
case '?':
case ':':
case '@':
case '=':
case '&':
return true;
}
return false;
}
/// Extracts host from given url.
inline StringRef getURLHost(const char * data, size_t size)
{
@ -82,13 +98,13 @@ inline StringRef getURLHost(const char * data, size_t size)
start_of_host = pos + 1;
else if (*pos == '.')
{
if (pos + 1 == end || isEndOfUrl(*(pos + 1)))
if (pos + 1 == end || isCharEndOfUrl(*(pos + 1)))
return StringRef{};
has_dot_delimiter = true;
}
else if (isEndOfUrl(*pos))
break;
else if (isUnsafeCharUrl(*pos))
else if (isUnsafeCharUrl(*pos) || isReservedCharUrl(*pos))
return StringRef{};
}