ruby-usdt 0.0.1

data/ext/libusdt/usdt_dof.c

@@ -0,0 +1,118 @@
+#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 += usdt_probedef_argc(pd);
+                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);
+}
+
+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 (-1);
+
+        memcpy((char *) (strtab->data + index), (char *)string, length + 1);
+        strtab->size = index + length + 1;
+
+        return index;
+}
+

data/ext/libusdt/usdt_dof_file.c

@@ -0,0 +1,240 @@
+#include "usdt_internal.h"
+
+#include <stdio.h>
+#include <stdlib.h>
+
+static uint8_t
+dof_version(uint8_t header_version)
+{
+        uint8_t dof_version;
+        /* DOF versioning: Apple always needs version 3, but Solaris can use
+           1 or 2 depending on whether is-enabled probes are needed. */
+#ifdef __APPLE__
+        dof_version = DOF_VERSION_3;
+#else
+        switch(header_version) {
+        case 1:
+                dof_version = DOF_VERSION_1;
+                break;
+        case 2:
+                dof_version = DOF_VERSION_2;
+                break;
+        default:
+                dof_version = DOF_VERSION;
+        }
+#endif
+        return dof_version;
+}
+
+#ifdef __APPLE__
+static const char *helper = "/dev/dtracehelper";
+
+static int
+load_dof(int fd, dof_helper_t *dh)
+{
+        int ret;
+        uint8_t buffer[sizeof(dof_ioctl_data_t) + sizeof(dof_helper_t)];
+        dof_ioctl_data_t* ioctlData = (dof_ioctl_data_t*)buffer;
+        user_addr_t val;
+
+        ioctlData->dofiod_count = 1;
+        memcpy(&ioctlData->dofiod_helpers[0], dh, sizeof(dof_helper_t));
+
+        val = (user_addr_t)(unsigned long)ioctlData;
+        ret = ioctl(fd, DTRACEHIOC_ADDDOF, &val);
+
+        return ret;
+}
+
+#else /* Solaris */
+
+/* ignore Sol10 GA ... */
+static const char *helper = "/dev/dtrace/helper";
+
+static int
+load_dof(int fd, dof_helper_t *dh)
+{
+        return ioctl(fd, DTRACEHIOC_ADDDOF, dh);
+}
+
+#endif
+
+static void
+pad_section(usdt_dof_section_t *sec)
+{
+        size_t i, pad;
+
+        if (sec->align > 1) {
+                i = sec->offset % sec->align;
+                if (i > 0) {
+                        pad = sec->align - i;
+                        sec->offset = (pad + sec->offset);
+                        sec->pad = pad;
+                }
+        }
+}
+
+static void
+dof_header(dof_hdr_t *header)
+{
+        int i;
+
+        header->dofh_ident[DOF_ID_MAG0] = DOF_MAG_MAG0;
+        header->dofh_ident[DOF_ID_MAG1] = DOF_MAG_MAG1;
+        header->dofh_ident[DOF_ID_MAG2] = DOF_MAG_MAG2;
+        header->dofh_ident[DOF_ID_MAG3] = DOF_MAG_MAG3;
+
+        header->dofh_ident[DOF_ID_MODEL]    = DOF_MODEL_NATIVE;
+        header->dofh_ident[DOF_ID_ENCODING] = DOF_ENCODE_NATIVE;
+        header->dofh_ident[DOF_ID_VERSION]  = dof_version(2);
+        header->dofh_ident[DOF_ID_DIFVERS]  = DIF_VERSION;
+        header->dofh_ident[DOF_ID_DIFIREG]  = DIF_DIR_NREGS;
+        header->dofh_ident[DOF_ID_DIFTREG]  = DIF_DTR_NREGS;
+
+        for (i = DOF_ID_PAD; i < DOF_ID_SIZE; i++)
+                header->dofh_ident[i] = 0;
+
+        header->dofh_flags = 0;
+
+        header->dofh_hdrsize = sizeof(dof_hdr_t);
+        header->dofh_secsize = sizeof(dof_sec_t);
+        header->dofh_secoff = sizeof(dof_hdr_t);
+
+        header->dofh_loadsz = 0;
+        header->dofh_filesz = 0;
+        header->dofh_pad = 0;
+}
+
+static size_t
+add_header(usdt_dof_file_t *file, size_t offset, usdt_dof_section_t *section)
+{
+        dof_sec_t header;
+
+        header.dofs_flags   = section->flags;
+        header.dofs_type    = section->type;
+        header.dofs_offset  = section->offset;
+        header.dofs_size    = section->size;
+        header.dofs_entsize = section->entsize;
+        header.dofs_align   = section->align;
+
+        memcpy((file->dof + offset), &header, sizeof(dof_sec_t));
+        return (offset + sizeof(dof_sec_t));
+}
+
+static size_t
+add_section(usdt_dof_file_t *file, size_t offset, usdt_dof_section_t *section)
+{
+        if (section->pad > 0) {
+                memcpy((file->dof + offset), "\0", section->pad);
+                offset += section->pad;
+        }
+
+        memcpy((file->dof + offset), section->data, section->size);
+        return (offset + section->size);
+}
+
+int
+usdt_dof_file_load(usdt_dof_file_t *file, const char *module)
+{
+        dof_helper_t dh;
+        dof_hdr_t *dof;
+        int fd;
+
+        dof = (dof_hdr_t *) file->dof;
+
+        dh.dofhp_dof  = (uintptr_t)dof;
+        dh.dofhp_addr = (uintptr_t)dof;
+        (void) strncpy(dh.dofhp_mod, module, sizeof (dh.dofhp_mod));
+
+        if ((fd = open(helper, O_RDWR)) < 0)
+                return (-1);
+
+        file->gen = load_dof(fd, &dh);
+
+        if ((close(fd)) < 0)
+                return (-1);
+
+        return (0);
+}
+
+void
+usdt_dof_file_append_section(usdt_dof_file_t *file, usdt_dof_section_t *section)
+{
+        usdt_dof_section_t *s;
+
+        if (file->sections == NULL) {
+                file->sections = section;
+        }
+        else {
+                for (s = file->sections; (s->next != NULL); s = s->next) ;
+                s->next = section;
+        }
+}
+
+void
+usdt_dof_file_generate(usdt_dof_file_t *file, usdt_strtab_t *strtab)
+{
+        dof_hdr_t header;
+        uint64_t filesz;
+        uint64_t loadsz;
+        usdt_dof_section_t *sec;
+        size_t offset;
+
+        dof_header(&header);
+        header.dofh_secnum = 6;
+
+        filesz = sizeof(dof_hdr_t) + (sizeof(dof_sec_t) * header.dofh_secnum);
+        loadsz = filesz;
+
+        strtab->offset = filesz;
+        pad_section((usdt_dof_section_t *)strtab);
+        filesz += strtab->size + strtab->pad;
+
+        if (strtab->flags & 1)
+                loadsz += strtab->size + strtab->pad;
+
+        for (sec = file->sections; sec != NULL; sec = sec->next) {
+                sec->offset = filesz;
+                pad_section(sec);
+                filesz += sec->size + sec->pad;
+                if (sec->flags & 1)
+                        loadsz += sec->size + sec->pad;
+        }
+
+        header.dofh_loadsz = loadsz;
+        header.dofh_filesz = filesz;
+        memcpy(file->dof, &header, sizeof(dof_hdr_t));
+
+        offset = sizeof(dof_hdr_t);
+
+        offset = add_header(file, offset, (usdt_dof_section_t *)strtab);
+
+        for (sec = file->sections; sec != NULL; sec = sec->next)
+                offset = add_header(file, offset, sec);
+
+        offset = add_section(file, offset, (usdt_dof_section_t *)strtab);
+
+        for (sec = file->sections; sec != NULL; sec = sec->next)
+                offset = add_section(file, offset, sec);
+}
+
+usdt_dof_file_t *
+usdt_dof_file_init(usdt_provider_t *provider, size_t size)
+{
+        usdt_dof_file_t *file;
+
+        if ((file = malloc(sizeof(*file))) == NULL) {
+                usdt_error(provider, USDT_ERROR_MALLOC);
+                return (NULL);
+        }
+
+        if ((file->dof = valloc(size)) == NULL) {
+                usdt_error(provider, USDT_ERROR_VALLOC);
+                return (NULL);
+        }
+
+        file->sections = NULL;
+        file->size = size;
+
+        return (file);
+}

