File: 100kpackets.rb

package info (click to toggle)
ruby-packetfu 2.0.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 3,520 kB
  • sloc: ruby: 8,344; makefile: 2
file content (43 lines) | stat: -rw-r--r-- 1,322 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#!/usr/bin/env ruby
# -*- coding: binary -*-

# Used mainly to test for memory leaks and to demo the preferred ways of
# reading and writing packets to and from pcap files.

# Usage:
# ruby examples/100kpackets.rb

# Path setting slight of hand:
$: << File.expand_path("../../lib", __FILE__)
require 'packetfu'

puts "Generating packets... (#{Time.now.utc})"

File.unlink("/tmp/out.pcap") if File.exist? "/tmp/out.pcap"
start_time = Time.now.utc
count = 0

100.times do
  @pcaps = []
  1000.times do
    u = PacketFu::UDPPacket.new
    u.ip_src = [rand(2**32-1)].pack("N")
    u.ip_dst = [rand(2**32-1)].pack("N")
    u.recalc
    @pcaps << u
  end
  pfile = PacketFu::PcapFile.new
  res = pfile.array_to_file(:filename => "/tmp/out.pcap", :array => @pcaps, :append => true)
  count += res.last
  puts "Wrote #{count} packets in #{Time.now.utc - start_time} seconds"
end

read_bytes_start = Time.now.utc
puts "Reading packet bytes..."
packet_bytes = PacketFu::PcapFile.read_packet_bytes "/tmp/out.pcap"
puts "Read #{packet_bytes.size} packet byte blobs in #{Time.now.utc - read_bytes_start} seconds."

read_packets_start = Time.now.utc
puts "Reading packets..."
packet_bytes = PacketFu::PcapFile.read_packets "/tmp/out.pcap"
puts "Read #{packet_bytes.size} parsed packets in #{Time.now.utc - read_packets_start} seconds."