set.c 32 KB

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