data/ext/libusdt/usdt_dof_sections.c

@@ -0,0 +1,172 @@
+#include "usdt_internal.h"
+
+int
+usdt_dof_probes_sect(usdt_dof_section_t *probes,
+                     usdt_provider_t *provider, usdt_strtab_t *strtab)
+{
+        usdt_probedef_t *pd;
+        dof_probe_t p;
+        dof_stridx_t type, argv;
+        uint8_t argc, i;
+        uint32_t argidx = 0;
+        uint32_t offidx = 0;
+
+        usdt_dof_section_init(probes, DOF_SECT_PROBES, 1);
+
+        for (pd = provider->probedefs; pd != NULL; pd = pd->next) {
+                argc = 0;
+                argv = 0;
+
+                for (i = 0; pd->types[i] != USDT_ARGTYPE_NONE && i < 6; i++) {
+                        switch(pd->types[i]) {
+                        case USDT_ARGTYPE_INTEGER:
+                                type = usdt_strtab_add(strtab, "int");
+                                break;
+                        case USDT_ARGTYPE_STRING:
+                                type = usdt_strtab_add(strtab, "char *");
+                                break;
+                        default:
+                                break;
+                        }
+
+                        argc++;
+                        if (argv == 0)
+                                argv = type;
+                }
+
+                if (usdt_create_tracepoints(pd->probe) < 0) {
+                        usdt_error(provider, USDT_ERROR_VALLOC);
+                        return (-1);
+                }
+
+#ifdef __x86_64__
+                p.dofpr_addr     = (uint64_t) pd->probe->isenabled_addr;
+#elif __i386__
+                p.dofpr_addr     = (uint32_t) pd->probe->isenabled_addr;
+#else
+#error "only x86_64 and i386 supported"
+#endif
+                p.dofpr_func     = usdt_strtab_add(strtab, pd->function);
+                p.dofpr_name     = usdt_strtab_add(strtab, pd->name);
+                p.dofpr_nargv    = argv;
+                p.dofpr_xargv    = argv;
+                p.dofpr_argidx   = argidx;
+                p.dofpr_offidx   = offidx;
+                p.dofpr_nargc    = argc;
+                p.dofpr_xargc    = argc;
+                p.dofpr_noffs    = 1;
+                p.dofpr_enoffidx = offidx;
+                p.dofpr_nenoffs  = 1;
+
+                usdt_dof_section_add_data(probes, &p, sizeof(dof_probe_t));
+                probes->entsize = sizeof(dof_probe_t);
+
+                argidx += argc;
+                offidx++;
+        }
+
+        return (0);
+}
+
+int
+usdt_dof_prargs_sect(usdt_dof_section_t *prargs, usdt_provider_t *provider)
+{
+        usdt_probedef_t *pd;
+        uint8_t i;
+
+        usdt_dof_section_init(prargs, DOF_SECT_PRARGS, 2);
+        prargs->entsize = 1;
+
+        for (pd = provider->probedefs; pd != NULL; pd = pd->next) {
+                for (i = 0; pd->types[i] != USDT_ARGTYPE_NONE && i < 6; i++)
+                        usdt_dof_section_add_data(prargs, &i, 1);
+        }
+        if (prargs->size == 0) {
+                i = 0;
+                if (usdt_dof_section_add_data(prargs, &i, 1) < 0) {
+                        usdt_error(provider, USDT_ERROR_MALLOC);
+                        return (-1);
+                }
+        }
+
+        return (0);
+}
+
+int
+usdt_dof_proffs_sect(usdt_dof_section_t *proffs,
+                     usdt_provider_t *provider, char *dof)
+{
+        usdt_probedef_t *pd;
+        uint32_t off;
+
+        usdt_dof_section_init(proffs, DOF_SECT_PROFFS, 3);
+        proffs->entsize = 4;
+
+        for (pd = provider->probedefs; pd != NULL; pd = pd->next) {
+                off = usdt_probe_offset(pd->probe, dof, usdt_probedef_argc(pd));
+                if (usdt_dof_section_add_data(proffs, &off, 4) < 0) {
+                        usdt_error(provider, USDT_ERROR_MALLOC);
+                        return (-1);
+                }
+        }
+
+        return (0);
+}
+
+int
+usdt_dof_prenoffs_sect(usdt_dof_section_t *prenoffs,
+                       usdt_provider_t *provider, char *dof)
+{
+        usdt_probedef_t *pd;
+        uint32_t off;
+
+        usdt_dof_section_init(prenoffs, DOF_SECT_PRENOFFS, 4);
+        prenoffs->entsize = 4;
+
+        for (pd = provider->probedefs; pd != NULL; pd = pd->next) {
+                off = usdt_is_enabled_offset(pd->probe, dof);
+                if (usdt_dof_section_add_data(prenoffs, &off, 4) < 0) {
+                        usdt_error(provider, USDT_ERROR_MALLOC);
+                        return (-1);
+                }
+        }
+
+        return (0);
+}
+
+int
+usdt_dof_provider_sect(usdt_dof_section_t *provider_s, usdt_provider_t *provider)
+{
+        dof_provider_t p;
+
+        usdt_dof_section_init(provider_s, DOF_SECT_PROVIDER, 5);
+
+        p.dofpv_strtab   = 0;
+        p.dofpv_probes   = 1;
+        p.dofpv_prargs   = 2;
+        p.dofpv_proffs   = 3;
+        p.dofpv_prenoffs = 4;
+        p.dofpv_name     = 1; // provider name always first strtab entry.
+        p.dofpv_provattr = DOF_ATTR(DTRACE_STABILITY_EVOLVING,
+                                    DTRACE_STABILITY_EVOLVING,
+                                    DTRACE_STABILITY_EVOLVING);
+        p.dofpv_modattr  = DOF_ATTR(DTRACE_STABILITY_PRIVATE,
+                                    DTRACE_STABILITY_PRIVATE,
+                                    DTRACE_STABILITY_EVOLVING);
+        p.dofpv_funcattr = DOF_ATTR(DTRACE_STABILITY_PRIVATE,
+                                    DTRACE_STABILITY_PRIVATE,
+                                    DTRACE_STABILITY_EVOLVING);
+        p.dofpv_nameattr = DOF_ATTR(DTRACE_STABILITY_EVOLVING,
+                                    DTRACE_STABILITY_EVOLVING,
+                                    DTRACE_STABILITY_EVOLVING);
+        p.dofpv_argsattr = DOF_ATTR(DTRACE_STABILITY_EVOLVING,
+                                    DTRACE_STABILITY_EVOLVING,
+                                    DTRACE_STABILITY_EVOLVING);
+
+        if ((usdt_dof_section_add_data(provider_s, &p, sizeof(p))) < 0) {
+                usdt_error(provider, USDT_ERROR_MALLOC);
+                return (-1);
+        }
+
+        return (0);
+}

