You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
251 lines
6.4 KiB
251 lines
6.4 KiB
/* 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 <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 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;
|
|
}
|