// 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.

#ifndef ARC_NETWORK_ADDRESS_MANAGER_H_
#define ARC_NETWORK_ADDRESS_MANAGER_H_

#include <map>
#include <memory>

#include <base/callback.h>
#include <base/macros.h>
#include <base/memory/weak_ptr.h>
#include <brillo/brillo_export.h>

#include "arc/network/mac_address_generator.h"
#include "arc/network/subnet.h"
#include "arc/network/subnet_pool.h"

namespace arc_networkd {

// Responsible for address provisioning for guest networks.
class BRILLO_EXPORT AddressManager {
 public:
  enum class Guest {
    ARC,
    ARC_NET,
    VM_ARC,
    VM_TERMINA,
    VM_PLUGIN,
    CONTAINER,
  };

  AddressManager(std::initializer_list<Guest> guests);
  virtual ~AddressManager() = default;

  // Generates a MAC address guaranteed to be unique for the lifetime of this
  // object.
  // Virtual for testing only.
  virtual MacAddress GenerateMacAddress();

  // 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.
  std::unique_ptr<Subnet> AllocateIPv4Subnet(Guest guest);

 private:
  MacAddressGenerator mac_addrs_;
  std::map<Guest, std::unique_ptr<SubnetPool>> pools_;

  base::WeakPtrFactory<AddressManager> weak_ptr_factory_{this};

  DISALLOW_COPY_AND_ASSIGN(AddressManager);
};

}  // namespace arc_networkd

#endif  // ARC_NETWORK_ADDRESS_MANAGER_H_