after some exam days its nice to code again
--HG-- extra : convert_revision : svn%3A39bc706e-5318-0410-9160-8a85361fbb7c/trunk%40321
This commit is contained in:
parent
c5d0848177
commit
2ea877f14b
969
extensions/geoip/GeoIP.c
Normal file
969
extensions/geoip/GeoIP.c
Normal file
@ -0,0 +1,969 @@
|
|||||||
|
/* -*- Mode: C; indent-tabs-mode: t; c-basic-offset: 2; tab-width: 2 -*- */
|
||||||
|
/* GeoIP.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"
|
||||||
|
|
||||||
|
#ifndef _WIN32
|
||||||
|
#include <netdb.h>
|
||||||
|
#include <sys/socket.h>
|
||||||
|
#include <netinet/in.h> /* For ntohl */
|
||||||
|
#include <arpa/inet.h>
|
||||||
|
#else
|
||||||
|
#include <windows.h>
|
||||||
|
#define snprintf _snprintf
|
||||||
|
#endif
|
||||||
|
#include <errno.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <assert.h>
|
||||||
|
#include <sys/types.h> /* for fstat */
|
||||||
|
#include <sys/stat.h> /* for fstat */
|
||||||
|
|
||||||
|
#ifdef HAVE_STDINT_H
|
||||||
|
#include <stdint.h> /* For uint32_t */
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#ifndef INADDR_NONE
|
||||||
|
#define INADDR_NONE -1
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#define COUNTRY_BEGIN 16776960
|
||||||
|
#define STATE_BEGIN_REV0 16700000
|
||||||
|
#define STATE_BEGIN_REV1 16000000
|
||||||
|
#define STRUCTURE_INFO_MAX_SIZE 20
|
||||||
|
#define DATABASE_INFO_MAX_SIZE 100
|
||||||
|
#define MAX_ORG_RECORD_LENGTH 300
|
||||||
|
#define US_OFFSET 1
|
||||||
|
#define CANADA_OFFSET 677
|
||||||
|
#define WORLD_OFFSET 1353
|
||||||
|
#define FIPS_RANGE 360
|
||||||
|
|
||||||
|
#define CHECK_ERR(err, msg) { \
|
||||||
|
if (err != Z_OK) { \
|
||||||
|
fprintf(stderr, "%s error: %d\n", msg, err); \
|
||||||
|
exit(1); \
|
||||||
|
} \
|
||||||
|
}
|
||||||
|
|
||||||
|
const char GeoIP_country_code[251][3] = { "--","AP","EU","AD","AE","AF","AG","AI","AL","AM","AN",
|
||||||
|
"AO","AQ","AR","AS","AT","AU","AW","AZ","BA","BB",
|
||||||
|
"BD","BE","BF","BG","BH","BI","BJ","BM","BN","BO",
|
||||||
|
"BR","BS","BT","BV","BW","BY","BZ","CA","CC","CD",
|
||||||
|
"CF","CG","CH","CI","CK","CL","CM","CN","CO","CR",
|
||||||
|
"CU","CV","CX","CY","CZ","DE","DJ","DK","DM","DO",
|
||||||
|
"DZ","EC","EE","EG","EH","ER","ES","ET","FI","FJ",
|
||||||
|
"FK","FM","FO","FR","FX","GA","GB","GD","GE","GF",
|
||||||
|
"GH","GI","GL","GM","GN","GP","GQ","GR","GS","GT",
|
||||||
|
"GU","GW","GY","HK","HM","HN","HR","HT","HU","ID",
|
||||||
|
"IE","IL","IN","IO","IQ","IR","IS","IT","JM","JO",
|
||||||
|
"JP","KE","KG","KH","KI","KM","KN","KP","KR","KW",
|
||||||
|
"KY","KZ","LA","LB","LC","LI","LK","LR","LS","LT",
|
||||||
|
"LU","LV","LY","MA","MC","MD","MG","MH","MK","ML",
|
||||||
|
"MM","MN","MO","MP","MQ","MR","MS","MT","MU","MV",
|
||||||
|
"MW","MX","MY","MZ","NA","NC","NE","NF","NG","NI",
|
||||||
|
"NL","NO","NP","NR","NU","NZ","OM","PA","PE","PF",
|
||||||
|
"PG","PH","PK","PL","PM","PN","PR","PS","PT","PW",
|
||||||
|
"PY","QA","RE","RO","RU","RW","SA","SB","SC","SD",
|
||||||
|
"SE","SG","SH","SI","SJ","SK","SL","SM","SN","SO",
|
||||||
|
"SR","ST","SV","SY","SZ","TC","TD","TF","TG","TH",
|
||||||
|
"TJ","TK","TM","TN","TO","TL","TR","TT","TV","TW",
|
||||||
|
"TZ","UA","UG","UM","US","UY","UZ","VA","VC","VE",
|
||||||
|
"VG","VI","VN","VU","WF","WS","YE","YT","RS","ZA",
|
||||||
|
"ZM","ME","ZW","A1","A2","O1","AX","GG","IM","JE"};
|
||||||
|
|
||||||
|
const char GeoIP_country_code3[251][4] = { "--","AP","EU","AND","ARE","AFG","ATG","AIA","ALB","ARM","ANT",
|
||||||
|
"AGO","AQ","ARG","ASM","AUT","AUS","ABW","AZE","BIH","BRB",
|
||||||
|
"BGD","BEL","BFA","BGR","BHR","BDI","BEN","BMU","BRN","BOL",
|
||||||
|
"BRA","BHS","BTN","BV","BWA","BLR","BLZ","CAN","CC","COD",
|
||||||
|
"CAF","COG","CHE","CIV","COK","CHL","CMR","CHN","COL","CRI",
|
||||||
|
"CUB","CPV","CX","CYP","CZE","DEU","DJI","DNK","DMA","DOM",
|
||||||
|
"DZA","ECU","EST","EGY","ESH","ERI","ESP","ETH","FIN","FJI",
|
||||||
|
"FLK","FSM","FRO","FRA","FX","GAB","GBR","GRD","GEO","GUF",
|
||||||
|
"GHA","GIB","GRL","GMB","GIN","GLP","GNQ","GRC","GS","GTM",
|
||||||
|
"GUM","GNB","GUY","HKG","HM","HND","HRV","HTI","HUN","IDN",
|
||||||
|
"IRL","ISR","IND","IO","IRQ","IRN","ISL","ITA","JAM","JOR",
|
||||||
|
"JPN","KEN","KGZ","KHM","KIR","COM","KNA","PRK","KOR","KWT",
|
||||||
|
"CYM","KAZ","LAO","LBN","LCA","LIE","LKA","LBR","LSO","LTU",
|
||||||
|
"LUX","LVA","LBY","MAR","MCO","MDA","MDG","MHL","MKD","MLI",
|
||||||
|
"MMR","MNG","MAC","MNP","MTQ","MRT","MSR","MLT","MUS","MDV",
|
||||||
|
"MWI","MEX","MYS","MOZ","NAM","NCL","NER","NFK","NGA","NIC",
|
||||||
|
"NLD","NOR","NPL","NRU","NIU","NZL","OMN","PAN","PER","PYF",
|
||||||
|
"PNG","PHL","PAK","POL","SPM","PCN","PRI","PSE","PRT","PLW",
|
||||||
|
"PRY","QAT","REU","ROU","RUS","RWA","SAU","SLB","SYC","SDN",
|
||||||
|
"SWE","SGP","SHN","SVN","SJM","SVK","SLE","SMR","SEN","SOM",
|
||||||
|
"SUR","STP","SLV","SYR","SWZ","TCA","TCD","TF","TGO","THA",
|
||||||
|
"TJK","TKL","TKM","TUN","TON","TLS","TUR","TTO","TUV","TWN",
|
||||||
|
"TZA","UKR","UGA","UM","USA","URY","UZB","VAT","VCT","VEN",
|
||||||
|
"VGB","VIR","VNM","VUT","WLF","WSM","YEM","YT","SRB","ZAF",
|
||||||
|
"ZMB","MNE","ZWE","A1","A2","O1","ALA","GGY","IMN","JEY"};
|
||||||
|
|
||||||
|
const char * GeoIP_country_name[251] = {"N/A","Asia/Pacific Region","Europe","Andorra","United Arab Emirates","Afghanistan","Antigua and Barbuda","Anguilla","Albania","Armenia","Netherlands Antilles",
|
||||||
|
"Angola","Antarctica","Argentina","American Samoa","Austria","Australia","Aruba","Azerbaijan","Bosnia and Herzegovina","Barbados",
|
||||||
|
"Bangladesh","Belgium","Burkina Faso","Bulgaria","Bahrain","Burundi","Benin","Bermuda","Brunei Darussalam","Bolivia",
|
||||||
|
"Brazil","Bahamas","Bhutan","Bouvet Island","Botswana","Belarus","Belize","Canada","Cocos (Keeling) Islands","Congo, The Democratic Republic of the",
|
||||||
|
"Central African Republic","Congo","Switzerland","Cote D'Ivoire","Cook Islands","Chile","Cameroon","China","Colombia","Costa Rica",
|
||||||
|
"Cuba","Cape Verde","Christmas Island","Cyprus","Czech Republic","Germany","Djibouti","Denmark","Dominica","Dominican Republic",
|
||||||
|
"Algeria","Ecuador","Estonia","Egypt","Western Sahara","Eritrea","Spain","Ethiopia","Finland","Fiji",
|
||||||
|
"Falkland Islands (Malvinas)","Micronesia, Federated States of","Faroe Islands","France","France, Metropolitan","Gabon","United Kingdom","Grenada","Georgia","French Guiana",
|
||||||
|
"Ghana","Gibraltar","Greenland","Gambia","Guinea","Guadeloupe","Equatorial Guinea","Greece","South Georgia and the South Sandwich Islands","Guatemala",
|
||||||
|
"Guam","Guinea-Bissau","Guyana","Hong Kong","Heard Island and McDonald Islands","Honduras","Croatia","Haiti","Hungary","Indonesia",
|
||||||
|
"Ireland","Israel","India","British Indian Ocean Territory","Iraq","Iran, Islamic Republic of","Iceland","Italy","Jamaica","Jordan",
|
||||||
|
"Japan","Kenya","Kyrgyzstan","Cambodia","Kiribati","Comoros","Saint Kitts and Nevis","Korea, Democratic People's Republic of","Korea, Republic of","Kuwait",
|
||||||
|
"Cayman Islands","Kazakhstan","Lao People's Democratic Republic","Lebanon","Saint Lucia","Liechtenstein","Sri Lanka","Liberia","Lesotho","Lithuania",
|
||||||
|
"Luxembourg","Latvia","Libyan Arab Jamahiriya","Morocco","Monaco","Moldova, Republic of","Madagascar","Marshall Islands","Macedonia","Mali",
|
||||||
|
"Myanmar","Mongolia","Macau","Northern Mariana Islands","Martinique","Mauritania","Montserrat","Malta","Mauritius","Maldives",
|
||||||
|
"Malawi","Mexico","Malaysia","Mozambique","Namibia","New Caledonia","Niger","Norfolk Island","Nigeria","Nicaragua",
|
||||||
|
"Netherlands","Norway","Nepal","Nauru","Niue","New Zealand","Oman","Panama","Peru","French Polynesia",
|
||||||
|
"Papua New Guinea","Philippines","Pakistan","Poland","Saint Pierre and Miquelon","Pitcairn Islands","Puerto Rico","Palestinian Territory","Portugal","Palau",
|
||||||
|
"Paraguay","Qatar","Reunion","Romania","Russian Federation","Rwanda","Saudi Arabia","Solomon Islands","Seychelles","Sudan",
|
||||||
|
"Sweden","Singapore","Saint Helena","Slovenia","Svalbard and Jan Mayen","Slovakia","Sierra Leone","San Marino","Senegal","Somalia","Suriname",
|
||||||
|
"Sao Tome and Principe","El Salvador","Syrian Arab Republic","Swaziland","Turks and Caicos Islands","Chad","French Southern Territories","Togo","Thailand",
|
||||||
|
"Tajikistan","Tokelau","Turkmenistan","Tunisia","Tonga","Timor-Leste","Turkey","Trinidad and Tobago","Tuvalu","Taiwan",
|
||||||
|
"Tanzania, United Republic of","Ukraine","Uganda","United States Minor Outlying Islands","United States","Uruguay","Uzbekistan","Holy See (Vatican City State)","Saint Vincent and the Grenadines","Venezuela",
|
||||||
|
"Virgin Islands, British","Virgin Islands, U.S.","Vietnam","Vanuatu","Wallis and Futuna","Samoa","Yemen","Mayotte","Serbia","South Africa",
|
||||||
|
"Zambia","Montenegro","Zimbabwe","Anonymous Proxy","Satellite Provider","Other","Aland Islands","Guernsey","Isle of Man","Jersey"};
|
||||||
|
|
||||||
|
const char GeoIP_country_continent[251][3] = {"--","AS","EU","EU","AS","AS","SA","SA","EU","AS","SA",
|
||||||
|
"AF","AN","SA","OC","EU","OC","SA","AS","EU","SA",
|
||||||
|
"AS","EU","AF","EU","AS","AF","AF","SA","AS","SA",
|
||||||
|
"SA","SA","AS","AF","AF","EU","SA","NA","AS","AF",
|
||||||
|
"AF","AF","EU","AF","OC","SA","AF","AS","SA","SA",
|
||||||
|
"SA","AF","AS","AS","EU","EU","AF","EU","SA","SA",
|
||||||
|
"AF","SA","EU","AF","AF","AF","EU","AF","EU","OC",
|
||||||
|
"SA","OC","EU","EU","EU","AF","EU","SA","AS","SA",
|
||||||
|
"AF","EU","SA","AF","AF","SA","AF","EU","SA","SA",
|
||||||
|
"OC","AF","SA","AS","AF","SA","EU","SA","EU","AS",
|
||||||
|
"EU","AS","AS","AS","AS","AS","EU","EU","SA","AS",
|
||||||
|
"AS","AF","AS","AS","OC","AF","SA","AS","AS","AS",
|
||||||
|
"SA","AS","AS","AS","SA","EU","AS","AF","AF","EU",
|
||||||
|
"EU","EU","AF","AF","EU","EU","AF","OC","EU","AF",
|
||||||
|
"AS","AS","AS","OC","SA","AF","SA","EU","AF","AS",
|
||||||
|
"AF","NA","AS","AF","AF","OC","AF","OC","AF","SA",
|
||||||
|
"EU","EU","AS","OC","OC","OC","AS","SA","SA","OC",
|
||||||
|
"OC","AS","AS","EU","SA","OC","SA","AS","EU","OC",
|
||||||
|
"SA","AS","AF","EU","AS","AF","AS","OC","AF","AF",
|
||||||
|
"EU","AS","AF","EU","EU","EU","AF","EU","AF","AF",
|
||||||
|
"SA","AF","SA","AS","AF","SA","AF","AF","AF","AS",
|
||||||
|
"AS","OC","AS","AF","OC","AS","AS","SA","OC","AS",
|
||||||
|
"AF","EU","AF","OC","NA","SA","AS","EU","SA","SA",
|
||||||
|
"SA","SA","AS","OC","OC","OC","AS","AF","EU","AF",
|
||||||
|
"AF","EU","AF","--","--","--","EU","EU","EU","EU"};
|
||||||
|
|
||||||
|
const char * GeoIPDBDescription[NUM_DB_TYPES] = {NULL, "GeoIP Country Edition", "GeoIP City Edition, Rev 1", "GeoIP Region Edition, Rev 1", "GeoIP ISP Edition", "GeoIP Organization Edition", "GeoIP City Edition, Rev 0", "GeoIP Region Edition, Rev 0","GeoIP Proxy Edition","GeoIP ASNum Edition","GeoIP Netspeed Edition","GeoIP Domain Name Edition"};
|
||||||
|
|
||||||
|
char * custom_directory = NULL;
|
||||||
|
|
||||||
|
void GeoIP_setup_custom_directory (char * dir) {
|
||||||
|
custom_directory = dir;
|
||||||
|
}
|
||||||
|
|
||||||
|
char *_GeoIP_full_path_to(const char *file_name) {
|
||||||
|
char *path = malloc(sizeof(char) * 1024);
|
||||||
|
|
||||||
|
if (custom_directory == NULL){
|
||||||
|
#ifndef _WIN32
|
||||||
|
memset(path, 0, sizeof(char) * 1024);
|
||||||
|
snprintf(path, sizeof(char) * 1024 - 1, "%s/%s", GEOIPDATADIR, file_name);
|
||||||
|
#else
|
||||||
|
char buf[MAX_PATH], *p, *q = NULL;
|
||||||
|
int len;
|
||||||
|
memset(buf, 0, sizeof(buf));
|
||||||
|
len = GetModuleFileName(GetModuleHandle(NULL), buf, sizeof(buf) - 1);
|
||||||
|
for (p = buf + len; p > buf; p--)
|
||||||
|
if (*p == '\\')
|
||||||
|
{
|
||||||
|
if (!q)
|
||||||
|
q = p;
|
||||||
|
else
|
||||||
|
*p = '/';
|
||||||
|
}
|
||||||
|
*q = 0;
|
||||||
|
memset(path, 0, sizeof(char) * 1024);
|
||||||
|
snprintf(path, sizeof(char) * 1024 - 1, "%s/%s", buf, file_name);
|
||||||
|
#endif
|
||||||
|
} else {
|
||||||
|
size_t l = strlen(custom_directory);
|
||||||
|
if (custom_directory[l-1] != '/') {
|
||||||
|
snprintf(path, sizeof(char) * 1024 - 1, "%s/%s",custom_directory, file_name);
|
||||||
|
} else {
|
||||||
|
snprintf(path, sizeof(char) * 1024 - 1, "%s%s", custom_directory, file_name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return path;
|
||||||
|
}
|
||||||
|
|
||||||
|
char ** GeoIPDBFileName = NULL;
|
||||||
|
|
||||||
|
void _GeoIP_setup_dbfilename() {
|
||||||
|
if (NULL == GeoIPDBFileName) {
|
||||||
|
GeoIPDBFileName = malloc(sizeof(char *) * NUM_DB_TYPES);
|
||||||
|
memset(GeoIPDBFileName, 0, sizeof(char *) * NUM_DB_TYPES);
|
||||||
|
|
||||||
|
GeoIPDBFileName[GEOIP_COUNTRY_EDITION] = _GeoIP_full_path_to("GeoIP.dat");
|
||||||
|
GeoIPDBFileName[GEOIP_REGION_EDITION_REV0] = _GeoIP_full_path_to("GeoIPRegion.dat");
|
||||||
|
GeoIPDBFileName[GEOIP_REGION_EDITION_REV1] = _GeoIP_full_path_to("GeoIPRegion.dat");
|
||||||
|
GeoIPDBFileName[GEOIP_CITY_EDITION_REV0] = _GeoIP_full_path_to("GeoIPCity.dat");
|
||||||
|
GeoIPDBFileName[GEOIP_CITY_EDITION_REV1] = _GeoIP_full_path_to("GeoIPCity.dat");
|
||||||
|
GeoIPDBFileName[GEOIP_ISP_EDITION] = _GeoIP_full_path_to("GeoIPISP.dat");
|
||||||
|
GeoIPDBFileName[GEOIP_ORG_EDITION] = _GeoIP_full_path_to("GeoIPOrg.dat");
|
||||||
|
GeoIPDBFileName[GEOIP_PROXY_EDITION] = _GeoIP_full_path_to("GeoIPProxy.dat");
|
||||||
|
GeoIPDBFileName[GEOIP_ASNUM_EDITION] = _GeoIP_full_path_to("GeoIPASNum.dat");
|
||||||
|
GeoIPDBFileName[GEOIP_NETSPEED_EDITION] = _GeoIP_full_path_to("GeoIPNetSpeed.dat");
|
||||||
|
GeoIPDBFileName[GEOIP_DOMAIN_EDITION] = _GeoIP_full_path_to("GeoIPDomain.dat");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static
|
||||||
|
int _file_exists(const char *file_name) {
|
||||||
|
struct stat file_stat;
|
||||||
|
return( (stat(file_name, &file_stat) == 0) ? 1:0);
|
||||||
|
}
|
||||||
|
|
||||||
|
int GeoIP_db_avail(int type) {
|
||||||
|
const char * filePath;
|
||||||
|
if (type < 0 || type >= NUM_DB_TYPES) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
filePath = GeoIPDBFileName[type];
|
||||||
|
if (NULL == filePath) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
return _file_exists(filePath);
|
||||||
|
}
|
||||||
|
|
||||||
|
static
|
||||||
|
void _setup_segments(GeoIP * gi) {
|
||||||
|
int i, j;
|
||||||
|
unsigned char delim[3];
|
||||||
|
unsigned char buf[SEGMENT_RECORD_LENGTH];
|
||||||
|
|
||||||
|
gi->databaseSegments = NULL;
|
||||||
|
|
||||||
|
/* default to GeoIP Country Edition */
|
||||||
|
gi->databaseType = GEOIP_COUNTRY_EDITION;
|
||||||
|
gi->record_length = STANDARD_RECORD_LENGTH;
|
||||||
|
fseek(gi->GeoIPDatabase, -3l, SEEK_END);
|
||||||
|
for (i = 0; i < STRUCTURE_INFO_MAX_SIZE; i++) {
|
||||||
|
fread(delim, 1, 3, gi->GeoIPDatabase);
|
||||||
|
if (delim[0] == 255 && delim[1] == 255 && delim[2] == 255) {
|
||||||
|
fread(&gi->databaseType, 1, 1, gi->GeoIPDatabase);
|
||||||
|
if (gi->databaseType >= 106) {
|
||||||
|
/* backwards compatibility with databases from April 2003 and earlier */
|
||||||
|
gi->databaseType -= 105;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (gi->databaseType == GEOIP_REGION_EDITION_REV0) {
|
||||||
|
/* Region Edition, pre June 2003 */
|
||||||
|
gi->databaseSegments = malloc(sizeof(int));
|
||||||
|
gi->databaseSegments[0] = STATE_BEGIN_REV0;
|
||||||
|
} else if (gi->databaseType == GEOIP_REGION_EDITION_REV1) {
|
||||||
|
/* Region Edition, post June 2003 */
|
||||||
|
gi->databaseSegments = malloc(sizeof(int));
|
||||||
|
gi->databaseSegments[0] = STATE_BEGIN_REV1;
|
||||||
|
} else if (gi->databaseType == GEOIP_CITY_EDITION_REV0 ||
|
||||||
|
gi->databaseType == GEOIP_CITY_EDITION_REV1 ||
|
||||||
|
gi->databaseType == GEOIP_ORG_EDITION ||
|
||||||
|
gi->databaseType == GEOIP_ISP_EDITION ||
|
||||||
|
gi->databaseType == GEOIP_ASNUM_EDITION) {
|
||||||
|
/* City/Org Editions have two segments, read offset of second segment */
|
||||||
|
gi->databaseSegments = malloc(sizeof(int));
|
||||||
|
gi->databaseSegments[0] = 0;
|
||||||
|
fread(buf, SEGMENT_RECORD_LENGTH, 1, gi->GeoIPDatabase);
|
||||||
|
for (j = 0; j < SEGMENT_RECORD_LENGTH; j++) {
|
||||||
|
gi->databaseSegments[0] += (buf[j] << (j * 8));
|
||||||
|
}
|
||||||
|
if (gi->databaseType == GEOIP_ORG_EDITION ||
|
||||||
|
gi->databaseType == GEOIP_ISP_EDITION)
|
||||||
|
gi->record_length = ORG_RECORD_LENGTH;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
} else {
|
||||||
|
fseek(gi->GeoIPDatabase, -4l, SEEK_CUR);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (gi->databaseType == GEOIP_COUNTRY_EDITION ||
|
||||||
|
gi->databaseType == GEOIP_PROXY_EDITION ||
|
||||||
|
gi->databaseType == GEOIP_NETSPEED_EDITION) {
|
||||||
|
gi->databaseSegments = malloc(sizeof(int));
|
||||||
|
gi->databaseSegments[0] = COUNTRY_BEGIN;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static
|
||||||
|
int _check_mtime(GeoIP *gi) {
|
||||||
|
struct stat buf;
|
||||||
|
if (gi->flags & GEOIP_CHECK_CACHE) {
|
||||||
|
if (stat(gi->file_path, &buf) != -1) {
|
||||||
|
if (buf.st_mtime != gi->mtime) {
|
||||||
|
/* GeoIP Database file updated */
|
||||||
|
if (gi->flags & GEOIP_MEMORY_CACHE) {
|
||||||
|
/* reload database into memory cache */
|
||||||
|
if ((gi->cache = (unsigned char*) realloc(gi->cache, buf.st_size)) == NULL) {
|
||||||
|
fprintf(stderr,"Out of memory when reloading %s\n",gi->file_path);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
/* refresh filehandle */
|
||||||
|
fclose(gi->GeoIPDatabase);
|
||||||
|
gi->GeoIPDatabase = fopen(gi->file_path,"rb");
|
||||||
|
if (gi->GeoIPDatabase == NULL) {
|
||||||
|
fprintf(stderr,"Error Opening file %s when reloading\n",gi->file_path);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
gi->mtime = buf.st_mtime;
|
||||||
|
if (gi->flags & GEOIP_MEMORY_CACHE) {
|
||||||
|
if (fread(gi->cache, sizeof(unsigned char), buf.st_size, gi->GeoIPDatabase) != (size_t) buf.st_size) {
|
||||||
|
fprintf(stderr,"Error reading file %s when reloading\n",gi->file_path);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (gi->databaseSegments != NULL) {
|
||||||
|
free(gi->databaseSegments);
|
||||||
|
gi->databaseSegments = NULL;
|
||||||
|
}
|
||||||
|
_setup_segments(gi);
|
||||||
|
if (gi->databaseSegments == NULL) {
|
||||||
|
fprintf(stderr, "Error reading file %s -- corrupt\n", gi->file_path);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if (gi->flags & GEOIP_INDEX_CACHE) {
|
||||||
|
gi->index_cache = (unsigned char *) realloc(gi->index_cache, sizeof(unsigned char) * ((gi->databaseSegments[0] * (long)gi->record_length * 2)));
|
||||||
|
if (gi->index_cache != NULL) {
|
||||||
|
fseek(gi->GeoIPDatabase, 0, SEEK_SET);
|
||||||
|
if (fread(gi->index_cache, sizeof(unsigned char), gi->databaseSegments[0] * (long)gi->record_length * 2, gi->GeoIPDatabase) != (size_t) (gi->databaseSegments[0]*(long)gi->record_length * 2)) {
|
||||||
|
fprintf(stderr,"Error reading file %s where reloading\n",gi->file_path);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
unsigned int _GeoIP_seek_record (GeoIP *gi, unsigned long ipnum) {
|
||||||
|
int depth;
|
||||||
|
unsigned int x;
|
||||||
|
unsigned char stack_buffer[2 * MAX_RECORD_LENGTH];
|
||||||
|
const unsigned char *buf = (gi->cache == NULL) ? stack_buffer : NULL;
|
||||||
|
unsigned int offset = 0;
|
||||||
|
|
||||||
|
const unsigned char * p;
|
||||||
|
int j;
|
||||||
|
|
||||||
|
_check_mtime(gi);
|
||||||
|
for (depth = 31; depth >= 0; depth--) {
|
||||||
|
if (gi->cache == NULL && gi->index_cache == NULL) {
|
||||||
|
/* read from disk */
|
||||||
|
fseek(gi->GeoIPDatabase, (long)gi->record_length * 2 * offset, SEEK_SET);
|
||||||
|
fread(stack_buffer,gi->record_length,2,gi->GeoIPDatabase);
|
||||||
|
} else if (gi->index_cache == NULL) {
|
||||||
|
/* simply point to record in memory */
|
||||||
|
buf = gi->cache + (long)gi->record_length * 2 *offset;
|
||||||
|
} else {
|
||||||
|
buf = gi->index_cache + (long)gi->record_length * 2 * offset;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (ipnum & (1 << depth)) {
|
||||||
|
/* Take the right-hand branch */
|
||||||
|
if ( gi->record_length == 3 ) {
|
||||||
|
/* Most common case is completely unrolled and uses constants. */
|
||||||
|
x = (buf[3*1 + 0] << (0*8))
|
||||||
|
+ (buf[3*1 + 1] << (1*8))
|
||||||
|
+ (buf[3*1 + 2] << (2*8));
|
||||||
|
|
||||||
|
} else {
|
||||||
|
/* General case */
|
||||||
|
j = gi->record_length;
|
||||||
|
p = &buf[2*j];
|
||||||
|
x = 0;
|
||||||
|
do {
|
||||||
|
x <<= 8;
|
||||||
|
x += *(--p);
|
||||||
|
} while ( --j );
|
||||||
|
}
|
||||||
|
|
||||||
|
} else {
|
||||||
|
/* Take the left-hand branch */
|
||||||
|
if ( gi->record_length == 3 ) {
|
||||||
|
/* Most common case is completely unrolled and uses constants. */
|
||||||
|
x = (buf[3*0 + 0] << (0*8))
|
||||||
|
+ (buf[3*0 + 1] << (1*8))
|
||||||
|
+ (buf[3*0 + 2] << (2*8));
|
||||||
|
} else {
|
||||||
|
/* General case */
|
||||||
|
j = gi->record_length;
|
||||||
|
p = &buf[1*j];
|
||||||
|
x = 0;
|
||||||
|
do {
|
||||||
|
x <<= 8;
|
||||||
|
x += *(--p);
|
||||||
|
} while ( --j );
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (x >= gi->databaseSegments[0]) {
|
||||||
|
return x;
|
||||||
|
}
|
||||||
|
offset = x;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* shouldn't reach here */
|
||||||
|
fprintf(stderr,"Error Traversing Database for ipnum = %lu - Perhaps database is corrupt?\n",ipnum);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
unsigned long _GeoIP_addr_to_num (const char *addr) {
|
||||||
|
int i;
|
||||||
|
char tok[4];
|
||||||
|
int octet;
|
||||||
|
int j = 0, k = 0;
|
||||||
|
unsigned long ipnum = 0;
|
||||||
|
char c = 0;
|
||||||
|
|
||||||
|
for (i=0; i<4; i++) {
|
||||||
|
for (;;) {
|
||||||
|
c = addr[k++];
|
||||||
|
if (c == '.' || c == '\0') {
|
||||||
|
tok[j] = '\0';
|
||||||
|
octet = atoi(tok);
|
||||||
|
if (octet > 255)
|
||||||
|
return 0;
|
||||||
|
ipnum += (octet << ((3-i)*8));
|
||||||
|
j = 0;
|
||||||
|
break;
|
||||||
|
} else if (c >= '0' && c<= '9') {
|
||||||
|
if (j > 2) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
tok[j++] = c;
|
||||||
|
} else {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if(c == '\0' && i<3) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ipnum;
|
||||||
|
}
|
||||||
|
|
||||||
|
GeoIP* GeoIP_open_type (int type, int flags) {
|
||||||
|
GeoIP * gi;
|
||||||
|
const char * filePath;
|
||||||
|
if (type < 0 || type >= NUM_DB_TYPES) {
|
||||||
|
printf("Invalid database type %d\n", type);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
_GeoIP_setup_dbfilename();
|
||||||
|
filePath = GeoIPDBFileName[type];
|
||||||
|
if (filePath == NULL) {
|
||||||
|
printf("Invalid database type %d\n", type);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
gi = GeoIP_open (filePath, flags);
|
||||||
|
return gi;
|
||||||
|
}
|
||||||
|
|
||||||
|
GeoIP* GeoIP_new (int flags) {
|
||||||
|
GeoIP * gi;
|
||||||
|
_GeoIP_setup_dbfilename();
|
||||||
|
gi = GeoIP_open (GeoIPDBFileName[GEOIP_COUNTRY_EDITION], flags);
|
||||||
|
return gi;
|
||||||
|
}
|
||||||
|
|
||||||
|
GeoIP* GeoIP_open (const char * filename, int flags) {
|
||||||
|
struct stat buf;
|
||||||
|
GeoIP * gi;
|
||||||
|
size_t len;
|
||||||
|
|
||||||
|
#ifdef _WIN32
|
||||||
|
WSADATA wsa;
|
||||||
|
if (WSAStartup(MAKEWORD(1, 1), &wsa) != 0)
|
||||||
|
return NULL;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
gi = (GeoIP *)malloc(sizeof(GeoIP));
|
||||||
|
if (gi == NULL)
|
||||||
|
return NULL;
|
||||||
|
len = sizeof(char) * (strlen(filename)+1);
|
||||||
|
gi->file_path = malloc(len);
|
||||||
|
if (gi->file_path == NULL) {
|
||||||
|
free(gi);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
strncpy(gi->file_path, filename, len);
|
||||||
|
gi->GeoIPDatabase = fopen(filename,"rb");
|
||||||
|
if (gi->GeoIPDatabase == NULL) {
|
||||||
|
fprintf(stderr,"Error Opening file %s\n",filename);
|
||||||
|
free(gi->file_path);
|
||||||
|
free(gi);
|
||||||
|
return NULL;
|
||||||
|
} else {
|
||||||
|
if (flags & GEOIP_MEMORY_CACHE) {
|
||||||
|
if (fstat(fileno(gi->GeoIPDatabase), &buf) == -1) {
|
||||||
|
fprintf(stderr,"Error stating file %s\n",filename);
|
||||||
|
free(gi->file_path);
|
||||||
|
free(gi);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
gi->mtime = buf.st_mtime;
|
||||||
|
gi->cache = (unsigned char *) malloc(sizeof(unsigned char) * buf.st_size);
|
||||||
|
if (gi->cache != NULL) {
|
||||||
|
if (fread(gi->cache, sizeof(unsigned char), buf.st_size, gi->GeoIPDatabase) != (size_t) buf.st_size) {
|
||||||
|
fprintf(stderr,"Error reading file %s\n",filename);
|
||||||
|
free(gi->cache);
|
||||||
|
free(gi->file_path);
|
||||||
|
free(gi);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if (flags & GEOIP_CHECK_CACHE) {
|
||||||
|
if (fstat(fileno(gi->GeoIPDatabase), &buf) == -1) {
|
||||||
|
fprintf(stderr,"Error stating file %s\n",filename);
|
||||||
|
free(gi->file_path);
|
||||||
|
free(gi);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
gi->mtime = buf.st_mtime;
|
||||||
|
}
|
||||||
|
gi->cache = NULL;
|
||||||
|
}
|
||||||
|
gi->flags = flags;
|
||||||
|
_setup_segments(gi);
|
||||||
|
if (flags & GEOIP_INDEX_CACHE) {
|
||||||
|
gi->index_cache = (unsigned char *) malloc(sizeof(unsigned char) * ((gi->databaseSegments[0] * (long)gi->record_length * 2)));
|
||||||
|
if (gi->index_cache != NULL) {
|
||||||
|
fseek(gi->GeoIPDatabase, 0, SEEK_SET);
|
||||||
|
if (fread(gi->index_cache, sizeof(unsigned char), gi->databaseSegments[0] * (long)gi->record_length * 2, gi->GeoIPDatabase) != (size_t) (gi->databaseSegments[0]*(long)gi->record_length * 2)) {
|
||||||
|
fprintf(stderr,"Error reading file %s\n",filename);
|
||||||
|
free(gi->databaseSegments);
|
||||||
|
free(gi->index_cache);
|
||||||
|
free(gi);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
gi->index_cache = NULL;
|
||||||
|
}
|
||||||
|
return gi;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void GeoIP_delete (GeoIP *gi) {
|
||||||
|
if (gi == NULL )
|
||||||
|
return;
|
||||||
|
if (gi->GeoIPDatabase != NULL)
|
||||||
|
fclose(gi->GeoIPDatabase);
|
||||||
|
if (gi->cache != NULL)
|
||||||
|
free(gi->cache);
|
||||||
|
if (gi->index_cache != NULL)
|
||||||
|
free(gi->index_cache);
|
||||||
|
if (gi->file_path != NULL)
|
||||||
|
free(gi->file_path);
|
||||||
|
if (gi->databaseSegments != NULL)
|
||||||
|
free(gi->databaseSegments);
|
||||||
|
free(gi);
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *GeoIP_country_code_by_name (GeoIP* gi, const char *name) {
|
||||||
|
int country_id;
|
||||||
|
country_id = GeoIP_id_by_name(gi, name);
|
||||||
|
return (country_id > 0) ? GeoIP_country_code[country_id] : NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *GeoIP_country_code3_by_name (GeoIP* gi, const char *name) {
|
||||||
|
int country_id;
|
||||||
|
country_id = GeoIP_id_by_name(gi, name);
|
||||||
|
return (country_id > 0) ? GeoIP_country_code3[country_id] : NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *GeoIP_country_name_by_name (GeoIP* gi, const char *name) {
|
||||||
|
int country_id;
|
||||||
|
country_id = GeoIP_id_by_name(gi, name);
|
||||||
|
return (country_id > 0) ? GeoIP_country_name[country_id] : NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
unsigned long _GeoIP_lookupaddress (const char *host) {
|
||||||
|
unsigned long addr = inet_addr(host);
|
||||||
|
struct hostent phe2;
|
||||||
|
struct hostent * phe = &phe2;
|
||||||
|
char *buf = NULL;
|
||||||
|
int buflength = 16384;
|
||||||
|
int herr = 0;
|
||||||
|
int result = 0;
|
||||||
|
#ifdef HAVE_GETHOSTBYNAME_R
|
||||||
|
buf = malloc(buflength);
|
||||||
|
#endif
|
||||||
|
if (addr == INADDR_NONE) {
|
||||||
|
#ifdef HAVE_GETHOSTBYNAME_R
|
||||||
|
while (1) {
|
||||||
|
/* we use gethostbyname_r here because it is thread-safe and gethostbyname is not */
|
||||||
|
#ifdef GETHOSTBYNAME_R_RETURNS_INT
|
||||||
|
result = gethostbyname_r(host,&phe2,buf,buflength,&phe,&herr);
|
||||||
|
#else
|
||||||
|
phe = gethostbyname_r(host,&phe2,buf,buflength,&herr);
|
||||||
|
#endif
|
||||||
|
if (herr != ERANGE)
|
||||||
|
break;
|
||||||
|
if (result == 0)
|
||||||
|
break;
|
||||||
|
/* double the buffer if the buffer is too small */
|
||||||
|
buflength = buflength * 2;
|
||||||
|
buf = realloc(buf,buflength);
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
#ifndef HAVE_GETHOSTBYNAME_R
|
||||||
|
/* Some systems do not support gethostbyname_r, such as Mac OS X */
|
||||||
|
phe = gethostbyname(host);
|
||||||
|
#endif
|
||||||
|
if (!phe || result != 0) {
|
||||||
|
free(buf);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
addr = *((unsigned long *) phe->h_addr_list[0]);
|
||||||
|
}
|
||||||
|
#ifdef HAVE_GETHOSTBYNAME_R
|
||||||
|
free(buf);
|
||||||
|
#endif
|
||||||
|
return ntohl(addr);
|
||||||
|
}
|
||||||
|
|
||||||
|
int GeoIP_id_by_name (GeoIP* gi, const char *name) {
|
||||||
|
unsigned long ipnum;
|
||||||
|
int ret;
|
||||||
|
if (name == NULL) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
if (gi->databaseType != GEOIP_COUNTRY_EDITION && gi->databaseType != GEOIP_PROXY_EDITION && gi->databaseType != GEOIP_NETSPEED_EDITION) {
|
||||||
|
printf("Invalid database type %s, expected %s\n", GeoIPDBDescription[(int)gi->databaseType], GeoIPDBDescription[GEOIP_COUNTRY_EDITION]);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
if (!(ipnum = _GeoIP_lookupaddress(name)))
|
||||||
|
return 0;
|
||||||
|
ret = _GeoIP_seek_record(gi, ipnum) - COUNTRY_BEGIN;
|
||||||
|
return ret;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *GeoIP_country_code_by_addr (GeoIP* gi, const char *addr) {
|
||||||
|
int country_id;
|
||||||
|
country_id = GeoIP_id_by_addr(gi, addr);
|
||||||
|
return (country_id > 0) ? GeoIP_country_code[country_id] : NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *GeoIP_country_code3_by_addr (GeoIP* gi, const char *addr) {
|
||||||
|
int country_id;
|
||||||
|
country_id = GeoIP_id_by_addr(gi, addr);
|
||||||
|
return (country_id > 0) ? GeoIP_country_code3[country_id] : NULL;
|
||||||
|
return GeoIP_country_code3[country_id];
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *GeoIP_country_name_by_addr (GeoIP* gi, const char *addr) {
|
||||||
|
int country_id;
|
||||||
|
country_id = GeoIP_id_by_addr(gi, addr);
|
||||||
|
return (country_id > 0) ? GeoIP_country_name[country_id] : NULL;
|
||||||
|
return GeoIP_country_name[country_id];
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *GeoIP_country_name_by_ipnum (GeoIP* gi, unsigned long ipnum) {
|
||||||
|
int country_id;
|
||||||
|
country_id = GeoIP_id_by_ipnum(gi, ipnum);
|
||||||
|
return (country_id > 0) ? GeoIP_country_name[country_id] : NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *GeoIP_country_code_by_ipnum (GeoIP* gi, unsigned long ipnum) {
|
||||||
|
int country_id;
|
||||||
|
country_id = GeoIP_id_by_ipnum(gi, ipnum);
|
||||||
|
return (country_id > 0) ? GeoIP_country_code[country_id] : NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *GeoIP_country_code3_by_ipnum (GeoIP* gi, unsigned long ipnum) {
|
||||||
|
int country_id;
|
||||||
|
country_id = GeoIP_id_by_ipnum(gi, ipnum);
|
||||||
|
return (country_id > 0) ? GeoIP_country_code3[country_id] : NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
int GeoIP_country_id_by_addr (GeoIP* gi, const char *addr) {
|
||||||
|
return GeoIP_id_by_addr(gi, addr);
|
||||||
|
}
|
||||||
|
|
||||||
|
int GeoIP_country_id_by_name (GeoIP* gi, const char *host) {
|
||||||
|
return GeoIP_id_by_name(gi, host);
|
||||||
|
}
|
||||||
|
|
||||||
|
int GeoIP_id_by_addr (GeoIP* gi, const char *addr) {
|
||||||
|
unsigned long ipnum;
|
||||||
|
int ret;
|
||||||
|
if (addr == NULL) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
if (gi->databaseType != GEOIP_COUNTRY_EDITION &&
|
||||||
|
gi->databaseType != GEOIP_PROXY_EDITION &&
|
||||||
|
gi->databaseType != GEOIP_NETSPEED_EDITION) {
|
||||||
|
printf("Invalid database type %s, expected %s\n",
|
||||||
|
GeoIPDBDescription[(int)gi->databaseType],
|
||||||
|
GeoIPDBDescription[GEOIP_COUNTRY_EDITION]);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
ipnum = _GeoIP_addr_to_num(addr);
|
||||||
|
ret = _GeoIP_seek_record(gi, ipnum) - COUNTRY_BEGIN;
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
int GeoIP_id_by_ipnum (GeoIP* gi, unsigned long ipnum) {
|
||||||
|
int ret;
|
||||||
|
if (ipnum == 0) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
if (gi->databaseType != GEOIP_COUNTRY_EDITION &&
|
||||||
|
gi->databaseType != GEOIP_PROXY_EDITION &&
|
||||||
|
gi->databaseType != GEOIP_NETSPEED_EDITION) {
|
||||||
|
printf("Invalid database type %s, expected %s\n",
|
||||||
|
GeoIPDBDescription[(int)gi->databaseType],
|
||||||
|
GeoIPDBDescription[GEOIP_COUNTRY_EDITION]);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
ret = _GeoIP_seek_record(gi, ipnum) - COUNTRY_BEGIN;
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
char *GeoIP_database_info (GeoIP* gi) {
|
||||||
|
int i;
|
||||||
|
unsigned char buf[3];
|
||||||
|
char *retval;
|
||||||
|
int hasStructureInfo = 0;
|
||||||
|
|
||||||
|
if(gi == NULL)
|
||||||
|
return NULL;
|
||||||
|
|
||||||
|
_check_mtime(gi);
|
||||||
|
fseek(gi->GeoIPDatabase, -3l, SEEK_END);
|
||||||
|
|
||||||
|
/* first get past the database structure information */
|
||||||
|
for (i = 0; i < STRUCTURE_INFO_MAX_SIZE; i++) {
|
||||||
|
fread(buf, 1, 3, gi->GeoIPDatabase);
|
||||||
|
if (buf[0] == 255 && buf[1] == 255 && buf[2] == 255) {
|
||||||
|
hasStructureInfo = 1;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
fseek(gi->GeoIPDatabase, -4l, SEEK_CUR);
|
||||||
|
}
|
||||||
|
if (hasStructureInfo == 1) {
|
||||||
|
fseek(gi->GeoIPDatabase, -3l, SEEK_CUR);
|
||||||
|
} else {
|
||||||
|
/* no structure info, must be pre Sep 2002 database, go back to end */
|
||||||
|
fseek(gi->GeoIPDatabase, -3l, SEEK_END);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (i = 0; i < DATABASE_INFO_MAX_SIZE; i++) {
|
||||||
|
fread(buf, 1, 3, gi->GeoIPDatabase);
|
||||||
|
if (buf[0] == 0 && buf[1] == 0 && buf[2] == 0) {
|
||||||
|
retval = malloc(sizeof(char) * (i+1));
|
||||||
|
if (retval == NULL) {
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
fread(retval, 1, i, gi->GeoIPDatabase);
|
||||||
|
retval[i] = '\0';
|
||||||
|
return retval;
|
||||||
|
}
|
||||||
|
fseek(gi->GeoIPDatabase, -4l, SEEK_CUR);
|
||||||
|
}
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* GeoIP Region Edition functions */
|
||||||
|
|
||||||
|
void GeoIP_assign_region_by_inetaddr(GeoIP* gi, unsigned long inetaddr, GeoIPRegion *region) {
|
||||||
|
unsigned int seek_region;
|
||||||
|
|
||||||
|
/* This also writes in the terminating NULs (if you decide to
|
||||||
|
* keep them) and clear any fields that are not set. */
|
||||||
|
memset(region, 0, sizeof(GeoIPRegion));
|
||||||
|
|
||||||
|
seek_region = _GeoIP_seek_record(gi, ntohl(inetaddr));
|
||||||
|
|
||||||
|
if (gi->databaseType == GEOIP_REGION_EDITION_REV0) {
|
||||||
|
/* Region Edition, pre June 2003 */
|
||||||
|
seek_region -= STATE_BEGIN_REV0;
|
||||||
|
if (seek_region >= 1000) {
|
||||||
|
region->country_code[0] = 'U';
|
||||||
|
region->country_code[1] = 'S';
|
||||||
|
region->region[0] = (char) ((seek_region - 1000)/26 + 65);
|
||||||
|
region->region[1] = (char) ((seek_region - 1000)%26 + 65);
|
||||||
|
} else {
|
||||||
|
memcpy(region->country_code, GeoIP_country_code[seek_region], 2);
|
||||||
|
}
|
||||||
|
} else if (gi->databaseType == GEOIP_REGION_EDITION_REV1) {
|
||||||
|
/* Region Edition, post June 2003 */
|
||||||
|
seek_region -= STATE_BEGIN_REV1;
|
||||||
|
if (seek_region < US_OFFSET) {
|
||||||
|
/* Unknown */
|
||||||
|
/* we don't need to do anything here b/c we memset region to 0 */
|
||||||
|
} else if (seek_region < CANADA_OFFSET) {
|
||||||
|
/* USA State */
|
||||||
|
region->country_code[0] = 'U';
|
||||||
|
region->country_code[1] = 'S';
|
||||||
|
region->region[0] = (char) ((seek_region - US_OFFSET)/26 + 65);
|
||||||
|
region->region[1] = (char) ((seek_region - US_OFFSET)%26 + 65);
|
||||||
|
} else if (seek_region < WORLD_OFFSET) {
|
||||||
|
/* Canada Province */
|
||||||
|
region->country_code[0] = 'C';
|
||||||
|
region->country_code[1] = 'A';
|
||||||
|
region->region[0] = (char) ((seek_region - CANADA_OFFSET)/26 + 65);
|
||||||
|
region->region[1] = (char) ((seek_region - CANADA_OFFSET)%26 + 65);
|
||||||
|
} else {
|
||||||
|
/* Not US or Canada */
|
||||||
|
memcpy(region->country_code, GeoIP_country_code[(seek_region - WORLD_OFFSET) / FIPS_RANGE], 2);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static
|
||||||
|
GeoIPRegion * _get_region(GeoIP* gi, unsigned long ipnum) {
|
||||||
|
GeoIPRegion * region;
|
||||||
|
|
||||||
|
region = malloc(sizeof(GeoIPRegion));
|
||||||
|
if (region) {
|
||||||
|
GeoIP_assign_region_by_inetaddr(gi, htonl(ipnum), region);
|
||||||
|
}
|
||||||
|
return region;
|
||||||
|
}
|
||||||
|
|
||||||
|
GeoIPRegion * GeoIP_region_by_addr (GeoIP* gi, const char *addr) {
|
||||||
|
unsigned long ipnum;
|
||||||
|
if (addr == NULL) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
if (gi->databaseType != GEOIP_REGION_EDITION_REV0 &&
|
||||||
|
gi->databaseType != GEOIP_REGION_EDITION_REV1) {
|
||||||
|
printf("Invalid database type %s, expected %s\n", GeoIPDBDescription[(int)gi->databaseType], GeoIPDBDescription[GEOIP_REGION_EDITION_REV1]);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
ipnum = _GeoIP_addr_to_num(addr);
|
||||||
|
return _get_region(gi, ipnum);
|
||||||
|
}
|
||||||
|
|
||||||
|
GeoIPRegion * GeoIP_region_by_name (GeoIP* gi, const char *name) {
|
||||||
|
unsigned long ipnum;
|
||||||
|
if (name == NULL) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
if (gi->databaseType != GEOIP_REGION_EDITION_REV0 &&
|
||||||
|
gi->databaseType != GEOIP_REGION_EDITION_REV1) {
|
||||||
|
printf("Invalid database type %s, expected %s\n", GeoIPDBDescription[(int)gi->databaseType], GeoIPDBDescription[GEOIP_REGION_EDITION_REV1]);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
if (!(ipnum = _GeoIP_lookupaddress(name)))
|
||||||
|
return 0;
|
||||||
|
return _get_region(gi, ipnum);
|
||||||
|
}
|
||||||
|
|
||||||
|
GeoIPRegion * GeoIP_region_by_ipnum (GeoIP* gi, unsigned long ipnum) {
|
||||||
|
if (gi->databaseType != GEOIP_REGION_EDITION_REV0 &&
|
||||||
|
gi->databaseType != GEOIP_REGION_EDITION_REV1) {
|
||||||
|
printf("Invalid database type %s, expected %s\n", GeoIPDBDescription[(int)gi->databaseType], GeoIPDBDescription[GEOIP_REGION_EDITION_REV1]);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
return _get_region(gi, ipnum);
|
||||||
|
}
|
||||||
|
|
||||||
|
void GeoIPRegion_delete (GeoIPRegion *gir) {
|
||||||
|
free(gir);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* GeoIP Organization, ISP and AS Number Edition private method */
|
||||||
|
static
|
||||||
|
char *_get_name (GeoIP* gi, unsigned long ipnum) {
|
||||||
|
int seek_org;
|
||||||
|
char buf[MAX_ORG_RECORD_LENGTH];
|
||||||
|
char * org_buf, * buf_pointer;
|
||||||
|
int record_pointer;
|
||||||
|
size_t len;
|
||||||
|
|
||||||
|
if (gi->databaseType != GEOIP_ORG_EDITION &&
|
||||||
|
gi->databaseType != GEOIP_ISP_EDITION &&
|
||||||
|
gi->databaseType != GEOIP_ASNUM_EDITION) {
|
||||||
|
printf("Invalid database type %s, expected %s\n", GeoIPDBDescription[(int)gi->databaseType], GeoIPDBDescription[GEOIP_ORG_EDITION]);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
seek_org = _GeoIP_seek_record(gi, ipnum);
|
||||||
|
if (seek_org == gi->databaseSegments[0])
|
||||||
|
return NULL;
|
||||||
|
|
||||||
|
record_pointer = seek_org + (2 * gi->record_length - 1) * gi->databaseSegments[0];
|
||||||
|
|
||||||
|
if (gi->cache == NULL) {
|
||||||
|
fseek(gi->GeoIPDatabase, record_pointer, SEEK_SET);
|
||||||
|
fread(buf, sizeof(char), MAX_ORG_RECORD_LENGTH, gi->GeoIPDatabase);
|
||||||
|
len = sizeof(char) * (strlen(buf)+1);
|
||||||
|
org_buf = malloc(len);
|
||||||
|
strncpy(org_buf, buf, len);
|
||||||
|
} else {
|
||||||
|
buf_pointer = gi->cache + (long)record_pointer;
|
||||||
|
len = sizeof(char) * (strlen(buf_pointer)+1);
|
||||||
|
org_buf = malloc(len);
|
||||||
|
strncpy(org_buf, buf_pointer, len);
|
||||||
|
}
|
||||||
|
return org_buf;
|
||||||
|
}
|
||||||
|
|
||||||
|
char *GeoIP_name_by_ipnum (GeoIP* gi, unsigned long ipnum) {
|
||||||
|
return _get_name(gi,ipnum);
|
||||||
|
}
|
||||||
|
|
||||||
|
char *GeoIP_name_by_addr (GeoIP* gi, const char *addr) {
|
||||||
|
unsigned long ipnum;
|
||||||
|
if (addr == NULL) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
ipnum = _GeoIP_addr_to_num(addr);
|
||||||
|
return _get_name(gi, ipnum);
|
||||||
|
}
|
||||||
|
|
||||||
|
char *GeoIP_name_by_name (GeoIP* gi, const char *name) {
|
||||||
|
unsigned long ipnum;
|
||||||
|
if (name == NULL) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
if (!(ipnum = _GeoIP_lookupaddress(name)))
|
||||||
|
return 0;
|
||||||
|
return _get_name(gi, ipnum);
|
||||||
|
}
|
||||||
|
|
||||||
|
char *GeoIP_org_by_ipnum (GeoIP* gi, unsigned long ipnum) {
|
||||||
|
return GeoIP_name_by_ipnum(gi, ipnum);
|
||||||
|
}
|
||||||
|
|
||||||
|
char *GeoIP_org_by_addr (GeoIP* gi, const char *addr) {
|
||||||
|
return GeoIP_name_by_addr(gi, addr);
|
||||||
|
}
|
||||||
|
|
||||||
|
char *GeoIP_org_by_name (GeoIP* gi, const char *name) {
|
||||||
|
return GeoIP_name_by_name(gi, name);
|
||||||
|
}
|
||||||
|
|
||||||
|
unsigned char GeoIP_database_edition (GeoIP* gi) {
|
||||||
|
return gi->databaseType;
|
||||||
|
}
|
BIN
extensions/geoip/GeoIP.dat
Normal file
BIN
extensions/geoip/GeoIP.dat
Normal file
Binary file not shown.
169
extensions/geoip/GeoIP.h
Normal file
169
extensions/geoip/GeoIP.h
Normal file
@ -0,0 +1,169 @@
|
|||||||
|
/* -*- 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<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
|
||||||
|
|
||||||
|
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;
|
||||||
|
char record_length;
|
||||||
|
int record_iter; /* used in GeoIP_next_record */
|
||||||
|
} GeoIP;
|
||||||
|
|
||||||
|
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,
|
||||||
|
} 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
|
||||||
|
} 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;
|
||||||
|
|
||||||
|
extern const char GeoIP_country_code[251][3];
|
||||||
|
extern const char GeoIP_country_code3[251][4];
|
||||||
|
extern const char * GeoIP_country_name[251];
|
||||||
|
extern const char GeoIP_country_continent[251][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);
|
||||||
|
|
||||||
|
/* 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);
|
||||||
|
/* 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 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);
|
||||||
|
|
||||||
|
/* 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);
|
||||||
|
|
||||||
|
/* 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_database_info (GeoIP* gi);
|
||||||
|
GEOIP_API unsigned char GeoIP_database_edition (GeoIP* gi);
|
||||||
|
|
||||||
|
/* 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 */
|
83
extensions/geoip/extension.cpp
Normal file
83
extensions/geoip/extension.cpp
Normal file
@ -0,0 +1,83 @@
|
|||||||
|
#include "extension.h"
|
||||||
|
#include "GeoIP.h"
|
||||||
|
|
||||||
|
GeoIP_Extension g_GeoIP;
|
||||||
|
GeoIP *gi = NULL;
|
||||||
|
|
||||||
|
SMEXT_LINK(&g_GeoIP);
|
||||||
|
|
||||||
|
bool GeoIP_Extension::SDK_OnLoad(char *error, size_t err_max, bool late)
|
||||||
|
{
|
||||||
|
char *path = "GeoIP.dat"; //:TODO: build a real path here
|
||||||
|
//:TODO: log any errors on load.
|
||||||
|
gi = GeoIP_open(path, GEOIP_MEMORY_CACHE);
|
||||||
|
|
||||||
|
if (!gi)
|
||||||
|
{
|
||||||
|
//:TODO: log
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
void GeoIP_Extension::SDK_OnUnload()
|
||||||
|
{
|
||||||
|
GeoIP_delete(gi);
|
||||||
|
gi = NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
/*******************************
|
||||||
|
* *
|
||||||
|
* GEOIP NATIVE IMPLEMENTATIONS *
|
||||||
|
* *
|
||||||
|
*******************************/
|
||||||
|
|
||||||
|
inline void StripPort(char *ip)
|
||||||
|
{
|
||||||
|
char *tmp = strchr(ip, ':');
|
||||||
|
if (!tmp)
|
||||||
|
return;
|
||||||
|
*tmp = '\0';
|
||||||
|
}
|
||||||
|
|
||||||
|
static cell_t sm_Geoip_Code2(IPluginContext *pCtx, const cell_t *params)
|
||||||
|
{
|
||||||
|
char *ip;
|
||||||
|
const char *ccode;
|
||||||
|
|
||||||
|
pCtx->LocalToString(params[1], &ip);
|
||||||
|
StripPort(ip);
|
||||||
|
|
||||||
|
ccode = GeoIP_country_code_by_addr(gi, ip);
|
||||||
|
pCtx->StringToLocal(params[2], 3, (ccode) ? ccode : "er");
|
||||||
|
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
static cell_t sm_Geoip_Code3(IPluginContext *pCtx, const cell_t *params)
|
||||||
|
{
|
||||||
|
char *ip;
|
||||||
|
const char *ccode;
|
||||||
|
|
||||||
|
pCtx->LocalToString(params[1], &ip);
|
||||||
|
StripPort(ip);
|
||||||
|
|
||||||
|
ccode = GeoIP_country_code3_by_addr(gi, ip);
|
||||||
|
pCtx->StringToLocal(params[2], 4, (ccode) ? ccode : "err");
|
||||||
|
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
static cell_t sm_Geoip_Country(IPluginContext *pCtx, const cell_t *params)
|
||||||
|
{
|
||||||
|
char *ip;
|
||||||
|
const char *ccode;
|
||||||
|
|
||||||
|
pCtx->LocalToString(params[1], &ip);
|
||||||
|
StripPort(ip);
|
||||||
|
|
||||||
|
ccode = GeoIP_country_name_by_addr(gi, ip);
|
||||||
|
pCtx->StringToLocal(params[2], params[3], (ccode) ? ccode : "error");
|
||||||
|
|
||||||
|
return 1;
|
||||||
|
}
|
61
extensions/geoip/extension.h
Normal file
61
extensions/geoip/extension.h
Normal file
@ -0,0 +1,61 @@
|
|||||||
|
#ifndef _INCLUDE_SOURCEMOD_EXTENSION_PROPER_H_
|
||||||
|
#define _INCLUDE_SOURCEMOD_EXTENSION_PROPER_H_
|
||||||
|
|
||||||
|
#include "smsdk_ext.h"
|
||||||
|
|
||||||
|
#define GEOIPDATADIR ""
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief GeoIP implementation of the SDK Extension.
|
||||||
|
* Note: Uncomment one of the pre-defined virtual functions in order to use it.
|
||||||
|
*/
|
||||||
|
class GeoIP_Extension : public SDKExtension
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
/**
|
||||||
|
* @brief This is called after the initial loading sequence has been processed.
|
||||||
|
*
|
||||||
|
* @param error Error message buffer.
|
||||||
|
* @param err_max Size of error message buffer.
|
||||||
|
* @param late Whether or not the module was loaded after map load.
|
||||||
|
* @return True to succeed loading, false to fail.
|
||||||
|
*/
|
||||||
|
virtual bool SDK_OnLoad(char *error, size_t err_max, bool late);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief This is called right before the extension is unloaded.
|
||||||
|
*/
|
||||||
|
virtual void SDK_OnUnload();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief This is called once all known extensions have been loaded.
|
||||||
|
* Note: It is is a good idea to add natives here, if any are provided.
|
||||||
|
*/
|
||||||
|
//virtual void SDK_OnAllLoaded();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Called when the pause state is changed.
|
||||||
|
*/
|
||||||
|
//virtual void SDK_OnPauseChange(bool paused);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief this is called when Core wants to know if your extension is working.
|
||||||
|
*
|
||||||
|
* @param error Error message buffer.
|
||||||
|
* @param err_max Size of error message buffer.
|
||||||
|
* @return True if working, false otherwise.
|
||||||
|
*/
|
||||||
|
//virtual void QueryRunning(char *error, size_t maxlength);
|
||||||
|
public:
|
||||||
|
#if defined SMEXT_CONF_METAMOD
|
||||||
|
/**
|
||||||
|
* Read smext_base.h for documentation on these.
|
||||||
|
*/
|
||||||
|
|
||||||
|
//virtual bool SDK_OnMetamodLoad(char *error, size_t err_max, bool late);
|
||||||
|
//virtual bool SDK_OnMetamodUnload(char *error, size_t err_max);
|
||||||
|
//virtual bool SDK_OnMetamodPauseChange(bool paused, char *error, size_t err_max);
|
||||||
|
#endif
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif //_INCLUDE_SOURCEMOD_EXTENSION_PROPER_H_
|
26
extensions/geoip/geoip.sln
Normal file
26
extensions/geoip/geoip.sln
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
|
||||||
|
Microsoft Visual Studio Solution File, Format Version 9.00
|
||||||
|
# Visual Studio 2005
|
||||||
|
Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "geoip", "geoip.vcproj", "{B3E797CF-4E77-4C9D-B8A8-7589B6902206}"
|
||||||
|
EndProject
|
||||||
|
Global
|
||||||
|
GlobalSection(SolutionConfigurationPlatforms) = preSolution
|
||||||
|
Debug - Metamod|Win32 = Debug - Metamod|Win32
|
||||||
|
Debug|Win32 = Debug|Win32
|
||||||
|
Release - Metamod|Win32 = Release - Metamod|Win32
|
||||||
|
Release|Win32 = Release|Win32
|
||||||
|
EndGlobalSection
|
||||||
|
GlobalSection(ProjectConfigurationPlatforms) = postSolution
|
||||||
|
{B3E797CF-4E77-4C9D-B8A8-7589B6902206}.Debug - Metamod|Win32.ActiveCfg = Debug - Metamod|Win32
|
||||||
|
{B3E797CF-4E77-4C9D-B8A8-7589B6902206}.Debug - Metamod|Win32.Build.0 = Debug - Metamod|Win32
|
||||||
|
{B3E797CF-4E77-4C9D-B8A8-7589B6902206}.Debug|Win32.ActiveCfg = Debug|Win32
|
||||||
|
{B3E797CF-4E77-4C9D-B8A8-7589B6902206}.Debug|Win32.Build.0 = Debug|Win32
|
||||||
|
{B3E797CF-4E77-4C9D-B8A8-7589B6902206}.Release - Metamod|Win32.ActiveCfg = Release - Metamod|Win32
|
||||||
|
{B3E797CF-4E77-4C9D-B8A8-7589B6902206}.Release - Metamod|Win32.Build.0 = Release - Metamod|Win32
|
||||||
|
{B3E797CF-4E77-4C9D-B8A8-7589B6902206}.Release|Win32.ActiveCfg = Release|Win32
|
||||||
|
{B3E797CF-4E77-4C9D-B8A8-7589B6902206}.Release|Win32.Build.0 = Release|Win32
|
||||||
|
EndGlobalSection
|
||||||
|
GlobalSection(SolutionProperties) = preSolution
|
||||||
|
HideSolutionNode = FALSE
|
||||||
|
EndGlobalSection
|
||||||
|
EndGlobal
|
379
extensions/geoip/geoip.vcproj
Normal file
379
extensions/geoip/geoip.vcproj
Normal file
@ -0,0 +1,379 @@
|
|||||||
|
<?xml version="1.0" encoding="Windows-1252"?>
|
||||||
|
<VisualStudioProject
|
||||||
|
ProjectType="Visual C++"
|
||||||
|
Version="8,00"
|
||||||
|
Name="geoip"
|
||||||
|
ProjectGUID="{B3E797CF-4E77-4C9D-B8A8-7589B6902206}"
|
||||||
|
RootNamespace="geoip"
|
||||||
|
Keyword="Win32Proj"
|
||||||
|
>
|
||||||
|
<Platforms>
|
||||||
|
<Platform
|
||||||
|
Name="Win32"
|
||||||
|
/>
|
||||||
|
</Platforms>
|
||||||
|
<ToolFiles>
|
||||||
|
</ToolFiles>
|
||||||
|
<Configurations>
|
||||||
|
<Configuration
|
||||||
|
Name="Debug|Win32"
|
||||||
|
OutputDirectory="$(SolutionDir)$(ConfigurationName)"
|
||||||
|
IntermediateDirectory="$(ConfigurationName)"
|
||||||
|
ConfigurationType="2"
|
||||||
|
CharacterSet="2"
|
||||||
|
>
|
||||||
|
<Tool
|
||||||
|
Name="VCPreBuildEventTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCCustomBuildTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCXMLDataGeneratorTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCWebServiceProxyGeneratorTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCMIDLTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCCLCompilerTool"
|
||||||
|
Optimization="0"
|
||||||
|
PreprocessorDefinitions="WIN32;_DEBUG;_WINDOWS;_USRDLL;geoip_EXPORTS;_CRT_SECURE_NO_DEPRECATE;_CRT_NONSTDC_NO_DEPRECATE"
|
||||||
|
MinimalRebuild="true"
|
||||||
|
BasicRuntimeChecks="3"
|
||||||
|
RuntimeLibrary="1"
|
||||||
|
UsePrecompiledHeader="0"
|
||||||
|
WarningLevel="3"
|
||||||
|
Detect64BitPortabilityProblems="true"
|
||||||
|
DebugInformationFormat="4"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCManagedResourceCompilerTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCResourceCompilerTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCPreLinkEventTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCLinkerTool"
|
||||||
|
AdditionalDependencies="ws2_32.lib"
|
||||||
|
OutputFile="$(OutDir)\geoip.ext.dll"
|
||||||
|
LinkIncremental="2"
|
||||||
|
GenerateDebugInformation="true"
|
||||||
|
SubSystem="2"
|
||||||
|
TargetMachine="1"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCALinkTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCManifestTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCXDCMakeTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCBscMakeTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCFxCopTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCAppVerifierTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCWebDeploymentTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCPostBuildEventTool"
|
||||||
|
/>
|
||||||
|
</Configuration>
|
||||||
|
<Configuration
|
||||||
|
Name="Release|Win32"
|
||||||
|
OutputDirectory="$(SolutionDir)$(ConfigurationName)"
|
||||||
|
IntermediateDirectory="$(ConfigurationName)"
|
||||||
|
ConfigurationType="2"
|
||||||
|
CharacterSet="1"
|
||||||
|
WholeProgramOptimization="1"
|
||||||
|
>
|
||||||
|
<Tool
|
||||||
|
Name="VCPreBuildEventTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCCustomBuildTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCXMLDataGeneratorTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCWebServiceProxyGeneratorTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCMIDLTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCCLCompilerTool"
|
||||||
|
PreprocessorDefinitions="WIN32;NDEBUG;_WINDOWS;_USRDLL;geoip_EXPORTS;_CRT_SECURE_NO_DEPRECATE;_CRT_NONSTDC_NO_DEPRECATE"
|
||||||
|
RuntimeLibrary="0"
|
||||||
|
UsePrecompiledHeader="0"
|
||||||
|
WarningLevel="3"
|
||||||
|
Detect64BitPortabilityProblems="true"
|
||||||
|
DebugInformationFormat="3"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCManagedResourceCompilerTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCResourceCompilerTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCPreLinkEventTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCLinkerTool"
|
||||||
|
AdditionalDependencies="ws2_32.lib"
|
||||||
|
OutputFile="$(OutDir)\geoip.ext.dll"
|
||||||
|
LinkIncremental="1"
|
||||||
|
GenerateDebugInformation="true"
|
||||||
|
SubSystem="2"
|
||||||
|
OptimizeReferences="2"
|
||||||
|
EnableCOMDATFolding="2"
|
||||||
|
TargetMachine="1"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCALinkTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCManifestTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCXDCMakeTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCBscMakeTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCFxCopTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCAppVerifierTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCWebDeploymentTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCPostBuildEventTool"
|
||||||
|
/>
|
||||||
|
</Configuration>
|
||||||
|
<Configuration
|
||||||
|
Name="Debug - Metamod|Win32"
|
||||||
|
OutputDirectory="$(SolutionDir)$(ConfigurationName)"
|
||||||
|
IntermediateDirectory="$(ConfigurationName)"
|
||||||
|
ConfigurationType="2"
|
||||||
|
CharacterSet="2"
|
||||||
|
>
|
||||||
|
<Tool
|
||||||
|
Name="VCPreBuildEventTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCCustomBuildTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCXMLDataGeneratorTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCWebServiceProxyGeneratorTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCMIDLTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCCLCompilerTool"
|
||||||
|
Optimization="0"
|
||||||
|
PreprocessorDefinitions="WIN32;_DEBUG;_WINDOWS;_USRDLL;geoip_EXPORTS;_CRT_SECURE_NO_DEPRECATE;SMEXT_CONF_METAMOD;_CRT_NONSTDC_NO_DEPRECATE"
|
||||||
|
MinimalRebuild="true"
|
||||||
|
BasicRuntimeChecks="3"
|
||||||
|
RuntimeLibrary="1"
|
||||||
|
UsePrecompiledHeader="0"
|
||||||
|
WarningLevel="3"
|
||||||
|
Detect64BitPortabilityProblems="true"
|
||||||
|
DebugInformationFormat="4"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCManagedResourceCompilerTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCResourceCompilerTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCPreLinkEventTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCLinkerTool"
|
||||||
|
AdditionalDependencies="tier0.lib ws2_32.lib"
|
||||||
|
OutputFile="$(OutDir)\geoip.ext.dll"
|
||||||
|
LinkIncremental="2"
|
||||||
|
GenerateDebugInformation="true"
|
||||||
|
SubSystem="2"
|
||||||
|
TargetMachine="1"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCALinkTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCManifestTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCXDCMakeTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCBscMakeTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCFxCopTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCAppVerifierTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCWebDeploymentTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCPostBuildEventTool"
|
||||||
|
/>
|
||||||
|
</Configuration>
|
||||||
|
<Configuration
|
||||||
|
Name="Release - Metamod|Win32"
|
||||||
|
OutputDirectory="$(SolutionDir)$(ConfigurationName)"
|
||||||
|
IntermediateDirectory="$(ConfigurationName)"
|
||||||
|
ConfigurationType="2"
|
||||||
|
CharacterSet="1"
|
||||||
|
WholeProgramOptimization="1"
|
||||||
|
>
|
||||||
|
<Tool
|
||||||
|
Name="VCPreBuildEventTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCCustomBuildTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCXMLDataGeneratorTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCWebServiceProxyGeneratorTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCMIDLTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCCLCompilerTool"
|
||||||
|
PreprocessorDefinitions="WIN32;NDEBUG;_WINDOWS;_USRDLL;geoip_EXPORTS;_CRT_SECURE_NO_DEPRECATE;SMEXT_CONF_METAMOD;_CRT_NONSTDC_NO_DEPRECATE"
|
||||||
|
RuntimeLibrary="0"
|
||||||
|
UsePrecompiledHeader="0"
|
||||||
|
WarningLevel="3"
|
||||||
|
Detect64BitPortabilityProblems="true"
|
||||||
|
DebugInformationFormat="3"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCManagedResourceCompilerTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCResourceCompilerTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCPreLinkEventTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCLinkerTool"
|
||||||
|
AdditionalDependencies="ws2_32.lib"
|
||||||
|
OutputFile="$(OutDir)\geoip.ext.dll"
|
||||||
|
LinkIncremental="1"
|
||||||
|
GenerateDebugInformation="true"
|
||||||
|
SubSystem="2"
|
||||||
|
OptimizeReferences="2"
|
||||||
|
EnableCOMDATFolding="2"
|
||||||
|
TargetMachine="1"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCALinkTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCManifestTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCXDCMakeTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCBscMakeTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCFxCopTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCAppVerifierTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCWebDeploymentTool"
|
||||||
|
/>
|
||||||
|
<Tool
|
||||||
|
Name="VCPostBuildEventTool"
|
||||||
|
/>
|
||||||
|
</Configuration>
|
||||||
|
</Configurations>
|
||||||
|
<References>
|
||||||
|
</References>
|
||||||
|
<Files>
|
||||||
|
<Filter
|
||||||
|
Name="Source Files"
|
||||||
|
Filter="cpp;c;cc;cxx;def;odl;idl;hpj;bat;asm;asmx"
|
||||||
|
UniqueIdentifier="{4FC737F1-C7A5-4376-A066-2A32D752A2FF}"
|
||||||
|
>
|
||||||
|
<File
|
||||||
|
RelativePath=".\extension.cpp"
|
||||||
|
>
|
||||||
|
</File>
|
||||||
|
<File
|
||||||
|
RelativePath=".\GeoIP.c"
|
||||||
|
>
|
||||||
|
</File>
|
||||||
|
<File
|
||||||
|
RelativePath=".\smsdk_ext.cpp"
|
||||||
|
>
|
||||||
|
</File>
|
||||||
|
</Filter>
|
||||||
|
<Filter
|
||||||
|
Name="Header Files"
|
||||||
|
Filter="h;hpp;hxx;hm;inl;inc;xsd"
|
||||||
|
UniqueIdentifier="{93995380-89BD-4b04-88EB-625FBE52EBFB}"
|
||||||
|
>
|
||||||
|
<File
|
||||||
|
RelativePath=".\extension.h"
|
||||||
|
>
|
||||||
|
</File>
|
||||||
|
<File
|
||||||
|
RelativePath=".\GeoIP.h"
|
||||||
|
>
|
||||||
|
</File>
|
||||||
|
<File
|
||||||
|
RelativePath=".\smsdk_config.h"
|
||||||
|
>
|
||||||
|
</File>
|
||||||
|
<File
|
||||||
|
RelativePath=".\smsdk_ext.h"
|
||||||
|
>
|
||||||
|
</File>
|
||||||
|
</Filter>
|
||||||
|
<Filter
|
||||||
|
Name="Resource Files"
|
||||||
|
Filter="rc;ico;cur;bmp;dlg;rc2;rct;bin;rgs;gif;jpg;jpeg;jpe;resx;tiff;tif;png;wav"
|
||||||
|
UniqueIdentifier="{67DA6AB6-F800-4c08-8B7A-83BB121AAD01}"
|
||||||
|
>
|
||||||
|
</Filter>
|
||||||
|
</Files>
|
||||||
|
<Globals>
|
||||||
|
</Globals>
|
||||||
|
</VisualStudioProject>
|
27
extensions/geoip/smsdk_config.h
Normal file
27
extensions/geoip/smsdk_config.h
Normal file
@ -0,0 +1,27 @@
|
|||||||
|
#ifndef _INCLUDE_SOURCEMOD_EXTENSION_CONFIG_H_
|
||||||
|
#define _INCLUDE_SOURCEMOD_EXTENSION_CONFIG_H_
|
||||||
|
|
||||||
|
/* Basic information exposed publically */
|
||||||
|
#define SMEXT_CONF_NAME "GeoIP"
|
||||||
|
#define SMEXT_CONF_DESCRIPTION "NO IDEA WHAT THIS MODULE DOES" //:TODO:
|
||||||
|
#define SMEXT_CONF_VERSION "0.0.0.0"
|
||||||
|
#define SMEXT_CONF_AUTHOR "AlliedModders"
|
||||||
|
#define SMEXT_CONF_URL "http://www.sourcemod.net/"
|
||||||
|
#define SMEXT_CONF_LOGTAG "GEOIP"
|
||||||
|
#define SMEXT_CONF_LICENSE "GPL"
|
||||||
|
#define SMEXT_CONF_DATESTRING __DATE__
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Exposes plugin's main interface.
|
||||||
|
*/
|
||||||
|
#define SMEXT_LINK(name) SDKExtension *g_pExtensionIface = name;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Sets whether or not this plugin required Metamod.
|
||||||
|
* NOTE: Uncomment to enable, comment to disable.
|
||||||
|
* NOTE: This is enabled automatically if a Metamod build is chosen in
|
||||||
|
* the Visual Studio project.
|
||||||
|
*/
|
||||||
|
//#define SMEXT_CONF_METAMOD
|
||||||
|
|
||||||
|
#endif //_INCLUDE_SOURCEMOD_EXTENSION_CONFIG_H_
|
277
extensions/geoip/smsdk_ext.cpp
Normal file
277
extensions/geoip/smsdk_ext.cpp
Normal file
@ -0,0 +1,277 @@
|
|||||||
|
#include <stdio.h>
|
||||||
|
#include "smsdk_ext.h"
|
||||||
|
|
||||||
|
IShareSys *g_pShareSys = NULL;
|
||||||
|
IExtension *myself = NULL;
|
||||||
|
IHandleSys *g_pHandleSys = NULL;
|
||||||
|
|
||||||
|
PLATFORM_EXTERN_C IExtensionInterface *GetSMExtAPI()
|
||||||
|
{
|
||||||
|
return g_pExtensionIface;
|
||||||
|
}
|
||||||
|
|
||||||
|
SDKExtension::SDKExtension()
|
||||||
|
{
|
||||||
|
#if defined SMEXT_CONF_METAMOD
|
||||||
|
m_SourceMMLoaded = false;
|
||||||
|
m_WeAreUnloaded = false;
|
||||||
|
m_WeGotPauseChange = false;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
bool SDKExtension::OnExtensionLoad(IExtension *me, IShareSys *sys, char *error, size_t err_max, bool late)
|
||||||
|
{
|
||||||
|
g_pShareSys = sys;
|
||||||
|
myself = me;
|
||||||
|
|
||||||
|
#if defined SMEXT_CONF_METAMOD
|
||||||
|
m_WeAreUnloaded = true;
|
||||||
|
|
||||||
|
if (!m_SourceMMLoaded)
|
||||||
|
{
|
||||||
|
if (error)
|
||||||
|
{
|
||||||
|
snprintf(error, err_max, "Metamod attach failed");
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
SM_GET_IFACE(HANDLESYSTEM, g_pHandleSys);
|
||||||
|
|
||||||
|
if (SDK_OnLoad(error, err_max, late))
|
||||||
|
{
|
||||||
|
#if defined SMEXT_CONF_METAMOD
|
||||||
|
m_WeAreUnloaded = true;
|
||||||
|
#endif
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool SDKExtension::IsMetamodExtension()
|
||||||
|
{
|
||||||
|
#if defined SMEXT_CONF_METAMOD
|
||||||
|
return true;
|
||||||
|
#else
|
||||||
|
return false;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
void SDKExtension::OnExtensionPauseChange(bool state)
|
||||||
|
{
|
||||||
|
#if defined SMEXT_CONF_METAMOD
|
||||||
|
m_WeGotPauseChange = true;
|
||||||
|
#endif
|
||||||
|
SDK_OnPauseChange(state);
|
||||||
|
}
|
||||||
|
|
||||||
|
void SDKExtension::OnExtensionsAllLoaded()
|
||||||
|
{
|
||||||
|
SDK_OnAllLoaded();
|
||||||
|
}
|
||||||
|
|
||||||
|
void SDKExtension::OnExtensionUnload()
|
||||||
|
{
|
||||||
|
#if defined SMEXT_CONF_METAMOD
|
||||||
|
m_WeAreUnloaded = true;
|
||||||
|
#endif
|
||||||
|
SDK_OnUnload();
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *SDKExtension::GetExtensionAuthor()
|
||||||
|
{
|
||||||
|
return SMEXT_CONF_AUTHOR;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *SDKExtension::GetExtensionDateString()
|
||||||
|
{
|
||||||
|
return SMEXT_CONF_DATESTRING;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *SDKExtension::GetExtensionDescription()
|
||||||
|
{
|
||||||
|
return SMEXT_CONF_DESCRIPTION;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *SDKExtension::GetExtensionVerString()
|
||||||
|
{
|
||||||
|
return SMEXT_CONF_VERSION;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *SDKExtension::GetExtensionName()
|
||||||
|
{
|
||||||
|
return SMEXT_CONF_NAME;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *SDKExtension::GetExtensionTag()
|
||||||
|
{
|
||||||
|
return SMEXT_CONF_LOGTAG;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *SDKExtension::GetExtensionURL()
|
||||||
|
{
|
||||||
|
return SMEXT_CONF_URL;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool SDKExtension::SDK_OnLoad(char *error, size_t err_max, bool late)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
void SDKExtension::SDK_OnUnload()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
void SDKExtension::SDK_OnPauseChange(bool paused)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
void SDKExtension::SDK_OnAllLoaded()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
#if defined SMEXT_CONF_METAMOD
|
||||||
|
|
||||||
|
PluginId g_PLID = 0;
|
||||||
|
ISmmPlugin *g_PLAPI = NULL;
|
||||||
|
SourceHook::ISourceHook *g_SHPtr = NULL;
|
||||||
|
ISmmAPI *g_SMAPI = NULL;
|
||||||
|
|
||||||
|
IVEngineServer *engine = NULL;
|
||||||
|
IServerGameDLL *gamedll = NULL;
|
||||||
|
|
||||||
|
SMM_API void *PL_EXPOSURE(const char *name, int *code)
|
||||||
|
{
|
||||||
|
if (name && !strcmp(name, PLAPI_NAME))
|
||||||
|
{
|
||||||
|
if (code)
|
||||||
|
{
|
||||||
|
*code = IFACE_OK;
|
||||||
|
}
|
||||||
|
return static_cast<void *>(g_pExtensionIface);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (code)
|
||||||
|
{
|
||||||
|
*code = IFACE_FAILED;
|
||||||
|
}
|
||||||
|
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool SDKExtension::Load(PluginId id, ISmmAPI *ismm, char *error, size_t maxlen, bool late)
|
||||||
|
{
|
||||||
|
PLUGIN_SAVEVARS();
|
||||||
|
|
||||||
|
GET_V_IFACE_ANY(serverFactory, gamedll, IServerGameDLL, INTERFACEVERSION_SERVERGAMEDLL);
|
||||||
|
GET_V_IFACE_CURRENT(engineFactory, engine, IVEngineServer, INTERFACEVERSION_VENGINESERVER);
|
||||||
|
|
||||||
|
m_SourceMMLoaded = true;
|
||||||
|
|
||||||
|
return SDK_OnMetamodLoad(error, maxlen, late);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool SDKExtension::Unload(char *error, size_t maxlen)
|
||||||
|
{
|
||||||
|
if (!m_WeAreUnloaded)
|
||||||
|
{
|
||||||
|
if (error)
|
||||||
|
{
|
||||||
|
snprintf(error, maxlen, "This extension must be unloaded by SourceMod.");
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
return SDK_OnMetamodUnload(error, maxlen);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool SDKExtension::Pause(char *error, size_t maxlen)
|
||||||
|
{
|
||||||
|
if (!m_WeGotPauseChange)
|
||||||
|
{
|
||||||
|
if (error)
|
||||||
|
{
|
||||||
|
snprintf(error, maxlen, "This extension must be paused by SourceMod.");
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
m_WeGotPauseChange = false;
|
||||||
|
|
||||||
|
return SDK_OnMetamodPauseChange(true, error, maxlen);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool SDKExtension::Unpause(char *error, size_t maxlen)
|
||||||
|
{
|
||||||
|
if (!m_WeGotPauseChange)
|
||||||
|
{
|
||||||
|
if (error)
|
||||||
|
{
|
||||||
|
snprintf(error, maxlen, "This extension must be unpaused by SourceMod.");
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
m_WeGotPauseChange = false;
|
||||||
|
|
||||||
|
return SDK_OnMetamodPauseChange(false, error, maxlen);
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *SDKExtension::GetAuthor()
|
||||||
|
{
|
||||||
|
return GetExtensionAuthor();
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *SDKExtension::GetDate()
|
||||||
|
{
|
||||||
|
return GetExtensionDateString();
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *SDKExtension::GetDescription()
|
||||||
|
{
|
||||||
|
return GetExtensionDescription();
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *SDKExtension::GetLicense()
|
||||||
|
{
|
||||||
|
return SMEXT_CONF_LICENSE;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *SDKExtension::GetLogTag()
|
||||||
|
{
|
||||||
|
return GetExtensionTag();
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *SDKExtension::GetName()
|
||||||
|
{
|
||||||
|
return GetExtensionName();
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *SDKExtension::GetURL()
|
||||||
|
{
|
||||||
|
return GetExtensionURL();
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *SDKExtension::GetVersion()
|
||||||
|
{
|
||||||
|
return GetExtensionVerString();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool SDKExtension::SDK_OnMetamodLoad(char *error, size_t err_max, bool late)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool SDKExtension::SDK_OnMetamodUnload(char *error, size_t err_max)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool SDKExtension::SDK_OnMetamodPauseChange(bool paused, char *error, size_t err_max)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif
|
140
extensions/geoip/smsdk_ext.h
Normal file
140
extensions/geoip/smsdk_ext.h
Normal file
@ -0,0 +1,140 @@
|
|||||||
|
#ifndef _INCLUDE_SOURCEMOD_EXTENSION_BASESDK_H_
|
||||||
|
#define _INCLUDE_SOURCEMOD_EXTENSION_BASESDK_H_
|
||||||
|
|
||||||
|
#include "smsdk_config.h"
|
||||||
|
#include <IExtensionSys.h>
|
||||||
|
#include <IHandleSys.h>
|
||||||
|
#include <sp_vm_api.h>
|
||||||
|
#include <sm_platform.h>
|
||||||
|
|
||||||
|
#if defined SMEXT_CONF_METAMOD
|
||||||
|
#include <ISmmPlugin.h>
|
||||||
|
#include <eiface.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
|
using namespace SourceMod;
|
||||||
|
using namespace SourcePawn;
|
||||||
|
|
||||||
|
class SDKExtension :
|
||||||
|
#if defined SMEXT_CONF_METAMOD
|
||||||
|
public ISmmPlugin,
|
||||||
|
#endif
|
||||||
|
public IExtensionInterface
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
SDKExtension();
|
||||||
|
public:
|
||||||
|
/**
|
||||||
|
* @brief This is called after the initial loading sequence has been processed.
|
||||||
|
*
|
||||||
|
* @param error Error message buffer.
|
||||||
|
* @param err_max Size of error message buffer.
|
||||||
|
* @param late Whether or not the module was loaded after map load.
|
||||||
|
* @return True to succeed loading, false to fail.
|
||||||
|
*/
|
||||||
|
virtual bool SDK_OnLoad(char *error, size_t err_max, bool late);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief This is called right before the extension is unloaded.
|
||||||
|
*/
|
||||||
|
virtual void SDK_OnUnload();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief This is called once all known extensions have been loaded.
|
||||||
|
*/
|
||||||
|
virtual void SDK_OnAllLoaded();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Called when the pause state is changed.
|
||||||
|
*/
|
||||||
|
virtual void SDK_OnPauseChange(bool paused);
|
||||||
|
|
||||||
|
#if defined SMEXT_CONF_METAMOD
|
||||||
|
/**
|
||||||
|
* @brief Called when Metamod is attached, before the extension version is called.
|
||||||
|
*
|
||||||
|
* @param error Error buffer.
|
||||||
|
* @param err_max Maximum size of error buffer.
|
||||||
|
* @param late Whether or not Metamod considers this a late load.
|
||||||
|
* @return True to succeed, false to fail.
|
||||||
|
*/
|
||||||
|
virtual bool SDK_OnMetamodLoad(char *error, size_t err_max, bool late);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Called when Metamod is detaching, after the extension version is called.
|
||||||
|
* NOTE: By default this is blocked unless sent from SourceMod.
|
||||||
|
*
|
||||||
|
* @param error Error buffer.
|
||||||
|
* @param err_max Maximum size of error buffer.
|
||||||
|
* @return True to succeed, false to fail.
|
||||||
|
*/
|
||||||
|
virtual bool SDK_OnMetamodUnload(char *error, size_t err_max);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Called when Metamod's pause state is changing.
|
||||||
|
* NOTE: By default this is blocked unless sent from SourceMod.
|
||||||
|
*
|
||||||
|
* @param paused Pause state being set.
|
||||||
|
* @param error Error buffer.
|
||||||
|
* @param err_max Maximum size of error buffer.
|
||||||
|
* @return True to succeed, false to fail.
|
||||||
|
*/
|
||||||
|
virtual bool SDK_OnMetamodPauseChange(bool paused, char *error, size_t err_max);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
public: //IExtensionInterface
|
||||||
|
virtual bool OnExtensionLoad(IExtension *me, IShareSys *sys, char *error, size_t err_max, bool late);
|
||||||
|
virtual void OnExtensionUnload();
|
||||||
|
virtual void OnExtensionsAllLoaded();
|
||||||
|
virtual bool IsMetamodExtension();
|
||||||
|
virtual void OnExtensionPauseChange(bool state);
|
||||||
|
virtual const char *GetExtensionName();
|
||||||
|
virtual const char *GetExtensionURL();
|
||||||
|
virtual const char *GetExtensionTag();
|
||||||
|
virtual const char *GetExtensionAuthor();
|
||||||
|
virtual const char *GetExtensionVerString();
|
||||||
|
virtual const char *GetExtensionDescription();
|
||||||
|
virtual const char *GetExtensionDateString();
|
||||||
|
#if defined SMEXT_CONF_METAMOD
|
||||||
|
public: //ISmmPlugin
|
||||||
|
virtual bool Load(PluginId id, ISmmAPI *ismm, char *error, size_t maxlength, bool late);
|
||||||
|
virtual const char *GetAuthor();
|
||||||
|
virtual const char *GetName();
|
||||||
|
virtual const char *GetDescription();
|
||||||
|
virtual const char *GetURL();
|
||||||
|
virtual const char *GetLicense();
|
||||||
|
virtual const char *GetVersion();
|
||||||
|
virtual const char *GetDate();
|
||||||
|
virtual const char *GetLogTag();
|
||||||
|
virtual bool Unload(char *error, size_t maxlen);
|
||||||
|
virtual bool Pause(char *error, size_t maxlen);
|
||||||
|
virtual bool Unpause(char *error, size_t maxlen);
|
||||||
|
private:
|
||||||
|
bool m_SourceMMLoaded;
|
||||||
|
bool m_WeAreUnloaded;
|
||||||
|
bool m_WeGotPauseChange;
|
||||||
|
#endif
|
||||||
|
};
|
||||||
|
|
||||||
|
extern SDKExtension *g_pExtensionIface;
|
||||||
|
|
||||||
|
extern IShareSys *g_pShareSys;
|
||||||
|
extern IExtension *myself;
|
||||||
|
extern IHandleSys *g_pHandleSys;
|
||||||
|
|
||||||
|
#if defined SMEXT_CONF_METAMOD
|
||||||
|
PLUGIN_GLOBALVARS();
|
||||||
|
extern IVEngineServer *engine;
|
||||||
|
extern IServerGameDLL *gamedll;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#define SM_MKIFACE(name) SMINTERFACE_##name##_NAME, SMINTERFACE_##name##_VERSION
|
||||||
|
#define SM_GET_IFACE(prefix,addr) \
|
||||||
|
if (!g_pShareSys->RequestInterface(SM_MKIFACE(prefix), myself, (SMInterface **)&addr)) { \
|
||||||
|
if (error) { \
|
||||||
|
snprintf(error, err_max, "Could not find interface: %s", SMINTERFACE_##prefix##_NAME); \
|
||||||
|
} \
|
||||||
|
return false; \
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif //_INCLUDE_SOURCEMOD_EXTENSION_BASESDK_H_
|
Loading…
Reference in New Issue
Block a user