set.c 33 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099
  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 "match.h"
  11. #include "main.h"
  12. #include "misc.h"
  13. #include "net.h"
  14. #include "src/mod/server.mod/server.h"
  15. #include "src/mod/channels.mod/channels.h"
  16. #include "src/mod/ctcp.mod/ctcp.h"
  17. #include "users.h"
  18. #include "userrec.h"
  19. #include "userent.h"
  20. #include "rfc1459.h"
  21. #include <bdlib/src/Stream.h>
  22. #include <bdlib/src/String.h>
  23. #include "set_default.h"
  24. static bool parsing_botset = 0;
  25. char altchars[50] = "";
  26. char alias[1024] = "";
  27. char rbl_servers[1024] = "";
  28. bool auth_chan;
  29. char auth_key[51] = "";
  30. char auth_prefix[2] = "";
  31. bool auth_obscure;
  32. bool oidentd;
  33. bool ident_botnick;
  34. int dcc_autoaway;
  35. bool irc_autoaway;
  36. bool link_cleartext;
  37. bool dccauth = 0;
  38. bool use_deaf = 0;
  39. bool use_callerid = 0;
  40. int cloak_script = 0;
  41. rate_t close_threshold;
  42. int fight_threshold;
  43. int set_noshare = 0;
  44. int fork_interval;
  45. int hijack;
  46. int in_bots;
  47. int ison_time = 10;
  48. int kill_threshold;
  49. int lag_threshold;
  50. int login;
  51. char motd[512] = "";
  52. char msgident[21] = "";
  53. char msginvite[21] = "";
  54. char msgop[21] = "";
  55. char msgpass[21] = "";
  56. int op_bots;
  57. rate_t op_requests;
  58. int promisc;
  59. int trace;
  60. bool manop_warn;
  61. char homechan[51] = "";
  62. char usermode[15] = "";
  63. ////// THIS MUST REMAIN SORTED
  64. // VAR("bad-process", &badprocess, VAR_INT|VAR_DETECTED, 0, 4, "ignore"),
  65. // VAR("process-list", process_list, VAR_STRING|VAR_LIST, 0, 0, NULL),
  66. static variable_t vars[] = {
  67. VAR("alias", alias, VAR_STRING|VAR_LIST|VAR_NOLOC|VAR_PERM, 0, 0, DEFAULT_ALIAS),
  68. VAR("altchars", altchars, VAR_WORD|VAR_NOLHUB, 0, 0, "-_\\`^[]"),
  69. VAR("auth-chan", &auth_chan, VAR_INT|VAR_BOOL|VAR_NOLHUB, 0, 1, "1"),
  70. VAR("auth-key", auth_key, VAR_STRING|VAR_PERM, 0, 0, NULL),
  71. VAR("auth-obscure", &auth_obscure, VAR_INT|VAR_BOOL, 0, 1, "0"),
  72. VAR("auth-prefix", auth_prefix, VAR_WORD|VAR_NOLHUB|VAR_PERM, 0, 0, "+"),
  73. VAR("callerid", &use_callerid, VAR_INT|VAR_BOOL|VAR_NOLHUB, 0, 1, "1"),
  74. VAR("cloak-script", &cloak_script, VAR_INT|VAR_CLOAK|VAR_NOLHUB, 0, 10, "0"),
  75. VAR("close-threshold", &close_threshold, VAR_RATE|VAR_NOLOC, 0, 0, "0:0"),
  76. VAR("dcc-autoaway", &dcc_autoaway, VAR_INT|VAR_NOLOC, 0, (5*60*60), "1800"),
  77. VAR("dccauth", &dccauth, VAR_INT|VAR_BOOL, 0, 1, "0"),
  78. VAR("deaf", &use_deaf, VAR_INT|VAR_BOOL|VAR_NOLHUB, 0, 1, "1"),
  79. VAR("fight-threshold", &fight_threshold, VAR_INT|VAR_NOLOC, 0, 0, "0"),
  80. VAR("flood-ctcp", &flood_ctcp, VAR_RATE|VAR_NOLHUB, 0, 0, "3:60"),
  81. VAR("flood-callerid", &flood_callerid, VAR_RATE|VAR_NOLHUB, 0, 0, "6:2"),
  82. VAR("flood-msg", &flood_msg, VAR_RATE|VAR_NOLHUB, 0, 0, "5:60"),
  83. VAR("fork-interval", &fork_interval, VAR_INT, 10, 0, "0"),
  84. VAR("hijack", &hijack, VAR_INT|VAR_DETECTED|VAR_PERM, 0, 4, "die"),
  85. VAR("homechan", homechan, VAR_WORD|VAR_NOLOC|VAR_HIDE, 0, 0, NULL),
  86. VAR("ident-botnick", &ident_botnick, VAR_INT|VAR_BOOL|VAR_NOLHUB, 0, 1, "0"),
  87. VAR("in-bots", &in_bots, VAR_INT|VAR_NOLOC, 1, MAX_BOTS, "2"),
  88. VAR("irc-autoaway", &irc_autoaway, VAR_INT|VAR_NOLHUB|VAR_BOOL, 0, 1, "1"),
  89. VAR("jupenick", jupenick, VAR_WORD|VAR_NOHUB|VAR_JUPENICK|VAR_NODEF, 0, 0, NULL),
  90. VAR("kill-threshold", &kill_threshold, VAR_INT|VAR_NOLOC, 0, 0, "0"),
  91. VAR("lag-threshold", &lag_threshold, VAR_INT|VAR_NOLHUB, 0, 0, "15"),
  92. VAR("link_cleartext", &link_cleartext, VAR_INT|VAR_NOLOC|VAR_BOOL, 0, 1, "0"),
  93. VAR("login", &login, VAR_INT|VAR_DETECTED, 0, 4, "warn"),
  94. VAR("manop-warn", &manop_warn, VAR_INT|VAR_BOOL|VAR_NOLHUB, 0, 1, "1"),
  95. VAR("motd", motd, VAR_STRING|VAR_HIDE|VAR_NOLOC, 0, 0, NULL),
  96. VAR("msg-ident", msgident, VAR_WORD|VAR_NOLHUB, 0, 0, NULL),
  97. VAR("msg-invite", msginvite, VAR_WORD|VAR_NOLHUB, 0, 0, NULL),
  98. VAR("msg-op", msgop, VAR_WORD|VAR_NOLHUB, 0, 0, NULL),
  99. VAR("msg-pass", msgpass, VAR_WORD|VAR_NOLHUB, 0, 0, NULL),
  100. VAR("nick", origbotname, VAR_WORD|VAR_NOHUB|VAR_NICK|VAR_NODEF, 0, 0, NULL),
  101. VAR("notify-time", &ison_time, VAR_INT|VAR_NOLHUB, 1, 30, "10"),
  102. VAR("oidentd", &oidentd, VAR_INT|VAR_BOOL|VAR_NOLHUB, 0, 1, "0"),
  103. VAR("op-bots", &op_bots, VAR_INT|VAR_NOLOC, 1, MAX_BOTS, "1"),
  104. VAR("op-requests", &op_requests, VAR_RATE|VAR_NOLOC, 0, 0, "2:5"),
  105. VAR("promisc", &promisc, VAR_INT|VAR_DETECTED, 0, 4, "warn"),
  106. VAR("rbl-servers", rbl_servers, VAR_STRING|VAR_LIST|VAR_SHUFFLE|VAR_NOLHUB, 0, 0, DEFAULT_RBL),
  107. VAR("realname", botrealname, VAR_STRING|VAR_NOLHUB, 0, 0, "* I'm too lame to read BitchX.doc *"),
  108. VAR("server-port", &default_port, VAR_INT|VAR_SHORT|VAR_NOLHUB, 0, 65535, "6667"),
  109. VAR("servers", &serverlist, VAR_SERVERS|VAR_LIST|VAR_SHUFFLE|VAR_NOLHUB|VAR_NOLDEF, 0, 0, DEFAULT_SERVERS),
  110. VAR("servers6", &serverlist, VAR_SERVERS|VAR_LIST|VAR_SHUFFLE|VAR_NOLHUB|VAR_NOLDEF, 0, 0, DEFAULT_SERVERS6),
  111. VAR("trace", &trace, VAR_INT|VAR_DETECTED, 0, 4, "die"),
  112. VAR("usermode", &usermode, VAR_WORD|VAR_NOLHUB, 0, 0, "+iws"),
  113. VAR(NULL, NULL, 0, 0, 0, NULL)
  114. };
  115. static bool use_server_type(const char *name)
  116. {
  117. if (!conf.bot->hub) {
  118. if (!strcmp(name, "servers")) {
  119. if (conf.bot->net.host6 || conf.bot->net.ip6) /* we want to use the servers6 entry. */
  120. return 0;
  121. } else if (!strcmp(name, "servers6")) {
  122. if (!conf.bot->net.host6 && !conf.bot->net.ip6) /* we probably want to use the normal server list.. */
  123. return 0;
  124. }
  125. }
  126. return 1;
  127. }
  128. /* sanitize the variable data string */
  129. char *var_sanitize(variable_t *var, const char *data)
  130. {
  131. /* var->a var->b */
  132. char *dataout = NULL;
  133. if ((var->flags & VAR_INT) && !(var->flags & VAR_BOOL) && !(var->flags & VAR_DETECTED)) {
  134. bool isnumber = 0;
  135. int number = 0;
  136. isnumber = data ? str_isdigit(data) : 0;
  137. if (isnumber)
  138. number = atoi(data);
  139. /* Do limit enforcing... */
  140. if (number < var->a)
  141. number = var->a;
  142. else if (var->b && (number > var->b))
  143. number = var->b;
  144. dataout = (char*) my_calloc(1, 11);
  145. simple_snprintf(dataout, 11, "%d", number);
  146. } else if (var->flags & VAR_DETECTED) {
  147. if (data) {
  148. if (str_isdigit(data)) {
  149. int number = atoi(data);
  150. dataout = strdup(det_translate_num(number));
  151. } else if (!strcasecmp(data, "ignore") || !strcasecmp(data, "warn") ||
  152. !strcasecmp(data, "die") || !strcasecmp(data, "reject") ||
  153. !strcasecmp(data, "suicide")) {
  154. dataout = strdup(data);
  155. }
  156. } else
  157. dataout = strdup("ignore");
  158. } else if (var->flags & VAR_BOOL) {
  159. int num = 0;
  160. if (data && (str_isdigit(data) ||
  161. !strcasecmp(data, "true") ||
  162. !strcasecmp(data, "on") ||
  163. !strcasecmp(data, "off") ||
  164. !strcasecmp(data, "false"))) {
  165. if (str_isdigit(data))
  166. num = atoi(data);
  167. if (num > 0 || !strcasecmp(data, "true") || !strcasecmp(data, "on"))
  168. num = 1;
  169. else if (num < 0 || !strcasecmp(data, "false") || !strcasecmp(data, "off"))
  170. num = 0;
  171. }
  172. dataout = (char*) my_calloc(1, 2);
  173. simple_snprintf(dataout, 2, "%u", num ? 1 : 0);
  174. } else if (var->flags & VAR_STRING) {
  175. dataout = data ? strdup(data) : NULL;
  176. } else if (var->flags & VAR_WORD) {
  177. if (data) {
  178. const char *p = strchr(data, ' ');
  179. if (!p)
  180. dataout = strdup(data);
  181. else
  182. dataout = strldup(data, p - data);
  183. } else {
  184. data = NULL;
  185. }
  186. } else if (var->flags & VAR_RATE) {
  187. const char *p = NULL;
  188. rate_t rate = {0, 0};
  189. if (data && (p = strchr(data, ':'))) {
  190. char *p_count = strldup(data, p - data);
  191. if (str_isdigit(p_count))
  192. rate.count = atoi(p_count);
  193. free(p_count);
  194. if (str_isdigit(p + 1))
  195. rate.time = atoi(p + 1);
  196. }
  197. /* No limit enforcing yet */
  198. dataout = (char*) my_calloc(1, 21);
  199. simple_snprintf(dataout, 21, "%d:%d", rate.count, rate.time);
  200. } else if ((var->flags & VAR_SERVERS)) {
  201. dataout = data ? strdup(data) : NULL;
  202. }
  203. return dataout;
  204. }
  205. /* set the ptr to the new data */
  206. static bool var_set_mem(variable_t *var, const char *datain)
  207. {
  208. char *data = (datain && datain[0]) ? strdup(datain) : NULL, *datap = data;
  209. #ifdef DEBUG
  210. sdprintf("var (mem): %s -> %s", var->name, datain ? datain : "(NULL)");
  211. #endif
  212. if (data && var->flags & VAR_SHUFFLE) {
  213. // char *datadup = strdup(data);
  214. // shuffle(datadup, ",");
  215. // data = datadup;
  216. // freedata++;
  217. shuffle(data, ",", strlen(data) + 1);
  218. }
  219. /* figure out it's type and set it's variable to the data */
  220. if ((var->flags & VAR_INT) && !(var->flags & VAR_BOOL) && !(var->flags & VAR_DETECTED)) {
  221. int number = atoi(data);
  222. if (var->flags & VAR_CLOAK && !conf.bot->hub) {
  223. if (number == 0)
  224. number = randint(CLOAK_COUNT - 1) + 1;
  225. }
  226. if (var->flags & VAR_SHORT)
  227. *(short *) (var->mem) = (short) number;
  228. else
  229. *(int *) (var->mem) = number;
  230. if (var->flags & VAR_CLOAK && !conf.bot->hub)
  231. scriptchanged();
  232. } else if (var->flags & VAR_DETECTED) {
  233. int number = data ? det_translate(data) : DET_IGNORE;
  234. #ifndef DEBUG
  235. if (number < 2 && !strcasecmp(var->name, "trace"))
  236. number = DET_DIE;
  237. #endif
  238. *(int *) (var->mem) = number;
  239. } else if (var->flags & VAR_BOOL) {
  240. bool olddata = *(bool*)(var->mem);
  241. bool num = 0;
  242. if (data[0] == '0')
  243. num = 0;
  244. else if (data[0] == '1')
  245. num = 1;
  246. *(bool *) (var->mem) = num;
  247. if (!strcmp(var->name, "ident-botnick"))
  248. strlcpy(botuser, conf.username && !num ? conf.username : origbotname, 21);
  249. else if ((!strcmp(var->name, "deaf") && deaf_char) || (!strcmp(var->name, "callerid") && callerid_char)) {
  250. if (server_online) {
  251. char which = 0;
  252. if (num == 1 && olddata == 0)
  253. which = '+';
  254. else if (num == 0 && olddata == 1)
  255. which = '-';
  256. if (which) {
  257. char mode_char = strcmp(var->name, "deaf") == 0 ? deaf_char : callerid_char;
  258. dprintf(DP_SERVER, "MODE %s %c%c\n", botname, which, mode_char);
  259. }
  260. }
  261. }
  262. } else if (var->flags & (VAR_STRING|VAR_WORD)) {
  263. char *olddata = ((char*) var->mem) ? strdup((char*) var->mem) : NULL;
  264. if (data)
  265. strlcpy((char *) var->mem, data, var->size);
  266. else
  267. ((char *) var->mem)[0] = 0;
  268. if (!conf.bot->hub) {
  269. if (var->flags & VAR_JUPENICK) {
  270. //Don't allow setting to the same as origbotname
  271. if (data && !rfc_casecmp(jupenick, origbotname)) {
  272. ((char *) var->mem)[0] = 0;
  273. data = NULL;
  274. }
  275. //rolls = 0;
  276. //altnick_char = 0;
  277. // Don't send nick changes on restart, no need.
  278. if (server_online && !parsing_botset) {
  279. //New jupenick, no old value, or old value and it has changed.
  280. if (data && (!olddata || (olddata && strcmp(olddata, jupenick)))) {
  281. // If not on the new nick, jump to it
  282. if (!match_my_nick(jupenick)) {
  283. tried_jupenick = 1;
  284. dprintf(DP_SERVER, "NICK %s\n", jupenick);
  285. }
  286. // Unset and there was an old value
  287. } else if (!data && olddata) {
  288. // Unset jupenick, try for 'nick' now if we were on jupenick
  289. if (match_my_nick(olddata)) {
  290. altnick_char = rolls = 0;
  291. dprintf(DP_SERVER, "NICK %s\n", origbotname);
  292. }
  293. }
  294. }
  295. } else if (var->flags & VAR_NICK) {
  296. // Default to botnick
  297. if (!data)
  298. strlcpy((char *) var->mem, conf.bot->nick, var->size);
  299. // Only send nick changes if online and not loading
  300. if (server_online && !parsing_botset) {
  301. // the nick changed and not on the new nick already
  302. if ((olddata && strcmp(olddata, (char *) var->mem)) &&
  303. !match_my_nick((char *) var->mem) &&
  304. (!jupenick[0] || !match_my_nick(jupenick)))
  305. {
  306. // Unset the rolls/altnick stuff as we're starting over from scratch.
  307. altnick_char = rolls = 0;
  308. dprintf(DP_SERVER, "NICK %s\n", (char *) var->mem);
  309. }
  310. }
  311. }
  312. }
  313. if (olddata)
  314. free(olddata);
  315. } else if (var->flags & VAR_RATE) {
  316. char *p = NULL;
  317. if (data && (p = strchr(data, ':'))) {
  318. *p = 0;
  319. p++;
  320. (*(rate_t *) (var->mem)).count = atoi(data);
  321. (*(rate_t *) (var->mem)).time = atoi(p);
  322. --p;
  323. *p = ':';
  324. }
  325. } else if ((var->flags & VAR_SERVERS) && use_server_type(var->name)) {
  326. if (var->mem && *(struct server_list **)var->mem) {
  327. clearq(*(struct server_list **) var->mem);
  328. *(struct server_list **)var->mem = NULL;
  329. }
  330. if (data)
  331. add_server(data);
  332. if (server_online) {
  333. curserv = -1;
  334. next_server(&curserv, cursrvname, &curservport, NULL);
  335. }
  336. }
  337. if (datap)
  338. free(datap);
  339. // if (var->changed)
  340. // var->changed(var);
  341. return 1;
  342. }
  343. /* Returns the current string representation of the variable's value (if set)
  344. * otherwise, set's the variable to the value of the actual memory and returns that.
  345. */
  346. const char *var_string(variable_t *var)
  347. {
  348. /* see if we already have the data ready to return */
  349. if (var->ldata)
  350. return var->ldata;
  351. else if (var->gdata)
  352. return var->gdata;
  353. /* otherwise, we'll parse/convert the memory associated with our variable */
  354. char *data = NULL;
  355. /* else: fill gdata and return it */
  356. if ((var->flags & VAR_INT) && !(var->flags & VAR_BOOL)) {
  357. /* only bother setting if we have a value that's not 0 */
  358. if (var->flags & VAR_DETECTED) {
  359. data = strdup(det_translate_num(*(int *) (var->mem)));
  360. } else {
  361. if (*(int *) (var->mem)) {
  362. data = (char *) my_calloc(1, 11);
  363. simple_snprintf(data, 11, "%d", *(int *) (var->mem));
  364. }
  365. }
  366. } else if (var->flags & (VAR_BOOL)) {
  367. /* only bother setting if we have a value that's not 0 */
  368. if (*(int *) (var->mem)) {
  369. bool num = *(bool *) (var->mem);
  370. data = (char *) my_calloc(1, 2);
  371. /* Only actually set 0 or 1 */
  372. simple_snprintf(data, 2, "%d", num ? 1 : 0);
  373. }
  374. } else if (var->flags & (VAR_STRING|VAR_WORD)) {
  375. /* only bother setting if we have a non-empty string */
  376. if ((char *)var->mem && *(char *)var->mem)
  377. data = strdup((char *) var->mem);
  378. } else if (var->flags & VAR_RATE) {
  379. /* only bother setting if we don't have 0:0 */
  380. if ((*(rate_t *) (var->mem)).count && (*(rate_t *) (var->mem)).time) {
  381. data = (char *) my_calloc(1, 21);
  382. simple_snprintf(data, 21, "%d:%d", (*(rate_t *) (var->mem)).count, (*(rate_t *) (var->mem)).time);
  383. }
  384. } else if (var->flags & VAR_SERVERS) {
  385. /* only bother setting/checking if we have 'serverlist' alloc'd */
  386. if (*(struct server_list **)var->mem) {
  387. if (!use_server_type(var->name))
  388. return NULL;
  389. struct server_list *n = NULL;
  390. char list[2048] = "", buf[101] = "";
  391. for (n = (*(struct server_list **)var->mem); n; n = n->next) {
  392. if (n->port && n->port != default_port)
  393. simple_snprintf(buf, sizeof(buf), "%s:%d", n->name, n->port);
  394. else
  395. strlcat(buf, n->name, sizeof(buf));
  396. if (n->pass) {
  397. strlcat(buf, ":", sizeof(buf));
  398. strlcat(buf, n->pass, sizeof(buf));
  399. }
  400. strlcat(buf, ",", sizeof(buf));
  401. strlcat(list, buf, sizeof(list));
  402. buf[0] = 0;
  403. }
  404. list[strlen(list) - 1] = 0;
  405. data = strdup(list);
  406. }
  407. }
  408. if (data) {
  409. if (var->flags & VAR_SHUFFLE)
  410. shuffle(data, ",", strlen(data) + 1);
  411. if ((var->flags & VAR_NODEF) && !var->gdata) {
  412. free(data);
  413. data = NULL;
  414. }
  415. }
  416. return data;
  417. }
  418. static int comp_variable_t(const void *m1, const void *m2) {
  419. const variable_t *mi1 = (const variable_t *) m1;
  420. const variable_t *mi2 = (const variable_t *) m2;
  421. return strcasecmp(mi1->name, mi2->name);
  422. }
  423. static inline variable_t *var_get_var_by_name(const char *name)
  424. {
  425. variable_t key;
  426. key.name = name;
  427. return (variable_t*) bsearch(&key, &vars, (sizeof(vars)/sizeof(variable_t)) - 1, sizeof(variable_t), comp_variable_t);
  428. }
  429. void var_set(variable_t *var, const char *target, const char *datain)
  430. {
  431. /* Don't set locally if the variable doesn't permit it. */
  432. if (target) {
  433. if (var->flags & VAR_NOLOC)
  434. return;
  435. struct userrec *botu = NULL;
  436. botu = get_user_by_handle(userlist, (char *) target);
  437. if (botu && (bot_hublevel(botu) != 999) && (var->flags & VAR_NOLHUB))
  438. return;
  439. }
  440. if (conf.bot->hub && (var->flags & VAR_NOGHUB))
  441. return;
  442. bool domem = 1, clear = 0;
  443. // bool freedata = 0;
  444. char *data = (char *) datain;
  445. /* If no data or data is not -
  446. sanitize the data */
  447. char *sdata = NULL;
  448. /* Make a temporary to free at the end */
  449. if (data && strcmp(data, "-") && (!var->def || (var->def && strcmp(data, var->def)))) {
  450. sdata = var_sanitize(var, data);
  451. data = sdata;
  452. }
  453. /* If no data, or data is "-", do a clear */
  454. if (!data || !strcmp(data, "-") || !data[0])
  455. clear = 1;
  456. // if (data && var->flags & VAR_SHUFFLE) {
  457. // char *datadup = strdup(data);
  458. // shuffle(datadup, ",");
  459. // data = datadup;
  460. // freedata++;
  461. // }
  462. if (target) {
  463. bool me = 0;
  464. if (!strcasecmp(conf.bot->nick, target)) {
  465. me = 1;
  466. domem = 1; /* always set the mem if it's local */
  467. if (var->ldata)
  468. free(var->ldata);
  469. sdprintf("var: %s (local): %s", var->name, data ? data : "(NULL)");
  470. if (data && !clear)
  471. var->ldata = strdup(data);
  472. else
  473. var->ldata = NULL;
  474. /* if ldata is blank, see about setting the memory to the global setting... */
  475. if (domem && var->mem)
  476. var_set_mem(var, var->ldata ? var->ldata : var->gdata ? var->gdata : NULL);
  477. }
  478. } else if (target == NULL) {
  479. if (var->ldata)
  480. domem = 0;
  481. if (var->gdata)
  482. free(var->gdata);
  483. sdprintf("var: %s (global): %s", var->name, data ? data : "(NULL)");
  484. if (data && !clear)
  485. var->gdata = strdup(data);
  486. else
  487. var->gdata = var->def ? strdup(var->def) : NULL;
  488. if (domem && var->mem)
  489. var_set_mem(var, var->gdata);
  490. /* It's global, we need to update the botnet with it */
  491. if (!set_noshare)
  492. botnet_send_var_broad(-1, var);
  493. }
  494. // if (freedata)
  495. // free(data);
  496. if (sdata)
  497. free(sdata);
  498. }
  499. void var_set_by_name(const char *target, const char *name, const char *data)
  500. {
  501. variable_t *var = NULL;
  502. if ((var = var_get_var_by_name(name)))
  503. var_set(var, target, data);
  504. }
  505. void var_set_userentry(const char *target, const char *name, const char *data)
  506. {
  507. struct userrec *u = NULL;
  508. bool me = 0;
  509. if (!strcasecmp(conf.bot->nick, target))
  510. me = 1;
  511. if (me && conf.bot)
  512. u = conf.bot->u;
  513. else if (!me)
  514. u = get_user_by_handle(userlist, (char *) target);
  515. if (u) {
  516. struct xtra_key *xk = (struct xtra_key *) my_calloc(1, sizeof(struct xtra_key));
  517. xk->key = strdup(name);
  518. xk->data = data ? strdup(data) : NULL;
  519. set_user(&USERENTRY_SET, u, xk); //This will send the change to the specified bot
  520. }
  521. }
  522. const char *var_get_str_by_name(const char *name)
  523. {
  524. variable_t *var = NULL;
  525. if ((var = var_get_var_by_name(name)))
  526. return var_string(var);
  527. return NULL;
  528. }
  529. void *var_get_by_name(const char *name)
  530. {
  531. variable_t *var = NULL;
  532. if ((var = var_get_var_by_name(name)))
  533. return var->mem;
  534. return NULL;
  535. }
  536. void init_vars()
  537. {
  538. int i = 0;
  539. /* initialize vars: defaults -> gdata */
  540. for (i = 0; vars[i].name; i++) {
  541. if (!vars[i].gdata && !vars[i].ldata && !(!conf.bot->hub && (vars[i].flags & VAR_NOLDEF)))
  542. var_set(&vars[i], NULL, NULL); //empty out and set to defaults
  543. }
  544. #ifdef DEBUG
  545. if (!strncmp(conf.bot->nick, "wtest", 5))
  546. var_set_by_name(NULL, "homechan", "#bryan");
  547. #endif
  548. }
  549. /* This is used to parse (GLOBAL) userfile var lines and changes via .set from a remote hub */
  550. void var_userfile_share_line(char *line, int idx, bool share)
  551. {
  552. char *name = newsplit(&line);
  553. variable_t *var = NULL;
  554. if (!(var = var_get_var_by_name((char *) name))) {
  555. putlog(LOG_ERRORS, "*", "Unrecognized variable '%s' in userfile.", name);
  556. return;
  557. }
  558. set_noshare = 1;
  559. var_set(var, NULL, line);
  560. /* leaf bots don't need to bother attempting to share; there are no bots linked to us! */
  561. if (share && (conf.bot->hub || conf.bot->localhub))
  562. botnet_send_var_broad(idx, var);
  563. set_noshare = 0;
  564. }
  565. static const char *var_get_bot_data(struct userrec *u, const char *name)
  566. {
  567. if (!u)
  568. return NULL;
  569. struct xtra_key *xk = NULL;
  570. xk = (struct xtra_key *) get_user(&USERENTRY_SET, u);
  571. while (xk && strcmp(xk->key, name))
  572. xk = xk->next;
  573. return xk ? xk->data : NULL;
  574. }
  575. void var_parse_my_botset()
  576. {
  577. int i;
  578. struct xtra_key *xk = NULL, *x = NULL;
  579. /* look for local vars inside our own USERENTRY_SET and set them in our cfg struct */
  580. set_noshare = 1; /* why bother sharing out our LOCAL settings? */
  581. parsing_botset = 1;
  582. xk = x = (struct xtra_key *) get_user(&USERENTRY_SET, conf.bot->u);
  583. for (i = 0; vars[i].name; i++) {
  584. xk = x; /* reset pointer to beginning */
  585. while (xk && strcmp(xk->key, vars[i].name))
  586. xk = xk->next;
  587. if (xk && xk->key) {
  588. putlog(LOG_DEBUG, "*", "var_parse_my_botset: %s: %s", vars[i].name, xk->data);
  589. var_set(&vars[i], conf.bot->nick, xk->data);
  590. vars[i].flagged = 1;
  591. }
  592. }
  593. for (i = 0; vars[i].name; i++) {
  594. if (vars[i].ldata && !vars[i].flagged) {
  595. sdprintf("var[%s] nulled but we missed it!, reseting.", vars[i].name);
  596. var_set(&vars[i], conf.bot->nick, NULL);
  597. }
  598. vars[i].flagged = 0;
  599. }
  600. parsing_botset = 0;
  601. set_noshare = 0;
  602. }
  603. static void var_print_list(int idx, const char *name, const char *datain)
  604. {
  605. if (!datain)
  606. return;
  607. char *word = NULL, *data = datain ? strdup(datain) : NULL, *datap = data;
  608. const char *delim = ",";
  609. int i = 1;
  610. dprintf(idx, "%s list:\n", name);
  611. while ((word = strsep(&data, delim))) {
  612. dprintf(idx, " %d. %s\n", i, word);
  613. i++;
  614. }
  615. dprintf(idx, "End of %s list.\n", name);
  616. free(datap);
  617. }
  618. static bool var_find_list(const char *botnick, variable_t *var, const char *element) {
  619. char *olddata = NULL;
  620. if (botnick) { //fetch data from bot's USERENTRY_SET
  621. char *botdata = (char *) var_get_bot_data(get_user_by_handle(userlist, (char *) botnick), var->name);
  622. olddata = botdata ? botdata : NULL;
  623. } else //use global, no bot specified
  624. olddata = var->gdata ? var->gdata : NULL;
  625. if (!olddata)
  626. return false;
  627. char *item = NULL, *data = strdup(olddata), *datap = data;
  628. const char *delim = ",";
  629. size_t slen = 0;
  630. const char *p = NULL;
  631. /* The first word only .. */
  632. if (!strcmp(var->name, "alias") && (p = strchr(element, ' ')))
  633. slen = p - element;
  634. /* Or the whole word */
  635. else
  636. slen = strlen(element);
  637. while ((item = strsep(&data, delim)))
  638. if (!strncasecmp(item, element, slen)) {
  639. free(datap);
  640. return true;
  641. }
  642. free(datap);
  643. return false;
  644. }
  645. static int var_add_list(const char *botnick, variable_t *var, const char *element)
  646. {
  647. if (!element)
  648. return 0;
  649. char *data = NULL, *olddata = NULL, *botdata = NULL;
  650. if (botnick) { //fetch data from bot's USERENTRY_SET
  651. botdata = (char *) var_get_bot_data(get_user_by_handle(userlist, (char *) botnick), var->name);
  652. olddata = botdata ? botdata : NULL;
  653. } else //use global, no bot specified
  654. olddata = var->gdata ? var->gdata : NULL;
  655. /* Append to the olddata if there...*/
  656. size_t osiz = olddata ? strlen(olddata) : 0;
  657. if (olddata && osiz) {
  658. size_t esiz = strlen(element) + 1; // element + ,
  659. data = (char *) my_calloc(1, osiz + esiz + 1);
  660. simple_snprintf(data, osiz + esiz + 1, "%s,%s", olddata, element);
  661. } else /* otherwise, just set the data to the element to be added */
  662. data = strdup(element);
  663. var_set(var, botnick, data);
  664. if (botnick)
  665. var_set_userentry(botnick, var->name, data);
  666. free(data);
  667. return 1;
  668. }
  669. static char *var_rem_list(const char *botnick, variable_t *var, const char *element)
  670. {
  671. static char ret[101] = "";
  672. ret[0] = 0;
  673. if (!element)
  674. return ret;
  675. char *olddata = NULL, *botdata = NULL, *olddatap = NULL, *olddatacp = NULL;
  676. if (botnick) { //fetch data from bot's USERENTRY_SET
  677. botdata = (char *) var_get_bot_data(get_user_by_handle(userlist, (char *) botnick), var->name);
  678. olddatacp = botdata ? botdata : NULL;
  679. } else //use global, no bot specified
  680. olddatacp = var->gdata ? var->gdata : NULL;
  681. if (!olddatacp)
  682. return 0;
  683. const char *delim = ",";
  684. int num = 0, i = 0;
  685. size_t elen = 0;
  686. if (str_isdigit(element))
  687. num = atoi(element);
  688. else
  689. elen = strlen(element);
  690. olddata = olddatap = strdup(olddatacp);
  691. size_t tsiz = strlen(olddata) + 1;
  692. char *data = (char *) my_calloc(1, tsiz);
  693. char *word = NULL;
  694. while ((word = strsep(&olddata, delim))) {
  695. ++i;
  696. if ((num && num != i) || (!num && strncasecmp(word, element, elen))) {
  697. /* Reconstruct the left and right part of the list...*/
  698. if (data && data[0] && word && word[0]) {
  699. strlcat(data, ",", tsiz);
  700. strlcat(data, word, tsiz);
  701. } else
  702. strlcpy(data, word, tsiz);
  703. } else /* minus the part we are removing ... */
  704. strlcpy(ret, word, sizeof(ret));
  705. }
  706. if (num <= i && ret[0]) {
  707. var_set(var, botnick, data);
  708. if (botnick)
  709. var_set_userentry(botnick, var->name, data);
  710. } else
  711. ret[0] = 0;
  712. free(data);
  713. free(olddatap);
  714. return ret;
  715. }
  716. void write_vars_and_cmdpass(bd::Stream& stream, int idx)
  717. {
  718. bd::String buf;
  719. putlog(LOG_DEBUG, "@", "Writing set entries...");
  720. stream << buf.printf(SET_NAME " - -\n");
  721. int i = 0;
  722. for (i = 0; vars[i].name; i++) {
  723. // If we're a localhub, dont share variables set to not leaf default unless we're linked to a hub.
  724. // Otherwise, we share out servers, and the child bots connect to the default list,
  725. // before ever receiving the actual list from the hub. (Along with stuff like realname)
  726. if (conf.bot->hub ||
  727. (conf.bot->localhub &&
  728. (!(vars[i].flags & VAR_NOLDEF) ||
  729. ((vars[i].flags & VAR_NOLDEF) && have_linked_to_hub))
  730. )) {
  731. /* send blanks if our variable isn't set, theirs MIGHT be set and needs to be UNSET */
  732. stream << buf.printf("@ %s %s\n", vars[i].name, vars[i].gdata ? vars[i].gdata : "");
  733. }
  734. }
  735. for (struct cmd_pass *cp = cmdpass; cp; cp = cp->next)
  736. stream << buf.printf("- %s %s\n", cp->name, cp->pass);
  737. }
  738. #define LIST_ADD 1
  739. #define LIST_RM 2
  740. #define LIST_SHOW 3
  741. int cmd_set_real(const char *botnick, int idx, char *par)
  742. {
  743. variable_t *var = NULL;
  744. char *name = NULL;
  745. const char *data = NULL, *botdata = NULL;
  746. int list = 0, i = 0;
  747. bool notyes = 1, wildcard = 0;
  748. if (par[0] && !strncasecmp(par, "-yes", 4)) {
  749. notyes = 0;
  750. newsplit(&par);
  751. }
  752. if (botnick)
  753. putlog(LOG_CMDS, "*", "#%s# botset %s %s", dcc[idx].nick, botnick, par);
  754. else
  755. putlog(LOG_CMDS, "*", "#%s# set %s", dcc[idx].nick, par);
  756. if (par[0])
  757. name = newsplit(&par);
  758. if (name) {
  759. if (name[0] == '+')
  760. list = LIST_ADD;
  761. else if (name[0] == '-')
  762. list = LIST_RM;
  763. else if (!strcasecmp(name, "list"))
  764. list = LIST_SHOW;
  765. }
  766. if (list) {
  767. if (list != LIST_SHOW && name[1] && strcasecmp(&name[1], "list"))
  768. name = &name[1];
  769. else {
  770. if (!par[0]) {
  771. dprintf(idx, "A variable must be specified!\n");
  772. return 0;
  773. }
  774. name = newsplit(&par);
  775. }
  776. }
  777. if (name && strchr(name, '*'))
  778. wildcard = 1;
  779. if (wildcard && par[0]) {
  780. dprintf(idx, "Wildcards may only be used for listing matching variables.\n");
  781. return 0;
  782. } else if (!wildcard) {
  783. if (!list || (list && list != LIST_SHOW)) {
  784. if (par[0])
  785. data = (const char *) par;
  786. else if (!par[0] && list) {
  787. dprintf(idx, "A value must be specified!\n");
  788. return 0;
  789. }
  790. } else if (par[0] && list && list == LIST_SHOW) {
  791. dprintf(idx, "Data value ignored for listing.\n");
  792. }
  793. if (name && !(var = var_get_var_by_name(name))) {
  794. dprintf(idx, "No such variable: %s\n", name);
  795. return 0;
  796. }
  797. if (list && !(var->flags & VAR_LIST)) {
  798. dprintf(idx, "That variable is not a list!\n");
  799. return 0;
  800. }
  801. }
  802. struct userrec *botu = NULL;
  803. bool ishub = 0;
  804. if (botnick) {
  805. botu = get_user_by_handle(userlist, (char *) botnick);
  806. if (data)
  807. dprintf(idx, "%-10s:\n", botnick);
  808. ishub = bot_hublevel(botu) == 999 ? 0 : 1;
  809. }
  810. if (!data) {
  811. while (vars[i].name) {
  812. botdata = NULL;
  813. if (!name || wildcard) //not looping all, provided with one...
  814. if (vars[i].name)
  815. var = &vars[i];
  816. if (wildcard && !wild_match(name, var->name)) {
  817. ++i;
  818. continue;
  819. }
  820. if (!(var->flags & VAR_HIDE) && !((var->flags & VAR_PERM) && !isowner(dcc[idx].nick)) &&
  821. !(botnick && ((var->flags & VAR_NOLOC) || (ishub && (var->flags & VAR_NOLHUB))))
  822. ) { //Just display the data
  823. if (botnick) { //fetch data from bot's USERENTRY_SET
  824. botdata = var_get_bot_data(botu, var->name);
  825. data = botdata ? botdata : NULL;
  826. } else //use global, no bot specified
  827. data = var->gdata ? var->gdata : NULL;
  828. if (list && data)
  829. var_print_list(idx, var->name, data);
  830. else if (list && !data)
  831. dprintf(idx, "%s list not set.\n", var->name);
  832. else {
  833. char buf[51] = "";
  834. simple_snprintf(buf, sizeof(buf), "(%-6s) %-19s: ", var_type_name(var->flags), var->name);
  835. // dprintf(idx, " %-15s: %s\n", var->name, data);
  836. dumplots(idx, buf, data ? (char *) data : (char *) "(not set)");
  837. }
  838. }
  839. if (name && !wildcard)
  840. break;
  841. ++i;
  842. }
  843. } else { // need to set it!
  844. if (!list && var->flags & VAR_LIST) {
  845. if (notyes) {
  846. dprintf(idx, "Using non-list functions on a list variable can be dangerous, please retype with -YES if you're sure:\n");
  847. if (botnick)
  848. dprintf(idx, "%sbotset %s -YES %s ...\n", (dcc[idx].u.chat->channel >= 0) ? settings.dcc_prefix : "", botnick, var->name);
  849. else
  850. dprintf(idx, "%sset -YES %s ...\n", (dcc[idx].u.chat->channel >= 0) ? settings.dcc_prefix : "", var->name);
  851. return 0;
  852. }
  853. }
  854. if (botnick) {
  855. if (var->flags & VAR_NOLOC) {
  856. dprintf(idx, "Sorry, cannot set '%s' locally.\n", var->name);
  857. return 0;
  858. }
  859. if (ishub && (var->flags & VAR_NOLHUB)) {
  860. dprintf(idx, "Sorry, cannot set '%s' locally for hubs.\n", var->name);
  861. return 0;
  862. }
  863. }
  864. if ((var->flags & VAR_PERM) && !isowner(dcc[idx].nick)) {
  865. dprintf(idx, "Sorry, only permanent owners may set '%s'.\n", var->name);
  866. return 0;
  867. }
  868. if (!list && !strcmp(data, "-"))
  869. data = NULL;
  870. if (botnick) {
  871. char tmp[51] = "";
  872. if (list)
  873. simple_snprintf(tmp, sizeof(tmp), "%s %s %s", var->name, list == LIST_ADD ? "+" : "-", data);
  874. else
  875. simple_snprintf(tmp, sizeof(tmp), "%s %s", var->name, data);
  876. update_mod((char *) botnick, dcc[idx].nick, "botset", tmp);
  877. }
  878. if (list) {
  879. if (strchr(data, ',')) {
  880. dprintf(idx, "The ',' character may not be used with set list functions.\n");
  881. return 0;
  882. }
  883. if (list == LIST_ADD) {
  884. if (var_find_list(botnick, var, data)) {
  885. char *data_word = NULL;
  886. const char *p = NULL;
  887. if (!strcmp(var->name, "alias")) {
  888. if ((p = strchr(data, ' ')))
  889. data_word = strldup(data, p - data);
  890. } else
  891. data_word = (char*)data;
  892. dprintf(idx, "Item '%s' is already in the %s list.\n", data_word, var->name);
  893. if (p)
  894. free(data_word);
  895. return 0;
  896. } else if (var_add_list(botnick, var, data)) {
  897. dprintf(idx, "Added '%s' to %s list.\n", data, var->name);
  898. return 1;
  899. }
  900. } else if (list == LIST_RM) {
  901. char *expanded_data = NULL;
  902. if ((expanded_data = var_rem_list(botnick, var, data)) && expanded_data[0]) {
  903. dprintf(idx, "Removed '%s' from %s list.\n", expanded_data, var->name);
  904. return 1;
  905. } else if (!var_find_list(botnick, var, data)) {
  906. char *data_word = NULL;
  907. const char *p = NULL;
  908. if (!strcmp(var->name, "alias")) {
  909. if ((p = strchr(data, ' ')))
  910. data_word = strldup(data, p - data);
  911. } else
  912. data_word = (char*)data;
  913. dprintf(idx, "Item '%s' does not exist in the %s list.\n", data, var->name);
  914. if (p)
  915. free(data_word);
  916. return 0;
  917. }
  918. }
  919. dprintf(idx, "Failed to modify %s list.\n", var->name);
  920. return 0;
  921. } else {
  922. /* If no data or data is not -
  923. sanitize the data */
  924. char *sdata = NULL;
  925. /* Make a temporary to free at the end */
  926. if (data && strcmp(data, "-") && (!var->def || (var->def && strcmp(data, var->def)))) {
  927. sdata = var_sanitize(var, data);
  928. data = sdata;
  929. }
  930. var_set(var, botnick, data);
  931. if (botnick)
  932. var_set_userentry(botnick, name, data);
  933. dprintf(idx, "%s: %s\n", name, botnick ? (!data || (data[0] == '-' && !data[1]) ? "(not set)" : data) : (var->gdata ? var->gdata : "(not set)"));
  934. if (sdata)
  935. free(sdata);
  936. }
  937. return 1;
  938. }
  939. return 0;
  940. }