mirror of
https://github.com/apple/container.git
synced 2026-07-14 13:37:02 +00:00
Kathryn Baldauf
8e9670c8f8
Co-authored-by: Aditya Ramani <a_ramani@apple.com> Co-authored-by: Agam Dua <agam_dua@apple.com> Co-authored-by: Danny Canter <danny_canter@apple.com> Co-authored-by: Dmitry Kovba <dkovba@apple.com> Co-authored-by: Eric Ernst <eric_ernst@apple.com> Co-authored-by: Evan Hazlett <ehazlett@apple.com> Co-authored-by: Gilbert Song <gilbertsong@apple.com> Co-authored-by: Hugh Bussell <hbussell@apple.com> Co-authored-by: John Logan <john_logan@apple.com> Co-authored-by: Kathryn Baldauf <k_baldauf@apple.com> Co-authored-by: Madhu Venugopal <mvenugopal@apple.com> Co-authored-by: Michael Crosby <michael_crosby@apple.com> Co-authored-by: Sidhartha Mani <sidhartha_mani@apple.com> Co-authored-by: Tanweer Noor <tnoor@apple.com> Co-authored-by: Ximena Perez Diaz <xperez528@gmail.com> Co-authored-by: Yibo Zhuang <yzhuang@apple.com>
85 lines
3.0 KiB
Swift
85 lines
3.0 KiB
Swift
//===----------------------------------------------------------------------===//
|
|
// Copyright © 2025 Apple Inc. and the container project authors. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// https://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
import Foundation
|
|
import NIOCore
|
|
import NIOPosix
|
|
|
|
extension DNSServer {
|
|
/// Handles the DNS request.
|
|
/// - Parameters:
|
|
/// - outbound: The NIOAsyncChannelOutboundWriter for which to respond.
|
|
/// - packet: The request packet.
|
|
func handle(
|
|
outbound: NIOAsyncChannelOutboundWriter<AddressedEnvelope<ByteBuffer>>,
|
|
packet: inout AddressedEnvelope<ByteBuffer>
|
|
) async throws {
|
|
let chunkSize = 512
|
|
var data = Data()
|
|
|
|
self.log?.debug("reading data")
|
|
while packet.data.readableBytes > 0 {
|
|
if let chunk = packet.data.readBytes(length: min(chunkSize, packet.data.readableBytes)) {
|
|
data.append(contentsOf: chunk)
|
|
}
|
|
}
|
|
|
|
self.log?.debug("deserializing message")
|
|
let query = try Message(deserialize: data)
|
|
self.log?.debug("processing query: \(query.questions)")
|
|
|
|
// always send response
|
|
let responseData: Data
|
|
do {
|
|
self.log?.debug("awaiting processing")
|
|
var response =
|
|
try await handler.answer(query: query)
|
|
?? Message(
|
|
id: query.id,
|
|
type: .response,
|
|
returnCode: .notImplemented,
|
|
questions: query.questions,
|
|
answers: []
|
|
)
|
|
|
|
// no responses
|
|
if response.answers.isEmpty {
|
|
response.returnCode = .nonExistentDomain
|
|
}
|
|
|
|
self.log?.debug("serializing response")
|
|
responseData = try response.serialize()
|
|
} catch {
|
|
self.log?.error("error processing message from \(query): \(error)")
|
|
let response = Message(
|
|
id: query.id,
|
|
type: .response,
|
|
returnCode: .notImplemented,
|
|
questions: query.questions,
|
|
answers: []
|
|
)
|
|
responseData = try response.serialize()
|
|
}
|
|
|
|
self.log?.debug("sending response for \(query.id)")
|
|
let rData = ByteBuffer(bytes: responseData)
|
|
try? await outbound.write(AddressedEnvelope(remoteAddress: packet.remoteAddress, data: rData))
|
|
|
|
self.log?.debug("processing done")
|
|
|
|
}
|
|
}
|