/*
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Copyright 2016 - 2017 Red Hat, Inc.
 */

#include "nm-default.h"

#include <sys/mount.h>
#include <sched.h>
#include <sys/wait.h>
#include <fcntl.h>
#include <linux/if_tun.h>

#include "test-common.h"

#define SIGNAL_DATA_FMT "'%s-%s' ifindex %d%s%s%s (%d times received)"
#define SIGNAL_DATA_ARG(data) (data)->name, nm_platform_signal_change_type_to_string ((data)->change_type), (data)->ifindex, (data)->ifname ? " ifname '" : "", (data)->ifname ?: "", (data)->ifname ? "'" : "", (data)->received_count

int NMTSTP_ENV1_IFINDEX = -1;
int NMTSTP_ENV1_EX = -1;

/*****************************************************************************/

void
nmtstp_setup_platform (void)
{
	g_assert (_nmtstp_setup_platform_func);
	_nmtstp_setup_platform_func ();
}

gboolean
nmtstp_is_root_test (void)
{
	g_assert (_nmtstp_setup_platform_func);
	return _nmtstp_setup_platform_func == nm_linux_platform_setup;
}

gboolean
nmtstp_is_sysfs_writable (void)
{
	return    !nmtstp_is_root_test ()
	       || (access ("/sys/devices", W_OK) == 0);
}

static void
_init_platform (NMPlatform **platform, gboolean external_command)
{
	g_assert (platform);
	if (!*platform)
		*platform = NM_PLATFORM_GET;
	g_assert (NM_IS_PLATFORM (*platform));

	if (external_command)
		g_assert (NM_IS_LINUX_PLATFORM (*platform));
}

/*****************************************************************************/

static GArray *
_ipx_address_get_all (NMPlatform *self, int ifindex, NMPObjectType obj_type)
{
	NMPLookup lookup;

	g_assert (NM_IS_PLATFORM (self));
	g_assert (ifindex > 0);
	g_assert (NM_IN_SET (obj_type, NMP_OBJECT_TYPE_IP4_ADDRESS, NMP_OBJECT_TYPE_IP6_ADDRESS));
	nmp_lookup_init_object (&lookup,
	                        obj_type,
	                        ifindex);
	return nmp_cache_lookup_to_array (nm_platform_lookup (self, &lookup),
	                                  obj_type,
	                                  FALSE /*addresses are always visible. */);
}

GArray *
nmtstp_platform_ip4_address_get_all (NMPlatform *self, int ifindex)
{
	return _ipx_address_get_all (self, ifindex, NMP_OBJECT_TYPE_IP4_ADDRESS);
}

GArray *
nmtstp_platform_ip6_address_get_all (NMPlatform *self, int ifindex)
{
	return _ipx_address_get_all (self, ifindex, NMP_OBJECT_TYPE_IP6_ADDRESS);
}

/*****************************************************************************/

gboolean
nmtstp_platform_ip4_route_delete (NMPlatform *platform, int ifindex, in_addr_t network, guint8 plen, guint32 metric)
{
	NMDedupMultiIter iter;

	nm_platform_process_events (platform);

	nm_dedup_multi_iter_for_each (&iter,
	                              nm_platform_lookup_object (platform,
	                                                         NMP_OBJECT_TYPE_IP4_ROUTE,
	                                                         ifindex)) {
		const NMPlatformIP4Route *r = NMP_OBJECT_CAST_IP4_ROUTE (iter.current->obj);

		if (   r->ifindex != ifindex
		    || r->network != network
		    || r->plen != plen
		    || r->metric != metric) {
			continue;
		}

		return nm_platform_object_delete (platform, NMP_OBJECT_UP_CAST (r));
	}

	return TRUE;
}

gboolean
nmtstp_platform_ip6_route_delete (NMPlatform *platform, int ifindex, struct in6_addr network, guint8 plen, guint32 metric)
{
	NMDedupMultiIter iter;

	nm_platform_process_events (platform);

	nm_dedup_multi_iter_for_each (&iter,
	                              nm_platform_lookup_object (platform,
	                                                         NMP_OBJECT_TYPE_IP6_ROUTE,
	                                                         ifindex)) {
		const NMPlatformIP6Route *r = NMP_OBJECT_CAST_IP6_ROUTE (iter.current->obj);

		if (   r->ifindex != ifindex
		    || !IN6_ARE_ADDR_EQUAL (&r->network, &network)
		    || r->plen != plen
		    || r->metric != metric) {
			continue;
		}

		return nm_platform_object_delete (platform, NMP_OBJECT_UP_CAST (r));
	}

	return TRUE;
}

/*****************************************************************************/

SignalData *
add_signal_full (const char *name, NMPlatformSignalChangeType change_type, GCallback callback, int ifindex, const char *ifname)
{
	SignalData *data = g_new0 (SignalData, 1);

	data->name = name;
	data->change_type = change_type;
	data->received_count = 0;
	data->handler_id = g_signal_connect (NM_PLATFORM_GET, name, callback, data);
	data->ifindex = ifindex;
	data->ifname = ifname;

	g_assert (data->handler_id > 0);

	return data;
}

void
_accept_signal (const char *file, int line, const char *func, SignalData *data)
{
	_LOGD ("NMPlatformSignalAssert: %s:%d, %s(): Accepting signal one time: "SIGNAL_DATA_FMT, file, line, func, SIGNAL_DATA_ARG (data));
	if (data->received_count != 1)
		g_error ("NMPlatformSignalAssert: %s:%d, %s(): failure to accept signal one time: "SIGNAL_DATA_FMT, file, line, func, SIGNAL_DATA_ARG (data));
	data->received_count = 0;
}

void
_accept_signals (const char *file, int line, const char *func, SignalData *data, int min, int max)
{
	_LOGD ("NMPlatformSignalAssert: %s:%d, %s(): Accepting signal [%d,%d] times: "SIGNAL_DATA_FMT, file, line, func, min, max, SIGNAL_DATA_ARG (data));
	if (data->received_count < min || data->received_count > max)
		g_error ("NMPlatformSignalAssert: %s:%d, %s(): failure to accept signal [%d,%d] times: "SIGNAL_DATA_FMT, file, line, func, min, max, SIGNAL_DATA_ARG (data));
	data->received_count = 0;
}

void
_ensure_no_signal (const char *file, int line, const char *func, SignalData *data)
{
	_LOGD ("NMPlatformSignalAssert: %s:%d, %s(): Accepting signal 0 times: "SIGNAL_DATA_FMT, file, line, func, SIGNAL_DATA_ARG (data));
	if (data->received_count > 0)
		g_error ("NMPlatformSignalAssert: %s:%d, %s(): failure to accept signal 0 times: "SIGNAL_DATA_FMT, file, line, func, SIGNAL_DATA_ARG (data));
}

void
_accept_or_wait_signal (const char *file, int line, const char *func, SignalData *data)
{
	_LOGD ("NMPlatformSignalAssert: %s:%d, %s(): accept-or-wait signal: "SIGNAL_DATA_FMT, file, line, func, SIGNAL_DATA_ARG (data));
	if (data->received_count == 0) {
		data->loop = g_main_loop_new (NULL, FALSE);
		g_main_loop_run (data->loop);
		g_clear_pointer (&data->loop, g_main_loop_unref);
	}

	_accept_signal (file, line, func, data);
}

void
_wait_signal (const char *file, int line, const char *func, SignalData *data)
{
	_LOGD ("NMPlatformSignalAssert: %s:%d, %s(): wait signal: "SIGNAL_DATA_FMT, file, line, func, SIGNAL_DATA_ARG (data));
	if (data->received_count)
		g_error ("NMPlatformSignalAssert: %s:%d, %s(): failure to wait for signal: "SIGNAL_DATA_FMT, file, line, func, SIGNAL_DATA_ARG (data));

	data->loop = g_main_loop_new (NULL, FALSE);
	g_main_loop_run (data->loop);
	g_clear_pointer (&data->loop, g_main_loop_unref);

	_accept_signal (file, line, func, data);
}

void
_free_signal (const char *file, int line, const char *func, SignalData *data)
{
	_LOGD ("NMPlatformSignalAssert: %s:%d, %s(): free signal: "SIGNAL_DATA_FMT, file, line, func, SIGNAL_DATA_ARG (data));
	if (data->received_count != 0)
		g_error ("NMPlatformSignalAssert: %s:%d, %s(): failure to free non-accepted signal: "SIGNAL_DATA_FMT, file, line, func, SIGNAL_DATA_ARG (data));

	g_signal_handler_disconnect (NM_PLATFORM_GET, data->handler_id);
	g_free (data);
}