data/ext/libusdt/usdt_internal.h

@@ -0,0 +1,100 @@
+#include <sys/dtrace.h>
+#include <sys/types.h>
+#include <sys/mman.h>
+
+#include <errno.h>
+#include <string.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <stdint.h>
+#include <assert.h>
+
+#ifndef __APPLE__
+/* solaris and freebsd */
+#include <stdlib.h>
+#endif
+
+#define FUNC_SIZE 32
+
+#include "usdt.h"
+
+extern void usdt_tracepoint_isenabled(void);
+extern void usdt_tracepoint_probe(void);
+extern void usdt_tracepoint_end(void);
+extern void usdt_probe_args(void *, int, void**);
+
+/*
+  typedef enum usdt_argtype {
+  USDT_ARGTYPE_NONE = 0,
+  USDT_ARGTYPE_STRING,
+  USDT_ARGTYPE_INTEGER
+  } usdt_argtype_t;
+*/
+
+uint32_t usdt_probe_offset(usdt_probe_t *probe, char *dof, uint8_t argc);
+uint32_t usdt_is_enabled_offset(usdt_probe_t *probe, char *dof);
+int usdt_create_tracepoints(usdt_probe_t *probe);
+
+uint8_t usdt_probedef_argc(usdt_probedef_t *probedef);
+
+typedef struct usdt_dof_section {
+        dof_secidx_t index;
+        uint32_t type;
+        uint32_t flags;
+        uint32_t align;
+        uint64_t offset;
+        uint64_t size;
+        uint32_t entsize;
+        size_t pad;
+        struct usdt_dof_section *next;
+        char *data;
+} usdt_dof_section_t;
+
+int usdt_dof_section_init(usdt_dof_section_t *section,
+                          uint32_t type, dof_secidx_t index);
+int usdt_dof_section_add_data(usdt_dof_section_t *section,
+                              void *data, size_t length);
+
+typedef struct usdt_strtab {
+        dof_secidx_t index;
+        uint32_t type;
+        uint32_t flags;
+        uint32_t align;
+        uint64_t offset;
+        uint64_t size;
+        uint32_t entsize;
+        size_t pad;
+        int strindex;
+        char *data;
+} usdt_strtab_t;
+
+int usdt_strtab_init(usdt_strtab_t *strtab, dof_secidx_t index);
+dof_stridx_t usdt_strtab_add(usdt_strtab_t *strtab, const char *string);
+char *usdt_strtab_header(usdt_strtab_t *strtab);
+size_t usdt_strtab_size(usdt_strtab_t *strtab);
+
+size_t usdt_provider_dof_size(usdt_provider_t *provider, usdt_strtab_t *strtab);
+
+typedef struct usdt_dof_file {
+        char *dof;
+        int gen;
+        size_t size;
+        usdt_dof_section_t *sections;
+} usdt_dof_file_t;
+
+usdt_dof_file_t *usdt_dof_file_init(usdt_provider_t *provider, size_t size);
+void usdt_dof_file_append_section(usdt_dof_file_t *file, usdt_dof_section_t *section);
+void usdt_dof_file_generate(usdt_dof_file_t *file, usdt_strtab_t *strtab);
+int usdt_dof_file_load(usdt_dof_file_t *file, const char *module);
+
+int usdt_dof_probes_sect(usdt_dof_section_t *probes,
+                         usdt_provider_t *provider, usdt_strtab_t *strtab);
+int usdt_dof_prargs_sect(usdt_dof_section_t *prargs,
+                         usdt_provider_t *provider);
+int usdt_dof_proffs_sect(usdt_dof_section_t *proffs,
+                         usdt_provider_t *provider, char *dof);
+int usdt_dof_prenoffs_sect(usdt_dof_section_t *prenoffs,
+                           usdt_provider_t *provider, char *dof);
+int usdt_dof_provider_sect(usdt_dof_section_t *provider_s,
+                           usdt_provider_t *provider);
+

