#!/usr/bin/env python

import os
from waflib import Logs

# quic_ko_wrapper has no system library yet
VERSION = "0.99.0"

def configure(conf):
    if not conf.CONFIG_GET('HAVE_LIBQUIC'):
        Logs.warn('quic_ko_wrapper requires libquic (and libngtcp2)')
        return

    if not conf.CONFIG_GET('HAVE_LIBNGTCP2'):
        Logs.warn('quic_ko_wrapper requires libngtcp2 (and libquic)')
        return

    # quic_ko_wrapper has no system library yet
    # so no USING_SYSTEM_QUIC_KO_WRAPPER
    if True:
        conf.CHECK_HEADERS('gnu/lib-names.h')

        if conf.CONFIG_SET("HAVE___THREAD"):
            conf.DEFINE("HAVE_GCC_THREAD_LOCAL_STORAGE", 1)

        # check HAVE_FUNCTION_ATTRIBUTE_FORMAT
        conf.CHECK_CODE('''
            void log_fn(const char *format, ...) __attribute__ ((format (printf, 1, 2)));

            int main(void) {
                return 0;
            }
            ''',
            'HAVE_FUNCTION_ATTRIBUTE_FORMAT',
            addmain=False,
            strict=True,
            msg='Checking for printf format validation support')

        conf.CHECK_HEADERS('sys/signalfd.h')
        conf.CHECK_HEADERS('sys/eventfd.h')
        conf.CHECK_HEADERS('sys/timerfd.h')
        conf.CHECK_HEADERS('gnu/lib-names.h')

        conf.CHECK_STRUCTURE_MEMBER('struct msghdr',
                                    'msg_control',
                                    headers='sys/types.h sys/socket.h',
                                    define='HAVE_STRUCT_MSGHDR_MSG_CONTROL')

        conf.CHECK_FUNCS('getaddrinfo')
        conf.CHECK_FUNCS('signalfd eventfd timerfd_create')
        conf.CHECK_FUNCS('bindresvport')
        conf.CHECK_FUNCS('pledge')
        conf.CHECK_FUNCS('accept4')
        conf.CHECK_FUNCS('sendmmsg recvmmsg')
        conf.CHECK_FUNCS('__close_nocancel')
        conf.CHECK_FUNCS('openat64')
        conf.CHECK_FUNCS('fopen64')
        conf.CHECK_FUNCS('open64')
        conf.CHECK_FUNCS('fcntl64')

        conf.CHECK_FUNCS_IN('bind',
                            'socket',
                            checklibc=True,
                            headers='sys/types.h sys/socket.h')

        conf.CHECK_C_PROTOTYPE('accept',
                               'int accept(int s, struct sockaddr *addr, Psocklen_t addrlen)',
                               define='HAVE_ACCEPT_PSOCKLEN_T', headers='sys/types.h sys/socket.h')

        # Linux legacy glibc < 2.21
        conf.CHECK_C_PROTOTYPE('recvmmsg',
                               'int recvmmsg(int __fd, struct mmsghdr *__vmessages, unsigned int __vlen, int __flags, const struct timespec *__tmo)',
                               define='HAVE_RECVMMSG_CONST_TIMEOUT', headers='sys/types.h sys/socket.h')
        # FreeBSD
        conf.CHECK_C_PROTOTYPE('recvmmsg',
                               'ssize_t recvmmsg(int __fd, struct mmsghdr * __restrict __vmessages, size_t __vlen, int __flags, const struct timespec * __restrict __tmo)',
                               define='HAVE_RECVMMSG_SSIZE_T_CONST_TIMEOUT', headers='sys/types.h sys/socket.h')
        # FreeBSD
        conf.CHECK_C_PROTOTYPE('sendmmsg',
                               'ssize_t sendmmsg(int __fd, struct mmsghdr * __restrict __vmessages, size_t __vlen, int __flags)',
                               define='HAVE_SENDMMSG_SSIZE_T', headers='sys/types.h sys/socket.h')

        conf.CHECK_C_PROTOTYPE('ioctl',
                               'int ioctl(int s, int r, ...)',
                               define='HAVE_IOCTL_INT', headers='unistd.h sys/ioctl.h')

        if conf.CONFIG_SET("HAVE_EVENTFD"):
            conf.CHECK_C_PROTOTYPE('eventfd',
                                   'int eventfd(unsigned int count, int flags)',
                                   define='HAVE_EVENTFD_UNSIGNED_INT', headers='sys/eventfd.h')

        # FreeBSD exports each syscall also with '_' as prefix
        # and these symbols are used if called by system libraries itself.
        # That means quic_ko_wrapper needs to implement these too
        # in order to inject itself into system libraries,
        # we just check for _socket and _close and assume the rest
        # is also there...
        conf.CHECK_FUNCS('_socket _close')

        # Create full path to quic_ko_wrapper
        blddir = os.path.realpath(conf.bldnode.abspath())
        libquic_ko_wrapper_so_path = blddir + '/default/third_party/quic_ko_wrapper/libquic-ko-wrapper.so'

    conf.DEFINE('LIBQUIC_KO_WRAPPER_SO_PATH', libquic_ko_wrapper_so_path)
    conf.DEFINE('QUIC_KO_WRAPPER', 1)

def build(bld):
    # We need to do it this way or the library wont work.
    # We need force_unversioned=True as symbol versioning
    # breaks preloading!
    bld.SAMBA_LIBRARY('quic_ko_wrapper',
                      source='quic_ko_wrapper.c',
                      cflags='-D%s="%s" -D%s="%s"' % (
                          "QUIC_KO_WRAPPER_PACKAGE",
                          "samba_quic_ko_wrapper",
                          "QUIC_KO_WRAPPER_VERSION",
                          VERSION),
                      deps='dl pthread gnutls libquic libngtcp2 libngtcp2_crypto_gnutls',
                      install=False,
                      force_unversioned=True,
                      realname='libquic-ko-wrapper.so')