build-a-router/custom_proto.lua at master · cswpy/build-a-router · GitHub

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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
-- Define the custom CPU Metadata and PWOSPF protocols
local p_cpu_metadata = Proto("CPU_Metadata", "CPU Metadata Protocol")
local p_pwospf = Proto("PWOSPF", "Pee Wee OSPF Protocol")

local ethertype_arp = 0x0806 -- EtherType for ARP
local ethertype_cpu_metadata = 0x080a

-- CPU Metadata fields
local f_fromCpu = ProtoField.uint8("cpu_metadata.fromCpu", "From CPU", base.HEX)
local f_origEtherType = ProtoField.uint16("cpu_metadata.origEtherType", "Original EtherType", base.HEX)
local f_srcPort = ProtoField.uint16("cpu_metadata.srcPort", "Source Port", base.DEC)
local f_dstPort = ProtoField.uint16("cpu_metadata.dstPort", "Destination Port", base.DEC)

-- PWOSPF common fields
local f_version = ProtoField.uint8("pwospf.version", "Version", base.DEC)
local f_type = ProtoField.uint8("pwospf.type", "Type", base.DEC)
local f_packet_length = ProtoField.uint16("pwospf.packet_length", "Packet Length", base.DEC)
local f_router_id = ProtoField.ipv4("pwospf.router_id", "Router ID")
local f_area_id = ProtoField.uint32("pwospf.area_id", "Area ID")
local f_checksum = ProtoField.uint16("pwospf.checksum", "Checksum", base.HEX)
local f_autype = ProtoField.uint16("pwospf.autype", "Authentication Type", base.DEC)
local f_authentication = ProtoField.uint64("pwospf.authentication", "Authentication", base.HEX)

-- Hello Packet Fields
local f_netmask = ProtoField.ipv4("pwospf.hello.netmask", "Network Mask")
local f_helloint = ProtoField.uint16("pwospf.hello.helloint", "Hello Interval", base.DEC)
local f_padding = ProtoField.uint16("pwospf.hello.padding", "Padding", base.DEC)

-- LSU Packet Fields
local f_sequence = ProtoField.uint16("pwospf.lsu.sequence", "Sequence Number", base.DEC)
local f_ttl = ProtoField.uint16("pwospf.lsu.ttl", "TTL", base.DEC)
local f_numlsa = ProtoField.uint32("pwospf.lsu.numlsa", "Number of LSAs", base.DEC)

-- LSA Fields
local f_subnet = ProtoField.ipv4("pwospf.lsu.lsa.subnet", "Subnet")
local f_mask = ProtoField.ipv4("pwospf.lsu.lsa.mask", "Mask")
local f_lsa_router_id = ProtoField.ipv4("pwospf.lsu.lsa.router_id", "Router ID")

-- Add fields to their respective protocol
p_cpu_metadata.fields = {f_fromCpu, f_origEtherType, f_srcPort, f_dstPort}
p_pwospf.fields = {f_version, f_type, f_packet_length, f_router_id, f_area_id, f_checksum, f_autype, f_authentication,
                   f_netmask, f_helloint, f_padding, f_sequence, f_ttl, f_numlsa, f_subnet, f_mask, f_lsa_router_id}

-- CPU Metadata dissector function
function p_cpu_metadata.dissector(buffer, pinfo, tree)
    pinfo.cols.protocol = "CPU Metadata"
    local subtree = tree:add(p_cpu_metadata, buffer(), "CPU Metadata Header")

    subtree:add(f_fromCpu, buffer(0,1))
    local origEtherTypeField = subtree:add(f_origEtherType, buffer(1,2))
    subtree:add(f_srcPort, buffer(3,2))
    subtree:add(f_dstPort, buffer(5,2))

    local origEtherType = buffer(1,2):uint()

    -- Depending on the original EtherType, call appropriate dissector
    local dissector_table = DissectorTable.get("ethertype")
    local dissector = dissector_table:get_dissector(origEtherType)
    if dissector then
        dissector:call(buffer(7):tvb(), pinfo, tree)
    end
end

-- Create dissect function
function p_pwospf.dissector(buffer, pinfo, tree)
    if buffer:len() == 0 then return end
    pinfo.cols.protocol = "PWOSPF"
    local subtree = tree:add(p_pwospf, buffer(), "PWOSPF Protocol Data")
    local packet_length = buffer(2,2):uint()

    if packet_length > buffer:len() then
        pinfo.desegment_len = packet_length - buffer:len()
        pinfo.desegment_offset = 0
        return
    end

    subtree:add(f_version, buffer(0,1))
    local type = buffer(1,1):uint()
    subtree:add(f_type, buffer(1,1))
    subtree:add(f_packet_length, buffer(2,2))
    subtree:add(f_router_id, buffer(4,4))
    subtree:add(f_area_id, buffer(8,4))
    subtree:add(f_checksum, buffer(12,2))
    subtree:add(f_autype, buffer(14,2))
    subtree:add(f_authentication, buffer(16,8))

    local payload_offset = 24
    if buffer:len() > payload_offset then

        -- Handling different types based on type field
        if type == 1 then
            local payload_tree = subtree:add(buffer(payload_offset), "PWOSPF HELLO Packet")
            -- PWOSPF Hello packet
            pinfo.cols.info:set("PWOSPF Hello")
            subtree:add(f_netmask, buffer(payload_offset, 4))
            subtree:add(f_helloint, buffer(payload_offset + 4, 2))
            subtree:add(f_padding, buffer(payload_offset + 6, 2))
        elseif type == 4 then
            local payload_tree = subtree:add(buffer(payload_offset), "PWOSPF LSU Packet")
            -- PWOSPF LSU packet
            pinfo.cols.info:set("PWOSPF LSU")
            subtree:add(f_sequence, buffer(payload_offset, 2))
            subtree:add(f_ttl, buffer(payload_offset + 2, 2))
            local numlsa = buffer(payload_offset + 4, 4):uint()
            subtree:add(f_numlsa, buffer(payload_offset + 4, 4))
            local lsa_offset = payload_offset + 8
            for i = 1, numlsa do
                local lsa_tree = subtree:add(p_pwospf, buffer(lsa_offset, 12), "Link State Advertisement "..i)
                lsa_tree:add(f_subnet, buffer(lsa_offset, 4))
                lsa_tree:add(f_mask, buffer(lsa_offset + 4, 4))
                lsa_tree:add(f_lsa_router_id, buffer(lsa_offset + 8, 4))
                lsa_offset = lsa_offset + 12 -- move to the next LSA
            end
        else
            pinfo.cols.info:set("Unknown PWOSPF Type")
        end
    end
    return true
end

-- Register the protocol dissectors to Ethernet type
local ethertype_table = DissectorTable.get("ethertype")
ethertype_table:add(ethertype_cpu_metadata, p_cpu_metadata)
local ip_proto_table = DissectorTable.get("ip.proto")
ip_proto_table:add(89, p_pwospf)