ruby-usdt 0.1.2 → 0.2.0

data/ext/libusdt/test_mem_usage.c

@@ -1,77 +0,0 @@
-/*
- * Copyright (c) 2012, Chris Andrews. All rights reserved.
- */
-
-#include "usdt.h"
-
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-
-static void
-create_and_free_provider(int argc, char **argv)
-{
-        usdt_provider_t *provider;
-        usdt_probedef_t *probedef;
-
-        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);
-        }
-
-        if ((usdt_provider_disable(provider)) < 0) {
-                fprintf(stderr, "unable to disable provider: %s\n", usdt_errstr(provider));
-                exit (1);
-        }
-
-        usdt_probe_release(probedef);
-        usdt_provider_free(provider);
-}
-
-int
-main(int argc, char **argv)
-{
-        char char_argv[USDT_ARG_MAX];
-        int int_argv[USDT_ARG_MAX * 2];
-        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);
-        }
-
-        for (i = 0; i < USDT_ARG_MAX; i++) {
-                if (argv[i+3] != NULL && i+3 < argc) {
-                        if (strncmp("c", argv[i+3], 1) == 0) {
-                                argv[i+3] = strdup("char *");
-                        }
-                        if (strncmp("i", argv[i+3], 1) == 0) {
-                                argv[i+3] = strdup("int");
-                        }
-                }
-        }
-
-        for (i = 0; i < 100000; i++)
-                create_and_free_provider(argc, argv);
-
-        return 0;
-}

data/ext/libusdt/test_usdt.c

