ruby-dtrace-consumer 0.4.0

data/ext/dtrace_probedesc.c

@@ -0,0 +1,93 @@
+/* Ruby-DTrace
+ * (c) 2007 Chris Andrews <chris@nodnol.org>
+ */
+
+#include "dtrace_api.h"
+
+RUBY_EXTERN VALUE cDTraceProbeDesc;
+
+static void free_probedesc(void *p);
+
+/* :nodoc: */
+VALUE dtraceprobedesc_init(dtrace_probedesc_t *p)
+{
+  dtrace_probedesc_t *pdp;
+  VALUE self;
+
+  pdp = ALLOC(dtrace_probedesc_t);
+  memcpy(pdp, p, sizeof(dtrace_probedesc_t));
+
+  self = Data_Wrap_Struct(cDTraceProbeDesc, 0, free_probedesc, pdp);
+  return self;
+}
+
+/*
+ * Returns the id of the probedesc. Corresponds to the ID displayed by
+ * dtrace -l
+ */
+VALUE dtraceprobedesc_probe_id(VALUE self)
+{
+  dtrace_probedesc_t *pdp;
+
+  Data_Get_Struct(self, dtrace_probedesc_t, pdp);
+  return INT2NUM(pdp->dtpd_id);
+}
+
+/*
+ * Returns the name of the probe's provider.
+ */
+VALUE dtraceprobedesc_provider(VALUE self)
+{
+  VALUE string;
+  dtrace_probedesc_t *pdp;
+
+  Data_Get_Struct(self, dtrace_probedesc_t, pdp);
+  string = rb_str_new2(pdp->dtpd_provider);
+  return string;
+}
+
+/*
+ * Returns the name of the module where the probe is defined.
+ */
+VALUE dtraceprobedesc_mod(VALUE self)
+{
+  VALUE string;
+  dtrace_probedesc_t *pdp;
+
+  Data_Get_Struct(self, dtrace_probedesc_t, pdp);
+  string = rb_str_new2(pdp->dtpd_mod);
+  return string;
+}
+
+/*
+ * Returns the name of the function where the probe is defined.
+ */
+VALUE dtraceprobedesc_func(VALUE self)
+{
+  VALUE string;
+  dtrace_probedesc_t *pdp;
+
+  Data_Get_Struct(self, dtrace_probedesc_t, pdp);
+  string = rb_str_new2(pdp->dtpd_func);
+  return string;
+}
+
+/*
+ * Returns the name of the probe.
+ */
+VALUE dtraceprobedesc_name(VALUE self)
+{
+  VALUE string;
+  dtrace_probedesc_t *pdp;
+
+  Data_Get_Struct(self, dtrace_probedesc_t, pdp);
+  string = rb_str_new2(pdp->dtpd_name);
+  return string;
+}
+
+static void free_probedesc(void *p)
+{
+  dtrace_probedesc_t *pdp = p;
+  free(pdp);
+}
+

data/ext/dtrace_probedesc.c~

@@ -0,0 +1,78 @@
+/* Ruby-DTrace
+ * (c) 2007 Chris Andrews <chris@nodnol.org>
+ */
+
+#include "dtrace_api.h"
+
+/* :nodoc: */
+VALUE dtraceprobedesc_init(VALUE self)
+{
+  dtrace_probedesc_t *pdp;
+
+  Data_Get_Struct(self, dtrace_probedesc_t, pdp);
+  return self;
+}
+
+/*
+ * Returns the id of the probedesc. Corresponds to the ID displayed by
+ * dtrace -l
+ */
+VALUE dtraceprobedesc_probe_id(VALUE self)
+{
+  dtrace_probedesc_t *pdp;
+
+  Data_Get_Struct(self, dtrace_probedesc_t, pdp);
+  return INT2NUM(pdp->dtpd_id);
+}
+
+/*
+ * Returns the name of the probe's provider.
+ */
+VALUE dtraceprobedesc_provider(VALUE self)
+{
+  VALUE string;
+  dtrace_probedesc_t *pdp;
+
+  Data_Get_Struct(self, dtrace_probedesc_t, pdp);
+  string = rb_str_new2(pdp->dtpd_provider);
+  return string;
+}
+
+/*
+ * Returns the name of the module where the probe is defined.
+ */
+VALUE dtraceprobedesc_mod(VALUE self)
+{
+  VALUE string;
+  dtrace_probedesc_t *pdp;
+
+  Data_Get_Struct(self, dtrace_probedesc_t, pdp);
+  string = rb_str_new2(pdp->dtpd_mod);
+  return string;
+}
+
+/*
+ * Returns the name of the function where the probe is defined.
+ */
+VALUE dtraceprobedesc_func(VALUE self)
+{
+  VALUE string;
+  dtrace_probedesc_t *pdp;
+
+  Data_Get_Struct(self, dtrace_probedesc_t, pdp);
+  string = rb_str_new2(pdp->dtpd_func);
+  return string;
+}
+
+/*
+ * Returns the name of the probe.
+ */
+VALUE dtraceprobedesc_name(VALUE self)
+{
+  VALUE string;
+  dtrace_probedesc_t *pdp;
+
+  Data_Get_Struct(self, dtrace_probedesc_t, pdp);
+  string = rb_str_new2(pdp->dtpd_name);
+  return string;
+}

