thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-25 00:52:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Doing more URL check on domain and topLevelDomain function

Browse Source
This commit is contained in:
Guillaume Tassery 2019-04-23 12:23:59 +07:00
parent 5964646971
commit b567127f49
1 changed files with 49 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -8,6 +8,42 @@
namespace DB
{
static inline bool isUnsafeCharUrl(char c)
{
switch (c)
{
case ' ':
case '\t':
case '<':
case '>':
case '#':
case '%':
case '{':
case '}':
case '|':
case '\\':
case '^':
case '~':
case '[':
case ']':
return true;
}
return false;
}
static inline bool isEndOfUrl(char c)
{
switch (c)
{
case ':':
case '/':
case '?':
case '#':
return true;
}
return false;
}
/// Extracts host from given url.
inline StringRef getURLHost(const char * data, size_t size)
{
@ -39,14 +75,26 @@ inline StringRef getURLHost(const char * data, size_t size)
pos += 2;
const char * start_of_host = pos;
bool has_dot_delimiter = false;
for (; pos < end; ++pos)
{
if (*pos == '@')
start_of_host = pos + 1;
else if (*pos == ':' || *pos == '/' || *pos == '?' || *pos == '#')
else if (*pos == '.')
{
if (pos + 1 == end || isEndOfUrl(*(pos + 1)))
return StringRef{};
has_dot_delimiter = true;
}
else if (isEndOfUrl(*pos))
break;
else if (isUnsafeCharUrl(*pos))
return StringRef{};
}
if (!has_dot_delimiter)
return StringRef{};
return (pos == start_of_host) ? StringRef{} : StringRef(start_of_host, pos - start_of_host);
}