void
link_callback (NMPlatform *platform, int obj_type_i, int ifindex, NMPlatformLink *received, int change_type_i, SignalData *data)
{
	const NMPObjectType obj_type = obj_type_i;
	const NMPlatformSignalChangeType change_type = change_type_i;
	NMPLookup lookup;
	NMDedupMultiIter iter;
	const NMPlatformLink *cached;

	g_assert_cmpint (obj_type, ==, NMP_OBJECT_TYPE_LINK);
	g_assert (received);
	g_assert_cmpint (received->ifindex, ==, ifindex);
	g_assert (data && data->name);
	g_assert_cmpstr (data->name, ==, NM_PLATFORM_SIGNAL_LINK_CHANGED);

	if (data->ifindex && data->ifindex != received->ifindex)
		return;
	if (data->ifname && g_strcmp0 (data->ifname, nm_platform_link_get_name (NM_PLATFORM_GET, ifindex)) != 0)
		return;
	if (change_type != data->change_type)
		return;

	if (data->loop) {
		_LOGD ("Quitting main loop.");
		g_main_loop_quit (data->loop);
	}

	data->received_count++;
	_LOGD ("Received signal '%s-%s' ifindex %d ifname '%s' %dth time.", data->name, nm_platform_signal_change_type_to_string (data->change_type), ifindex, received->name, data->received_count);

	if (change_type == NM_PLATFORM_SIGNAL_REMOVED)
		g_assert (!nm_platform_link_get_name (NM_PLATFORM_GET, ifindex));
	else
		g_assert (nm_platform_link_get_name (NM_PLATFORM_GET, ifindex));

	/* Check the data */
	g_assert (received->ifindex > 0);

	nmp_lookup_init_obj_type (&lookup, NMP_OBJECT_TYPE_LINK);
	nmp_cache_iter_for_each_link (&iter,
	                              nm_platform_lookup (platform, &lookup),
	                              &cached) {
		if (!nmp_object_is_visible (NMP_OBJECT_UP_CAST (cached)))
			continue;
		if (cached->ifindex == received->ifindex) {
			g_assert_cmpint (nm_platform_link_cmp (cached, received), ==, 0);
			g_assert (!memcmp (cached, received, sizeof (*cached)));
			if (data->change_type == NM_PLATFORM_SIGNAL_REMOVED)
				g_error ("Deleted link still found in the local cache.");
			return;
		}
	}

	if (data->change_type != NM_PLATFORM_SIGNAL_REMOVED)
		g_error ("Added/changed link not found in the local cache.");
}

/*****************************************************************************/

static const NMPlatformIP4Route *
_ip4_route_get (NMPlatform *platform,
                int ifindex,
                guint32 network,
                int plen,
                guint32 metric,
                guint8 tos,
                guint *out_c_exists)
{
	NMDedupMultiIter iter;
	NMPLookup lookup;
	const NMPObject *o = NULL;
	guint c;
	const NMPlatformIP4Route *r = NULL;

	_init_platform (&platform, FALSE);

	nmp_lookup_init_ip4_route_by_weak_id (&lookup,
	                                      network,
	                                      plen,
	                                      metric,
	                                      tos);

	c = 0;
	nmp_cache_iter_for_each (&iter,
	                         nm_platform_lookup (platform, &lookup),
	                         &o) {
		if (   NMP_OBJECT_CAST_IP4_ROUTE (o)->ifindex != ifindex
		    && ifindex > 0)
			continue;
		if (!r)
			r = NMP_OBJECT_CAST_IP4_ROUTE (o);
		c++;
	}

	NM_SET_OUT (out_c_exists, c);
	return r;
}

const NMPlatformIP4Route *
_nmtstp_assert_ip4_route_exists (const char *file,
                                 guint line,
                                 const char *func,
                                 NMPlatform *platform,
                                 int c_exists,
                                 const char *ifname,
                                 guint32 network,
                                 int plen,
                                 guint32 metric,
                                 guint8 tos)
{
	int ifindex;
	guint c;
	const NMPlatformIP4Route *r = NULL;

	_init_platform (&platform, FALSE);

	ifindex = -1;
	if (ifname) {
		ifindex = nm_platform_link_get_ifindex (platform, ifname);
		g_assert (ifindex > 0);
	}

	r = _ip4_route_get (platform,
	                    ifindex,
	                    network,
	                    plen,
	                    metric,
	                    tos,
	                    &c);

	if (c != c_exists && c_exists != -1) {
		char sbuf[NM_UTILS_INET_ADDRSTRLEN];

		g_error ("[%s:%u] %s(): The ip4 route %s/%d metric %u tos %u shall exist %u times, but platform has it %u times",
		         file, line, func,
		         nm_utils_inet4_ntop (network, sbuf),
		         plen,
		         metric,
		         tos,
		         c_exists,
		         c);
	}

	return r;
}

const NMPlatformIP4Route *
nmtstp_ip4_route_get (NMPlatform *platform,
                      int ifindex,
                      guint32 network,
                      int plen,
                      guint32 metric,
                      guint8 tos)
{
	return _ip4_route_get (platform,
	                       ifindex,
	                       network,
	                       plen,
	                       metric,
	                       tos,
	                       NULL);
}

/*****************************************************************************/

static const NMPlatformIP6Route *
_ip6_route_get (NMPlatform *platform,
                int ifindex,
                const struct in6_addr *network,
                guint plen,
                guint32 metric,
                const struct in6_addr *src,
                guint8 src_plen,
                guint *out_c_exists)
{
	NMDedupMultiIter iter;
	NMPLookup lookup;
	const NMPObject *o = NULL;
	guint c;
	const NMPlatformIP6Route *r = NULL;

	_init_platform (&platform, FALSE);

	nmp_lookup_init_ip6_route_by_weak_id (&lookup,
	                                      network,
	                                      plen,
	                                      metric,
	                                      src,
	                                      src_plen);

	c = 0;
	nmp_cache_iter_for_each (&iter,
	                         nm_platform_lookup (platform, &lookup),
	                         &o) {
		if (   NMP_OBJECT_CAST_IP6_ROUTE (o)->ifindex != ifindex
		    && ifindex > 0)
			continue;
		if (!r)
			r = NMP_OBJECT_CAST_IP6_ROUTE (o);
		c++;
	}

	NM_SET_OUT (out_c_exists, c);
	return r;
}

const NMPlatformIP6Route *
_nmtstp_assert_ip6_route_exists (const char *file,
                                 guint line,
                                 const char *func,
                                 NMPlatform *platform,
                                 int c_exists,
                                 const char *ifname,
                                 const struct in6_addr *network,
                                 guint plen,
                                 guint32 metric,
                                 const struct in6_addr *src,
                                 guint8 src_plen)
{
	int ifindex;
	guint c;
	const NMPlatformIP6Route *r = NULL;

	_init_platform (&platform, FALSE);

	ifindex = -1;
	if (ifname) {
		ifindex = nm_platform_link_get_ifindex (platform, ifname);
		g_assert (ifindex > 0);
	}

	r = _ip6_route_get (platform,
	                    ifindex,
	                    network,
	                    plen,
	                    metric,
	                    src,
	                    src_plen,
	                    &c);

	if (c != c_exists && c_exists != -1) {
		char s_src[NM_UTILS_INET_ADDRSTRLEN];
		char s_network[NM_UTILS_INET_ADDRSTRLEN];

		g_error ("[%s:%u] %s(): The ip6 route %s/%d metric %u src %s/%d shall exist %u times, but platform has it %u times",
		         file, line, func,
		         nm_utils_inet6_ntop (network, s_network),
		         plen,
		         metric,
		         nm_utils_inet6_ntop (src, s_src),
		         src_plen,
		         c_exists,
		         c);
	}

	return r;
}

const NMPlatformIP6Route *
nmtstp_ip6_route_get (NMPlatform *platform,
                      int ifindex,
                      const struct in6_addr *network,
                      guint plen,
                      guint32 metric,
                      const struct in6_addr *src,
                      guint8 src_plen)
{
	return _ip6_route_get (platform,
	                       ifindex,
	                       network,
	                       plen,
	                       metric,
	                       src,
	                       src_plen,
	                       NULL);
}

/*****************************************************************************/

int
nmtstp_run_command (const char *format, ...)
{
	int result;
	gs_free char *command = NULL;
	va_list ap;

	va_start (ap, format);
	command = g_strdup_vprintf (format, ap);
	va_end (ap);

	_LOGD ("Running command: %s", command);
	result = system (command);
	_LOGD ("Command finished: result=%d", result);

	return result;
}

/*****************************************************************************/

typedef struct {
	GMainLoop *loop;
	guint signal_counts;
	guint id;
} WaitForSignalData;

