#!/bin/sh

# SPDX-License-Identifier: GPL-2.0

# This is not an autoconf generated configure

#

INCLUDE=${1:-"$PWD/include"}

# Output file which is input to Makefile

CONFIG=config.mk

# Make a temp directory in build tree.

TMPDIR=$(mktemp -d config.XXXXXX)

trap 'status=$?; rm -rf $TMPDIR; exit $status' EXIT HUP INT QUIT TERM

check_toolchain()

{

    : ${PKG_CONFIG:=pkg-config}

    : ${AR=ar}

    : ${CC=gcc}

    : ${YACC=bison}

    echo "PKG_CONFIG:=${PKG_CONFIG}" >>$CONFIG

    echo "AR:=${AR}" >>$CONFIG

    echo "CC:=${CC}" >>$CONFIG

    echo "YACC:=${YACC}" >>$CONFIG

}

check_atm()

{

    cat >$TMPDIR/atmtest.c <

#include <atm.h>

int main(int argc, char **argv) {

	struct atm_qos qos;

	(void) text2qos("aal5,ubr:sdu=9180,rx:none",&qos,0);

	return 0;

}

EOF

    if $CC -I$INCLUDE -o $TMPDIR/atmtest $TMPDIR/atmtest.c -latm >/dev/null 2>&1; then

	echo "TC_CONFIG_ATM:=y" >>$CONFIG

	echo yes

    else

	echo no

    fi

    rm -f $TMPDIR/atmtest.c $TMPDIR/atmtest

}

check_xtables()

{

	if ! ${PKG_CONFIG} xtables --exists; then

		echo "TC_CONFIG_NO_XT:=y" >>$CONFIG

	fi

}

check_xt()

{

    #check if we have xtables from iptables >= 1.4.5.

    cat >$TMPDIR/ipttest.c <

#include <xtables.h>

#include <linux/netfilter.h>

static struct xtables_globals test_globals = {

	.option_offset = 0,

	.program_name = "tc-ipt",

	.program_version = XTABLES_VERSION,

	.orig_opts = NULL,

	.opts = NULL,

	.exit_err = NULL,

};

int main(int argc, char **argv)

{

	xtables_init_all(&test_globals, NFPROTO_IPV4);

	return 0;

}

EOF

    if $CC -I$INCLUDE $IPTC -o $TMPDIR/ipttest $TMPDIR/ipttest.c $IPTL \

	$(${PKG_CONFIG} xtables --cflags --libs) -ldl >/dev/null 2>&1; then

	echo "TC_CONFIG_XT:=y" >>$CONFIG

	echo "using xtables"

    fi

    rm -f $TMPDIR/ipttest.c $TMPDIR/ipttest

}

check_xt_old()

{

    # bail if previous XT checks has already succeeded.

    grep -q TC_CONFIG_XT $CONFIG && return

    #check if we don't need our internal header ..

    cat >$TMPDIR/ipttest.c <

#include <xtables.h>

char *lib_dir;

unsigned int global_option_offset = 0;

const char *program_version = XTABLES_VERSION;

const char *program_name = "tc-ipt";

struct afinfo afinfo = {

	.libprefix      = "libxt_",

};

void exit_error(enum exittype status, const char *msg, ...)

{

}

int main(int argc, char **argv) {

	return 0;

}

EOF

    if $CC -I$INCLUDE $IPTC -o $TMPDIR/ipttest $TMPDIR/ipttest.c $IPTL -ldl >/dev/null 2>&1; then

	echo "TC_CONFIG_XT_OLD:=y" >>$CONFIG

	echo "using old xtables (no need for xt-internal.h)"

    fi

    rm -f $TMPDIR/ipttest.c $TMPDIR/ipttest

}

check_xt_old_internal_h()

