Amir Abbas
98 lines
2.4 KiB
Swift
98 lines
2.4 KiB
Swift
//
|
|
// SMB2DataTypes.swift
|
|
// ExtDownloader
|
|
//
|
|
// Created by Amir Abbas Mousavian on 4/30/95.
|
|
// Copyright © 1395 Mousavian. All rights reserved.
|
|
//
|
|
|
|
import Foundation
|
|
|
|
protocol SMBRequestBody {
|
|
func data() -> Data
|
|
}
|
|
|
|
extension SMBRequestBody {
|
|
func data() -> Data {
|
|
return Data(value: self)
|
|
}
|
|
}
|
|
|
|
protocol SMBResponseBody {
|
|
init? (data: Data)
|
|
}
|
|
|
|
extension SMBResponseBody {
|
|
init? (data: Data) {
|
|
if let v: Self = data.scanValue() {
|
|
self = v
|
|
} else {
|
|
return nil
|
|
}
|
|
}
|
|
}
|
|
|
|
typealias SMBRequest = (header: SMB2.Header, body: SMBRequestBody?)
|
|
typealias SMBResponse = (header: SMB2.Header, body: SMBResponseBody?)
|
|
|
|
protocol IOCtlRequestProtocol: SMBRequestBody {}
|
|
protocol IOCtlResponseProtocol: SMBResponseBody {}
|
|
|
|
|
|
struct SMBTime {
|
|
var time: Int64
|
|
|
|
init(time: Int64) {
|
|
self.time = time
|
|
}
|
|
|
|
init(unixTime: UInt) {
|
|
self.time = (Int64(unixTime) + 11644473600) * 10000000
|
|
}
|
|
|
|
init(timeIntervalSince1970: TimeInterval) {
|
|
self.time = Int64((timeIntervalSince1970 + 11644473600) * 10000000)
|
|
}
|
|
|
|
init(date: Date) {
|
|
self.init(timeIntervalSince1970: date.timeIntervalSince1970)
|
|
}
|
|
|
|
var unixTime: UInt {
|
|
return UInt(self.time / 10000000 - 11644473600)
|
|
}
|
|
|
|
var date: Date {
|
|
return Date(timeIntervalSince1970: Double(self.time) / 10000000 - 11644473600)
|
|
}
|
|
}
|
|
|
|
extension Data {
|
|
init<T>(value: T) {
|
|
var value = value
|
|
self = Data(buffer: UnsafeBufferPointer(start: &value, count: 1))
|
|
}
|
|
|
|
func scanValue<T>() -> T? {
|
|
guard MemoryLayout<T>.size <= self.count else { return nil }
|
|
return self.withUnsafeBytes { $0.pointee }
|
|
}
|
|
|
|
func scanValue<T>(start: Int) -> T? {
|
|
let length = MemoryLayout<T>.size
|
|
guard self.count >= start + length else { return nil }
|
|
return self.subdata(in: start..<start+length).withUnsafeBytes { $0.pointee }
|
|
}
|
|
|
|
func scanString(start: Int = 0, length: Int, encoding: String.Encoding) -> String? {
|
|
guard self.count >= start + length else { return nil }
|
|
return String(data: self.subdata(in: start..<start+length), encoding: encoding)
|
|
}
|
|
|
|
static func mapMemory<T, U>(from: T) -> U? {
|
|
guard MemoryLayout<T>.size >= MemoryLayout<U>.size else { return nil }
|
|
let data = Data(value: from)
|
|
return data.scanValue()
|
|
}
|
|
}
|