@@ -1,87 +0,0 @@
-/*
- * 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;
-}

data/ext/libusdt/usdt.c

@@ -1,296 +0,0 @@
-/*
- * Copyright (c) 2012, Chris Andrews. All rights reserved.
- */
-
-#include "usdt_internal.h"
-
-#include <stdlib.h>
-#include <stdarg.h>
-#include <string.h>
-#include <errno.h>
-#include <stdio.h>
-
-char *usdt_errors[] = {
-  "failed to allocate memory",
-  "failed to allocate page-aligned memory",
-  "no probes defined",
-  "failed to load DOF: %s",
-  "provider is already enabled",
-  "failed to unload DOF: %s",
-  "probe named %s:%s:%s:%s already exists",
-  "failed to remove probe %s:%s:%s:%s"
-};
-
-static void
-free_probedef(usdt_probedef_t *pd)
-{
-        switch (pd->refcnt) {
-        case 1:
-                free((char *)pd->function);
-                free((char *)pd->name);
-                if (pd->probe) {
-                        free(pd->probe->isenabled_addr);
-                        free(pd->probe);
-                }
-                free(pd);
-                break;
-        case 2:
-                pd->refcnt = 1;
-                break;
-        default:
-                break;
-        }
-}
-
-usdt_provider_t *
-usdt_create_provider(const char *name, const char *module)
-{
-        usdt_provider_t *provider;
-
-        if ((provider = malloc(sizeof *provider)) == NULL) 
-                return NULL;
-
-        provider->name = strdup(name);
-        provider->module = strdup(module);
-        provider->probedefs = NULL;
-        provider->enabled = 0;
-
-        return provider;
-}
-
-usdt_probedef_t *
-usdt_create_probe(const char *func, const char *name, size_t argc, const char **types)
-{
-        int i;
-        usdt_probedef_t *p;
-
-        if (argc > USDT_ARG_MAX)
-                argc = USDT_ARG_MAX;
-
-        if ((p = malloc(sizeof *p)) == NULL)
-                return (NULL);
-
-        p->refcnt = 2;
-        p->function = strdup(func);
-        p->name = strdup(name);
-        p->argc = argc;
-        p->probe = NULL;
-
-        for (i = 0; i < argc; i++) {
-                if (strncmp("char *", types[i], 6) == 0)
-                        p->types[i] = USDT_ARGTYPE_STRING;
-                if (strncmp("int", types[i], 3) == 0)
-                        p->types[i] = USDT_ARGTYPE_INTEGER;
-        }
-
-        return (p);
-}
-
-void
-usdt_probe_release(usdt_probedef_t *probedef)
-{
-        free_probedef(probedef);
-}
-
-int
-usdt_provider_add_probe(usdt_provider_t *provider, usdt_probedef_t *probedef)
-{
-        usdt_probedef_t *pd;
-
-        if (provider->probedefs != NULL) {
-                for (pd = provider->probedefs; (pd != NULL); pd = pd->next) {
-                if ((strcmp(pd->name, probedef->name) == 0) &&
-                    (strcmp(pd->function, probedef->function) == 0)) {
-                                usdt_error(provider, USDT_ERROR_DUP_PROBE,
-                                           provider->name, provider->module,
-                                           probedef->function, probedef->name);
-                                return (-1);
-                        }
-                }
-        }
-
-        probedef->next = NULL;
-        if (provider->probedefs == NULL)
-                provider->probedefs = probedef;
-        else {
-                for (pd = provider->probedefs; (pd->next != NULL); pd = pd->next) ;
-                pd->next = probedef;
-        }
-
-        return (0);
-}
-
-int
-usdt_provider_remove_probe(usdt_provider_t *provider, usdt_probedef_t *probedef)
-{
-        usdt_probedef_t *pd, *prev_pd = NULL;
-
-        if (provider->probedefs == NULL) {
-                usdt_error(provider, USDT_ERROR_NOPROBES);
-                return (-1);
-        }
-
-        for (pd = provider->probedefs; (pd != NULL);
-             prev_pd = pd, pd = pd->next) {
-
-                if ((strcmp(pd->name, probedef->name) == 0) &&
-                    (strcmp(pd->function, probedef->function) == 0)) {
-
-                        if (prev_pd == NULL)
-                                provider->probedefs = pd->next;
-                        else
-                                prev_pd->next = pd->next;
-
-                        return (0);
-                }
-        }
-
-        usdt_error(provider, USDT_ERROR_REMOVE_PROBE,
-                   provider->name, provider->module,
-                   probedef->function, probedef->name);
-        return (-1);
-}
-
-int
-usdt_provider_enable(usdt_provider_t *provider)
-{
-        usdt_strtab_t strtab;
-        usdt_dof_file_t *file;
-        usdt_probedef_t *pd;
-        int i;
-        size_t size;
-        usdt_dof_section_t sects[5];
-
-        if (provider->enabled == 1) {
-                usdt_error(provider, USDT_ERROR_ALREADYENABLED);
-                return (0); /* not fatal */
-        }
-
-        if (provider->probedefs == NULL) {
-                usdt_error(provider, USDT_ERROR_NOPROBES);
-                return (-1);
-        }
-
-        for (pd = provider->probedefs; pd != NULL; pd = pd->next) {
-                if ((pd->probe = malloc(sizeof(*pd->probe))) == NULL) {
-                        usdt_error(provider, USDT_ERROR_MALLOC);
-                        return (-1);
-                }
-        }
-
-        if ((usdt_strtab_init(&strtab, 0)) < 0) {
-                usdt_error(provider, USDT_ERROR_MALLOC);
-                return (-1);
-        }
-
-        if ((usdt_strtab_add(&strtab, provider->name)) == 0) {
-                usdt_error(provider, USDT_ERROR_MALLOC);
-                return (-1);
-        }
-
-        if ((usdt_dof_probes_sect(&sects[0], provider, &strtab)) < 0)
-                return (-1);
-        if ((usdt_dof_prargs_sect(&sects[1], provider)) < 0)
-                return (-1);
-
-        size = usdt_provider_dof_size(provider, &strtab);
-        if ((file = usdt_dof_file_init(provider, size)) == NULL)
-                return (-1);
-
-        if ((usdt_dof_proffs_sect(&sects[2], provider, file->dof)) < 0)
-                return (-1);
-        if ((usdt_dof_prenoffs_sect(&sects[3], provider, file->dof)) < 0)
-                return (-1);
-        if ((usdt_dof_provider_sect(&sects[4], provider)) < 0)
-                return (-1);
-
-        for (i = 0; i < 5; i++)
-                usdt_dof_file_append_section(file, &sects[i]);
-
-        usdt_dof_file_generate(file, &strtab);
-
-        usdt_dof_section_free(&strtab);
-        for (i = 0; i < 5; i++)
-                usdt_dof_section_free(&sects[i]);
-
-        if ((usdt_dof_file_load(file, provider->module)) < 0) {
-                usdt_error(provider, USDT_ERROR_LOADDOF, strerror(errno));
-                return (-1);
-        }
-
-        provider->enabled = 1;
-        provider->file = file;
-
-        return (0);
-}
-
-int
-usdt_provider_disable(usdt_provider_t *provider)
-{
-        if (provider->enabled == 0)
-                return (0);
-
-        if ((usdt_dof_file_unload((usdt_dof_file_t *)provider->file)) < 0) {
-                usdt_error(provider, USDT_ERROR_UNLOADDOF, strerror(errno));
-                return (-1);
-        }
-
-        usdt_dof_file_free(provider->file);
-        provider->file = NULL;
-        provider->enabled = 0;
-
-        return (0);
-}
-
-void
-usdt_provider_free(usdt_provider_t *provider)
-{
-        usdt_probedef_t *pd, *next;
-
-        for (pd = provider->probedefs; pd != NULL; pd = next) {
-                next = pd->next;
-                free_probedef(pd);
-        }
-
-        free((char *)provider->name);
-        free((char *)provider->module);
-        free(provider);
-}
-
-int
-usdt_is_enabled(usdt_probe_t *probe)
-{
-        if (probe != NULL)
-                return (*probe->isenabled_addr)();
-        else
-                return 0;
-}
-
-void
-usdt_fire_probe(usdt_probe_t *probe, size_t argc, void **nargv)
-{
-        if (probe != NULL)
-                usdt_probe_args(probe->probe_addr, argc, nargv);
-}
-
-static void
-usdt_verror(usdt_provider_t *provider, usdt_error_t error, va_list argp)
-{
-        vasprintf(&provider->error, usdt_errors[error], argp);
-}
-
-void
-usdt_error(usdt_provider_t *provider, usdt_error_t error, ...)
-{
-        va_list argp;
-
-        va_start(argp, error);
-        usdt_verror(provider, error, argp);
-        va_end(argp);
-}
-
-char *
-usdt_errstr(usdt_provider_t *provider)
-{
-        return (provider->error);
-}

