fastgeoip 0.1.0

data/ext/GeoIP.h

@@ -0,0 +1,252 @@
+/* -*- Mode: C; indent-tabs-mode: t; c-basic-offset: 2; tab-width: 2 -*- */
+/* GeoIP.h
+ *
+ * Copyright (C) 2006 MaxMind LLC
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+ */
+
+#ifndef GEOIP_H
+#define GEOIP_H
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+#include <sys/types.h>
+
+#if !defined(_WIN32) 
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+#else /* !defined(_WIN32) */ 
+#include <winsock2.h> 
+#include <ws2tcpip.h> 
+#include <windows.h> 
+#define snprintf _snprintf 
+#define FILETIME_TO_USEC(ft) (((unsigned __int64) ft.dwHighDateTime << 32 | ft.dwLowDateTime) / 10) 
+#endif /* !defined(_WIN32) */ 
+ 
+#include<stdio.h>
+#include<stdlib.h>
+#include<string.h>
+#include <sys/types.h> /* for fstat */
+#include <sys/stat.h>	/* for fstat */
+
+#define SEGMENT_RECORD_LENGTH 3
+#define STANDARD_RECORD_LENGTH 3
+#define ORG_RECORD_LENGTH 4
+#define MAX_RECORD_LENGTH 4
+#define NUM_DB_TYPES 20
+
+/* 128 bit address in network order */
+typedef struct in6_addr geoipv6_t;
+ 
+#define GEOIP_CHKBIT_V6(bit,ptr) (ptr[((127UL - bit) >> 3)] & (1UL << (~(127 - bit) & 7)))
+
+typedef struct GeoIPTag {
+  FILE *GeoIPDatabase;
+  char *file_path;
+	unsigned char *cache;
+	unsigned char *index_cache;
+	unsigned int *databaseSegments;
+	char databaseType;
+	time_t mtime;
+	int flags;
+	off_t	size;
+	char record_length;
+	int charset; /* 0 iso-8859-1 1 utf8 */
+	int record_iter; /* used in GeoIP_next_record */
+	int netmask; /* netmask of last lookup - set using depth in _GeoIP_seek_record */
+	time_t last_mtime_check;
+} GeoIP;
+
+
+typedef enum {
+	GEOIP_CHARSET_ISO_8859_1 = 0,
+	GEOIP_CHARSET_UTF8       = 1
+} GeoIPCharset;
+
+typedef struct GeoIPRegionTag {
+	char country_code[3];
+	char region[3];
+} GeoIPRegion;
+
+typedef enum {
+	GEOIP_STANDARD = 0,
+	GEOIP_MEMORY_CACHE = 1,
+	GEOIP_CHECK_CACHE = 2,
+	GEOIP_INDEX_CACHE = 4,
+	GEOIP_MMAP_CACHE = 8,
+} GeoIPOptions;
+
+typedef enum {
+	GEOIP_COUNTRY_EDITION     = 1,
+	GEOIP_REGION_EDITION_REV0 = 7,
+	GEOIP_CITY_EDITION_REV0   = 6,
+	GEOIP_ORG_EDITION         = 5,
+	GEOIP_ISP_EDITION         = 4,
+	GEOIP_CITY_EDITION_REV1   = 2,
+	GEOIP_REGION_EDITION_REV1 = 3,
+	GEOIP_PROXY_EDITION       = 8,
+	GEOIP_ASNUM_EDITION       = 9,
+	GEOIP_NETSPEED_EDITION    = 10,
+	GEOIP_DOMAIN_EDITION      = 11,
+        GEOIP_COUNTRY_EDITION_V6  = 12,
+} GeoIPDBTypes;
+
+typedef enum {
+	GEOIP_ANON_PROXY = 1,
+	GEOIP_HTTP_X_FORWARDED_FOR_PROXY = 2,
+	GEOIP_HTTP_CLIENT_IP_PROXY = 3,
+} GeoIPProxyTypes;
+
+typedef enum {
+	GEOIP_UNKNOWN_SPEED = 0,
+	GEOIP_DIALUP_SPEED = 1,
+	GEOIP_CABLEDSL_SPEED = 2,
+	GEOIP_CORPORATE_SPEED = 3,
+} GeoIPNetspeedValues;
+
+extern char **GeoIPDBFileName;
+extern const char * GeoIPDBDescription[NUM_DB_TYPES];
+extern const char *GeoIPCountryDBFileName;
+extern const char *GeoIPRegionDBFileName;
+extern const char *GeoIPCityDBFileName;
+extern const char *GeoIPOrgDBFileName;
+extern const char *GeoIPISPDBFileName;
+
+/* Warning: do not use those arrays as doing so may break your
+ * program with newer GeoIP versions */
+extern const char GeoIP_country_code[253][3];
+extern const char GeoIP_country_code3[253][4];
+extern const char * GeoIP_country_name[253];
+extern const char GeoIP_country_continent[253][3];
+
+#ifdef DLL
+#define GEOIP_API __declspec(dllexport)
+#else
+#define GEOIP_API
+#endif  /* DLL */
+
+GEOIP_API void GeoIP_setup_custom_directory(char *dir);
+GEOIP_API GeoIP* GeoIP_open_type (int type, int flags);
+GEOIP_API GeoIP* GeoIP_new(int flags);
+GEOIP_API GeoIP* GeoIP_open(const char * filename, int flags);
+GEOIP_API int GeoIP_db_avail(int type);
+GEOIP_API void GeoIP_delete(GeoIP* gi);
+GEOIP_API const char *GeoIP_country_code_by_addr (GeoIP* gi, const char *addr);
+GEOIP_API const char *GeoIP_country_code_by_name (GeoIP* gi, const char *host);
+GEOIP_API const char *GeoIP_country_code3_by_addr (GeoIP* gi, const char *addr);
+GEOIP_API const char *GeoIP_country_code3_by_name (GeoIP* gi, const char *host);
+GEOIP_API const char *GeoIP_country_name_by_addr (GeoIP* gi, const char *addr);
+GEOIP_API const char *GeoIP_country_name_by_name (GeoIP* gi, const char *host);
+GEOIP_API const char *GeoIP_country_name_by_ipnum (GeoIP* gi, unsigned long ipnum);
+GEOIP_API const char *GeoIP_country_code_by_ipnum (GeoIP* gi, unsigned long ipnum);
+GEOIP_API const char *GeoIP_country_code3_by_ipnum (GeoIP* gi, unsigned long ipnum);
+
+/* */
+GEOIP_API const char *GeoIP_country_name_by_ipnum_v6 (GeoIP* gi, geoipv6_t ipnum);
+GEOIP_API const char *GeoIP_country_code_by_ipnum_v6 (GeoIP* gi, geoipv6_t ipnum);
+GEOIP_API const char *GeoIP_country_code3_by_ipnum_v6 (GeoIP* gi, geoipv6_t ipnum);
+
+/* Deprecated - for backwards compatibility only */
+GEOIP_API int GeoIP_country_id_by_addr (GeoIP* gi, const char *addr);
+GEOIP_API int GeoIP_country_id_by_name (GeoIP* gi, const char *host);
+GEOIP_API char *GeoIP_org_by_addr (GeoIP* gi, const char *addr);
+GEOIP_API char *GeoIP_org_by_name (GeoIP* gi, const char *host);
+GEOIP_API char *GeoIP_org_by_ipnum (GeoIP* gi, unsigned long ipnum);
+
+GEOIP_API char *GeoIP_org_by_ipnum_v6 (GeoIP* gi, geoipv6_t ipnum);
+GEOIP_API char *GeoIP_org_by_addr_v6 (GeoIP* gi, const char *addr);
+GEOIP_API char *GeoIP_org_by_name_v6 (GeoIP* gi, const char *name);
+
+/* End deprecated */
+
+GEOIP_API int GeoIP_id_by_addr (GeoIP* gi, const char *addr);
+GEOIP_API int GeoIP_id_by_name (GeoIP* gi, const char *host);
+GEOIP_API int GeoIP_id_by_ipnum (GeoIP* gi, unsigned long ipnum);
+
+GEOIP_API int GeoIP_id_by_addr_v6 (GeoIP* gi, const char *addr);
+GEOIP_API int GeoIP_id_by_name_v6 (GeoIP* gi, const char *host);
+GEOIP_API int GeoIP_id_by_ipnum_v6 (GeoIP* gi, geoipv6_t ipnum);
+
+GEOIP_API GeoIPRegion * GeoIP_region_by_addr (GeoIP* gi, const char *addr);
+GEOIP_API GeoIPRegion * GeoIP_region_by_name (GeoIP* gi, const char *host);
+GEOIP_API GeoIPRegion * GeoIP_region_by_ipnum (GeoIP *gi, unsigned long ipnum);
+
+GEOIP_API GeoIPRegion * GeoIP_region_by_addr_v6 (GeoIP* gi, const char *addr);
+GEOIP_API GeoIPRegion * GeoIP_region_by_name_v6 (GeoIP* gi, const char *host);
+GEOIP_API GeoIPRegion * GeoIP_region_by_ipnum_v6 (GeoIP *gi, geoipv6_t ipnum);
+
+/* Warning - don't call this after GeoIP_assign_region_by_inetaddr calls */
+GEOIP_API void GeoIPRegion_delete (GeoIPRegion *gir);
+
+GEOIP_API void GeoIP_assign_region_by_inetaddr(GeoIP* gi, unsigned long inetaddr, GeoIPRegion *gir);
+
+GEOIP_API void GeoIP_assign_region_by_inetaddr_v6(GeoIP* gi, geoipv6_t inetaddr, GeoIPRegion *gir);
+
+/* Used to query GeoIP Organization, ISP and AS Number databases */
+GEOIP_API char *GeoIP_name_by_ipnum (GeoIP* gi, unsigned long ipnum);
+GEOIP_API char *GeoIP_name_by_addr (GeoIP* gi, const char *addr);
+GEOIP_API char *GeoIP_name_by_name (GeoIP* gi, const char *host);
+
+GEOIP_API char *GeoIP_name_by_ipnum_v6 (GeoIP* gi, geoipv6_t ipnum);
+GEOIP_API char *GeoIP_name_by_addr_v6 (GeoIP* gi, const char *addr);
+GEOIP_API char *GeoIP_name_by_name_v6 (GeoIP* gi, const char *name);
+
+/** return two letter country code */
+GEOIP_API const char* GeoIP_code_by_id(int id);
+
+/** return three letter country code */
+GEOIP_API const char* GeoIP_code3_by_id(int id);
+
+/** return full name of country */
+GEOIP_API const char* GeoIP_name_by_id(int id);
+
+/** return continent of country */
+GEOIP_API const char* GeoIP_continent_by_id(int id);
+
+/** return id by country code **/
+GEOIP_API int GeoIP_id_by_code(const char *country);
+
+/** return return number of known countries */
+GEOIP_API unsigned GeoIP_num_countries(void);
+
+GEOIP_API char *GeoIP_database_info (GeoIP* gi);
+GEOIP_API unsigned char GeoIP_database_edition (GeoIP* gi);
+
+GEOIP_API int GeoIP_charset (GeoIP* gi);
+GEOIP_API int GeoIP_set_charset (GeoIP* gi, int charset);
+
+GEOIP_API int GeoIP_last_netmask (GeoIP* gi);
+GEOIP_API char **GeoIP_range_by_ip (GeoIP* gi, const char *addr);
+GEOIP_API void GeoIP_range_by_ip_delete(char **ptr);
+
+/* Convert region code to region name */
+GEOIP_API const char * GeoIP_region_name_by_code(const char *country_code, const char *region_code);
+
+/* Get timezone from country and region code */
+GEOIP_API const char * GeoIP_time_zone_by_country_and_region(const char *country_code, const char *region_code);
+
+#ifdef BSD
+#define memcpy(dest, src, n) bcopy(src, dest, n)
+#endif
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif /* GEOIP_H */

