kyanos/agent/common/options.go at 0448cea9ce6121979d0ebe37e69fad2125c0ae04 · hengyoush/kyanos · 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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
package common

import (
	"container/list"
	"context"
	"fmt"
	anc "kyanos/agent/analysis/common"
	"kyanos/agent/compatible"
	"kyanos/agent/metadata"
	"kyanos/agent/protocol"
	"kyanos/agent/render/watch"
	"kyanos/bpf"
	"kyanos/common"
	"os"
	"runtime"
	"strings"
)

type LoadBpfProgramFunction func() *list.List
type InitCompletedHook func()
type ConnManagerInitHook func(any)

var Options *AgentOptions

const perfEventDataBufferSize = 30 * 1024 * 1024
const perfEventControlBufferSize = 1 * 1024 * 1024

type AgentOptions struct {
	Stopper                chan os.Signal
	CustomSyscallEventHook bpf.SyscallEventHook
	CustomConnEventHook    bpf.ConnEventHook
	CustomKernEventHook    bpf.KernEventHook
	CustomSslEventHook     bpf.SslEventHook
	InitCompletedHook      InitCompletedHook
	ConnManagerInitHook    ConnManagerInitHook
	LoadBpfProgramFunction LoadBpfProgramFunction
	ProcessorsNum          int
	MessageFilter          protocol.ProtocolFilter
	LatencyFilter          protocol.LatencyFilter
	TraceSide              common.SideEnum
	IfName                 string
	BTFFilePath            string
	protocol.SizeFilter
	AnalysisEnable bool
	anc.AnalysisOptions
	PerfEventBufferSizeForData  int
	PerfEventBufferSizeForEvent int
	WatchOptions                watch.WatchOptions
	PerformanceMode             bool
	ConntrackCloseWaitTimeMills int
	MaxAllowStuckTimeMills      int
	StartGopsServer             bool
	EnablePprof                 bool
	PprofAddr                   string

	FilterComm              string
	ProcessExecEventChannel chan *bpf.AgentProcessExecEvent
	DockerEndpoint          string
	ContainerdEndpoint      string
	CriRuntimeEndpoint      string
	ContainerId             string
	ContainerName           string
	PodName                 string
	PodNameSpace            string

	Cc                  *metadata.ContainerCache
	Objs                any
	Ctx                 context.Context
	Kv                  *compatible.KernelVersion
	LoadPorgressChannel chan string

	SyscallPerfEventMapPageNum int
	SslPerfEventMapPageNum     int
	ConnPerfEventMapPageNum    int
	KernPerfEventMapPageNum    int
	FirstPacketEventMapPageNum int
}

func (o AgentOptions) GetPprofAddr() string {
	if o.PprofAddr == "" {
		return "localhost:6060"
	}
	return o.PprofAddr
}

func (o AgentOptions) FilterByContainer() bool {
	return o.ContainerId != "" || o.ContainerName != "" || o.PodName != ""
}

func (o AgentOptions) FilterByK8s() bool {
	return o.PodName != ""
}

func getPodNameFilter(raw string) (name, ns string) {
	if !strings.Contains(raw, ".") {
		return raw, "default"
	}
	index := strings.LastIndex(raw, ".")
	return raw[:index], raw[index+1:]
}

func getEndpoint(raw string) string {
	if strings.HasPrefix(raw, "http") {
		return raw
	}
	if strings.HasPrefix(raw, "unix://") {
		return raw
	}
	return fmt.Sprintf("unix://%s", raw)
}

func ValidateAndRepairOptions(options AgentOptions) AgentOptions {
	var newOptions = options
	if newOptions.Stopper == nil {
		newOptions.Stopper = make(chan os.Signal)
	}
	if newOptions.ProcessorsNum == 0 {
		newOptions.ProcessorsNum = runtime.NumCPU()
	}
	if newOptions.MessageFilter == nil {
		newOptions.MessageFilter = protocol.BaseFilter{}
	}
	if newOptions.PerfEventBufferSizeForData <= 0 {
		newOptions.PerfEventBufferSizeForData = perfEventDataBufferSize
	}
	if newOptions.PerfEventBufferSizeForEvent <= 0 {
		newOptions.PerfEventBufferSizeForEvent = perfEventControlBufferSize
	}
	if newOptions.PodName != "" {
		newOptions.PodName, newOptions.PodNameSpace = getPodNameFilter(newOptions.PodName)
	}
	if newOptions.DockerEndpoint != "" {
		newOptions.DockerEndpoint = getEndpoint(newOptions.DockerEndpoint)
	}
	if newOptions.CriRuntimeEndpoint != "" {
		newOptions.CriRuntimeEndpoint = getEndpoint(newOptions.CriRuntimeEndpoint)
	}
	newOptions.WatchOptions.Init()
	newOptions.LoadPorgressChannel = make(chan string, 10)
	return newOptions
}