data/ext/dtrace_process.c

@@ -0,0 +1,44 @@
+/* Ruby-DTrace
+ * (c) 2007 Chris Andrews <chris@nodnol.org>
+ */
+
+#include "dtrace_api.h"
+
+static void _release_process(dtrace_process_t *process)
+{
+  if (process->handle->hdl != NULL && process->proc != NULL) {
+    dtrace_proc_release(process->handle->hdl, process->proc);
+    process->proc = NULL;
+  }
+}
+
+/* :nodoc: */
+void dtrace_process_free(dtrace_process_t *process)
+{
+  _release_process(process);
+  free(process);
+}
+
+/* Release the traced process. */
+VALUE dtrace_process_release(VALUE self)
+{
+  dtrace_process_t *process;
+
+  Data_Get_Struct(self, dtrace_process_t, process);
+  _release_process(process);
+  return Qnil;
+}
+
+/*
+ * Start or restart the process. Call this having configured tracing
+ * for the process, using $target in the D program.
+ */
+VALUE dtrace_process_continue(VALUE self)
+{
+  dtrace_process_t *process;
+
+  Data_Get_Struct(self, dtrace_process_t, process);
+  dtrace_proc_continue(process->handle->hdl, process->proc);
+
+  return Qnil;
+}

data/ext/dtrace_process.c~

@@ -0,0 +1,56 @@
+/* Ruby-DTrace
+ * (c) 2007 Chris Andrews <chris@nodnol.org>
+ */
+
+#include "dtrace_api.h"
+
+/* :nodoc: */
+VALUE dtrace_process_init(VALUE self)
+{
+  dtrace_process_t *process;
+
+  Data_Get_Struct(self, dtrace_process_t, process);
+  if (process)
+    return self;
+  else
+    return Qnil;
+}
+
+static void _release_process(dtrace_process_t *process)
+{
+  if (process->handle->hdl != NULL && process->proc != NULL) {
+    dtrace_proc_release(process->handle->hdl, process->proc);
+    process->proc = NULL;
+  }
+}
+
+/* :nodoc: */
+void dtrace_process_free(dtrace_process_t *process)
+{
+  _release_process(process);
+  free(process);
+}
+
+/* Release the traced process. */
+VALUE dtrace_process_release(VALUE self)
+{
+  dtrace_process_t *process;
+
+  Data_Get_Struct(self, dtrace_process_t, process);
+  _release_process(process);
+  return Qnil;
+}
+
+/*
+ * Start or restart the process. Call this having configured tracing
+ * for the process, using $target in the D program.
+ */
+VALUE dtrace_process_continue(VALUE self)
+{
+  dtrace_process_t *process;
+
+  Data_Get_Struct(self, dtrace_process_t, process);
+  dtrace_proc_continue(process->handle->hdl, process->proc);
+
+  return Qnil;
+}

data/ext/dtrace_program.c

@@ -0,0 +1,52 @@
+/* Ruby-DTrace
+ * (c) 2007 Chris Andrews <chris@nodnol.org>
+ */
+
+#include "dtrace_api.h"
+
+RUBY_EXTERN VALUE eDTraceException;
+
+/*
+ * Execute the D program. Returns a DTraceProgramInfo object if
+ * successful, otherwise raises a DTraceException.
+ */
+VALUE dtraceprogram_exec(VALUE self)
+{
+  dtrace_prog_t *prog;
+  dtrace_proginfo_t *proginfo;
+  dtrace_handle_t *handle;
+  VALUE dtrace;
+  VALUE dtraceprograminfo;
+  int ret;
+
+  Data_Get_Struct(self, dtrace_prog_t, prog);
+  dtrace = rb_iv_get(self, "@handle");
+  Data_Get_Struct(dtrace, dtrace_handle_t, handle);
+
+  proginfo = ALLOC(dtrace_proginfo_t);
+  if (!proginfo) {
+    rb_raise(eDTraceException, "alloc failed");
+    return Qnil;
+  }
+
+  ret = dtrace_program_exec(handle->hdl, prog, proginfo);
+
+  if (ret == 0) {
+    dtraceprograminfo = dtraceprograminfo_init(proginfo);
+    rb_iv_set(self, "@proginfo", dtraceprograminfo);
+  }
+
+  if (ret < 0)
+    rb_raise(eDTraceException, "%s", dtrace_errmsg(handle->hdl, dtrace_errno(handle->hdl)));
+
+  return Qnil;
+}
+
+/*
+ * Return this program's DTraceProgramInfo object. Returns nil unless
+ * the program has been executed.
+ */
+VALUE dtraceprogram_info(VALUE self)
+{
+  return rb_iv_get(self, "@proginfo");
+}

