$Date: 2010/03/25 14:37:43 $
*/
-#include <unistd.h>
-#include <getopt.h>
-#include <iconv.h>
-#include <langinfo.h>
+#include "request.h"
+#include "common_sg.h"
+#include "sg_error_codes.h"
+
+#include "stg/user_conf.h"
+#include "stg/user_stat.h"
+#include "stg/common.h"
#include <cerrno>
#include <clocale>
#include <cstdio>
+#include <cstdlib>
#include <cstring>
#include <string>
#include <sstream>
-#include "stg/common.h"
-#include "request.h"
-#include "common_sg.h"
-#include "sg_error_codes.h"
+#include <unistd.h>
+#include <getopt.h>
+#include <iconv.h>
+#include <langinfo.h>
namespace
{
template <typename T>
-struct ARRAY_TYPE;
+struct ARRAY_TYPE
+{
+typedef typename T::value_type type;
+};
template <typename T>
struct ARRAY_TYPE<T[]>
return true;
}
+void Usage();
+void UsageConnection();
+void UsageAdmins();
+void UsageTariffs();
+void UsageUsers();
+void UsageServices();
+void UsageCorporations();
+
} // namespace anonymous
time_t stgTime;
{0, 0, 0, 0}};
//-----------------------------------------------------------------------------
-double ParseCash(const char * c, string * message)
+CASH_INFO ParseCash(const char * str)
{
//-c 123.45:log message
-double cash;
-char * msg;
-char * str;
-str = new char[strlen(c) + 1];
-
-strncpy(str, c, strlen(c));
-str[strlen(c)] = 0;
-
-msg = strchr(str, ':');
-
-if (msg)
+std::string cashString;
+std::string message;
+const char * pos = strchr(str, ':');
+if (pos != NULL)
{
- *message = msg + 1;
- str[msg - str] = 0;
+ cashString.append(str, pos);
+ message.append(pos + 1);
}
else
- *message = "";
+ cashString = str;
-if (strtodouble2(str, cash) != 0)
+double cash = 0;
+if (strtodouble2(cashString, cash) != 0)
{
- printf("Incorrect cash value %s\n", c);
+ printf("Incorrect cash value %s\n", str);
exit(PARAMETER_PARSING_ERR_CODE);
}
-delete[] str;
-return cash;
+return CASH_INFO(cash, message);
}
//-----------------------------------------------------------------------------
double ParseCredit(const char * c)
return dp[0] - '0';
}
//-----------------------------------------------------------------------------
-string ParseTariff(const char * t, int &chgType)
+void ParseTariff(const char * str, RESETABLE<std::string> & tariffName, RESETABLE<std::string> & nextTariff)
{
-int l = strlen(t);
-char * s;
-s = new char[l];
-char * s1, * s2;
-string ss;
-
-strcpy(s, t);
-
-s1 = strtok(s, ":");
-
-if (strlen(s1) >= TARIFF_NAME_LEN)
+const char * pos = strchr(str, ':');
+if (pos != NULL)
{
- printf("Tariff name too big %s\n", s1);
- exit(PARAMETER_PARSING_ERR_CODE);
- }
-
-//*tariff = s;
-
-if (CheckLogin(s1))
- {
- printf("Incorrect tariff value %s\n", t);
- exit(PARAMETER_PARSING_ERR_CODE);
- }
-
-s2 = strtok(NULL, ":");
-
-chgType = -1;
-
-if (s2 == NULL)
- {
- chgType = TARIFF_NOW;
- ss = s;
- delete[] s;
- return ss;
- }
-
-
-if (strcmp(s2, "now") == 0)
- chgType = TARIFF_NOW;
-
-if (strcmp(s2, "delayed") == 0)
- chgType = TARIFF_DEL;
-
-if (strcmp(s2, "recalc") == 0)
- chgType = TARIFF_REC;
-
-if (chgType < 0)
- {
- printf("Incorrect tariff value %s\n", t);
- exit(PARAMETER_PARSING_ERR_CODE);
+ std::string tariff(str, pos);
+ if (strcmp(pos + 1, "now") == 0)
+ tariffName = tariff;
+ else if (strcmp(pos + 1, "delayed") == 0)
+ nextTariff = tariff;
+ else
+ {
+ printf("Incorrect tariff value '%s'. Should be '<tariff>', '<tariff>:now' or '<tariff>:delayed'.\n", str);
+ exit(PARAMETER_PARSING_ERR_CODE);
+ }
}
-
-ss = s;
-delete[] s;
-return ss;
+else
+ tariffName = str;
}
//-----------------------------------------------------------------------------
time_t ParseCreditExpire(const char * str)
int missedOptionArg = false;
+USER_CONF_RES conf;
+USER_STAT_RES stat;
while (1)
{
int option_index = -1;
break;
case 'o': //change user password
- req.usrPasswd = ParsePassword(optarg);
+ conf.password = ParsePassword(optarg);
break;
case 'u': //user
break;
case 'c': //add cash
- req.cash = ParseCash(optarg, &req.message);
+ stat.cashAdd = ParseCash(optarg);
break;
case 'v': //set cash
- req.setCash = ParseCash(optarg, &req.message);
+ stat.cashSet = ParseCash(optarg);
break;
case 'r': //credit
- req.credit = ParseCredit(optarg);
+ conf.credit = ParseCredit(optarg);
break;
case 'E': //credit expire
- req.creditExpire = ParseCreditExpire(optarg);
+ conf.creditExpire = ParseCreditExpire(optarg);
break;
case 'd': //down
- req.down = ParseDownPassive(optarg);
+ conf.disabled = ParseDownPassive(optarg);
break;
case 'i': //passive
- req.passive = ParseDownPassive(optarg);
+ conf.passive = ParseDownPassive(optarg);
break;
case 't': //tariff
- req.tariff = ParseTariff(optarg, req.chgTariff);
+ ParseTariff(optarg, conf.tariffName, conf.nextTariff);
break;
case 'm': //message
break;
case 'e': //Prepaid Traffic
- req.prepaidTraff = ParsePrepaidTraffic(optarg);
+ stat.freeMb = ParsePrepaidTraffic(optarg);
break;
case 'n': //Create User
case 'N': //Note
ParseAnyString(optarg, &str, "koi8-ru");
- req.note = str;
+ conf.note = str;
break;
case 'A': //nAme
ParseAnyString(optarg, &str, "koi8-ru");
- req.name = str;
+ conf.realName = str;
break;
case 'D': //aDdress
ParseAnyString(optarg, &str, "koi8-ru");
- req.address = str;
+ conf.address = str;
break;
case 'L': //emaiL
ParseAnyString(optarg, &str, "koi8-ru");
- req.email = str;
- //printf("EMAIL=%s\n", optarg);
+ conf.email = str;
break;
case 'P': //phone
ParseAnyString(optarg, &str);
- req.phone = str;
+ conf.phone = str;
break;
case 'G': //Group
ParseAnyString(optarg, &str, "koi8-ru");
- req.group = str;
+ conf.group = str;
break;
case 'I': //IP-address of user
ParseAnyString(optarg, &str);
- req.ips = str;
+ conf.ips = StrToIPS(str);
break;
case 'S':
- req.disableDetailStat = ParseDownPassive(optarg);
+ conf.disabledDetailStat = ParseDownPassive(optarg);
break;
case 'O':
- req.alwaysOnline = ParseDownPassive(optarg);
+ conf.alwaysOnline = ParseDownPassive(optarg);
break;
case 500: //U
- SetArrayItem(req.sessionUpload, optarg, ParseTraff(argv[optind++]));
- //req.sessionUpload[optarg] = ParseTraff(argv[optind++]);
+ SetArrayItem(stat.sessionUp, optarg, ParseTraff(argv[optind++]));
break;
case 501:
- SetArrayItem(req.sessionDownload, optarg, ParseTraff(argv[optind++]));
- //req.sessionDownload[optarg] = ParseTraff(argv[optind++]);
+ SetArrayItem(stat.sessionDown, optarg, ParseTraff(argv[optind++]));
break;
case 502:
- SetArrayItem(req.monthUpload, optarg, ParseTraff(argv[optind++]));
- //req.monthUpload[optarg] = ParseTraff(argv[optind++]);
+ SetArrayItem(stat.monthUp, optarg, ParseTraff(argv[optind++]));
break;
case 503:
- SetArrayItem(req.monthDownload, optarg, ParseTraff(argv[optind++]));
- //req.monthDownload[optarg] = ParseTraff(argv[optind++]);
+ SetArrayItem(stat.monthDown, optarg, ParseTraff(argv[optind++]));
break;
case 700: //UserData
ParseAnyString(argv[optind++], &str);
- SetArrayItem(req.userData, optarg, str);
- //req.userData[optarg] = str;
+ SetArrayItem(conf.userdata, optarg, str);
break;
case '?':
char rstr[rLen];
memset(rstr, 0, rLen);
-CreateRequestSet(&req, rstr);
if (isMessage)
- return ProcessSendMessage(req.server.data(), req.port.data(), req.admLogin.data(), req.admPasswd.data(), rstr);
+ return ProcessSendMessage(req.server.data(), req.port.data(), req.admLogin.data(), req.admPasswd.data(), req.login.data(), req.usrMsg.data());
-return ProcessSetUser(req.server.data(), req.port.data(), req.admLogin.data(), req.admPasswd.data(), rstr);
+return ProcessSetUser(req.server.data(), req.port.data(), req.admLogin.data(), req.admPasswd.data(), req.login.data(), conf, stat);
}
//-----------------------------------------------------------------------------
int main(int argc, char **argv)
{
+Usage();
+exit(0);
if (argc <= 2)
{
UsageConf();
}
//-----------------------------------------------------------------------------
+namespace
+{
+
+void Usage()
+{
+std::cout << "sgconf is the Stargazer management utility.\n\n"
+ << "Usage:\n"
+ << "\tsgconf [options]\n\n"
+ << "General options:\n"
+ << "\t-f, --config <config file>\t\toverride default config file (default: \"~/.config/stg/sgconf.conf\")\n"
+ << "\t-h, --help\t\t\t\tshow this help and exit\n"
+ << "\t-v, --version\t\t\t\tshow version information and exit\n\n";
+UsageConnection();
+UsageAdmins();
+UsageTariffs();
+UsageUsers();
+UsageServices();
+UsageCorporations();
+}
+//-----------------------------------------------------------------------------
+void UsageConnection()
+{
+std::cout << "Connection options:\n"
+ << "\t-s, --server <address>\t\t\thost to connect (ip or domain name, default: \"localhost\")\n"
+ << "\t-p, --port <port>\t\t\tport to connect (default: \"5555\")\n"
+ << "\t-u, --username <username>\t\tadministrative login (default: \"admin\")\n"
+ << "\t-w, --userpass <password>\t\tpassword for administrative login\n"
+ << "\t-c, --connect <connection string>\tconnection params as a single string in format: <login>:<password>@<host>:<port>\n\n";
+}
+//-----------------------------------------------------------------------------
+void UsageAdmins()
+{
+std::cout << "Admins management options:\n"
+ << "\t--get-admins\t\t\t\tget a list of admins (subsequent options will define what to show)\n"
+ << "\t\t--login\t\t\t\tshow admin's login\n"
+ << "\t\t--priv\t\t\t\tshow admin's priviledges\n"
+ << "\t--get-admin\t\t\t\tget the information about admin\n"
+ << "\t\t--login <login>\t\t\t\tlogin of the admin to show\n"
+ << "\t\t--priv\t\t\t\tshow admin's priviledges\n"
+ << "\t--add-admin\t\t\t\tadd a new admin\n"
+ << "\t\t--login <login>\t\t\tlogin of the admin to add\n"
+ << "\t\t--password <password>\t\tpassword of the admin to add\n"
+ << "\t\t--priv <priv number>\t\tpriviledges of the admin to add\n"
+ << "\t--del-admin\t\t\t\tdelete an existing admin\n"
+ << "\t\t--login <login>\t\t\tlogin of the admin to delete\n"
+ << "\t--chg-admin\t\t\t\tchange an existing admin\n"
+ << "\t\t--login <login>\t\t\tlogin of the admin to change\n"
+ << "\t\t--priv <priv number>\t\tnew priviledges\n\n";
+}
+//-----------------------------------------------------------------------------
+void UsageTariffs()
+{
+std::cout << "Tariffs management options:\n"
+ << "\t--get-tariffs\t\t\t\tget a list of tariffs (subsequent options will define what to show)\n"
+ << "\t--get-tariff\t\t\t\tget the information about tariff\n"
+ << "\t--add-tariff\t\t\t\tadd a new tariff\n"
+ << "\t--del-tariff\t\t\t\tdelete an existing tariff\n"
+ << "\t--chg-tariff\t\t\t\tchange an existing tariff\n"
+ << "\n\n";
+}
+//-----------------------------------------------------------------------------
+void UsageUsers()
+{
+std::cout << "Users management options:\n"
+ << "\t--get-users\t\t\t\tget a list of users (subsequent options will define what to show)\n"
+ << "\t--get-user\t\t\t\tget the information about user\n"
+ << "\t--add-user\t\t\t\tadd a new user\n"
+ << "\t--del-user\t\t\t\tdelete an existing user\n"
+ << "\t--chg-user\t\t\t\tchange an existing user\n"
+ << "\t--check-user\t\t\t\tcheck credentials is valid\n"
+ << "\t--send-message\t\t\t\tsend a message to a user\n"
+ << "\n\n";
+}
+//-----------------------------------------------------------------------------
+void UsageServices()
+{
+std::cout << "Services management options:\n"
+ << "\t--get-services\t\t\t\tget a list of services (subsequent options will define what to show)\n"
+ << "\t--get-service\t\t\t\tget the information about service\n"
+ << "\t--add-service\t\t\t\tadd a new service\n"
+ << "\t--del-service\t\t\t\tdelete an existing service\n"
+ << "\t--chg-service\t\t\t\tchange an existing service\n"
+ << "\n\n";
+}
+//-----------------------------------------------------------------------------
+void UsageCorporations()
+{
+std::cout << "Corporations management options:\n"
+ << "\t--get-corporations\t\t\tget a list of corporations (subsequent options will define what to show)\n"
+ << "\t\t--name\t\t\t\tshow corporation's name\n"
+ << "\t\t--cash\t\t\t\tshow corporation's cash\n"
+ << "\t--get-corp\t\t\t\tget the information about corporation\n"
+ << "\t\t--name <name>\t\t\tname of the corporation to show\n"
+ << "\t\t--cash\t\t\t\tshow corporation's cash\n"
+ << "\t--add-corp\t\t\t\tadd a new corporation\n"
+ << "\t\t--name <name>\t\t\tname of the corporation to add\n"
+ << "\t\t--cash <cash>\t\t\tinitial corporation's cash (default: \"0\")\n"
+ << "\t--del-corp\t\t\t\tdelete an existing corporation\n"
+ << "\t\t--name <name>\t\t\tname of the corporation to delete\n"
+ << "\t--chg-corp\t\t\t\tchange an existing corporation\n"
+ << "\t\t--name <name>\t\t\tname of the corporation to change\n"
+ << "\t\t--add-cash <amount>[:<message>]\tadd cash to the corporation's account and optional comment message\n"
+ << "\t\t--set-cash <cash>[:<message>]\tnew corporation's cash and optional comment message\n"
+ << "\n\n";
+}
+
+} // namespace anonymous