chunk uploads for large sast scan uploads (#64)

* chunk uploadds for large sast scan upload

* change limi to 50 mb for chunk upload

* bump version

Author: abronte

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "corgea"
-version = "1.7.1"
+version = "1.7.2"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

Cargo.toml

@@ -223,11 +223,43 @@ pub fn upload_scan(config: &Config, paths: Vec<String>, scanner: String, input:
     println!("Uploading the scan...");
 
     // main scan upload
-    debug(&format!("POST: {}", scan_upload_url));
-    let res = client.post(scan_upload_url)
-        .header(header::CONTENT_TYPE, "application/json")
-        .body(input.clone())
-        .send();
+    let input_bytes = input.as_bytes();
+    let input_size = input_bytes.len();
+    let max_upload_size = 50 * 1024 * 1024; // 50mb
+    let chunk_size = 1024 * 1024; // 1mb
+    let res = if input_size > max_upload_size {
+        let total_chunks = (input_size + chunk_size - 1) / chunk_size;
+        debug(&format!("Uploading scan in {} chunks", total_chunks));
+        let mut offset = 0usize;
+        let mut last_response = None;
+
+        for (index, chunk) in input_bytes.chunks(chunk_size).enumerate() {
+            debug(&format!("POST: {} (chunk {}/{})", scan_upload_url, index + 1, total_chunks));
+            let response = client.post(&scan_upload_url)
+                .header(header::CONTENT_TYPE, "application/json")
+                .header("Upload-Offset", offset.to_string())
+                .header("Upload-Length", input_size.to_string())
+                .body(chunk.to_vec())
+                .send();
+            let should_break = match &response {
+                Ok(res) => !res.status().is_success(),
+                Err(_) => true,
+            };
+            last_response = Some(response);
+            if should_break {
+                break;
+            }
+            offset += chunk.len();
+        }
+
+        last_response.expect("Failed to upload scan.")
+    } else {
+        debug(&format!("POST: {}", scan_upload_url));
+        client.post(&scan_upload_url)
+            .header(header::CONTENT_TYPE, "application/json")
+            .body(input.clone())
+            .send()
+    };
 
     let mut sast_scan_id: Option<String> = None;