]>
Commit | Line | Data |
---|---|---|
8e965b0c | 1 | /* Copyright (C) Chris Porter 2005-2007 */ |
2da56f0d CP |
2 | /* ALL RIGHTS RESERVED. */ |
3 | /* Don't put this into the SVN repo. */ | |
4 | ||
d0d1e2a3 CP |
5 | /* |
6 | @todo | |
7 | - Write a nick printf type thing for pcalled functions. | |
8 | - Make commands register as apposed to blinding calling. | |
9 | - Use numerics instead of huge structures, and add lookup functions. | |
10 | */ | |
11 | ||
2da56f0d CP |
12 | #include "../channel/channel.h" |
13 | #include "../control/control.h" | |
14 | #include "../nick/nick.h" | |
15 | #include "../localuser/localuser.h" | |
16 | #include "../localuser/localuserchannel.h" | |
17 | #include "../lib/irc_string.h" | |
8e965b0c | 18 | #include "../lib/flags.h" |
2da56f0d CP |
19 | |
20 | #include "lua.h" | |
21 | #include "luabot.h" | |
22 | ||
23 | #include <stdarg.h> | |
30a59b87 | 24 | #include <stddef.h> |
2da56f0d | 25 | |
ec91b2e4 CP |
26 | #define MAX_PUSHER 50 |
27 | ||
2da56f0d CP |
28 | static int lua_smsg(lua_State *ps); |
29 | static int lua_skill(lua_State *ps); | |
30 | ||
ec91b2e4 | 31 | typedef struct lua_pusher { |
30a59b87 CP |
32 | short argtype; |
33 | short offset; | |
34 | const char *structname; | |
ec91b2e4 CP |
35 | } lua_pusher; |
36 | ||
37 | struct lua_pusher nickpusher[MAX_PUSHER]; | |
38 | struct lua_pusher chanpusher[MAX_PUSHER]; | |
ebdfcaa9 | 39 | int nickpushercount, chanpushercount; |
ec91b2e4 | 40 | |
ebdfcaa9 | 41 | INLINE void lua_setuppusher(struct lua_pusher *pusherlist, lua_State *l, int index, struct lua_pusher **lp, int max, int pcount); |
ec91b2e4 CP |
42 | INLINE int lua_usepusher(lua_State *l, struct lua_pusher **lp, void *np); |
43 | ||
ebdfcaa9 | 44 | void lua_initnickpusher(void); |
ec91b2e4 | 45 | void lua_initchanpusher(void); |
ec91b2e4 | 46 | |
ebdfcaa9 CP |
47 | #define lua_setupnickpusher(L2, I2, P2, M2) lua_setuppusher(&nickpusher[0], L2, I2, P2, M2, nickpushercount) |
48 | #define lua_setupchanpusher(L2, I2, P2, M2) lua_setuppusher(&chanpusher[0], L2, I2, P2, M2, chanpushercount) | |
30a59b87 | 49 | |
2da56f0d CP |
50 | int lua_cmsg(char *channell, char *message, ...) { |
51 | char buf[512]; | |
52 | va_list va; | |
53 | channel *cp; | |
54 | ||
55 | va_start(va, message); | |
56 | vsnprintf(buf, sizeof(buf), message, va); | |
57 | va_end(va); | |
58 | ||
59 | cp = findchannel(channell); | |
60 | if(!cp) | |
61 | return LUA_FAIL; | |
62 | ||
63 | if(!lua_lineok(buf)) | |
64 | return LUA_FAIL; | |
65 | ||
66 | lua_channelmessage(cp, "%s", buf); | |
67 | ||
68 | return LUA_OK; | |
69 | } | |
70 | ||
71 | static int lua_chanmsg(lua_State *ps) { | |
72 | if(!lua_isstring(ps, 1)) | |
73 | LUA_RETURN(ps, LUA_FAIL); | |
74 | ||
75 | LUA_RETURN(ps, lua_cmsg(LUA_PUKECHAN, "lua: %s", lua_tostring(ps, 1))); | |
76 | } | |
77 | ||
2da56f0d CP |
78 | static int lua_ctcp(lua_State *ps) { |
79 | const char *n, *msg; | |
80 | nick *np; | |
81 | ||
82 | if(!lua_isstring(ps, 1) || !lua_isstring(ps, 2)) | |
83 | LUA_RETURN(ps, LUA_FAIL); | |
84 | ||
85 | n = lua_tostring(ps, 1); | |
86 | msg = lua_tostring(ps, 2); | |
87 | ||
88 | np = getnickbynick(n); | |
89 | if(!np || !lua_lineok(msg)) | |
90 | LUA_RETURN(ps, LUA_FAIL); | |
91 | ||
92 | lua_message(np, "\001%s\001", msg); | |
93 | ||
94 | LUA_RETURN(ps, lua_cmsg(LUA_PUKECHAN, "lua-ctcp: %s (%s)", np->nick, msg)); | |
95 | } | |
96 | ||
97 | static int lua_noticecmd(lua_State *ps) { | |
98 | const char *n, *msg; | |
99 | nick *np; | |
100 | ||
101 | if(!lua_isstring(ps, 1) || !lua_isstring(ps, 2)) | |
102 | LUA_RETURN(ps, LUA_FAIL); | |
103 | ||
104 | n = lua_tostring(ps, 1); | |
105 | msg = lua_tostring(ps, 2); | |
106 | ||
107 | np = getnickbynick(n); | |
108 | if(!np || !lua_lineok(msg)) | |
109 | LUA_RETURN(ps, LUA_FAIL); | |
110 | ||
111 | lua_notice(np, "%s", msg); | |
112 | ||
113 | LUA_RETURN(ps, LUA_OK); | |
114 | } | |
115 | ||
47990928 CP |
116 | static int lua_privmsgcmd(lua_State *ps) { |
117 | const char *n, *msg; | |
118 | nick *np; | |
119 | ||
120 | if(!lua_isstring(ps, 1) || !lua_isstring(ps, 2)) | |
121 | LUA_RETURN(ps, LUA_FAIL); | |
122 | ||
123 | n = lua_tostring(ps, 1); | |
124 | msg = lua_tostring(ps, 2); | |
125 | ||
126 | np = getnickbynick(n); | |
127 | if(!np || !lua_lineok(msg)) | |
128 | LUA_RETURN(ps, LUA_FAIL); | |
129 | ||
130 | lua_message(np, "%s", msg); | |
131 | ||
132 | LUA_RETURN(ps, LUA_OK); | |
133 | } | |
134 | ||
2da56f0d CP |
135 | static int lua_kill(lua_State *ps) { |
136 | const char *n, *msg; | |
137 | nick *np; | |
138 | ||
139 | if(!lua_isstring(ps, 1) || !lua_isstring(ps, 2)) | |
140 | LUA_RETURN(ps, LUA_FAIL); | |
141 | ||
142 | n = lua_tostring(ps, 1); | |
143 | msg = lua_tostring(ps, 2); | |
144 | ||
145 | np = getnickbynick(n); | |
146 | if(!np) | |
147 | LUA_RETURN(ps, LUA_FAIL); | |
148 | ||
149 | if(IsOper(np) || IsService(np) || IsXOper(np)) | |
150 | LUA_RETURN(ps, LUA_FAIL); | |
151 | ||
152 | if(!lua_lineok(msg)) | |
153 | LUA_RETURN(ps, LUA_FAIL); | |
154 | ||
155 | killuser(lua_nick, np, "%s", msg); | |
156 | ||
157 | LUA_RETURN(ps, lua_cmsg(LUA_PUKECHAN, "lua-KILL: %s (%s)", np->nick, msg)); | |
158 | } | |
159 | ||
511316a4 CP |
160 | static int lua_kick(lua_State *ps) { |
161 | const char *n, *msg, *chan; | |
162 | nick *np; | |
163 | channel *cp; | |
46751959 | 164 | int dochecks = 1; |
511316a4 CP |
165 | |
166 | if(!lua_isstring(ps, 1) || !lua_isstring(ps, 2) || !lua_isstring(ps, 3)) | |
167 | LUA_RETURN(ps, LUA_FAIL); | |
168 | ||
169 | chan = lua_tostring(ps, 1); | |
170 | n = lua_tostring(ps, 2); | |
171 | msg = lua_tostring(ps, 3); | |
172 | ||
46751959 CP |
173 | if(lua_isboolean(ps, 4) && !lua_toboolean(ps, 4)) |
174 | dochecks = 0; | |
175 | ||
511316a4 CP |
176 | np = getnickbynick(n); |
177 | if(!np) | |
178 | LUA_RETURN(ps, LUA_FAIL); | |
179 | ||
46751959 CP |
180 | if(dochecks && (IsOper(np) || IsXOper(np) || IsService(np))) |
181 | LUA_RETURN(ps, LUA_FAIL); | |
182 | ||
511316a4 CP |
183 | cp = findchannel((char *)chan); |
184 | if(!cp) | |
185 | LUA_RETURN(ps, LUA_FAIL); | |
186 | ||
187 | if(!lua_lineok(msg)) | |
188 | LUA_RETURN(ps, LUA_FAIL); | |
189 | ||
190 | localkickuser(lua_nick, cp, np, msg); | |
191 | ||
192 | LUA_RETURN(ps, LUA_OK); | |
193 | } | |
194 | ||
195 | static int lua_invite(lua_State *ps) { | |
196 | nick *np; | |
197 | channel *cp; | |
198 | ||
199 | if(!lua_isstring(ps, 1) || !lua_isstring(ps, 2)) | |
200 | LUA_RETURN(ps, LUA_FAIL); | |
201 | ||
202 | np = getnickbynick((char *)lua_tostring(ps, 1)); | |
203 | if(!np) | |
204 | LUA_RETURN(ps, LUA_FAIL); | |
205 | ||
206 | cp = findchannel((char *)lua_tostring(ps, 2)); | |
207 | if(!cp) | |
208 | LUA_RETURN(ps, LUA_FAIL); | |
209 | ||
210 | localinvite(lua_nick, cp, np); | |
211 | ||
212 | LUA_RETURN(ps, LUA_OK); | |
213 | } | |
214 | ||
2da56f0d CP |
215 | static int lua_gline(lua_State *ps) { |
216 | const char *reason; | |
217 | nick *target; | |
218 | char mask[512]; | |
219 | int duration, usercount = 0; | |
220 | host *hp; | |
221 | ||
222 | if(!lua_isstring(ps, 1) || !lua_isint(ps, 2) || !lua_isstring(ps, 3)) | |
223 | LUA_RETURN(ps, LUA_FAIL); | |
224 | ||
225 | duration = lua_toint(ps, 2); | |
79e512d7 | 226 | if((duration < 1) || (duration > 31 * 86400)) |
2da56f0d CP |
227 | LUA_RETURN(ps, LUA_FAIL); |
228 | ||
229 | reason = lua_tostring(ps, 3); | |
230 | if(!lua_lineok(reason) || !reason) | |
231 | LUA_RETURN(ps, LUA_FAIL); | |
232 | ||
233 | target = getnickbynick(lua_tostring(ps, 1)); | |
234 | if(!target || (IsOper(target) || IsXOper(target) || IsService(target))) | |
235 | LUA_RETURN(ps, LUA_FAIL); | |
236 | ||
237 | hp = target->host; | |
238 | if(!hp) | |
239 | LUA_RETURN(ps, LUA_FAIL); | |
240 | ||
241 | usercount = hp->clonecount; | |
242 | if(usercount > 10) { /* (decent) trusted host */ | |
243 | int j; | |
244 | nick *np; | |
245 | ||
246 | usercount = 0; | |
247 | ||
248 | for (j=0;j<NICKHASHSIZE;j++) | |
249 | for (np=nicktable[j];np;np=np->next) | |
250 | if (np && (np->host == hp) && (!ircd_strcmp(np->ident, target->ident))) | |
251 | usercount++; | |
252 | ||
253 | if(usercount > 50) | |
254 | LUA_RETURN(ps, LUA_FAIL); | |
255 | ||
526e7c1d | 256 | snprintf(mask, sizeof(mask), "*%s@%s", target->ident, IPtostr(target->p_ipaddr)); |
2da56f0d | 257 | } else { |
526e7c1d | 258 | snprintf(mask, sizeof(mask), "*@%s", IPtostr(target->p_ipaddr)); |
2da56f0d CP |
259 | } |
260 | ||
b822d7a4 | 261 | irc_send("%s GL * +%s %d :%s", mynumeric->content, mask, duration, reason); |
2da56f0d CP |
262 | LUA_RETURN(ps, lua_cmsg(LUA_PUKECHAN, "lua-GLINE: %s (%d users, %d seconds -- %s)", mask, usercount, duration, reason)); |
263 | } | |
264 | ||
265 | static int lua_getchaninfo(lua_State *ps) { | |
266 | channel *cp; | |
267 | ||
268 | if(!lua_isstring(ps, 1)) | |
269 | return 0; | |
270 | ||
271 | cp = findchannel((char *)lua_tostring(ps, 1)); | |
272 | if(!cp) | |
273 | return 0; | |
274 | ||
275 | LUA_PUSHCHAN(ps, cp); | |
276 | ||
277 | return 1; | |
278 | } | |
279 | ||
280 | static int lua_opchan(lua_State *ps) { | |
281 | channel *cp; | |
282 | nick *np; | |
283 | ||
284 | if(!lua_isstring(ps, 1) || !lua_isstring(ps, 2)) | |
285 | LUA_RETURN(ps, LUA_FAIL); | |
286 | ||
287 | cp = findchannel((char *)lua_tostring(ps, 1)); | |
288 | if(!cp) | |
289 | LUA_RETURN(ps, LUA_FAIL); | |
290 | ||
291 | np = getnickbynick((char *)lua_tostring(ps, 2)); | |
292 | if(!np) | |
293 | LUA_RETURN(ps, LUA_FAIL); | |
294 | ||
295 | localsetmodes(lua_nick, cp, np, MC_OP); | |
296 | LUA_RETURN(ps, LUA_OK); | |
297 | } | |
298 | ||
d6998bd1 CP |
299 | static int lua_deopchan(lua_State *ps) { |
300 | channel *cp; | |
301 | nick *np; | |
302 | ||
303 | if(!lua_isstring(ps, 1) || !lua_isstring(ps, 2)) | |
304 | LUA_RETURN(ps, LUA_FAIL); | |
305 | ||
306 | cp = findchannel((char *)lua_tostring(ps, 1)); | |
307 | if(!cp) | |
308 | LUA_RETURN(ps, LUA_FAIL); | |
309 | ||
310 | np = getnickbynick((char *)lua_tostring(ps, 2)); | |
311 | if(!np) | |
312 | LUA_RETURN(ps, LUA_FAIL); | |
313 | ||
314 | localsetmodes(lua_nick, cp, np, MC_DEOP); | |
315 | LUA_RETURN(ps, LUA_OK); | |
316 | } | |
317 | ||
2da56f0d CP |
318 | static int lua_voicechan(lua_State *ps) { |
319 | channel *cp; | |
320 | nick *np; | |
321 | ||
322 | if(!lua_isstring(ps, 1) || !lua_isstring(ps, 2)) | |
323 | LUA_RETURN(ps, LUA_FAIL); | |
324 | ||
325 | cp = findchannel((char *)lua_tostring(ps, 1)); | |
326 | if(!cp) | |
327 | LUA_RETURN(ps, LUA_FAIL); | |
328 | ||
329 | np = getnickbynick((char *)lua_tostring(ps, 2)); | |
330 | if(!np) | |
331 | LUA_RETURN(ps, LUA_FAIL); | |
332 | ||
333 | localsetmodes(lua_nick, cp, np, MC_VOICE); | |
334 | LUA_RETURN(ps, LUA_OK); | |
335 | } | |
336 | ||
337 | static int lua_counthost(lua_State *ps) { | |
338 | long numeric; | |
339 | nick *np; | |
340 | ||
341 | if(!lua_islong(ps, 1)) | |
342 | return 0; | |
343 | ||
344 | numeric = lua_tolong(ps, 1); | |
345 | ||
346 | np = getnickbynumeric(numeric); | |
347 | if(!np) | |
348 | return 0; | |
349 | ||
350 | lua_pushint(ps, np->host->clonecount); | |
351 | return 1; | |
352 | } | |
353 | ||
354 | static int lua_versioninfo(lua_State *ps) { | |
355 | lua_pushstring(ps, LUA_VERSION); | |
356 | lua_pushstring(ps, LUA_BOTVERSION); | |
357 | lua_pushstring(ps, __DATE__); | |
358 | lua_pushstring(ps, __TIME__); | |
359 | ||
0225bed3 | 360 | lua_pushstring(ps, LUA_AUXVERSION); |
6269435f CP |
361 | |
362 | return 5; | |
363 | } | |
364 | ||
365 | static int lua_basepath(lua_State *ps) { | |
366 | lua_pushfstring(ps, "%s/", cpath->content); | |
367 | ||
368 | return 1; | |
2da56f0d CP |
369 | } |
370 | ||
371 | /* O(n) */ | |
372 | static int lua_getuserbyauth(lua_State *l) { | |
373 | const char *acc; | |
374 | nick *np; | |
375 | int i, found = 0; | |
376 | ||
377 | if(!lua_isstring(l, 1)) | |
378 | return 0; | |
379 | ||
1fbb1306 CP |
380 | acc = lua_tostring(l, 1); |
381 | ||
2da56f0d CP |
382 | for(i=0;i<NICKHASHSIZE;i++) { |
383 | for(np=nicktable[i];np;np=np->next) { | |
384 | if(np && np->authname && !ircd_strcmp(np->authname, acc)) { | |
385 | LUA_PUSHNICK(l, np); | |
386 | found++; | |
387 | } | |
388 | } | |
389 | } | |
390 | ||
391 | return found; | |
392 | } | |
393 | ||
394 | static int lua_getnickchans(lua_State *l) { | |
395 | nick *np; | |
396 | int i; | |
397 | channel **channels; | |
398 | ||
399 | if(!lua_islong(l, 1)) | |
400 | return 0; | |
401 | ||
402 | np = getnickbynumeric(lua_tolong(l, 1)); | |
403 | if(!np) | |
404 | return 0; | |
405 | ||
406 | channels = (channel **)np->channels->content; | |
407 | for(i=0;i<np->channels->cursi;i++) | |
408 | lua_pushstring(l, channels[i]->index->name->content); | |
409 | ||
410 | return np->channels->cursi; | |
411 | } | |
412 | ||
4561a594 CP |
413 | static int lua_getnickchanindex(lua_State *l) { |
414 | nick *np; | |
415 | int offset; | |
416 | ||
417 | if(!lua_islong(l, 1) || !lua_isint(l, 2)) | |
418 | return 0; | |
419 | ||
420 | np = getnickbynumeric(lua_tolong(l, 1)); | |
421 | if(!np) | |
422 | return 0; | |
423 | ||
424 | offset = lua_toint(l, 2); | |
425 | if((offset < 0) || (offset >= np->channels->cursi)) | |
426 | return 0; | |
427 | ||
428 | lua_pushstring(l, ((channel **)np->channels->content)[offset]->index->name->content); | |
429 | ||
430 | return 1; | |
431 | } | |
432 | ||
c5706d5b CP |
433 | int hashindex; |
434 | nick *lasthashnick; | |
f6a3d1f7 | 435 | static int geoipext; |
c5706d5b | 436 | |
ec91b2e4 | 437 | struct lua_pusher *nickhashpusher[MAX_PUSHER]; |
30a59b87 | 438 | |
c5706d5b CP |
439 | static int lua_getnextnick(lua_State *l) { |
440 | if(!lasthashnick && (hashindex != -1)) | |
441 | return 0; | |
442 | ||
443 | do { | |
444 | if(!lasthashnick) { | |
445 | hashindex++; | |
446 | if(hashindex >= NICKHASHSIZE) | |
447 | return 0; | |
448 | lasthashnick = nicktable[hashindex]; | |
449 | } else { | |
450 | lasthashnick = lasthashnick->next; | |
451 | } | |
452 | } while(!lasthashnick); | |
453 | ||
ec91b2e4 | 454 | return lua_usepusher(l, nickhashpusher, lasthashnick); |
c5706d5b CP |
455 | } |
456 | ||
457 | static int lua_getfirstnick(lua_State *l) { | |
c5706d5b CP |
458 | hashindex = -1; |
459 | lasthashnick = NULL; | |
460 | ||
ec91b2e4 | 461 | lua_setupnickpusher(l, 1, nickhashpusher, MAX_PUSHER); |
f6a3d1f7 | 462 | geoipext = findnickext("geoip"); |
30a59b87 | 463 | |
c5706d5b CP |
464 | return lua_getnextnick(l); |
465 | } | |
466 | ||
ec91b2e4 CP |
467 | int chanhashindex; |
468 | chanindex *lasthashchan; | |
469 | ||
470 | struct lua_pusher *chanhashpusher[MAX_PUSHER]; | |
471 | ||
472 | static int lua_getnextchan(lua_State *l) { | |
473 | if(!lasthashchan && (chanhashindex != -1)) | |
474 | return 0; | |
475 | ||
476 | do { | |
477 | if(!lasthashchan) { | |
478 | chanhashindex++; | |
479 | if(chanhashindex >= CHANNELHASHSIZE) | |
480 | return 0; | |
481 | lasthashchan = chantable[chanhashindex]; | |
482 | } else { | |
483 | lasthashchan = lasthashchan->next; | |
484 | } | |
485 | } while(!lasthashchan || !lasthashchan->channel); | |
486 | ||
487 | return lua_usepusher(l, chanhashpusher, lasthashchan); | |
488 | } | |
489 | ||
490 | static int lua_getfirstchan(lua_State *l) { | |
491 | chanhashindex = -1; | |
492 | lasthashchan = NULL; | |
493 | ||
494 | lua_setupchanpusher(l, 1, chanhashpusher, MAX_PUSHER); | |
495 | ||
496 | return lua_getnextchan(l); | |
497 | } | |
498 | ||
8ac8cd09 CP |
499 | static int lua_getnickchancount(lua_State *l) { |
500 | nick *np; | |
501 | ||
502 | if(!lua_islong(l, 1)) | |
503 | return 0; | |
504 | ||
505 | np = getnickbynumeric(lua_tolong(l, 1)); | |
506 | if(!np) | |
507 | return 0; | |
508 | ||
509 | lua_pushint(l, np->channels->cursi); | |
510 | ||
511 | return 1; | |
512 | } | |
513 | ||
2da56f0d CP |
514 | static int lua_gethostusers(lua_State *l) { |
515 | nick *np; | |
516 | int count; | |
517 | ||
518 | if(!lua_islong(l, 1)) | |
519 | return 0; | |
520 | ||
521 | np = getnickbynumeric(lua_tolong(l, 1)); | |
522 | if(!np || !np->host || !np->host->nicks) | |
523 | return 0; | |
524 | ||
525 | np = np->host->nicks; | |
526 | count = np->host->clonecount; | |
527 | ||
528 | do { | |
529 | LUA_PUSHNICK(l, np); | |
530 | np = np->nextbyhost; | |
531 | } while(np); | |
532 | ||
533 | return count; | |
534 | } | |
535 | ||
46751959 CP |
536 | static int lua_getnickcountry(lua_State *l) { |
537 | nick *np; | |
538 | int ext; | |
539 | ||
540 | ext = findnickext("geoip"); | |
541 | if(ext == -1) | |
542 | return 0; | |
543 | ||
544 | if(!lua_islong(l, 1)) | |
545 | return 0; | |
546 | ||
547 | np = getnickbynumeric(lua_tolong(l, 1)); | |
548 | if(!np) | |
549 | return 0; | |
550 | ||
abc75f54 | 551 | lua_pushint(l, (long)np->exts[ext]); |
46751959 CP |
552 | return 1; |
553 | } | |
554 | ||
ebe20bb8 CP |
555 | static int lua_chanfix(lua_State *ps) { |
556 | channel *cp; | |
557 | nick *np; | |
558 | ||
559 | if(!lua_isstring(ps, 1)) | |
560 | LUA_RETURN(ps, LUA_FAIL); | |
561 | ||
562 | cp = findchannel((char *)lua_tostring(ps, 1)); | |
1f57d0e2 | 563 | if(!cp || !cp->index) |
ebe20bb8 CP |
564 | LUA_RETURN(ps, LUA_FAIL); |
565 | ||
566 | np = getnickbynick(LUA_CHANFIXBOT); | |
567 | if(!np) | |
568 | LUA_RETURN(ps, LUA_FAIL); | |
569 | ||
570 | lua_message(np, "chanfix %s", cp->index->name->content); | |
571 | ||
572 | LUA_RETURN(ps, LUA_OK); | |
573 | } | |
574 | ||
575 | static int lua_clearmode(lua_State *ps) { | |
576 | channel *cp; | |
577 | int i; | |
578 | nick *np; | |
579 | unsigned long *lp; | |
580 | modechanges changes; | |
581 | ||
582 | if(!lua_isstring(ps, 1)) | |
583 | LUA_RETURN(ps, LUA_FAIL); | |
584 | ||
585 | cp = findchannel((char *)lua_tostring(ps, 1)); | |
1f57d0e2 | 586 | if(!cp || !cp->users) |
ebe20bb8 CP |
587 | LUA_RETURN(ps, LUA_FAIL); |
588 | ||
589 | localsetmodeinit(&changes, cp, lua_nick); | |
590 | ||
591 | localdosetmode_key(&changes, NULL, MCB_DEL); | |
592 | localdosetmode_simple(&changes, 0, CHANMODE_INVITEONLY | CHANMODE_LIMIT); | |
593 | ||
594 | while(cp->bans) | |
595 | localdosetmode_ban(&changes, bantostring(cp->bans), MCB_DEL); | |
596 | ||
597 | for(i=0,lp=cp->users->content;i<cp->users->hashsize;i++,lp++) | |
598 | if((*lp != nouser) && (*lp & CUMODE_OP)) { | |
599 | np = getnickbynumeric(*lp); | |
600 | if(np && !IsService(np)) | |
601 | localdosetmode_nick(&changes, np, MC_DEOP); | |
602 | } | |
603 | ||
604 | localsetmodeflush(&changes, 1); | |
605 | ||
606 | LUA_RETURN(ps, LUA_OK); | |
607 | } | |
608 | ||
46751959 CP |
609 | static int lua_ban(lua_State *ps) { |
610 | channel *cp; | |
611 | const char *mask; | |
612 | modechanges changes; | |
613 | int dir = MCB_ADD; | |
614 | ||
615 | if(!lua_isstring(ps, 1) || !lua_isstring(ps, 2)) | |
616 | LUA_RETURN(ps, LUA_FAIL); | |
617 | ||
618 | if(lua_isboolean(ps, 3) && lua_toboolean(ps, 3)) | |
619 | dir = MCB_DEL; | |
620 | ||
621 | cp = findchannel((char *)lua_tostring(ps, 1)); | |
622 | if(!cp) | |
623 | LUA_RETURN(ps, LUA_FAIL); | |
624 | ||
625 | mask = lua_tostring(ps, 2); | |
626 | if(!mask || !mask[0] || !lua_lineok(mask)) | |
627 | LUA_RETURN(ps, LUA_FAIL); | |
628 | ||
629 | localsetmodeinit(&changes, cp, lua_nick); | |
630 | localdosetmode_ban(&changes, mask, dir); | |
631 | localsetmodeflush(&changes, 1); | |
632 | ||
633 | LUA_RETURN(ps, LUA_OK); | |
634 | } | |
635 | ||
d0d1e2a3 CP |
636 | static int lua_topic(lua_State *ps) { |
637 | channel *cp; | |
638 | char *topic; | |
639 | ||
640 | if(!lua_isstring(ps, 1) || !lua_isstring(ps, 2)) | |
641 | LUA_RETURN(ps, LUA_FAIL); | |
642 | ||
643 | cp = findchannel((char *)lua_tostring(ps, 1)); | |
644 | if(!cp) | |
645 | LUA_RETURN(ps, LUA_FAIL); | |
646 | ||
647 | topic = (char *)lua_tostring(ps, 2); | |
648 | if(!topic || !lua_lineok(topic)) | |
649 | LUA_RETURN(ps, LUA_FAIL); | |
650 | ||
651 | localsettopic(lua_nick, cp, topic); | |
652 | ||
653 | LUA_RETURN(ps, LUA_OK); | |
654 | } | |
655 | ||
656 | static int lua_getuserchanmodes(lua_State *l) { | |
657 | nick *np; | |
658 | channel *cp; | |
659 | unsigned long *lp; | |
660 | ||
661 | if(!lua_islong(l, 1) || !lua_isstring(l, 2)) | |
662 | return 0; | |
663 | ||
664 | np = getnickbynumeric(lua_tolong(l, 1)); | |
665 | if(!np) | |
666 | return 0; | |
667 | ||
668 | cp = findchannel((char *)lua_tostring(l, 2)); | |
1f57d0e2 | 669 | if(!cp || !cp->users) |
d0d1e2a3 CP |
670 | return 0; |
671 | ||
672 | lp = getnumerichandlefromchanhash(cp->users, np->numeric); | |
673 | if(!lp) | |
674 | return 0; | |
675 | ||
676 | LUA_PUSHNICKCHANMODES(l, lp); | |
677 | return 1; | |
678 | } | |
679 | ||
c0d574fa CP |
680 | static int lua_getusermodes(lua_State *l) { |
681 | nick *np; | |
682 | ||
683 | if(!lua_islong(l, 1)) | |
684 | return 0; | |
685 | ||
686 | np = getnickbynumeric(lua_tolong(l, 1)); | |
687 | if(!np) | |
688 | return 0; | |
689 | ||
690 | lua_pushstring(l, printflags(np->umodes, umodeflags)); | |
691 | return 1; | |
692 | } | |
693 | ||
d0d1e2a3 CP |
694 | static int lua_getnickbynick(lua_State *l) { |
695 | nick *np; | |
696 | ||
697 | if(!lua_isstring(l, 1)) | |
698 | return 0; | |
699 | ||
700 | np = getnickbynick(lua_tostring(l, 1)); | |
701 | if(!np) | |
702 | return 0; | |
703 | ||
704 | LUA_PUSHNICK(l, np); | |
705 | return 1; | |
706 | } | |
707 | ||
d4aef858 CP |
708 | static int lua_getnickbynumeric(lua_State *l) { |
709 | nick *np; | |
710 | ||
711 | if(!lua_islong(l, 1)) | |
712 | return 0; | |
713 | ||
714 | np = getnickbynumeric(lua_tolong(l, 1)); | |
715 | if(!np) | |
716 | return 0; | |
717 | ||
718 | LUA_PUSHNICK(l, np); | |
719 | return 1; | |
720 | } | |
721 | ||
1f57d0e2 CP |
722 | static int lua_fastgetnickbynumeric(lua_State *l) { |
723 | static struct lua_pusher *ourpusher[MAX_PUSHER]; | |
724 | nick *np; | |
725 | ||
726 | if(!lua_islong(l, 1)) | |
727 | return 0; | |
728 | ||
729 | np = getnickbynumeric(lua_tolong(l, 1)); | |
730 | if(!np) | |
731 | return 0; | |
732 | ||
733 | lua_setupnickpusher(l, 2, ourpusher, MAX_PUSHER); | |
734 | return lua_usepusher(l, ourpusher, np); | |
735 | } | |
736 | ||
c0e81f9a CP |
737 | int channelnicklistindex, channelnicklistcount = -1; |
738 | channel *channelnicklist; | |
739 | ||
740 | struct lua_pusher *channelnickpusher[MAX_PUSHER]; | |
741 | ||
742 | static int lua_getnextchannick(lua_State *l) { | |
743 | nick *np; | |
744 | ||
745 | do { | |
746 | channelnicklistindex++; | |
747 | ||
748 | if(channelnicklistindex >= channelnicklistcount) | |
749 | return 0; | |
750 | } while((channelnicklist->users->content[channelnicklistindex] == nouser) || !(np = getnickbynumeric(channelnicklist->users->content[channelnicklistindex]))); | |
751 | ||
752 | return lua_usepusher(l, channelnickpusher, np); | |
753 | } | |
754 | ||
755 | static int lua_getfirstchannick(lua_State *l) { | |
756 | if(!lua_isstring(l, 1)) | |
757 | return 0; | |
758 | ||
759 | channelnicklist = findchannel((char *)lua_tostring(l, 1)); | |
760 | if(!channelnicklist|| !channelnicklist->users) | |
761 | return 0; | |
762 | ||
763 | channelnicklistindex = -1; | |
764 | channelnicklistcount = channelnicklist->users->hashsize; | |
765 | ||
766 | lua_setupnickpusher(l, 2, channelnickpusher, MAX_PUSHER); | |
767 | ||
768 | return lua_getnextchannick(l); | |
769 | } | |
770 | ||
1f57d0e2 CP |
771 | static int lua_nickonchan(lua_State *l) { |
772 | int success = 0; | |
773 | if(lua_islong(l, 1) && lua_isstring(l, 2)) { | |
774 | channel *cp = findchannel((char *)lua_tostring(l, 2)); | |
775 | if(cp && cp->users) { | |
776 | unsigned long *lp = getnumerichandlefromchanhash(cp->users, lua_tolong(l, 1)); | |
777 | if(lp) | |
778 | success = 1; | |
779 | } | |
780 | } | |
781 | ||
782 | lua_pushboolean(l, success); | |
783 | return 1; | |
784 | } | |
785 | ||
8e965b0c CP |
786 | static int lua_simplechanmode(lua_State *ps) { |
787 | channel *cp; | |
788 | char *modes; | |
789 | flag_t add = 0, del = ~add; | |
790 | flag_t permitted = CHANMODE_NOEXTMSG | CHANMODE_TOPICLIMIT | CHANMODE_SECRET | CHANMODE_PRIVATE | CHANMODE_INVITEONLY | CHANMODE_MODERATE | CHANMODE_NOCOLOUR | CHANMODE_NOCTCP | CHANMODE_REGONLY | CHANMODE_DELJOINS | CHANMODE_NOQUITMSG | CHANMODE_NONOTICE; | |
791 | modechanges changes; | |
792 | ||
793 | if(!lua_isstring(ps, 1) || !lua_isstring(ps, 2)) | |
794 | LUA_RETURN(ps, LUA_FAIL); | |
795 | ||
796 | cp = findchannel((char *)lua_tostring(ps, 1)); | |
797 | if(!cp) | |
798 | LUA_RETURN(ps, LUA_FAIL); | |
799 | ||
800 | modes = (char *)lua_tostring(ps, 2); | |
801 | if(!modes) | |
802 | LUA_RETURN(ps, LUA_FAIL); | |
803 | ||
804 | if(setflags(&add, permitted, modes, cmodeflags, REJECT_DISALLOWED|REJECT_UNKNOWN) != REJECT_NONE) | |
805 | LUA_RETURN(ps, LUA_FAIL); | |
806 | ||
807 | if(setflags(&del, permitted, modes, cmodeflags, REJECT_DISALLOWED|REJECT_UNKNOWN) != REJECT_NONE) | |
808 | LUA_RETURN(ps, LUA_FAIL); | |
809 | ||
810 | localsetmodeinit(&changes, cp, lua_nick); | |
811 | localdosetmode_simple(&changes, add, ~del); | |
812 | localsetmodeflush(&changes, 1); | |
813 | ||
814 | LUA_RETURN(ps, LUA_OK); | |
815 | } | |
816 | ||
524fb1bd CP |
817 | static int lua_sethost(lua_State *ps) { |
818 | char *ident, *host; | |
819 | nick *np; | |
820 | ||
821 | if(!lua_islong(ps, 1) || !lua_isstring(ps, 2) || !lua_isstring(ps, 3)) | |
822 | LUA_RETURN(ps, LUA_FAIL); | |
823 | ||
824 | np = getnickbynumeric(lua_tolong(ps, 1)); | |
825 | if(!np) | |
826 | LUA_RETURN(ps, LUA_FAIL); | |
827 | ||
828 | ident = (char *)lua_tostring(ps, 2); | |
829 | host = (char *)lua_tostring(ps, 3); | |
830 | if(!lua_lineok(ident) || !lua_lineok(host)) | |
831 | LUA_RETURN(ps, LUA_FAIL); | |
832 | ||
833 | sethostuser(np, ident, host); | |
834 | ||
835 | LUA_RETURN(ps, LUA_OK); | |
836 | } | |
837 | ||
2da56f0d CP |
838 | void lua_registercommands(lua_State *l) { |
839 | lua_register(l, "irc_smsg", lua_smsg); | |
840 | lua_register(l, "irc_skill", lua_skill); | |
841 | ||
842 | lua_register(l, "chanmsg", lua_chanmsg); | |
2da56f0d | 843 | lua_register(l, "versioninfo", lua_versioninfo); |
6269435f | 844 | lua_register(l, "basepath", lua_basepath); |
2da56f0d CP |
845 | |
846 | lua_register(l, "irc_report", lua_chanmsg); | |
847 | lua_register(l, "irc_ctcp", lua_ctcp); | |
848 | lua_register(l, "irc_kill", lua_kill); | |
511316a4 CP |
849 | lua_register(l, "irc_kick", lua_kick); |
850 | lua_register(l, "irc_invite", lua_invite); | |
2da56f0d CP |
851 | lua_register(l, "irc_gline", lua_gline); |
852 | lua_register(l, "irc_getchaninfo", lua_getchaninfo); | |
853 | lua_register(l, "irc_counthost", lua_counthost); | |
854 | lua_register(l, "irc_getuserbyauth", lua_getuserbyauth); | |
855 | lua_register(l, "irc_notice", lua_noticecmd); | |
47990928 | 856 | lua_register(l, "irc_privmsg", lua_privmsgcmd); |
2da56f0d CP |
857 | lua_register(l, "irc_opchan", lua_opchan); |
858 | lua_register(l, "irc_voicechan", lua_voicechan); | |
ebe20bb8 CP |
859 | lua_register(l, "irc_chanfix", lua_chanfix); |
860 | lua_register(l, "irc_clearmode", lua_clearmode); | |
46751959 | 861 | lua_register(l, "irc_ban", lua_ban); |
d6998bd1 | 862 | lua_register(l, "irc_deopchan", lua_deopchan); |
d0d1e2a3 CP |
863 | lua_register(l, "irc_topic", lua_topic); |
864 | ||
865 | lua_register(l, "irc_getnickbynick", lua_getnickbynick); | |
d4aef858 | 866 | lua_register(l, "irc_getnickbynumeric", lua_getnickbynumeric); |
d0d1e2a3 CP |
867 | lua_register(l, "irc_getfirstnick", lua_getfirstnick); |
868 | lua_register(l, "irc_getnextnick", lua_getnextnick); | |
2da56f0d CP |
869 | |
870 | lua_register(l, "irc_getnickchans", lua_getnickchans); | |
4561a594 | 871 | lua_register(l, "irc_getnickchanindex", lua_getnickchanindex); |
8ac8cd09 | 872 | lua_register(l, "irc_getnickchancount", lua_getnickchancount); |
c5706d5b | 873 | |
d0d1e2a3 CP |
874 | lua_register(l, "irc_getuserchanmodes", lua_getuserchanmodes); |
875 | ||
c0e81f9a CP |
876 | lua_register(l, "irc_getfirstchannick", lua_getfirstchannick); |
877 | lua_register(l, "irc_getnextchannick", lua_getnextchannick); | |
c5706d5b | 878 | |
2da56f0d | 879 | lua_register(l, "irc_gethostusers", lua_gethostusers); |
46751959 | 880 | lua_register(l, "irc_getnickcountry", lua_getnickcountry); |
ec91b2e4 CP |
881 | |
882 | lua_register(l, "irc_getfirstchan", lua_getfirstchan); | |
883 | lua_register(l, "irc_getnextchan", lua_getnextchan); | |
c0d574fa | 884 | lua_register(l, "irc_getusermodes", lua_getusermodes); |
1f57d0e2 CP |
885 | lua_register(l, "irc_nickonchan", lua_nickonchan); |
886 | ||
887 | lua_register(l, "irc_fastgetnickbynumeric", lua_fastgetnickbynumeric); | |
8e965b0c CP |
888 | |
889 | lua_register(l, "irc_simplechanmode", lua_simplechanmode); | |
524fb1bd | 890 | lua_register(l, "irc_sethost", lua_sethost); |
2da56f0d CP |
891 | } |
892 | ||
893 | /* --- */ | |
894 | ||
895 | static int lua_smsg(lua_State *ps) { | |
896 | if(!lua_isstring(ps, 1) || !lua_isstring(ps, 2)) | |
897 | LUA_RETURN(ps, LUA_FAIL); | |
898 | ||
899 | LUA_RETURN(ps, lua_cmsg((char *)lua_tostring(ps, 2), "%s", lua_tostring(ps, 1))); | |
900 | } | |
901 | ||
902 | static int lua_skill(lua_State *ps) { | |
903 | const char *n, *msg; | |
904 | nick *np; | |
905 | ||
906 | if(!lua_isstring(ps, 1) || !lua_isstring(ps, 2)) | |
907 | LUA_RETURN(ps, LUA_FAIL); | |
908 | ||
909 | n = lua_tostring(ps, 1); | |
910 | msg = lua_tostring(ps, 2); | |
911 | ||
912 | np = getnickbynick(n); | |
913 | if(!np) | |
914 | LUA_RETURN(ps, LUA_FAIL); | |
915 | ||
916 | if(IsOper(np) || IsService(np) || IsXOper(np)) | |
917 | LUA_RETURN(ps, LUA_FAIL); | |
918 | ||
919 | if(!lua_lineok(msg)) | |
920 | LUA_RETURN(ps, LUA_FAIL); | |
921 | ||
922 | killuser(lua_nick, np, "%s", msg); | |
923 | ||
924 | LUA_RETURN(ps, LUA_OK); | |
925 | } | |
926 | ||
30a59b87 | 927 | #define PUSHER_STRING 1 |
eebdff1f | 928 | #define PUSHER_REALNAME 2 |
30a59b87 CP |
929 | #define PUSHER_IP 3 |
930 | #define PUSHER_LONG 4 | |
eebdff1f CP |
931 | #define PUSHER_HOSTNAME 5 |
932 | #define PUSHER_SSTRING 6 | |
ec91b2e4 CP |
933 | #define PUSHER_TOTALUSERS 7 |
934 | #define PUSHER_TOPIC 8 | |
de2bb012 | 935 | #define PUSHER_UMODES 9 |
f6a3d1f7 | 936 | #define PUSHER_COUNTRY 10 |
30a59b87 CP |
937 | |
938 | void lua_initnickpusher(void) { | |
939 | int i = 0; | |
940 | ||
941 | #define PUSH_NICKPUSHER(F2, O2) nickpusher[i].argtype = F2; nickpusher[i].structname = #O2; nickpusher[i].offset = offsetof(nick, O2); i++; | |
f6a3d1f7 | 942 | #define PUSH_NICKPUSHER_CUSTOM(F2, custom) nickpusher[i].argtype = F2; nickpusher[i].structname = custom; nickpusher[i].offset = 0; i++; |
30a59b87 CP |
943 | |
944 | PUSH_NICKPUSHER(PUSHER_STRING, nick); | |
945 | PUSH_NICKPUSHER(PUSHER_STRING, ident); | |
eebdff1f CP |
946 | PUSH_NICKPUSHER(PUSHER_HOSTNAME, host); |
947 | PUSH_NICKPUSHER(PUSHER_REALNAME, realname); | |
30a59b87 | 948 | PUSH_NICKPUSHER(PUSHER_STRING, authname); |
526e7c1d | 949 | PUSH_NICKPUSHER(PUSHER_IP, ipnode); |
30a59b87 | 950 | PUSH_NICKPUSHER(PUSHER_LONG, numeric); |
de2bb012 CP |
951 | PUSH_NICKPUSHER(PUSHER_LONG, timestamp); |
952 | PUSH_NICKPUSHER(PUSHER_LONG, accountts); | |
953 | PUSH_NICKPUSHER(PUSHER_UMODES, umodes); | |
f6a3d1f7 | 954 | PUSH_NICKPUSHER_CUSTOM(PUSHER_COUNTRY, "country"); |
30a59b87 | 955 | |
ebdfcaa9 | 956 | nickpushercount = i; |
30a59b87 CP |
957 | nickpusher[i].argtype = 0; |
958 | } | |
959 | ||
ebdfcaa9 | 960 | INLINE void lua_setuppusher(struct lua_pusher *pusherlist, lua_State *l, int index, struct lua_pusher **lp, int max, int pcount) { |
30a59b87 | 961 | int current = 0; |
30a59b87 CP |
962 | |
963 | if(max > 0) | |
964 | lp[0] = NULL; | |
965 | ||
533af02d | 966 | if(!lua_istable(l, index) || (max < 2)) |
30a59b87 CP |
967 | return; |
968 | ||
969 | lua_pushnil(l); | |
970 | ||
971 | max--; | |
972 | ||
533af02d | 973 | while(lua_next(l, index)) { |
ebdfcaa9 CP |
974 | if(lua_isint(l, -1)) { |
975 | int index = lua_toint(l, -1); | |
976 | if((index >= 0) && (index < pcount)) | |
977 | lp[current++] = &pusherlist[index]; | |
30a59b87 CP |
978 | } |
979 | ||
980 | lua_pop(l, 1); | |
981 | ||
982 | if(current == max) | |
983 | break; | |
984 | } | |
985 | ||
986 | lp[current] = NULL; | |
987 | } | |
988 | ||
ec91b2e4 | 989 | INLINE int lua_usepusher(lua_State *l, struct lua_pusher **lp, void *np) { |
1f38b1d7 CP |
990 | int i = 0; |
991 | ||
30a59b87 CP |
992 | while(*lp) { |
993 | void *offset = (void *)np + (*lp)->offset; | |
994 | ||
30a59b87 CP |
995 | switch((*lp)->argtype) { |
996 | case PUSHER_STRING: | |
997 | lua_pushstring(l, (char *)offset); | |
998 | break; | |
eebdff1f CP |
999 | case PUSHER_HOSTNAME: |
1000 | lua_pushstring(l, (*(host **)offset)->name->content); | |
1001 | break; | |
1002 | case PUSHER_REALNAME: | |
1003 | lua_pushstring(l, (*(realname **)offset)->name->content); | |
1004 | break; | |
30a59b87 | 1005 | case PUSHER_SSTRING: |
ec91b2e4 | 1006 | lua_pushstring(l, (*((sstring **)offset))->content); |
30a59b87 CP |
1007 | break; |
1008 | case PUSHER_LONG: | |
1009 | lua_pushlong(l, *((long *)offset)); | |
1010 | break; | |
1011 | case PUSHER_IP: | |
1c7f0e03 | 1012 | lua_pushstring(l, IPtostr((*((patricia_node_t **)offset))->prefix->sin)); |
30a59b87 | 1013 | break; |
ec91b2e4 CP |
1014 | case PUSHER_TOTALUSERS: |
1015 | lua_pushint(l, (*((channel **)offset))->users->totalusers); | |
1016 | break; | |
de2bb012 CP |
1017 | case PUSHER_UMODES: |
1018 | lua_pushstring(l, printflags(*((flag_t *)offset), umodeflags)); | |
1019 | break; | |
ec91b2e4 CP |
1020 | case PUSHER_TOPIC: |
1021 | if((*((channel **)offset))->topic) { | |
1022 | lua_pushstring(l, (*((channel **)offset))->topic->content); | |
1023 | } else { | |
1024 | lua_pushnil(l); | |
1025 | } | |
1026 | break; | |
f6a3d1f7 CP |
1027 | case PUSHER_COUNTRY: |
1028 | if(geoipext < 0) { | |
1029 | lua_pushint(l, -1); | |
1030 | } else { | |
1031 | lua_pushint(l, (long)((nick *)offset)->exts[geoipext]); | |
1032 | } | |
1033 | break; | |
30a59b87 CP |
1034 | } |
1035 | ||
1f38b1d7 | 1036 | i++; |
30a59b87 CP |
1037 | lp++; |
1038 | } | |
1f38b1d7 CP |
1039 | |
1040 | return i; | |
30a59b87 | 1041 | } |
24da5817 | 1042 | |
ec91b2e4 CP |
1043 | void lua_initchanpusher(void) { |
1044 | int i = 0; | |
1045 | ||
1046 | #define PUSH_CHANPUSHER(F2, O2, N2) chanpusher[i].argtype = F2; chanpusher[i].structname = N2; chanpusher[i].offset = offsetof(chanindex, O2); i++; | |
1047 | ||
1048 | PUSH_CHANPUSHER(PUSHER_SSTRING, name, "name"); | |
1049 | PUSH_CHANPUSHER(PUSHER_TOTALUSERS, channel, "totalusers"); | |
1050 | PUSH_CHANPUSHER(PUSHER_TOPIC, channel, "topic"); | |
1051 | ||
ebdfcaa9 | 1052 | chanpushercount = i; |
ec91b2e4 CP |
1053 | chanpusher[i].argtype = 0; |
1054 | } |