Fix fuzzing issues regarding masking

uNetworking · Jan 3, 2022 · 5df695e · 5df695e
1 parent 0c8be4d
commit 5df695e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/WebSocketProtocol.h b/src/WebSocketProtocol.h
@@ -402,7 +402,12 @@ struct WIN32_EXPORT WebSocketProtocol {
                 /* No need to unmask if mask is 0 */
                 uint32_t nullmask = 0;
                 if (!memcmp(wState->mask, &nullmask, sizeof(uint32_t))) {
-                    unmaskAll(src, wState->mask);
+                    if /*constexpr*/ (LIBUS_RECV_BUFFER_LENGTH == length) {
+                        unmaskAll(src, wState->mask);
+                    } else {
+                        // Slow path
+                        unmaskInplace(src, src + ((length >> 2) + 1) * 4, wState->mask);
+                    }
                 }
             }