set.c 29 KB

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