set.c 29 KB

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