88 lines
2.7 KiB
C
88 lines
2.7 KiB
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;
|
|
}
|