data/ext/GeoIPCity.c

@@ -0,0 +1,297 @@
+/* -*- Mode: C; indent-tabs-mode: t; c-basic-offset: 2; tab-width: 2 -*- */
+/* GeoIPCity.c
+ *
+ * Copyright (C) 2006 MaxMind LLC
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+ */
+
+#include <GeoIP.h>
+#include <GeoIP_internal.h>
+#include <GeoIPCity.h>
+#if !defined(_WIN32)
+#include <netdb.h>
+#include <netinet/in.h> /* For ntohl */
+#else
+#include <windows.h>
+#include <winsock.h>
+#endif
+#include <sys/types.h> /* For uint32_t */
+#ifdef HAVE_STDINT_H
+#include <stdint.h>     /* For uint32_t */
+#endif
+
+static
+const int FULL_RECORD_LENGTH = 50;
+
+static
+GeoIPRecord * _extract_record(GeoIP* gi, unsigned int seek_record, int *next_record_ptr) {
+	int record_pointer;
+	unsigned char *record_buf = NULL;
+	unsigned char *begin_record_buf = NULL;
+	GeoIPRecord * record;
+	int str_length = 0;
+	int j;
+	double latitude = 0, longitude = 0;
+	int metroarea_combo = 0;
+	int bytes_read = 0;
+	if (seek_record == gi->databaseSegments[0])		
+		return NULL;
+
+	record = malloc(sizeof(GeoIPRecord));
+	memset(record, 0, sizeof(GeoIPRecord));
+
+	record->charset = gi->charset;
+
+	record_pointer = seek_record + (2 * gi->record_length - 1) * gi->databaseSegments[0];
+
+	if (gi->cache == NULL) {
+		fseek(gi->GeoIPDatabase, record_pointer, SEEK_SET);
+		begin_record_buf = record_buf = malloc(sizeof(char) * FULL_RECORD_LENGTH);
+		bytes_read = fread(record_buf, sizeof(char), FULL_RECORD_LENGTH, gi->GeoIPDatabase);
+		if (bytes_read == 0) {
+			/* eof or other error */
+			free(begin_record_buf);
+			free(record);
+			return NULL;
+		}
+	} else {
+		record_buf = gi->cache + (long)record_pointer;
+	}
+
+	/* get country */
+	record->continent_code = (char *) GeoIP_country_continent[record_buf[0]];
+	record->country_code	= (char *) GeoIP_country_code [record_buf[0]];
+	record->country_code3 = (char *) GeoIP_country_code3[record_buf[0]];
+	record->country_name	= (char *) GeoIP_country_name [record_buf[0]];
+	record_buf++;
+
+	/* get region */
+	while (record_buf[str_length] != '\0')
+		str_length++;
+	if (str_length > 0) {
+		record->region = malloc(str_length+1);
+		strncpy(record->region, (char *)record_buf, str_length+1);
+	}
+	record_buf += str_length + 1;
+	str_length = 0;
+
+	/* get city */
+	while (record_buf[str_length] != '\0')
+		str_length++;
+	if (str_length > 0) {
+		if ( gi->charset == GEOIP_CHARSET_UTF8 ) {
+			record->city = _iso_8859_1__utf8( (const char * ) record_buf );
+		} else {
+			record->city = malloc(str_length+1);
+			strncpy(record->city, ( const char * ) record_buf, str_length+1);
+		}
+	}
+	record_buf += (str_length + 1);
+	str_length = 0;
+
+	/* get postal code */
+	while (record_buf[str_length] != '\0')
+		str_length++;
+	if (str_length > 0) {
+		record->postal_code = malloc(str_length+1);
+		strncpy(record->postal_code, (char *)record_buf, str_length+1);
+	}
+	record_buf += (str_length + 1);
+
+	/* get latitude */
+for (j = 0; j < 3; ++j)
+		latitude += (record_buf[j] << (j * 8));
+	record->latitude = latitude/10000 - 180;
+	record_buf += 3;
+
+	/* get longitude */
+	for (j = 0; j < 3; ++j)
+		longitude += (record_buf[j] << (j * 8));
+	record->longitude = longitude/10000 - 180;
+
+	/* get area code and metro code for post April 2002 databases and for US locations */
+	if (GEOIP_CITY_EDITION_REV1 == gi->databaseType) {
+		if (!strcmp(record->country_code, "US")) {
+			record_buf += 3;
+			for (j = 0; j < 3; ++j)
+				metroarea_combo += (record_buf[j] << (j * 8));
+			record->metro_code = metroarea_combo/1000;
+			record->area_code = metroarea_combo % 1000;
+		}
+	}
+
+	if (gi->cache == NULL)
+		free(begin_record_buf);
+
+	/* Used for GeoIP_next_record */
+	if (next_record_ptr != NULL)
+		*next_record_ptr = seek_record + record_buf - begin_record_buf + 3;
+
+	return record;
+}
+
+static
+GeoIPRecord * _get_record(GeoIP* gi, unsigned long ipnum) {
+	unsigned int seek_record;
+
+	if (gi->databaseType != GEOIP_CITY_EDITION_REV0 &&
+			gi->databaseType != GEOIP_CITY_EDITION_REV1) {
+		printf("Invalid database type %s, expected %s\n", GeoIPDBDescription[(int)gi->databaseType], GeoIPDBDescription[GEOIP_CITY_EDITION_REV1]);
+		return 0;
+	}
+
+	seek_record = _GeoIP_seek_record(gi, ipnum);
+	return _extract_record(gi, seek_record, NULL);
+}
+
+static
+GeoIPRecord * _get_record_v6(GeoIP* gi, geoipv6_t ipnum) {
+       unsigned int seek_record;
+
+       if (gi->databaseType != GEOIP_CITY_EDITION_REV0 &&
+                       gi->databaseType != GEOIP_CITY_EDITION_REV1) {
+               printf("Invalid database type %s, expected %s\n", GeoIPDBDescription[(int)gi->databaseType], GeoIPDBDescription[GEOIP_CITY_EDITION_REV1]);
+               return 0;
+       }
+
+       seek_record = _GeoIP_seek_record_v6(gi, ipnum);
+       return _extract_record(gi, seek_record, NULL);
+}
+
+
+
+GeoIPRecord * GeoIP_record_by_ipnum (GeoIP* gi, unsigned long ipnum) {
+	return _get_record(gi, ipnum);
+}
+
+GeoIPRecord * GeoIP_record_by_ipnum_v6 (GeoIP* gi, geoipv6_t ipnum) {
+       return _get_record_v6(gi, ipnum);
+}
+
+GeoIPRecord * GeoIP_record_by_addr (GeoIP* gi, const char *addr) {
+	unsigned long ipnum;
+	if (addr == NULL) {
+		return 0;
+	}
+	ipnum = _GeoIP_addr_to_num(addr);
+	return _get_record(gi, ipnum);
+}
+
+GeoIPRecord * GeoIP_record_by_addr_v6 (GeoIP* gi, const char *addr) {
+       geoipv6_t ipnum;
+       if (addr == NULL) {
+               return 0;
+       }
+       ipnum = _GeoIP_addr_to_num_v6(addr);
+       return _get_record_v6(gi, ipnum);
+}
+
+GeoIPRecord * GeoIP_record_by_name (GeoIP* gi, const char *name) {
+	unsigned long ipnum;
+	if (name == NULL) {
+		return 0;
+	}
+	ipnum = _GeoIP_lookupaddress(name);
+	return _get_record(gi, ipnum);
+}
+
+GeoIPRecord * GeoIP_record_by_name_v6 (GeoIP* gi, const char *name) {
+        geoipv6_t ipnum;
+       if (name == NULL) {
+               return 0;
+       }
+       ipnum = _GeoIP_lookupaddress_v6(name);
+       return _get_record_v6(gi, ipnum);
+}
+
+int GeoIP_record_id_by_addr (GeoIP* gi, const char *addr) {
+	unsigned long ipnum;
+	if (gi->databaseType != GEOIP_CITY_EDITION_REV0 &&
+			gi->databaseType != GEOIP_CITY_EDITION_REV1) {
+		printf("Invalid database type %s, expected %s\n", GeoIPDBDescription[(int)gi->databaseType], GeoIPDBDescription[GEOIP_CITY_EDITION_REV1]);
+		return 0;
+	}
+	if (addr == NULL) {
+		return 0;
+	}
+	ipnum = _GeoIP_addr_to_num(addr);
+	return _GeoIP_seek_record(gi, ipnum);
+}
+
+int GeoIP_record_id_by_addr_v6 (GeoIP* gi, const char *addr) {
+       geoipv6_t ipnum;
+       if (gi->databaseType != GEOIP_CITY_EDITION_REV0 &&
+                       gi->databaseType != GEOIP_CITY_EDITION_REV1) {
+               printf("Invalid database type %s, expected %s\n", GeoIPDBDescription[(int)gi->databaseType], GeoIPDBDescription[GEOIP_CITY_EDITION_REV1]);
+               return 0;
+       }
+       if (addr == NULL) {
+               return 0;
+       }
+       ipnum = _GeoIP_addr_to_num_v6(addr);
+       return _GeoIP_seek_record_v6(gi, ipnum);
+}
+
+int GeoIP_init_record_iter (GeoIP* gi) {
+	return gi->databaseSegments[0] + 1;
+}
+
+int GeoIP_next_record (GeoIP* gi, GeoIPRecord **gir, int *record_iter) {
+	if (gi->cache != NULL) {
+		printf("GeoIP_next_record not supported in memory cache mode\n");
+		return 1;
+	}
+	*gir = _extract_record(gi, *record_iter, record_iter);
+	return 0;
+}
+
+void GeoIPRecord_delete (GeoIPRecord *gir) {
+	free(gir->region);
+	free(gir->city);
+	free(gir->postal_code);
+	free(gir);
+}
+
+
+
+char * _iso_8859_1__utf8(const char * iso) {
+	signed char c;
+	char k;
+	char * p;
+	char * t = (char *)iso;
+	int len = 0;
+	while ( ( c = *t++) ){
+		if ( c < 0 )
+			len++; 
+	}
+	len += t - iso;
+	t = p = malloc( len );
+	
+	if ( p ){
+		while ( ( c = *iso++ ) ) {
+			if (c < 0 ) {
+				k = 0xc2;
+				if (c >= -64 )
+					k++;
+				*t++ = k;
+				c &= ~0x40;
+			}
+			*t++ = c;
+		}
+		*t++ = 0x00;
+	}
+	return p;
+}