/* 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 */ #ifdef HAVE_CONFIG_H #include "config.h" #endif #include "GeoIP.h" #include "GeoIP_internal.h" #include "GeoIPCity.h" #ifndef WIN32 #include #include /* For ntohl */ #else #include #include #endif #include /* For uint32_t */ #ifdef HAVE_STDINT_H #include /* 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 dmaarea_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 dma 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) dmaarea_combo += (record_buf[j] << (j * 8)); record->dma_code = dmaarea_combo/1000; record->area_code = dmaarea_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); } GeoIPRecord * GeoIP_record_by_ipnum (GeoIP* gi, unsigned long ipnum) { return _get_record(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_name (GeoIP* gi, const char *name) { unsigned long ipnum; if (name == NULL) { return 0; } ipnum = _GeoIP_lookupaddress(name); return _get_record(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_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) { char c, 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; }