ruby-static-tracing 0.0.12 → 0.0.13

data/ext/ruby-static-tracing/lib/libusdt/usdt_internal.h

@@ -0,0 +1,107 @@
+/*
+ * Copyright (c) 2012, Chris Andrews. All rights reserved.
+ */
+
+#ifndef __LIB_USDT_USDT_INTERNAL_H__
+#define __LIB_USDT_USDT_INTERNAL_H__
+
+#ifdef __linux__
+#include <endian.h>
+#if __BYTE_ORDER == __LITTLE_ENDIAN
+#ifndef _LITTLE_ENDIAN
+#define _LITTLE_ENDIAN
+#endif
+#endif
+#endif
+
+#include <sys/dtrace.h>
+#include <sys/types.h>
+#include <sys/mman.h>
+
+#include <stdlib.h>
+#include <errno.h>
+#include <string.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <stdint.h>
+#include <assert.h>
+
+#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**);
+
+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);
+void usdt_free_tracepoints(usdt_probe_t *probe);
+
+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);
+void usdt_dof_section_free(usdt_dof_section_t *section);
+
+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_file_unload(usdt_dof_file_t *file);
+void usdt_dof_file_free(usdt_dof_file_t *file);
+
+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);
+
+#endif // __LIB_USDT_USDT_INTERNAL_H__

data/ext/ruby-static-tracing/lib/libusdt/usdt_probe.c

@@ -0,0 +1,133 @@
+/*
+ * Copyright (c) 2012, Chris Andrews. All rights reserved.
+ */
+
+#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 + 6);
+#elif __i386__
+        offset = ((uint32_t) probe->isenabled_addr - (uint32_t) dof + 6);
+#else
+#error "only x86_64 and i386 supported"
+#endif
+        return (offset);
+}
+
+#elif defined __linux__
+
+uint32_t
+usdt_probe_offset(usdt_probe_t *probe, char *dof, uint8_t argc)
+{
+        return (16);
+}
+
+uint32_t
+usdt_is_enabled_offset(usdt_probe_t *probe, char *dof)
+{
+        return (10);
+}
+
+#else /* solaris and freebsd */
+
+uint32_t
+usdt_probe_offset(usdt_probe_t *probe, char *dof, uint8_t argc)
+{
+        return (16);
+}
+
+uint32_t
+usdt_is_enabled_offset(usdt_probe_t *probe, char *dof)
+{
+        return (8);
+}
+
+#endif
+
+int
+usdt_create_tracepoints(usdt_probe_t *probe)
+{
+        /* Prepare the tracepoints - for each probe, a separate chunk
+         * of memory with the tracepoint code copied into it, to give
+         * us unique addresses for each tracepoint.
+         *
+         * On Oracle Linux, this must be an mmapped file because USDT
+         * probes there are implemented as uprobes, which are
+         * addressed by inode and offset. The file used is a small
+         * mkstemp'd file we immediately unlink.
+         *
+         * Elsewhere, we can use the heap directly because USDT will
+         * instrument any memory mapped by the process.
+         */
+
+        size_t size;
+#ifdef __linux__
+        int fd;
+        char tmp[20] = "/tmp/libusdtXXXXXX";
+
+        if ((fd = mkstemp(tmp)) < 0)
+                return (-1);
+        if (unlink(tmp) < 0)
+                return (-1);
+        if (write(fd, "\0", FUNC_SIZE) < FUNC_SIZE)
+                return (-1);
+
+        probe->isenabled_addr = (int (*)())mmap(NULL, FUNC_SIZE,
+                                                PROT_READ | PROT_WRITE | PROT_EXEC,
+                                                MAP_PRIVATE, fd, 0);
+#else
+        probe->isenabled_addr = (int (*)())valloc(FUNC_SIZE);
+#endif
+        if (probe->isenabled_addr == NULL)
+                return (-1);
+
+        /* ensure that the tracepoints will fit the heap we're allocating */
+        size = ((char *)usdt_tracepoint_end - (char *)usdt_tracepoint_isenabled);
+        assert(size < FUNC_SIZE);
+
+        size = ((char *)usdt_tracepoint_probe - (char *)usdt_tracepoint_isenabled);
+        probe->probe_addr = (char *)probe->isenabled_addr + size;
+
+        memcpy((void *)probe->isenabled_addr,
+               (const void *)usdt_tracepoint_isenabled, FUNC_SIZE);
+
+#ifdef __linux__
+        mprotect((void *)probe->isenabled_addr, FUNC_SIZE,
+                 PROT_READ | PROT_EXEC);
+#else
+        mprotect((void *)probe->isenabled_addr, FUNC_SIZE,
+                 PROT_READ | PROT_WRITE | PROT_EXEC);
+#endif
+
+        return (0);
+}
+
+void
+usdt_free_tracepoints(usdt_probe_t *probe)
+{
+#ifdef __linux__
+        (void) munmap(probe->isenabled_addr, FUNC_SIZE);
+#else
+        free(probe->isenabled_addr);
+#endif
+}

