Skip to content

Add Windows Support to SKSupport #99

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
May 3, 2019
Merged

Add Windows Support to SKSupport #99

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
47 changes: 37 additions & 10 deletions Sources/SKSupport/dlopen.swift
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,9 +13,14 @@
import SPMLibc

public final class DLHandle {
var rawValue: UnsafeMutableRawPointer? = nil
#if os(Windows)
typealias Handle = HMODULE
#else
typealias Handle = UnsafeMutableRawPointer
#endif
var rawValue: Handle? = nil

init(rawValue: UnsafeMutableRawPointer) {
init(rawValue: Handle) {
self.rawValue = rawValue
}

Expand All @@ -25,9 +30,15 @@ public final class DLHandle {

public func close() throws {
if let handle = rawValue {
guard dlclose(handle) == 0 else {
throw DLError.dlclose(dlerror() ?? "unknown error")
}
#if os(Windows)
guard FreeLibrary(handle) != 0 else {
throw DLError.close("Failed to FreeLibrary: \(GetLastError())")
}
#else
guard dlclose(handle) == 0 else {
throw DLError.close(dlerror() ?? "unknown error")
}
#endif
}
rawValue = nil
}
Expand All @@ -39,6 +50,7 @@ public final class DLHandle {

public struct DLOpenFlags: RawRepresentable, OptionSet {

#if !os(Windows)
public static let lazy: DLOpenFlags = DLOpenFlags(rawValue: RTLD_LAZY)
public static let now: DLOpenFlags = DLOpenFlags(rawValue: RTLD_NOW)
public static let local: DLOpenFlags = DLOpenFlags(rawValue: RTLD_LOCAL)
Expand All @@ -52,6 +64,7 @@ public struct DLOpenFlags: RawRepresentable, OptionSet {
public static let first: DLOpenFlags = DLOpenFlags(rawValue: 0)
public static let deepBind: DLOpenFlags = DLOpenFlags(rawValue: RTLD_DEEPBIND)
#endif
#endif

public var rawValue: Int32

Expand All @@ -61,31 +74,45 @@ public struct DLOpenFlags: RawRepresentable, OptionSet {
}

public enum DLError: Swift.Error {
case dlopen(String)
case dlclose(String)
case `open`(String)
case close(String)
}

public func dlopen(_ path: String?, mode: DLOpenFlags) throws -> DLHandle {
guard let handle = SPMLibc.dlopen(path, mode.rawValue) else {
throw DLError.dlopen(dlerror() ?? "unknown error")
}
#if os(Windows)
guard let handle = path?.withCString(encodedAs: UTF16.self, LoadLibraryW) else {
throw DLError.open("LoadLibraryW failed: \(GetLastError())")
}
#else
guard let handle = SPMLibc.dlopen(path, mode.rawValue) else {
throw DLError.open(dlerror() ?? "unknown error")
}
#endif
return DLHandle(rawValue: handle)
}

public func dlsym<T>(_ handle: DLHandle, symbol: String) -> T? {
#if os(Windows)
guard let ptr = GetProcAddress(handle.rawValue!, symbol) else {
return nil
}
#else
guard let ptr = dlsym(handle.rawValue!, symbol) else {
return nil
}
#endif
return unsafeBitCast(ptr, to: T.self)
}

public func dlclose(_ handle: DLHandle) throws {
try handle.close()
}

#if !os(Windows)
public func dlerror() -> String? {
if let err: UnsafeMutablePointer<Int8> = dlerror() {
return String(cString: err)
}
return nil
}
#endif