data/ext/libusdt/usdt.h

@@ -1,66 +0,0 @@
-/*
- * Copyright (c) 2012, Chris Andrews. All rights reserved.
- */
-
-#include <stdint.h>
-#include <unistd.h>
-
-typedef uint8_t usdt_argtype_t;
-#define USDT_ARGTYPE_NONE    0
-#define USDT_ARGTYPE_STRING  1
-#define USDT_ARGTYPE_INTEGER 2
-
-#define USDT_ARG_MAX 32
-
-typedef enum usdt_error {
-        USDT_ERROR_MALLOC = 0,
-        USDT_ERROR_VALLOC,
-        USDT_ERROR_NOPROBES,
-        USDT_ERROR_LOADDOF,
-        USDT_ERROR_ALREADYENABLED,
-        USDT_ERROR_UNLOADDOF,
-        USDT_ERROR_DUP_PROBE,
-        USDT_ERROR_REMOVE_PROBE
-} usdt_error_t;
-
-typedef struct usdt_probe {
-        int (*isenabled_addr)(void);
-        void *probe_addr;
-} usdt_probe_t;
-
-int usdt_is_enabled(usdt_probe_t *probe);
-void usdt_fire_probe(usdt_probe_t *probe, size_t argc, void **argv);
-
-typedef struct usdt_probedef {
-        const char *name;
-        const char *function;
-        size_t argc;
-        usdt_argtype_t types[USDT_ARG_MAX];
-        struct usdt_probe *probe;
-        struct usdt_probedef *next;
-        int refcnt;
-} usdt_probedef_t;
-
-usdt_probedef_t *usdt_create_probe(const char *func, const char *name,
-                                   size_t argc, const char **types);
-void usdt_probe_release(usdt_probedef_t *probedef);
-
-typedef struct usdt_provider {
-        const char *name;
-        const char *module;
-        usdt_probedef_t *probedefs;
-        char *error;
-        int enabled;
-        void *file;
-} usdt_provider_t;
-
-usdt_provider_t *usdt_create_provider(const char *name, const char *module);
-int usdt_provider_add_probe(usdt_provider_t *provider, usdt_probedef_t *probedef);
-int usdt_provider_remove_probe(usdt_provider_t *provider, usdt_probedef_t *probedef);
-int usdt_provider_enable(usdt_provider_t *provider);
-int usdt_provider_disable(usdt_provider_t *provider);
-void usdt_provider_free(usdt_provider_t *provider);
-
-void usdt_error(usdt_provider_t *provider, usdt_error_t error, ...);
-char *usdt_errstr(usdt_provider_t *provider);
-

