/
scatter.go
148 lines (128 loc) · 3.55 KB
/
scatter.go
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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
/*
* NETCAP - Traffic Analysis Framework
* Copyright (c) 2017-2020 Philipp Mieden <dreadl0ck [at] protonmail [dot] ch>
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
package manager
import (
"io"
"log"
"os"
"path/filepath"
"sort"
"strings"
"time"
"github.com/go-echarts/go-echarts/v2/charts"
"github.com/go-echarts/go-echarts/v2/components"
"github.com/go-echarts/go-echarts/v2/opts"
)
var instance *LabelManager
type scatterData struct {
time time.Time
value int
}
// scatterDataSlice implements sort.Interface to sort scatter data points based on their timestamp.
type scatterDataSlice []scatterData
// Len will return the length.
func (d scatterDataSlice) Len() int {
return len(d)
}
// Less will return true if the value at index i is smaller than the other one.
func (d scatterDataSlice) Less(i, j int) bool {
return d[i].time.Before(d[j].time)
}
// Swap will switch the values.
func (d scatterDataSlice) Swap(i, j int) {
d[i], d[j] = d[j], d[i]
}
func generateScatterItems(values []scatterData) []opts.ScatterData {
items := make([]opts.ScatterData, len(values))
for i := 0; i < len(values); i++ {
items[i] = opts.ScatterData{
Value: values[i].value,
Symbol: "circle",
SymbolSize: 7,
SymbolRotate: 0,
}
}
return items
}
func scatterShowLabel(filename string, attacks map[time.Time]int, normal map[time.Time]int) *charts.Scatter {
scatter := charts.NewScatter()
scatter.SetGlobalOptions(charts.WithTitleOpts(
opts.Title{
Title: "Labels for " + filename,
Subtitle: "Displayed is the label count in " + instance.scatterDuration.String() + " intervals",
}),
charts.WithLegendOpts(
opts.Legend{
Show: true,
},
),
)
var attackData scatterDataSlice
for t, v := range attacks {
attackData = append(attackData, scatterData{
time: t,
value: v,
})
}
sort.Sort(attackData)
var normalData scatterDataSlice
for t, v := range normal {
normalData = append(normalData, scatterData{
time: t,
value: v,
})
}
sort.Sort(normalData)
var times []time.Time
for _, s := range normalData {
times = append(times, s.time)
}
scatter.SetXAxis(times).
AddSeries("Attack", generateScatterItems(attackData)).
AddSeries("Normal", generateScatterItems(normalData)).
SetSeriesOptions(charts.WithLabelOpts(
opts.Label{
Show: true,
Position: "right",
}),
)
return scatter
}
// Render will render the current label manager
func Render(out string) {
if instance != nil {
// TODO make configurable
filename := filepath.Base(out)
if filename == "" {
filename = "label"
}
// discard all file extensions if there are any
parts := strings.Split(filename, ".")
if len(parts) > 1 {
filename = parts[0]
}
page := components.NewPage()
page.PageTitle = "NETCAP label scatterplot"
page.AddCharts(
scatterShowLabel(filename, instance.scatterAttackMap, instance.scatterNormalMap),
)
f, err := os.Create(filepath.Join(out, filename+"-scatter.html"))
if err != nil {
panic(err)
}
err = page.Render(io.MultiWriter(f))
if err != nil {
log.Println("failed to render label scatter plot", err)
}
}
}