-
Notifications
You must be signed in to change notification settings - Fork 11
Expand file tree
/
Copy pathdivert_dll.go
More file actions
83 lines (71 loc) · 1.54 KB
/
divert_dll.go
File metadata and controls
83 lines (71 loc) · 1.54 KB
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
//go:build windows && !divert_cgo && !divert_rsrc && !divert_embed && (amd64 || 386 || arm64)
package divert
import (
"fmt"
"strconv"
"strings"
"golang.org/x/sys/windows"
)
var (
winDivert = (*windows.DLL)(nil)
winDivertOpen = (*windows.Proc)(nil)
winDivertCalcChecksums = (*windows.Proc)(nil)
)
// Open is ...
func Open(filter string, layer Layer, priority int16, flags uint64) (h *Handle, err error) {
once.Do(func() {
dll, er := windows.LoadDLL("WinDivert.dll")
if er != nil {
err = er
return
}
winDivert = dll
proc, er := winDivert.FindProc("WinDivertOpen")
if er != nil {
err = er
return
}
winDivertOpen = proc
proc, er = winDivert.FindProc("WinDivertHelperCalcChecksums")
if er != nil {
err = er
return
}
winDivertCalcChecksums = proc
vers := map[string]struct{}{
"2.0": {},
"2.1": {},
"2.2": {},
}
ver, er := func() (ver string, err error) {
h, err := open("false", LayerNetwork, PriorityDefault, FlagDefault)
if err != nil {
return
}
defer func() {
err = h.Close()
}()
major, err := h.GetParam(VersionMajor)
if err != nil {
return
}
minor, err := h.GetParam(VersionMinor)
if err != nil {
return
}
ver = strings.Join([]string{strconv.Itoa(int(major)), strconv.Itoa(int(minor))}, ".")
return
}()
if er != nil {
err = er
return
}
if _, ok := vers[ver]; !ok {
err = fmt.Errorf("unsupported windivert version: %v", ver)
}
})
if err != nil {
return
}
return open(filter, layer, priority, flags)
}