00b67f09dd
Also known as ISC bind. This import adds utilities such as host(1), dig(1), and nslookup(1), as well as many other tools and libraries. Change-Id: I035ca46e64f1965d57019e773f4ff0ef035e4aa3
151 lines
3.9 KiB
C
151 lines
3.9 KiB
C
/* $NetBSD: geoip.c,v 1.1.1.3 2014/12/10 03:34:24 christos Exp $ */
|
|
|
|
/*
|
|
* Copyright (C) 2013, 2014 Internet Systems Consortium, Inc. ("ISC")
|
|
*
|
|
* Permission to use, copy, modify, and/or distribute this software for any
|
|
* purpose with or without fee is hereby granted, provided that the above
|
|
* copyright notice and this permission notice appear in all copies.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
|
|
* REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
|
|
* AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
|
|
* INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
|
|
* LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
|
|
* OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
|
|
* PERFORMANCE OF THIS SOFTWARE.
|
|
*/
|
|
|
|
/*! \file */
|
|
|
|
#include <config.h>
|
|
|
|
#include <isc/util.h>
|
|
|
|
#include <named/log.h>
|
|
#include <named/geoip.h>
|
|
|
|
#include <dns/geoip.h>
|
|
|
|
#ifdef HAVE_GEOIP
|
|
static dns_geoip_databases_t geoip_table = {
|
|
NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL
|
|
};
|
|
|
|
static void
|
|
init_geoip_db(GeoIP **dbp, GeoIPDBTypes edition, GeoIPDBTypes fallback,
|
|
GeoIPOptions method, const char *name)
|
|
{
|
|
char *info;
|
|
GeoIP *db;
|
|
|
|
REQUIRE(dbp != NULL);
|
|
|
|
db = *dbp;
|
|
|
|
if (db != NULL) {
|
|
GeoIP_delete(db);
|
|
db = *dbp = NULL;
|
|
}
|
|
|
|
if (! GeoIP_db_avail(edition)) {
|
|
isc_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL,
|
|
NS_LOGMODULE_SERVER, ISC_LOG_INFO,
|
|
"GeoIP %s (type %d) DB not available", name, edition);
|
|
goto fail;
|
|
}
|
|
|
|
isc_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL,
|
|
NS_LOGMODULE_SERVER, ISC_LOG_INFO,
|
|
"initializing GeoIP %s (type %d) DB", name, edition);
|
|
|
|
db = GeoIP_open_type(edition, method);
|
|
if (db == NULL) {
|
|
isc_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL,
|
|
NS_LOGMODULE_SERVER, ISC_LOG_ERROR,
|
|
"failed to initialize GeoIP %s (type %d) DB%s",
|
|
name, edition, fallback == 0
|
|
? "geoip matches using this database will fail" : "");
|
|
goto fail;
|
|
}
|
|
|
|
info = GeoIP_database_info(db);
|
|
if (info != NULL)
|
|
isc_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL,
|
|
NS_LOGMODULE_SERVER, ISC_LOG_INFO,
|
|
"%s", info);
|
|
|
|
*dbp = db;
|
|
return;
|
|
fail:
|
|
if (fallback != 0)
|
|
init_geoip_db(dbp, fallback, 0, method, name);
|
|
|
|
}
|
|
#endif /* HAVE_GEOIP */
|
|
|
|
void
|
|
ns_geoip_init(void) {
|
|
#ifndef HAVE_GEOIP
|
|
return;
|
|
#else
|
|
GeoIP_cleanup();
|
|
if (ns_g_geoip == NULL)
|
|
ns_g_geoip = &geoip_table;
|
|
#endif
|
|
}
|
|
|
|
void
|
|
ns_geoip_load(char *dir) {
|
|
#ifndef HAVE_GEOIP
|
|
|
|
UNUSED(dir);
|
|
|
|
return;
|
|
#else
|
|
GeoIPOptions method;
|
|
|
|
#ifdef _WIN32
|
|
method = GEOIP_STANDARD;
|
|
#else
|
|
method = GEOIP_MMAP_CACHE;
|
|
#endif
|
|
|
|
ns_geoip_init();
|
|
if (dir != NULL) {
|
|
isc_log_write(ns_g_lctx, NS_LOGCATEGORY_GENERAL,
|
|
NS_LOGMODULE_SERVER, ISC_LOG_INFO,
|
|
"using \"%s\" as GeoIP directory", dir);
|
|
GeoIP_setup_custom_directory(dir);
|
|
}
|
|
|
|
init_geoip_db(&ns_g_geoip->country_v4, GEOIP_COUNTRY_EDITION, 0,
|
|
method, "Country (IPv4)");
|
|
#ifdef HAVE_GEOIP_V6
|
|
init_geoip_db(&ns_g_geoip->country_v6, GEOIP_COUNTRY_EDITION_V6, 0,
|
|
method, "Country (IPv6)");
|
|
#endif
|
|
|
|
init_geoip_db(&ns_g_geoip->city_v4, GEOIP_CITY_EDITION_REV1,
|
|
GEOIP_CITY_EDITION_REV0, method, "City (IPv4)");
|
|
#if defined(HAVE_GEOIP_V6) && defined(HAVE_GEOIP_CITY_V6)
|
|
init_geoip_db(&ns_g_geoip->city_v6, GEOIP_CITY_EDITION_REV1_V6,
|
|
GEOIP_CITY_EDITION_REV0_V6, method, "City (IPv6)");
|
|
#endif
|
|
|
|
init_geoip_db(&ns_g_geoip->region, GEOIP_REGION_EDITION_REV1,
|
|
GEOIP_REGION_EDITION_REV0, method, "Region");
|
|
|
|
init_geoip_db(&ns_g_geoip->isp, GEOIP_ISP_EDITION, 0,
|
|
method, "ISP");
|
|
init_geoip_db(&ns_g_geoip->org, GEOIP_ORG_EDITION, 0,
|
|
method, "Org");
|
|
init_geoip_db(&ns_g_geoip->as, GEOIP_ASNUM_EDITION, 0,
|
|
method, "AS");
|
|
init_geoip_db(&ns_g_geoip->domain, GEOIP_DOMAIN_EDITION, 0,
|
|
method, "Domain");
|
|
init_geoip_db(&ns_g_geoip->netspeed, GEOIP_NETSPEED_EDITION, 0,
|
|
method, "NetSpeed");
|
|
#endif /* HAVE_GEOIP */
|
|
}
|