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

Check for protocol prefix

Browse Source
This commit is contained in:
Konstantin Bogdanov 2024-04-26 16:45:30 +02:00
parent 7326e2a0fa
commit 07e4e7fa73
Signed by: thevar1able
GPG Key ID: DB399448D9FE52F1
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/Server/GRPCServer.cpp
View File

@ -1,6 +1,7 @@
#include "GRPCServer.h"
#include <limits>
#include <memory>
#include <Poco/Net/SocketAddress.h>
#if USE_GRPC
#include <Columns/ColumnString.h>
@ -322,12 +323,18 @@ namespace
{
/// Returns a string like ipv4:127.0.0.1:55930 or ipv6:%5B::1%5D:55930
String uri_encoded_peer = grpc_context.peer();
uri_encoded_peer = uri_encoded_peer.substr(uri_encoded_peer.find(':') + 1);
String peer;
Poco::URI::decode(uri_encoded_peer, peer);
if (uri_encoded_peer.starts_with("ipv4") || uri_encoded_peer.starts_with("ipv6"))
{
uri_encoded_peer = uri_encoded_peer.substr(uri_encoded_peer.find(':') + 1);
return Poco::Net::SocketAddress{peer};
String peer;
Poco::URI::decode(uri_encoded_peer, peer);
return Poco::Net::SocketAddress{Poco::Net::AddressFamily::Family::IPv4, uri_encoded_peer};
}
else
throw Exception(ErrorCodes::LOGICAL_ERROR, "Expected ipv4 or ipv6 protocol in peer address, got {}", uri_encoded_peer);
}
std::optional<String> getClientHeader(const String & key) const