projects/stargazer/services_impl.cpp

   1 /*
   2  *    This program is free software; you can redistribute it and/or modify
   3  *    it under the terms of the GNU General Public License as published by
   4  *    the Free Software Foundation; either version 2 of the License, or
   5  *    (at your option) any later version.
   6  *
   7  *    This program is distributed in the hope that it will be useful,
   8  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
   9  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  10  *    GNU General Public License for more details.
  11  *
  12  *    You should have received a copy of the GNU General Public License
  13  *    along with this program; if not, write to the Free Software
  14  *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  15  */
  16
  17 /*
  18  *    Author : Maxim Mamontov <faust@stargazer.dp.ua>
  19  */
  20
  21 #include <cerrno>
  22 #include <cassert>
  23 #include <algorithm>
  24
  25 #include "stg/admin.h"
  26 #include "stg/common.h"
  27 #include "services_impl.h"
  28
  29 //-----------------------------------------------------------------------------
  30 SERVICES_IMPL::SERVICES_IMPL(STORE * st)
  31     : SERVICES(),
  32       data(),
  33       store(st),
  34       WriteServLog(GetStgLogger()),
  35       searchDescriptors(),
  36       handle(0)
  37 {
  38 pthread_mutex_init(&mutex, NULL);
  39 Read();
  40 }
  41 //-----------------------------------------------------------------------------
  42 int SERVICES_IMPL::Add(const SERVICE_CONF & service, const ADMIN * admin)
  43 {
  44 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
  45 const PRIV * priv = admin->GetPriv();
  46
  47 if (!priv->serviceChg)
  48     {
  49     string s = admin->GetLogStr() + " Add service \'" + service.name + "\'. Access denied.";
  50     strError = "Access denied.";
  51     WriteServLog(s.c_str());
  52     return -1;
  53     }
  54
  55 srv_iter si(find(data.begin(), data.end(), service));
  56
  57 if (si != data.end())
  58     {
  59     strError = "Service \'" + service.name + "\' cannot not be added. Service already exist.";
  60     WriteServLog("%s %s", admin->GetLogStr().c_str(), strError.c_str());
  61
  62     return -1;
  63     }
  64
  65 data.push_back(service);
  66
  67 if (store->AddService(service.name) == 0)
  68     {
  69     WriteServLog("%s Service \'%s\' added.",
  70                  admin->GetLogStr().c_str(), service.name.c_str());
  71     return 0;
  72     }
  73
  74 strError = "Service \'" + service.name + "\' was not added. Error: " + store->GetStrError();
  75 WriteServLog("%s %s", admin->GetLogStr().c_str(), strError.c_str());
  76
  77 return -1;
  78 }
  79 //-----------------------------------------------------------------------------
  80 int SERVICES_IMPL::Del(const string & name, const ADMIN * admin)
  81 {
  82 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
  83 const PRIV * priv = admin->GetPriv();
  84
  85 if (!priv->serviceChg)
  86     {
  87     string s = admin->GetLogStr() + " Delete service \'" + name + "\'. Access denied.";
  88     strError = "Access denied.";
  89     WriteServLog(s.c_str());
  90     return -1;
  91     }
  92
  93 srv_iter si(find(data.begin(), data.end(), SERVICE_CONF(name)));
  94
  95 if (si == data.end())
  96     {
  97     strError = "Service \'" + name + "\' cannot be deleted. Service does not exist.";
  98     WriteServLog("%s %s", admin->GetLogStr().c_str(), strError.c_str());
  99     return -1;
 100     }
 101
 102 map<int, const_srv_iter>::iterator csi;
 103 csi = searchDescriptors.begin();
 104 while (csi != searchDescriptors.end())
 105     {
 106     if (csi->second == si)
 107         (csi->second)++;
 108     csi++;
 109     }
 110
 111 data.remove(*si);
 112 if (store->DelService(name) < 0)
 113     {
 114     strError = "Service \'" + name + "\' was not deleted. Error: " + store->GetStrError();
 115     WriteServLog("%s %s", admin->GetLogStr().c_str(), strError.c_str());
 116
 117     return -1;
 118     }
 119
 120 WriteServLog("%s Service \'%s\' deleted.", admin->GetLogStr().c_str(), name.c_str());
 121 return 0;
 122 }
 123 //-----------------------------------------------------------------------------
 124 int SERVICES_IMPL::Change(const SERVICE_CONF & service, const ADMIN * admin)
 125 {
 126 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 127 const PRIV * priv = admin->GetPriv();
 128
 129 if (!priv->serviceChg)
 130     {
 131     string s = admin->GetLogStr() + " Change service \'" + service.name + "\'. Access denied.";
 132     strError = "Access denied.";
 133     WriteServLog(s.c_str());
 134     return -1;
 135     }
 136
 137 srv_iter si(find(data.begin(), data.end(), service));
 138
 139 if (si == data.end())
 140     {
 141     strError = "Service \'" + service.name + "\' cannot be changed " + ". Service does not exist.";
 142     WriteServLog("%s %s", admin->GetLogStr().c_str(), strError.c_str());
 143     return -1;
 144     }
 145
 146 *si = service;
 147 if (store->SaveService(service))
 148     {
 149     WriteServLog("Cannot write service %s.", service.name.c_str());
 150     WriteServLog("%s", store->GetStrError().c_str());
 151     return -1;
 152     }
 153
 154 WriteServLog("%s Service \'%s\' changed.",
 155              admin->GetLogStr().c_str(), service.name.c_str());
 156
 157 return 0;
 158 }
 159 //-----------------------------------------------------------------------------
 160 bool SERVICES_IMPL::Read()
 161 {
 162 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 163 vector<string> servicesList;
 164 if (store->GetServicesList(&servicesList) < 0)
 165     {
 166     WriteServLog(store->GetStrError().c_str());
 167     return true;
 168     }
 169
 170 for (size_t i = 0; i < servicesList.size(); i++)
 171     {
 172     SERVICE_CONF service;
 173
 174     if (store->RestoreService(&service, servicesList[i]))
 175         {
 176         WriteServLog(store->GetStrError().c_str());
 177         return true;
 178         }
 179
 180     data.push_back(service);
 181     }
 182 return false;
 183 }
 184 //-----------------------------------------------------------------------------
 185 bool SERVICES_IMPL::Find(const string & name, SERVICE_CONF * service)
 186 {
 187 assert(service != NULL && "Pointer to service is not null");
 188
 189 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 190 if (data.empty())
 191     return false;
 192
 193 srv_iter si(find(data.begin(), data.end(), SERVICE_CONF(name)));
 194
 195 if (si != data.end())
 196     {
 197     *service = *si;
 198     return false;
 199     }
 200
 201 return true;
 202 }
 203 //-----------------------------------------------------------------------------
 204 bool SERVICES_IMPL::Exists(const string & name) const
 205 {
 206 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 207 if (data.empty())
 208     {
 209     printfd(__FILE__, "no admin in system!\n");
 210     return true;
 211     }
 212
 213 const_srv_iter si(find(data.begin(), data.end(), SERVICE_CONF(name)));
 214
 215 if (si != data.end())
 216     return true;
 217
 218 return false;
 219 }
 220 //-----------------------------------------------------------------------------
 221 int SERVICES_IMPL::OpenSearch() const
 222 {
 223 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 224 handle++;
 225 searchDescriptors[handle] = data.begin();
 226 return handle;
 227 }
 228 //-----------------------------------------------------------------------------
 229 int SERVICES_IMPL::SearchNext(int h, SERVICE_CONF * service) const
 230 {
 231 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 232 if (searchDescriptors.find(h) == searchDescriptors.end())
 233     {
 234     WriteServLog("SERVICES. Incorrect search handle.");
 235     return -1;
 236     }
 237
 238 if (searchDescriptors[h] == data.end())
 239     return -1;
 240
 241 *service = *searchDescriptors[h]++;
 242
 243 return 0;
 244 }
 245 //-----------------------------------------------------------------------------
 246 int SERVICES_IMPL::CloseSearch(int h) const
 247 {
 248 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 249 if (searchDescriptors.find(h) != searchDescriptors.end())
 250     {
 251     searchDescriptors.erase(searchDescriptors.find(h));
 252     return 0;
 253     }
 254
 255 WriteServLog("SERVICES. Incorrect search handle.");
 256 return -1;
 257 }
 258 //-----------------------------------------------------------------------------