File: test-HomeFlow.py

package info (click to toggle)
python-libtrace 1.6%2Bgit20161027-1
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 2,124 kB
  • ctags: 1,357
  • sloc: ansic: 6,890; python: 3,228; makefile: 70; sh: 49
file content (89 lines) | stat: -rwxr-xr-x 2,415 bytes parent folder | download | duplicates (2)
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#!/usr/bin/env python

# Thu, 13 Mar 14 (PDT)
# ip6.py:  Demonstrate IPv6 objects
# Copyright (C) 2015, Nevil Brownlee, U Auckland | WAND

#import ipp
import plt
import natkit
from plt_testing import *
import sys

UAv4 = ipp.from_s("130.216.0.0/16")
UAv6 = ipp.from_s("2001:df0::/47")
linklocal = ipp.from_s("fe80::/10")
multicast = ipp.from_s("ff00::/8")

fh = natkit.FlowHome(UAv4, UAv6)  # List of 'home' prefixes'
#  hf = fh(pkt) returns:
#    hf.src_in_home
#    hf.dst_in_home
#
#    hf.home_key = if src in home or dst in home 
#                   flow key with home address as dest
#                  else  None
#    hf.inward   = True if pkt dest was in home
#                    (outward home_key has addresses and ports swapped)

def print_flow(n, ipf, tag=''):
    test_println("%5d: %d %3d  %5d %5d  %s  %s" % (n,  # v6
        ipf.version, ipf.proto, ipf.src_port, ipf.dst_port,
        ipf.src_prefix, ipf.dst_prefix), tag+get_tag())

    #key = ipf.fwd_key
    #for b in key:
    #    print " %02x" % ord(b),
    #print

    test_println("          src_home=%s, dst_home=%s, is_inward=%s" % (
        ipf.src_in_home, ipf.dst_in_home, ipf.is_inward), tag+get_tag())
            

def test_uri(uri, tag=''):
    t = plt.trace(uri)
    t.start()

    nip6 = nip = n = 0
    for pkt in t:
        n += 1

        if not (pkt.ip or pkt.ip6):
            continue

        try:
            ipf = fh.flow(pkt)
        except:
            test_println("%4d: probably not an IP packet" % n, tag+get_tag("n:"+str(n)))
            continue

        if not ipf.is_inward:
            print_flow(n, ipf, tag+get_tag("n:"+str(n)))
        else:
            if pkt.ip:
                nip += 1
                #if nip <= 4:
                print_flow(n, ipf, tag+get_tag("n:"+str(n)))
            elif pkt.ip6:
                nip6 += 1
                #if nip6 <= 4:
                print_flow(n, ipf, tag+get_tag("n:"+str(n)))

            fwd = ipf.fwd_key
            test_print(" fwd =", tag+get_tag("n:"+str(n)))
            for b in fwd:
                test_print(" %02x" % ord(b))

            hk = ipf.home_key
            test_println('')
            test_print(" hom =", tag+get_tag("n:"+str(n)))
            for b in hk:
                test_print(" %02x" % ord(b))
            test_println('')

        #if n == 480:
        #    break

    t.close()

test_uri("pcapfile:home-test.pcap", get_tag())