set.c 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825
  1. /*
  2. * set.c
  3. *
  4. */
  5. #include "common.h"
  6. #include "set.h"
  7. #include "shell.h"
  8. #include "botmsg.h"
  9. #include "chanprog.h"
  10. #include "misc.h"
  11. #include "src/mod/server.mod/server.h"
  12. #include "src/mod/channels.mod/channels.h"
  13. #include "src/mod/ctcp.mod/ctcp.h"
  14. #include "users.h"
  15. #include "userrec.h"
  16. #include "userent.h"
  17. #include "rfc1459.h"
  18. int set_noshare = 0;
  19. int dcc_autoaway = 1800;
  20. bool auth_obscure = 0;
  21. bool auth_chan = 1;
  22. char alias[1024] = "bc botcmd,bl botcmd ?,+list set +,-list set -,list set list";
  23. char auth_key[51] = "";
  24. char auth_prefix[2] = "";
  25. int badprocess = DET_IGNORE;
  26. bool dccauth = 0;
  27. char *def_chanset = "+enforcebans +dynamicbans +userbans -bitch -protectops +cycle -inactive +userexempts -dynamicexempts +userinvites -dynamicinvites -nodesynch -closed -take -voice -private -fastop";
  28. int cloak_script = 0;
  29. rate_t close_threshold = { 0, 0 };
  30. int fight_threshold;
  31. int fork_interval;
  32. int hijack = DET_DIE;
  33. int in_bots = 2;
  34. int kill_threshold;
  35. int lag_threshold = 15;
  36. int login = DET_WARN;
  37. char motd[512] = "";
  38. char msgident[21] = "";
  39. char msginvite[21] = "";
  40. char msgop[21] = "";
  41. char msgpass[21] = "";
  42. int op_bots = 1;
  43. rate_t op_requests = { 2, 5 };
  44. char process_list[1024] = "";
  45. int promisc = DET_WARN;
  46. int trace = DET_DIE;
  47. bool offensive_bans = 1;
  48. bool manop_warn = 1;
  49. char homechan[51] = "";
  50. static variable_t vars[] = {
  51. VAR("alias", alias, sizeof(alias), VAR_STRING|VAR_LIST|VAR_NOLOC|VAR_PERM),
  52. VAR("auth-chan", &auth_chan, 0, VAR_INT|VAR_BOOL|VAR_NOLHUB),
  53. VAR("auth-key", auth_key, sizeof(auth_key), VAR_STRING|VAR_PERM),
  54. VAR("auth-prefix", auth_prefix, sizeof(auth_prefix), VAR_STRING|VAR_NOLHUB|VAR_PERM),
  55. VAR("auth-obscure", &auth_obscure, 0, VAR_INT|VAR_BOOL),
  56. VAR("dcc-autoaway", &dcc_autoaway, 0, VAR_INT|VAR_NOLOC),
  57. VAR("bad-process", &badprocess, 0, VAR_INT|VAR_DETECTED),
  58. VAR("dccauth", &dccauth, 0, VAR_INT|VAR_BOOL),
  59. VAR("chanset", glob_chanset, sizeof(glob_chanset), VAR_STRING|VAR_CHANSET|VAR_NOLHUB),
  60. VAR("cloak-script", &cloak_script, 0, VAR_INT|VAR_CLOAK|VAR_NOLHUB),
  61. VAR("close-threshold", &close_threshold, 0, VAR_RATE|VAR_NOLOC),
  62. VAR("fight-threshold", &fight_threshold, 0, VAR_INT|VAR_NOLOC),
  63. VAR("flood-msg", &flood_msg, 0, VAR_RATE|VAR_NOLHUB),
  64. VAR("flood-ctcp", &flood_ctcp, 0, VAR_RATE|VAR_NOLHUB),
  65. VAR("flood-g", &flood_g, 0, VAR_RATE|VAR_NOLHUB),
  66. VAR("fork-interval", &fork_interval, 0, VAR_INT),
  67. VAR("hijack", &hijack, 0, VAR_INT|VAR_DETECTED|VAR_PERM),
  68. VAR("homechan", homechan, sizeof(homechan), VAR_STRING|VAR_NOLOC|VAR_HIDE),
  69. VAR("in-bots", &in_bots, 0, VAR_INT|VAR_NOLOC),
  70. VAR("kill-threshold", &kill_threshold, 0, VAR_INT|VAR_NOLOC),
  71. VAR("lag-threshold", &lag_threshold, 0, VAR_INT|VAR_NOLHUB),
  72. VAR("login", &login, 0, VAR_INT|VAR_DETECTED),
  73. VAR("manop-warn", &manop_warn, 0, VAR_INT|VAR_BOOL|VAR_NOLHUB),
  74. VAR("mean-kicks", &offensive_bans, 0, VAR_INT|VAR_BOOL|VAR_NOLHUB),
  75. VAR("motd", motd, sizeof(motd), VAR_STRING|VAR_HIDE|VAR_NOLOC),
  76. VAR("msg-ident", msgident, sizeof(msgident), VAR_STRING|VAR_NOLHUB),
  77. VAR("msg-invite", msginvite, sizeof(msginvite), VAR_STRING|VAR_NOLHUB),
  78. VAR("msg-op", msgop, sizeof(msgop), VAR_STRING|VAR_NOLHUB),
  79. VAR("msg-pass", msgpass, sizeof(msgpass), VAR_STRING|VAR_NOLHUB),
  80. VAR("nick", origbotname, sizeof(origbotname), VAR_STRING|VAR_NOLHUB|VAR_NICK|VAR_NODEF|VAR_NOGHUB),
  81. VAR("op-bots", &op_bots, 0, VAR_INT|VAR_NOLOC),
  82. VAR("op-requests", &op_requests, 0, VAR_RATE|VAR_NOLOC),
  83. VAR("process-list", process_list, sizeof(process_list), VAR_STRING|VAR_LIST),
  84. VAR("promisc", &promisc, 0, VAR_INT|VAR_DETECTED),
  85. VAR("realname", botrealname, sizeof(botrealname), VAR_STRING|VAR_NOLHUB),
  86. VAR("server-port", &default_port, 0, VAR_INT|VAR_NOLHUB),
  87. VAR("servers", &serverlist, 0, VAR_SERVERS|VAR_LIST|VAR_SHUFFLE|VAR_NOLHUB),
  88. VAR("servers6", &serverlist, 0, VAR_SERVERS|VAR_LIST|VAR_SHUFFLE|VAR_NOLHUB),
  89. VAR("trace", &trace, 0, VAR_INT|VAR_DETECTED),
  90. {NULL, NULL, 0, 0, NULL, NULL, 0}
  91. };
  92. static bool use_server_type(const char *name)
  93. {
  94. if (!conf.bot->hub) {
  95. if (!strcmp(name, "servers")) {
  96. if (conf.bot->net.host6 || conf.bot->net.ip6) /* we want to use the servers6 entry. */
  97. return 0;
  98. } else if (!strcmp(name, "servers6")) {
  99. if (!conf.bot->net.host6 && !conf.bot->net.ip6) /* we probably want to use the normal server list.. */
  100. return 0;
  101. }
  102. }
  103. return 1;
  104. }
  105. const char *var_find_by_mem(void *mem)
  106. {
  107. int i = 0;
  108. for (i = 0; vars[i].name; i++) {
  109. if (vars[i].mem == mem)
  110. return vars[i].name;
  111. }
  112. return "";
  113. }
  114. /* set the ptr to the new data */
  115. static bool var_set_mem(variable_t *var, const char *datain)
  116. {
  117. char *data = (datain && datain[0]) ? strdup(datain) : NULL, *datap = data;
  118. sdprintf("var (mem): %s -> %s", var->name, datain);
  119. if (data && var->flags & VAR_SHUFFLE) {
  120. // char *datadup = strdup(data);
  121. // shuffle(datadup, ",");
  122. // data = datadup;
  123. // freedata++;
  124. shuffle(data, ",");
  125. }
  126. /* figure out it's type and set it's variable to the data */
  127. if (var->flags & VAR_INT) {
  128. bool isnumber = 0;
  129. int number = 0;
  130. isnumber = data ? str_isdigit(data) : 0;
  131. if (isnumber)
  132. number = atoi(data);
  133. else if (!isnumber && (var->flags & VAR_DETECTED)) {
  134. number = data ? det_translate(data) : DET_IGNORE;
  135. #ifndef DEBUG
  136. if (number < 2 && !strcmp(var->name, "trace"))
  137. number = DET_DIE;
  138. #endif
  139. }
  140. else if (!isnumber)
  141. number = 0;
  142. if (var->flags & VAR_CLOAK && !conf.bot->hub) {
  143. if (number == 0)
  144. number = randint(CLOAK_COUNT) + 1;
  145. }
  146. *(int *) (var->mem) = number;
  147. if (var->flags & VAR_CLOAK && !conf.bot->hub)
  148. scriptchanged();
  149. } else if (var->flags & VAR_BOOL) {
  150. if (data && str_isdigit(data)) {
  151. int num = atoi(data);
  152. if (num == 0 || num == 1)
  153. *(bool *) (var->mem) = (bool) num;
  154. } else if (!data) {
  155. *(bool *) (var->mem) = 0;
  156. } else
  157. goto end;
  158. } else if (var->flags & VAR_STRING) {
  159. if (data)
  160. strlcpy((char *) var->mem, data, var->size);
  161. else
  162. ((char *) var->mem)[0] = 0;
  163. if (var->flags & VAR_NICK && !conf.bot->hub) {
  164. if (!data)
  165. strlcpy((char *) var->mem, conf.bot->nick, var->size);
  166. if (server_online && rfc_casecmp(botname, (char *) var->mem))
  167. dprintf(DP_SERVER, "NICK %s\n", (char *) var->mem);
  168. }
  169. } else if (var->flags & VAR_RATE) {
  170. char *p = NULL;
  171. if (data && (p = strchr(data, ':'))) {
  172. *p = 0;
  173. p++;
  174. (*(rate_t *) (var->mem)).count = atoi(data);
  175. (*(rate_t *) (var->mem)).time = atoi(p);
  176. } else if (!data) {
  177. (*(rate_t *) (var->mem)).count = 0;
  178. (*(rate_t *) (var->mem)).time = 0;
  179. } else
  180. goto end;
  181. } else if (var->flags & VAR_SERVERS) {
  182. if (!use_server_type(var->name))
  183. goto end;
  184. if (var->mem && *(struct server_list **)var->mem) {
  185. clearq(*(struct server_list **) var->mem);
  186. *(struct server_list **)var->mem = NULL;
  187. }
  188. if (data)
  189. add_server(data);
  190. if (server_online) {
  191. curserv = -1;
  192. next_server(&curserv, cursrvname, &curservport, NULL);
  193. }
  194. }
  195. end:
  196. if (datap)
  197. free(datap);
  198. // if (var->changed)
  199. // var->changed(var);
  200. return 1;
  201. }
  202. /* Returns the current string representation of the variable's value (if set)
  203. * otherwise, set's the variable to the value of the actual memory and returns that.
  204. */
  205. const char *var_string(variable_t *var)
  206. {
  207. /* see if we already have the data ready to return */
  208. if (var->ldata)
  209. return var->ldata;
  210. else if (var->gdata)
  211. return var->gdata;
  212. /* otherwise, we'll parse/convert the memory associated with our variable */
  213. char *data = NULL;
  214. /* else: fill gdata and return it */
  215. if (var->flags & (VAR_INT|VAR_BOOL)) {
  216. /* only bother setting if we have a value that's not 0 */
  217. if (var->flags & VAR_DETECTED) {
  218. data = strdup(det_translate_num(*(int *) (var->mem)));
  219. } else {
  220. if (*(int *) (var->mem)) {
  221. data = (char *) my_calloc(1, 11);
  222. simple_snprintf(data, 11, "%d", *(int *) (var->mem));
  223. }
  224. }
  225. }
  226. else if (var->flags & VAR_STRING) {
  227. /* only bother setting if we have a non-empty string */
  228. if ((char *)var->mem && *(char *)var->mem)
  229. data = strdup((char *) var->mem);
  230. } else if (var->flags & VAR_RATE) {
  231. /* only bother setting if we don't have 0:0 */
  232. if ((*(rate_t *) (var->mem)).count && (*(rate_t *) (var->mem)).time) {
  233. data = (char *) my_calloc(1, 21);
  234. egg_snprintf(data, 21, "%d:%li", (*(rate_t *) (var->mem)).count, (*(rate_t *) (var->mem)).time);
  235. }
  236. } else if (var->flags & VAR_SERVERS) {
  237. /* only bother setting/checking if we have 'serverlist' alloc'd */
  238. if (*(struct server_list **)var->mem) {
  239. if (!use_server_type(var->name))
  240. return NULL;
  241. struct server_list *n = NULL;
  242. char list[2048] = "", buf[101] = "";
  243. for (n = (*(struct server_list **)var->mem); n; n = n->next) {
  244. if (n->port && n->port != default_port)
  245. simple_snprintf(buf, sizeof(buf), "%s:%d", n->name, n->port);
  246. else
  247. strlcat(buf, n->name, sizeof(buf));
  248. if (n->pass) {
  249. strlcat(buf, ":", sizeof(buf));
  250. strlcat(buf, n->pass, sizeof(buf));
  251. }
  252. strlcat(buf, ",", sizeof(buf));
  253. strlcat(list, buf, sizeof(list));
  254. buf[0] = 0;
  255. }
  256. list[strlen(list) - 1] = 0;
  257. data = strdup(list);
  258. }
  259. }
  260. if (data) {
  261. if (var->flags & VAR_SHUFFLE)
  262. shuffle(data, ",");
  263. if (!(var->flags & VAR_NODEF) && !var->gdata)
  264. var->gdata = data;
  265. else
  266. free(data);
  267. }
  268. return var->gdata ? var->gdata : NULL;
  269. }
  270. static variable_t *var_get_var_by_name(const char *name)
  271. {
  272. int i = 0;
  273. for (i = 0; vars[i].name; i++)
  274. if (!egg_strcasecmp(vars[i].name, name))
  275. return &(vars[i]);
  276. return NULL;
  277. }
  278. void var_set(variable_t *var, const char *target, const char *datain)
  279. {
  280. /* Don't set locally if the variable doesn't permit it. */
  281. if (target) {
  282. if (var->flags & VAR_NOLOC)
  283. return;
  284. struct userrec *botu = NULL;
  285. botu = get_user_by_handle(userlist, (char *) target);
  286. if (botu && (bot_hublevel(botu) != 999) && (var->flags & VAR_NOLHUB))
  287. return;
  288. }
  289. if (conf.bot->hub && (var->flags & VAR_NOGHUB))
  290. return;
  291. bool domem = 1, clear = 0;
  292. // bool freedata = 0;
  293. char *data = (char *) datain;
  294. if (!data || !strcmp(data, "-") || !data[0])
  295. clear = 1;
  296. // if (data && var->flags & VAR_SHUFFLE) {
  297. // char *datadup = strdup(data);
  298. // shuffle(datadup, ",");
  299. // data = datadup;
  300. // freedata++;
  301. // }
  302. if (target) {
  303. bool me = 0;
  304. if (!egg_strcasecmp(conf.bot->nick, target)) {
  305. me = 1;
  306. domem = 1; /* always set the mem if it's local */
  307. if (var->ldata)
  308. free(var->ldata);
  309. sdprintf("var: %s (local): %s", var->name, data);
  310. if (data && !clear)
  311. var->ldata = strdup(data);
  312. else
  313. var->ldata = NULL;
  314. /* if ldata is blank, see about setting the memory to the global setting... */
  315. if (domem && var->mem)
  316. var_set_mem(var, var->ldata ? var->ldata : var->gdata ? var->gdata : NULL);
  317. }
  318. } else if (target == NULL) {
  319. if (var->ldata)
  320. domem = 0;
  321. if (var->gdata)
  322. free(var->gdata);
  323. sdprintf("var: %s (global): %s", var->name, data);
  324. if (data && !clear)
  325. var->gdata = strdup(data);
  326. else {
  327. if (var->flags & VAR_CHANSET)
  328. var->gdata = strdup(def_chanset);
  329. else
  330. var->gdata = NULL;
  331. }
  332. if (domem && var->mem)
  333. var_set_mem(var, var->gdata);
  334. /* It's global, we need to update the botnet with it */
  335. if (!set_noshare)
  336. botnet_send_var_broad(-1, var);
  337. }
  338. // if (freedata)
  339. // free(data);
  340. }
  341. void var_set_by_name(const char *target, const char *name, const char *data)
  342. {
  343. variable_t *var = NULL;
  344. if ((var = var_get_var_by_name(name)))
  345. var_set(var, target, data);
  346. }
  347. void var_set_userentry(const char *target, const char *name, const char *data)
  348. {
  349. struct userrec *u = NULL;
  350. bool me = 0;
  351. if (!egg_strcasecmp(conf.bot->nick, target))
  352. me = 1;
  353. if (me && conf.bot)
  354. u = conf.bot->u;
  355. else if (!me)
  356. u = get_user_by_handle(userlist, (char *) target);
  357. if (u) {
  358. struct xtra_key *xk = (struct xtra_key *) my_calloc(1, sizeof(struct xtra_key));
  359. xk->key = strdup(name);
  360. xk->data = data ? strdup(data) : NULL;
  361. set_user(&USERENTRY_SET, u, xk); //This will send the change to the specified bot
  362. }
  363. }
  364. const char *var_get_str_by_name(const char *name)
  365. {
  366. variable_t *var = NULL;
  367. if ((var = var_get_var_by_name(name)))
  368. return var_string(var);
  369. return NULL;
  370. }
  371. void *var_get_by_name(const char *name)
  372. {
  373. variable_t *var = NULL;
  374. if ((var = var_get_var_by_name(name)))
  375. return var->mem;
  376. return NULL;
  377. }
  378. void init_vars()
  379. {
  380. int i = 0;
  381. if (conf.bot->hub) {
  382. var_set_by_name(NULL, "servers6", "efnet.port80.se,irc.efnet.nl,irc.ipv6.homelien.no,efnet.ipv6.xs4all.nl,irc.ipv6.inter.net.il,irc.choopa.net,irc.ptptech.com");
  383. var_set_by_name(NULL, "servers", "irc.umich.edu,irc.kagmir.ca,irc.dataphone.se,irc.easynews.com,efnet.cs.hut.fi,irc.umn.edu,irc.blackened.com,irc.homelien.no,irc.blessed.net,irc.he.net,irc.inter.net.il,irc.du.se,irc.csbnet.se,efnet.xs4all.nl,irc.efnet.nl,irc.banetele.no,irc.daxnet.no,irc.inet.tele.dk,irc.dks.ca,irc.scnet.net,irc.arcti.ca,irc.avalonworks.ca,irc.foxlink.net,irc2.choopa.net,irc.dkom.at,efnet.demon.co.uk,irc.efnet.pl,irc.nac.net,irc.concentric.net,irc.choopa.net,irc.wh.verio.net,irc.mindspring.com,irc.desync.com,irc.mzima.net,irc.ptptech.com,efnet.port80.se,irc.pte.hu,irc.efnet.fr");
  384. }
  385. /* check mem for all vars and copy to our gdata */
  386. for (i = 0; vars[i].name; i++) {
  387. if (!vars[i].gdata && !vars[i].ldata)
  388. var_string(&vars[i]);
  389. }
  390. var_set_by_name(NULL, "chanset", def_chanset);
  391. if (!strncmp(conf.bot->nick, "wtest", 5))
  392. var_set_by_name(NULL, "homechan", "#bryan");
  393. }
  394. /* This is used to parse (GLOBAL) userfile var lines and changes via .set from a remote hub */
  395. void var_userfile_share_line(char *line, int idx, bool share)
  396. {
  397. char *name = newsplit(&line);
  398. variable_t *var = NULL;
  399. if (!(var = var_get_var_by_name((char *) name))) {
  400. putlog(LOG_ERRORS, "*", "Unrecognized variable '%s' in userfile.", name);
  401. return;
  402. }
  403. set_noshare = 1;
  404. var_set(var, NULL, line);
  405. /* leaf bots don't need to bother attempting to share; there are no bots linked to us! */
  406. if (share && conf.bot->hub)
  407. botnet_send_var_broad(idx, var);
  408. set_noshare = 0;
  409. }
  410. static const char *var_get_bot_data(struct userrec *u, const char *name)
  411. {
  412. if (!u)
  413. return NULL;
  414. struct xtra_key *xk = NULL;
  415. xk = (struct xtra_key *) get_user(&USERENTRY_SET, u);
  416. while (xk && strcmp(xk->key, name))
  417. xk = xk->next;
  418. return xk ? xk->data : NULL;
  419. }
  420. void var_parse_my_botset()
  421. {
  422. int i;
  423. struct xtra_key *xk = NULL, *x = NULL;
  424. /* look for local vars inside our own USERENTRY_SET and set them in our cfg struct */
  425. set_noshare = 1; /* why bother sharing out our LOCAL settings? */
  426. xk = x = (struct xtra_key *) get_user(&USERENTRY_SET, conf.bot->u);
  427. for (i = 0; vars[i].name; i++) {
  428. xk = x; /* reset pointer to beginning */
  429. while (xk && strcmp(xk->key, vars[i].name))
  430. xk = xk->next;
  431. if (xk && xk->key) {
  432. putlog(LOG_DEBUG, "*", "var_parse_my_botset: %s: %s", vars[i].name, xk->data);
  433. var_set(&vars[i], conf.bot->nick, xk->data);
  434. vars[i].flagged = 1;
  435. }
  436. }
  437. for (i = 0; vars[i].name; i++) {
  438. if (vars[i].ldata && !vars[i].flagged) {
  439. sdprintf("var[%s] nulled but we missed it!, reseting.", vars[i].name);
  440. var_set(&vars[i], conf.bot->nick, NULL);
  441. }
  442. vars[i].flagged = 0;
  443. }
  444. set_noshare = 0;
  445. }
  446. static void var_print_list(int idx, const char *name, const char *datain)
  447. {
  448. if (!datain)
  449. return;
  450. char *word = NULL, *data = datain ? strdup(datain) : NULL, *datap = data;
  451. const char *delim = ",";
  452. int i = 1;
  453. dprintf(idx, "%s list:\n", name);
  454. while ((word = strsep(&data, delim))) {
  455. dprintf(idx, " %d. %s\n", i, word);
  456. i++;
  457. }
  458. dprintf(idx, "End of %s list.\n", name);
  459. free(datap);
  460. }
  461. static int var_add_list(const char *botnick, variable_t *var, const char *element)
  462. {
  463. if (!element)
  464. return 0;
  465. char *data = NULL, *olddata = NULL, *botdata = NULL;
  466. if (botnick) { //fetch data from bot's USERENTRY_SET
  467. botdata = (char *) var_get_bot_data(get_user_by_handle(userlist, (char *) botnick), var->name);
  468. olddata = botdata ? botdata : NULL;
  469. } else //use global, no bot specified
  470. olddata = var->gdata ? var->gdata : NULL;
  471. /* Append to the olddata if there...*/
  472. size_t osiz = olddata ? strlen(olddata) : 0;
  473. if (olddata && osiz) {
  474. size_t esiz = strlen(element) + 1; // element + ,
  475. data = (char *) my_calloc(1, osiz + esiz + 1);
  476. simple_snprintf(data, osiz + esiz + 1, "%s,%s", olddata, element);
  477. } else /* otherwise, just set the data to the element to be added */
  478. data = strdup(element);
  479. var_set(var, botnick, data);
  480. if (botnick)
  481. var_set_userentry(botnick, var->name, data);
  482. free(data);
  483. return 1;
  484. }
  485. static char *var_rem_list(const char *botnick, variable_t *var, const char *element)
  486. {
  487. static char ret[101] = "";
  488. if (!element)
  489. return ret;
  490. char *data = NULL, *olddata = NULL, *botdata = NULL, *olddatap = NULL, *olddatacp = NULL, *word = NULL;
  491. if (botnick) { //fetch data from bot's USERENTRY_SET
  492. botdata = (char *) var_get_bot_data(get_user_by_handle(userlist, (char *) botnick), var->name);
  493. olddatacp = botdata ? botdata : NULL;
  494. } else //use global, no bot specified
  495. olddatacp = var->gdata ? var->gdata : NULL;
  496. if (!olddatacp)
  497. return 0;
  498. const char *delim = ",";
  499. int num = 0, i = 1;
  500. if (str_isdigit(element))
  501. num = atoi(element);
  502. olddata = olddatap = strdup(olddatacp);
  503. size_t osiz = strlen(olddata), esiz = strlen(element) + 1, tsiz = osiz - esiz + 1; // element + ,
  504. data = (char *) my_calloc(1, tsiz);
  505. while ((word = strsep(&olddata, delim))) {
  506. if ((num && num != i) || (!num && egg_strcasecmp(word, element))) {
  507. if (data && data[0])
  508. simple_snprintf(data, tsiz, "%s,%s", data, word);
  509. else
  510. simple_snprintf(data, tsiz, "%s", word);
  511. } else
  512. simple_snprintf(ret, sizeof(ret), "%s", word);
  513. i++;
  514. }
  515. var_set(var, botnick, data);
  516. if (botnick)
  517. var_set_userentry(botnick, var->name, data);
  518. free(data);
  519. free(olddatap);
  520. return ret;
  521. }
  522. bool write_vars_and_cmdpass(FILE *f, int idx)
  523. {
  524. putlog(LOG_DEBUG, "@", "Writing set entries...");
  525. if (lfprintf(f, SET_NAME " - -\n") == EOF) /* Daemus */
  526. return 0;
  527. int i = 0;
  528. for (i = 0; vars[i].name; i++) {
  529. /* send blanks if our variable isn't set, theirs MIGHT be set and needs to be UNSET */
  530. if (lfprintf(f, "@ %s %s\n", vars[i].name, vars[i].gdata ? vars[i].gdata : "") == EOF)
  531. return 0;
  532. }
  533. for (struct cmd_pass *cp = cmdpass; cp; cp = cp->next)
  534. if (lfprintf(f, "- %s %s\n", cp->name, cp->pass) == EOF)
  535. return 0;
  536. return 1;
  537. }
  538. #define LIST_ADD 1
  539. #define LIST_RM 2
  540. #define LIST_SHOW 3
  541. void cmd_set_real(const char *botnick, int idx, char *par)
  542. {
  543. variable_t *var = NULL;
  544. char *name = NULL;
  545. const char *data = NULL, *botdata = NULL;
  546. int list = 0, i = 0;
  547. bool notyes = 1;
  548. if (par[0] && !egg_strncasecmp(par, "-yes", 4)) {
  549. notyes = 0;
  550. newsplit(&par);
  551. }
  552. if (botnick)
  553. putlog(LOG_CMDS, "*", "#%s# botset %s %s", dcc[idx].nick, botnick, par);
  554. else
  555. putlog(LOG_CMDS, "*", "#%s# set %s", dcc[idx].nick, par);
  556. if (par[0])
  557. name = newsplit(&par);
  558. if (name) {
  559. if (name[0] == '+')
  560. list = LIST_ADD;
  561. else if (name[0] == '-')
  562. list = LIST_RM;
  563. else if (!egg_strcasecmp(name, "list"))
  564. list = LIST_SHOW;
  565. }
  566. if (list) {
  567. if (list != LIST_SHOW && name[1] && egg_strcasecmp(&name[1], "list"))
  568. name = &name[1];
  569. else {
  570. if (!par[0]) {
  571. dprintf(idx, "A variable must be specified!\n");
  572. return;
  573. }
  574. name = newsplit(&par);
  575. }
  576. }
  577. if (!list || (list && list != LIST_SHOW)) {
  578. if (par[0])
  579. data = (const char *) par;
  580. else if (!par[0] && list) {
  581. dprintf(idx, "A value must be specified!\n");
  582. return;
  583. }
  584. } else if (par[0] && list && list == LIST_SHOW) {
  585. dprintf(idx, "Data value ignored for listing.\n");
  586. }
  587. if (name && !(var = var_get_var_by_name(name))) {
  588. dprintf(idx, "No such variable: %s\n", name);
  589. return;
  590. }
  591. if (list && !(var->flags & VAR_LIST)) {
  592. dprintf(idx, "That variable is not a list!\n");
  593. return;
  594. }
  595. struct userrec *botu = NULL;
  596. bool ishub = 0;
  597. if (botnick) {
  598. botu = get_user_by_handle(userlist, (char *) botnick);
  599. if (data)
  600. dprintf(idx, "%-10s:\n", botnick);
  601. ishub = bot_hublevel(botu) == 999 ? 0 : 1;
  602. }
  603. if (!data) {
  604. while (vars[i].name) {
  605. botdata = NULL;
  606. if (!name) //not looping all, provided with one...
  607. if (vars[i].name)
  608. var = &vars[i];
  609. if (!(var->flags & VAR_HIDE) && !((var->flags & VAR_PERM) && !isowner(dcc[idx].nick)) &&
  610. !(botnick && ((var->flags & VAR_NOLOC) || (ishub && (var->flags & VAR_NOLHUB))))
  611. ) { //Just display the data
  612. if (botnick) { //fetch data from bot's USERENTRY_SET
  613. botdata = var_get_bot_data(botu, var->name);
  614. data = botdata ? botdata : NULL;
  615. } else //use global, no bot specified
  616. data = var->gdata ? var->gdata : NULL;
  617. if (list && data)
  618. var_print_list(idx, var->name, data);
  619. else if (list && !data)
  620. dprintf(idx, "%s list not set.\n", var->name);
  621. else {
  622. char buf[51] = "";
  623. egg_snprintf(buf, sizeof(buf), "(%-6s) %-19s: ", var_type_name(var->flags), var->name);
  624. // dprintf(idx, " %-15s: %s\n", var->name, data);
  625. dumplots(idx, buf, data ? (char *) data : (char *) "(not set)");
  626. }
  627. }
  628. if (name)
  629. break;
  630. i++;
  631. }
  632. } else { // need to set it!
  633. if (!list && var->flags & VAR_LIST) {
  634. if (notyes) {
  635. dprintf(idx, "Using non-list functions on a list variable can be dangerous, please retype with -YES if you're sure:\n");
  636. if (botnick)
  637. dprintf(idx, "%sbotset %s -YES %s ...\n", settings.dcc_prefix, botnick, var->name);
  638. else
  639. dprintf(idx, "%sset -YES %s ...\n", settings.dcc_prefix, var->name);
  640. return;
  641. }
  642. }
  643. if (botnick) {
  644. if (var->flags & VAR_NOLOC) {
  645. dprintf(idx, "Sorry, cannot set '%s' locally.\n", var->name);
  646. return;
  647. }
  648. if (ishub && (var->flags & VAR_NOLHUB)) {
  649. dprintf(idx, "Sorry, cannot set '%s' locally for hubs.\n", var->name);
  650. return;
  651. }
  652. }
  653. if ((var->flags & VAR_PERM) && !isowner(dcc[idx].nick)) {
  654. dprintf(idx, "Sorry, only permanent owners may set '%s'.\n", var->name);
  655. return;
  656. }
  657. if (!strcmp(data, "-"))
  658. data = NULL;
  659. if (botnick) {
  660. char tmp[51] = "";
  661. if (list)
  662. simple_snprintf(tmp, sizeof(tmp), "%s %s %s", var->name, list == LIST_ADD ? "+" : "-", data);
  663. else
  664. simple_snprintf(tmp, sizeof(tmp), "%s %s", var->name, data);
  665. update_mod((char *) botnick, dcc[idx].nick, "botset", tmp);
  666. }
  667. if (list) {
  668. if (list == LIST_ADD) {
  669. if (var_add_list(botnick, var, data)) {
  670. dprintf(idx, "Added '%s' to %s list.\n", data, var->name);
  671. return;
  672. }
  673. } else if (list == LIST_RM) {
  674. char *expanded_data = NULL;
  675. if ((expanded_data = var_rem_list(botnick, var, data)) && expanded_data[0]) {
  676. dprintf(idx, "Removed '%s' from %s list.\n", expanded_data, var->name);
  677. return;
  678. }
  679. }
  680. dprintf(idx, "Failed to modify %s list.\n", var->name);
  681. } else {
  682. var_set(var, botnick, data);
  683. if (botnick)
  684. var_set_userentry(botnick, name, data);
  685. dprintf(idx, "%s: %s\n", name, botnick ? (!data || (data[0] == '-') ? "(not set)" : data) : (var->gdata ? var->gdata : "(not set)"));
  686. }
  687. if (conf.bot->hub)
  688. write_userfile(idx);
  689. }
  690. }