extract method to yield packets back from each capture.

1 files changed, 9 insertions, 1 deletions

diff --git a/lib/fake_agent.rb b/lib/fake_agent.rb
index 227ae2b..1f8b221 100644
--- a/lib/fake_agent.rb
+++ b/lib/fake_agent.rb
@@ -45,7 +45,15 @@ class FakeAgent
     capture.stream.each do |p|
       packet = Packet.parse(p)
       if packet.is_ip?
-        next if packet.ip_saddr == Utils.ifconfig(interface)[:ip_saddr]
+        yield packet if block_given?
+      end
+    end
+  end
+
+  def packet_capture(interface)
+    sniff(interface) do |packet|
+      if packet.ip_saddr == Utils.ifconfig(interface)[:ip_saddr]
+      else
         packet_info = [packet.ip_saddr, packet.ip_daddr, packet.size, packet.proto.last]
         puts "%-15s -> %-15s %-4d %s" % packet_info
       end