data/ext/dtrace_program.c~

@@ -0,0 +1,62 @@
+/* Ruby-DTrace
+ * (c) 2007 Chris Andrews <chris@nodnol.org>
+ */
+
+#include "dtrace_api.h"
+
+RUBY_EXTERN VALUE eDTraceException;
+RUBY_EXTERN VALUE cDTraceProgramInfo;
+
+/* :nodoc: */
+VALUE dtraceprogram_init(VALUE self)
+{
+  dtrace_prog_t *prog;
+
+  Data_Get_Struct(self, dtrace_prog_t, prog);
+  return self;
+}
+
+/*
+ * Execute the D program. Returns a DTraceProgramInfo object if
+ * successful, otherwise raises a DTraceException.
+ */
+VALUE dtraceprogram_exec(VALUE self)
+{
+  dtrace_prog_t *prog;
+  dtrace_proginfo_t *proginfo;
+  dtrace_handle_t *handle;
+  VALUE dtrace;
+  VALUE dtraceprograminfo;
+  int ret;
+
+  Data_Get_Struct(self, dtrace_prog_t, prog);
+  dtrace = rb_iv_get(self, "@handle");
+  Data_Get_Struct(dtrace, dtrace_handle_t, handle);
+
+  proginfo = ALLOC(dtrace_proginfo_t);
+  if (!proginfo) {
+    rb_raise(eDTraceException, "alloc failed");
+    return Qnil;
+  }
+
+  ret = dtrace_program_exec(handle->hdl, prog, proginfo);
+
+  if (ret == 0) {
+    dtraceprograminfo = Data_Wrap_Struct(cDTraceProgramInfo, 0, NULL, proginfo);
+    rb_iv_set(self, "@proginfo", dtraceprograminfo);
+  }
+
+  if (ret < 0)
+          rb_raise(eDTraceException, "%s", dtrace_errmsg(handle->hdl, dtrace_errno(handle->hdl)));
+
+  return Qnil;
+}
+
+/*
+ * Return this program's DTraceProgramInfo object. Returns nil unless
+ * the program has been executed.
+ */
+VALUE dtraceprogram_info(VALUE self)
+{
+  return rb_iv_get(self, "@proginfo");
+}

data/ext/dtrace_programinfo.c

@@ -0,0 +1,70 @@
+/* Ruby-DTrace
+ * (c) 2007 Chris Andrews <chris@nodnol.org>
+ */
+
+#include "dtrace_api.h"
+
+RUBY_EXTERN VALUE cDTraceProgramInfo;
+
+static void free_proginfo(void *p);
+
+/* :nodoc: */
+VALUE dtraceprograminfo_init(dtrace_proginfo_t *proginfo)
+{
+  VALUE self;
+
+  self = Data_Wrap_Struct(cDTraceProgramInfo, 0, free_proginfo, proginfo);
+  return self;
+}
+
+/*
+ * Returns the number of aggregates associated with this program.
+ */
+VALUE dtraceprograminfo_aggregates_count(VALUE self) 
+{
+  dtrace_proginfo_t *proginfo;
+
+  Data_Get_Struct(self, dtrace_proginfo_t, proginfo);
+  return INT2NUM(proginfo->dpi_aggregates);
+}
+
+/*
+ * Returns the number of record generating probes associated with this
+ * program.
+ */
+VALUE dtraceprograminfo_recgens_count(VALUE self)
+{
+  dtrace_proginfo_t *proginfo;
+
+  Data_Get_Struct(self, dtrace_proginfo_t, proginfo);
+  return INT2NUM(proginfo->dpi_recgens);
+}
+
+/*
+ * Returns the number of probes matched by this program.
+ */
+VALUE dtraceprograminfo_matches_count(VALUE self)
+{
+  dtrace_proginfo_t *proginfo;
+
+  Data_Get_Struct(self, dtrace_proginfo_t, proginfo);
+  return INT2NUM(proginfo->dpi_matches);
+}
+
+/*
+ * Returns the number of speculations specified by this program.
+ */
+VALUE dtraceprograminfo_speculations_count(VALUE self)
+{
+  dtrace_proginfo_t *proginfo;
+
+  Data_Get_Struct(self, dtrace_proginfo_t, proginfo);
+  return INT2NUM(proginfo->dpi_speculations);
+}
+
+static void free_proginfo(void *p)
+{
+  dtrace_proginfo_t *proginfo = p;
+  free(proginfo);
+}
+