ProxyIpResolver.hpp

#pragma once
 
#include "util/config/ConfigDefinition.hpp"
#include "util/config/ValueView.hpp"
 
#include <boost/beast/http/field.hpp>
#include <boost/beast/http/message.hpp>
#include <boost/beast/http/string_body.hpp>
#include <xrpl/basics/base_uint.h>
 
#include <optional>
#include <string>
#include <string_view>
#include <unordered_set>
 
namespace web {

class ProxyIpResolver {
    std::unordered_set<std::string> proxyIps_;
    // ripple::uint256 doesn't have hash implementation
    std::unordered_set<ripple::uint256, ripple::uint256::hasher> proxyTokens_;
 
public:
    ProxyIpResolver(
        std::unordered_set<std::string> proxyIps,
        std::unordered_set<std::string> proxyTokens
    );

    static ProxyIpResolver
    fromConfig(util::config::ClioConfigDefinition const& config);
 
    using HttpHeaders = boost::beast::http::request<boost::beast::http::string_body>::header_type;
 
    static constexpr std::string_view kPROXY_TOKEN_HEADER = "X-Proxy-Token";

    std::string
    resolveClientIp(std::string const& connectionIp, HttpHeaders const& headers) const;
 
private:
    static std::optional<std::string>
    extractClientIp(HttpHeaders const& headers);
};
 
}  // namespace web