data/ext/libusdt/usdt_dof.c

@@ -1,128 +0,0 @@
-/*
- * Copyright (c) 2012, Chris Andrews. All rights reserved.
- */
-
-#include "usdt_internal.h"
-
-#include <stdlib.h>
-
-int
-usdt_dof_section_add_data(usdt_dof_section_t *section, void *data, size_t length)
-{
-        int newlen = section->size + length;
-
-        if ((section->data = realloc((char *)section->data, newlen)) == NULL)
-                return (-1);
-
-        memcpy(section->data + section->size, data, length);
-        section->size = newlen;
-        return (0);
-}
-
-size_t
-usdt_provider_dof_size(usdt_provider_t *provider, usdt_strtab_t *strtab)
-{
-        uint8_t i, j;
-        int args = 0;
-        int probes = 0;
-        size_t size = 0;
-        usdt_probedef_t *pd;
-        size_t sections[8];
-
-        for (pd = provider->probedefs; pd != NULL; pd = pd->next) {
-                args += pd->argc;
-                probes++;
-        }
-
-        sections[0] = sizeof(dof_hdr_t);
-        sections[1] = sizeof(dof_sec_t) * 6;
-        sections[2] = strtab->size;
-        sections[3] = sizeof(dof_probe_t) * probes;
-        sections[4] = sizeof(uint8_t) * args;
-        sections[5] = sizeof(uint32_t) * probes;
-        sections[6] = sizeof(uint32_t) * probes;
-        sections[7] = sizeof(dof_provider_t);
-
-        for (i = 0; i < 8; i++) {
-                size += sections[i];
-                j = size % 8;
-                if (j > 0)
-                        size += (8 - j);
-        }
-
-        return size;
-}
-
-int
-usdt_dof_section_init(usdt_dof_section_t *section, uint32_t type, dof_secidx_t index)
-{
-        section->type    = type;
-        section->index   = index;
-        section->flags   = DOF_SECF_LOAD;
-        section->offset  = 0;
-        section->size    = 0;
-        section->entsize = 0;
-        section->pad	 = 0;
-        section->next    = NULL;
-
-        if ((section->data = malloc(1)) == NULL)
-                return (-1);
-
-        switch(type) {
-        case DOF_SECT_PROBES:   section->align = 8; break;
-        case DOF_SECT_PRARGS:   section->align = 1; break;
-        case DOF_SECT_PROFFS:   section->align = 4; break;
-        case DOF_SECT_PRENOFFS: section->align = 4; break;
-        case DOF_SECT_PROVIDER: section->align = 4; break;
-        }
-
-        return (0);
-}
-
-void
-usdt_dof_section_free(usdt_dof_section_t *section)
-{
-        free(section->data);
-}
-
-int
-usdt_strtab_init(usdt_strtab_t *strtab, dof_secidx_t index)
-{
-        strtab->type    = DOF_SECT_STRTAB;;
-        strtab->index   = index;
-        strtab->flags   = DOF_SECF_LOAD;
-        strtab->offset  = 0;
-        strtab->size    = 0;
-        strtab->entsize = 0;
-        strtab->pad	  = 0;
-        strtab->data    = NULL;
-        strtab->align   = 1;
-        strtab->strindex = 1;
-
-        if ((strtab->data = (char *) malloc(1)) == NULL)
-                return (-1);
-
-        *strtab->data = '\0';
-
-        return (0);
-}
-
-dof_stridx_t
-usdt_strtab_add(usdt_strtab_t *strtab, const char *string)
-{
-        size_t length;
-        int index;
-
-        length = strlen(string);
-        index = strtab->strindex;
-        strtab->strindex += (length + 1);
-
-        if ((strtab->data = realloc(strtab->data, strtab->strindex)) == NULL)
-                return (0);
-
-        memcpy((char *) (strtab->data + index), (char *)string, length + 1);
-        strtab->size = index + length + 1;
-
-        return (index);
-}
-