# Maxmind GeoIP2 API bindings
# Copyright (c) 2013 Timo Teräs
# GPLv2 applies

# Use from your VCL by adding to your vcl_recv():
#  call maxminddb_lookup;

# You also need to modify the cc_command parameter to include
# Maxmind database library (-lmaxminddb). In Alpine Linux this is
# done by uncomminting the relevant VARNISHD_PLUGIN_CFLAGS line in
# /etc/conf.d/varnishd.

C{

#include <maxminddb.h>
#include <string.h>

static MMDB_s mmdb;

static void __attribute__((constructor)) __geoip_on_load(void)
{
	if (MMDB_open("/var/lib/libmaxminddb/GeoLite2-City.mmdb", MMDB_MODE_MMAP, &mmdb) != MMDB_SUCCESS)
		mmdb.filename = NULL;
}

static void __attribute__((destructor)) __geoip_on_unload(void)
{
	if (mmdb.filename != NULL)
		MMDB_close(&mmdb);
}

static inline void __geoip_set_header(struct sess *sp, const char *hdr, MMDB_lookup_result_s *res, char **path)
{
	MMDB_entry_data_s entry;
	char buf[64];
	int s;

	s = MMDB_aget_value(&res->entry, &entry, path);
	if (s != MMDB_SUCCESS || !entry.has_data)
		return;

	if (entry.type != MMDB_DATA_TYPE_UTF8_STRING)
		return;

	if (entry.data_size+1 >= sizeof(buf))
		return;

	memcpy(buf, entry.utf8_string, entry.data_size);
	buf[entry.data_size] = 0;

	VRT_SetHdr(sp, HDR_REQ, hdr, buf, vrt_magic_string_end);
}

static void __geoip_set_headers(struct sess *sp)
{
	static const char *continent_path[] = { "continent", "code", NULL };
	static const char *country_path[] = { "country", "iso_code", NULL };
	MMDB_lookup_result_s res;
	int mmdb_error;

	if (mmdb.filename == NULL)
		return;

	res = MMDB_lookup_sockaddr(&mmdb, (struct sockaddr*) VRT_r_client_ip(sp), &mmdb_error);
	if (mmdb_error == MMDB_SUCCESS) {
		__geoip_set_header(sp, "\022X-GeoIP-Continent:", &res, continent_path);
		__geoip_set_header(sp, "\020X-GeoIP-Country:", &res, country_path);
	}
}

}C

sub maxminddb_lookup {
	C{ __geoip_set_headers(sp); }C
	return(lookup);
}