extension/testarg.c


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

#include "awk.h"

int plugin_is_GPL_compatible;

static NODE *
do_check_arg(int nargs)
{
        int ret = 0, argc;
        NODE *arg1, *arg2, *arg3;

        argc = get_curfunc_arg_count();
        printf("arg count: defined = %d, supplied = %d\n",
                   nargs, argc);

        arg1 = get_scalar_argument(0, FALSE);
        arg2 = get_array_argument(1, FALSE);
        arg3 = get_scalar_argument(2, TRUE);	/* optional */
        if (argc > 3) {	/* try to use an extra arg */
                NODE *arg4;
                arg4 = get_array_argument(3, TRUE);
        }
        if (arg3 != NULL)
                printf("3rd arg present\n\n");
        else
                printf("no 3rd arg\n\n");

        /* Set the return value */
        return make_number((AWKNUM) ret);
}

/* dlload --- load new builtins in this library */

NODE *
dlload(tree, dl)
NODE *tree;
void *dl;
{
        make_builtin("check_arg", do_check_arg, 3);
        return make_number((AWKNUM) 0);
}