optimize: reduce memory usage by reading kallsyms only in the trace c…

…ommand (#700)
daeuniverse · Nov 18, 2024 · 53b5fed · 53b5fed
1 parent d3ab0b2
commit 53b5fed
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/cmd/trace.go b/cmd/trace.go
@@ -31,6 +31,9 @@ func init() {
 	traceCmd := &cobra.Command{
 		Use:   "trace",
 		Short: "To trace traffic",
+		PreRun: func(cmd *cobra.Command, args []string) {
+			trace.ReadKallsyms()
+		},
 		Run: func(cmd *cobra.Command, args []string) {
 			internal.AutoSu()
 

diff --git a/trace/kallsyms.go b/trace/kallsyms.go
@@ -26,11 +26,7 @@ var kallsyms []Symbol
 var kallsymsByName map[string]Symbol = make(map[string]Symbol)
 var kallsymsByAddr map[uint64]Symbol = make(map[uint64]Symbol)
 
-func init() {
-	readKallsyms()
-}
-
-func readKallsyms() {
+func ReadKallsyms() {
 	file, err := os.Open("/proc/kallsyms")
 	if err != nil {
 		logrus.Fatalf("failed to open /proc/kallsyms: %v", err)