Function PlanUpload

internal/sessions/transfer.go:116–143

--------------------------------------------------------------------------- Transfer planning --------------------------------------------------------------------------- PlanUpload decides the transfer strategy and chunk count for an upload.

(data []byte, userAgent string)

// PlanUpload decides the transfer strategy and chunk count for an upload.
func PlanUpload(data []byte, userAgent string) TransferPlan {
	binary := IsBinary(data)
	size := len(data)
	profile := MatchAgentProfile(userAgent)

	plan := TransferPlan{
		TotalSize: size,
		IsBinary:  binary,
		AgentName: profile.Name,
	}

	if !binary && size <= smallFileThreshold {
		plan.Strategy = StrategyDirectTool
		plan.NumChunks = 1
		return plan
	}

	// Shell + base64 path.
	plan.Strategy = StrategyShellBase64
	plan.ChunkSize = profile.ChunkSizeBytes

	if size <= profile.ChunkSizeBytes {
		plan.NumChunks = 1
	} else {
		plan.NumChunks = int(math.Ceil(float64(size) / float64(profile.ChunkSizeBytes)))
	}
	return plan
}

// PlanDownload decides the transfer strategy and chunk count for a download.
// fileSize should be obtained from a probe command; if 0 or negative the plan