raw_net_capture 1.1.0 → 2.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA1:
-  metadata.gz: 1d6cf8bbc933a2477cb83cbcb3d6dcbfdd24a31d
-  data.tar.gz: 2f5ed8a2ffa6af154039ea1d2ec27ee6ec255a58
+SHA256:
+  metadata.gz: 5bbb3ccfe245bd93d89e6ddbcca677d9f3117d3e4438f4e0a0848b6d88a938d2
+  data.tar.gz: fd845b1ee17db4a389c00986209b9f8cb31cdec1ccca5df2669f1fbf6939bb0e
 SHA512:
-  metadata.gz: aea4d1138a7f2cdd04a8675f6975011218069e74fd5fa35078389c158a7b1dafffd5c2e18a80dc857a4d221fe6065381dd892e032ecf6eb9e7c51ff1634f8f52
-  data.tar.gz: e5948bcdf2da1be5915bd72ac32458ab2ecf5e95cba15fbd07e41a9b5add9cc3dbe7189d6200f987deffb6aa430b709f9a031268c57064b25afc3b7099c4cc80
+  metadata.gz: 3fcb6307695e48b90530b5fdff48b8715c360cd9b85e95fb9ebc6d33138189a61450c1c1dec6932729cf5bd435d492bf255d671cfc607fe242f46e2b2cf720c2
+  data.tar.gz: 7e876740adee3f08c227054aec99f183f3c838b65adbe40b72884f0093ca8cc74f8f799b4a8bfdca48645510d7bbc5a90bd4939ce3ed9ddf736f2b2e73cd6909

data/lib/raw_net_capture/net/buffered_io.rb

@@ -0,0 +1,44 @@
+# encoding: utf-8
+
+module Net
+  class BufferedIO
+    private
+    alias_method :orig_rbuf_consume, :rbuf_consume
+    alias_method :orig_write0, :write0
+
+    def rbuf_consume(len = nil)
+      return orig_rbuf_consume(len) unless @debug_output
+
+      begin
+        saved_debug_output = @debug_output
+        @debug_output = nil
+        str = orig_rbuf_consume(len)
+      ensure
+        @debug_output = saved_debug_output
+      end
+
+      @debug_output << %Q[-> #{str.dump}\n]
+      @debug_output.received(str) if @debug_output.respond_to?(:received)
+      str
+    end
+
+    def write0(*strs)
+      return orig_write0(*strs) unless @debug_output
+
+      begin
+        saved_debug_output = @debug_output
+        @debug_output = nil
+        ret = orig_write0(*strs)
+      ensure
+        @debug_output = saved_debug_output
+      end
+
+      strs.each do |str|
+        @debug_output << str.dump
+        @debug_output.sent(str) if @debug_output.respond_to?(:sent)
+      end
+
+      ret
+    end
+  end
+end

data/lib/raw_net_capture/raw_http_capture.rb

@@ -0,0 +1,70 @@
+# encoding: utf-8
+
+class RawHTTPCapture < StringIO
+  def initialize
+    super
+    reset
+  end
+
+  def received(data)
+    transactions.last.response.raw_io << data
+  end
+
+  def sent(data)
+    # when there are multiple requests on the same connection, E.g. redirection, we want the last one.
+    if transactions.last.response.raw_io.length > 0
+      reset
+    end
+
+    transactions.last.request.raw_io << data
+  end
+
+  def transactions
+    @transactions ||= []
+  end
+
+  private
+
+  def reset
+    transactions << Transaction.new
+  end
+
+  class Transaction
+    attr_reader :response, :request
+
+    def initialize
+      @request = Request.new
+      @response = Response.new
+    end
+
+    class Part
+      def body
+        @body ||= partitions[2]
+      end
+
+      def headers
+        @headers ||= partitions[0]
+      end
+
+      def raw_io
+        @raw_io ||= StringIO.new
+      end
+
+      def raw
+        @raw ||= raw_io.string
+      end
+
+      private
+
+      # headers and body
+      def partitions
+        @partitions ||= raw.partition("\r\n\r\n").map do |part|
+          part.empty? ? nil : part
+        end
+      end
+    end
+
+    class Response < Part; end
+    class Request < Part; end
+  end
+end

data/lib/raw_net_capture/raw_net_capture.rb

@@ -0,0 +1,24 @@
+# encoding: utf-8
+
+class RawNetCapture < StringIO
+  attr_reader :raw_traffic
+
+  def initialize
+    super
+    reset
+  end
+
+  def received(data)
+    @raw_traffic << [:received, data]
+  end
+
+  def sent(data)
+    @raw_traffic << [:sent, data]
+  end
+
+  private
+
+  def reset
+    @raw_traffic = []
+  end
+end

data/lib/raw_net_capture.rb

@@ -1,79 +1,7 @@
 # encoding: utf-8
 
-class RawNetCapture < StringIO
-  attr_reader :raw_traffic
+gem 'net-protocol'
 
-  def initialize
-    super
-    reset
-  end
-
-  def received(data)
-    @raw_traffic << [:received, data]
-  end
-
-  def sent(data)
-    @raw_traffic << [:sent, data]
-  end
-
-  def reset
-    @raw_traffic = []
-  end
-end
-
-class RawHTTPCapture < StringIO
-  attr_reader :raw_received, :raw_sent
-
-  def initialize
-    super
-    reset
-  end
-
-  def reset
-    @raw_received = StringIO.new
-    @raw_sent = StringIO.new
-  end
-
-  def received(data)
-    @raw_received << data
-  end
-
-  def sent(data)
-    @raw_sent << data
-  end
-
-  def headers
-    separator = "\r\n\r\n"
-    raw_string = @raw_received.string
-    if headers_end_index = raw_string.index(separator)
-      raw_string[0...(headers_end_index + separator.length)]
-    else
-      raw_string
-    end
-  end
-end
-
-module Net
-  class BufferedIO
-    private
-
-    def rbuf_consume(len)
-      s = @rbuf.slice!(0, len)
-      if @debug_output
-        @debug_output << %Q[-> #{s.dump}\n]
-        @debug_output.received(s) if @debug_output.respond_to?(:received)
-      end
-      s
-    end
-
-    def write0(str)
-      if @debug_output
-        @debug_output << str.dump
-        @debug_output.sent(str) if @debug_output.respond_to?(:sent)
-      end
-      len = @io.write(str)
-      @written_bytes += len
-      len
-    end
-  end
-end
+require 'raw_net_capture/raw_net_capture'
+require 'raw_net_capture/raw_http_capture'
+require 'raw_net_capture/net/buffered_io'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: raw_net_capture
 version: !ruby/object:Gem::Version
-  version: 1.1.0
+  version: 2.1.1
 platform: ruby
 authors:
 - Gary Grossman
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-01-26 00:00:00.000000000 Z
+date: 2023-04-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: appraisal
@@ -62,6 +62,9 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/raw_net_capture.rb
+- lib/raw_net_capture/net/buffered_io.rb
+- lib/raw_net_capture/raw_http_capture.rb
+- lib/raw_net_capture/raw_net_capture.rb
 homepage: https://github.com/zendesk/raw_net_capture
 licenses:
 - Apache License Version 2.0
@@ -81,8 +84,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.2.2
+rubygems_version: 3.0.3.1
 signing_key: 
 specification_version: 4
 summary: Capture raw data in net debug_output