Small tweaks

beardedspice · Mar 30, 2016 · 8887cd7 · 8887cd7
1 parent f6f6b34
commit 8887cd7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/PocketSocket/PSWebSocketDriver.m b/PocketSocket/PSWebSocketDriver.m
@@ -307,8 +307,8 @@ - (void)writeMessageWithOpCode:(PSWebSocketOpCode)opcode data:(NSData *)data {
     uint8_t *headerBytes = header.mutableBytes;
 
     headerBytes[0] |= PSWebSocketFinMask;
-    //  headerBytes[0] |= (ZWPWebSocketRsv2Mask);
-    //  headerBytes[0] |= (ZWPWebSocketRsv3Mask);
+    //  headerBytes[0] |= (PSWebSocketRsv2Mask);
+    //  headerBytes[0] |= (PSWebSocketRsv3Mask);
     headerBytes[0] |= (PSWebSocketOpCodeMask & opcode);
 
     // determine payload payload
@@ -604,6 +604,7 @@ - (NSInteger)readBytes:(void *)bytes maxLength:(NSUInteger)maxLength error:(NSEr
 
             if(!frame->control && _frames.count > 0) {
                 _initialFrame = [_frames lastObject];
+                frame->rsv1 = _initialFrame->rsv1;
                 frame->opcode = _initialFrame->opcode;
                 frame->buffer = _initialFrame->buffer;
             } else {