/* Prosody IM

--

-- This project is MIT/X11 licensed. Please see the

-- COPYING file in the source package for more information.

--

-- Copyright (C) 2012 Paul Aurich

-- Copyright (C) 2013 Matthew Wild

-- Copyright (C) 2013 Florian Zeitz

--

*/

#define _GNU_SOURCE

#include <stddef.h>

#include <string.h>

#include <errno.h>

#ifndef _WIN32

#include <sys/ioctl.h>

#include <sys/types.h>

#include <sys/socket.h>

#include <net/if.h>

#include <ifaddrs.h>

#include <arpa/inet.h>

#include <netinet/in.h>

#endif

#include <lua.h>

#include <lauxlib.h>

#if (LUA_VERSION_NUM == 501)

#define luaL_setfuncs(L, R, N) luaL_register(L, NULL, R)

#endif

/* Enumerate all locally configured IP addresses */

const char *const type_strings[] = {

	"both",

	"ipv4",

	"ipv6",

	NULL

};

static int lc_local_addresses(lua_State *L) {

#ifndef _WIN32

	/* Link-local IPv4 addresses; see RFC 3927 and RFC 5735 */

	const long ip4_linklocal = htonl(0xa9fe0000); /* 169.254.0.0 */

	const long ip4_mask      = htonl(0xffff0000);

	struct ifaddrs *addr = NULL, *a;

#endif

	int n = 1;

	int type = luaL_checkoption(L, 1, "both", type_strings);

	const char link_local = lua_toboolean(L, 2); /* defaults to 0 (false) */

	const char ipv4 = (type == 0 || type == 1);

	const char ipv6 = (type == 0 || type == 2);

#ifndef _WIN32

	if(getifaddrs(&addr) < 0) {

		lua_pushnil(L);

		lua_pushfstring(L, "getifaddrs failed (%d): %s", errno,

		                strerror(errno));

		return 2;

	}

#endif

	lua_newtable(L);

#ifndef _WIN32

	for(a = addr; a; a = a->ifa_next) {

		int family;

		char ipaddr[INET6_ADDRSTRLEN];

		const char *tmp = NULL;

		if(a->ifa_addr == NULL || a->ifa_flags & IFF_LOOPBACK) {

			continue;

		}

		family = a->ifa_addr->sa_family;

		if(ipv4 && family == AF_INET) {

			struct sockaddr_in *sa = (struct sockaddr_in *)a->ifa_addr;

			if(!link_local && ((sa->sin_addr.s_addr & ip4_mask) == ip4_linklocal)) {

				continue;

			}

			tmp = inet_ntop(family, &sa->sin_addr, ipaddr, sizeof(ipaddr));

		} else if(ipv6 && family == AF_INET6) {

			struct sockaddr_in6 *sa = (struct sockaddr_in6 *)a->ifa_addr;

			if(!link_local && IN6_IS_ADDR_LINKLOCAL(&sa->sin6_addr)) {

				continue;

			}

			if(IN6_IS_ADDR_V4MAPPED(&sa->sin6_addr) || IN6_IS_ADDR_V4COMPAT(&sa->sin6_addr)) {

				continue;

			}

			tmp = inet_ntop(family, &sa->sin6_addr, ipaddr, sizeof(ipaddr));

		}

		if(tmp != NULL) {

			lua_pushstring(L, tmp);

			lua_rawseti(L, -2, n++);

		}

		/* TODO: Error reporting? */

	}

	freeifaddrs(addr);

#else

	if(ipv4) {

		lua_pushstring(L, "0.0.0.0");

		lua_rawseti(L, -2, n++);

	}

	if(ipv6) {

		lua_pushstring(L, "::");

		lua_rawseti(L, -2, n++);

	}

#endif

	return 1;

}

static int lc_pton(lua_State *L) {

	char buf[16];

	const char *ipaddr = luaL_checkstring(L, 1);

	int errno_ = 0;

	int family = strchr(ipaddr, ':') ? AF_INET6 : AF_INET;

	switch(inet_pton(family, ipaddr, &buf)) {

		case 1:

			lua_pushlstring(L, buf, family == AF_INET6 ? 16 : 4);

			return 1;

		case -1:

			errno_ = errno;

			lua_pushnil(L);

			lua_pushstring(L, strerror(errno_));

			lua_pushinteger(L, errno_);

			return 3;

		default:

		case 0:

			lua_pushnil(L);

			lua_pushstring(L, strerror(EINVAL));

			lua_pushinteger(L, EINVAL);

			return 3;

	}

}

static int lc_ntop(lua_State *L) {

	char buf[INET6_ADDRSTRLEN];

	int family;

	int errno_;

	size_t l;

	const char *ipaddr = luaL_checklstring(L, 1, &l);

	if(l == 16) {

		family = AF_INET6;

	}

	else if(l == 4) {

		family = AF_INET;

	}

	else {

		lua_pushnil(L);

		lua_pushstring(L, strerror(EAFNOSUPPORT));

		lua_pushinteger(L, EAFNOSUPPORT);

		return 3;

	}

	if(!inet_ntop(family, ipaddr, buf, INET6_ADDRSTRLEN))

	{

		errno_ = errno;

		lua_pushnil(L);

		lua_pushstring(L, strerror(errno_));

		lua_pushinteger(L, errno_);

		return 3;

	}

	lua_pushstring(L, (const char *)(&buf));

	return 1;

}

int luaopen_util_net(lua_State *L) {

#if (LUA_VERSION_NUM > 501)

	luaL_checkversion(L);

#endif

	luaL_Reg exports[] = {

		{ "local_addresses", lc_local_addresses },

		{ "pton", lc_pton },

		{ "ntop", lc_ntop },

		{ NULL, NULL }

	};

	lua_createtable(L, 0, 1);

	luaL_setfuncs(L, exports, 0);

	return 1;

}