set.c 35 KB

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