Function Aplugin

cli/ctl/agent_check.go:647–699

(cmd *cobra.Command)

}

func Aplugin(cmd *cobra.Command) {
	conn := agentGetConn(cmd)
	if conn == nil {
		return
	}
	defer conn.Close()
	fmt.Printf("request trisolaris(%s), params(%+v)\n", conn.Target(), agentParamData)
	var pluginType agent.PluginType
	switch agentParamData.PluginType {
	case "wasm":
		pluginType = agent.PluginType_WASM
	default:
		fmt.Printf("request pluginType(%s) not supported, pluginType must be in %s\n",
			agentParamData.PluginType, []string{"wasm"})
		return
	}
	c := agent.NewSynchronizerClient(conn)
	reqData := &agent.PluginRequest{
		CtrlIp:     &agentParamData.CtrlIP,
		CtrlMac:    &agentParamData.CtrlMac,
		TeamId:     &agentParamData.TeamID,
		PluginType: &pluginType,
		PluginName: &agentParamData.PluginName,
	}
	stream, err := c.Plugin(context.Background(), reqData)
	if err != nil {
		fmt.Println(err)
		return
	}
	var (
		data []byte
		md5  string
	)
	for {
		if res, err := stream.Recv(); err == nil {
			data = append(data, res.GetContent()...)
			md5 = res.GetMd5()
		} else {
			if errors.Is(err, io.EOF) {
				break
			}
			fmt.Println(res, err)
			return
		}
	}
	fileName := agentParamData.PluginType + "-" + agentParamData.PluginName
	err = os.WriteFile(fileName, data, 0666)
	if err != nil {
		fmt.Printf("save plugin(%s) fail %s\n", fileName, err)
		return
	}
	fmt.Printf("save plugin(%s) success, md5=%s\n", fileName, md5)
}