Merge pull request #11 from milseman/aperture_adapters

Windows syscall adapters

Author: milseman

Sources/SystemInternals/Mocking.swift

@@ -92,7 +92,26 @@ import Glibc
 #endif
 
 // TLS helper functions
-#if !os(Windows)
+#if os(Windows)
+internal typealias TLSKey = DWORD
+internal func makeTLSKey() -> TLSKey {
+  var raw: DWORD = FlsAlloc(nil)
+  if raw == FLS_OUT_OF_INDEXES {
+    fatalError("Unable to create key")
+  }
+  return raw
+}
+internal func setTLS(_ key: TLSKey, _ p: UnsafeMutableRawPointer?) {
+  guard 0 != FlsSetValue(key, p) else {
+    fatalError("Unable to set TLS")
+  }
+}
+internal func getTLS(_ key: TLSKey) -> UnsafeMutableRawPointer? {
+  FlsGetValue(key)
+}
+
+#else
+
 internal typealias TLSKey = pthread_key_t
 internal func makeTLSKey() -> TLSKey {
   var raw = pthread_key_t()
@@ -109,9 +128,6 @@ internal func setTLS(_ key: TLSKey, _ p: UnsafeMutableRawPointer?) {
 internal func getTLS(_ key: TLSKey) -> UnsafeMutableRawPointer? {
   pthread_getspecific(key)
 }
-#else
-// TODO: Windows version...
-#error("Unsupported Platform")
 #endif
 
 private let driverKey: TLSKey = { makeTLSKey() }()

Sources/SystemInternals/Syscalls.swift

@@ -35,11 +35,11 @@ private func mockImpl(
   switch driver.forceErrno {
   case .none: break
   case .always(let e):
-    errno = e
+    system_errno = e
     return -1
   case .counted(let e, let count):
     assert(count >= 1)
-    errno = e
+    system_errno = e
     driver.forceErrno = count > 1 ? .counted(errno: e, count: count-1) : .none
     return -1
   }

Sources/SystemInternals/WindowsSyscallAdapters.swift

@@ -0,0 +1,102 @@
+/*
+ This source file is part of the Swift System open source project
+
+ Copyright (c) 2020 Apple Inc. and the Swift System project authors
+ Licensed under Apache License v2.0 with Runtime Library Exception
+
+ See https://swift.org/LICENSE.txt for license information
+*/
+
+#if os(Windows)
+
+@inline(__always)
+internal func open(_ path: UnsafePointer<CChar>, _ oflag: Int32) {
+  var fh: CInt = -1
+  _ = _sopen_s(&fh, path, oflag, _SH_DENYNO, _S_IREAD | _S_IWRITE)
+  return fh
+}
+
+@inline(__always)
+internal func open(
+  _ path: UnsafePointer<CChar>, _ oflag: Int32, _ mode: mode_t
+) -> CInt {
+  // TODO(compnerd): Apply read/write permissions
+  var fh: CInt = -1
+  _ = _sopen_s(&fh, path, oflag, _SH_DENYNO, _S_IREAD | _S_IWRITE)
+  return fh
+}
+
+@inline(__always)
+internal func close(_ fd: Int32) -> Int32 {
+  _close(fd)
+}
+
+@inline(__always)
+internal func lseek(
+  _ fd: Int32, _ off: off_t, _ whence: Int32
+) -> off_t {
+  _lseek(fd, off, whence)
+}
+
+@inline(__always)
+internal func read(
+  _ fd: Int32, _ buf: UnsafeMutableRawPointer!, _ nbyte: Int
+) -> Int {
+  Int(_read(fd, buf, numericCast(nbyte)))
+}
+
+@inline(__always)
+internal func write(
+  _ fd: Int32, _ buf: UnsafeRawPointer!, _ nbyte: Int
+) -> Int {
+  Int(_write(fd, buf, numericCast(nbyte)))
+}
+
+@inline(__always)
+internal func pread(
+  _ fd: Int32, _ buf: UnsafeMutableRawPointer!, _ nbyte: Int, _ offset: off_t
+) -> Int {
+  let handle: intptr_t = _get_osfhandle(fd)
+  if handle == /* INVALID_HANDLE_VALUE */ -1 { return Int(EBADF) }
+
+  // NOTE: this is a non-owning handle, do *not* call CloseHandle on it
+  let hFile: HANDLE = HANDLE(bitPattern: handle)!
+
+  var ovlOverlapped: OVERLAPPED = OVERLAPPED()
+  ovlOverlapped.OffsetHigh = DWORD(UInt32(offset >> 32) & 0xffffffff)
+  ovlOverlapped.Offset = DWORD(UInt32(offset >> 0) & 0xffffffff)
+
+  var nNumberOfBytesRead: DWORD = 0
+  if !ReadFile(hFile, buf, DWORD(nbyte), &nNumberOfBytesRead, &ovlOverlapped) {
+    let _ = GetLastError()
+    // TODO(compnerd) map windows error to errno
+    return Int(-1)
+  }
+  return Int(nNumberOfBytesRead)
+}
+
+@inline(__always)
+internal func pwrite(
+  _ fd: Int32, _ buf: UnsafeRawPointer!, _ nbyte: Int, _ offset: off_t
+) -> Int {
+  let handle: intptr_t = _get_osfhandle(fd)
+  if handle == /* INVALID_HANDLE_VALUE */ -1 { return Int(EBADF) }
+
+  // NOTE: this is a non-owning handle, do *not* call CloseHandle on it
+  let hFile: HANDLE = HANDLE(bitPattern: handle)!
+
+  var ovlOverlapped: OVERLAPPED = OVERLAPPED()
+  ovlOverlapped.OffsetHigh = DWORD(UInt32(offset >> 32) & 0xffffffff)
+  ovlOverlapped.Offset = DWORD(UInt32(offset >> 0) & 0xffffffff)
+
+  var nNumberOfBytesWritten: DWORD = 0
+  if !WriteFile(hFile, buf, DWORD(nbyte), &nNumberOfBytesWritten,
+                &ovlOverlapped) {
+    let _ = GetLastError()
+    // TODO(compnerd) map windows error to errno
+    return Int(-1)
+  }
+  return Int(nNumberOfBytesWritten)
+}
+
+#endif