Function runFunc

cli/cmd/root.go:369–449

(command *cobra.Command, args []string)

}

func runFunc(command *cobra.Command, args []string) {
    stopper := make(chan os.Signal, 1)
    signal.Notify(stopper, os.Interrupt, syscall.SIGTERM)
    ctx, cancelFun := context.WithCancel(context.TODO())
    if gconfig.Rpc {
        rpc.SetupRpc(ctx, Logger, gconfig)
        rpc.StartRpcServer(stopper, gconfig.RpcPath)
        os.Exit(0)
    }
    var runMods uint8
    var runModules = make(map[string]module.IModule)
    var wg sync.WaitGroup

    var modNames []string
    if mconfig.BrkAddr != 0 {
        modNames = append(modNames, module.MODULE_NAME_BRK)
    } else if mconfig.SysCallConf.Enable {
        modNames = append(modNames, module.MODULE_NAME_PERF)
        modNames = append(modNames, module.MODULE_NAME_SYSCALL)
    } else if len(mconfig.StackUprobeConf.Points) > 0 {
        modNames = append(modNames, module.MODULE_NAME_PERF)
        modNames = append(modNames, module.MODULE_NAME_STACK)
    } else {
        Logger.Fatal("hook nothing, plz set -w/--point or -s/--syscall or --brk")
    }
    for _, modName := range modNames {
        // 现在合并成只有一个模块了 所以直接通过名字获取
        mod := module.GetModuleByName(modName)

        mod.Init(ctx, Logger, mconfig)
        err := mod.Run()
        if err != nil {
            Logger.Printf("%s\tmodule Run failed, [skip it]. error:%+v", mod.Name(), err)
            os.Exit(1)
        }
        runModules[mod.Name()] = mod
        if gconfig.Debug {
            Logger.Printf("%s\tmodule started successfully", mod.Name())
        }
        wg.Add(1)
        runMods++

    }
    if runMods > 0 {
        Logger.Printf("start %d modules", runMods)
        go func() {
            scanner := bufio.NewScanner(os.Stdin)
            for {
                scanner.Scan()
                err := scanner.Err()
                if err != nil {
                    Logger.Printf("get input from console failed, err:%v", err)
                    syscall.Kill(syscall.Getpid(), syscall.SIGTERM)
                }
                input_text := scanner.Text()
                if input_text == "c" {
                    event.LetItRun()
                }