]>
Commit | Line | Data |
---|---|---|
1 | /* | |
2 | * CIDR functionality | |
3 | */ | |
4 | ||
5 | #include "newsearch.h" | |
6 | ||
7 | #include <stdio.h> | |
8 | #include <stdlib.h> | |
9 | #include <string.h> | |
10 | ||
11 | #include "../irc/irc_config.h" | |
12 | #include "../lib/irc_string.h" | |
13 | #include "../lib/irc_ipv6.h" | |
14 | ||
15 | struct cidr_localdata { | |
16 | struct irc_in_addr ip; | |
17 | unsigned char bits; | |
18 | }; | |
19 | ||
20 | void *cidr_exe(searchCtx *ctx, struct searchNode *thenode, void *theinput); | |
21 | void cidr_free(searchCtx *ctx, struct searchNode *thenode); | |
22 | ||
23 | struct searchNode *cidr_parse(searchCtx *ctx, int argc, char **argv) { | |
24 | struct searchNode *thenode, *convsn; | |
25 | struct cidr_localdata *c; | |
26 | struct irc_in_addr ip; | |
27 | unsigned char bits; | |
28 | char *p; | |
29 | int ret; | |
30 | ||
31 | if(argc != 1) { | |
32 | parseError = "usage: cidr ip/mask"; | |
33 | return NULL; | |
34 | } | |
35 | ||
36 | if (!(convsn=argtoconststr("cidr", ctx, argv[0], &p))) | |
37 | return NULL; | |
38 | ||
39 | ret = ipmask_parse(p, &ip, &bits); | |
40 | convsn->free(ctx, convsn); | |
41 | ||
42 | if(!ret) { | |
43 | parseError = "usage: cidr ip/mask"; | |
44 | return NULL; | |
45 | } | |
46 | ||
47 | if (!(thenode=(struct searchNode *)malloc(sizeof (struct searchNode)))) { | |
48 | parseError = "malloc: could not allocate memory for this search."; | |
49 | return NULL; | |
50 | } | |
51 | ||
52 | thenode->returntype = RETURNTYPE_BOOL; | |
53 | if (!(c = malloc(sizeof(struct cidr_localdata)))) { | |
54 | /* couldn't malloc() memory for thenode->localdata, so free thenode to avoid leakage */ | |
55 | parseError = "malloc: could not allocate memory for this search."; | |
56 | free(thenode); | |
57 | return NULL; | |
58 | } | |
59 | ||
60 | memcpy(&c->ip, &ip, sizeof(struct irc_in_addr)); | |
61 | c->bits = bits; | |
62 | ||
63 | thenode->localdata = (void *)c; | |
64 | thenode->free = cidr_free; | |
65 | thenode->exe = cidr_exe; | |
66 | ||
67 | return thenode; | |
68 | } | |
69 | ||
70 | void *cidr_exe(searchCtx *ctx, struct searchNode *thenode, void *theinput) { | |
71 | nick *np = (nick *)theinput; | |
72 | struct cidr_localdata *c = thenode->localdata; | |
73 | ||
74 | if(!ipmask_check(&np->ipaddress, &c->ip, c->bits)) | |
75 | return (void *)0; | |
76 | ||
77 | return (void *)1; | |
78 | } | |
79 | ||
80 | void cidr_free(searchCtx *ctx, struct searchNode *thenode) { | |
81 | free(thenode->localdata); | |
82 | free(thenode); | |
83 | } |