raw_net_capture 1.1.0 → 2.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 1d6cf8bbc933a2477cb83cbcb3d6dcbfdd24a31d
-  data.tar.gz: 2f5ed8a2ffa6af154039ea1d2ec27ee6ec255a58
+  metadata.gz: 0cddcb536ac8cc22edf1b0b2605deb406f783019
+  data.tar.gz: 59abcb2f70e8f5e61a7958bd867ab4107c8aaf02
 SHA512:
-  metadata.gz: aea4d1138a7f2cdd04a8675f6975011218069e74fd5fa35078389c158a7b1dafffd5c2e18a80dc857a4d221fe6065381dd892e032ecf6eb9e7c51ff1634f8f52
-  data.tar.gz: e5948bcdf2da1be5915bd72ac32458ab2ecf5e95cba15fbd07e41a9b5add9cc3dbe7189d6200f987deffb6aa430b709f9a031268c57064b25afc3b7099c4cc80
+  metadata.gz: cb2a51061d463901d575be9384a734838cbcbaa925b20046936a321cc1c739030a1bab8381d242bbd3023db056dc7233bdf853b7029deee7aa24fc7893239250
+  data.tar.gz: aa6cd0a9fc9bbb80f6045417b4bcf5a59ef7046bbad8020899bdc92cbfac9f6f927c9162d3ef31f83002880294103ce0775876ec7a619a79906e40c059c6d724

data/lib/raw_net_capture.rb

@@ -1,79 +1,5 @@
 # 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
-
-  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'

data/lib/raw_net_capture/net/buffered_io.rb

@@ -0,0 +1,27 @@
+# encoding: utf-8
+
+module Net
+  class BufferedIO
+    private
+
+    def rbuf_consume(len)
+      str = @rbuf.slice!(0, len)
+      if @debug_output
+        @debug_output << %Q[-> #{str.dump}\n]
+        @debug_output.received(str) if @debug_output.respond_to?(:received)
+      end
+      str
+    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

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: raw_net_capture
 version: !ruby/object:Gem::Version
-  version: 1.1.0
+  version: 2.0.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: 2015-02-25 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