Fix(cf curl): Prevent errors when handling binary API responses

Author: prkalle

command/v7/curl_command.go

@@ -1,6 +1,7 @@
 package v7
 
 import (
+	"bytes"
 	"net/http/httputil"
 	"os"
 
@@ -39,9 +40,9 @@ func (cmd CurlCommand) Execute(args []string) error {
 	}
 
 	var bytesToWrite []byte
-
-	if cmd.IncludeResponseHeaders {
-		headerBytes, _ := httputil.DumpResponse(httpResponse, false)
+	var headerBytes []byte
+	if cmd.IncludeResponseHeaders && httpResponse != nil {
+		headerBytes, _ = httputil.DumpResponse(httpResponse, false)
 		bytesToWrite = append(bytesToWrite, headerBytes...)
 	}
 
@@ -54,9 +55,24 @@ func (cmd CurlCommand) Execute(args []string) error {
 		}
 
 		cmd.UI.DisplayOK()
+		return nil
+	}
+
+	// Check if the response contains binary data
+	if IsBinary(responseBodyBytes) {
+		// For binary data, write response headers with string conversion
+		// and the response body without string conversion
+		if cmd.IncludeResponseHeaders {
+			cmd.UI.DisplayTextLiteral(string(headerBytes))
+		}
+		cmd.UI.GetOut().Write(responseBodyBytes)
 	} else {
 		cmd.UI.DisplayText(string(bytesToWrite))
 	}
 
 	return nil
 }
+
+func IsBinary(data []byte) bool {
+	return bytes.ContainsRune(data, 0x00) // Check for null byte
+}

command/v7/curl_command_test.go

@@ -149,4 +149,95 @@ var _ = Describe("curl Command", func() {
 		})
 	})
 
+	When("the response contains binary data", func() {
+		var binaryData []byte
+
+		BeforeEach(func() {
+			// Create binary data with null bytes (like a droplet file)
+			binaryData = []byte{0x50, 0x4B, 0x03, 0x04, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00}
+			fakeActor.MakeCurlRequestReturns(binaryData, &http.Response{
+				Header: http.Header{
+					"Content-Type": []string{"application/octet-stream"},
+				},
+			}, nil)
+		})
+
+		It("writes binary data directly to stdout without string conversion", func() {
+			Expect(executeErr).NotTo(HaveOccurred())
+			Expect(testUI.Out).To(Say(string(binaryData)))
+		})
+
+		When("include-response-headers flag is set", func() {
+			BeforeEach(func() {
+				cmd.IncludeResponseHeaders = true
+			})
+
+			It("writes headers as text and binary data separately", func() {
+				Expect(executeErr).NotTo(HaveOccurred())
+
+				// Check that headers are written as text (using DisplayTextLiteral)
+				Expect(testUI.Out).To(Say("Content-Type: application/octet-stream"))
+
+				// Check that binary data is preserved
+				Expect(testUI.Out).To(Say(string(binaryData)))
+			})
+		})
+
+		When("output file is specified", func() {
+			BeforeEach(func() {
+				outputFile, err := os.CreateTemp("", "binary-output")
+				Expect(err).NotTo(HaveOccurred())
+				cmd.OutputFile = flag.Path(outputFile.Name())
+			})
+
+			AfterEach(func() {
+				os.RemoveAll(string(cmd.OutputFile))
+			})
+
+			It("writes binary data to the file correctly", func() {
+				Expect(executeErr).NotTo(HaveOccurred())
+
+				fileContents, err := os.ReadFile(string(cmd.OutputFile))
+				Expect(err).ToNot(HaveOccurred())
+				Expect(fileContents).To(Equal(binaryData))
+			})
+		})
+	})
+
+	When("the response is empty", func() {
+		BeforeEach(func() {
+			fakeActor.MakeCurlRequestReturns([]byte{}, &http.Response{
+				Header: http.Header{},
+			}, nil)
+		})
+
+		It("handles empty response correctly", func() {
+			Expect(executeErr).NotTo(HaveOccurred())
+			Expect(testUI.Out).To(Say(""))
+		})
+	})
+
+	// Test the IsBinary function directly
+	Describe("IsBinary function", func() {
+		It("returns true for data containing null bytes", func() {
+			data := []byte{0x48, 0x65, 0x6C, 0x6C, 0x6F, 0x00, 0x57, 0x6F, 0x72, 0x6C, 0x64}
+			Expect(v7.IsBinary(data)).To(BeTrue())
+		})
+
+		It("returns false for data without null bytes", func() {
+			data := []byte("Hello World")
+			Expect(v7.IsBinary(data)).To(BeFalse())
+		})
+
+		It("returns true for data that is all null bytes", func() {
+			data := []byte{0x00, 0x00, 0x00, 0x00}
+			Expect(v7.IsBinary(data)).To(BeTrue())
+		})
+
+		It("returns false for empty data", func() {
+			data := []byte{}
+			Expect(v7.IsBinary(data)).To(BeFalse())
+		})
+	})
+
 })