static void
_wait_for_signal_cb (NMPlatform *platform,
                     int obj_type_i,
                     int ifindex,
                     NMPlatformLink *plink,
                     int change_type_i,
                     gpointer user_data)
{
	WaitForSignalData *data = user_data;

	data->signal_counts++;
	nm_clear_g_source (&data->id);
	g_main_loop_quit (data->loop);
}

static gboolean
_wait_for_signal_timeout (gpointer user_data)
{
	WaitForSignalData *data = user_data;

	g_assert (data->id);
	data->id = 0;
	g_main_loop_quit (data->loop);
	return G_SOURCE_REMOVE;
}

guint
nmtstp_wait_for_signal (NMPlatform *platform, gint64 timeout_ms)
{
	WaitForSignalData data = { 0 };
	gulong id_link, id_ip4_address, id_ip6_address, id_ip4_route, id_ip6_route;

	_init_platform (&platform, FALSE);

	data.loop = g_main_loop_new (NULL, FALSE);

	id_link        = g_signal_connect (platform, NM_PLATFORM_SIGNAL_LINK_CHANGED, G_CALLBACK (_wait_for_signal_cb), &data);
	id_ip4_address = g_signal_connect (platform, NM_PLATFORM_SIGNAL_IP4_ADDRESS_CHANGED, G_CALLBACK (_wait_for_signal_cb), &data);
	id_ip6_address = g_signal_connect (platform, NM_PLATFORM_SIGNAL_IP6_ADDRESS_CHANGED, G_CALLBACK (_wait_for_signal_cb), &data);
	id_ip4_route   = g_signal_connect (platform, NM_PLATFORM_SIGNAL_IP4_ROUTE_CHANGED, G_CALLBACK (_wait_for_signal_cb), &data);
	id_ip6_route   = g_signal_connect (platform, NM_PLATFORM_SIGNAL_IP6_ROUTE_CHANGED, G_CALLBACK (_wait_for_signal_cb), &data);

	/* if timeout_ms is negative, it means the wait-time already expired.
	 * Maybe, we should do nothing and return right away, without even
	 * processing events from platform. However, that inconsistency (of not
	 * processing events from mainloop) is inconvenient.
	 *
	 * It's better that on the return of nmtstp_wait_for_signal(), we always
	 * have no events pending. So, a negative timeout is treated the same as
	 * a zero timeout: we check whether there are any events pending in platform,
	 * and quite the mainloop immediately afterwards. But we always check. */

	data.id = g_timeout_add (CLAMP (timeout_ms, 0, G_MAXUINT32),
	                         _wait_for_signal_timeout, &data);

	g_main_loop_run (data.loop);

	g_assert (!data.id);
	g_assert (nm_clear_g_signal_handler (platform, &id_link));
	g_assert (nm_clear_g_signal_handler (platform, &id_ip4_address));
	g_assert (nm_clear_g_signal_handler (platform, &id_ip6_address));
	g_assert (nm_clear_g_signal_handler (platform, &id_ip4_route));
	g_assert (nm_clear_g_signal_handler (platform, &id_ip6_route));

	g_clear_pointer (&data.loop, g_main_loop_unref);

	/* return the number of signals, or 0 if timeout was reached .*/
	return data.signal_counts;
}

guint
nmtstp_wait_for_signal_until (NMPlatform *platform, gint64 until_ms)
{
	gint64 now;
	guint signal_counts;

	while (TRUE) {
		now = nm_utils_get_monotonic_timestamp_ms ();

		if (until_ms < now)
			return 0;

		signal_counts = nmtstp_wait_for_signal (platform, until_ms - now);
		if (signal_counts)
			return signal_counts;
	}
}

const NMPlatformLink *
nmtstp_wait_for_link (NMPlatform *platform, const char *ifname, NMLinkType expected_link_type, gint64 timeout_ms)
{
	return nmtstp_wait_for_link_until (platform, ifname, expected_link_type,
	                                   timeout_ms
	                                     ? nm_utils_get_monotonic_timestamp_ms () + timeout_ms
	                                     : 0);
}

const NMPlatformLink *
nmtstp_wait_for_link_until (NMPlatform *platform, const char *ifname, NMLinkType expected_link_type, gint64 until_ms)
{
	const NMPlatformLink *plink;
	gint64 now;
	gboolean waited_once = FALSE;

	_init_platform (&platform, FALSE);

	while (TRUE) {
		now = nm_utils_get_monotonic_timestamp_ms ();

		plink = nm_platform_link_get_by_ifname (platform, ifname);
		if (   plink
		    && (expected_link_type == NM_LINK_TYPE_NONE || plink->type == expected_link_type))
			return plink;

		if (until_ms == 0) {
			/* don't wait, don't even poll the socket. */
			return NULL;
		}

		if (   waited_once
		    && until_ms < now) {
			/* timeout reached (+ we already waited for a signal at least once). */
			return NULL;
		}

		waited_once = TRUE;
		/* regardless of whether timeout is already reached, we poll the netlink
		 * socket a bit. */
		nmtstp_wait_for_signal (platform, until_ms - now);
	}
}

/*****************************************************************************/

int
nmtstp_run_command_check_external_global (void)
{
	if (!nmtstp_is_root_test ())
		return FALSE;
	switch (nmtst_get_rand_uint32 () % 3) {
	case 0:
		return -1;
	case 1:
		return FALSE;
	default:
		return TRUE;
	}
}

gboolean
nmtstp_run_command_check_external (int external_command)
{
	if (external_command != -1) {
		g_assert (NM_IN_SET (external_command, FALSE, TRUE));
		g_assert (!external_command || nmtstp_is_root_test ());
		return !!external_command;
	}
	if (!nmtstp_is_root_test ())
		return FALSE;
	return (nmtst_get_rand_uint32 () % 2) == 0;
}

/*****************************************************************************/

#define CHECK_LIFETIME_MAX_DIFF    2

gboolean
nmtstp_ip_address_check_lifetime (const NMPlatformIPAddress *addr,
                                  gint64 now,
                                  guint32 expected_lifetime,
                                  guint32 expected_preferred)
{
	gint64 offset;
	int i;

	g_assert (addr);

	if (now == -1)
		now = nm_utils_get_monotonic_timestamp_s ();
	g_assert (now > 0);

	g_assert (expected_preferred <= expected_lifetime);

	if (   expected_lifetime == NM_PLATFORM_LIFETIME_PERMANENT
	    && expected_preferred == NM_PLATFORM_LIFETIME_PERMANENT) {
		return    addr->timestamp == 0
		       && addr->lifetime == NM_PLATFORM_LIFETIME_PERMANENT
		       && addr->preferred == NM_PLATFORM_LIFETIME_PERMANENT;
	}

	if (addr->timestamp == 0)
		return FALSE;

	offset = (gint64) now - addr->timestamp;

	for (i = 0; i < 2; i++) {
		guint32 lft = i ? expected_lifetime : expected_preferred;
		guint32 adr = i ? addr->lifetime : addr->preferred;

		if (lft == NM_PLATFORM_LIFETIME_PERMANENT) {
			if (adr != NM_PLATFORM_LIFETIME_PERMANENT)
				return FALSE;
		} else {
			if (   adr - offset <= lft - CHECK_LIFETIME_MAX_DIFF
			    || adr - offset >= lft + CHECK_LIFETIME_MAX_DIFF)
				return FALSE;
		}
	}
	return TRUE;
}

void
nmtstp_ip_address_assert_lifetime (const NMPlatformIPAddress *addr,
                                   gint64 now,
                                   guint32 expected_lifetime,
                                   guint32 expected_preferred)
{
	gint64 n = now;
	gint64 offset;
	int i;

	g_assert (addr);

	if (now == -1)
		now = nm_utils_get_monotonic_timestamp_s ();
	g_assert (now > 0);

	g_assert (expected_preferred <= expected_lifetime);

	if (   expected_lifetime == NM_PLATFORM_LIFETIME_PERMANENT
	    && expected_preferred == NM_PLATFORM_LIFETIME_PERMANENT) {
		g_assert_cmpint (addr->timestamp, ==, 0);
		g_assert_cmpint (addr->lifetime, ==, NM_PLATFORM_LIFETIME_PERMANENT);
		g_assert_cmpint (addr->preferred, ==, NM_PLATFORM_LIFETIME_PERMANENT);
		return;
	}

	g_assert_cmpint (addr->timestamp, >, 0);
	g_assert_cmpint (addr->timestamp, <=, now);

	offset = (gint64) now - addr->timestamp;
	g_assert_cmpint (offset, >=, 0);

	for (i = 0; i < 2; i++) {
		guint32 lft = i ? expected_lifetime : expected_preferred;
		guint32 adr = i ? addr->lifetime : addr->preferred;

		if (lft == NM_PLATFORM_LIFETIME_PERMANENT)
			g_assert_cmpint (adr, ==, NM_PLATFORM_LIFETIME_PERMANENT);
		else {
			g_assert_cmpint (adr - offset, <=, lft + CHECK_LIFETIME_MAX_DIFF);
			g_assert_cmpint (adr - offset, >=, lft - CHECK_LIFETIME_MAX_DIFF);
		}
	}

	g_assert (nmtstp_ip_address_check_lifetime (addr, n, expected_lifetime, expected_preferred));
}

