ruby-usdt 0.2.0 → 0.2.1

data/ext/libusdt/usdt_dof.c

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

data/ext/libusdt/usdt_dof_file.c

@@ -0,0 +1,277 @@
+/*
+ * Copyright (c) 2012, Chris Andrews. All rights reserved.
+ */
+
+#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);
+
+        if (ret < 0)
+                return ret;
+
+        return (int)(ioctlData->dofiod_helpers[0].dofhp_dof);
+}
+
+#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) {
+                /* maximum padding required is 7 */
+                memcpy((file->dof + offset), "\0\0\0\0\0\0\0", section->pad);
+                offset += section->pad;
+        }
+
+        memcpy((file->dof + offset), section->data, section->size);
+        return (offset + section->size);
+}
+
+int
+usdt_dof_file_unload(usdt_dof_file_t *file)
+{
+        int fd, ret;
+
+        if ((fd = open(helper, O_RDWR)) < 0)
+                return (-1);
+
+        ret = ioctl(fd, DTRACEHIOC_REMOVE, file->gen);
+
+        if (ret < 0)
+                return (-1);
+
+        if ((close(fd)) < 0)
+                return (-1);
+
+        return (0);
+}
+
+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);
+
+        if (file->gen < 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);
+}
+
+void
+usdt_dof_file_free(usdt_dof_file_t *file)
+{
+        free(file->dof);
+        free(file);
+}

data/ext/libusdt/usdt_dof_sections.c

@@ -0,0 +1,190 @@
+/*
+ * Copyright (c) 2012, Chris Andrews. All rights reserved.
+ */
+
+#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;
+                type = 0;
+
+                for (i = 0; i < pd->argc; 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__ || __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;
+                p.dofpr_pad1     = 0;
+                p.dofpr_pad2     = 0;
+
+                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; i < pd->argc; 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, pd->argc);
+                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. */
+
+        /*
+         * Stability is something of a hack. Everything is marked *
+         * "stable" here to permit use of the "args" array, which is *
+         * needed to access arguments past "arg9".
+         *
+         * It should be up to the creator of the provider to decide
+         * this, though, and it should be possible to set the
+         * appropriate stability at creation time.
+         */
+
+        p.dofpv_provattr = DOF_ATTR(DTRACE_STABILITY_STABLE,
+                                    DTRACE_STABILITY_STABLE,
+                                    DTRACE_STABILITY_STABLE);
+        p.dofpv_modattr  = DOF_ATTR(DTRACE_STABILITY_STABLE,
+                                    DTRACE_STABILITY_STABLE,
+                                    DTRACE_STABILITY_STABLE);
+        p.dofpv_funcattr = DOF_ATTR(DTRACE_STABILITY_STABLE,
+                                    DTRACE_STABILITY_STABLE,
+                                    DTRACE_STABILITY_STABLE);
+        p.dofpv_nameattr = DOF_ATTR(DTRACE_STABILITY_STABLE,
+                                    DTRACE_STABILITY_STABLE,
+                                    DTRACE_STABILITY_STABLE);
+        p.dofpv_argsattr = DOF_ATTR(DTRACE_STABILITY_STABLE,
+                                    DTRACE_STABILITY_STABLE,
+                                    DTRACE_STABILITY_STABLE);
+
+        if ((usdt_dof_section_add_data(provider_s, &p, sizeof(p))) < 0) {
+                usdt_error(provider, USDT_ERROR_MALLOC);
+                return (-1);
+        }
+
+        return (0);
+}