Kitura/Kitura#33 Add marshaling of frames with more than 125 bytes of…

… payload
Kitura · Nov 3, 2016 · d7f52b2 · d7f52b2
1 parent 1789c2c
commit d7f52b2
Showing 1 changed file with 22 additions and 2 deletions.
diff --git a/Sources/KituraWebSocket/WSFrame.swift b/Sources/KituraWebSocket/WSFrame.swift
@@ -35,9 +35,29 @@ struct WSFrame {
             bytes[1] = UInt8(payloadLength)
             length += 1
         } else if payloadLength < Int(UInt16.max) {
-
+            bytes[1] = 126
+            let tempPayloadLengh = UInt16(payloadLength)
+            var payloadLengthUInt16: UInt16
+            #if os(Linux)
+                payloadLengthUInt16 = Glibc.htons(tempPayloadLengh)
+            #else
+                payloadLengthUInt16 = CFSwapInt16HostToBig(tempPayloadLengh)
+            #endif
+            let asBytes = UnsafeMutablePointer(&payloadLengthUInt16)
+            (UnsafeMutableRawPointer(mutating: bytes)+length+1).copyBytes(from: asBytes, count: 2)
+            length += 3
         } else {
-
+            bytes[1] = 127
+            let tempPayloadLengh = UInt32(payloadLength)
+            var payloadLengthUInt32: UInt32
+            #if os(Linux)
+                payloadLengthUInt32 = Glibc.htonl(tempPayloadLengh)
+            #else
+                payloadLengthUInt32 = CFSwapInt32HostToBig(tempPayloadLengh)
+            #endif
+            let asBytes = UnsafeMutablePointer(&payloadLengthUInt32)
+            (UnsafeMutableRawPointer(mutating: bytes)+length+5).copyBytes(from: asBytes, count: 4)
+            length += 9
         }
         buffer.append(bytes, length: length)
     }