data/ext/ruby-static-tracing/lib/libusdt/usdt_tracepoints_i386.s

@@ -0,0 +1,69 @@
+/*
+ * Copyright (c) 2012, Chris Andrews. All rights reserved.
+ */
+
+/*
+ * Stub functions containing DTrace tracepoints for probes and
+ * is-enabled probes. These functions are copied for each probe
+ * dynamically created.
+ *
+ */
+        .text
+
+        .align 4, 0x90
+        .globl usdt_tracepoint_isenabled
+        .globl _usdt_tracepoint_isenabled
+        .globl usdt_tracepoint_probe
+        .globl _usdt_tracepoint_probe
+        .globl usdt_tracepoint_end
+        .globl _usdt_tracepoint_end
+        .globl usdt_probe_args
+        .globl _usdt_probe_args
+
+usdt_tracepoint_isenabled:
+_usdt_tracepoint_isenabled:
+        pushl   %ebp
+        movl    %esp, %ebp
+        subl    $8, %esp
+        xorl    %eax, %eax
+        nop
+        nop
+        leave
+        ret
+usdt_tracepoint_probe:
+_usdt_tracepoint_probe:
+        nop
+        nop
+        nop
+        nop
+        nop
+        addl    $0x20,%esp
+        leave
+usdt_tracepoint_end:
+_usdt_tracepoint_end:
+        ret
+
+/*
+ * Probe argument marshalling, i386 style
+ *
+ */
+
+usdt_probe_args:
+_usdt_probe_args:
+        pushl   %ebp
+        movl    %esp,%ebp
+        subl    $8,%esp
+        subl    $8,%esp
+        movl    8(%ebp),%edx
+        movl    0xc(%ebp),%ecx
+        test    %ecx,%ecx
+        je      fire
+args:   movl    %ecx,%eax
+        sal     $2,%eax
+        subl    $4,%eax
+        addl    0x10(%ebp),%eax
+        pushl   (%eax)
+        dec     %ecx
+        jne     args
+fire:   jmp     *%edx
+

data/ext/ruby-static-tracing/lib/libusdt/usdt_tracepoints_x86_64.s

@@ -0,0 +1,123 @@
+/*
+ * Copyright (c) 2012, Chris Andrews. All rights reserved.
+ */
+
+/*
+ * Stub functions containing DTrace tracepoints for probes and
+ * is-enabled probes. These functions are copied for each probe
+ * dynamically created.
+ *
+ */
+        .text
+
+        .align 4, 0x90
+        .globl usdt_tracepoint_isenabled
+        .globl _usdt_tracepoint_isenabled
+        .globl usdt_tracepoint_probe
+        .globl _usdt_tracepoint_probe
+        .globl usdt_tracepoint_end
+        .globl _usdt_tracepoint_end
+        .globl usdt_probe_args
+        .globl _usdt_probe_args
+
+usdt_tracepoint_isenabled:
+_usdt_tracepoint_isenabled:
+        pushq   %rbp
+        movq    %rsp, %rbp
+        addq    $1, %rax
+        xorq    %rax, %rax
+        nop
+        nop
+        leave
+        ret
+usdt_tracepoint_probe:
+_usdt_tracepoint_probe:
+        nop
+        nop
+        nop
+        nop
+        nop
+        addq	%r14,%rsp
+        popq    %rbx
+        popq    %r14
+        popq    %r13
+        popq    %r12
+        leave
+usdt_tracepoint_end:
+_usdt_tracepoint_end:
+        ret
+
+/*
+ * Probe argument marshalling, x86_64 style
+ *
+ */
+
+usdt_probe_args:
+_usdt_probe_args:
+        pushq   %rbp
+        movq    %rsp,%rbp
+        pushq   %r12
+        pushq   %r13
+        pushq   %r14
+        pushq   %rbx
+
+        movq    %rdi,%r12
+        movq    %rsi,%rbx
+        movq    %rdx,%r11
+        movq    $0,%r14
+
+        test    %rbx,%rbx
+        je      fire
+        movq    (%r11),%rdi
+        dec     %rbx
+        test    %rbx,%rbx
+        je      fire
+        addq    $8,%r11
+        movq    (%r11),%rsi
+        dec     %rbx
+        test    %rbx,%rbx
+        je      fire
+        addq    $8,%r11
+        movq    (%r11),%rdx
+        dec     %rbx
+        test    %rbx,%rbx
+        je      fire
+        addq    $8,%r11
+        movq    (%r11),%rcx
+        dec     %rbx
+        test    %rbx,%rbx
+        je      fire
+        addq    $8,%r11
+        movq    (%r11),%r8
+        dec     %rbx
+        test    %rbx,%rbx
+        je      fire
+        addq    $8,%r11
+        movq    (%r11),%r9
+
+        movq    %rbx,%r13
+morestack:
+        dec     %rbx
+        test    %rbx,%rbx
+        je      args
+        subq    $16,%rsp
+        addq    $16,%r14
+        dec     %rbx
+        test    %rbx,%rbx
+        je      args
+        jmp     morestack
+
+args:
+        movq    %r13,%rbx
+        movq    $0,%r13
+moreargs:
+        dec     %rbx
+        test    %rbx,%rbx
+        je      fire
+        addq    $8,%r11
+        movq    (%r11),%rax
+        movq    %rax,(%rsp,%r13)
+        addq    $8,%r13
+        jmp     moreargs
+
+fire:   jmp     *%r12

