mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-21 15:12:02 +00:00
Fix error
This commit is contained in:
parent
467b7e24d3
commit
d1ed1fb83a
@ -1,5 +1,6 @@
|
||||
#pragma once
|
||||
|
||||
#include <cassert>
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include <functional>
|
||||
@ -27,7 +28,11 @@ struct StringRef
|
||||
size_t size = 0;
|
||||
|
||||
template <typename CharT, typename = std::enable_if_t<sizeof(CharT) == 1>>
|
||||
constexpr StringRef(const CharT * data_, size_t size_) : data(reinterpret_cast<const char *>(data_)), size(size_) {}
|
||||
constexpr StringRef(const CharT * data_, size_t size_) : data(reinterpret_cast<const char *>(data_)), size(size_)
|
||||
{
|
||||
/// Sanity check for overflowed values.
|
||||
assert(size < 0x8000000000000000ULL);
|
||||
}
|
||||
|
||||
StringRef(const std::string & s) : data(s.data()), size(s.size()) {}
|
||||
constexpr explicit StringRef(const std::string_view & s) : data(s.data()), size(s.size()) {}
|
||||
|
@ -20,7 +20,8 @@ struct ExtractNetloc
|
||||
Pos pos = data;
|
||||
Pos end = data + size;
|
||||
|
||||
if (*pos == '/' && *(pos + 1) == '/')
|
||||
/// Skip scheme.
|
||||
if (pos + 2 < end && pos[0] == '/' && pos[1] == '/')
|
||||
{
|
||||
pos += 2;
|
||||
}
|
||||
@ -55,16 +56,18 @@ struct ExtractNetloc
|
||||
case '&':
|
||||
return StringRef{};
|
||||
default:
|
||||
goto exloop;
|
||||
pos = scheme_end; /// exit from the loop
|
||||
}
|
||||
}
|
||||
}
|
||||
exloop: if ((scheme_end - pos) > 2 && *pos == ':' && *(pos + 1) == '/' && *(pos + 2) == '/')
|
||||
pos += 3;
|
||||
else
|
||||
pos = data;
|
||||
if (pos + 2 < scheme_end && pos[0] == ':' && pos[1] == '/' && pos[2] == '/')
|
||||
pos += 3;
|
||||
else
|
||||
pos = data;
|
||||
}
|
||||
|
||||
/// Now pos points to the first byte after scheme (if there is).
|
||||
|
||||
bool has_identification = false;
|
||||
Pos question_mark_pos = end;
|
||||
Pos slash_pos = end;
|
||||
@ -106,7 +109,9 @@ exloop: if ((scheme_end - pos) > 2 && *pos == ':' && *(pos + 1) == '/' && *(pos
|
||||
case ';':
|
||||
case '=':
|
||||
case '&':
|
||||
return StringRef(start_of_host, std::min(std::min(pos - 1, question_mark_pos), slash_pos) - start_of_host);
|
||||
return pos > start_of_host
|
||||
? StringRef(start_of_host, std::min(std::min(pos - 1, question_mark_pos), slash_pos) - start_of_host)
|
||||
: StringRef{};
|
||||
}
|
||||
}
|
||||
|
||||
|
1
tests/queries/0_stateless/01434_netloc_fuzz.reference
Normal file
1
tests/queries/0_stateless/01434_netloc_fuzz.reference
Normal file
@ -0,0 +1 @@
|
||||
|
1
tests/queries/0_stateless/01434_netloc_fuzz.sql
Normal file
1
tests/queries/0_stateless/01434_netloc_fuzz.sql
Normal file
@ -0,0 +1 @@
|
||||
SELECT netloc('<\'[%UzO');
|
Loading…
Reference in New Issue
Block a user