/*****************************************************************************/

static void
_ip_address_add (NMPlatform *platform,
                 gboolean external_command,
                 gboolean is_v4,
                 int ifindex,
                 const NMIPAddr *address,
                 int plen,
                 const NMIPAddr *peer_address,
                 guint32 lifetime,
                 guint32 preferred,
                 guint32 flags,
                 const char *label)
{
	gint64 end_time;

	external_command = nmtstp_run_command_check_external (external_command);

	_init_platform (&platform, external_command);

	if (external_command) {
		const char *ifname;
		gs_free char *s_valid = NULL;
		gs_free char *s_preferred = NULL;
		gs_free char *s_label = NULL;
		char b1[NM_UTILS_INET_ADDRSTRLEN];
		char b2[NM_UTILS_INET_ADDRSTRLEN];

		ifname = nm_platform_link_get_name (platform, ifindex);
		g_assert (ifname);

		if (lifetime != NM_PLATFORM_LIFETIME_PERMANENT)
			s_valid = g_strdup_printf (" valid_lft %d", lifetime);
		if (preferred != NM_PLATFORM_LIFETIME_PERMANENT)
			s_preferred = g_strdup_printf (" preferred_lft %d", preferred);
		if (label)
			s_label = g_strdup_printf ("%s:%s", ifname, label);

		if (is_v4) {
			char s_peer[NM_UTILS_INET_ADDRSTRLEN + 50];

			g_assert (flags == 0);

			if (   peer_address->addr4 != address->addr4
			    || nmtst_get_rand_uint32 () % 2) {
				/* If the peer is the same as the local address, we can omit it. The result should be identical */
				nm_sprintf_buf (s_peer, " peer %s", nm_utils_inet4_ntop (peer_address->addr4, b2));
			} else
				s_peer[0] = '\0';

			nmtstp_run_command_check ("ip address change %s%s/%d dev %s%s%s%s",
			                          nm_utils_inet4_ntop (address->addr4, b1),
			                          s_peer,
			                          plen,
			                          ifname,
			                          s_valid ?: "",
			                          s_preferred ?: "",
			                          s_label ?: "");
		} else {
			g_assert (label == NULL);

			/* flags not implemented (yet) */
			g_assert (flags == 0);
			nmtstp_run_command_check ("ip address change %s%s%s/%d dev %s%s%s%s",
			                          nm_utils_inet6_ntop (&address->addr6, b1),
			                          !IN6_IS_ADDR_UNSPECIFIED (&peer_address->addr6) ? " peer " : "",
			                          !IN6_IS_ADDR_UNSPECIFIED (&peer_address->addr6) ? nm_utils_inet6_ntop (&peer_address->addr6, b2) : "",
			                          plen,
			                          ifname,
			                          s_valid ?: "",
			                          s_preferred ?: "",
			                          s_label ?: "");
		}
	} else {
		gboolean success;

		if (is_v4) {
			success = nm_platform_ip4_address_add (platform,
			                                       ifindex,
			                                       address->addr4,
			                                       plen,
			                                       peer_address->addr4,
			                                       lifetime,
			                                       preferred,
			                                       flags,
			                                       label);
		} else {
			g_assert (label == NULL);
			success = nm_platform_ip6_address_add (platform,
			                                       ifindex,
			                                       address->addr6,
			                                       plen,
			                                       peer_address->addr6,
			                                       lifetime,
			                                       preferred,
			                                       flags);
		}
		g_assert (success);
	}

	/* Let's wait until we see the address. */
	end_time = nm_utils_get_monotonic_timestamp_ms () + 500;
	do {

		if (external_command)
			nm_platform_process_events (platform);

		/* let's wait until we see the address as we added it. */
		if (is_v4) {
			const NMPlatformIP4Address *a;

			g_assert (flags == 0);
			a = nm_platform_ip4_address_get (platform, ifindex, address->addr4, plen, peer_address->addr4);
			if (   a
			    && a->peer_address == peer_address->addr4
			    && nmtstp_ip_address_check_lifetime ((NMPlatformIPAddress*) a, -1, lifetime, preferred)
			    && strcmp (a->label, label ?: "") == 0)
				break;
		} else {
			const NMPlatformIP6Address *a;

			g_assert (label == NULL);
			g_assert (flags == 0);

			a = nm_platform_ip6_address_get (platform, ifindex, address->addr6);
			if (   a
			    && !memcmp (nm_platform_ip6_address_get_peer (a),
			                (IN6_IS_ADDR_UNSPECIFIED (&peer_address->addr6) || IN6_ARE_ADDR_EQUAL (&address->addr6, &peer_address->addr6))
			                    ? &address->addr6 : &peer_address->addr6,
			                sizeof (struct in6_addr))
			    && nmtstp_ip_address_check_lifetime ((NMPlatformIPAddress*) a, -1, lifetime, preferred))
				break;
		}

		/* for internal command, we expect not to reach this line.*/
		g_assert (external_command);

		nmtstp_assert_wait_for_signal_until (platform, end_time);
	} while (TRUE);
}

void
nmtstp_ip4_address_add (NMPlatform *platform,
                        gboolean external_command,
                        int ifindex,
                        in_addr_t address,
                        int plen,
                        in_addr_t peer_address,
                        guint32 lifetime,
                        guint32 preferred,
                        guint32 flags,
                        const char *label)
{
	_ip_address_add (platform,
	                 external_command,
	                 TRUE,
	                 ifindex,
	                 (NMIPAddr *) &address,
	                 plen,
	                 (NMIPAddr *) &peer_address,
	                 lifetime,
	                 preferred,
	                 flags,
	                 label);
}

void
nmtstp_ip6_address_add (NMPlatform *platform,
                        gboolean external_command,
                        int ifindex,
                        struct in6_addr address,
                        int plen,
                        struct in6_addr peer_address,
                        guint32 lifetime,
                        guint32 preferred,
                        guint32 flags)
{
	_ip_address_add (platform,
	                 external_command,
	                 FALSE,
	                 ifindex,
	                 (NMIPAddr *) &address,
	                 plen,
	                 (NMIPAddr *) &peer_address,
	                 lifetime,
	                 preferred,
	                 flags,
	                 NULL);
}

void nmtstp_ip4_route_add (NMPlatform *platform,
                           int ifindex,
                           NMIPConfigSource source,
                           in_addr_t network,
                           guint8 plen,
                           in_addr_t gateway,
                           in_addr_t pref_src,
                           guint32 metric,
                           guint32 mss)
{
	NMPlatformIP4Route route = { };

	route.ifindex = ifindex;
	route.rt_source = source;
	route.network = network;
	route.plen = plen;
	route.gateway = gateway;
	route.pref_src = pref_src;
	route.metric = metric;
	route.mss = mss;

	g_assert (NMTST_NM_ERR_SUCCESS (nm_platform_ip4_route_add (platform, NMP_NLM_FLAG_REPLACE, &route)));
}

void nmtstp_ip6_route_add (NMPlatform *platform,
                           int ifindex,
                           NMIPConfigSource source,
                           struct in6_addr network,
                           guint8 plen,
                           struct in6_addr gateway,
                           struct in6_addr pref_src,
                           guint32 metric,
                           guint32 mss)
{
	NMPlatformIP6Route route = { };

	route.ifindex = ifindex;
	route.rt_source = source;
	route.network = network;
	route.plen = plen;
	route.gateway = gateway;
	route.pref_src = pref_src;
	route.metric = metric;
	route.mss = mss;

	g_assert (NMTST_NM_ERR_SUCCESS (nm_platform_ip6_route_add (platform, NMP_NLM_FLAG_REPLACE, &route)));
}

/*****************************************************************************/

