rbtrace 0.4.13 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 942f750987d2faf32e6d104d097bb8dfa68dda43644750259a1bc0d3a23a47d6
-  data.tar.gz: 1fb44c1bbf4936871cef319331d27f9743debe12d28ca05235fc1750e70aa261
+  metadata.gz: dc6fc7320951955050f41babf4a1effbec1eaac42e9103a0c8b08c69cc2ad090
+  data.tar.gz: 105f410c0c78e6f23c3d6617165b7a1c3c9a72976735cfd185481c91acb3bb17
 SHA512:
-  metadata.gz: df57be716aff5cc91cd2ce54e3bb55015b76a4390fe86b7f68b91a63794364387cd9edae630f6f83584c5288f8d2fcdd3a191c6c43e75a1cdabfb064ae5cd0be
-  data.tar.gz: 60cf530a8271c124dd1cd8df54761812e951aa9a722df89da12c5be163a87838b7fcce03b12244699df732424a5eeaf26e6ac80270f4afaa79f756875654521a
+  metadata.gz: a4e42ec27236d32067f52fa3c0ed1a1a3eb518f1f0d4de9c42dcbec787dc62346f39c790ae2b74fcca6835bc1413c3e44216398c8ca46d74d760f0d75a551873
+  data.tar.gz: 4602057b9797b58337d2308a195ee1d1e6e9247a7e08bebf40d153bd9c6f4fb1f4871851b514b0fef471cf87df002632b0c75ff6511dc926e58c86be453d8b67

data/.github/workflows/ci.yml

@@ -0,0 +1,26 @@
+---
+name: Test
+on: [push, pull_request]
+
+jobs:
+  ruby:
+    name: Ruby ${{ matrix.ruby }}
+    timeout-minutes: 15
+    strategy:
+      fail-fast: false
+      matrix:
+        os: ["ubuntu-latest"]
+        ruby: ["3.2", "3.1", "3.0", "2.7", "2.6"]
+    runs-on: ubuntu-latest
+    steps:
+      - name: Check out code
+        uses: actions/checkout@v3
+
+      - name: Set up Ruby
+        uses: ruby/setup-ruby@v1
+        with:
+          ruby-version: ${{ matrix.ruby }}
+          bundler-cache: true
+
+      - name: Tests
+        run: ./test.sh

data/CHANGELOG

@@ -0,0 +1,7 @@
+- 0.5.0 - 02-12-2023
+
+- Improved --backtraces implementation
+- Added --shapedump which is useful for lighter weight object shape dumps on Ruby 3.2 and up
+- Move heap dumping to a fork for better performance
+- Use TypedData API which has been supported for a very long time
+- Update CI and fix some small errors

data/ext/extconf.rb

@@ -13,7 +13,7 @@ require 'fileutils'
 
 libdir = File.basename RbConfig::CONFIG['libdir']
 
-unless File.exists?("#{CWD}/dst/#{libdir}/libmsgpackc.a")
+unless File.exist?("#{CWD}/dst/#{libdir}/libmsgpackc.a")
   Logging.message "Building msgpack\n"
 
   msgpack = File.basename('msgpack-1.1.0.tar.gz')
@@ -28,7 +28,7 @@ unless File.exists?("#{CWD}/dst/#{libdir}/libmsgpackc.a")
   end
 
   Dir.chdir('src') do
-    FileUtils.rm_rf(dir) if File.exists?(dir)
+    FileUtils.rm_rf(dir) if File.exist?(dir)
 
     sys("tar zxvfo #{msgpack}")
     Dir.chdir(dir) do
@@ -63,11 +63,6 @@ have_func('rb_during_gc', 'ruby.h')
 have_func('rb_gc_add_event_hook', ['ruby.h', 'node.h'])
 have_func('rb_postponed_job_register_one', 'ruby.h')
 
-# increase message size on linux
-if RUBY_PLATFORM =~ /linux/
-  $defs.push("-DBUF_SIZE=256")
-end
-
 # warnings save lives
 $CFLAGS << " -Wall "
 

data/ext/rbtrace.c

@@ -22,6 +22,7 @@
 
 #include <msgpack.h>
 #include <ruby.h>
+#include <ruby/debug.h>
 
 #ifndef RUBY_VM
 #include <env.h>
@@ -51,6 +52,11 @@
 #endif
 
 
