mirror of
https://github.com/permissionlesstech/bitchat.git
synced 2026-07-25 14:05:18 +00:00
Fix compressed BLE file transfers
This commit is contained in:
@@ -108,4 +108,67 @@ final class NotificationStreamAssemblerTests: XCTestCase {
|
||||
}
|
||||
XCTAssertEqual(decoded.timestamp, packet.timestamp)
|
||||
}
|
||||
|
||||
func testAssemblesCompressedLargeFrame() throws {
|
||||
var assembler = NotificationStreamAssembler()
|
||||
|
||||
let largeContent = Data(repeating: 0x41, count: 2_500_000)
|
||||
let filePacket = BitchatFilePacket(
|
||||
fileName: "large.bin",
|
||||
fileSize: UInt64(largeContent.count),
|
||||
mimeType: "application/octet-stream",
|
||||
content: largeContent
|
||||
)
|
||||
guard let tlvPayload = filePacket.encode() else {
|
||||
return XCTFail("Failed to encode file packet")
|
||||
}
|
||||
|
||||
let senderID = Data(repeating: 0xAA, count: BinaryProtocol.senderIDSize)
|
||||
let packet = BitchatPacket(
|
||||
type: MessageType.fileTransfer.rawValue,
|
||||
senderID: senderID,
|
||||
recipientID: nil,
|
||||
timestamp: 0x010203040506,
|
||||
payload: tlvPayload,
|
||||
signature: nil,
|
||||
ttl: 3,
|
||||
version: 2
|
||||
)
|
||||
|
||||
guard let frame = packet.toBinaryData(padding: false) else {
|
||||
return XCTFail("Failed to encode packet frame")
|
||||
}
|
||||
|
||||
let flagsOffset = 1 + 1 + 1 + 8
|
||||
XCTAssertLessThan(flagsOffset, frame.count)
|
||||
let flags = frame[frame.startIndex + flagsOffset]
|
||||
XCTAssertNotEqual(flags & BinaryProtocol.Flags.isCompressed, 0, "Frame should be compressed for large payloads")
|
||||
|
||||
let splitIndex = min(4096, frame.count / 2)
|
||||
var result = assembler.append(frame.prefix(splitIndex))
|
||||
XCTAssertTrue(result.frames.isEmpty)
|
||||
|
||||
result = assembler.append(frame.suffix(from: splitIndex))
|
||||
XCTAssertEqual(result.frames.count, 1)
|
||||
XCTAssertTrue(result.droppedPrefixes.isEmpty)
|
||||
XCTAssertFalse(result.reset)
|
||||
|
||||
guard let assembled = result.frames.first else {
|
||||
return XCTFail("Missing assembled frame")
|
||||
}
|
||||
XCTAssertEqual(assembled.count, frame.count)
|
||||
|
||||
guard let decodedPacket = BinaryProtocol.decode(assembled) else {
|
||||
return XCTFail("Failed to decode compressed frame")
|
||||
}
|
||||
XCTAssertEqual(decodedPacket.payload.count, tlvPayload.count)
|
||||
|
||||
guard let decodedFile = BitchatFilePacket.decode(decodedPacket.payload) else {
|
||||
return XCTFail("Failed to decode TLV payload")
|
||||
}
|
||||
XCTAssertEqual(decodedFile.fileName, filePacket.fileName)
|
||||
XCTAssertEqual(decodedFile.mimeType, filePacket.mimeType)
|
||||
XCTAssertEqual(decodedFile.content.count, largeContent.count)
|
||||
XCTAssertEqual(decodedFile.content.prefix(32), largeContent.prefix(32))
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,47 @@
|
||||
import XCTest
|
||||
@testable import bitchat
|
||||
|
||||
final class BitchatFilePacketTests: XCTestCase {
|
||||
|
||||
func testRoundTripPreservesFields() throws {
|
||||
let content = Data((0..<4096).map { UInt8($0 % 251) })
|
||||
let packet = BitchatFilePacket(
|
||||
fileName: "sample.jpg",
|
||||
fileSize: UInt64(content.count),
|
||||
mimeType: "image/jpeg",
|
||||
content: content
|
||||
)
|
||||
|
||||
guard let encoded = packet.encode() else {
|
||||
return XCTFail("Failed to encode file packet")
|
||||
}
|
||||
guard let decoded = BitchatFilePacket.decode(encoded) else {
|
||||
return XCTFail("Failed to decode file packet")
|
||||
}
|
||||
|
||||
XCTAssertEqual(decoded.fileName, packet.fileName)
|
||||
XCTAssertEqual(decoded.fileSize, packet.fileSize)
|
||||
XCTAssertEqual(decoded.mimeType, packet.mimeType)
|
||||
XCTAssertEqual(decoded.content, packet.content)
|
||||
}
|
||||
|
||||
func testDecodeFallsBackToContentSizeWhenFileSizeMissing() throws {
|
||||
let content = Data(repeating: 0x7F, count: 1024)
|
||||
let packet = BitchatFilePacket(
|
||||
fileName: nil,
|
||||
fileSize: nil,
|
||||
mimeType: nil,
|
||||
content: content
|
||||
)
|
||||
|
||||
guard let encoded = packet.encode() else {
|
||||
return XCTFail("Failed to encode file packet")
|
||||
}
|
||||
guard let decoded = BitchatFilePacket.decode(encoded) else {
|
||||
return XCTFail("Failed to decode file packet")
|
||||
}
|
||||
|
||||
XCTAssertEqual(decoded.fileSize, UInt64(content.count))
|
||||
XCTAssertEqual(decoded.content, content)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user