@@ -32,22 +32,25 @@ ip2proxy_free(void *ptr)
3232VCL_VOID
3333vmod_init_db (VRT_CTX , struct vmod_priv * priv , char * filename , char * memtype )
3434{
35+ IP2Proxy * IP2ProxyObj ;
3536
3637 CHECK_OBJ_NOTNULL (ctx , VRT_CTX_MAGIC );
37- printf ("The filename accepted is %s.\n" , (char * ) filename );
38- if (priv -> priv != NULL ) {
39- IP2Proxy_close (priv -> priv );
40- }
41- IP2Proxy * IP2ProxyObj = IP2Proxy_open ( (char * ) filename );
42- if (strcmp (memtype , "IP2PROXY_FILE_IO" ) == 0 ) {
43- IP2Proxy_open_mem (priv -> priv , IP2PROXY_FILE_IO );
44- } else if (strcmp (memtype , "IP2PROXY_SHARED_MEMORY" ) == 0 ) {
45- IP2Proxy_open_mem (priv -> priv , IP2PROXY_SHARED_MEMORY );
46- } else if (strcmp (memtype , "IP2PROXY_CACHE_MEMORY" ) == 0 ) {
47- IP2Proxy_open_mem (priv -> priv , IP2PROXY_CACHE_MEMORY );
48- }
38+ AN (priv );
39+
40+ if (priv -> priv != NULL )
41+ IP2Proxy_close ((IP2Proxy * )priv -> priv );
42+
43+ IP2ProxyObj = IP2Proxy_open (filename );
44+ AN (IP2ProxyObj );
45+
46+ if (strcmp (memtype , "IP2PROXY_FILE_IO" ) == 0 )
47+ IP2Proxy_open_mem (IP2ProxyObj , IP2PROXY_FILE_IO );
48+ else if (strcmp (memtype , "IP2PROXY_SHARED_MEMORY" ) == 0 )
49+ IP2Proxy_open_mem (IP2ProxyObj , IP2PROXY_SHARED_MEMORY );
50+ else if (strcmp (memtype , "IP2PROXY_CACHE_MEMORY" ) == 0 )
51+ IP2Proxy_open_mem (IP2ProxyObj , IP2PROXY_CACHE_MEMORY );
52+
4953 priv -> priv = IP2ProxyObj ;
50- AN (priv -> priv );
5154 priv -> free = ip2proxy_free ;
5255}
5356
@@ -57,15 +60,20 @@ vmod_ ## lower(VRT_CTX, struct vmod_priv *priv, char * ip) \
5760{ \
5861 char *result; \
5962 IP2ProxyRecord *r; \
63+ \
6064 CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC); \
6165 AN(priv); \
66+ \
6267 if (!ip || !priv->priv) \
6368 return ("-"); \
69+ \
6470 r = IP2Proxy_get_all((IP2Proxy *)priv->priv, ip); \
6571 if (!r) \
6672 return ("-"); \
73+ \
6774 result = WS_Copy(ctx->ws, r->field, -1); \
6875 IP2Proxy_free_record(r); \
76+ \
6977 return (result); \
7078}
7179
0 commit comments