change BPF config detect logic.

Signed-off-by: CFC4N <cfc4n.cs@gmail.com>

Author: cfc4n

cli/cmd/root.go

@@ -6,6 +6,8 @@ package cmd
 
 import (
 	"ecapture/cli/cobrautl"
+	"ecapture/pkg/util/ebpf"
+	"log"
 	"os"
 
 	"github.com/spf13/cobra"
@@ -57,6 +59,15 @@ func Execute() {
 	rootCmd.Version = GitVersion
 	rootCmd.SetVersionTemplate(`{{with .Name}}{{printf "%s " .}}{{end}}{{printf "version:\t%s" .Version}}
 `)
+	enable, e := ebpf.IsEnableBPF()
+	if e != nil {
+		log.Fatalf("Kernel config read failed, error:%v", e)
+	}
+
+	if !enable {
+		log.Fatalf("Kernel not support, error:%v", e)
+	}
+
 	err := rootCmd.Execute()
 	if err != nil {
 		os.Exit(1)

main.go

@@ -27,15 +27,6 @@ func main() {
 		log.Fatalf("Linux Kernel version %v is not supported. Need > 4.18 .", kv)
 	}
 
-	enable, e := ebpf.IsEnableBPF()
-	if e != nil {
-		log.Fatalf("Kernel config read failed, error:%v", e)
-	}
-
-	if !enable {
-		log.Fatalf("Kernel not support, error:%v", e)
-	}
-
 	// changed by go build '-ldflags X'
 	if enableCORE == "true" {
 		// BTF支持情况检测
@@ -49,7 +40,5 @@ func main() {
 		}
 	}
 
-	// TODO check UPROBE
-
 	cli.Start()
 }

pkg/util/ebpf/bpf.go

@@ -114,6 +114,7 @@ func IsEnableBTF() (bool, error) {
 	return true, nil
 }
 
+// check BPF CONFIG
 func IsEnableBPF() (bool, error) {
 	var e error
 	var KernelConfig = make(map[string]string)
@@ -127,13 +128,13 @@ func IsEnableBPF() (bool, error) {
 		bc, found := KernelConfig[item]
 		if !found {
 			// 没有这个配置项
-			return false, fmt.Errorf("Notfound config item:%s", item)
+			return false, fmt.Errorf("Config not found,  item:%s.", item)
 		}
 
 		//如果有，在判断配置项的值
 		if bc != "y" {
 			// 没有开启
-			return false, fmt.Errorf("config item:%s not enable.", item)
+			return false, fmt.Errorf("Config disabled, item :%s.", item)
 		}
 	}