Method Handle

internal/bridge/module_shell.go:44–98

Handle processes a shell module command.

(ctx ModuleContext, sessionID string, taskID uint32, spite *implantpb.Spite)

// Handle processes a shell module command.
func (m *ShellModule) Handle(ctx ModuleContext, sessionID string, taskID uint32, spite *implantpb.Spite) {
	sess, toolName := acquireShellSession(ctx, sessionID, taskID, m.name)
	if sess == nil {
		return
	}

	// Determine OS from User-Agent.
	info := parseUserAgentFull(sess.UserAgent)
	osName := info.osName

	// Extract request args.
	var req *implantpb.Request
	if r := spite.GetRequest(); r != nil {
		req = r
	} else {
		req = &implantpb.Request{}
	}

	// Build the OS-specific command.
	command := m.buildCommand(osName, req)
	if command == "" {
		ctx.SendSpite(sessionID, taskID, execSpite(fmt.Sprintf("cannot build command for module %s on %s", m.name, osName)))
		ctx.Tasks.Fail(sessionID, taskID, "empty command")
		return
	}

	result := enqueueAndAwait(ctx, sessionID, taskID, sess, toolName, command)
	if result == nil {
		return
	}

	output := extractPlainOutput(result.Output)

	var resultSpite *implantpb.Spite
	if m.buildResponse != nil {
		resultSpite = m.buildResponse(output, osName, req)
	}
	if resultSpite == nil {
		// Fallback to ExecResponse.
		resp := parseToolOutput(result.Output)
		resp.End = true
		resultSpite = &implantpb.Spite{
			Name: consts.ModuleExecute,
			Body: &implantpb.Spite_ExecResponse{ExecResponse: resp},
		}
	}

	if err := ctx.SendSpite(sessionID, taskID, resultSpite); err != nil {
		log.Errorf("[bridge] failed to forward module %s task %d: %v", m.name, taskID, err)
		ctx.Tasks.Fail(sessionID, taskID, err.Error())
	} else {
		log.Infof("[bridge] forwarded module %s task %d result for session %s", m.name, taskID, sessionID)
		ctx.Tasks.Complete(sessionID, taskID)
	}
}