/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright (c) 2019, Joyent, Inc.
 * Copyright 2021 OmniOS Community Edition (OmniOSce) Association.
 */

/*
 * Check that we properly handle functions and function pointers.
 */

#include "check-common.h"

static const char *one_args[] = { "int" };
static const char *two_args[] = { "int", "const char *" };
static const char *three_args[] = { "int", "const char *", "float" };
static const char *argument_args[] = { "uintptr_t" };
static const char *vararg_args[] = { "const char *" };
static const char *vla1_args[] = { "int", "int *" };
static const char *vla2_args[] = { "int", "int (*)[0]" };
static const char *vla3_args[] = { "int", "int (*)[7]" };
static const char *vla4_args[] = { "int", "int (*)[0]" };
static const char *vla5_args[] = { "int", "int", "int (*)[3][0]" };
static const char *vla6_args[] = { "int", "int", "int (*)[0][4]" };


static check_function_test_t functions[] = {
        { "simple_func", "void", 0, 0, NULL },
        { "one", "void", 1, 0, one_args },
        { "two", "void", 2, 0, two_args },
        { "three", "void", 3, 0, three_args },
        { "noarg", "const char *", 0, 0, NULL },
        { "argument", "const char *", 1, 0, argument_args },
        { "vararg", "void", 1, CTF_FUNC_VARARG, vararg_args },
        { "vararg_ret", "uintptr_t", 1, CTF_FUNC_VARARG, vararg_args },
        { "vla1", "int", 2, 0, vla1_args },
        { "vla2", "int", 2, 0, vla2_args },
        { "vla3", "int", 2, 0, vla3_args },
        { "vla4", "int", 2, 0, vla4_args },
        { "vla5", "int", 3, 0, vla5_args },
        { "vla6", "int", 3, 0, vla6_args },
        { NULL }
};

static const char *strfunc_args[] = { "const char *", "const char *" };

static check_function_test_t fptrs[] = {
        { "strfunc_t", "int", 2, 0, strfunc_args },
        { "vararg_t", "void", 1, CTF_FUNC_VARARG, vararg_args },
        { NULL }
};

int
main(int argc, char *argv[])
{
        int i, ret = 0;

        if (argc < 2) {
                errx(EXIT_FAILURE, "missing test files");
        }

        for (i = 1; i < argc; i++) {
                ctf_file_t *fp;
                uint_t j;

                if ((fp = ctf_open(argv[i], &ret)) == NULL) {
                        warnx("failed to open %s: %s", argv[i],
                            ctf_errmsg(ret));
                        ret = EXIT_FAILURE;
                        continue;
                }

                for (j = 0; functions[j].cft_name != NULL; j++) {
                        if (!ctftest_check_function(functions[j].cft_name, fp,
                            functions[j].cft_rtype, functions[j].cft_nargs,
                            functions[j].cft_flags, functions[j].cft_args)) {
                                ret = EXIT_FAILURE;
                        }
                }

                for (j = 0; fptrs[j].cft_name != NULL; j++) {
                        if (!ctftest_check_fptr(fptrs[j].cft_name, fp,
                            fptrs[j].cft_rtype, fptrs[j].cft_nargs,
                            fptrs[j].cft_flags, fptrs[j].cft_args)) {
                                ret = EXIT_FAILURE;
                        }
                }

                ctf_close(fp);
        }

        return (ret);
}