data/ext/libusdt/usdt_probe.c

@@ -0,0 +1,66 @@
+#include "usdt_internal.h"
+
+#ifdef __APPLE__
+
+uint32_t
+usdt_probe_offset(usdt_probe_t *probe, char *dof, uint8_t argc)
+{
+        uint32_t offset;
+#ifdef __x86_64__
+        offset = ((uint64_t) probe->probe_addr - (uint64_t) dof + 2);
+#elif __i386__
+        offset = ((uint32_t) probe->probe_addr - (uint32_t) dof + 2);
+#else
+#error "only x86_64 and i386 supported"
+#endif
+        return (offset);
+}
+
+uint32_t
+usdt_is_enabled_offset(usdt_probe_t *probe, char *dof) 
+{
+        uint32_t offset;
+#ifdef __x86_64__
+        offset = ((uint64_t) probe->isenabled_addr - (uint64_t) dof + 2);
+#elif __i386__
+        offset = ((uint32_t) probe->isenabled_addr - (uint32_t) dof + 6);
+#else
+#error "only x86_64 and i386 supported"
+#endif
+        return (offset);
+}
+
+#else /* solaris and freebsd */
+
+uint32_t
+usdt_probe_offset(usdt_probe_t *probe, char *dof, uint8_t argc)
+{
+        return (12);
+}
+
+uint32_t
+usdt_is_enabled_offset(usdt_probe_t *probe, char *dof)
+{
+        return (6);
+}
+
+#endif
+
+int
+usdt_create_tracepoints(usdt_probe_t *probe)
+{
+        /* ensure that the tracepoints will fit the heap we're allocating */
+        assert((usdt_tracepoint_end - usdt_tracepoint_isenabled) < FUNC_SIZE);
+
+        if ((probe->isenabled_addr = valloc(FUNC_SIZE)) == NULL) {
+                return (-1);
+        }
+        probe->probe_addr = probe->isenabled_addr +
+                (usdt_tracepoint_probe - usdt_tracepoint_isenabled);
+
+        memcpy(probe->isenabled_addr, usdt_tracepoint_isenabled, FUNC_SIZE);
+        mprotect((void *)probe->isenabled_addr, FUNC_SIZE,
+                 PROT_READ | PROT_WRITE | PROT_EXEC);
+
+        return (0);
+}