fix: Gracefully Handle Invalid Message Data

mParticle · Aug 30, 2021 · b618e5b · b618e5b
1 parent 953f063
commit b618e5b
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/mParticle-Apple-SDK/Utils/MPMessageBuilder.mm b/mParticle-Apple-SDK/Utils/MPMessageBuilder.mm
@@ -382,6 +382,16 @@ - (MPMessage *)build {
     messageDictionary[kMPMessageIdKey] = uuid ? uuid : [[NSUUID UUID] UUIDString];
 
     NSNumber *userId = _session.userId.integerValue ? _session.userId : [MPPersistenceController mpId];
+    for (NSString *key in messageDictionary) {
+        if ([messageDictionary[key] isKindOfClass:[NSNumber class]]) {
+            NSNumber *value = (NSNumber *)messageDictionary[key];
+            if(value.doubleValue == INFINITY) {
+                MPILogVerbose(@"Invalid Message Data for key: %@", key);
+                MPILogVerbose(@"Value should not be infinite. Removing value from message data");
+                messageDictionary[key] = nil;
+            }
+        }
+    }
 
     message = [[MPMessage alloc] initWithSession:_session
                                      messageType:_messageType