Gemini/GeminiProtocol/GeminiProtocol.swift

114 lines
4.3 KiB
Swift
Raw Normal View History

2020-07-13 03:09:37 +00:00
//
// GeminiProtocol.swift
// Gemini
//
// Created by Shadowfacts on 7/12/20.
//
import Network
class GeminiProtocol: NWProtocolFramerImplementation {
static let definition = NWProtocolFramer.Definition(implementation: GeminiProtocol.self)
static let label = "Gemini"
2020-07-23 13:18:59 +00:00
private var tempStatusCode: GeminiResponseHeader.StatusCode?
private var tempMeta: String?
2020-07-13 03:09:37 +00:00
required init(framer: NWProtocolFramer.Instance) {
}
func start(framer: NWProtocolFramer.Instance) -> NWProtocolFramer.StartResult {
return .ready
}
func wakeup(framer: NWProtocolFramer.Instance) {
}
func stop(framer: NWProtocolFramer.Instance) -> Bool {
return true
}
func cleanup(framer: NWProtocolFramer.Instance) {
}
func handleInput(framer: NWProtocolFramer.Instance) -> Int {
2020-07-23 13:18:59 +00:00
if tempStatusCode == nil {
_ = framer.parseInput(minimumIncompleteLength: 3, maximumLength: 3) { (buffer, isComplete) -> Int in
2020-07-13 03:09:37 +00:00
guard let buffer = buffer,
buffer.count == 3 else { return 0 }
2020-07-23 13:18:59 +00:00
self.tempStatusCode = GeminiResponseHeader.StatusCode(buffer)
2020-07-13 03:09:37 +00:00
return 3
}
2020-07-23 13:18:59 +00:00
}
guard let statusCode = tempStatusCode else {
return 3
}
var attemptedMetaLength: Int?
if tempMeta == nil {
2020-07-13 03:09:37 +00:00
// Minimum length is 2 bytes, spec does not say meta string is required
2020-07-23 13:18:59 +00:00
_ = framer.parseInput(minimumIncompleteLength: 2, maximumLength: 1024 + 2) { (buffer, isComplete) -> Int in
2020-07-13 03:09:37 +00:00
guard let buffer = buffer,
buffer.count >= 2 else { return 0 }
2020-07-23 13:18:59 +00:00
attemptedMetaLength = buffer.count
2020-07-13 03:09:37 +00:00
let lastPossibleCRIndex = buffer.index(before: buffer.index(before: buffer.endIndex))
var index = buffer.startIndex
var found = false
while index <= lastPossibleCRIndex {
// <CR><LF>
if buffer[index] == 13 && buffer[buffer.index(after: index)] == 10 {
found = true
break
}
index = buffer.index(after: index)
}
2020-07-23 13:18:59 +00:00
if !found {
if buffer.count < 1026 {
return 0
} else {
fatalError("Didn't find <CR><LF> in buffer. Meta string was longer than 1024 bytes")
}
}
2020-07-13 03:09:37 +00:00
2020-07-23 13:18:59 +00:00
self.tempMeta = String(bytes: buffer[..<index], encoding: .utf8)
return buffer.startIndex.distance(to: index) + 2
2020-07-13 03:09:37 +00:00
}
2020-07-23 13:18:59 +00:00
}
guard let meta = tempMeta else {
if let attempted = attemptedMetaLength {
return attempted + 1
} else {
2020-07-13 03:09:37 +00:00
return 2
}
2020-07-23 13:18:59 +00:00
}
let header = GeminiResponseHeader(status: statusCode, meta: meta)
let message = NWProtocolFramer.Message(geminiResponseHeader: header)
2021-06-16 02:24:16 +00:00
// Deliver all the input (the response body) to the client without copying.
_ = framer.deliverInputNoCopy(length: statusCode.isSuccess ? .max : 0, message: message, isComplete: true)
2021-06-16 02:24:16 +00:00
// Just in case, set the framer to pass-through input so it never invokes this method again.
// todo: this should work according to an apple engineer, but the request seems to hang forever on a real device
// sometimes works fine when stepping through w/ debugger => race condition?
// framer.passThroughInput()
return 0
2020-07-13 03:09:37 +00:00
}
func handleOutput(framer: NWProtocolFramer.Instance, message: NWProtocolFramer.Message, messageLength: Int, isComplete: Bool) {
guard let request = message.geminiRequest else { fatalError("GeminiProtocol can't send message that doesn't have an associated GeminiRequest") }
framer.writeOutput(data: request.data)
}
}
2020-07-15 02:13:09 +00:00
fileprivate extension GeminiResponseHeader.StatusCode {
2020-07-13 03:09:37 +00:00
init?(_ buffer: UnsafeMutableRawBufferPointer) {
guard let str = String(bytes: buffer[...buffer.index(after: buffer.startIndex)], encoding: .utf8),
let value = Int(str, radix: 10) else { return nil }
self.init(rawValue: value)
}
}