Merge pull request #36 from network-quality/Enforce-Encoding

Force No Compression

1 files changed, 8 insertions, 0 deletions

diff --git a/rpm/rpm.go b/rpm/rpm.go
index 607cffa..3b32273 100644
--- a/rpm/rpm.go
+++ b/rpm/rpm.go
@@ -127,11 +127,19 @@ func Probe(
 		return err
 	}
 
+	// Used to disable compression
+	probe_req.Header.Set("Accept-Encoding", "identity")
+
 	probe_resp, err := client.Do(probe_req)
 	if err != nil {
 		return err
 	}
 
+	// Header.Get returns "" when not set
+	if probe_resp.Header.Get("Content-Encoding") != "" {
+		return fmt.Errorf("Content-Encoding header was set (compression not allowed)")
+	}
+
 	// TODO: Make this interruptable somehow by using _ctx_.
 	_, err = io.ReadAll(probe_resp.Body)
 	if err != nil {