ruby-usdt 0.0.1

data/ext/libusdt/usdt_tracepoints_i386.s

@@ -0,0 +1,65 @@
+/*
+ * 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),%ebx 
+        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     *%ebx
+

data/ext/libusdt/usdt_tracepoints_x86_64.s

@@ -0,0 +1,83 @@
+/*
+ * 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
+        xorq    %rax, %rax
+        nop
+        nop
+        leave
+        ret
+usdt_tracepoint_probe:
+_usdt_tracepoint_probe:
+        nop
+        nop
+        nop
+        nop
+        nop
+        popq %r11
+        popq %rbx
+        popq %r12
+        addq $0x18,%rsp
+        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
+        subq    $0x18,%rsp
+        pushq   %r12
+        pushq   %rbx
+        pushq   %r11
+        movq    %rdi,%r12
+        movq    %rsi,%rbx
+        movq    %rdx,%r11
+        test    %rbx,%rbx
+        je      fire
+        movq    (%r11),%rdi
+        dec     %rbx
+        test    %rbx,%rbx
+        je      fire
+        movq    8(%r11),%rsi
+        dec     %rbx
+        test    %rbx,%rbx
+        je      fire
+        movq    16(%r11),%rdx
+        dec     %rbx
+        test    %rbx,%rbx
+        je      fire
+        movq    24(%r11),%rcx
+        dec     %rbx
+        test    %rbx,%rbx
+        je      fire
+        movq    32(%r11),%r8
+        dec     %rbx
+        test    %rbx,%rbx
+        je      fire
+        movq    40(%r11),%r9
+fire:   jmp     *%r12

data/ext/usdt/Makefile

@@ -0,0 +1,157 @@
+
+SHELL = /bin/sh
+
+#### Start of system configuration section. ####
+
+srcdir = .
+topdir = /Users/kevin/.rvm/rubies/ruby-1.8.7-p358/lib/ruby/1.8/i686-darwin11.3.0
+hdrdir = $(topdir)
+VPATH = $(srcdir):$(topdir):$(hdrdir)
+prefix = $(DESTDIR)/Users/kevin/.rvm/rubies/ruby-1.8.7-p358
+exec_prefix = $(prefix)
+htmldir = $(docdir)
+vendorarchdir = $(vendorlibdir)/$(sitearch)
+sitedir = $(libdir)/ruby/site_ruby
+pdfdir = $(docdir)
+dvidir = $(docdir)
+sharedstatedir = $(prefix)/com
+vendordir = $(libdir)/ruby/vendor_ruby
+psdir = $(docdir)
+libexecdir = $(exec_prefix)/libexec
+datarootdir = $(prefix)/share
+bindir = $(exec_prefix)/bin
+infodir = $(datarootdir)/info
+sitearchdir = $(sitelibdir)/$(sitearch)
+sysconfdir = $(prefix)/etc
+archdir = $(rubylibdir)/$(arch)
+sbindir = $(exec_prefix)/sbin
+includedir = $(prefix)/include
+mandir = $(datarootdir)/man
+rubylibdir = $(libdir)/ruby/$(ruby_version)
+localstatedir = $(prefix)/var
+vendorlibdir = $(vendordir)/$(ruby_version)
+sitelibdir = $(sitedir)/$(ruby_version)
+docdir = $(datarootdir)/doc/$(PACKAGE)
+datadir = $(datarootdir)
+localedir = $(datarootdir)/locale
+libdir = $(exec_prefix)/lib
+oldincludedir = $(DESTDIR)/usr/include
+
+CC = /usr/bin/gcc-4.2
+LIBRUBY = $(LIBRUBY_SO)
+LIBRUBY_A = lib$(RUBY_SO_NAME)-static.a
+LIBRUBYARG_SHARED = -l$(RUBY_SO_NAME)
+LIBRUBYARG_STATIC = -l$(RUBY_SO_NAME)-static
+
+RUBY_EXTCONF_H = 
+CFLAGS   =  -fno-common -g -O2  -fno-common -pipe -fno-common $(cflags) 
+INCFLAGS = -I. -I. -I/Users/kevin/.rvm/rubies/ruby-1.8.7-p358/lib/ruby/1.8/i686-darwin11.3.0 -I.
+DEFS     = 
+CPPFLAGS = -DHAVE_USDT_H  -D_XOPEN_SOURCE -D_DARWIN_C_SOURCE  
+CXXFLAGS = $(CFLAGS) 
+ldflags  = -L. 
+dldflags = 
+archflag = 
+DLDFLAGS = $(ldflags) $(dldflags) $(archflag)
+LDSHARED = cc -dynamic -bundle -undefined suppress -flat_namespace
+AR = ar
+EXEEXT = 
+
+RUBY_INSTALL_NAME = ruby
+RUBY_SO_NAME = ruby
+arch = i686-darwin11.3.0
+sitearch = i686-darwin11.3.0
+ruby_version = 1.8
+ruby = /Users/kevin/.rvm/rubies/ruby-1.8.7-p358/bin/ruby
+RUBY = $(ruby)
+RM = rm -f
+MAKEDIRS = mkdir -p
+INSTALL = /usr/bin/install -c
+INSTALL_PROG = $(INSTALL) -m 0755
+INSTALL_DATA = $(INSTALL) -m 644
+COPY = cp
+
+#### End of system configuration section. ####
+
+preload = 
+
+libpath = . $(libdir)
+LIBPATH =  -L. -L$(libdir)
+DEFFILE = 
+
+CLEANFILES = mkmf.log
+DISTCLEANFILES = 
+
+extout = 
+extout_prefix = 
+target_prefix = 
+LOCAL_LIBS = 
+LIBS = $(LIBRUBYARG_SHARED) -lusdt -ldtrace  -ldl -lobjc  
+SRCS = usdt.c
+OBJS = usdt.o
+TARGET = usdt
+DLLIB = $(TARGET).bundle
+EXTSTATIC = 
+STATIC_LIB = 
+
+BINDIR        = $(bindir)
+RUBYCOMMONDIR = $(sitedir)$(target_prefix)
+RUBYLIBDIR    = $(sitelibdir)$(target_prefix)
+RUBYARCHDIR   = $(sitearchdir)$(target_prefix)
+
+TARGET_SO     = $(DLLIB)
+CLEANLIBS     = $(TARGET).bundle $(TARGET).il? $(TARGET).tds $(TARGET).map
+CLEANOBJS     = *.o *.a *.s[ol] *.pdb *.exp *.bak
+
+all:		$(DLLIB)
+static:		$(STATIC_LIB)
+
+clean:
+		@-$(RM) $(CLEANLIBS) $(CLEANOBJS) $(CLEANFILES)
+
+distclean:	clean
+		@-$(RM) Makefile $(RUBY_EXTCONF_H) conftest.* mkmf.log
+		@-$(RM) core ruby$(EXEEXT) *~ $(DISTCLEANFILES)
+
+realclean:	distclean
+install: install-so install-rb
+
+install-so: $(RUBYARCHDIR)
+install-so: $(RUBYARCHDIR)/$(DLLIB)
+$(RUBYARCHDIR)/$(DLLIB): $(DLLIB)
+	$(INSTALL_PROG) $(DLLIB) $(RUBYARCHDIR)
+install-rb: pre-install-rb install-rb-default
+install-rb-default: pre-install-rb-default
+pre-install-rb: Makefile
+pre-install-rb-default: Makefile
+$(RUBYARCHDIR):
+	$(MAKEDIRS) $@
+
+site-install: site-install-so site-install-rb
+site-install-so: install-so
+site-install-rb: install-rb
+
+.SUFFIXES: .c .m .cc .cxx .cpp .C .o
+
+.cc.o:
+	$(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) -c $<
+
+.cxx.o:
+	$(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) -c $<
+
+.cpp.o:
+	$(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) -c $<
+
+.C.o:
+	$(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) -c $<
+
+.c.o:
+	$(CC) $(INCFLAGS) $(CPPFLAGS) $(CFLAGS) -c $<
+
+$(DLLIB): $(OBJS) Makefile
+	@-$(RM) $@
+	$(LDSHARED) -o $@ $(OBJS) $(LIBPATH) $(DLDFLAGS) $(LOCAL_LIBS) $(LIBS)
+
+
+
+$(OBJS): ruby.h defines.h

data/ext/usdt/conftest.dSYM/Contents/Info.plist

@@ -0,0 +1,20 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple Computer//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+	<dict>
+		<key>CFBundleDevelopmentRegion</key>
+		<string>English</string>
+		<key>CFBundleIdentifier</key>
+		<string>com.apple.xcode.dsym.conftest</string>
+		<key>CFBundleInfoDictionaryVersion</key>
+		<string>6.0</string>
+		<key>CFBundlePackageType</key>
+		<string>dSYM</string>
+		<key>CFBundleSignature</key>
+		<string>????</string>
+		<key>CFBundleShortVersionString</key>
+		<string>1.0</string>
+		<key>CFBundleVersion</key>
+		<string>1</string>
+	</dict>
+</plist>

data/ext/usdt/extconf.rb

@@ -0,0 +1,9 @@
+require 'mkmf'
+
+have_library("dtrace", "dtrace_open", "/usr/include/dtrace.h")
+system("cd ../libusdt && make")
+system("cp ../libusdt/usdt.h ../libusdt/*.a ./")
+have_library('usdt')
+have_header('usdt.h')
+
+create_makefile('usdt')

data/ext/usdt/test.rb

@@ -0,0 +1,23 @@
+require File.expand_path(File.dirname(__FILE__)) + '/usdt'
+
+provider = USDT::Provider.create :ruby, :test
+puts provider
+p = provider.probe("myfn", "probe",
+                   :string, :string, :integer)
+
+provider.enable
+
+puts "run:\nsudo dtrace -n 'ruby*:test:myfn:probe { printf(\"%s %s %d\",
+  copyinstr(arg0),
+  copyinstr(arg1),
+  args[2])
+}'"
+
+while true
+  if p.enabled? 
+    p.fire("one", "two", 3)
+    puts("Fired!")
+  end
+  sleep 0.5
+end
+

data/ext/usdt/usdt.c

@@ -0,0 +1,141 @@
+#include <ruby.h>
+#include "usdt.h"
+
+VALUE USDT;
+VALUE USDT_Provider;
+VALUE USDT_Probe;
+VALUE USDT_Error;
+
+static VALUE provider_create(VALUE self, VALUE name, VALUE mod);
+static VALUE provider_probe(int argc, VALUE *argv, VALUE self);
+static VALUE provider_enable(VALUE self);
+static VALUE probe_enabled(VALUE self);
+static VALUE probe_fire(int argc, VALUE *argv, VALUE self);
+
+void Init_usdt() {
+  USDT = rb_define_module("USDT");
+
+  USDT_Error = rb_define_class_under(USDT, "Error", rb_eRuntimeError);
+
+  USDT_Provider = rb_define_class_under(USDT, "Provider", rb_cObject);
+  rb_define_singleton_method(USDT_Provider, "create", provider_create, 2);
+  rb_define_method(USDT_Provider, "probe", provider_probe, -1);
+  rb_define_method(USDT_Provider, "enable", provider_enable, 0);
+
+  USDT_Probe = rb_define_class_under(USDT, "Probe", rb_cObject);
+  rb_define_method(USDT_Probe, "enabled?", probe_enabled, 0);
+  rb_define_method(USDT_Probe, "fire", probe_fire, -1);
+}
+
+/**
+ * USDT::Provider.create :name, :modname
+ */
+static VALUE provider_create(VALUE self, VALUE name, VALUE mod) {
+  Check_Type(name, T_SYMBOL);
+  Check_Type(mod, T_SYMBOL);
+
+  char *namestr = rb_id2name(rb_to_id(name));
+  char *modstr = rb_id2name(rb_to_id(mod));
+
+  usdt_provider_t* p = usdt_create_provider(namestr, modstr);
+
+  VALUE rbProvider = Data_Wrap_Struct(USDT_Provider, NULL, free, p);
+
+  if (rb_block_given_p()) {
+    rb_yield(rbProvider);
+  }
+
+  return rbProvider;
+}
+
+/**
+ * USDT::Provider#probe(func, name, pargs*)
+ */
+static VALUE provider_probe(int argc, VALUE *argv, VALUE self) {
+  const char *func = RSTRING_PTR(argv[0]);
+  const char *name = RSTRING_PTR(argv[1]);
+  const char *types[6];
+  size_t i, pargc = 0;
+  size_t t_int = rb_intern("integer");
+  size_t t_str = rb_intern("string");
+
+  for (i = 0; i < 6; i++) {
+    if (i < argc - 2) {
+      Check_Type(argv[i+2], T_SYMBOL);
+      if (t_int == rb_to_id(argv[i+2])) {
+        types[i] = "int";
+        pargc++;
+      } else if (t_str == rb_to_id(argv[i+2])) {
+        types[i] = "char *";
+        pargc++;
+      } else {
+        types[i] = NULL;
+      }
+    } else {
+      types[i] = NULL;
+    }
+  }
+
+  usdt_provider_t *provider = DATA_PTR(self);
+
+  usdt_probedef_t **probe;
+  probe = ALLOC(usdt_probedef_t *);
+  *probe = usdt_create_probe(func, name, pargc, types);
+
+  usdt_provider_add_probe(provider, *probe);
+  VALUE rbProbe = Data_Wrap_Struct(USDT_Probe, NULL, free, probe);
+  return rbProbe;
+}
+
+/**
+ * USDT::Provider#enable
+ */
+static VALUE provider_enable(VALUE self) {
+  usdt_provider_t *provider = DATA_PTR(self);
+  int status = usdt_provider_enable(provider);
+  if (status == 0) {
+    return Qtrue;
+  } else {
+    rb_raise(USDT_Error, "%s", usdt_errstr(provider));
+  }
+}
+
+/**
+ * USDT::Probe#enabled?
+ */
+static VALUE probe_enabled(VALUE self) {
+  usdt_probedef_t **p = DATA_PTR(self);
+  usdt_probedef_t *pd = *p;
+
+  if (usdt_is_enabled(pd->probe) == 0) {
+    return Qfalse;
+  } else {
+    return Qtrue;
+  }
+}
+
+/**
+ * USDT::Probe#fire *args
+ */
+static VALUE probe_fire(int argc, VALUE *argv, VALUE self) {
+  usdt_probedef_t **p = DATA_PTR(self);
+  usdt_probedef_t *probedef = *p;
+
+  void *pargs[6];
+  int i;
+
+  for (i = 0; i < probedef->argc; i++) {
+    if (probedef->types[i] == USDT_ARGTYPE_STRING) {
+      Check_Type(argv[i], T_STRING);
+      pargs[i] = (void *) RSTRING_PTR(argv[i]);
+    } else if (probedef->types[i] == USDT_ARGTYPE_INTEGER) {
+      Check_Type(argv[i], T_FIXNUM);
+      pargs[i] = (void *) FIX2INT(argv[i]);
+    } else {
+      pargs[i] = NULL;
+    }
+  }
+
+  usdt_fire_probe(probedef->probe, probedef->argc, pargs);
+  return Qtrue;
+}

