Amir Abbas
137 lines
4.7 KiB
Swift
137 lines
4.7 KiB
Swift
//
|
|
// SMB2Tree.swift
|
|
// ExtDownloader
|
|
//
|
|
// Created by Amir Abbas Mousavian on 4/30/95.
|
|
// Copyright © 1395 Mousavian. All rights reserved.
|
|
//
|
|
|
|
import Foundation
|
|
|
|
extension SMB2 {
|
|
// MARK: SMB2 Tree Connect
|
|
|
|
struct TreeConnectRequest: SMBRequestBody {
|
|
let header: TreeConnectRequest.Header
|
|
let buffer: Data?
|
|
var path: String {
|
|
return ""
|
|
}
|
|
var share: String {
|
|
return ""
|
|
}
|
|
|
|
init? (header: TreeConnectRequest.Header, host: String, share: String) {
|
|
guard !host.contains("/") && !share.contains("/") else {
|
|
return nil
|
|
}
|
|
self.header = header
|
|
let path = "\\\\" + host + "\\" + share
|
|
self.buffer = path.data(using: .utf16)
|
|
}
|
|
|
|
func data() -> Data {
|
|
var header = self.header
|
|
header.pathOffset = UInt16(MemoryLayout<SMB2.Header>.size + MemoryLayout<TreeConnectRequest.Header>.size)
|
|
header.pathLength = UInt16(buffer?.count ?? 0)
|
|
var result = Data(value: header)
|
|
if let buffer = self.buffer {
|
|
result.append(buffer)
|
|
}
|
|
return result
|
|
}
|
|
|
|
struct Header {
|
|
let size: UInt16
|
|
let flags: TreeConnectRequest.Flags
|
|
var pathOffset: UInt16
|
|
var pathLength: UInt16
|
|
|
|
init(flags: TreeConnectRequest.Flags) {
|
|
self.size = 9
|
|
self.flags = flags
|
|
self.pathOffset = 0
|
|
self.pathLength = 0
|
|
}
|
|
}
|
|
|
|
struct Flags: OptionSet {
|
|
let rawValue: UInt16
|
|
|
|
init(rawValue: UInt16) {
|
|
self.rawValue = rawValue
|
|
}
|
|
|
|
static let SHAREFLAG_CLUSTER_RECONNECT = Flags(rawValue: 0x0001)
|
|
}
|
|
}
|
|
|
|
struct TreeConnectResponse: SMBResponseBody {
|
|
let size: UInt16 // = 16
|
|
fileprivate let _type: UInt8
|
|
var type: ShareType {
|
|
return ShareType(rawValue: _type) ?? .UNKNOWN
|
|
}
|
|
fileprivate let reserved: UInt8
|
|
let flags: TreeConnectResponse.ShareFlags
|
|
let capabilities: TreeConnectResponse.Capabilities
|
|
let maximalAccess: FileAccessMask
|
|
|
|
enum ShareType: UInt8 {
|
|
case UNKNOWN = 0x00
|
|
case DISK = 0x01
|
|
case PIPE = 0x02
|
|
case PRINT = 0x03
|
|
}
|
|
|
|
struct ShareFlags: OptionSet {
|
|
let rawValue: UInt32
|
|
|
|
init(rawValue: UInt32) {
|
|
self.rawValue = rawValue
|
|
}
|
|
|
|
static let DFS = ShareFlags(rawValue: 0x00000001)
|
|
static let DFS_ROOT = ShareFlags(rawValue: 0x00000002)
|
|
static let MANUAL_CACHING = ShareFlags(rawValue: 0x00000000)
|
|
static let AUTO_CACHING = ShareFlags(rawValue: 0x00000010)
|
|
static let VDO_CACHING = ShareFlags(rawValue: 0x00000020)
|
|
static let NO_CACHING = ShareFlags(rawValue: 0x00000030)
|
|
static let RESTRICT_EXCLUSIVE_OPENS = ShareFlags(rawValue: 0x00000100)
|
|
static let FORCE_SHARED_DELETE = ShareFlags(rawValue: 0x00000200)
|
|
static let ALLOW_NAMESPACE_CACHING = ShareFlags(rawValue: 0x00000400)
|
|
static let ACCESS_BASED_DIRECTORY_ENUM = ShareFlags(rawValue: 0x00000800)
|
|
static let FORCE_LEVELII_OPLOCK = ShareFlags(rawValue: 0x00001000)
|
|
static let ENABLE_HASH_V1 = ShareFlags(rawValue: 0x00002000)
|
|
static let ENABLE_HASH_V2 = ShareFlags(rawValue: 0x00004000)
|
|
static let ENCRYPT_DATA = ShareFlags(rawValue: 0x00008000)
|
|
}
|
|
|
|
struct Capabilities: OptionSet {
|
|
let rawValue: UInt32
|
|
|
|
init(rawValue: UInt32) {
|
|
self.rawValue = rawValue
|
|
}
|
|
|
|
static let DFS = Capabilities(rawValue: 0x00000008)
|
|
static let CONTINUOUS_AVAILABILITY = Capabilities(rawValue: 0x00000010)
|
|
static let SCALEOUT = Capabilities(rawValue: 0x00000020)
|
|
static let CLUSTER = Capabilities(rawValue: 0x00000040)
|
|
static let ASYMMETRIC = Capabilities(rawValue: 0x00000080)
|
|
}
|
|
}
|
|
|
|
// MARK: SMB2 Tree Disconnect
|
|
|
|
struct TreeDisconnect: SMBRequestBody, SMBResponseBody {
|
|
let size: UInt16
|
|
let reserved: UInt16
|
|
|
|
init() {
|
|
self.size = 4
|
|
self.reserved = 0
|
|
}
|
|
}
|
|
}
|