mirror of
https://github.com/OpenNebula/one.git
synced 2024-12-24 21:34:01 +03:00
192 lines
5.3 KiB
C++
192 lines
5.3 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_POOL_H_
|
|
#define USER_POOL_H_
|
|
|
|
#include "PoolSQL.h"
|
|
#include "User.h"
|
|
|
|
#include <time.h>
|
|
#include <sstream>
|
|
|
|
#include <iostream>
|
|
|
|
#include <vector>
|
|
|
|
using namespace std;
|
|
|
|
class AuthRequest; //Forward definition of AuthRequest
|
|
|
|
/**
|
|
* The User Pool class. ...
|
|
*/
|
|
class UserPool : public PoolSQL
|
|
{
|
|
public:
|
|
|
|
UserPool(SqlDB * db);
|
|
|
|
~UserPool(){};
|
|
|
|
/**
|
|
* Function to allocate a new User object
|
|
* @param oid the id assigned to the User
|
|
* @return the oid assigned to the object or -1 in case of failure
|
|
*/
|
|
int allocate (
|
|
int * oid,
|
|
string hostname,
|
|
string password,
|
|
bool enabled,
|
|
string& error_str);
|
|
|
|
/**
|
|
* Function to get a User from the pool, if the object is not in memory
|
|
* it is loaded from the DB
|
|
* @param oid User unique id
|
|
* @param lock locks the User mutex
|
|
* @return a pointer to the Host, 0 if the User could not be loaded
|
|
*/
|
|
User * get(
|
|
int oid,
|
|
bool lock)
|
|
{
|
|
User * user = static_cast<User *>(PoolSQL::get(oid,lock));
|
|
|
|
return user;
|
|
}
|
|
|
|
|
|
/**
|
|
* Function to get a User from the pool, if the object is not in memory
|
|
* it is loaded from the DB
|
|
* @param username
|
|
* @param lock locks the User mutex
|
|
* @return a pointer to the User, 0 if the User could not be loaded
|
|
*/
|
|
User * get(
|
|
string username,
|
|
bool lock)
|
|
{
|
|
map<string, int>::iterator index;
|
|
|
|
index = known_users.find(username);
|
|
|
|
if ( index != known_users.end() )
|
|
{
|
|
return get((int)index->second,lock);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/** Update a particular User
|
|
* @param user pointer to User
|
|
* @return 0 on success
|
|
*/
|
|
int update(User * user)
|
|
{
|
|
return user->update(db);
|
|
};
|
|
|
|
|
|
/** Drops a user from the DB, the user mutex MUST BE locked
|
|
* @param user pointer to User
|
|
*/
|
|
int drop(User * user)
|
|
{
|
|
int rc = PoolSQL::drop(user);
|
|
|
|
if ( rc == 0)
|
|
{
|
|
known_users.erase(user->get_name());
|
|
}
|
|
|
|
return rc;
|
|
};
|
|
|
|
/**
|
|
* Bootstraps the database table(s) associated to the User pool
|
|
*/
|
|
static void bootstrap(SqlDB * _db)
|
|
{
|
|
User::bootstrap(_db);
|
|
};
|
|
|
|
/**
|
|
* Returns whether there is a user with given username/password or not
|
|
* @param session, colon separated username and password string
|
|
* @return -1 if authn failed, uid of the user in other case
|
|
*/
|
|
int authenticate(string& session);
|
|
|
|
/**
|
|
* Returns whether there is a user with given username/password or not
|
|
* @param ar, an Authorization Request
|
|
* @return -1 if authz failed, 0 otherwise
|
|
*/
|
|
static int authorize(AuthRequest& ar);
|
|
|
|
/**
|
|
* Dumps the User pool in XML format. A filter can be also added to the
|
|
* query
|
|
* @param oss the output stream to dump the pool contents
|
|
* @param where filter for the objects, defaults to all
|
|
*
|
|
* @return 0 on success
|
|
*/
|
|
int dump(ostringstream& oss, const string& where);
|
|
|
|
private:
|
|
/**
|
|
* Factory method to produce User objects
|
|
* @return a pointer to the new User
|
|
*/
|
|
PoolObjectSQL * create()
|
|
{
|
|
return new User;
|
|
};
|
|
|
|
/**
|
|
* This map stores the association between UIDs and Usernames
|
|
*/
|
|
map<string, int> known_users;
|
|
|
|
/**
|
|
* Callback function to get output the user pool in XML format
|
|
* (User::dump)
|
|
* @param _oss pointer to 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
|
|
*/
|
|
int dump_cb(void * _oss, int num, char **values, char **names);
|
|
|
|
/**
|
|
* Callback function to build the knwon_user map (User::User)
|
|
* @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 init_cb(void *nil, int num, char **values, char **names);
|
|
};
|
|
|
|
#endif /*USER_POOL_H_*/
|
|
|