pcap2fuzz.py

#!/usr/bin/python

import sys

import Diameter as dm
from Pdml import PdmlLoader

from Dia import Directory

from struct import unpack, pack
import random
from fuzz import fuzz_msg

import pcapng

if __name__ == '__main__':
  random.seed(0)

  if len(sys.argv) != 2:
    print >>sys.stderr, 'usage: %s <.pcap>' % sys.argv[0]
    sys.exit(1)

  f = sys.stdout

  pcapng.write_shblock(f)
  pcapng.write_idblock(f)

  pcap = sys.argv[1]
  c = PdmlLoader(pcap)
  for pdu in c.pdus:
    m = dm.Msg.decode(pdu.content)
    Directory.tag(m)

    pcapng.write_epblock(f, 0, m.encode(), 'No fuzzing', 'inbound')

    for mutated, comment in fuzz_msg(m):
      mutated.e2e_id = None
      mutated.h2h_id = None

      raw = mutated.encode()
      pcapng.write_epblock(f, 0, raw, comment, 'inbound')

  f.flush()
  f.close()