all: updated FSF address in GPL text
[sip-router] / modules / geoip / geoip_mod.c
1 /**
2  * $Id$
3  *
4  * Copyright (C) 2010 Daniel-Constantin Mierla (asipto.com)
5  *
6  * This file is part of Kamailio, a free SIP server.
7  *
8  * This file is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation; either version 2 of the License, or
11  * (at your option) any later version
12  *
13  *
14  * This file is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
22  *
23  */
24
25 #include <stdio.h>
26 #include <unistd.h>
27 #include <stdlib.h>
28 #include <string.h>
29
30 #include "../../sr_module.h"
31 #include "../../dprint.h"
32 #include "../../ut.h"
33 #include "../../pvar.h"
34 #include "../../mod_fix.h"
35
36 #include "geoip_pv.h"
37
38 MODULE_VERSION
39
40 static char *geoip_path = NULL;
41
42 static int  mod_init(void);
43 static void mod_destroy(void);
44
45 static int w_geoip_match(struct sip_msg* msg, char* str1, char* str2);
46 static int geoip_match(struct sip_msg *msg, gparam_t *target, gparam_t *pvname);
47
48 static pv_export_t mod_pvs[] = {
49         { {"gip", sizeof("git")-1}, PVT_OTHER, pv_get_geoip, 0,
50                 pv_parse_geoip_name, 0, 0, 0 },
51         { {0, 0}, 0, 0, 0, 0, 0, 0, 0 }
52 };
53
54
55 static cmd_export_t cmds[]={
56         {"geoip_match", (cmd_function)w_geoip_match, 2, fixup_spve_spve,
57                 0, ANY_ROUTE},
58         {0, 0, 0, 0, 0, 0}
59 };
60
61 static param_export_t params[]={
62         {"path",     STR_PARAM, &geoip_path},
63         {0, 0, 0}
64 };
65
66 struct module_exports exports = {
67         "geoip",
68         DEFAULT_DLFLAGS, /* dlopen flags */
69         cmds,
70         params,
71         0,
72         0,              /* exported MI functions */
73         mod_pvs,        /* exported pseudo-variables */
74         0,              /* extra processes */
75         mod_init,       /* module initialization function */
76         0,              /* response function */
77         mod_destroy,    /* destroy function */
78         0               /* per child init function */
79 };
80
81
82
83 /**
84  * init module function
85  */
86 static int mod_init(void)
87 {
88
89         if(geoip_path==NULL || strlen(geoip_path)==0)
90         {
91                 LM_ERR("path to GeoIP database file not set\n");
92                 return -1;
93         }
94
95         if(geoip_init_pv(geoip_path)!=0)
96         {
97                 LM_ERR("cannot init for database file at: %s\n", geoip_path);
98                 return -1;
99         }
100         return 0;
101 }
102
103 /**
104  * destroy module function
105  */
106 static void mod_destroy(void)
107 {
108         geoip_destroy_pv();
109 }
110
111 static int w_geoip_match(struct sip_msg* msg, char* str1, char* str2)
112 {
113         return geoip_match(msg, (gparam_t*)str1, (gparam_t*)str2);
114 }
115
116 static int geoip_match(struct sip_msg *msg, gparam_t *target, gparam_t *pvname)
117 {
118         str tomatch;
119         str pvclass;
120         
121         if(msg==NULL)
122         {
123                 LM_ERR("received null msg\n");
124                 return -1;
125         }
126
127         if(fixup_get_svalue(msg, target, &tomatch)<0)
128         {
129                 LM_ERR("cannot get the address\n");
130                 return -1;
131         }
132         if(fixup_get_svalue(msg, pvname, &pvclass)<0)
133         {
134                 LM_ERR("cannot get the pv class\n");
135                 return -1;
136         }
137         geoip_pv_reset(&pvclass);
138
139         return geoip_update_pv(&tomatch, &pvclass);
140 }
141