configure

#!/bin/bash

CONFIGH="config.h"

if [ -t 1 ]; then
	RED=""
	GREEN=""
	COL_RESET=""
fi

MISSING_DEFS=0

[ -z "$CC" ] && CC=gcc

CC="${CROSS_COMPILE}${CC}"

CFLAGS="${CFLAGS}"
if [ "${SYSROOT}xx" != "xx" ]; then
	CFLAGS="${CFLAGS} --sysroot=${SYSROOT}"
fi

echo "#pragma once" > $CONFIGH
echo "/* This file is auto-generated by configure.sh */" >> $CONFIGH

TMP=$(mktemp)
trap 'rm -f "$TMP" "$TMP.log" "$TMP.c"' EXIT

check_header()
{
	echo -n "[*] Checking header $1 ... "

	rm -f "$TMP" || exit 1
	echo "#include <$1>" >"$TMP.c"

	${CC} ${CFLAGS} "$TMP.c" -E &>"$TMP.log"
	if [ $? -eq 0 ]; then
		echo $GREEN "[YES]" $COL_RESET
		echo "#define $2 1" >> $CONFIGH
	else
		echo $RED "[NO]" $COL_RESET
		MISSING_DEFS=1
	fi
}

#############################################################################################

echo "[*] Checking system headers."

#############################################################################################
# Are ipv6 headers usable ?
[ -z "$IPV6" ] && IPV6=yes
if [[ "$IPV6" == "yes" ]]; then
	echo -n "[*] Checking ipv6 headers ... "
	rm -f "$TMP" || exit 1

	cat >"$TMP.c" << EOF
#include <netinet/in.h>
#include <netinet/ip6.h>
#include <linux/in6.h>
#include <linux/ipv6.h>

int main()
{
	struct in6_addr test;
}
EOF
	${CC} ${CFLAGS} -D_GNU_SOURCE "$TMP.c" -o "$TMP" &>"$TMP.log"
	if [ ! -x "$TMP" ]; then
		echo $RED "[BROKEN]" $COL_RESET "See https://patchwork.ozlabs.org/patch/425881/"
	else
		echo $GREEN "[OK]" $COL_RESET
		echo "#define USE_IPV6 1" >> $CONFIGH
	fi
fi

#############################################################################################
# is /usr/include/linux/if_pppox.h new enough to feature pppol2tpin6/pppol2tpv3in6
#
echo -n "[*] Checking if pppox can use pppol2tpin6.. "
rm -f "$TMP" || exit 1

cat >"$TMP.c" << EOF
#include <stdio.h>
#include <netinet/in.h>
#include <linux/if.h>
#include <linux/if_pppox.h>

int main(void)
{
	struct sockaddr_pppol2tpin6 *pppox;
	printf("%d\n", pppox->sa_family);
}
EOF

