blob: 535ac754136adde2a830587f3d984aec84bf631e [file] [log] [blame]
// Copyright 2019 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <map>
#include <memory>
#include <base/callback.h>
#include <base/macros.h>
#include <base/memory/weak_ptr.h>
#include <brillo/brillo_export.h>
#include "patchpanel/guest_type.h"
#include "patchpanel/mac_address_generator.h"
#include "patchpanel/subnet.h"
#include "patchpanel/subnet_pool.h"
namespace patchpanel {
// Responsible for address provisioning for guest networks.
class BRILLO_EXPORT AddressManager {
AddressManager(const AddressManager&) = delete;
AddressManager& operator=(const AddressManager&) = delete;
virtual ~AddressManager() = default;
// Generates a MAC address guaranteed to be unique for the lifetime of this
// object.
// If |index| is provided, a MAC address will be returned that is stable
// across all invocations and instantions.
// Virtual for testing only.
virtual MacAddress GenerateMacAddress(uint8_t index = kAnySubnetIndex);
// Allocates a subnet from the specified guest network pool if available.
// Returns nullptr if the guest was configured or no more subnets are
// available for allocation.
// |index| is used to acquire a particular subnet from the pool, if supported
// for |guest|, it is 1-based, so 0 indicates no preference.
std::unique_ptr<Subnet> AllocateIPv4Subnet(GuestType guest_type,
uint32_t index = kAnySubnetIndex);
MacAddressGenerator mac_addrs_;
std::map<GuestType, std::unique_ptr<SubnetPool>> pools_;
base::WeakPtrFactory<AddressManager> weak_ptr_factory_{this};
} // namespace patchpanel