]>
Commit | Line | Data |
---|---|---|
84ca44e7 P |
1 | /* patricianick.c */ |
2 | ||
3 | #include "patricianick.h" | |
4 | #include "../irc/irc_config.h" | |
5 | #include "../lib/irc_string.h" | |
6 | #include "../core/error.h" | |
7 | #include "../core/nsmalloc.h" | |
8 | #include "../control/control.h" | |
9 | #include "../core/schedule.h" | |
10 | ||
11 | #include <stdio.h> | |
12 | #include <string.h> | |
13 | ||
32a0a978 | 14 | #define ALLOCUNIT 100 |
84ca44e7 P |
15 | |
16 | patricianick_t *freepatricianicks; | |
17 | int pnode_ext; | |
18 | int pnick_ext; | |
19 | ||
35a8a596 | 20 | int pn_cmd_nodeuserlist(void *source, int cargc, char **cargv); |
84ca44e7 P |
21 | |
22 | void _init() { | |
23 | nick *np, *nnp; | |
24 | int i; | |
25 | ||
26 | pnode_ext = registernodeext("patricianick"); | |
32a0a978 | 27 | if (pnode_ext == -1) { |
84ca44e7 P |
28 | Error("patricianick", ERR_FATAL, "Could not register a required node extension"); |
29 | return; | |
30 | } | |
31 | ||
32 | pnick_ext = registernickext("patricianick"); | |
32a0a978 | 33 | if (pnick_ext == -1) { |
84ca44e7 P |
34 | Error("patricianick", ERR_FATAL, "Could not register a required nick extension"); |
35 | return; | |
36 | } | |
37 | ||
38 | for (i=0;i<NICKHASHSIZE;i++) | |
39 | for (np=nicktable[i];np;np=nnp) { | |
40 | nnp=np->next; | |
41 | addnicktonode(np->ipnode, np); | |
42 | } | |
43 | ||
44 | registerhook(HOOK_NICK_NEWNICK, &pn_hook_newuser); | |
45 | registerhook(HOOK_NICK_LOSTNICK, &pn_hook_lostuser); | |
46 | ||
2394d0d8 | 47 | registercontrolhelpcmd("nodeuserlist", NO_OPER, 1, &pn_cmd_nodeuserlist, "Usage: nodeuserlist <ipv4|ipv6|cidr4|cidr6>\nLists all users on a given IP address or CIDR range."); |
84ca44e7 P |
48 | } |
49 | ||
50 | void _fini() { | |
51 | nsfreeall(POOL_PATRICIANICK); | |
32a0a978 | 52 | |
84ca44e7 P |
53 | if (pnode_ext != -1) { |
54 | releasenodeext(pnode_ext); | |
55 | } | |
56 | ||
57 | if (pnick_ext != -1) { | |
58 | releasenickext(pnick_ext); | |
59 | } | |
60 | ||
61 | deregisterhook(HOOK_NICK_NEWNICK, &pn_hook_newuser); | |
62 | deregisterhook(HOOK_NICK_LOSTNICK, &pn_hook_lostuser); | |
63 | ||
35a8a596 | 64 | deregistercontrolcmd("nodeuserlist", &pn_cmd_nodeuserlist); |
84ca44e7 P |
65 | } |
66 | ||
67 | patricianick_t *getpatricianick() { | |
68 | int i; | |
69 | patricianick_t *pnp; | |
70 | ||
71 | if (freepatricianicks==NULL) { | |
72 | freepatricianicks=(patricianick_t *)nsmalloc(POOL_PATRICIANICK, ALLOCUNIT*sizeof(patricianick_t)); | |
73 | for(i=0;i<ALLOCUNIT-1;i++) { | |
32a0a978 | 74 | freepatricianicks[i].identhash[0]=(nick *)&(freepatricianicks[i+1]); |
84ca44e7 | 75 | } |
32a0a978 | 76 | freepatricianicks[ALLOCUNIT-1].identhash[0]=NULL; |
84ca44e7 P |
77 | } |
78 | ||
79 | pnp=freepatricianicks; | |
32a0a978 | 80 | freepatricianicks=(patricianick_t *)pnp->identhash[0]; |
84ca44e7 | 81 | |
32a0a978 | 82 | memset(pnp, 0, sizeof(patricianick_t)); |
84ca44e7 P |
83 | return pnp; |
84 | } | |
85 | ||
86 | void addnicktonode(patricia_node_t *node, nick *np) { | |
32a0a978 C |
87 | unsigned long hash; |
88 | ||
89 | if (!(node->exts[pnode_ext])) { | |
84ca44e7 P |
90 | node->exts[pnode_ext] = getpatricianick(); |
91 | } | |
32a0a978 C |
92 | |
93 | hash = pn_getidenthash(np->ident); | |
94 | np->exts[pnick_ext] = ((patricianick_t *)node->exts[pnode_ext])->identhash[hash]; | |
95 | ((patricianick_t *)node->exts[pnode_ext])->identhash[hash] = np; | |
84ca44e7 P |
96 | } |
97 | ||
98 | void deletenickfromnode(patricia_node_t *node, nick *np) { | |
99 | nick **tnp; | |
32a0a978 C |
100 | int found, i; |
101 | ||
102 | found = 0; | |
84ca44e7 | 103 | |
32a0a978 C |
104 | for (tnp = &(((patricianick_t *)node->exts[pnode_ext])->identhash[pn_getidenthash(np->ident)]); *tnp; tnp = (nick **)(&((*tnp)->exts[pnick_ext]))) { |
105 | if (*tnp == np) { | |
84ca44e7 P |
106 | *tnp = np->exts[pnick_ext]; |
107 | found = 1; | |
108 | break; | |
109 | } | |
110 | } | |
111 | ||
112 | if (!found) { | |
32a0a978 C |
113 | Error("patricianick", ERR_ERROR, "Could not remove %s!%s from %s", np->nick, np->ident, IPtostr(node->prefix->sin)); |
114 | return; | |
84ca44e7 P |
115 | } |
116 | ||
32a0a978 C |
117 | for (i = 0; i < PATRICIANICK_HASHSIZE; i++) { |
118 | if (((patricianick_t *)node->exts[pnode_ext])->identhash[i]) { | |
119 | return; | |
120 | } | |
84ca44e7 P |
121 | } |
122 | ||
32a0a978 C |
123 | freepatricianick(node->exts[pnode_ext]); |
124 | node->exts[pnode_ext]= NULL; | |
84ca44e7 P |
125 | } |
126 | ||
127 | void freepatricianick(patricianick_t *pnp) { | |
32a0a978 | 128 | pnp->identhash[0]=(nick *)freepatricianicks; |
84ca44e7 P |
129 | freepatricianicks=pnp; |
130 | } | |
131 | ||
132 | void pn_hook_newuser(int hook, void *arg) { | |
133 | nick *np = (nick *)arg; | |
134 | ||
135 | addnicktonode(np->ipnode, np); | |
136 | } | |
137 | ||
138 | void pn_hook_lostuser(int hook, void *arg) { | |
139 | nick *np = (nick *)arg; | |
140 | ||
141 | deletenickfromnode(np->ipnode, np); | |
142 | } | |
143 | ||
35a8a596 | 144 | int pn_cmd_nodeuserlist(void *source, int cargc, char **cargv) { |
84ca44e7 P |
145 | nick *np=(nick *)source; |
146 | struct irc_in_addr sin; | |
147 | unsigned char bits; | |
32a0a978 | 148 | unsigned int count, i; |
84ca44e7 P |
149 | patricia_node_t *head, *node; |
150 | patricianick_t *pnp; | |
151 | nick *npp; | |
84ca44e7 P |
152 | |
153 | if (cargc < 1) { | |
2394d0d8 | 154 | return CMD_USAGE; |
84ca44e7 P |
155 | } |
156 | ||
157 | if (ipmask_parse(cargv[0], &sin, &bits) == 0) { | |
158 | controlreply(np, "Invalid mask."); | |
2394d0d8 | 159 | return CMD_ERROR; |
84ca44e7 P |
160 | } |
161 | ||
162 | head = refnode(iptree, &sin, bits); | |
32a0a978 | 163 | count = 0; |
84ca44e7 P |
164 | |
165 | PATRICIA_WALK(head, node) | |
166 | { | |
167 | pnp = node->exts[pnode_ext]; | |
32a0a978 C |
168 | if (pnp) { |
169 | if (count < PATRICIANICK_MAXRESULTS) { | |
170 | for (i = 0; i < PATRICIANICK_HASHSIZE; i++) { | |
171 | for (npp = pnp->identhash[i]; npp; npp=npp->exts[pnick_ext]) { | |
172 | controlreply(np, "%s!%s@%s%s%s (%s)", npp->nick, npp->ident, npp->host->name->content, IsAccount(npp) ? "/" : "", npp->authname, IPtostr(node->prefix->sin)); | |
173 | } | |
174 | } | |
175 | ||
176 | count += node->usercount; | |
177 | ||
178 | if (count >= PATRICIANICK_MAXRESULTS) { | |
179 | controlreply(np, "Too many results, output truncated"); | |
180 | } | |
181 | } else { | |
182 | count += node->usercount; | |
183 | } | |
84ca44e7 P |
184 | } |
185 | } | |
186 | PATRICIA_WALK_END; | |
8ac1acd3 | 187 | derefnode(iptree, head); |
84ca44e7 | 188 | |
32a0a978 | 189 | controlreply(np, "Total users on %s: %d", cargv[0], count); |
84ca44e7 P |
190 | return CMD_OK; |
191 | } |