data/ext/usdt/usdt.h

@@ -0,0 +1,50 @@
+#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
+
+typedef enum usdt_error {
+        USDT_ERROR_MALLOC = 0,
+        USDT_ERROR_VALLOC,
+        USDT_ERROR_NOPROBES,
+        USDT_ERROR_LOADDOF
+} 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[6];
+        struct usdt_probe *probe;
+        struct usdt_probedef *next;
+} usdt_probedef_t;
+
+usdt_probedef_t *usdt_create_probe_varargs(const char *func, const char *name, ...);
+usdt_probedef_t *usdt_create_probe(const char *func, const char *name,
+                                   size_t argc, const char **types);
+
+typedef struct usdt_provider {
+        const char *name;
+        const char *module;
+        usdt_probedef_t *probedefs;
+        char *error;
+} usdt_provider_t;
+
+usdt_provider_t *usdt_create_provider(const char *name, const char *module);
+void usdt_provider_add_probe(usdt_provider_t *provider, usdt_probedef_t *probedef);
+int usdt_provider_enable(usdt_provider_t *provider);
+
+void usdt_error(usdt_provider_t *provider, usdt_error_t error);
+char *usdt_errstr(usdt_provider_t *provider);
+

data/lib/usdt/version.rb

@@ -0,0 +1,3 @@
+module USDT
+  VERSION = '0.0.1'
+end

