foundationdb/fdbrpc/IPAllowList.h

87 lines
2.4 KiB
C
Raw Normal View History

2022-02-18 23:48:44 +08:00
/*
* IPAllowList.h
*
* This source file is part of the FoundationDB open source project
*
* Copyright 2013-2022 Apple Inc. and the FoundationDB project authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#pragma once
#ifndef FDBRPC_IP_ALLOW_LIST_H
#define FDBRPC_IP_ALLOW_LIST_H
#include "flow/network.h"
#include "flow/Arena.h"
struct AuthAllowedSubnet {
IPAddress baseAddress;
IPAddress addressMask;
2022-04-11 23:17:10 +08:00
AuthAllowedSubnet(IPAddress const& baseAddress, IPAddress const& addressMask);
2022-02-18 23:48:44 +08:00
static AuthAllowedSubnet fromString(std::string_view addressString);
template <std::size_t sz>
2022-04-11 23:17:10 +08:00
static std::array<unsigned char, sz> createBitMask(std::array<unsigned char, sz> const& addr, int netmaskWeight);
2022-02-18 23:48:44 +08:00
2022-02-23 17:23:27 +08:00
bool operator()(IPAddress const& address) const {
2022-02-18 23:48:44 +08:00
if (addressMask.isV4() != address.isV4()) {
return false;
}
if (addressMask.isV4()) {
return (addressMask.toV4() & address.toV4()) == baseAddress.toV4();
} else {
auto res = address.toV6();
auto const& mask = addressMask.toV6();
for (int i = 0; i < res.size(); ++i) {
res[i] &= mask[i];
}
return res == baseAddress.toV6();
}
}
IPAddress netmask() const;
2022-02-24 02:02:29 +08:00
int netmaskWeight() const;
2022-02-18 23:48:44 +08:00
// some useful helper functions if we need to debug ip masks etc
static void printIP(std::string_view txt, IPAddress const& address);
};
class IPAllowList {
std::vector<AuthAllowedSubnet> subnetList;
2022-02-23 17:23:27 +08:00
2022-02-18 23:48:44 +08:00
public:
2022-02-23 17:23:27 +08:00
void addTrustedSubnet(std::string_view str) { subnetList.push_back(AuthAllowedSubnet::fromString(str)); }
2022-02-18 23:48:44 +08:00
2022-02-23 17:23:27 +08:00
void addTrustedSubnet(AuthAllowedSubnet const& subnet) { subnetList.push_back(subnet); }
2022-02-18 23:48:44 +08:00
2022-02-23 17:23:27 +08:00
std::vector<AuthAllowedSubnet> const& subnets() const { return subnetList; }
2022-02-18 23:48:44 +08:00
2022-02-23 17:23:27 +08:00
bool operator()(IPAddress address) const {
2022-02-18 23:48:44 +08:00
if (subnetList.empty()) {
return true;
}
for (auto const& subnet : subnetList) {
if (subnet(address)) {
return true;
}
}
return false;
}
};
#endif // FDBRPC_IP_ALLOW_LIST_H