${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo $RED "[NO]" $COL_RESET
	MISSING_DEFS=1
else
	echo $GREEN "[YES]" $COL_RESET
	echo "#define USE_PPPOL2TPIN6 1" >> $CONFIGH
fi

#############################################################################################
# is /usr/include/linux/if_pppox.h new enough to feature pppol2tpv3
#
echo -n "[*] Checking if pppox can use pppol2tpv3.. "
rm -f "$TMP" || exit 1

cat >"$TMP.c" << EOF
#include <stdio.h>
#include <netinet/in.h>
#include <linux/if.h>
#include <linux/if_pppox.h>

int main(void)
{
	struct sockaddr_pppol2tpv3 *pppox;
	printf("%d\n", pppox->sa_family);
}
EOF

${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo $RED "[NO]" $COL_RESET
	MISSING_DEFS=1
else
	echo $GREEN "[YES]" $COL_RESET
	echo "#define USE_PPPOL2TPV3 1" >> $CONFIGH
fi

#############################################################################################
# is /usr/include/linux/if_pppox.h new enough to feature pptp
#
echo -n "[*] Checking if pppox can use pptp.. "
rm -f "$TMP" || exit 1

cat >"$TMP.c" << EOF
#include <stdio.h>
#include <netinet/in.h>
#include <linux/if.h>
#include <linux/if_pppox.h>

int main(void)
{
	struct sockaddr_pppox *pppox;
	printf("%d\n", pppox->sa_addr.pptp.call_id);
}
EOF

${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo $RED "[NO]" $COL_RESET
	MISSING_DEFS=1
else
	echo $GREEN "[YES]" $COL_RESET
	echo "#define USE_PPPOX_PPTP 1" >> $CONFIGH
fi

#############################################################################################
# is /usr/include/linux/llc.h new enough to feature LLC_OPT_PKTINFO
#
echo -n "[*] Checking if llc can use LLC_OPT_PKTINFO.. "
rm -f "$TMP" || exit 1

cat >"$TMP.c" << EOF
#include <stdio.h>
#include <net/if.h>
#include <linux/llc.h>

int main(void)
{
	printf("%d\n", LLC_OPT_PKTINFO);
}
EOF

${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo $RED "[NO]" $COL_RESET
	MISSING_DEFS=1
else
	echo $GREEN "[YES]" $COL_RESET
	echo "#define USE_LLC_OPT_PKTINFO 1" >> $CONFIGH
fi

#############################################################################################
# Do glibc headers provides struct termios2

echo -n "[*] Checking if glibc headers provide termios2.. "
rm -f "$TMP" || exit 1

cat >"$TMP.c" << EOF
#include <sys/ioctl.h>
#include <sys/vt.h>
#include <termios.h>

int main()
{
	struct termios2 test;
}
EOF

${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo $RED "[NO]" $COL_RESET
	MISSING_DEFS=1
else
	echo $GREEN "[YES]" $COL_RESET
	echo "#define HAVE_TERMIOS2 1" >> $CONFIGH
fi

#############################################################################################
# Do glibc headers provides nvme ioctls

echo -n "[*] Checking if glibc headers provide nvme ioctls.. "
rm -f "$TMP" || exit 1

cat >"$TMP.c" << EOF
#include <sys/ioctl.h>
#include <linux/nvme_ioctl.h>

int main(int argc, char* argv[])
{
	unsigned int foo = NVME_IOCTL_IO_CMD;
}
EOF

${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo $RED "[NO]" $COL_RESET
	MISSING_DEFS=1
else
	echo $GREEN "[YES]" $COL_RESET
	echo "#define USE_NVME 1" >> $CONFIGH
fi

#############################################################################################
# is /usr/include/linux/bpf.h new enough to feature bpf
#
echo -n "[*] Checking if bpf_attr can use map_flags.. "
rm -f "$TMP" || exit 1

cat >"$TMP.c" << EOF
#include <stdio.h>
#include <linux/bpf.h>

int main(void)
{
	union bpf_attr attr = {
		.map_flags = 0,
	};

	enum bpf_prog_type type = BPF_PROG_TYPE_PERF_EVENT;
}
EOF

${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo $RED "[NO]" $COL_RESET
	MISSING_DEFS=1
else
	echo $GREEN "[YES]" $COL_RESET
	echo "#define USE_BPF 1" >> $CONFIGH
fi

#############################################################################################
# is /usr/include/linux/btrfs.h new enough to feature btrfs
#
echo -n "[*] Checking if btrfs can use btrfs_ioctl_defrag_range_args.. "
rm -f "$TMP" || exit 1

cat >"$TMP.c" << EOF
#include <stdio.h>
#include <linux/btrfs.h>

int main(void)
{
	struct btrfs_ioctl_defrag_range_args args;
}
EOF

${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo $RED "[NO]" $COL_RESET
	MISSING_DEFS=1
else
	echo $GREEN "[YES]" $COL_RESET
	echo "#define USE_BTRFS 1" >> $CONFIGH
fi

#############################################################################################
# Does linux/xattr.h define struct xattr_args (added in 6.13)
#
echo -n "[*] Checking if struct xattr_args is defined.. "
rm -f "$TMP" || exit 1

cat >"$TMP.c" << EOF
#include <linux/xattr.h>

int main(void)
{
	struct xattr_args args = { .flags = 0 };
	return args.flags;
}
EOF

${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo $RED "[NO]" $COL_RESET
	MISSING_DEFS=1
else
	echo $GREEN "[YES]" $COL_RESET
	echo "#define USE_XATTR_ARGS 1" >> $CONFIGH
fi

#############################################################################################
# Does linux/io_uring.h give io_uring_clone_buffers the src_off/dst_off/nr
# members (added in 6.13)
#
echo -n "[*] Checking if io_uring_clone_buffers has src_off.. "
rm -f "$TMP" || exit 1

cat >"$TMP.c" << EOF
#include <linux/io_uring.h>

int main(void)
{
	struct io_uring_clone_buffers b = { .src_off = 0, .dst_off = 0, .nr = 0 };
	return b.src_off + b.dst_off + b.nr;
}
EOF

${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo $RED "[NO]" $COL_RESET
	MISSING_DEFS=1
else
	echo $GREEN "[YES]" $COL_RESET
	echo "#define USE_IO_URING_CLONE_BUFFERS_OFF 1" >> $CONFIGH
fi

#############################################################################################
# Does linux/io_uring.h give io_uring_napi the opcode/op_param members
# (added in 6.13)
#
echo -n "[*] Checking if io_uring_napi has opcode/op_param.. "
rm -f "$TMP" || exit 1

cat >"$TMP.c" << EOF
#include <linux/io_uring.h>

int main(void)
{
	struct io_uring_napi n = { .opcode = 0, .op_param = 0 };
	return n.opcode + n.op_param;
}
EOF

${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo $RED "[NO]" $COL_RESET
	MISSING_DEFS=1
else
	echo $GREEN "[YES]" $COL_RESET
	echo "#define USE_IO_URING_NAPI_OP 1" >> $CONFIGH
fi

#############################################################################################
# Does linux/bpf.h give union bpf_attr the fd_by_id_token_fd member
# (added in 6.13)
#
echo -n "[*] Checking if bpf_attr has fd_by_id_token_fd.. "
rm -f "$TMP" || exit 1

cat >"$TMP.c" << EOF
#include <linux/bpf.h>

int main(void)
{
	union bpf_attr a = { .fd_by_id_token_fd = 0 };
	return a.fd_by_id_token_fd;
}
EOF

${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo $RED "[NO]" $COL_RESET
	MISSING_DEFS=1
else
	echo $GREEN "[YES]" $COL_RESET
	echo "#define USE_BPF_FD_BY_ID_TOKEN_FD 1" >> $CONFIGH
fi

#############################################################################################
# Does linux/bpf.h give union bpf_attr the prog_stream_read member
# (BPF_PROG_STREAM_READ_BY_FD, added post-6.18)
#
echo -n "[*] Checking if bpf_attr has prog_stream_read.. "
rm -f "$TMP" || exit 1

cat >"$TMP.c" << EOF
#include <linux/bpf.h>

int main(void)
{
	union bpf_attr a = { .prog_stream_read.stream_id = 0 };
	return a.prog_stream_read.stream_id;
}
EOF

${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo $RED "[NO]" $COL_RESET
	MISSING_DEFS=1
else
	echo $GREEN "[YES]" $COL_RESET
	echo "#define USE_BPF_PROG_STREAM_READ 1" >> $CONFIGH
fi

#############################################################################################
# Does linux/bpf.h give union bpf_attr the link_create.cgroup member
# (added in 6.13)
#
echo -n "[*] Checking if bpf_attr has link_create.cgroup.. "
rm -f "$TMP" || exit 1

cat >"$TMP.c" << EOF
#include <linux/bpf.h>

int main(void)
{
	union bpf_attr a = { .link_create.cgroup.relative_fd = 0 };
	return a.link_create.cgroup.relative_fd;
}
EOF

${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo $RED "[NO]" $COL_RESET
	MISSING_DEFS=1
else
	echo $GREEN "[YES]" $COL_RESET
	echo "#define USE_BPF_LINK_CREATE_CGROUP 1" >> $CONFIGH
fi

#############################################################################################
# Does linux/iommufd.h give the IOMMU_HWPT_ALLOC_PASID flag
# (added in 6.14). The flag is an enum value, not a #define, so we
# need a compile test rather than #ifdef.
#
echo -n "[*] Checking if iommufd has IOMMU_HWPT_ALLOC_PASID.. "
rm -f "$TMP" || exit 1

cat >"$TMP.c" << EOF
#include <linux/iommufd.h>

int main(void)
{
	return IOMMU_HWPT_ALLOC_PASID;
}
EOF

${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo $RED "[NO]" $COL_RESET
	MISSING_DEFS=1
else
	echo $GREEN "[YES]" $COL_RESET
	echo "#define HAVE_IOMMU_HWPT_ALLOC_PASID 1" >> $CONFIGH
fi

#############################################################################################
# Does linux/iommufd.h give the IOMMU_HW_INFO_FLAG_INPUT_TYPE flag
# and the in_data_type member of iommu_hw_info (added in 6.14).
# Enum value, so compile-test.
#
echo -n "[*] Checking if iommufd has IOMMU_HW_INFO_FLAG_INPUT_TYPE.. "
rm -f "$TMP" || exit 1

cat >"$TMP.c" << EOF
#include <linux/iommufd.h>

int main(void)
{
	struct iommu_hw_info h = { .in_data_type = 0 };
	return h.in_data_type + IOMMU_HW_INFO_FLAG_INPUT_TYPE;
}
EOF

${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo $RED "[NO]" $COL_RESET
	MISSING_DEFS=1
else
	echo $GREEN "[YES]" $COL_RESET
	echo "#define HAVE_IOMMU_HW_INFO_INPUT_TYPE 1" >> $CONFIGH
fi

#############################################################################################
# struct mnt_id_req's reserved offset-4 __u32 is named `spare' on older
# linux-libc-dev and `mnt_ns_fd' on newer headers (e.g. Debian 13).
# Same slot, different name -- compile-test which one this host has so
# statmount.c writes the STATMOUNT_BY_FD fd to the member that exists.
#
echo -n "[*] Checking mnt_id_req offset-4 field name.. "
rm -f "$TMP" || exit 1

cat >"$TMP.c" << EOF
#include <linux/mount.h>

int main(void)
{
	struct mnt_id_req r = { .mnt_ns_fd = 0 };
	return (int) r.mnt_ns_fd;
}
EOF

${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo "[spare]"
else
	echo $GREEN "[mnt_ns_fd]" $COL_RESET
	echo "#define HAVE_MNT_ID_REQ_MNT_NS_FD 1" >> $CONFIGH
fi

#############################################################################################
# xfrm UAPI enum probes -- XFRMA_SA_DIR, XFRMA_NAT_KEEPALIVE_INTERVAL, and
# XFRM_SA_DIR_{IN,OUT} are enumerators, not macros; #ifndef guards are always
# true and the shim fires regardless.  Probe via compile-test so the xfrm
# headers can switch to #ifndef HAVE_* guards that correctly reflect
# whether the kernel UAPI headers already supply the values.
#
echo -n "[*] Checking for XFRM_SA_DIR_{IN,OUT} (enum xfrm_sa_dir).. "
rm -f "$TMP" || exit 1
cat >"$TMP.c" << EOF
#include <linux/xfrm.h>
int main(void) { int x = XFRM_SA_DIR_OUT; return x; }
EOF
${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo "[missing]"
else
	echo $GREEN "[present]" $COL_RESET
	echo "#define HAVE_XFRM_SA_DIR 1" >> $CONFIGH
fi

echo -n "[*] Checking for XFRMA_SA_DIR (enum xfrm_attr_type_t).. "
rm -f "$TMP" || exit 1
cat >"$TMP.c" << EOF
#include <linux/xfrm.h>
int main(void) { int x = XFRMA_SA_DIR; return x; }
EOF
${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo "[missing]"
else
	echo $GREEN "[present]" $COL_RESET
	echo "#define HAVE_XFRMA_SA_DIR 1" >> $CONFIGH
fi

echo -n "[*] Checking for XFRMA_NAT_KEEPALIVE_INTERVAL (enum xfrm_attr_type_t).. "
rm -f "$TMP" || exit 1
cat >"$TMP.c" << EOF
#include <linux/xfrm.h>
int main(void) { int x = XFRMA_NAT_KEEPALIVE_INTERVAL; return x; }
EOF
${CC} ${CFLAGS} "$TMP.c" -o "$TMP" &>"$TMP.log"
if [ ! -x "$TMP" ]; then
	echo "[missing]"
else
	echo $GREEN "[present]" $COL_RESET
	echo "#define HAVE_XFRMA_NAT_KEEPALIVE_INTERVAL 1" >> $CONFIGH
fi

#############################################################################################

check_header linux/fsmap.h USE_FSMAP
check_header linux/if_alg.h USE_IF_ALG
check_header linux/rds.h USE_RDS
check_header linux/vfio.h USE_VFIO
check_header linux/iommufd.h USE_IOMMUFD
check_header linux/vduse.h USE_VDUSE
check_header drm/drm.h USE_DRM
check_header drm/exynos_drm.h USE_DRM_EXYNOS
check_header sound/compress_offload.h USE_SNDDRV_COMPRESS_OFFLOAD
check_header linux/kvm.h USE_KVM
check_header linux/seccomp.h USE_SECCOMP
check_header linux/vhost.h USE_VHOST
check_header execinfo.h USE_BACKTRACE
check_header bluetooth/bluetooth.h USE_BLUETOOTH
check_header linux/vm_sockets.h USE_VSOCK
check_header linux/if_pppox.h USE_PPPOX
check_header linux/can.h USE_CAN
check_header linux/rxrpc.h USE_RXRPC
check_header linux/llc.h USE_LLC
check_header linux/if_xdp.h USE_XDP
check_header linux/mctp.h USE_MCTP
check_header linux/sctp.h USE_SCTP
check_header linux/tcp.h USE_TCP_REPAIR_OPT
check_header linux/x25.h USE_X25
check_header linux/phonet.h USE_PHONET
check_header linux/atm.h USE_ATM

#############################################################################################
# Optional: guard-page armour for alloc_shared() regions.  Off-by-default
# hunt instrument that wraps tagged shared regions in PROT_NONE pages so
# a forward overflow (buf[size]) faults at the writer PC instead of
# scribbling an adjacent pool and crashing an unrelated consumer many
# operations later.
#
# Opt in at configure time with GUARD_SHARED=1.  Without the build flag
# the guard machinery compiles out entirely and alloc_shared keeps its
# legacy single-mmap path; with the build flag the runtime --guard-shared
# scope (off|pools|all, default off) gates whether any guarding fires.
echo -n "[*] Guard-page armour for alloc_shared (GUARD_SHARED): "
if [ "${GUARD_SHARED:-0}" = "1" ]; then
	echo $GREEN "[ON]" $COL_RESET
	echo "#define CONFIG_GUARD_SHARED 1" >> $CONFIGH
else
	echo "[off]"
fi

#############################################################################################

if [ "$MISSING_DEFS" == "1" ]; then
  echo "[-] Some header definitions were missing. This is not fatal."
  echo "    It usually means you're building on an older distribution which doesn't"
  echo "    have header files describing newer kernel features."
  echo "    Trinity will still compile and run, it just won't use those new features."
  echo "    Go ahead, and run 'make'"
fi

exit 0