2 #include <sys/socket.h>
3 #include <netinet/in.h>
13 #include <ostream> // xmlrpc-c devs have missed something :)
15 #include "stg/common.h"
16 #include "stg/admin.h"
17 #include "stg/module_settings.h"
18 #include "stg/settings.h"
19 #include "stg/plugin_creator.h"
21 #include "rpcconfig.h"
22 #include "info_methods.h"
23 #include "users_methods.h"
24 #include "tariffs_methods.h"
25 #include "admins_methods.h"
26 #include "messages_methods.h"
28 PLUGIN_CREATOR<RPC_CONFIG> rpcc;
30 RPC_CONFIG_SETTINGS::RPC_CONFIG_SETTINGS()
37 int RPC_CONFIG_SETTINGS::ParseSettings(const MODULE_SETTINGS & s)
41 std::vector<PARAM_VALUE>::const_iterator pvi;
44 pvi = std::find(s.moduleParams.begin(), s.moduleParams.end(), pv);
45 if (pvi == s.moduleParams.end())
47 errorStr = "Parameter \'Port\' not found.";
48 printfd(__FILE__, "Parameter 'Port' not found\n");
51 if (ParseIntInRange(pvi->value[0], 2, 65535, &p))
53 errorStr = "Cannot parse parameter \'Port\': " + errorStr;
54 printfd(__FILE__, "Cannot parse parameter 'Port'\n");
59 pv.param = "CookieTimeout";
60 pvi = std::find(s.moduleParams.begin(), s.moduleParams.end(), pv);
61 if (pvi == s.moduleParams.end())
63 cookieTimeout = 1800; // 30 * 60
67 if (str2x(pvi->value[0], cookieTimeout))
69 errorStr = "Incorrect value of CookieTimeout: \'" + pvi->value[0] + "\'";
70 printfd(__FILE__, "Incorrect value of 'CookieTimeout'\n");
80 return rpcc.GetPlugin();
83 RPC_CONFIG::RPC_CONFIG()
100 logger(GetPluginLogger(GetStgLogger(), "conf_rpc"))
104 RPC_CONFIG::~RPC_CONFIG()
110 int RPC_CONFIG::ParseSettings()
112 int ret = rpcConfigSettings.ParseSettings(settings);
115 errorStr = rpcConfigSettings.GetStrError();
120 void RPC_CONFIG::SetStgSettings(const SETTINGS * settings)
122 dayFee = settings->GetDayFee();
123 dirNames.erase(dirNames.begin(), dirNames.end());
124 for (size_t i = 0; i < DIR_NUM; ++i) {
125 dirNames.push_back(settings->GetDirName(i));
129 int RPC_CONFIG::Start()
134 fd = socket(AF_INET, SOCK_STREAM, 0);
137 errorStr = "Failed to create socket";
138 printfd(__FILE__, "Failed to create listening socket: %s\n", strerror(errno));
144 if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &flag, sizeof(flag)))
146 errorStr = "Setsockopt failed.";
147 printfd(__FILE__, "Setsockopt failed: %s\n", strerror(errno));
151 struct sockaddr_in addr;
152 addr.sin_family = AF_INET;
153 addr.sin_port = htons(rpcConfigSettings.GetPort());
154 addr.sin_addr.s_addr = inet_addr("0.0.0.0");
156 if (bind(fd, (struct sockaddr *)&addr, sizeof(addr)))
158 errorStr = "Failed to bind socket";
159 printfd(__FILE__, "Failed to bind listening socket: %s\n", strerror(errno));
165 errorStr = "Failed to listen socket";
166 printfd(__FILE__, "Failed to listen listening socket: %s\n", strerror(errno));
170 rpcServer = new xmlrpc_c::serverAbyss(
171 xmlrpc_c::serverAbyss::constrOpt()
172 .registryP(&rpcRegistry)
173 .logFileName("/var/log/stargazer_rpc.log")
177 if (pthread_create(&tid, NULL, Run, this))
179 errorStr = "Failed to create RPC thread";
180 printfd(__FILE__, "Failed to crate RPC thread\n");
187 int RPC_CONFIG::Stop()
190 for (int i = 0; i < 5 && !stopped; ++i)
192 struct timespec ts = {0, 200000000};
193 nanosleep(&ts, NULL);
199 printfd(__FILE__, "Failed to stop RPC thread\n");
200 errorStr = "Failed to stop RPC thread";
205 pthread_join(tid, NULL);
213 void * RPC_CONFIG::Run(void * rc)
216 sigfillset(&signalSet);
217 pthread_sigmask(SIG_BLOCK, &signalSet, NULL);
219 RPC_CONFIG * config = static_cast<RPC_CONFIG *>(rc);
221 config->stopped = false;
222 while (config->running)
224 if (WaitPackets(config->fd))
225 config->rpcServer->runOnce();
227 config->stopped = true;
232 bool RPC_CONFIG::GetAdminInfo(const std::string & cookie,
235 std::map<std::string,
236 ADMIN_INFO>::iterator it;
238 it = cookies.find(cookie);
240 if (it == cookies.end())
245 if (difftime(it->second.accessTime, time(NULL)) >
246 rpcConfigSettings.GetCookieTimeout())
252 // Update access time
253 time(&it->second.accessTime);
258 bool RPC_CONFIG::CheckAdmin(const std::string & login,
259 const std::string & password,
260 std::string * cookie)
262 ADMIN * admin = NULL;
264 if (!admins->Correct(login, password, &admin))
270 time(&info.accessTime);
272 info.priviledges = *admin->GetPriv();
273 *cookie = GetCookie();
274 cookies[*cookie] = info;
279 bool RPC_CONFIG::LogoutAdmin(const std::string & cookie)
281 std::map<std::string,
282 ADMIN_INFO>::iterator it;
284 it = cookies.find(cookie);
286 if (it == cookies.end())
296 std::string RPC_CONFIG::GetCookie() const
298 std::string charset("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890");
301 for (int i = 0; i < 64; ++i)
303 cookie += charset[rand() % charset.length()];
309 void RPC_CONFIG::InitiateRegistry()
312 xmlrpc_c::methodPtr const methodInfoPtr(new METHOD_INFO(
318 rpcRegistry.addMethod("stargazer.info", methodInfoPtr);
320 xmlrpc_c::methodPtr const methodLoginPtr(new METHOD_LOGIN(
323 rpcRegistry.addMethod("stargazer.login", methodLoginPtr);
325 xmlrpc_c::methodPtr const methodLogoutPtr(new METHOD_LOGOUT(
328 rpcRegistry.addMethod("stargazer.logout", methodLogoutPtr);
330 xmlrpc_c::methodPtr const methodGetUserPtr(new METHOD_USER_GET(
334 rpcRegistry.addMethod("stargazer.get_user", methodGetUserPtr);
336 xmlrpc_c::methodPtr const methodAddUserPtr(new METHOD_USER_ADD(
341 rpcRegistry.addMethod("stargazer.add_user", methodAddUserPtr);
343 xmlrpc_c::methodPtr const methodDelUserPtr(new METHOD_USER_DEL(
348 rpcRegistry.addMethod("stargazer.del_user", methodDelUserPtr);
350 xmlrpc_c::methodPtr const methodGetUsersPtr(new METHOD_USERS_GET(
354 rpcRegistry.addMethod("stargazer.get_users", methodGetUsersPtr);
356 xmlrpc_c::methodPtr const methodChgUserPtr(new METHOD_USER_CHG(
363 rpcRegistry.addMethod("stargazer.chg_user", methodChgUserPtr);
365 xmlrpc_c::methodPtr const methodAddCashPtr(new METHOD_USER_CASH_ADD(
371 rpcRegistry.addMethod("stargazer.add_user_cash", methodAddCashPtr);
373 xmlrpc_c::methodPtr const methodSetCashPtr(new METHOD_USER_CASH_SET(
379 rpcRegistry.addMethod("stargazer.set_user_cash", methodSetCashPtr);
381 xmlrpc_c::methodPtr const methodTariffChangePtr(new METHOD_USER_TARIFF_CHANGE(
388 rpcRegistry.addMethod("stargazer.chg_user_tariff", methodTariffChangePtr);
390 xmlrpc_c::methodPtr const methodGetTariffPtr(new METHOD_TARIFF_GET(
394 rpcRegistry.addMethod("stargazer.get_tariff", methodGetTariffPtr);
396 xmlrpc_c::methodPtr const methodChgTariffPtr(new METHOD_TARIFF_CHG(
401 rpcRegistry.addMethod("stargazer.chg_tariff", methodChgTariffPtr);
403 xmlrpc_c::methodPtr const methodGetTariffsPtr(new METHOD_TARIFFS_GET(
407 rpcRegistry.addMethod("stargazer.get_tariffs", methodGetTariffsPtr);
409 xmlrpc_c::methodPtr const methodAddTariffPtr(new METHOD_TARIFF_ADD(
414 rpcRegistry.addMethod("stargazer.add_tariff", methodAddTariffPtr);
416 xmlrpc_c::methodPtr const methodDelTariffPtr(new METHOD_TARIFF_DEL(
422 rpcRegistry.addMethod("stargazer.del_tariff", methodDelTariffPtr);
424 xmlrpc_c::methodPtr const methodGetAdminPtr(new METHOD_ADMIN_GET(
428 rpcRegistry.addMethod("stargazer.get_admin", methodGetAdminPtr);
430 xmlrpc_c::methodPtr const methodAddAdminPtr(new METHOD_ADMIN_ADD(
434 rpcRegistry.addMethod("stargazer.add_admin", methodAddAdminPtr);
436 xmlrpc_c::methodPtr const methodDelAdminPtr(new METHOD_ADMIN_DEL(
440 rpcRegistry.addMethod("stargazer.del_admin", methodDelAdminPtr);
442 xmlrpc_c::methodPtr const methodChgAdminPtr(new METHOD_ADMIN_CHG(
446 rpcRegistry.addMethod("stargazer.chg_admin", methodChgAdminPtr);
448 xmlrpc_c::methodPtr const methodGetAdminsPtr(new METHOD_ADMINS_GET(
452 rpcRegistry.addMethod("stargazer.get_admins", methodGetAdminsPtr);
454 xmlrpc_c::methodPtr const methodSendMessagePtr(new METHOD_MESSAGE_SEND(
458 rpcRegistry.addMethod("stargazer.send_user_message", methodSendMessagePtr);
460 xmlrpc_c::methodPtr const methodGetOnlinIPsPtr(new METHOD_GET_ONLINE_IPS(
464 rpcRegistry.addMethod("stargazer.get_online_ips", methodGetOnlinIPsPtr);