From 86cca041603c69a46c503c9707cc55ed8f9bf325 Mon Sep 17 00:00:00 2001
From: Chris Dillard <cdillard@newrelic.com>
Date: Thu, 16 Jan 2025 16:32:41 -0700
Subject: [PATCH] NR-359272: add one safety to path that recordTask takes
 through noticeNetworkRequest

---
 Agent/Network/NRMANetworkFacade.mm | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/Agent/Network/NRMANetworkFacade.mm b/Agent/Network/NRMANetworkFacade.mm
index 5cd1b5e0..f406f3fc 100644
--- a/Agent/Network/NRMANetworkFacade.mm
+++ b/Agent/Network/NRMANetworkFacade.mm
@@ -122,10 +122,12 @@ + (void) noticeNetworkRequest:(NSURLRequest*)request
                                                                                             contentType:[NRMANetworkFacade contentType:response]
                                                                                               bytesSent:bytesSent];
         NSUInteger modifiedBytesReceived = bytesReceived;
-        NSHTTPURLResponse *httpResponse = (NSHTTPURLResponse*) response;
-        NSString* header = httpResponse.allHeaderFields[@"Content-Encoding"];
-        if ([header isEqualToString:@"gzip"]) {
-            modifiedBytesReceived = [[NRMAHarvesterConnection gzipData:responseData] length];
+        if (([response isKindOfClass:[NSHTTPURLResponse class]])) {
+            NSHTTPURLResponse *httpResponse = (NSHTTPURLResponse*) response;
+            NSString* header = httpResponse.allHeaderFields[@"Content-Encoding"];
+            if ([header isEqualToString:@"gzip"]) {
+                modifiedBytesReceived = [[NRMAHarvesterConnection gzipData:responseData] length];
+            }
         }