mirror of
https://github.com/OpenNebula/one.git
synced 2024-12-24 21:34:01 +03:00
266 lines
7.1 KiB
C++
266 lines
7.1 KiB
C++
/* -------------------------------------------------------------------------- */
|
|
/* Copyright 2002-2010, OpenNebula Project Leads (OpenNebula.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 USER_H_
|
|
#define USER_H_
|
|
|
|
#include "PoolSQL.h"
|
|
|
|
using namespace std;
|
|
|
|
/* -------------------------------------------------------------------------- */
|
|
/* -------------------------------------------------------------------------- */
|
|
|
|
/**
|
|
* The User class.
|
|
*/
|
|
class User : public PoolObjectSQL
|
|
{
|
|
public:
|
|
|
|
/**
|
|
* Function to write a User on an output stream
|
|
*/
|
|
friend ostream& operator<<(ostream& os, User& u);
|
|
|
|
/**
|
|
* Function to print the User 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 User 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;
|
|
|
|
/**
|
|
* Get the User unique identifier UID, that matches the OID of the object
|
|
* @return UID User identifier
|
|
*/
|
|
int get_uid() const
|
|
{
|
|
return oid;
|
|
};
|
|
|
|
/**
|
|
* Check if the user is enabled
|
|
* @return true if the user is enabled
|
|
*/
|
|
bool isEnabled() const
|
|
{
|
|
return enabled;
|
|
}
|
|
|
|
/**
|
|
* Returns user username
|
|
* @return username User's hostname
|
|
*/
|
|
const string& get_username() const
|
|
{
|
|
return username;
|
|
};
|
|
|
|
/**
|
|
* Returns user password
|
|
* @return username User's hostname
|
|
*/
|
|
const string& get_password() const
|
|
{
|
|
return password;
|
|
};
|
|
|
|
/**
|
|
* Enables the current user
|
|
*/
|
|
void enable()
|
|
{
|
|
enabled = true;
|
|
};
|
|
|
|
/**
|
|
* Disables the current user
|
|
*/
|
|
void disable()
|
|
{
|
|
enabled = false;
|
|
};
|
|
|
|
/**
|
|
* Sets user username
|
|
*/
|
|
void set_username(string _username)
|
|
{
|
|
username = _username;
|
|
};
|
|
|
|
/**
|
|
* Sets user password
|
|
*/
|
|
void set_password(string _password)
|
|
{
|
|
password = _password;
|
|
};
|
|
|
|
/**
|
|
* Splits an authentication token (<usr>:<pass>)
|
|
* @param secret, the authentication token
|
|
* @param username
|
|
* @param password
|
|
* @return 0 on success
|
|
**/
|
|
static int split_secret(const string secret, string& user, string& pass);
|
|
|
|
/**
|
|
* "Encrypts" the password with SHA1 digest
|
|
* @param password
|
|
* @return sha1 encrypted password
|
|
*/
|
|
static string sha1_digest(const string& pass);
|
|
|
|
private:
|
|
// -------------------------------------------------------------------------
|
|
// Friends
|
|
// -------------------------------------------------------------------------
|
|
|
|
friend class UserPool;
|
|
|
|
// -------------------------------------------------------------------------
|
|
// User Attributes
|
|
// -------------------------------------------------------------------------
|
|
|
|
/**
|
|
* User's username
|
|
*/
|
|
string username;
|
|
|
|
/**
|
|
* User's password
|
|
*/
|
|
string password;
|
|
|
|
/**
|
|
* Flag marking user enabled/disabled
|
|
*/
|
|
bool enabled;
|
|
|
|
// *************************************************************************
|
|
// DataBase implementation (Private)
|
|
// *************************************************************************
|
|
|
|
/**
|
|
* Execute an INSERT or REPLACE Sql query.
|
|
* @param db The SQL DB
|
|
* @param replace Execute an INSERT or a REPLACE
|
|
* @return 0 one success
|
|
*/
|
|
int insert_replace(SqlDB *db, bool replace);
|
|
|
|
/**
|
|
* Callback function to unmarshall a User object (User::select)
|
|
* @param num the number of columns read from the DB
|
|
* @param names the column names
|
|
* @param vaues the column values
|
|
* @return 0 on success
|
|
*/
|
|
int select_cb(void *nil, int num, char **values, char **names);
|
|
|
|
/**
|
|
* Bootstraps the database table(s) associated to the User
|
|
*/
|
|
static void bootstrap(SqlDB * db)
|
|
{
|
|
ostringstream oss_user(User::db_bootstrap);
|
|
|
|
db->exec(oss_user);
|
|
};
|
|
|
|
protected:
|
|
|
|
// *************************************************************************
|
|
// Constructor
|
|
// *************************************************************************
|
|
|
|
User(int id=-1,
|
|
string _username="",
|
|
string _password="",
|
|
bool _enabled=true);
|
|
|
|
virtual ~User();
|
|
|
|
// *************************************************************************
|
|
// DataBase implementation
|
|
// *************************************************************************
|
|
|
|
enum ColNames
|
|
{
|
|
OID = 0,
|
|
USERNAME = 1,
|
|
PASSWORD = 2,
|
|
ENABLED = 3, // 0 = false, 1 = true
|
|
LIMIT = 4
|
|
};
|
|
|
|
static const char * db_names;
|
|
|
|
static const char * db_bootstrap;
|
|
|
|
static const char * table;
|
|
|
|
/**
|
|
* Reads the User (identified with its OID=UID) from the database.
|
|
* @param db pointer to the db
|
|
* @return 0 on success
|
|
*/
|
|
virtual int select(SqlDB *db);
|
|
|
|
/**
|
|
* Writes the User in the database.
|
|
* @param db pointer to the db
|
|
* @return 0 on success
|
|
*/
|
|
virtual int insert(SqlDB *db);
|
|
|
|
/**
|
|
* Writes/updates the User data fields in the database.
|
|
* @param db pointer to the db
|
|
* @return 0 on success
|
|
*/
|
|
virtual int update(SqlDB *db);
|
|
|
|
/**
|
|
* Drops USer from the database
|
|
* @param db pointer to the db
|
|
* @return 0 on success
|
|
*/
|
|
virtual int drop(SqlDB *db);
|
|
|
|
/**
|
|
* Function to output a User object in to an stream in XML format
|
|
* @param oss the output stream
|
|
* @param num the number of columns read from the DB
|
|
* @param names the column names
|
|
* @param vaues the column values
|
|
* @return 0 on success
|
|
*/
|
|
static int dump(ostringstream& oss, int num, char **values, char **names);
|
|
};
|
|
|
|
#endif /*USER_H_*/
|