+// The SUN_LEN macro is not available on Android
+#ifndef SUN_LEN
+#define SUN_LEN(ptr) ((size_t) (((struct sockaddr_un *) 0)->sun_path) + strlen((ptr)->sun_path))
+#endif
+
 static uint64_t
 ru_utime_usec()
 {
@@ -73,7 +79,7 @@ timeofday_usec()
 #define MAX_TRACERS 100 // max method tracers
 #define MAX_EXPRS 10    // max expressions per tracer
 #ifndef BUF_SIZE        // msgq buffer size
-#define BUF_SIZE 120
+#define BUF_SIZE 1024
 #endif
 
 typedef struct {
@@ -688,7 +694,7 @@ rbtracer_add(char *query, bool is_slow)
       *idx = NULL,
       *method = NULL;
 
-    if (NULL != (idx = rindex(query, '.'))) {
+    if (NULL != (idx = strrchr(query, '.'))) {
       klass_begin = 0;
       klass_end = idx - query;
       is_singleton = true;
@@ -700,7 +706,7 @@ rbtracer_add(char *query, bool is_slow)
 
       method = idx+1;
 
-    } else if (NULL != (idx = rindex(query, '#'))) {
+    } else if (NULL != (idx = strrchr(query, '#'))) {
       klass_begin = 0;
       klass_end = idx - query;
       is_singleton = false;
@@ -870,9 +876,7 @@ eval_inspect(VALUE rb_code) {
 }
 
 static VALUE
-rescue_inspect(VALUE arg) {
-  VALUE exception = rb_errinfo(); /* get last exception */
-  rb_set_errinfo(Qnil);
+rescue_inspect(VALUE arg, VALUE exception) {
   return rb_funcall(exception, rb_intern("inspect"), 0);
 }
 
@@ -885,7 +889,6 @@ rbtrace__process_event(msgpack_object cmd)
   static int last_tracer_id = -1; // hax
   char query[BUF_SIZE];
 
-  char code[BUF_SIZE+150];
   VALUE val = Qnil;
 
   msgpack_object_array ary;
@@ -1129,31 +1132,26 @@ send_write(VALUE klass, VALUE val) {
   return Qnil;
 }
 
+static const rb_data_type_t rbtrace_type = {
+    "RBTrace",
+    {
+        rbtrace_gc_mark,
+    }
+};
+
 void
 Init_rbtrace()
 {
   rbtrace_module = rb_define_module("RBTrace");
   VALUE output = rb_define_module_under(rbtrace_module, "OUT");
 
-  rb_define_singleton_method(output, "write", send_write, 1);
+  rb_const_set(rbtrace_module, rb_intern("BUF_SIZE"), INT2NUM(BUF_SIZE));
 
-  rb_eval_string(
-    "module RBTrace\n"
-    "  def self.eval_context\n"
-    "    @eval_context ||= binding\n"
-    "  end\n"
-
-    "  def self.eval_and_inspect(code)\n"
-    "    t = Thread.new { Thread.current[:output] = eval_context.eval(code).inspect }\n"
-    "    t.join\n"
-    "    t[:output]\n"
-    "  end\n"
-    "end\n"
-  );
+  rb_define_singleton_method(output, "write", send_write, 1);
 
   // hook into the gc
-  gc_hook = Data_Wrap_Struct(rb_cObject, rbtrace_gc_mark, NULL, NULL);
   rb_global_variable(&gc_hook);
+  gc_hook = TypedData_Wrap_Struct(rb_cObject, &rbtrace_type, NULL);
 
   // catch signal telling us to read from the msgq
 #if defined(HAVE_RB_POSTPONED_JOB_REGISTER_ONE)

data/lib/rbtrace/cli.rb

@@ -12,7 +12,7 @@ class RBTraceCLI
   #
   # Returns nothing.
   def self.check_msgmnb
-    if File.exists?(msgmnb = "/proc/sys/kernel/msgmnb")
+    if File.exist?(msgmnb = "/proc/sys/kernel/msgmnb")
       curr = File.read(msgmnb).to_i
       max = 1024*1024
       cmd = "sysctl kernel.msgmnb=#{max}"
@@ -225,6 +225,9 @@ EOS
         :default => "AUTO",
         :short => "-h"
 
+      opt :shapesdump,
+        "generate a shapes dump for the process in FILENAME",
+        :default => "AUTO"
     end
 
     opts = Optimist.with_standard_exception_handling(parser) do
@@ -241,7 +244,7 @@ EOS
     end
 
     unless %w[ fork eval interactive backtrace backtraces slow slowcpu firehose methods config gc memory heapdump].find{ |n| opts[:"#{n}_given"] }
-      $stderr.puts "Error: --slow, --slowcpu, --gc, --firehose, --methods, --interactive, --backtraces, --backtrace, --memory, --heapdump or --config required."
+      $stderr.puts "Error: --slow, --slowcpu, --gc, --firehose, --methods, --interactive, --backtraces, --backtrace, --memory, --heapdump, --shapesdump or --config required."
       $stderr.puts "Try --help for help."
       exit(-1)
     end
@@ -273,7 +276,7 @@ EOS
         file = [
           config,
           File.expand_path("../../../tracers/#{config}.tracer", __FILE__)
-        ].find{ |f| File.exists?(f) }
+        ].find{ |f| File.exist?(f) }
 
         unless file
           parser.die :config, '(file does not exist)'
@@ -449,7 +452,7 @@ EOS
 
         delim = "146621c9d681409aa"
 
-        code = "Thread.list.map{|t| t.backtrace[0...#{num}].join(\"#{delim}\")}.join(\"#{delim*2}\")"
+        code = "Thread.list.reject { |t| t.name == '__RBTrace__' }.map{ |t| t.backtrace[0...#{num}].join(\"#{delim}\")}.join(\"#{delim*2}\")"
 
         if res = tracer.eval(code)
           tracer.puts res.split(delim).join("\n")
@@ -499,9 +502,47 @@ EOS
           temp.unlink
         end
 
-        tracer.eval("file = File.open('#{filename}', 'w'); ObjectSpace.dump_all(output: file); file.close")
+        tracer.eval(<<-RUBY)
+          Thread.new do
+            Thread.current.name = '__RBTrace__'
+            pid = ::Process.fork do
+              file = File.open('#{filename}.tmp', 'w')
+              ObjectSpace.dump_all(output: file)
+              file.close
+              File.rename('#{filename}.tmp', '#{filename}')
+              exit!(0)
+            end
+            Process.waitpid(pid)
+          end
+        RUBY
         puts "Heapdump being written to #{filename}"
 
+      elsif opts[:shapesdump_given]
+        filename = opts[:shapesdump]
+
+        if filename == "AUTO"
+          require 'tempfile'
+          temp = Tempfile.new("dump")
+          filename = temp.path
+          temp.close
+          temp.unlink
+        end
+
+        tracer.eval(<<-RUBY)
+          Thread.new do
+            Thread.current.name = '__RBTrace__'
+            pid = ::Process.fork do
+              file = File.open('#{filename}.tmp', 'w')
+              ObjectSpace.dump_shapes(output: file)
+              file.close
+              File.rename('#{filename}.tmp', '#{filename}')
+              exit!(0)
+            end
+            Process.waitpid(pid)
+          end
+        RUBY
+        puts "Shapes dump being written to #{filename}"
+
       elsif opts[:eval_given]
         if res = tracer.eval(code = opts[:eval])
           tracer.puts ">> #{code}"

data/lib/rbtrace/core_ext.rb

@@ -1,7 +1,3 @@
-class String
-  alias :bytesize :size
-end unless ''.respond_to?(:bytesize)
-
 module FFI::LastError
   Errnos = Errno::constants.map(&Errno.method(:const_get)).inject({}) do |hash, c|
     hash[ c.const_get(:Errno) ] = c

data/lib/rbtrace/msgq.rb

@@ -1,3 +1,4 @@
+require 'rbtrace'
 require 'ffi'
 
 module MsgQ
@@ -5,7 +6,7 @@ module MsgQ
   ffi_lib FFI::CURRENT_PROCESS
 
   class EventMsg < FFI::Struct
-    BUF_SIZE = RUBY_PLATFORM =~ /linux/ ? 256 : 120
+    BUF_SIZE = RBTrace::BUF_SIZE
     IPC_NOWAIT = 004000
 
     layout :mtype, :long,

data/lib/rbtrace/rbtracer.rb

@@ -96,7 +96,7 @@ class RBTracer
   end
 
   def clean_socket_path
-    FileUtils.rm(socket_path) if File.exists?(socket_path)
+    FileUtils.rm(socket_path) if File.exist?(socket_path)
   end
 
   # Watch for method calls slower than a threshold.
@@ -323,7 +323,9 @@ class RBTracer
       msg = cmd.to_msgpack
       # A message is null-terminated, but bytesize gives the unterminated
       # length.
-      raise ArgumentError, 'command is too long' if msg.bytesize >= MsgQ::EventMsg::BUF_SIZE
+      if msg.bytesize >= RbTrace::BUF_SIZE
+        raise ArgumentError, "command is too long (#{msg.bytesize}B >= #{MsgQ::EventMsg::BUF_SIZE}B)"
+      end
       MsgQ::EventMsg.send_cmd(@qo, msg)
     rescue Errno::EINTR
       retry

data/lib/rbtrace/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 class RBTracer
-  VERSION = '0.4.13'
+  VERSION = "0.5.0"
 end

data/lib/rbtrace.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module RBTrace
+  class << self
+    def eval_and_inspect(code)
+      t = Thread.new do
+        Thread.current.name = '__RBTrace__'
+        Thread.current[:output] = eval_context.eval(code).inspect
+      end
+      t.join
+      t[:output]
+    end
+
+    private
+
+    def eval_context
+      @eval_context ||= binding
+    end
+  end
+end
+
+require 'rbtrace.so'

data/rbtrace.gemspec

@@ -23,7 +23,7 @@ Gem::Specification.new do |s|
   s.add_dependency 'optimist', '>= 3.0.0'
   s.add_dependency 'msgpack',  '>= 0.4.3'
 
-  s.add_development_dependency "rake", "~> 10.0"
+  s.add_development_dependency "rake"
 
   s.license = "MIT"
   s.summary = 'rbtrace: like strace but for ruby code'

data/server.rb

@@ -1,4 +1,5 @@
-require 'ext/rbtrace'
+require 'rbtrace'
+require 'tmpdir'
 
 class String
   def multiply_vowels(num)
@@ -26,19 +27,21 @@ end
 
 while true
   proc {
-    Dir.chdir("/tmp") do
-      Dir.pwd
-      Process.pid
-      'hello'.multiply_vowels(3){ :ohai }
-      sleep rand*0.5
+    Dir.mktmpdir do |tmp|
+      Dir.chdir(tmp) do
+        Dir.pwd
+        Process.pid
+        'hello'.multiply_vowels(3){ :ohai }
+        sleep rand*0.5
 
-      ENV['blah']
-      GC.start
+        ENV['blah']
+        GC.start
 
-      reload_test.call
-      Test.run
+        reload_test.call
+        Test.run
 
-      #fib(1024*100)
+        #fib(1024*100)
+      end
     end
   }.call
 end

data/test.sh

@@ -12,7 +12,7 @@ cd ..
 bundle check
 export RUBYOPT="-I.:lib"
 
-ruby server.rb &
+bundle exec ruby server.rb &
 export PID=$!
 
 trap cleanup INT TERM
@@ -23,11 +23,11 @@ cleanup() {
 
 trace() {
   echo ------------------------------------------
-  echo ./bin/rbtrace -p $PID $*
+  echo ./bin/rbtrace -p $PID "$@"
   echo ------------------------------------------
-  ./bin/rbtrace -p $PID $* &
+  bundle exec ./bin/rbtrace -p $PID "$@" &
   sleep 2
-  kill $!
+  kill $! || true
   wait $! || true
   echo
 }
@@ -37,6 +37,8 @@ trace -m sleep
 trace -m sleep Dir.chdir Dir.pwd Process.pid "String#gsub" "String#*"
 trace -m "Kernel#"
 trace -m "String#gsub(self,@test)" "String#*(self,__source__)" "String#multiply_vowels(self,self.length,num)"
+trace -e 'p(1 + 1)'
+trace -h
 trace --gc --slow=200
 trace --gc -m Dir.
 trace --slow=250

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rbtrace
 version: !ruby/object:Gem::Version
-  version: 0.4.13
+  version: 0.5.0
 platform: ruby
 authors:
 - Aman Gupta
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2020-05-19 00:00:00.000000000 Z
+date: 2023-12-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ffi
@@ -56,16 +56,16 @@ dependencies:
   name: rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '10.0'
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '10.0'
+        version: '0'
 description: rbtrace shows you method calls happening inside another ruby process
   in real time.
 email: aman@tmm1.net
@@ -75,7 +75,9 @@ extensions:
 - ext/extconf.rb
 extra_rdoc_files: []
 files:
+- ".github/workflows/ci.yml"
 - ".gitignore"
+- CHANGELOG
 - Gemfile
 - LICENSE
 - README.md
@@ -85,6 +87,7 @@ files:
 - ext/extconf.rb
 - ext/rbtrace.c
 - ext/src/msgpack-1.1.0.tar.gz
+- lib/rbtrace.rb
 - lib/rbtrace/cli.rb
 - lib/rbtrace/core_ext.rb
 - lib/rbtrace/interactive/irb.rb
@@ -107,7 +110,7 @@ homepage: http://github.com/tmm1/rbtrace
 licenses:
 - MIT
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -123,8 +126,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
-signing_key: 
+rubygems_version: 3.4.19
+signing_key:
 specification_version: 4
 summary: 'rbtrace: like strace but for ruby code'
 test_files: []