data/lib/ruby-static-tracing/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module StaticTracing
-  VERSION = '0.0.12'
+  VERSION = '0.0.13'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby-static-tracing
 version: !ruby/object:Gem::Version
-  version: 0.0.12
+  version: 0.0.13
 platform: ruby
 authors:
 - Dale Hamel
@@ -113,6 +113,41 @@ files:
 - ext/ruby-static-tracing/extconf.rb
 - ext/ruby-static-tracing/include/ruby_static_tracing.h
 - ext/ruby-static-tracing/lib/deps-extconf.rb
+- ext/ruby-static-tracing/lib/libstapsdt/Makefile
+- ext/ruby-static-tracing/lib/libstapsdt/example/demo.c
+- ext/ruby-static-tracing/lib/libstapsdt/src/asm/libstapsdt-x86_64.s
+- ext/ruby-static-tracing/lib/libstapsdt/src/dynamic-symbols.c
+- ext/ruby-static-tracing/lib/libstapsdt/src/dynamic-symbols.h
+- ext/ruby-static-tracing/lib/libstapsdt/src/errors.c
+- ext/ruby-static-tracing/lib/libstapsdt/src/errors.h
+- ext/ruby-static-tracing/lib/libstapsdt/src/hash-table.c
+- ext/ruby-static-tracing/lib/libstapsdt/src/hash-table.h
+- ext/ruby-static-tracing/lib/libstapsdt/src/libstapsdt.c
+- ext/ruby-static-tracing/lib/libstapsdt/src/libstapsdt.h
+- ext/ruby-static-tracing/lib/libstapsdt/src/sdtnote.c
+- ext/ruby-static-tracing/lib/libstapsdt/src/sdtnote.h
+- ext/ruby-static-tracing/lib/libstapsdt/src/section.c
+- ext/ruby-static-tracing/lib/libstapsdt/src/section.h
+- ext/ruby-static-tracing/lib/libstapsdt/src/shared-lib.c
+- ext/ruby-static-tracing/lib/libstapsdt/src/shared-lib.h
+- ext/ruby-static-tracing/lib/libstapsdt/src/string-table.c
+- ext/ruby-static-tracing/lib/libstapsdt/src/string-table.h
+- ext/ruby-static-tracing/lib/libstapsdt/src/util.c
+- ext/ruby-static-tracing/lib/libstapsdt/src/util.h
+- ext/ruby-static-tracing/lib/libstapsdt/tests/test-errors.c
+- ext/ruby-static-tracing/lib/libstapsdt/tests/test-memory-leaks.c
+- ext/ruby-static-tracing/lib/libusdt/Makefile
+- ext/ruby-static-tracing/lib/libusdt/test_mem_usage.c
+- ext/ruby-static-tracing/lib/libusdt/test_usdt.c
+- ext/ruby-static-tracing/lib/libusdt/usdt.c
+- ext/ruby-static-tracing/lib/libusdt/usdt.h
+- ext/ruby-static-tracing/lib/libusdt/usdt_dof.c
+- ext/ruby-static-tracing/lib/libusdt/usdt_dof_file.c
+- ext/ruby-static-tracing/lib/libusdt/usdt_dof_sections.c
+- ext/ruby-static-tracing/lib/libusdt/usdt_internal.h
+- ext/ruby-static-tracing/lib/libusdt/usdt_probe.c
+- ext/ruby-static-tracing/lib/libusdt/usdt_tracepoints_i386.s
+- ext/ruby-static-tracing/lib/libusdt/usdt_tracepoints_x86_64.s
 - ext/ruby-static-tracing/lib/post-extconf.rb
 - ext/ruby-static-tracing/linux/provider.c
 - ext/ruby-static-tracing/linux/provider.h
@@ -154,7 +189,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
+rubyforge_project: 
+rubygems_version: 2.7.6.2
 signing_key: 
 specification_version: 4
 summary: USDT tracing for Ruby