datalab/web/node_modules/dtrace-provider/libusdt/test_usdt.c

/*
 * Copyright (c) 2012, Chris Andrews. All rights reserved.
 */

#include "usdt.h"

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

static void
fire_probe(usdt_probedef_t *probedef, int argc, void **argv)
{
        if (usdt_is_enabled(probedef->probe))
                usdt_fire_probe(probedef->probe, argc, argv);
}

int main(int argc, char **argv) {
        usdt_provider_t *provider;
        usdt_probedef_t *probedef;
        char char_argv[USDT_ARG_MAX];
        int int_argv[USDT_ARG_MAX * 2];
        void **args = NULL;
        int i;
        char buf[255];

        for (i = 0; i < USDT_ARG_MAX; i++)
                int_argv[i] = i + 1;
        for (i = 0; i < USDT_ARG_MAX; i++)
                char_argv[i] = (char) i + 65;

        if (argc < 3) {
                fprintf(stderr, "usage: %s func name [types ...]\n", argv[0]);
                return(1);
        }

        if (argc > 3) {
                args = malloc((argc-3) * sizeof(void *));
        }

        for (i = 0; i < USDT_ARG_MAX; i++) {
                if (argv[i+3] != NULL && i+3 < argc) {
                        if (strncmp("c", argv[i+3], 1) == 0) {
                                args[i] = (void *)strndup(&char_argv[i], 1);
                                argv[i+3] = strdup("char *");
                        }
                        if (strncmp("i", argv[i+3], 1) == 0) {
                                args[i] = (void *)(long)int_argv[i];
                                argv[i+3] = strdup("int");
                        }
                }
        }

        if ((provider = usdt_create_provider("testlibusdt", "modname")) == NULL) {
                fprintf(stderr, "unable to create provider\n");
                exit (1);
        }
        if ((probedef = usdt_create_probe((const char *)argv[1],
                                          (const char *)argv[2],
                                          (argc-3), (const char **)&argv[3])) == NULL)
        {
                fprintf(stderr, "unable to create probe\n");
                exit (1);
        }
        usdt_provider_add_probe(provider, probedef);

        if ((usdt_provider_enable(provider)) < 0) {
                fprintf(stderr, "unable to enable provider: %s\n", usdt_errstr(provider));
                exit (1);
        }

        fprintf(stdout, "enabled\n");
        fflush(stdout);
        fgets(buf, 255, stdin);

        fire_probe(probedef, (argc-3), (void **)args);
        usdt_probe_release(probedef);

        if ((usdt_provider_disable(provider)) < 0) {
                fprintf(stderr, "unable to disable provider: %s\n", usdt_errstr(provider));
                exit (1);
        }

        usdt_provider_free(provider);

        return 0;
}