{

    # bail if previous XT checks has already succeeded.

    grep -q TC_CONFIG_XT $CONFIG && return

    #check if we need our own internal.h

    cat >$TMPDIR/ipttest.c <

#include <xtables.h>

#include "xt-internal.h"

char *lib_dir;

unsigned int global_option_offset = 0;

const char *program_version = XTABLES_VERSION;

const char *program_name = "tc-ipt";

struct afinfo afinfo = {

	.libprefix      = "libxt_",

};

void exit_error(enum exittype status, const char *msg, ...)

{

}

int main(int argc, char **argv) {

	return 0;

}

EOF

	if $CC -I$INCLUDE $IPTC -o $TMPDIR/ipttest $TMPDIR/ipttest.c $IPTL -ldl >/dev/null 2>&1; then

	    echo "using old xtables with xt-internal.h"

	    echo "TC_CONFIG_XT_OLD_H:=y" >>$CONFIG

	fi

	rm -f $TMPDIR/ipttest.c $TMPDIR/ipttest

}

check_ipt()

{

	if ! grep TC_CONFIG_XT $CONFIG > /dev/null; then

		echo "using iptables"

	fi

}

check_ipt_lib_dir()

{

	IPT_LIB_DIR=$(${PKG_CONFIG} --variable=xtlibdir xtables)

	if [ -n "$IPT_LIB_DIR" ]; then

		echo $IPT_LIB_DIR

		echo "IPT_LIB_DIR:=$IPT_LIB_DIR" >> $CONFIG

		return

	fi

	for dir in /lib /usr/lib /usr/local/lib; do

		for file in "xtables" "iptables"; do

			file="$dir/$file/lib*t_*so"

			if [ -f $file ]; then

				echo ${file%/*}

				echo "IPT_LIB_DIR:=${file%/*}" >> $CONFIG

				return

			fi

		done

	done

	echo "not found!"

}

check_setns()

{

    cat >$TMPDIR/setnstest.c <

#include <sched.h>

int main(int argc, char **argv)

{

	(void)setns(0,0);

	return 0;

}

EOF

    if $CC -I$INCLUDE -o $TMPDIR/setnstest $TMPDIR/setnstest.c >/dev/null 2>&1; then

	echo "IP_CONFIG_SETNS:=y" >>$CONFIG

	echo "yes"

	echo "CFLAGS += -DHAVE_SETNS" >>$CONFIG

    else

	echo "no"

    fi

    rm -f $TMPDIR/setnstest.c $TMPDIR/setnstest

}

check_ipset()

{

    cat >$TMPDIR/ipsettest.c <

#include <linux/netfilter/ipset/ip_set.h>

#ifndef IP_SET_INVALID

#define IPSET_DIM_MAX 3

typedef unsigned short ip_set_id_t;

#endif

#include <linux/netfilter/xt_set.h>

struct xt_set_info info;

#if IPSET_PROTOCOL == 6 || IPSET_PROTOCOL == 7

int main(void)

{

	return IPSET_MAXNAMELEN;

}

#else

#error unknown ipset version

#endif

EOF

    if $CC -I$INCLUDE -o $TMPDIR/ipsettest $TMPDIR/ipsettest.c >/dev/null 2>&1; then

	echo "TC_CONFIG_IPSET:=y" >>$CONFIG

	echo "yes"

    else

	echo "no"

    fi

    rm -f $TMPDIR/ipsettest.c $TMPDIR/ipsettest

}

check_elf()

{

    if ${PKG_CONFIG} libelf --exists; then

	echo "HAVE_ELF:=y" >>$CONFIG

	echo "yes"

	echo 'CFLAGS += -DHAVE_ELF' `${PKG_CONFIG} libelf --cflags` >> $CONFIG

	echo 'LDLIBS += ' `${PKG_CONFIG} libelf --libs` >>$CONFIG

    else

	echo "no"

    fi

}

check_selinux()

# SELinux is a compile time option in the ss utility

{

	if ${PKG_CONFIG} libselinux --exists; then

		echo "HAVE_SELINUX:=y" >>$CONFIG

		echo "yes"

		echo 'LDLIBS +=' `${PKG_CONFIG} --libs libselinux` >>$CONFIG

		echo 'CFLAGS += -DHAVE_SELINUX' `${PKG_CONFIG} --cflags libselinux` >>$CONFIG

	else

		echo "no"

	fi

}

check_mnl()

{

	if ${PKG_CONFIG} libmnl --exists; then

		echo "HAVE_MNL:=y" >>$CONFIG

		echo "yes"

		echo 'CFLAGS += -DHAVE_LIBMNL' `${PKG_CONFIG} libmnl --cflags` >>$CONFIG

		echo 'LDLIBS +=' `${PKG_CONFIG} libmnl --libs` >> $CONFIG

	else

		echo "no"

	fi

}

check_berkeley_db()

{

    cat >$TMPDIR/dbtest.c <

#include <fcntl.h>

#include <stdlib.h>

#include <db_185.h>

int main(int argc, char **argv) {

	dbopen("/tmp/xxx_test_db.db", O_CREAT|O_RDWR, 0644, DB_HASH, NULL);

	return 0;

}

EOF

    if $CC -I$INCLUDE -o $TMPDIR/dbtest $TMPDIR/dbtest.c -ldb >/dev/null 2>&1; then

	echo "HAVE_BERKELEY_DB:=y" >>$CONFIG

	echo "yes"

    else

	echo "no"

    fi

    rm -f $TMPDIR/dbtest.c $TMPDIR/dbtest

}

check_strlcpy()

{

    cat >$TMPDIR/strtest.c <

#include <string.h>

int main(int argc, char **argv) {

	char dst[10];

	strlcpy(dst, "test", sizeof(dst));

	return 0;

}

EOF

    if $CC -I$INCLUDE -o $TMPDIR/strtest $TMPDIR/strtest.c >/dev/null 2>&1; then

	echo "no"

    else

	if ${PKG_CONFIG} libbsd --exists; then

		echo 'CFLAGS += -DHAVE_LIBBSD' `${PKG_CONFIG} libbsd --cflags` >>$CONFIG

		echo 'LDLIBS +=' `${PKG_CONFIG} libbsd --libs` >> $CONFIG

		echo "no"

	else

		echo 'CFLAGS += -DNEED_STRLCPY' >>$CONFIG

		echo "yes"

	fi

    fi

    rm -f $TMPDIR/strtest.c $TMPDIR/strtest

}

check_cap()

{

	if ${PKG_CONFIG} libcap --exists; then

		echo "HAVE_CAP:=y" >>$CONFIG

		echo "yes"

		echo 'CFLAGS += -DHAVE_LIBCAP' `${PKG_CONFIG} libcap --cflags` >>$CONFIG

		echo 'LDLIBS +=' `${PKG_CONFIG} libcap --libs` >> $CONFIG

	else

		echo "no"

	fi

}

quiet_config()

{

	cat <

# user can control verbosity similar to kernel builds (e.g., V=1)

ifeq ("\$(origin V)", "command line")

  VERBOSE = \$(V)

endif

ifndef VERBOSE

  VERBOSE = 0

endif

ifeq (\$(VERBOSE),1)

  Q =

else

  Q = @

endif

ifeq (\$(VERBOSE), 0)

    QUIET_CC       = @echo '    CC       '\$@;

    QUIET_AR       = @echo '    AR       '\$@;

    QUIET_LINK     = @echo '    LINK     '\$@;

    QUIET_YACC     = @echo '    YACC     '\$@;

    QUIET_LEX      = @echo '    LEX      '\$@;

endif

EOF

}

echo "# Generated config based on" $INCLUDE >$CONFIG

quiet_config >> $CONFIG

check_toolchain

echo "TC schedulers"

echo -n " ATM	"

check_atm

check_xtables

if ! grep -q TC_CONFIG_NO_XT $CONFIG; then

	echo -n " IPT	"

	check_xt

	check_xt_old

	check_xt_old_internal_h

	check_ipt

	echo -n " IPSET  "

	check_ipset

fi

echo

if ! grep -q TC_CONFIG_NO_XT $CONFIG; then

	echo -n "iptables modules directory: "

	check_ipt_lib_dir

fi

echo -n "libc has setns: "

check_setns

echo -n "SELinux support: "

check_selinux

echo -n "ELF support: "

check_elf

echo -n "libmnl support: "

check_mnl

echo -n "Berkeley DB: "

check_berkeley_db

echo -n "need for strlcpy: "

check_strlcpy

echo -n "libcap support: "

check_cap

echo >> $CONFIG

echo "%.o: %.c" >> $CONFIG

echo '	$(QUIET_CC)$(CC) $(CFLAGS) $(EXTRA_CFLAGS) $(CPPFLAGS) -c -o $@ $<' >> $CONFIG