static void
_ip_address_del (NMPlatform *platform,
                 gboolean external_command,
                 gboolean is_v4,
                 int ifindex,
                 const NMIPAddr *address,
                 int plen,
                 const NMIPAddr *peer_address)
{
	gint64 end_time;

	external_command = nmtstp_run_command_check_external (external_command);

	_init_platform (&platform, external_command);

	if (external_command) {
		const char *ifname;
		char b1[NM_UTILS_INET_ADDRSTRLEN];
		char b2[NM_UTILS_INET_ADDRSTRLEN];
		int success;
		gboolean had_address;

		ifname = nm_platform_link_get_name (platform, ifindex);
		g_assert (ifname);

		/* let's wait until we see the address as we added it. */
		if (is_v4)
			had_address = !!nm_platform_ip4_address_get (platform, ifindex, address->addr4, plen, peer_address->addr4);
		else
			had_address = !!nm_platform_ip6_address_get (platform, ifindex, address->addr6);

		if (is_v4) {
			success = nmtstp_run_command ("ip address delete %s%s%s/%d dev %s",
			                              nm_utils_inet4_ntop (address->addr4, b1),
			                              peer_address->addr4 != address->addr4 ? " peer " : "",
			                              peer_address->addr4 != address->addr4 ? nm_utils_inet4_ntop (peer_address->addr4, b2) : "",
			                              plen,
			                              ifname);
		} else {
			g_assert (!peer_address);
			success = nmtstp_run_command ("ip address delete %s/%d dev %s",
			                              nm_utils_inet6_ntop (&address->addr6, b1),
			                              plen,
			                              ifname);
		}
		g_assert (success == 0 || !had_address);
	} else {
		gboolean success;

		if (is_v4) {
			success = nm_platform_ip4_address_delete (platform,
			                                          ifindex,
			                                          address->addr4,
			                                          plen,
			                                          peer_address->addr4);
		} else {
			g_assert (!peer_address);
			success = nm_platform_ip6_address_delete (platform,
			                                          ifindex,
			                                          address->addr6,
			                                          plen);
		}
		g_assert (success);
	}

	/* Let's wait until we get the result */
	end_time = nm_utils_get_monotonic_timestamp_ms () + 250;
	do {
		if (external_command)
			nm_platform_process_events (platform);

		/* let's wait until we see the address as we added it. */
		if (is_v4) {
			const NMPlatformIP4Address *a;

			a = nm_platform_ip4_address_get (platform, ifindex, address->addr4, plen, peer_address->addr4);
			if (!a)
				break;
		} else {
			const NMPlatformIP6Address *a;

			a = nm_platform_ip6_address_get (platform, ifindex, address->addr6);
			if (!a)
				break;
		}

		/* for internal command, we expect not to reach this line.*/
		g_assert (external_command);

		nmtstp_assert_wait_for_signal_until (platform, end_time);
	} while (TRUE);
}

void
nmtstp_ip4_address_del (NMPlatform *platform,
                        gboolean external_command,
                        int ifindex,
                        in_addr_t address,
                        int plen,
                        in_addr_t peer_address)
{
	_ip_address_del (platform,
	                 external_command,
	                 TRUE,
	                 ifindex,
	                 (NMIPAddr *) &address,
	                 plen,
	                 (NMIPAddr *) &peer_address);
}

void
nmtstp_ip6_address_del (NMPlatform *platform,
                        gboolean external_command,
                        int ifindex,
                        struct in6_addr address,
                        int plen)
{
	_ip_address_del (platform,
	                 external_command,
	                 FALSE,
	                 ifindex,
	                 (NMIPAddr *) &address,
	                 plen,
	                 NULL);
}

/*****************************************************************************/

#define _assert_pllink(platform, success, pllink, name, type) \
	G_STMT_START { \
		const NMPlatformLink *_pllink = (pllink); \
		\
		if ((success)) { \
			g_assert (_pllink); \
			g_assert (_pllink == nmtstp_link_get_typed (platform, _pllink->ifindex, (name), (type))); \
		} else { \
			g_assert (!_pllink); \
			g_assert (!nmtstp_link_get (platform, 0, (name))); \
		} \
	} G_STMT_END

const NMPlatformLink *
nmtstp_link_veth_add (NMPlatform *platform,
                      gboolean external_command,
                      const char *name,
                      const char *peer)
{
	const NMPlatformLink *pllink = NULL;
	gboolean success;

	g_assert (nm_utils_is_valid_iface_name (name, NULL));

	external_command = nmtstp_run_command_check_external (external_command);

	_init_platform (&platform, external_command);

	if (external_command) {
		success = !nmtstp_run_command ("ip link add dev %s type veth peer name %s",
		                                name, peer);
		if (success) {
			pllink = nmtstp_assert_wait_for_link (platform, name, NM_LINK_TYPE_VETH, 100);
			nmtstp_assert_wait_for_link (platform, peer, NM_LINK_TYPE_VETH, 10);
		}
	} else
		success = NMTST_NM_ERR_SUCCESS (nm_platform_link_veth_add (platform, name, peer, &pllink));

	g_assert (success);
	_assert_pllink (platform, success, pllink, name, NM_LINK_TYPE_VETH);
	return pllink;
}

const NMPlatformLink *
nmtstp_link_dummy_add (NMPlatform *platform,
                       gboolean external_command,
                       const char *name)
{
	const NMPlatformLink *pllink = NULL;
	gboolean success;

	g_assert (nm_utils_is_valid_iface_name (name, NULL));

	external_command = nmtstp_run_command_check_external (external_command);

	_init_platform (&platform, external_command);

	if (external_command) {
		success = !nmtstp_run_command ("ip link add %s type dummy",
		                                name);
		if (success)
			pllink = nmtstp_assert_wait_for_link (platform, name, NM_LINK_TYPE_DUMMY, 100);
	} else
		success = NMTST_NM_ERR_SUCCESS (nm_platform_link_dummy_add (platform, name, &pllink));

	g_assert (success);
	_assert_pllink (platform, success, pllink, name, NM_LINK_TYPE_DUMMY);
	return pllink;
}

const NMPlatformLink *
nmtstp_link_gre_add (NMPlatform *platform,
                     gboolean external_command,
                     const char *name,
                     const NMPlatformLnkGre *lnk)
{
	const NMPlatformLink *pllink = NULL;
	gboolean success;
	char b1[INET_ADDRSTRLEN];
	char b2[INET_ADDRSTRLEN];
	NMLinkType link_type;

	g_assert (nm_utils_is_valid_iface_name (name, NULL));

	external_command = nmtstp_run_command_check_external (external_command);
	link_type = lnk->is_tap ? NM_LINK_TYPE_GRETAP : NM_LINK_TYPE_GRE;

	_init_platform (&platform, external_command);

	if (external_command) {
		gs_free char *dev = NULL;
		char *obj, *type;

		if (lnk->parent_ifindex)
			dev = g_strdup_printf ("dev %s", nm_platform_link_get_name (platform, lnk->parent_ifindex));

		obj = lnk->is_tap ? "link" : "tunnel";
		type = lnk->is_tap ? "type gretap" : "mode gre";

		success = !nmtstp_run_command ("ip %s add %s %s %s local %s remote %s ttl %u tos %02x %s",
		                                obj,
		                                name,
		                                type,
		                                dev ?: "",
		                                nm_utils_inet4_ntop (lnk->local, b1),
		                                nm_utils_inet4_ntop (lnk->remote, b2),
		                                lnk->ttl,
		                                lnk->tos,
		                                lnk->path_mtu_discovery ? "pmtudisc" : "nopmtudisc");
		if (success)
			pllink = nmtstp_assert_wait_for_link (platform, name, link_type, 100);
	} else
		success = NMTST_NM_ERR_SUCCESS (nm_platform_link_gre_add (platform, name, lnk, &pllink));

	_assert_pllink (platform, success, pllink, name, link_type);

	return pllink;
}

const NMPlatformLink *
nmtstp_link_ip6tnl_add (NMPlatform *platform,
                        gboolean external_command,
                        const char *name,
                        const NMPlatformLnkIp6Tnl *lnk)
{
	const NMPlatformLink *pllink = NULL;
	gboolean success;
	char b1[NM_UTILS_INET_ADDRSTRLEN];
	char b2[NM_UTILS_INET_ADDRSTRLEN];
	char encap[20];
	char tclass[20];
	gboolean encap_ignore;
	gboolean tclass_inherit;

	g_assert (nm_utils_is_valid_iface_name (name, NULL));
	g_assert (!lnk->is_gre);

	external_command = nmtstp_run_command_check_external (external_command);

	_init_platform (&platform, external_command);

	if (external_command) {
		gs_free char *dev = NULL;
		const char *mode;

		if (lnk->parent_ifindex)
			dev = g_strdup_printf ("dev %s", nm_platform_link_get_name (platform, lnk->parent_ifindex));

		switch (lnk->proto) {
		case IPPROTO_IPIP:
			mode = "ipip6";
			break;
		case IPPROTO_IPV6:
			mode = "ip6ip6";
			break;
		default:
			g_assert_not_reached ();
		}

		encap_ignore = NM_FLAGS_HAS (lnk->flags, IP6_TNL_F_IGN_ENCAP_LIMIT);
		tclass_inherit = NM_FLAGS_HAS (lnk->flags, IP6_TNL_F_USE_ORIG_TCLASS);

		success = !nmtstp_run_command ("ip -6 tunnel add %s mode %s %s local %s remote %s ttl %u tclass %s encaplimit %s flowlabel %x",
		                                name,
		                                mode,
		                                dev,
		                                nm_utils_inet6_ntop (&lnk->local, b1),
		                                nm_utils_inet6_ntop (&lnk->remote, b2),
		                                lnk->ttl,
		                                tclass_inherit ? "inherit" : nm_sprintf_buf (tclass, "%02x", lnk->tclass),
		                                encap_ignore ? "none" : nm_sprintf_buf (encap, "%u", lnk->encap_limit),
		                                lnk->flow_label);
		if (success)
			pllink = nmtstp_assert_wait_for_link (platform, name, NM_LINK_TYPE_IP6TNL, 100);
	} else
		success = NMTST_NM_ERR_SUCCESS (nm_platform_link_ip6tnl_add (platform, name, lnk, &pllink));

	_assert_pllink (platform, success, pllink, name, NM_LINK_TYPE_IP6TNL);

	return pllink;
}

