From fcafc446aa1f15975b1802e47952ed371446e4d4 Mon Sep 17 00:00:00 2001 From: Maxim Mamontov Date: Sat, 13 Aug 2011 15:42:34 +0300 Subject: [PATCH] Initial services implementation added --- projects/stargazer/services_impl.cpp | 307 +++++++++++++++++++++++++++ projects/stargazer/services_impl.h | 67 ++++++ 2 files changed, 374 insertions(+) create mode 100644 projects/stargazer/services_impl.cpp create mode 100644 projects/stargazer/services_impl.h diff --git a/projects/stargazer/services_impl.cpp b/projects/stargazer/services_impl.cpp new file mode 100644 index 00000000..400b34b3 --- /dev/null +++ b/projects/stargazer/services_impl.cpp @@ -0,0 +1,307 @@ +/* + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + +/* + * Author : Maxim Mamontov + */ + +#include +#include +#include + +#include "stg/admin.h" +#include "stg/common.h" +#include "services_impl.h" + +//----------------------------------------------------------------------------- +SERVICES_IMPL::SERVICES_IMPL(STORE * st) + : SERVICES(), + data(), + store(st), + WriteServLog(GetStgLogger()), + searchDescriptors(), + handle(0) +{ +pthread_mutex_init(&mutex, NULL); +ReadServices(); +} +//----------------------------------------------------------------------------- +int SERVICES_IMPL::Add(const SERVICE_CONF & service, const ADMIN * admin) +{ +STG_LOCKER lock(&mutex, __FILE__, __LINE__); +const PRIV * priv = admin->GetPriv(); + +if (!priv->adminChg) + { + string s = admin->GetLogStr() + " Add administrator \'" + login + "\'. Access denied."; + strError = "Access denied."; + WriteServLog(s.c_str()); + return -1; + } + +ADMIN_IMPL adm(0, login, ""); +admin_iter ai(find(data.begin(), data.end(), adm)); + +if (ai != data.end()) + { + strError = "Administrator \'" + login + "\' cannot not be added. Administrator already exist."; + WriteServLog("%s %s", admin->GetLogStr().c_str(), strError.c_str()); + + return -1; + } + +data.push_back(adm); + +if (store->AddAdmin(login) == 0) + { + WriteServLog("%s Administrator \'%s\' added.", + admin->GetLogStr().c_str(), login.c_str()); + return 0; + } + +strError = "Administrator \'" + login + "\' was not added. Error: " + store->GetStrError(); +WriteServLog("%s %s", admin->GetLogStr().c_str(), strError.c_str()); + +return -1; +} +//----------------------------------------------------------------------------- +int SERVICES_IMPL::Del(const string & login, const ADMIN * admin) +{ +STG_LOCKER lock(&mutex, __FILE__, __LINE__); +ADMIN_IMPL adm(0, login, ""); +const PRIV * priv = admin->GetPriv(); + +if (!priv->adminChg) + { + string s = admin->GetLogStr() + " Delete administrator \'" + login + "\'. Access denied."; + strError = "Access denied."; + WriteServLog(s.c_str()); + return -1; + } + +admin_iter ai(find(data.begin(), data.end(), adm)); + +if (ai == data.end()) + { + strError = "Administrator \'" + login + "\' cannot be deleted. Administrator does not exist."; + WriteServLog("%s %s", admin->GetLogStr().c_str(), strError.c_str()); + return -1; + } + +map::iterator si; +si = searchDescriptors.begin(); +while (si != searchDescriptors.end()) + { + if (si->second == ai) + (si->second)++; + si++; + } + +data.remove(*ai); +if (store->DelAdmin(login) < 0) + { + strError = "Administrator \'" + login + "\' was not deleted. Error: " + store->GetStrError(); + WriteServLog("%s %s", admin->GetLogStr().c_str(), strError.c_str()); + + return -1; + } + +WriteServLog("%s Administrator \'%s\' deleted.", admin->GetLogStr().c_str(), login.c_str()); +return 0; +} +//----------------------------------------------------------------------------- +int SERVICES_IMPL::Change(const ADMIN_CONF & ac, const ADMIN * admin) +{ +STG_LOCKER lock(&mutex, __FILE__, __LINE__); +const PRIV * priv = admin->GetPriv(); + +if (!priv->adminChg) + { + string s = admin->GetLogStr() + " Change administrator \'" + ac.login + "\'. Access denied."; + strError = "Access denied."; + WriteServLog(s.c_str()); + return -1; + } + +ADMIN_IMPL adm(0, ac.login, ""); +admin_iter ai(find(data.begin(), data.end(), adm)); + +if (ai == data.end()) + { + strError = "Administrator \'" + ac.login + "\' cannot be changed " + ". Administrator does not exist."; + WriteServLog("%s %s", admin->GetLogStr().c_str(), strError.c_str()); + return -1; + } + +*ai = ac; +if (store->SaveAdmin(ac)) + { + WriteServLog("Cannot write admin %s.", ac.login.c_str()); + WriteServLog("%s", store->GetStrError().c_str()); + return -1; + } + +WriteServLog("%s Administrator \'%s\' changed.", + admin->GetLogStr().c_str(), ac.login.c_str()); + +return 0; +} +//----------------------------------------------------------------------------- +int SERVICES_IMPL::ReadAdmins() +{ +STG_LOCKER lock(&mutex, __FILE__, __LINE__); +vector adminsList; +if (store->GetAdminsList(&adminsList) < 0) + { + WriteServLog(store->GetStrError().c_str()); + return -1; + } + +for (unsigned int i = 0; i < adminsList.size(); i++) + { + ADMIN_CONF ac(0, adminsList[i], ""); + + if (store->RestoreAdmin(&ac, adminsList[i])) + { + WriteServLog(store->GetStrError().c_str()); + return -1; + } + + data.push_back(ADMIN_IMPL(ac)); + } +return 0; +} +//----------------------------------------------------------------------------- +void SERVICES_IMPL::PrintAdmins() const +{ +STG_LOCKER lock(&mutex, __FILE__, __LINE__); +const_admin_iter ai(data.begin()); +while (ai != data.end()) + { + ai->Print(); + ai++; + } +} +//----------------------------------------------------------------------------- +bool SERVICES_IMPL::FindAdmin(const string & l, ADMIN ** admin) +{ +assert(admin != NULL && "Pointer to admin is not null"); + +STG_LOCKER lock(&mutex, __FILE__, __LINE__); +if (data.empty()) + { + printfd(__FILE__, "no admin in system!\n"); + *admin = &noAdmin; + return false; + } + +ADMIN_IMPL adm(0, l, ""); +admin_iter ai(find(data.begin(), data.end(), adm)); + +if (ai != data.end()) + { + *admin = &(*ai); + return false; + } + +return true; +} +//----------------------------------------------------------------------------- +bool SERVICES_IMPL::AdminExists(const string & login) const +{ +STG_LOCKER lock(&mutex, __FILE__, __LINE__); +if (data.empty()) + { + printfd(__FILE__, "no admin in system!\n"); + return true; + } + +ADMIN_IMPL adm(0, login, ""); +const_admin_iter ai(find(data.begin(), data.end(), adm)); + +if (ai != data.end()) + return true; + +return false; +} +//----------------------------------------------------------------------------- +bool SERVICES_IMPL::AdminCorrect(const string & login, const std::string & password, ADMIN ** admin) +{ +STG_LOCKER lock(&mutex, __FILE__, __LINE__); +if (data.empty()) + { + printfd(__FILE__, "no admin in system!\n"); + return true; + } + +ADMIN_IMPL adm(0, login, ""); +admin_iter ai(find(data.begin(), data.end(), adm)); + +if (ai == data.end()) + { + return false; + } + +if (ai->GetPassword() != password) + { + return false; + } + +*admin = &(*ai); + +return true; +} +//----------------------------------------------------------------------------- +int SERVICES_IMPL::OpenSearch() const +{ +STG_LOCKER lock(&mutex, __FILE__, __LINE__); +handle++; +searchDescriptors[handle] = data.begin(); +return handle; +} +//----------------------------------------------------------------------------- +int SERVICES_IMPL::SearchNext(int h, ADMIN_CONF * ac) const +{ +STG_LOCKER lock(&mutex, __FILE__, __LINE__); +if (searchDescriptors.find(h) == searchDescriptors.end()) + { + WriteServLog("SERVICES. Incorrect search handle."); + return -1; + } + +if (searchDescriptors[h] == data.end()) + return -1; + +ADMIN_IMPL a = *searchDescriptors[h]++; + +*ac = a.GetConf(); + +return 0; +} +//----------------------------------------------------------------------------- +int SERVICES_IMPL::CloseSearch(int h) const +{ +STG_LOCKER lock(&mutex, __FILE__, __LINE__); +if (searchDescriptors.find(h) != searchDescriptors.end()) + { + searchDescriptors.erase(searchDescriptors.find(h)); + return 0; + } + +WriteServLog("SERVICES. Incorrect search handle."); +return -1; +} +//----------------------------------------------------------------------------- diff --git a/projects/stargazer/services_impl.h b/projects/stargazer/services_impl.h new file mode 100644 index 00000000..4c7044fa --- /dev/null +++ b/projects/stargazer/services_impl.h @@ -0,0 +1,67 @@ +/* + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + +/* + * Author : Maxim Mamontov + */ + +#ifndef SERVICES_IMPL_H +#define SERVICES_IMPL_H + +#include + +#include +#include +#include + +#include "stg/services.h" +#include "stg/service_conf.h" +#include "stg/locker.h" +#include "stg/store.h" +#include "stg/noncopyable.h" + +class ADMIN; + +class SERVICES_IMPL : private NONCOPYABLE, public SERVICES { +public: + SERVICES_IMPL(STORE * st); + virtual SERVICES_IMPL() {} + + int Add(const SERVICE_CONF & service, const ADMIN * admin); + int Del(const std::string & name, const ADMIN * admin); + int Change(const SERVICE_CONF & service, const ADMIN * admin); + bool FindService(const std::string & name, SERVICE_CONF & service); + bool ServiceExists(const std::string & name) const; + const std::string & GetStrError() const; + + int OpenSearch() const; + int SearchNext(int, SERVICE_CONF * service) const; + int CloseSearch(int) const; + +private: + typedef list::iterator srv_iter; + typedef list::const_iterator const_srv_iter; + + bool ReadServices(); + + std::list data; + STORE * store; + STG_LOGGER & WriteServLog; + mutable std::map searchDescriptors; + mutable unsigned int handle; + mutable pthread_mutex_t mutex; + std::string strError; +}; -- 2.44.2