data/lib/usdt.rb

@@ -0,0 +1,2 @@
+require "usdt/version"
+require 'usdt/usdt'

metadata

@@ -0,0 +1,71 @@
+--- !ruby/object:Gem::Specification
+name: ruby-usdt
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Kevin Chan
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-04-20 00:00:00.000000000 Z
+dependencies: []
+description: 
+email:
+- kevin@yinkei.com
+executables: []
+extensions:
+- ext/usdt/extconf.rb
+extra_rdoc_files: []
+files:
+- lib/usdt/version.rb
+- lib/usdt.rb
+- ext/libusdt/Makefile
+- ext/libusdt/README
+- ext/libusdt/test.pl
+- ext/libusdt/test_usdt.c
+- ext/libusdt/usdt.c
+- ext/libusdt/usdt.h
+- ext/libusdt/usdt_dof.c
+- ext/libusdt/usdt_dof_file.c
+- ext/libusdt/usdt_dof_sections.c
+- ext/libusdt/usdt_internal.h
+- ext/libusdt/usdt_probe.c
+- ext/libusdt/usdt_tracepoints_i386.s
+- ext/libusdt/usdt_tracepoints_x86_64.s
+- ext/usdt/conftest.dSYM/Contents/Info.plist
+- ext/usdt/conftest.dSYM/Contents/Resources/DWARF/conftest
+- ext/usdt/extconf.rb
+- ext/usdt/Makefile
+- ext/usdt/test.rb
+- ext/usdt/usdt.c
+- ext/usdt/usdt.h
+- README.md
+- LICENSE.md
+homepage: http://github.com/kevinykchan/ruby-usdt
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+- ext
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.21
+signing_key: 
+specification_version: 3
+summary: ruby bindings to libusdt
+test_files: []