const NMPlatformLink *
nmtstp_link_ip6gre_add (NMPlatform *platform,
                        gboolean external_command,
                        const char *name,
                        const NMPlatformLnkIp6Tnl *lnk)
{
	const NMPlatformLink *pllink = NULL;
	gboolean success;
	char b1[NM_UTILS_INET_ADDRSTRLEN];
	char b2[NM_UTILS_INET_ADDRSTRLEN];
	char tclass[20];
	gboolean tclass_inherit;

	g_assert (nm_utils_is_valid_iface_name (name, NULL));
	g_assert (lnk->is_gre);

	external_command = nmtstp_run_command_check_external (external_command);

	_init_platform (&platform, external_command);

	if (external_command) {
		gs_free char *dev = NULL;

		if (lnk->parent_ifindex)
			dev = g_strdup_printf ("dev %s", nm_platform_link_get_name (platform, lnk->parent_ifindex));

		tclass_inherit = NM_FLAGS_HAS (lnk->flags, IP6_TNL_F_USE_ORIG_TCLASS);

		success = !nmtstp_run_command ("ip link add %s type %s %s local %s remote %s ttl %u tclass %s flowlabel %x",
		                                name,
		                                lnk->is_tap ? "ip6gretap" : "ip6gre",
		                                dev,
		                                nm_utils_inet6_ntop (&lnk->local, b1),
		                                nm_utils_inet6_ntop (&lnk->remote, b2),
		                                lnk->ttl,
		                                tclass_inherit ? "inherit" : nm_sprintf_buf (tclass, "%02x", lnk->tclass),
		                                lnk->flow_label);
		if (success) {
			pllink = nmtstp_assert_wait_for_link (platform,
			                                      name,
			                                      lnk->is_tap ? NM_LINK_TYPE_IP6GRETAP : NM_LINK_TYPE_IP6GRE,
			                                      100);
		}
	} else
		success = NMTST_NM_ERR_SUCCESS (nm_platform_link_ip6gre_add (platform, name, lnk, &pllink));

	_assert_pllink (platform, success, pllink, name, lnk->is_tap ? NM_LINK_TYPE_IP6GRETAP : NM_LINK_TYPE_IP6GRE);

	return pllink;
}

const NMPlatformLink *
nmtstp_link_ipip_add (NMPlatform *platform,
                      gboolean external_command,
                      const char *name,
                      const NMPlatformLnkIpIp *lnk)
{
	const NMPlatformLink *pllink = NULL;
	gboolean success;
	char b1[INET_ADDRSTRLEN];
	char b2[INET_ADDRSTRLEN];

	g_assert (nm_utils_is_valid_iface_name (name, NULL));

	external_command = nmtstp_run_command_check_external (external_command);

	_init_platform (&platform, external_command);

	if (external_command) {
		gs_free char *dev = NULL;

		if (lnk->parent_ifindex)
			dev = g_strdup_printf ("dev %s", nm_platform_link_get_name (platform, lnk->parent_ifindex));

		success = !nmtstp_run_command ("ip tunnel add %s mode ipip %s local %s remote %s ttl %u tos %02x %s",
		                                name,
		                                dev,
		                                nm_utils_inet4_ntop (lnk->local, b1),
		                                nm_utils_inet4_ntop (lnk->remote, b2),
		                                lnk->ttl,
		                                lnk->tos,
		                                lnk->path_mtu_discovery ? "pmtudisc" : "nopmtudisc");
		if (success)
			pllink = nmtstp_assert_wait_for_link (platform, name, NM_LINK_TYPE_IPIP, 100);
	} else
		success = NMTST_NM_ERR_SUCCESS (nm_platform_link_ipip_add (platform, name, lnk, &pllink));

	_assert_pllink (platform, success, pllink, name, NM_LINK_TYPE_IPIP);

	return pllink;
}

const NMPlatformLink *
nmtstp_link_macvlan_add (NMPlatform *platform,
                         gboolean external_command,
                         const char *name,
                         int parent,
                         const NMPlatformLnkMacvlan *lnk)
{
	const NMPlatformLink *pllink = NULL;
	gboolean success;
	NMLinkType link_type;

	g_assert (nm_utils_is_valid_iface_name (name, NULL));

	external_command = nmtstp_run_command_check_external (external_command);

	_init_platform (&platform, external_command);

	link_type = lnk->tap ? NM_LINK_TYPE_MACVTAP : NM_LINK_TYPE_MACVLAN;

	if (external_command) {
		const char *dev;
		char *modes[] = {
				[MACVLAN_MODE_BRIDGE]   = "bridge",
				[MACVLAN_MODE_VEPA]     = "vepa",
				[MACVLAN_MODE_PRIVATE]  = "private",
				[MACVLAN_MODE_PASSTHRU] = "passthru",
		};

		dev = nm_platform_link_get_name (platform, parent);
		g_assert (dev);
		g_assert_cmpint (lnk->mode, <, G_N_ELEMENTS (modes));

		success = !nmtstp_run_command ("ip link add name %s link %s type %s mode %s %s",
		                                name,
		                                dev,
		                                lnk->tap ? "macvtap" : "macvlan",
		                                modes[lnk->mode],
		                                lnk->no_promisc ? "nopromisc" : "");
		if (success)
			pllink = nmtstp_assert_wait_for_link (platform, name, link_type, 100);
	} else
		success = NMTST_NM_ERR_SUCCESS (nm_platform_link_macvlan_add (platform, name, parent, lnk, &pllink));

	_assert_pllink (platform, success, pllink, name, link_type);

	return pllink;
}

const NMPlatformLink *
nmtstp_link_sit_add (NMPlatform *platform,
                     gboolean external_command,
                     const char *name,
                     const NMPlatformLnkSit *lnk)
{
	const NMPlatformLink *pllink = NULL;
	gboolean success;
	char b1[INET_ADDRSTRLEN];
	char b2[INET_ADDRSTRLEN];

	g_assert (nm_utils_is_valid_iface_name (name, NULL));

	external_command = nmtstp_run_command_check_external (external_command);

	_init_platform (&platform, external_command);

	if (external_command) {
		const char *dev = "";

		if (lnk->parent_ifindex) {
			const char *parent_name;

			parent_name = nm_platform_link_get_name (platform, lnk->parent_ifindex);
			g_assert (parent_name);
			dev = nm_sprintf_bufa (100, " dev %s", parent_name);
		}

		success = !nmtstp_run_command ("ip tunnel add %s mode sit%s local %s remote %s ttl %u tos %02x %s",
		                                name,
		                                dev,
		                                nm_utils_inet4_ntop (lnk->local, b1),
		                                nm_utils_inet4_ntop (lnk->remote, b2),
		                                lnk->ttl,
		                                lnk->tos,
		                                lnk->path_mtu_discovery ? "pmtudisc" : "nopmtudisc");
		if (success)
			pllink = nmtstp_assert_wait_for_link (platform, name, NM_LINK_TYPE_SIT, 100);
	} else
		success = NMTST_NM_ERR_SUCCESS (nm_platform_link_sit_add (platform, name, lnk, &pllink));

	_assert_pllink (platform, success, pllink, name, NM_LINK_TYPE_SIT);

	return pllink;
}

