]>
Commit | Line | Data |
---|---|---|
cc3768c4 CP |
1 | /* |
2 | Geoip module | |
3 | Copyright (C) 2004-2006 Chris Porter. | |
4 | */ | |
5 | ||
6 | #include "../nick/nick.h" | |
7 | #include "../core/error.h" | |
8 | #include "../core/config.h" | |
9 | #include "../core/hooks.h" | |
10 | #include "../control/control.h" | |
87698d77 | 11 | #include "../lib/version.h" |
cc3768c4 | 12 | |
b2182a68 | 13 | #include "libGeoIP/GeoIP.h" |
cc3768c4 CP |
14 | #include "geoip.h" |
15 | ||
0a55e401 | 16 | MODULE_VERSION("$Id: geoip.c 663 2006-05-16 17:27:36Z newserv $") |
87698d77 | 17 | |
cc3768c4 CP |
18 | int geoip_totals[COUNTRY_MAX + 1]; |
19 | int geoip_nickext = -1; | |
20 | GeoIP *gi = NULL; | |
21 | ||
22 | void geoip_setupuser(nick *np); | |
23 | ||
24 | void geoip_new_nick(int hook, void *args); | |
25 | void geoip_quit(int hook, void *args); | |
26 | void geoip_whois_handler(int hooknum, void *arg); | |
27 | ||
28 | void _init(void) { | |
29 | int i; | |
30 | nick *np; | |
31 | sstring *filename; | |
32 | ||
33 | filename = getcopyconfigitem("geoip", "db", "GeoIP.dat", 256); | |
34 | gi = GeoIP_new(GEOIP_MEMORY_CACHE, filename->content); | |
35 | freesstring(filename); | |
36 | ||
37 | if(!gi) | |
38 | return; | |
39 | ||
40 | geoip_nickext = registernickext("geoip"); | |
41 | if(geoip_nickext == -1) | |
42 | return; /* PPA: registerchanext produces an error, however the module should stop loading */ | |
43 | ||
44 | memset(geoip_totals, 0, sizeof(geoip_totals)); | |
45 | ||
46 | for(i=0;i<NICKHASHSIZE;i++) | |
47 | for(np=nicktable[i];np;np=np->next) | |
48 | geoip_setupuser(np); | |
49 | ||
50 | registerhook(HOOK_NICK_LOSTNICK, &geoip_quit); | |
51 | registerhook(HOOK_NICK_NEWNICK, &geoip_new_nick); | |
52 | registerhook(HOOK_CONTROL_WHOISREQUEST, &geoip_whois_handler); | |
53 | } | |
54 | ||
55 | void _fini(void) { | |
56 | if(gi) | |
57 | GeoIP_delete(gi); | |
58 | ||
59 | if(geoip_nickext == -1) | |
60 | return; | |
61 | ||
62 | releasenickext(geoip_nickext); | |
63 | ||
64 | deregisterhook(HOOK_NICK_NEWNICK, &geoip_new_nick); | |
65 | deregisterhook(HOOK_NICK_LOSTNICK, &geoip_quit); | |
66 | deregisterhook(HOOK_CONTROL_WHOISREQUEST, &geoip_whois_handler); | |
67 | } | |
68 | ||
69 | void geoip_setupuser(nick *np) { | |
70 | int country = GeoIP_id_by_ipnum(gi, np->ipaddress); | |
71 | if((country < COUNTRY_MIN) || (country > COUNTRY_MAX)) | |
72 | return; | |
73 | ||
74 | geoip_totals[country]++; | |
75 | np->exts[geoip_nickext] = (void *)(long)country; | |
76 | } | |
77 | ||
78 | void geoip_new_nick(int hook, void *args) { | |
79 | geoip_setupuser((nick *)args); | |
80 | } | |
81 | ||
82 | void geoip_quit(int hook, void *args) { | |
83 | int item; | |
84 | nick *np = (nick *)args; | |
85 | ||
86 | item = (int)((long)np->exts[geoip_nickext]); | |
87 | ||
88 | if((item < COUNTRY_MIN) || (item > COUNTRY_MAX)) | |
89 | return; | |
90 | ||
91 | geoip_totals[item]--; | |
92 | } | |
93 | ||
94 | void geoip_whois_handler(int hooknum, void *arg) { | |
95 | int item; | |
96 | char message[512], *longcountry, *shortcountry, *shortcountry3; | |
97 | nick *np = (nick *)arg; | |
98 | ||
99 | if(!np) | |
100 | return; | |
101 | ||
102 | item = (int)((long)np->exts[geoip_nickext]); | |
103 | if((item < COUNTRY_MIN) || (item > COUNTRY_MAX)) | |
104 | return; | |
105 | ||
106 | if(GeoIP_country_info_by_id(item, &shortcountry, &shortcountry3, &longcountry)) { | |
107 | snprintf(message, sizeof(message), "Country : %s (%s)", shortcountry, longcountry); | |
108 | triggerhook(HOOK_CONTROL_WHOISREPLY, message); | |
109 | } | |
110 | } | |
111 |