summaryrefslogtreecommitdiff
path: root/src/main/lua/pcap/tcpDataAmountStats.lua
blob: 496687a49948407db5246f1d60d0faa1c30afc76 (plain)
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

local newPcapParser = assert(require("pcapit").newPcapParser)

local main, onPcapFrame, vapourizeUrlVariables, printResult


function main()
    local app = {
        parser = false,
        youngestEpochSec = -math.huge,
        oldestEpochSec = math.huge,
        nextStreamNr = 1,
        httpStreams = {},
    }
    app.parser = newPcapParser{
        dumpFilePath = "-",
        onFrame = function(f)onPcapFrame(app, f)end,
    }
    app.parser:resume()
    printResult(app)
end


function onPcapFrame( app, it )
    local out = io.stdout
    --
    if not it:tcpSeqNr() then return end
    --
    --
    local sec, usec = it:frameArrivalTime()
    if sec < app.oldestEpochSec then app.oldestEpochSec = sec end
    if sec > app.youngestEpochSec then app.youngestEpochSec = sec end
    --
    local srcIp, dstIp = it:netSrcIpStr(), it:netDstIpStr()
    local srcPort, dstPort = it:trspSrcPort(), it:trspDstPort()
    local lowIp = (srcIp < dstIp)and(srcIp)or(dstIp)
    local higIp = (lowIp == dstIp)and(srcIp)or(dstIp)
    local lowPort = math.min(srcPort, dstPort)
    local streamId = lowIp .."\0".. higIp .."\0".. lowPort
    local stream = app.httpStreams[streamId]
    if not stream then
        stream = {
            srcIp = srcIp, dstIp = dstIp, srcPort = srcPort, dstPort = dstPort,
            streamNr = app.nextStreamNr, numBytes = 0,
        }
        app.nextStreamNr = app.nextStreamNr + 1
        app.httpStreams[streamId] = stream
    end
    local trspPayload = it:trspPayload()
    stream.numBytes = stream.numBytes + trspPayload:len()
end


function printResult( app )
    local out = io.stdout
    local sorted = {}
    local overalValue, maxValue = 0, 0
    for _, stream in pairs(app.httpStreams) do
        if stream.numBytes > maxValue then maxValue = stream.numBytes end
        overalValue = overalValue + stream.numBytes
        table.insert(sorted, stream)
    end
    table.sort(sorted, function(a, b)return a.numBytes > b.numBytes end)
    local dumpDurationSec = app.youngestEpochSec - app.oldestEpochSec
    local overallBytesPerSec = overalValue / dumpDurationSec
    local maxValuePerSec = maxValue / dumpDurationSec
    local timeFmt = "!%Y-%m-%d_%H:%M:%SZ"
    out:write("\n")
    out:write(string.format("   Subject  TCP data throughput\n"))
    out:write(string.format("     Begin  %s\n", os.date(timeFmt,app.oldestEpochSec)))
    out:write(string.format("  Duration  %d seconds\n", dumpDurationSec))
    out:write(string.format("   Overall  %.3f KiB per second (%.3f KiBit per second)\n",
        overallBytesPerSec/1024, overallBytesPerSec/1024*8))
    out:write("\n")
    out:write("   .-- KiB per Second\n")
    out:write("   |            .-- IP endpoints\n")
    out:write("   |            |                          .-- TCP server port\n")
    out:write("   |            |                          |       .-- TCP Payload (less is better)\n")
    out:write("   |            |                          |       |\n")
    out:write(".--+----.  .----+----------------------.  .+--.  .-+------------\n")
    local bar = "XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX"
    for i, elem in ipairs(sorted) do
        local streamNr, srcIp, dstIp, srcPort, dstPort, numBytes =
            elem.streamNr, elem.srcIp, elem.dstIp, elem.srcPort, elem.dstPort, elem.numBytes
        local lowPort = math.min(srcPort, dstPort)
        local bytesPerSecond = math.floor((numBytes / dumpDurationSec)*10+.5)/10
        out:write(string.format("%9.3f  %-14s %-14s  %5d ", bytesPerSecond/1024, srcIp, dstIp, lowPort))
        local part = bytesPerSecond / maxValuePerSec;
        out:write(bar:sub(0, math.floor(part * bar:len())))
        out:write("\n")
    end
    out:write("\n")
end


main()