const NMPlatformLink *
nmtstp_link_tun_add (NMPlatform *platform,
                     gboolean external_command,
                     const char *name,
                     const NMPlatformLnkTun *lnk,
                     int *out_fd)
{
	const NMPlatformLink *pllink = NULL;
	int err;
	int r;

	g_assert (nm_utils_is_valid_iface_name (name, NULL));
	g_assert (lnk);
	g_assert (NM_IN_SET (lnk->type, IFF_TUN, IFF_TAP));
	g_assert (!out_fd || *out_fd == -1);

	if (!lnk->persist) {
		/* ip tuntap does not support non-persistent devices.
		 *
		 * Add this device only via NMPlatform. */
		if (external_command == -1)
			external_command = FALSE;
	}

	external_command = nmtstp_run_command_check_external (external_command);

	_init_platform (&platform, external_command);

	if (external_command) {
		g_assert (lnk->persist);

		err = nmtstp_run_command ("ip tuntap add"
		                          " mode %s"
		                          "%s" /* user */
		                          "%s" /* group */
		                          "%s" /* pi */
		                          "%s" /* vnet_hdr */
		                          "%s" /* multi_queue */
		                          " name %s",
		                          lnk->type == IFF_TUN ? "tun" : "tap",
		                          lnk->owner_valid ? nm_sprintf_bufa (100, " user %u", (guint) lnk->owner) : "",
		                          lnk->group_valid ? nm_sprintf_bufa (100, " group %u", (guint) lnk->group) : "",
		                          lnk->pi ? " pi" : "",
		                          lnk->vnet_hdr ? " vnet_hdr" : "",
		                          lnk->multi_queue ? " multi_queue" : "",
		                          name);
		/* Older versions of iproute2 don't support adding  devices.
		 * On failure, fallback to using platform code. */
		if (err == 0)
			pllink = nmtstp_assert_wait_for_link (platform, name, NM_LINK_TYPE_TUN, 100);
		else
			g_error ("failure to add tun/tap device via ip-route");
	} else {
		g_assert (lnk->persist || out_fd);
		r = nm_platform_link_tun_add (platform, name, lnk, &pllink, out_fd);
		g_assert_cmpint (r, ==, 0);
	}

	g_assert (pllink);
	g_assert_cmpint (pllink->type, ==, NM_LINK_TYPE_TUN);
	g_assert_cmpstr (pllink->name, ==, name);
	return pllink;
}

const NMPlatformLink *
nmtstp_link_vxlan_add (NMPlatform *platform,
                       gboolean external_command,
                       const char *name,
                       const NMPlatformLnkVxlan *lnk)
{
	const NMPlatformLink *pllink = NULL;
	int err;
	int r;

	g_assert (nm_utils_is_valid_iface_name (name, NULL));

	external_command = nmtstp_run_command_check_external (external_command);

	_init_platform (&platform, external_command);

	if (external_command) {
		gs_free char *dev = NULL;
		char local[NM_UTILS_INET_ADDRSTRLEN];
		char group[NM_UTILS_INET_ADDRSTRLEN];

		if (lnk->parent_ifindex)
			dev = g_strdup_printf ("dev %s", nm_platform_link_get_name (platform, lnk->parent_ifindex));

		if (lnk->local)
			nm_utils_inet4_ntop (lnk->local, local);
		else if (memcmp (&lnk->local6, &in6addr_any, sizeof (in6addr_any)))
			nm_utils_inet6_ntop (&lnk->local6, local);
		else
			local[0] = '\0';

		if (lnk->group)
			nm_utils_inet4_ntop (lnk->group, group);
		else if (memcmp (&lnk->group6, &in6addr_any, sizeof (in6addr_any)))
			nm_utils_inet6_ntop (&lnk->group6, group);
		else
			group[0] = '\0';

		err = nmtstp_run_command ("ip link add %s type vxlan id %u %s local %s group %s ttl %u tos %02x dstport %u srcport %u %u ageing %u",
		                          name,
		                          lnk->id,
		                          dev ?: "",
		                          local,
		                          group,
		                          lnk->ttl,
		                          lnk->tos,
		                          lnk->dst_port,
		                          lnk->src_port_min, lnk->src_port_max,
		                          lnk->ageing);
		/* Older versions of iproute2 don't support adding vxlan devices.
		 * On failure, fallback to using platform code. */
		if (err == 0)
			pllink = nmtstp_assert_wait_for_link (platform, name, NM_LINK_TYPE_VXLAN, 100);
		else
			_LOGI ("Adding vxlan device via iproute2 failed. Assume iproute2 is not up to the task.");
	}
	if (!pllink) {
		r = nm_platform_link_vxlan_add (platform, name, lnk, &pllink);
		g_assert (NMTST_NM_ERR_SUCCESS (r));
		g_assert (pllink);
	}

	g_assert_cmpint (pllink->type, ==, NM_LINK_TYPE_VXLAN);
	g_assert_cmpstr (pllink->name, ==, name);
	return pllink;
}

/*****************************************************************************/

const NMPlatformLink *
nmtstp_link_get_typed (NMPlatform *platform,
                       int ifindex,
                       const char *name,
                       NMLinkType link_type)
{
	const NMPlatformLink *pllink = NULL;

	_init_platform (&platform, FALSE);

	if (ifindex > 0) {
		pllink = nm_platform_link_get (platform, ifindex);

		if (pllink) {
			g_assert_cmpint (pllink->ifindex, ==, ifindex);
			if (name)
				g_assert_cmpstr (name, ==, pllink->name);
		} else {
			if (name)
				g_assert (!nm_platform_link_get_by_ifname (platform, name));
		}
	} else {
		g_assert (name);

		pllink = nm_platform_link_get_by_ifname (platform, name);

		if (pllink)
			g_assert_cmpstr (name, ==, pllink->name);
	}

	g_assert (!name || nm_utils_is_valid_iface_name (name, NULL));

	if (pllink && link_type != NM_LINK_TYPE_NONE)
		g_assert_cmpint (pllink->type, ==, link_type);

	return pllink;
}

const NMPlatformLink *
nmtstp_link_get (NMPlatform *platform,
                 int ifindex,
                 const char *name)
{
	return nmtstp_link_get_typed (platform, ifindex, name, NM_LINK_TYPE_NONE);
}

/*****************************************************************************/

void
nmtstp_link_delete (NMPlatform *platform,
                    gboolean external_command,
                    int ifindex,
                    const char *name,
                    gboolean require_exist)
{
	gint64 end_time;
	const NMPlatformLink *pllink;
	gboolean success;
	gs_free char *name_copy = NULL;

	external_command = nmtstp_run_command_check_external (external_command);

	_init_platform (&platform, external_command);

	pllink = nmtstp_link_get (platform, ifindex, name);

	if (!pllink) {
		g_assert (!require_exist);
		return;
	}

	name = name_copy = g_strdup (pllink->name);
	ifindex = pllink->ifindex;

	if (external_command) {
		nmtstp_run_command_check ("ip link delete %s", name);
	} else {
		success = nm_platform_link_delete (platform, ifindex);
		g_assert (success);
	}

	/* Let's wait until we get the result */
	end_time = nm_utils_get_monotonic_timestamp_ms () + 250;
	do {
		if (external_command)
			nm_platform_process_events (platform);

		if (!nm_platform_link_get (platform, ifindex)) {
			g_assert (!nm_platform_link_get_by_ifname (platform, name));
			break;
		}

		/* for internal command, we expect not to reach this line.*/
		g_assert (external_command);

		nmtstp_assert_wait_for_signal_until (platform, end_time);
	} while (TRUE);
}

/*****************************************************************************/

void
nmtstp_link_set_updown (NMPlatform *platform,
                        gboolean external_command,
                        int ifindex,
                        gboolean up)
{
	const NMPlatformLink *plink;
	gint64 end_time;

	external_command = nmtstp_run_command_check_external (external_command);

	_init_platform (&platform, external_command);

	if (external_command) {
		const char *ifname;

		ifname = nm_platform_link_get_name (platform, ifindex);
		g_assert (ifname);

		nmtstp_run_command_check ("ip link set %s %s",
		                          ifname,
		                          up ? "up" : "down");
	} else {
		if (up)
			g_assert (nm_platform_link_set_up (platform, ifindex, NULL));
		else
			g_assert (nm_platform_link_set_down (platform, ifindex));
	}

	/* Let's wait until we get the result */
	end_time = nm_utils_get_monotonic_timestamp_ms () + 250;
	do {
		if (external_command)
			nm_platform_process_events (platform);

		/* let's wait until we see the address as we added it. */
		plink = nm_platform_link_get (platform, ifindex);
		g_assert (plink);

		if (NM_FLAGS_HAS (plink->n_ifi_flags, IFF_UP) == !!up)
			break;

		/* for internal command, we expect not to reach this line.*/
		g_assert (external_command);

		nmtstp_assert_wait_for_signal_until (platform, end_time);
	} while (TRUE);
}

/*****************************************************************************/

