mirror of
https://github.com/OpenNebula/one.git
synced 2024-12-28 07:21:29 +03:00
322 lines
9.9 KiB
C++
322 lines
9.9 KiB
C++
/* -------------------------------------------------------------------------- */
|
|
/* Copyright 2002-2009, Distributed Systems Architecture Group, Universidad */
|
|
/* Complutense de Madrid (dsa-research.org) */
|
|
/* */
|
|
/* 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. */
|
|
/* -------------------------------------------------------------------------- */
|
|
|
|
#ifndef LEASES_H_
|
|
#define LEASES_H_
|
|
|
|
#include <sqlite3.h>
|
|
#include "ObjectSQL.h"
|
|
#include "Attribute.h"
|
|
|
|
#include <map>
|
|
#include <vector>
|
|
#include <sstream>
|
|
|
|
using namespace std;
|
|
|
|
extern "C" int leases_select_cb (
|
|
void * _lease,
|
|
int num,
|
|
char ** values,
|
|
char ** names);
|
|
|
|
/**
|
|
* The Leases class represents all the IP and MAC addresses (lease) that can
|
|
* be asigned (or are asigned) in a Virtual Network
|
|
*/
|
|
class Leases : public ObjectSQL
|
|
{
|
|
public:
|
|
/**
|
|
* Creates a new Lease set for a given Virtual Network (oid)
|
|
* @param _db pinter to the DB object used to store the leases
|
|
* @param _oid the virtual network unique identifier
|
|
* @param _size the max number of leases
|
|
*/
|
|
Leases(SqliteDB * _db, int _oid, unsigned long _size):
|
|
oid(_oid), size(_size), db(_db){};
|
|
|
|
virtual ~Leases()
|
|
{
|
|
map<unsigned int, Lease *>::iterator it;
|
|
|
|
for(it=leases.begin();it!=leases.end();it++)
|
|
{
|
|
delete it->second;
|
|
}
|
|
};
|
|
|
|
friend ostream& operator<<(ostream& os, Leases& _leases);
|
|
|
|
/**
|
|
* Returns an unused lease, which becomes used
|
|
* @param vid identifier of the VM getting this lease
|
|
* @param ip ip of the returned lease
|
|
* @param mac mac of the returned lease
|
|
* @return 0 if success
|
|
*/
|
|
virtual int get(int vid, string& ip,string& mac) = 0;
|
|
|
|
/**
|
|
* Ask for a specific lease in the network
|
|
* @param vid identifier of the VM getting this lease
|
|
* @param ip ip of lease requested
|
|
* @param mac mac of the lease
|
|
* @return 0 if success
|
|
*/
|
|
virtual int set(int vid, const string& ip, string& mac) = 0;
|
|
|
|
/**
|
|
* Release an used lease, which becomes unused
|
|
* @param ip of the lease in use
|
|
*/
|
|
virtual void release(const string& ip) = 0;
|
|
|
|
// -------------------------------------------------------------------------
|
|
// -------------------------------------------------------------------------
|
|
|
|
protected:
|
|
/**
|
|
* The Lease class, it represents a pair of IP and MAC assigned to
|
|
* a Virtual Machine
|
|
*/
|
|
class Lease
|
|
{
|
|
public:
|
|
/**
|
|
* Creates a new lease, string form. This constructor throws a runtime
|
|
* exception if the IP or MAC format is wrong.
|
|
* @param _ip, the Lease IP in string format
|
|
* @param _mac, the Lease MAC in string format
|
|
* @param _vid, the ID of the VM owning the lease
|
|
* @param _used, the lease is in use
|
|
*/
|
|
//Lease(const string& _ip, const string& _mac,int _vid, bool _used=true);
|
|
|
|
/**
|
|
* Creates a new lease, numeric form.
|
|
* @param _ip, the Lease IP in numeric format
|
|
* @param _mac, the Lease MAC in numeric format
|
|
* @param _vid, the ID of the VM owning the lease
|
|
* @param _used, the lease is in use
|
|
*/
|
|
Lease(unsigned int _ip, unsigned int _mac[], int _vid, bool _used=true)
|
|
:ip(_ip), vid(_vid), used(_used)
|
|
{
|
|
// TODO check size
|
|
mac[PREFIX]=_mac[PREFIX];
|
|
mac[SUFFIX]=_mac[SUFFIX];
|
|
};
|
|
|
|
~Lease(){};
|
|
|
|
/**
|
|
* Converts this lease's IP and MAC to string
|
|
* @param ip ip of the lease in string
|
|
* @param mac mac of the lease in string
|
|
*/
|
|
void to_string(string& _ip, string& _mac) const;
|
|
|
|
/**
|
|
* Conversion from string IP to unsigned int IP
|
|
* @return 0 if success
|
|
*/
|
|
static int ip_to_number(const string& ip, unsigned int& i_ip);
|
|
|
|
/**
|
|
* Conversion from unsigned int IP to string IP
|
|
*/
|
|
static void ip_to_string(const unsigned int i_ip, string& ip);
|
|
|
|
/**
|
|
* Conversion from string MAC to unsigned int[] MAC
|
|
* @return 0 if success
|
|
*/
|
|
static int mac_to_number(const string& mac, unsigned int i_mac[]);
|
|
|
|
/**
|
|
* Conversion from string IP to unsigned int IP
|
|
*/
|
|
static void mac_to_string(const unsigned int i_mac[], string& mac);
|
|
|
|
/**
|
|
* Prints a Lease in a single line
|
|
*/
|
|
friend ostream& operator<<(ostream& os, Lease& _lease);
|
|
|
|
/**
|
|
* Function to print the Lease object into a string in
|
|
* plain text
|
|
* @param str the resulting string
|
|
* @return a reference to the generated string
|
|
*/
|
|
string& to_str(string& str) const;
|
|
|
|
/**
|
|
* Function to print the Lease object into a string in
|
|
* XML format
|
|
* @param xml the resulting XML string
|
|
* @return a reference to the generated string
|
|
*/
|
|
string& to_xml(string& xml) const;
|
|
|
|
/**
|
|
* Constants to access the array storing the MAC address
|
|
*/
|
|
enum MACIndex
|
|
{
|
|
SUFFIX = 0,/**< Lower significant 4 bytes */
|
|
PREFIX = 1 /**< Higher significant 2 bytes */
|
|
};
|
|
|
|
unsigned int ip;
|
|
|
|
unsigned int mac [2];
|
|
|
|
int vid;
|
|
|
|
bool used;
|
|
};
|
|
|
|
friend class VirtualNetwork;
|
|
|
|
|
|
// -------------------------------------------------------------------------
|
|
// Leases fields
|
|
// -------------------------------------------------------------------------
|
|
/**
|
|
* Leases indentifier. Connects it to a Virtual Network
|
|
*/
|
|
int oid;
|
|
|
|
/**
|
|
* Number of possible leases (free + asigned)
|
|
*/
|
|
unsigned int size;
|
|
|
|
/**
|
|
* Hash of leases, indexed by lease.ip
|
|
*/
|
|
map<unsigned int, Lease *> leases;
|
|
|
|
// -------------------------------------------------------------------------
|
|
// DataBase implementation variables
|
|
// -------------------------------------------------------------------------
|
|
/**
|
|
* Pointer to the DataBase
|
|
*/
|
|
SqliteDB * db;
|
|
|
|
enum ColNames
|
|
{
|
|
OID = 0,
|
|
IP = 1,
|
|
MAC_PREFIX = 2,
|
|
MAC_SUFFIX = 3,
|
|
VID = 4,
|
|
USED = 5,
|
|
LIMIT = 6
|
|
};
|
|
|
|
static const char * table;
|
|
|
|
static const char * db_names;
|
|
|
|
static const char * db_bootstrap;
|
|
|
|
// -------------------------------------------------------------------------
|
|
// Leases methods
|
|
// -------------------------------------------------------------------------
|
|
/**
|
|
* Check if the passed ip corresponds with a given lease
|
|
* @param ip of the lease to be checked
|
|
* @return true if the ip was already assigned
|
|
*/
|
|
bool check(const string& ip);
|
|
|
|
bool check(unsigned int ip);
|
|
|
|
/**
|
|
* Reads the leases from the DB, and updates the lease hash table
|
|
* @param db pointer to the database.
|
|
* @return 0 on success.
|
|
*/
|
|
virtual int select(SqliteDB * db);
|
|
|
|
friend ostream& operator<<(ostream& os, Lease& _lease);
|
|
|
|
/**
|
|
* Function to print the Leases object into a string in
|
|
* plain text
|
|
* @param str the resulting string
|
|
* @return a reference to the generated string
|
|
*/
|
|
string& to_str(string& str) const;
|
|
|
|
/**
|
|
* Function to print the Leases object into a string in
|
|
* XML format
|
|
* @param xml the resulting XML string
|
|
* @return a reference to the generated string
|
|
*/
|
|
string& to_xml(string& xml) const;
|
|
|
|
private:
|
|
|
|
friend int leases_select_cb (
|
|
void * _leases,
|
|
int num,
|
|
char ** values,
|
|
char ** names);
|
|
|
|
/**
|
|
* Function to unmarshall a leases object
|
|
* @param num the number of columns read from the DB
|
|
* @para names the column names
|
|
* @para vaues the column values
|
|
* @return 0 on success
|
|
*/
|
|
int unmarshall(int num, char **names, char ** values);
|
|
|
|
/**
|
|
* This method should not be called, leases are added/removed/updated
|
|
* through add/del interface
|
|
* @param db pointer to the database.
|
|
* @return 0 on success.
|
|
*/
|
|
int insert(SqliteDB * db);
|
|
|
|
/**
|
|
* Leases are added/removed/updated through add/del interface
|
|
* This method is for pool management.
|
|
* @param db pointer to the database.
|
|
* @return 0 on success.
|
|
*/
|
|
int drop(SqliteDB * db);
|
|
|
|
/**
|
|
* This method should not be called, leases are added/removed/updated
|
|
* through add/del interface
|
|
* @param db pointer to the database.
|
|
* @return 0 on success.
|
|
*/
|
|
int update(SqliteDB * db);
|
|
};
|
|
|
|
#endif /*LEASES_H_*/
|
|
|