struct _NMTstpNamespaceHandle {
	pid_t pid;
	int pipe_fd;
};

NMTstpNamespaceHandle *
nmtstp_namespace_create (int unshare_flags, GError **error)
{
	NMTstpNamespaceHandle *ns_handle;
	int e;
	int errsv;
	pid_t pid, pid2;
	int pipefd_c2p[2];
	int pipefd_p2c[2];
	ssize_t r;

	e = pipe2 (pipefd_c2p, O_CLOEXEC);
	if (e != 0) {
		errsv = errno;
		g_set_error (error, NM_UTILS_ERROR, NM_UTILS_ERROR_UNKNOWN,
		             "pipe() failed with %d (%s)", errsv, nm_strerror_native (errsv));
		return FALSE;
	}

	e = pipe2 (pipefd_p2c, O_CLOEXEC);
	if (e != 0) {
		errsv = errno;
		g_set_error (error, NM_UTILS_ERROR, NM_UTILS_ERROR_UNKNOWN,
		             "pipe() failed with %d (%s)", errsv, nm_strerror_native (errsv));
		nm_close (pipefd_c2p[0]);
		nm_close (pipefd_c2p[1]);
		return FALSE;
	}

	pid = fork ();
	if (pid < 0) {
		errsv = errno;
		g_set_error (error, NM_UTILS_ERROR, NM_UTILS_ERROR_UNKNOWN,
		             "fork() failed with %d (%s)", errsv, nm_strerror_native (errsv));
		nm_close (pipefd_c2p[0]);
		nm_close (pipefd_c2p[1]);
		nm_close (pipefd_p2c[0]);
		nm_close (pipefd_p2c[1]);
		return FALSE;
	}

	if (pid == 0) {
		char read_buf[1];

		nm_close (pipefd_c2p[0]); /* close read-end */
		nm_close (pipefd_p2c[1]); /* close write-end */

		if (unshare (unshare_flags) != 0) {
			errsv = errno;
			if (errsv == 0)
				errsv = -1;
		} else
			errsv = 0;

		/* sync with parent process and send result. */
		do {
			r = write (pipefd_c2p[1], &errsv, sizeof (errsv));
		} while (r < 0 && errno == EINTR);
		if (r != sizeof (errsv)) {
			errsv = errno;
			if (errsv == 0)
				errsv = -2;
		}
		nm_close (pipefd_c2p[1]);

		/* wait until parent process terminates (or kills us). */
		if (errsv == 0) {
			do {
				r = read (pipefd_p2c[0], read_buf, sizeof (read_buf));
			} while (r < 0 && errno == EINTR);
		}
		nm_close (pipefd_p2c[0]);
		_exit (0);
	}

	nm_close (pipefd_c2p[1]); /* close write-end */
	nm_close (pipefd_p2c[0]); /* close read-end */

	/* sync with child process. */
	do {
		r = read (pipefd_c2p[0], &errsv, sizeof (errsv));
	} while (r < 0 && errno == EINTR);

	nm_close (pipefd_c2p[0]);

	if (   r != sizeof (errsv)
	    || errsv != 0) {
		int status;

		if (r != sizeof (errsv)) {
			g_set_error (error, NM_UTILS_ERROR, NM_UTILS_ERROR_UNKNOWN,
			             "child process failed for unknown reason");
		} else {
			g_set_error (error, NM_UTILS_ERROR, NM_UTILS_ERROR_UNKNOWN,
			             "child process signaled failure %d (%s)", errsv, nm_strerror_native (errsv));
		}
		nm_close (pipefd_p2c[1]);
		kill (pid, SIGKILL);
		do {
			pid2 = waitpid (pid, &status, 0);
		} while (pid2 == -1 && errno == EINTR);
		return FALSE;
	}

	ns_handle = g_new0 (NMTstpNamespaceHandle, 1);
	ns_handle->pid = pid;
	ns_handle->pipe_fd = pipefd_p2c[1];
	return ns_handle;
}

pid_t
nmtstp_namespace_handle_get_pid (NMTstpNamespaceHandle *ns_handle)
{
	g_return_val_if_fail (ns_handle, 0);
	g_return_val_if_fail (ns_handle->pid > 0, 0);

	return ns_handle->pid;
}

void
nmtstp_namespace_handle_release (NMTstpNamespaceHandle *ns_handle)
{
	pid_t pid;
	int status;

	if (!ns_handle)
		return;

	g_return_if_fail (ns_handle->pid > 0);

	nm_close (ns_handle->pipe_fd);
	ns_handle->pipe_fd = 0;

	kill (ns_handle->pid, SIGKILL);

	do {
		pid = waitpid (ns_handle->pid, &status, 0);
	} while (pid == -1 && errno == EINTR);
	ns_handle->pid = 0;

	g_free (ns_handle);
}

int
nmtstp_namespace_get_fd_for_process (pid_t pid, const char *ns_name)
{
	char p[1000];

	g_return_val_if_fail (pid > 0, 0);
	g_return_val_if_fail (ns_name && ns_name[0] && strlen (ns_name) < 50, 0);

	nm_sprintf_buf (p, "/proc/%lu/ns/%s", (unsigned long) pid, ns_name);

	return open(p, O_RDONLY | O_CLOEXEC);
}

/*****************************************************************************/

void
nmtstp_netns_select_random (NMPlatform **platforms, gsize n_platforms, NMPNetns **netns)
{
	int i;

	g_assert (platforms);
	g_assert (n_platforms && n_platforms <= G_MAXINT32);
	g_assert (netns && !*netns);
	for (i = 0; i < n_platforms; i++)
		g_assert (NM_IS_PLATFORM (platforms[i]));

	i = nmtst_get_rand_uint32 () % (n_platforms + 1);
	if (i == 0)
		return;
	g_assert (nm_platform_netns_push (platforms[i - 1], netns));
}

/*****************************************************************************/

NMTST_DEFINE();

static gboolean
unshare_user (void)
{
	FILE *f;
	uid_t uid = geteuid ();
	gid_t gid = getegid ();

	/* Already a root? */
	if (gid == 0 && uid == 0)
		return TRUE;

	/* Become a root in new user NS. */
	if (unshare (CLONE_NEWUSER) != 0)
		return FALSE;

	/* Since Linux 3.19 we have to disable setgroups() in order to map users.
	 * Just proceed if the file is not there. */
	f = fopen ("/proc/self/setgroups", "we");
	if (f) {
		fprintf (f, "deny");
		fclose (f);
	}

	/* Map current UID to root in NS to be created. */
	f = fopen ("/proc/self/uid_map", "we");
	if (!f)
		return FALSE;
	fprintf (f, "0 %d 1", uid);
	fclose (f);

	/* Map current GID to root in NS to be created. */
	f = fopen ("/proc/self/gid_map", "we");
	if (!f)
		return FALSE;
	fprintf (f, "0 %d 1", gid);
	fclose (f);

	return TRUE;
}

int
main (int argc, char **argv)
{
	int result;
	const char *program = *argv;

	_nmtstp_init_tests (&argc, &argv);

	if (   nmtstp_is_root_test ()
	    && (geteuid () != 0 || getegid () != 0)) {
		if (   g_getenv ("NMTST_FORCE_REAL_ROOT")
		    || !unshare_user ()) {
			/* Try to exec as sudo, this function does not return, if a sudo-cmd is set. */
			nmtst_reexec_sudo ();

#ifdef REQUIRE_ROOT_TESTS
			g_print ("Fail test: requires root privileges (%s)\n", program);
			return EXIT_FAILURE;
#else
			g_print ("Skipping test: requires root privileges (%s)\n", program);
			return g_test_run ();
#endif
		}
	}

	if (nmtstp_is_root_test () && !g_getenv ("NMTST_NO_UNSHARE")) {
		int errsv;

		if (unshare (CLONE_NEWNET | CLONE_NEWNS) != 0) {
			errsv = errno;
			g_error ("unshare(CLONE_NEWNET|CLONE_NEWNS) failed with %s (%d)", nm_strerror_native (errsv), errsv);
		}

		/* We need a read-only /sys so that the platform knows there's no udev. */
		mount (NULL, "/sys", "sysfs", MS_SLAVE, NULL);
		if (mount ("sys", "/sys", "sysfs", MS_RDONLY, NULL) != 0) {
			errsv = errno;
			g_error ("mount(\"/sys\") failed with %s (%d)", nm_strerror_native (errsv), errsv);
		}
	}

	nmtstp_setup_platform ();

	_nmtstp_setup_tests ();

	result = g_test_run ();

	nmtstp_link_delete (NM_PLATFORM_GET, -1, -1, DEVICE_NAME, FALSE);

	g_object_unref (NM_PLATFORM_GET);
	return result;
}
