winlean.nim 46 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182
  1. #
  2. #
  3. # Nim's Runtime Library
  4. # (c) Copyright 2012 Andreas Rumpf
  5. #
  6. # See the file "copying.txt", included in this
  7. # distribution, for details about the copyright.
  8. #
  9. ## This module implements a small wrapper for some needed Win API procedures,
  10. ## so that the Nim compiler does not depend on the huge Windows module.
  11. import dynlib
  12. when defined(nimHasStyleChecks):
  13. {.push styleChecks: off.}
  14. {.passc: "-DWIN32_LEAN_AND_MEAN".}
  15. const
  16. useWinUnicode* = not defined(useWinAnsi)
  17. when useWinUnicode:
  18. type WinChar* = Utf16Char
  19. else:
  20. type WinChar* = char
  21. # See https://docs.microsoft.com/en-us/windows/win32/winprog/windows-data-types
  22. type
  23. Handle* = int
  24. LONG* = int32
  25. ULONG* = int32
  26. PULONG* = ptr int
  27. WINBOOL* = int32
  28. ## `WINBOOL` uses opposite convention as posix, !=0 meaning success.
  29. # xxx this should be distinct int32, distinct would make code less error prone
  30. PBOOL* = ptr WINBOOL
  31. DWORD* = int32
  32. PDWORD* = ptr DWORD
  33. LPINT* = ptr int32
  34. ULONG_PTR* = uint
  35. PULONG_PTR* = ptr uint
  36. HDC* = Handle
  37. HGLRC* = Handle
  38. BYTE* = uint8
  39. SECURITY_ATTRIBUTES* {.final, pure.} = object
  40. nLength*: int32
  41. lpSecurityDescriptor*: pointer
  42. bInheritHandle*: WINBOOL
  43. STARTUPINFO* {.final, pure.} = object
  44. cb*: int32
  45. lpReserved*: cstring
  46. lpDesktop*: cstring
  47. lpTitle*: cstring
  48. dwX*: int32
  49. dwY*: int32
  50. dwXSize*: int32
  51. dwYSize*: int32
  52. dwXCountChars*: int32
  53. dwYCountChars*: int32
  54. dwFillAttribute*: int32
  55. dwFlags*: int32
  56. wShowWindow*: int16
  57. cbReserved2*: int16
  58. lpReserved2*: pointer
  59. hStdInput*: Handle
  60. hStdOutput*: Handle
  61. hStdError*: Handle
  62. PROCESS_INFORMATION* {.final, pure.} = object
  63. hProcess*: Handle
  64. hThread*: Handle
  65. dwProcessId*: int32
  66. dwThreadId*: int32
  67. FILETIME* {.final, pure.} = object ## CANNOT BE int64 BECAUSE OF ALIGNMENT
  68. dwLowDateTime*: DWORD
  69. dwHighDateTime*: DWORD
  70. BY_HANDLE_FILE_INFORMATION* {.final, pure.} = object
  71. dwFileAttributes*: DWORD
  72. ftCreationTime*: FILETIME
  73. ftLastAccessTime*: FILETIME
  74. ftLastWriteTime*: FILETIME
  75. dwVolumeSerialNumber*: DWORD
  76. nFileSizeHigh*: DWORD
  77. nFileSizeLow*: DWORD
  78. nNumberOfLinks*: DWORD
  79. nFileIndexHigh*: DWORD
  80. nFileIndexLow*: DWORD
  81. OSVERSIONINFO* {.final, pure.} = object
  82. dwOSVersionInfoSize*: DWORD
  83. dwMajorVersion*: DWORD
  84. dwMinorVersion*: DWORD
  85. dwBuildNumber*: DWORD
  86. dwPlatformId*: DWORD
  87. szCSDVersion*: array[0..127, WinChar]
  88. Protoent* = object
  89. p_name*: cstring
  90. p_aliases*: cstringArray
  91. p_proto*: cshort
  92. const
  93. STARTF_USESHOWWINDOW* = 1'i32
  94. STARTF_USESTDHANDLES* = 256'i32
  95. HIGH_PRIORITY_CLASS* = 128'i32
  96. IDLE_PRIORITY_CLASS* = 64'i32
  97. NORMAL_PRIORITY_CLASS* = 32'i32
  98. REALTIME_PRIORITY_CLASS* = 256'i32
  99. WAIT_OBJECT_0* = 0'i32
  100. WAIT_TIMEOUT* = 0x00000102'i32
  101. WAIT_FAILED* = 0xFFFFFFFF'i32
  102. INFINITE* = -1'i32
  103. STILL_ACTIVE* = 0x00000103'i32
  104. STD_INPUT_HANDLE* = -10'i32
  105. STD_OUTPUT_HANDLE* = -11'i32
  106. STD_ERROR_HANDLE* = -12'i32
  107. DETACHED_PROCESS* = 8'i32
  108. SW_SHOWNORMAL* = 1'i32
  109. INVALID_HANDLE_VALUE* = Handle(-1)
  110. CREATE_UNICODE_ENVIRONMENT* = 1024'i32
  111. PIPE_ACCESS_DUPLEX* = 0x00000003'i32
  112. PIPE_ACCESS_INBOUND* = 1'i32
  113. PIPE_ACCESS_OUTBOUND* = 2'i32
  114. PIPE_NOWAIT* = 0x00000001'i32
  115. SYNCHRONIZE* = 0x00100000'i32
  116. CREATE_NO_WINDOW* = 0x08000000'i32
  117. HANDLE_FLAG_INHERIT* = 0x00000001'i32
  118. proc isSuccess*(a: WINBOOL): bool {.inline.} =
  119. ## Returns true if `a != 0`. Windows uses a different convention than POSIX,
  120. ## where `a == 0` is commonly used on success.
  121. a != 0
  122. proc getVersionExW*(lpVersionInfo: ptr OSVERSIONINFO): WINBOOL {.
  123. stdcall, dynlib: "kernel32", importc: "GetVersionExW", sideEffect.}
  124. proc getVersionExA*(lpVersionInfo: ptr OSVERSIONINFO): WINBOOL {.
  125. stdcall, dynlib: "kernel32", importc: "GetVersionExA", sideEffect.}
  126. proc getVersion*(): DWORD {.stdcall, dynlib: "kernel32", importc: "GetVersion", sideEffect.}
  127. proc closeHandle*(hObject: Handle): WINBOOL {.stdcall, dynlib: "kernel32",
  128. importc: "CloseHandle".}
  129. proc readFile*(hFile: Handle, buffer: pointer, nNumberOfBytesToRead: int32,
  130. lpNumberOfBytesRead: ptr int32, lpOverlapped: pointer): WINBOOL{.
  131. stdcall, dynlib: "kernel32", importc: "ReadFile", sideEffect.}
  132. proc writeFile*(hFile: Handle, buffer: pointer, nNumberOfBytesToWrite: int32,
  133. lpNumberOfBytesWritten: ptr int32,
  134. lpOverlapped: pointer): WINBOOL{.
  135. stdcall, dynlib: "kernel32", importc: "WriteFile", sideEffect.}
  136. proc createPipe*(hReadPipe, hWritePipe: var Handle,
  137. lpPipeAttributes: var SECURITY_ATTRIBUTES,
  138. nSize: int32): WINBOOL{.
  139. stdcall, dynlib: "kernel32", importc: "CreatePipe", sideEffect.}
  140. proc createNamedPipe*(lpName: WideCString,
  141. dwOpenMode, dwPipeMode, nMaxInstances, nOutBufferSize,
  142. nInBufferSize, nDefaultTimeOut: int32,
  143. lpSecurityAttributes: ptr SECURITY_ATTRIBUTES): Handle {.
  144. stdcall, dynlib: "kernel32", importc: "CreateNamedPipeW", sideEffect.}
  145. proc peekNamedPipe*(hNamedPipe: Handle, lpBuffer: pointer=nil,
  146. nBufferSize: int32 = 0,
  147. lpBytesRead: ptr int32 = nil,
  148. lpTotalBytesAvail: ptr int32 = nil,
  149. lpBytesLeftThisMessage: ptr int32 = nil): bool {.
  150. stdcall, dynlib: "kernel32", importc: "PeekNamedPipe".}
  151. when useWinUnicode:
  152. proc createProcessW*(lpApplicationName, lpCommandLine: WideCString,
  153. lpProcessAttributes: ptr SECURITY_ATTRIBUTES,
  154. lpThreadAttributes: ptr SECURITY_ATTRIBUTES,
  155. bInheritHandles: WINBOOL, dwCreationFlags: int32,
  156. lpEnvironment, lpCurrentDirectory: WideCString,
  157. lpStartupInfo: var STARTUPINFO,
  158. lpProcessInformation: var PROCESS_INFORMATION): WINBOOL{.
  159. stdcall, dynlib: "kernel32", importc: "CreateProcessW", sideEffect.}
  160. else:
  161. proc createProcessA*(lpApplicationName, lpCommandLine: cstring,
  162. lpProcessAttributes: ptr SECURITY_ATTRIBUTES,
  163. lpThreadAttributes: ptr SECURITY_ATTRIBUTES,
  164. bInheritHandles: WINBOOL, dwCreationFlags: int32,
  165. lpEnvironment: pointer, lpCurrentDirectory: cstring,
  166. lpStartupInfo: var STARTUPINFO,
  167. lpProcessInformation: var PROCESS_INFORMATION): WINBOOL{.
  168. stdcall, dynlib: "kernel32", importc: "CreateProcessA", sideEffect.}
  169. proc suspendThread*(hThread: Handle): int32 {.stdcall, dynlib: "kernel32",
  170. importc: "SuspendThread", sideEffect.}
  171. proc resumeThread*(hThread: Handle): int32 {.stdcall, dynlib: "kernel32",
  172. importc: "ResumeThread", sideEffect.}
  173. proc waitForSingleObject*(hHandle: Handle, dwMilliseconds: int32): int32 {.
  174. stdcall, dynlib: "kernel32", importc: "WaitForSingleObject", sideEffect.}
  175. proc terminateProcess*(hProcess: Handle, uExitCode: int): WINBOOL {.stdcall,
  176. dynlib: "kernel32", importc: "TerminateProcess", sideEffect.}
  177. proc getExitCodeProcess*(hProcess: Handle, lpExitCode: var int32): WINBOOL {.
  178. stdcall, dynlib: "kernel32", importc: "GetExitCodeProcess".}
  179. proc getStdHandle*(nStdHandle: int32): Handle {.stdcall, dynlib: "kernel32",
  180. importc: "GetStdHandle".}
  181. proc setStdHandle*(nStdHandle: int32, hHandle: Handle): WINBOOL {.stdcall,
  182. dynlib: "kernel32", importc: "SetStdHandle", sideEffect.}
  183. proc flushFileBuffers*(hFile: Handle): WINBOOL {.stdcall, dynlib: "kernel32",
  184. importc: "FlushFileBuffers", sideEffect.}
  185. proc getLastError*(): int32 {.importc: "GetLastError",
  186. stdcall, dynlib: "kernel32", sideEffect.}
  187. proc setLastError*(error: int32) {.importc: "SetLastError",
  188. stdcall, dynlib: "kernel32", sideEffect.}
  189. when useWinUnicode:
  190. proc formatMessageW*(dwFlags: int32, lpSource: pointer,
  191. dwMessageId, dwLanguageId: int32,
  192. lpBuffer: pointer, nSize: int32,
  193. arguments: pointer): int32 {.
  194. importc: "FormatMessageW", stdcall, dynlib: "kernel32".}
  195. else:
  196. proc formatMessageA*(dwFlags: int32, lpSource: pointer,
  197. dwMessageId, dwLanguageId: int32,
  198. lpBuffer: pointer, nSize: int32,
  199. arguments: pointer): int32 {.
  200. importc: "FormatMessageA", stdcall, dynlib: "kernel32".}
  201. proc localFree*(p: pointer) {.
  202. importc: "LocalFree", stdcall, dynlib: "kernel32".}
  203. when useWinUnicode:
  204. proc getCurrentDirectoryW*(nBufferLength: int32,
  205. lpBuffer: WideCString): int32 {.
  206. importc: "GetCurrentDirectoryW", dynlib: "kernel32", stdcall, sideEffect.}
  207. proc setCurrentDirectoryW*(lpPathName: WideCString): int32 {.
  208. importc: "SetCurrentDirectoryW", dynlib: "kernel32", stdcall, sideEffect.}
  209. proc createDirectoryW*(pathName: WideCString, security: pointer=nil): int32 {.
  210. importc: "CreateDirectoryW", dynlib: "kernel32", stdcall, sideEffect.}
  211. proc removeDirectoryW*(lpPathName: WideCString): int32 {.
  212. importc: "RemoveDirectoryW", dynlib: "kernel32", stdcall, sideEffect.}
  213. proc setEnvironmentVariableW*(lpName, lpValue: WideCString): int32 {.
  214. stdcall, dynlib: "kernel32", importc: "SetEnvironmentVariableW", sideEffect.}
  215. proc getModuleFileNameW*(handle: Handle, buf: WideCString,
  216. size: int32): int32 {.importc: "GetModuleFileNameW",
  217. dynlib: "kernel32", stdcall.}
  218. else:
  219. proc getCurrentDirectoryA*(nBufferLength: int32, lpBuffer: cstring): int32 {.
  220. importc: "GetCurrentDirectoryA", dynlib: "kernel32", stdcall, sideEffect.}
  221. proc setCurrentDirectoryA*(lpPathName: cstring): int32 {.
  222. importc: "SetCurrentDirectoryA", dynlib: "kernel32", stdcall, sideEffect.}
  223. proc createDirectoryA*(pathName: cstring, security: pointer=nil): int32 {.
  224. importc: "CreateDirectoryA", dynlib: "kernel32", stdcall, sideEffect.}
  225. proc removeDirectoryA*(lpPathName: cstring): int32 {.
  226. importc: "RemoveDirectoryA", dynlib: "kernel32", stdcall, sideEffect.}
  227. proc setEnvironmentVariableA*(lpName, lpValue: cstring): int32 {.
  228. stdcall, dynlib: "kernel32", importc: "SetEnvironmentVariableA", sideEffect.}
  229. proc getModuleFileNameA*(handle: Handle, buf: cstring, size: int32): int32 {.
  230. importc: "GetModuleFileNameA", dynlib: "kernel32", stdcall.}
  231. when useWinUnicode:
  232. proc createSymbolicLinkW*(lpSymlinkFileName, lpTargetFileName: WideCString,
  233. flags: DWORD): int32 {.
  234. importc:"CreateSymbolicLinkW", dynlib: "kernel32", stdcall, sideEffect.}
  235. proc createHardLinkW*(lpFileName, lpExistingFileName: WideCString,
  236. security: pointer=nil): int32 {.
  237. importc:"CreateHardLinkW", dynlib: "kernel32", stdcall, sideEffect.}
  238. else:
  239. proc createSymbolicLinkA*(lpSymlinkFileName, lpTargetFileName: cstring,
  240. flags: DWORD): int32 {.
  241. importc:"CreateSymbolicLinkA", dynlib: "kernel32", stdcall, sideEffect.}
  242. proc createHardLinkA*(lpFileName, lpExistingFileName: cstring,
  243. security: pointer=nil): int32 {.
  244. importc:"CreateHardLinkA", dynlib: "kernel32", stdcall, sideEffect.}
  245. const
  246. FILE_ATTRIBUTE_READONLY* = 0x00000001'i32
  247. FILE_ATTRIBUTE_HIDDEN* = 0x00000002'i32
  248. FILE_ATTRIBUTE_SYSTEM* = 0x00000004'i32
  249. FILE_ATTRIBUTE_DIRECTORY* = 0x00000010'i32
  250. FILE_ATTRIBUTE_ARCHIVE* = 0x00000020'i32
  251. FILE_ATTRIBUTE_DEVICE* = 0x00000040'i32
  252. FILE_ATTRIBUTE_NORMAL* = 0x00000080'i32
  253. FILE_ATTRIBUTE_TEMPORARY* = 0x00000100'i32
  254. FILE_ATTRIBUTE_SPARSE_FILE* = 0x00000200'i32
  255. FILE_ATTRIBUTE_REPARSE_POINT* = 0x00000400'i32
  256. FILE_ATTRIBUTE_COMPRESSED* = 0x00000800'i32
  257. FILE_ATTRIBUTE_OFFLINE* = 0x00001000'i32
  258. FILE_ATTRIBUTE_NOT_CONTENT_INDEXED* = 0x00002000'i32
  259. FILE_FLAG_FIRST_PIPE_INSTANCE* = 0x00080000'i32
  260. FILE_FLAG_OPEN_NO_RECALL* = 0x00100000'i32
  261. FILE_FLAG_OPEN_REPARSE_POINT* = 0x00200000'i32
  262. FILE_FLAG_POSIX_SEMANTICS* = 0x01000000'i32
  263. FILE_FLAG_BACKUP_SEMANTICS* = 0x02000000'i32
  264. FILE_FLAG_DELETE_ON_CLOSE* = 0x04000000'i32
  265. FILE_FLAG_SEQUENTIAL_SCAN* = 0x08000000'i32
  266. FILE_FLAG_RANDOM_ACCESS* = 0x10000000'i32
  267. FILE_FLAG_NO_BUFFERING* = 0x20000000'i32
  268. FILE_FLAG_OVERLAPPED* = 0x40000000'i32
  269. FILE_FLAG_WRITE_THROUGH* = 0x80000000'i32
  270. MAX_PATH* = 260
  271. MOVEFILE_COPY_ALLOWED* = 0x2'i32
  272. MOVEFILE_CREATE_HARDLINK* = 0x10'i32
  273. MOVEFILE_DELAY_UNTIL_REBOOT* = 0x4'i32
  274. MOVEFILE_FAIL_IF_NOT_TRACKABLE* = 0x20'i32
  275. MOVEFILE_REPLACE_EXISTING* = 0x1'i32
  276. MOVEFILE_WRITE_THROUGH* = 0x8'i32
  277. type
  278. WIN32_FIND_DATA* {.pure.} = object
  279. dwFileAttributes*: int32
  280. ftCreationTime*: FILETIME
  281. ftLastAccessTime*: FILETIME
  282. ftLastWriteTime*: FILETIME
  283. nFileSizeHigh*: int32
  284. nFileSizeLow*: int32
  285. dwReserved0: int32
  286. dwReserved1: int32
  287. cFileName*: array[0..(MAX_PATH) - 1, WinChar]
  288. cAlternateFileName*: array[0..13, WinChar]
  289. when useWinUnicode:
  290. proc findFirstFileW*(lpFileName: WideCString,
  291. lpFindFileData: var WIN32_FIND_DATA): Handle {.
  292. stdcall, dynlib: "kernel32", importc: "FindFirstFileW", sideEffect.}
  293. proc findNextFileW*(hFindFile: Handle,
  294. lpFindFileData: var WIN32_FIND_DATA): int32 {.
  295. stdcall, dynlib: "kernel32", importc: "FindNextFileW", sideEffect.}
  296. else:
  297. proc findFirstFileA*(lpFileName: cstring,
  298. lpFindFileData: var WIN32_FIND_DATA): Handle {.
  299. stdcall, dynlib: "kernel32", importc: "FindFirstFileA", sideEffect.}
  300. proc findNextFileA*(hFindFile: Handle,
  301. lpFindFileData: var WIN32_FIND_DATA): int32 {.
  302. stdcall, dynlib: "kernel32", importc: "FindNextFileA", sideEffect.}
  303. proc findClose*(hFindFile: Handle) {.stdcall, dynlib: "kernel32",
  304. importc: "FindClose".}
  305. when useWinUnicode:
  306. proc getFullPathNameW*(lpFileName: WideCString, nBufferLength: int32,
  307. lpBuffer: WideCString,
  308. lpFilePart: var WideCString): int32 {.
  309. stdcall, dynlib: "kernel32",
  310. importc: "GetFullPathNameW", sideEffect.}
  311. proc getFileAttributesW*(lpFileName: WideCString): int32 {.
  312. stdcall, dynlib: "kernel32",
  313. importc: "GetFileAttributesW", sideEffect.}
  314. proc setFileAttributesW*(lpFileName: WideCString,
  315. dwFileAttributes: int32): WINBOOL {.
  316. stdcall, dynlib: "kernel32", importc: "SetFileAttributesW", sideEffect.}
  317. proc copyFileW*(lpExistingFileName, lpNewFileName: WideCString,
  318. bFailIfExists: WINBOOL): WINBOOL {.
  319. importc: "CopyFileW", stdcall, dynlib: "kernel32", sideEffect.}
  320. proc moveFileW*(lpExistingFileName, lpNewFileName: WideCString): WINBOOL {.
  321. importc: "MoveFileW", stdcall, dynlib: "kernel32", sideEffect.}
  322. proc moveFileExW*(lpExistingFileName, lpNewFileName: WideCString,
  323. flags: DWORD): WINBOOL {.
  324. importc: "MoveFileExW", stdcall, dynlib: "kernel32", sideEffect.}
  325. proc getEnvironmentStringsW*(): WideCString {.
  326. stdcall, dynlib: "kernel32", importc: "GetEnvironmentStringsW", sideEffect.}
  327. proc freeEnvironmentStringsW*(para1: WideCString): int32 {.
  328. stdcall, dynlib: "kernel32", importc: "FreeEnvironmentStringsW", sideEffect.}
  329. proc getCommandLineW*(): WideCString {.importc: "GetCommandLineW",
  330. stdcall, dynlib: "kernel32", sideEffect.}
  331. else:
  332. proc getFullPathNameA*(lpFileName: cstring, nBufferLength: int32,
  333. lpBuffer: cstring, lpFilePart: var cstring): int32 {.
  334. stdcall, dynlib: "kernel32",
  335. importc: "GetFullPathNameA", sideEffect.}
  336. proc getFileAttributesA*(lpFileName: cstring): int32 {.
  337. stdcall, dynlib: "kernel32",
  338. importc: "GetFileAttributesA", sideEffect.}
  339. proc setFileAttributesA*(lpFileName: cstring,
  340. dwFileAttributes: int32): WINBOOL {.
  341. stdcall, dynlib: "kernel32", importc: "SetFileAttributesA", sideEffect.}
  342. proc copyFileA*(lpExistingFileName, lpNewFileName: cstring,
  343. bFailIfExists: cint): cint {.
  344. importc: "CopyFileA", stdcall, dynlib: "kernel32", sideEffect.}
  345. proc moveFileA*(lpExistingFileName, lpNewFileName: cstring): WINBOOL {.
  346. importc: "MoveFileA", stdcall, dynlib: "kernel32", sideEffect.}
  347. proc moveFileExA*(lpExistingFileName, lpNewFileName: cstring,
  348. flags: DWORD): WINBOOL {.
  349. importc: "MoveFileExA", stdcall, dynlib: "kernel32", sideEffect.}
  350. proc getEnvironmentStringsA*(): cstring {.
  351. stdcall, dynlib: "kernel32", importc: "GetEnvironmentStringsA", sideEffect.}
  352. proc freeEnvironmentStringsA*(para1: cstring): int32 {.
  353. stdcall, dynlib: "kernel32", importc: "FreeEnvironmentStringsA", sideEffect.}
  354. proc getCommandLineA*(): cstring {.
  355. importc: "GetCommandLineA", stdcall, dynlib: "kernel32", sideEffect.}
  356. proc rdFileTime*(f: FILETIME): int64 =
  357. result = ze64(f.dwLowDateTime) or (ze64(f.dwHighDateTime) shl 32)
  358. proc rdFileSize*(f: WIN32_FIND_DATA): int64 =
  359. result = ze64(f.nFileSizeLow) or (ze64(f.nFileSizeHigh) shl 32)
  360. proc getSystemTimeAsFileTime*(lpSystemTimeAsFileTime: var FILETIME) {.
  361. importc: "GetSystemTimeAsFileTime", dynlib: "kernel32", stdcall, sideEffect.}
  362. proc sleep*(dwMilliseconds: int32){.stdcall, dynlib: "kernel32",
  363. importc: "Sleep", sideEffect.}
  364. when useWinUnicode:
  365. proc shellExecuteW*(hwnd: Handle, lpOperation, lpFile,
  366. lpParameters, lpDirectory: WideCString,
  367. nShowCmd: int32): Handle{.
  368. stdcall, dynlib: "shell32.dll", importc: "ShellExecuteW", sideEffect.}
  369. else:
  370. proc shellExecuteA*(hwnd: Handle, lpOperation, lpFile,
  371. lpParameters, lpDirectory: cstring,
  372. nShowCmd: int32): Handle{.
  373. stdcall, dynlib: "shell32.dll", importc: "ShellExecuteA", sideEffect.}
  374. proc getFileInformationByHandle*(hFile: Handle,
  375. lpFileInformation: ptr BY_HANDLE_FILE_INFORMATION): WINBOOL{.
  376. stdcall, dynlib: "kernel32", importc: "GetFileInformationByHandle", sideEffect.}
  377. const
  378. WSADESCRIPTION_LEN* = 256
  379. WSASYS_STATUS_LEN* = 128
  380. FD_SETSIZE* = 64
  381. MSG_PEEK* = 2
  382. INADDR_ANY* = 0'u32
  383. INADDR_LOOPBACK* = 0x7F000001
  384. INADDR_BROADCAST* = -1
  385. INADDR_NONE* = -1
  386. ws2dll = "Ws2_32.dll"
  387. proc wsaGetLastError*(): cint {.importc: "WSAGetLastError", dynlib: ws2dll, sideEffect.}
  388. type
  389. SocketHandle* = distinct int
  390. type
  391. WSAData* {.importc: "WSADATA", header: "winsock2.h".} = object
  392. wVersion, wHighVersion: int16
  393. szDescription: array[0..WSADESCRIPTION_LEN, char]
  394. szSystemStatus: array[0..WSASYS_STATUS_LEN, char]
  395. iMaxSockets, iMaxUdpDg: int16
  396. lpVendorInfo: cstring
  397. SockAddr* {.importc: "SOCKADDR", header: "winsock2.h".} = object
  398. sa_family*: uint16
  399. sa_data*: array[0..13, char]
  400. PSockAddr = ptr SockAddr
  401. InAddr* {.importc: "IN_ADDR", header: "winsock2.h", union.} = object
  402. s_addr*: uint32 # IP address
  403. Sockaddr_in* {.importc: "SOCKADDR_IN",
  404. header: "winsock2.h".} = object
  405. sin_family*: uint16
  406. sin_port*: uint16
  407. sin_addr*: InAddr
  408. sin_zero*: array[0..7, char]
  409. In6_addr* {.importc: "IN6_ADDR", header: "winsock2.h".} = object
  410. bytes* {.importc: "u.Byte".}: array[0..15, char]
  411. Sockaddr_in6* {.importc: "SOCKADDR_IN6",
  412. header: "ws2tcpip.h".} = object
  413. sin6_family*: uint16
  414. sin6_port*: uint16
  415. sin6_flowinfo*: int32 # unsigned
  416. sin6_addr*: In6_addr
  417. sin6_scope_id*: int32 # unsigned
  418. Sockaddr_storage* {.importc: "SOCKADDR_STORAGE",
  419. header: "winsock2.h".} = object
  420. ss_family*: uint16
  421. ss_pad1: array[6, byte]
  422. ss_align: int64
  423. ss_pad2: array[112, byte]
  424. Servent* = object
  425. s_name*: cstring
  426. s_aliases*: cstringArray
  427. when defined(cpu64):
  428. s_proto*: cstring
  429. s_port*: int16
  430. else:
  431. s_port*: int16
  432. s_proto*: cstring
  433. Hostent* = object
  434. h_name*: cstring
  435. h_aliases*: cstringArray
  436. h_addrtype*: int16
  437. h_length*: int16
  438. h_addr_list*: cstringArray
  439. TFdSet* = object
  440. fd_count*: cint # unsigned
  441. fd_array*: array[0..FD_SETSIZE-1, SocketHandle]
  442. AddrInfo* = object
  443. ai_flags*: cint ## Input flags.
  444. ai_family*: cint ## Address family of socket.
  445. ai_socktype*: cint ## Socket type.
  446. ai_protocol*: cint ## Protocol of socket.
  447. ai_addrlen*: csize_t ## Length of socket address.
  448. ai_canonname*: cstring ## Canonical name of service location.
  449. ai_addr*: ptr SockAddr ## Socket address of socket.
  450. ai_next*: ptr AddrInfo ## Pointer to next in list.
  451. SockLen* = cuint
  452. when defined(cpp):
  453. type
  454. Timeval* {.importc: "timeval", header: "<time.h>".} = object
  455. tv_sec*, tv_usec*: int32
  456. else:
  457. type
  458. Timeval* = object
  459. tv_sec*, tv_usec*: int32
  460. var
  461. SOMAXCONN* {.importc, header: "winsock2.h".}: cint
  462. INVALID_SOCKET* {.importc, header: "winsock2.h".}: SocketHandle
  463. SOL_SOCKET* {.importc, header: "winsock2.h".}: cint
  464. SO_DEBUG* {.importc, header: "winsock2.h".}: cint ## turn on debugging info recording
  465. SO_ACCEPTCONN* {.importc, header: "winsock2.h".}: cint # socket has had listen()
  466. SO_REUSEADDR* {.importc, header: "winsock2.h".}: cint # allow local address reuse
  467. SO_REUSEPORT* {.importc: "SO_REUSEADDR", header: "winsock2.h".}: cint # allow port reuse. Since Windows does not really support it, mapped to SO_REUSEADDR. This shouldn't cause problems.
  468. SO_KEEPALIVE* {.importc, header: "winsock2.h".}: cint # keep connections alive
  469. SO_DONTROUTE* {.importc, header: "winsock2.h".}: cint # just use interface addresses
  470. SO_BROADCAST* {.importc, header: "winsock2.h".}: cint # permit sending of broadcast msgs
  471. SO_USELOOPBACK* {.importc, header: "winsock2.h".}: cint # bypass hardware when possible
  472. SO_LINGER* {.importc, header: "winsock2.h".}: cint # linger on close if data present
  473. SO_OOBINLINE* {.importc, header: "winsock2.h".}: cint # leave received OOB data in line
  474. SO_DONTLINGER* {.importc, header: "winsock2.h".}: cint
  475. SO_EXCLUSIVEADDRUSE* {.importc, header: "winsock2.h".}: cint # disallow local address reuse
  476. SO_ERROR* {.importc, header: "winsock2.h".}: cint
  477. TCP_NODELAY* {.importc, header: "winsock2.h".}: cint
  478. proc `==`*(x, y: SocketHandle): bool {.borrow.}
  479. proc getservbyname*(name, proto: cstring): ptr Servent {.
  480. stdcall, importc: "getservbyname", dynlib: ws2dll, sideEffect.}
  481. proc getservbyport*(port: cint, proto: cstring): ptr Servent {.
  482. stdcall, importc: "getservbyport", dynlib: ws2dll, sideEffect.}
  483. proc gethostbyaddr*(ip: ptr InAddr, len: cuint, theType: cint): ptr Hostent {.
  484. stdcall, importc: "gethostbyaddr", dynlib: ws2dll, sideEffect.}
  485. proc gethostbyname*(name: cstring): ptr Hostent {.
  486. stdcall, importc: "gethostbyname", dynlib: ws2dll, sideEffect.}
  487. proc gethostname*(hostname: cstring, len: cint): cint {.
  488. stdcall, importc: "gethostname", dynlib: ws2dll, sideEffect.}
  489. proc getprotobyname*(
  490. name: cstring
  491. ): ptr Protoent {.stdcall, importc: "getprotobyname", dynlib: ws2dll, sideEffect.}
  492. proc getprotobynumber*(
  493. proto: cint
  494. ): ptr Protoent {.stdcall, importc: "getprotobynumber", dynlib: ws2dll, sideEffect.}
  495. proc socket*(af, typ, protocol: cint): SocketHandle {.
  496. stdcall, importc: "socket", dynlib: ws2dll.}
  497. proc closesocket*(s: SocketHandle): cint {.
  498. stdcall, importc: "closesocket", dynlib: ws2dll.}
  499. proc accept*(s: SocketHandle, a: ptr SockAddr, addrlen: ptr SockLen): SocketHandle {.
  500. stdcall, importc: "accept", dynlib: ws2dll.}
  501. proc bindSocket*(s: SocketHandle, name: ptr SockAddr, namelen: SockLen): cint {.
  502. stdcall, importc: "bind", dynlib: ws2dll.}
  503. proc connect*(s: SocketHandle, name: ptr SockAddr, namelen: SockLen): cint {.
  504. stdcall, importc: "connect", dynlib: ws2dll.}
  505. proc getsockname*(s: SocketHandle, name: ptr SockAddr,
  506. namelen: ptr SockLen): cint {.
  507. stdcall, importc: "getsockname", dynlib: ws2dll.}
  508. proc getpeername*(s: SocketHandle, name: ptr SockAddr,
  509. namelen: ptr SockLen): cint {.
  510. stdcall, importc, dynlib: ws2dll.}
  511. proc getsockopt*(s: SocketHandle, level, optname: cint, optval: pointer,
  512. optlen: ptr SockLen): cint {.
  513. stdcall, importc: "getsockopt", dynlib: ws2dll.}
  514. proc setsockopt*(s: SocketHandle, level, optname: cint, optval: pointer,
  515. optlen: SockLen): cint {.
  516. stdcall, importc: "setsockopt", dynlib: ws2dll.}
  517. proc listen*(s: SocketHandle, backlog: cint): cint {.
  518. stdcall, importc: "listen", dynlib: ws2dll.}
  519. proc recv*(s: SocketHandle, buf: pointer, len, flags: cint): cint {.
  520. stdcall, importc: "recv", dynlib: ws2dll.}
  521. proc recvfrom*(s: SocketHandle, buf: cstring, len, flags: cint,
  522. fromm: ptr SockAddr, fromlen: ptr SockLen): cint {.
  523. stdcall, importc: "recvfrom", dynlib: ws2dll.}
  524. proc select*(nfds: cint, readfds, writefds, exceptfds: ptr TFdSet,
  525. timeout: ptr Timeval): cint {.
  526. stdcall, importc: "select", dynlib: ws2dll.}
  527. proc send*(s: SocketHandle, buf: pointer, len, flags: cint): cint {.
  528. stdcall, importc: "send", dynlib: ws2dll.}
  529. proc sendto*(s: SocketHandle, buf: pointer, len, flags: cint,
  530. to: ptr SockAddr, tolen: SockLen): cint {.
  531. stdcall, importc: "sendto", dynlib: ws2dll.}
  532. proc shutdown*(s: SocketHandle, how: cint): cint {.
  533. stdcall, importc: "shutdown", dynlib: ws2dll.}
  534. proc getnameinfo*(a1: ptr SockAddr, a2: SockLen,
  535. a3: cstring, a4: SockLen, a5: cstring,
  536. a6: SockLen, a7: cint): cint {.
  537. stdcall, importc: "getnameinfo", dynlib: ws2dll.}
  538. proc inet_addr*(cp: cstring): uint32 {.
  539. stdcall, importc: "inet_addr", dynlib: ws2dll.}
  540. proc WSAFDIsSet(s: SocketHandle, set: var TFdSet): bool {.
  541. stdcall, importc: "__WSAFDIsSet", dynlib: ws2dll, noSideEffect.}
  542. proc FD_ISSET*(socket: SocketHandle, set: var TFdSet): cint =
  543. result = if WSAFDIsSet(socket, set): 1'i32 else: 0'i32
  544. proc FD_SET*(socket: SocketHandle, s: var TFdSet) =
  545. if s.fd_count < FD_SETSIZE:
  546. s.fd_array[int(s.fd_count)] = socket
  547. inc(s.fd_count)
  548. proc FD_ZERO*(s: var TFdSet) =
  549. s.fd_count = 0
  550. proc wsaStartup*(wVersionRequired: int16, WSData: ptr WSAData): cint {.
  551. stdcall, importc: "WSAStartup", dynlib: ws2dll.}
  552. proc getaddrinfo*(nodename, servname: cstring, hints: ptr AddrInfo,
  553. res: var ptr AddrInfo): cint {.
  554. stdcall, importc: "getaddrinfo", dynlib: ws2dll.}
  555. proc freeAddrInfo*(ai: ptr AddrInfo) {.
  556. stdcall, importc: "freeaddrinfo", dynlib: ws2dll.}
  557. proc inet_ntoa*(i: InAddr): cstring {.
  558. stdcall, importc, dynlib: ws2dll.}
  559. const
  560. MAXIMUM_WAIT_OBJECTS* = 0x00000040
  561. type
  562. WOHandleArray* = array[0..MAXIMUM_WAIT_OBJECTS - 1, Handle]
  563. PWOHandleArray* = ptr WOHandleArray
  564. proc waitForMultipleObjects*(nCount: DWORD, lpHandles: PWOHandleArray,
  565. bWaitAll: WINBOOL, dwMilliseconds: DWORD): DWORD{.
  566. stdcall, dynlib: "kernel32", importc: "WaitForMultipleObjects".}
  567. # for memfiles.nim:
  568. const
  569. GENERIC_READ* = 0x80000000'i32
  570. GENERIC_WRITE* = 0x40000000'i32
  571. GENERIC_ALL* = 0x10000000'i32
  572. FILE_SHARE_READ* = 1'i32
  573. FILE_SHARE_DELETE* = 4'i32
  574. FILE_SHARE_WRITE* = 2'i32
  575. CREATE_ALWAYS* = 2'i32
  576. CREATE_NEW* = 1'i32
  577. OPEN_EXISTING* = 3'i32
  578. OPEN_ALWAYS* = 4'i32
  579. FILE_BEGIN* = 0'i32
  580. INVALID_SET_FILE_POINTER* = -1'i32
  581. NO_ERROR* = 0'i32
  582. PAGE_NOACCESS* = 0x01'i32
  583. PAGE_EXECUTE* = 0x10'i32
  584. PAGE_EXECUTE_READ* = 0x20'i32
  585. PAGE_EXECUTE_READWRITE* = 0x40'i32
  586. PAGE_READONLY* = 2'i32
  587. PAGE_READWRITE* = 4'i32
  588. FILE_MAP_READ* = 4'i32
  589. FILE_MAP_WRITE* = 2'i32
  590. INVALID_FILE_SIZE* = -1'i32
  591. DUPLICATE_SAME_ACCESS* = 2
  592. FILE_READ_DATA* = 0x00000001 # file & pipe
  593. FILE_WRITE_DATA* = 0x00000002 # file & pipe
  594. # Error Constants
  595. const
  596. ERROR_FILE_NOT_FOUND* = 2 ## https://docs.microsoft.com/en-us/windows/win32/debug/system-error-codes--0-499-
  597. ERROR_PATH_NOT_FOUND* = 3
  598. ERROR_ACCESS_DENIED* = 5
  599. ERROR_NO_MORE_FILES* = 18
  600. ERROR_LOCK_VIOLATION* = 33
  601. ERROR_HANDLE_EOF* = 38
  602. ERROR_FILE_EXISTS* = 80
  603. ERROR_BAD_ARGUMENTS* = 165
  604. proc duplicateHandle*(hSourceProcessHandle: Handle, hSourceHandle: Handle,
  605. hTargetProcessHandle: Handle,
  606. lpTargetHandle: ptr Handle,
  607. dwDesiredAccess: DWORD, bInheritHandle: WINBOOL,
  608. dwOptions: DWORD): WINBOOL{.stdcall, dynlib: "kernel32",
  609. importc: "DuplicateHandle".}
  610. proc getHandleInformation*(hObject: Handle, lpdwFlags: ptr DWORD): WINBOOL {.
  611. stdcall, dynlib: "kernel32", importc: "GetHandleInformation".}
  612. proc setHandleInformation*(hObject: Handle, dwMask: DWORD,
  613. dwFlags: DWORD): WINBOOL {.stdcall,
  614. dynlib: "kernel32", importc: "SetHandleInformation".}
  615. proc getCurrentProcess*(): Handle{.stdcall, dynlib: "kernel32",
  616. importc: "GetCurrentProcess".}
  617. proc createFileW*(lpFileName: WideCString, dwDesiredAccess, dwShareMode: DWORD,
  618. lpSecurityAttributes: pointer,
  619. dwCreationDisposition, dwFlagsAndAttributes: DWORD,
  620. hTemplateFile: Handle): Handle {.
  621. stdcall, dynlib: "kernel32", importc: "CreateFileW".}
  622. proc deleteFileW*(pathName: WideCString): int32 {.
  623. importc: "DeleteFileW", dynlib: "kernel32", stdcall.}
  624. proc createFileA*(lpFileName: cstring, dwDesiredAccess, dwShareMode: DWORD,
  625. lpSecurityAttributes: pointer,
  626. dwCreationDisposition, dwFlagsAndAttributes: DWORD,
  627. hTemplateFile: Handle): Handle {.
  628. stdcall, dynlib: "kernel32", importc: "CreateFileA".}
  629. proc deleteFileA*(pathName: cstring): int32 {.
  630. importc: "DeleteFileA", dynlib: "kernel32", stdcall.}
  631. proc setEndOfFile*(hFile: Handle): WINBOOL {.stdcall, dynlib: "kernel32",
  632. importc: "SetEndOfFile".}
  633. proc setFilePointer*(hFile: Handle, lDistanceToMove: LONG,
  634. lpDistanceToMoveHigh: ptr LONG,
  635. dwMoveMethod: DWORD): DWORD {.
  636. stdcall, dynlib: "kernel32", importc: "SetFilePointer".}
  637. proc getFileSize*(hFile: Handle, lpFileSizeHigh: ptr DWORD): DWORD{.stdcall,
  638. dynlib: "kernel32", importc: "GetFileSize".}
  639. when defined(cpu32):
  640. type
  641. WinSizeT* = uint32
  642. else:
  643. type
  644. WinSizeT* = uint64
  645. proc mapViewOfFileEx*(hFileMappingObject: Handle, dwDesiredAccess: DWORD,
  646. dwFileOffsetHigh, dwFileOffsetLow: DWORD,
  647. dwNumberOfBytesToMap: WinSizeT,
  648. lpBaseAddress: pointer): pointer{.
  649. stdcall, dynlib: "kernel32", importc: "MapViewOfFileEx".}
  650. proc createFileMappingW*(hFile: Handle,
  651. lpFileMappingAttributes: pointer,
  652. flProtect, dwMaximumSizeHigh: DWORD,
  653. dwMaximumSizeLow: DWORD,
  654. lpName: pointer): Handle {.
  655. stdcall, dynlib: "kernel32", importc: "CreateFileMappingW".}
  656. when not useWinUnicode:
  657. proc createFileMappingA*(hFile: Handle,
  658. lpFileMappingAttributes: pointer,
  659. flProtect, dwMaximumSizeHigh: DWORD,
  660. dwMaximumSizeLow: DWORD, lpName: cstring): Handle {.
  661. stdcall, dynlib: "kernel32", importc: "CreateFileMappingA".}
  662. proc unmapViewOfFile*(lpBaseAddress: pointer): WINBOOL {.stdcall,
  663. dynlib: "kernel32", importc: "UnmapViewOfFile".}
  664. proc flushViewOfFile*(lpBaseAddress: pointer, dwNumberOfBytesToFlush: DWORD): WINBOOL {.
  665. stdcall, dynlib: "kernel32", importc: "FlushViewOfFile".}
  666. type
  667. OVERLAPPED* {.pure, inheritable.} = object
  668. internal*: PULONG
  669. internalHigh*: PULONG
  670. offset*: DWORD
  671. offsetHigh*: DWORD
  672. hEvent*: Handle
  673. POVERLAPPED* = ptr OVERLAPPED
  674. POVERLAPPED_COMPLETION_ROUTINE* = proc (para1: DWORD, para2: DWORD,
  675. para3: POVERLAPPED){.stdcall.}
  676. GUID* {.final, pure.} = object
  677. D1*: int32
  678. D2*: int16
  679. D3*: int16
  680. D4*: array[0..7, int8]
  681. const
  682. ERROR_IO_PENDING* = 997 # a.k.a WSA_IO_PENDING
  683. WSAECONNABORTED* = 10053
  684. WSAEADDRINUSE* = 10048
  685. WSAECONNRESET* = 10054
  686. WSAEDISCON* = 10101
  687. WSAENETRESET* = 10052
  688. WSAETIMEDOUT* = 10060
  689. WSANOTINITIALISED* = 10093
  690. WSAENOTSOCK* = 10038
  691. WSAEINPROGRESS* = 10036
  692. WSAEINTR* = 10004
  693. WSAEWOULDBLOCK* = 10035
  694. WSAESHUTDOWN* = 10058
  695. ERROR_NETNAME_DELETED* = 64
  696. STATUS_PENDING* = 0x103
  697. proc createIoCompletionPort*(FileHandle: Handle, ExistingCompletionPort: Handle,
  698. CompletionKey: ULONG_PTR,
  699. NumberOfConcurrentThreads: DWORD): Handle{.stdcall,
  700. dynlib: "kernel32", importc: "CreateIoCompletionPort".}
  701. proc getQueuedCompletionStatus*(CompletionPort: Handle,
  702. lpNumberOfBytesTransferred: PDWORD, lpCompletionKey: PULONG_PTR,
  703. lpOverlapped: ptr POVERLAPPED,
  704. dwMilliseconds: DWORD): WINBOOL{.stdcall,
  705. dynlib: "kernel32", importc: "GetQueuedCompletionStatus".}
  706. proc getOverlappedResult*(hFile: Handle, lpOverlapped: POVERLAPPED,
  707. lpNumberOfBytesTransferred: var DWORD, bWait: WINBOOL): WINBOOL{.
  708. stdcall, dynlib: "kernel32", importc: "GetOverlappedResult".}
  709. # this is copy of HasOverlappedIoCompleted() macro from <winbase.h>
  710. # because we have declared own OVERLAPPED structure with member names not
  711. # compatible with original names.
  712. template hasOverlappedIoCompleted*(lpOverlapped): bool =
  713. (cast[uint](lpOverlapped.internal) != STATUS_PENDING)
  714. const
  715. IOC_OUT* = 0x40000000'i32
  716. IOC_IN* = 0x80000000'i32
  717. IOC_WS2* = 0x08000000'i32
  718. IOC_INOUT* = IOC_IN or IOC_OUT
  719. template WSAIORW*(x,y): untyped = (IOC_INOUT or x or y)
  720. const
  721. SIO_GET_EXTENSION_FUNCTION_POINTER* = WSAIORW(IOC_WS2,6).DWORD
  722. SO_UPDATE_ACCEPT_CONTEXT* = 0x700B
  723. AI_V4MAPPED* = 0x0008
  724. AF_UNSPEC* = 0
  725. AF_INET* = 2
  726. AF_INET6* = 23
  727. var
  728. WSAID_CONNECTEX*: GUID = GUID(D1: 0x25a207b9, D2: 0xddf3'i16, D3: 0x4660, D4: [
  729. 0x8e'i8, 0xe9'i8, 0x76'i8, 0xe5'i8, 0x8c'i8, 0x74'i8, 0x06'i8, 0x3e'i8])
  730. WSAID_ACCEPTEX*: GUID = GUID(D1: 0xb5367df1'i32, D2: 0xcbac'i16, D3: 0x11cf, D4: [
  731. 0x95'i8, 0xca'i8, 0x00'i8, 0x80'i8, 0x5f'i8, 0x48'i8, 0xa1'i8, 0x92'i8])
  732. WSAID_GETACCEPTEXSOCKADDRS*: GUID = GUID(D1: 0xb5367df2'i32, D2: 0xcbac'i16, D3: 0x11cf, D4: [
  733. 0x95'i8, 0xca'i8, 0x00'i8, 0x80'i8, 0x5f'i8, 0x48'i8, 0xa1'i8, 0x92'i8])
  734. proc WSAIoctl*(s: SocketHandle, dwIoControlCode: DWORD, lpvInBuffer: pointer,
  735. cbInBuffer: DWORD, lpvOutBuffer: pointer, cbOutBuffer: DWORD,
  736. lpcbBytesReturned: PDWORD, lpOverlapped: POVERLAPPED,
  737. lpCompletionRoutine: POVERLAPPED_COMPLETION_ROUTINE): cint
  738. {.stdcall, importc: "WSAIoctl", dynlib: "Ws2_32.dll".}
  739. type
  740. TWSABuf* {.importc: "WSABUF", header: "winsock2.h".} = object
  741. len*: ULONG
  742. buf*: cstring
  743. proc WSARecv*(s: SocketHandle, buf: ptr TWSABuf, bufCount: DWORD,
  744. bytesReceived, flags: PDWORD, lpOverlapped: POVERLAPPED,
  745. completionProc: POVERLAPPED_COMPLETION_ROUTINE): cint {.
  746. stdcall, importc: "WSARecv", dynlib: "Ws2_32.dll".}
  747. proc WSARecvFrom*(s: SocketHandle, buf: ptr TWSABuf, bufCount: DWORD,
  748. bytesReceived: PDWORD, flags: PDWORD, name: ptr SockAddr,
  749. namelen: ptr cint, lpOverlapped: POVERLAPPED,
  750. completionProc: POVERLAPPED_COMPLETION_ROUTINE): cint {.
  751. stdcall, importc: "WSARecvFrom", dynlib: "Ws2_32.dll".}
  752. proc WSASend*(s: SocketHandle, buf: ptr TWSABuf, bufCount: DWORD,
  753. bytesSent: PDWORD, flags: DWORD, lpOverlapped: POVERLAPPED,
  754. completionProc: POVERLAPPED_COMPLETION_ROUTINE): cint {.
  755. stdcall, importc: "WSASend", dynlib: "Ws2_32.dll".}
  756. proc WSASendTo*(s: SocketHandle, buf: ptr TWSABuf, bufCount: DWORD,
  757. bytesSent: PDWORD, flags: DWORD, name: ptr SockAddr,
  758. namelen: cint, lpOverlapped: POVERLAPPED,
  759. completionProc: POVERLAPPED_COMPLETION_ROUTINE): cint {.
  760. stdcall, importc: "WSASendTo", dynlib: "Ws2_32.dll".}
  761. proc get_osfhandle*(fd:FileHandle): Handle {.
  762. importc: "_get_osfhandle", header:"<io.h>".}
  763. proc getSystemTimes*(lpIdleTime, lpKernelTime,
  764. lpUserTime: var FILETIME): WINBOOL {.stdcall,
  765. dynlib: "kernel32", importc: "GetSystemTimes".}
  766. proc getProcessTimes*(hProcess: Handle; lpCreationTime, lpExitTime,
  767. lpKernelTime, lpUserTime: var FILETIME): WINBOOL {.stdcall,
  768. dynlib: "kernel32", importc: "GetProcessTimes".}
  769. proc getSystemTimePreciseAsFileTime*(lpSystemTimeAsFileTime: var FILETIME) {.
  770. importc: "GetSystemTimePreciseAsFileTime", dynlib: "kernel32", stdcall, sideEffect.}
  771. type inet_ntop_proc = proc(family: cint, paddr: pointer, pStringBuffer: cstring,
  772. stringBufSize: int32): cstring {.gcsafe, stdcall, tags: [].}
  773. var inet_ntop_real: inet_ntop_proc = nil
  774. let ws2 = loadLib(ws2dll)
  775. if ws2 != nil:
  776. inet_ntop_real = cast[inet_ntop_proc](symAddr(ws2, "inet_ntop"))
  777. proc WSAAddressToStringA(pAddr: ptr SockAddr, addrSize: DWORD, unused: pointer, pBuff: cstring, pBuffSize: ptr DWORD): cint {.stdcall, importc, dynlib: ws2dll.}
  778. proc inet_ntop_emulated(family: cint, paddr: pointer, pStringBuffer: cstring,
  779. stringBufSize: int32): cstring {.stdcall.} =
  780. case family
  781. of AF_INET:
  782. var sa: Sockaddr_in
  783. sa.sin_family = AF_INET
  784. sa.sin_addr = cast[ptr InAddr](paddr)[]
  785. var bs = stringBufSize.DWORD
  786. let r = WSAAddressToStringA(cast[ptr SockAddr](sa.addr), sa.sizeof.DWORD, nil, pStringBuffer, bs.addr)
  787. if r != 0:
  788. result = nil
  789. else:
  790. result = pStringBuffer
  791. of AF_INET6:
  792. var sa: Sockaddr_in6
  793. sa.sin6_family = AF_INET6
  794. sa.sin6_addr = cast[ptr In6_addr](paddr)[]
  795. var bs = stringBufSize.DWORD
  796. let r = WSAAddressToStringA(cast[ptr SockAddr](sa.addr), sa.sizeof.DWORD, nil, pStringBuffer, bs.addr)
  797. if r != 0:
  798. result = nil
  799. else:
  800. result = pStringBuffer
  801. else:
  802. setLastError(ERROR_BAD_ARGUMENTS)
  803. result = nil
  804. proc inet_ntop*(family: cint, paddr: pointer, pStringBuffer: cstring,
  805. stringBufSize: int32): cstring {.stdcall.} =
  806. var ver: OSVERSIONINFO
  807. ver.dwOSVersionInfoSize = sizeof(ver).DWORD
  808. let res = when useWinUnicode: getVersionExW(ver.addr) else: getVersionExA(ver.addr)
  809. if res == 0:
  810. result = nil
  811. elif ver.dwMajorVersion >= 6:
  812. if inet_ntop_real == nil:
  813. quit("Can't load inet_ntop proc from " & ws2dll)
  814. result = inet_ntop_real(family, paddr, pStringBuffer, stringBufSize)
  815. else:
  816. result = inet_ntop_emulated(family, paddr, pStringBuffer, stringBufSize)
  817. type
  818. WSAPROC_ACCEPTEX* = proc (sListenSocket: SocketHandle,
  819. sAcceptSocket: SocketHandle,
  820. lpOutputBuffer: pointer, dwReceiveDataLength: DWORD,
  821. dwLocalAddressLength: DWORD,
  822. dwRemoteAddressLength: DWORD,
  823. lpdwBytesReceived: ptr DWORD,
  824. lpOverlapped: POVERLAPPED): bool {.
  825. stdcall, gcsafe, raises: [].}
  826. WSAPROC_CONNECTEX* = proc (s: SocketHandle, name: ptr SockAddr, namelen: cint,
  827. lpSendBuffer: pointer, dwSendDataLength: DWORD,
  828. lpdwBytesSent: ptr DWORD,
  829. lpOverlapped: POVERLAPPED): bool {.
  830. stdcall, gcsafe, raises: [].}
  831. WSAPROC_GETACCEPTEXSOCKADDRS* = proc(lpOutputBuffer: pointer,
  832. dwReceiveDataLength: DWORD,
  833. dwLocalAddressLength: DWORD,
  834. dwRemoteAddressLength: DWORD,
  835. LocalSockaddr: ptr PSockAddr,
  836. LocalSockaddrLength: ptr cint,
  837. RemoteSockaddr: ptr PSockAddr,
  838. RemoteSockaddrLength: ptr cint) {.
  839. stdcall, gcsafe, raises: [].}
  840. const
  841. WT_EXECUTEDEFAULT* = 0x00000000'i32
  842. WT_EXECUTEINIOTHREAD* = 0x00000001'i32
  843. WT_EXECUTEINUITHREAD* = 0x00000002'i32
  844. WT_EXECUTEINWAITTHREAD* = 0x00000004'i32
  845. WT_EXECUTEONLYONCE* = 0x00000008'i32
  846. WT_EXECUTELONGFUNCTION* = 0x00000010'i32
  847. WT_EXECUTEINTIMERTHREAD* = 0x00000020'i32
  848. WT_EXECUTEINPERSISTENTIOTHREAD* = 0x00000040'i32
  849. WT_EXECUTEINPERSISTENTTHREAD* = 0x00000080'i32
  850. WT_TRANSFER_IMPERSONATION* = 0x00000100'i32
  851. PROCESS_TERMINATE* = 0x00000001'i32
  852. PROCESS_CREATE_THREAD* = 0x00000002'i32
  853. PROCESS_SET_SESSIONID* = 0x00000004'i32
  854. PROCESS_VM_OPERATION* = 0x00000008'i32
  855. PROCESS_VM_READ* = 0x00000010'i32
  856. PROCESS_VM_WRITE* = 0x00000020'i32
  857. PROCESS_DUP_HANDLE* = 0x00000040'i32
  858. PROCESS_CREATE_PROCESS* = 0x00000080'i32
  859. PROCESS_SET_QUOTA* = 0x00000100'i32
  860. PROCESS_SET_INFORMATION* = 0x00000200'i32
  861. PROCESS_QUERY_INFORMATION* = 0x00000400'i32
  862. PROCESS_SUSPEND_RESUME* = 0x00000800'i32
  863. PROCESS_QUERY_LIMITED_INFORMATION* = 0x00001000'i32
  864. PROCESS_SET_LIMITED_INFORMATION* = 0x00002000'i32
  865. type
  866. WAITORTIMERCALLBACK* = proc(para1: pointer, para2: int32) {.stdcall.}
  867. proc postQueuedCompletionStatus*(CompletionPort: Handle,
  868. dwNumberOfBytesTransferred: DWORD,
  869. dwCompletionKey: ULONG_PTR,
  870. lpOverlapped: pointer): bool
  871. {.stdcall, dynlib: "kernel32", importc: "PostQueuedCompletionStatus".}
  872. proc registerWaitForSingleObject*(phNewWaitObject: ptr Handle, hObject: Handle,
  873. Callback: WAITORTIMERCALLBACK,
  874. Context: pointer,
  875. dwMilliseconds: ULONG,
  876. dwFlags: ULONG): bool
  877. {.stdcall, dynlib: "kernel32", importc: "RegisterWaitForSingleObject".}
  878. proc unregisterWait*(WaitHandle: Handle): DWORD
  879. {.stdcall, dynlib: "kernel32", importc: "UnregisterWait".}
  880. proc openProcess*(dwDesiredAccess: DWORD, bInheritHandle: WINBOOL,
  881. dwProcessId: DWORD): Handle
  882. {.stdcall, dynlib: "kernel32", importc: "OpenProcess".}
  883. when defined(useWinAnsi):
  884. proc createEvent*(lpEventAttributes: ptr SECURITY_ATTRIBUTES,
  885. bManualReset: DWORD, bInitialState: DWORD,
  886. lpName: cstring): Handle
  887. {.stdcall, dynlib: "kernel32", importc: "CreateEventA".}
  888. else:
  889. proc createEvent*(lpEventAttributes: ptr SECURITY_ATTRIBUTES,
  890. bManualReset: DWORD, bInitialState: DWORD,
  891. lpName: ptr Utf16Char): Handle
  892. {.stdcall, dynlib: "kernel32", importc: "CreateEventW".}
  893. proc setEvent*(hEvent: Handle): cint
  894. {.stdcall, dynlib: "kernel32", importc: "SetEvent".}
  895. const
  896. FD_READ* = 0x00000001'i32
  897. FD_WRITE* = 0x00000002'i32
  898. FD_OOB* = 0x00000004'i32
  899. FD_ACCEPT* = 0x00000008'i32
  900. FD_CONNECT* = 0x00000010'i32
  901. FD_CLOSE* = 0x00000020'i32
  902. FD_QQS* = 0x00000040'i32
  903. FD_GROUP_QQS* = 0x00000080'i32
  904. FD_ROUTING_INTERFACE_CHANGE* = 0x00000100'i32
  905. FD_ADDRESS_LIST_CHANGE* = 0x00000200'i32
  906. FD_ALL_EVENTS* = 0x000003FF'i32
  907. proc wsaEventSelect*(s: SocketHandle, hEventObject: Handle,
  908. lNetworkEvents: clong): cint
  909. {.stdcall, importc: "WSAEventSelect", dynlib: "ws2_32.dll".}
  910. proc wsaCreateEvent*(): Handle
  911. {.stdcall, importc: "WSACreateEvent", dynlib: "ws2_32.dll".}
  912. proc wsaCloseEvent*(hEvent: Handle): bool
  913. {.stdcall, importc: "WSACloseEvent", dynlib: "ws2_32.dll".}
  914. proc wsaResetEvent*(hEvent: Handle): bool
  915. {.stdcall, importc: "WSAResetEvent", dynlib: "ws2_32.dll".}
  916. type
  917. KEY_EVENT_RECORD* {.final, pure.} = object
  918. eventType*: int16
  919. bKeyDown*: WINBOOL
  920. wRepeatCount*: int16
  921. wVirtualKeyCode*: int16
  922. wVirtualScanCode*: int16
  923. uChar*: int16
  924. dwControlKeyState*: DWORD
  925. when defined(useWinAnsi):
  926. proc readConsoleInput*(hConsoleInput: Handle, lpBuffer: pointer, nLength: cint,
  927. lpNumberOfEventsRead: ptr cint): cint
  928. {.stdcall, dynlib: "kernel32", importc: "ReadConsoleInputA".}
  929. else:
  930. proc readConsoleInput*(hConsoleInput: Handle, lpBuffer: pointer, nLength: cint,
  931. lpNumberOfEventsRead: ptr cint): cint
  932. {.stdcall, dynlib: "kernel32", importc: "ReadConsoleInputW".}
  933. type
  934. LPFIBER_START_ROUTINE* = proc (param: pointer) {.stdcall.}
  935. const
  936. FIBER_FLAG_FLOAT_SWITCH* = 0x01
  937. proc CreateFiber*(stackSize: int, fn: LPFIBER_START_ROUTINE, param: pointer): pointer {.stdcall, discardable, dynlib: "kernel32", importc.}
  938. proc CreateFiberEx*(stkCommit: int, stkReserve: int, flags: int32, fn: LPFIBER_START_ROUTINE, param: pointer): pointer {.stdcall, discardable, dynlib: "kernel32", importc.}
  939. proc ConvertThreadToFiber*(param: pointer): pointer {.stdcall, discardable, dynlib: "kernel32", importc.}
  940. proc ConvertThreadToFiberEx*(param: pointer, flags: int32): pointer {.stdcall, discardable, dynlib: "kernel32", importc.}
  941. proc DeleteFiber*(fiber: pointer) {.stdcall, discardable, dynlib: "kernel32", importc.}
  942. proc SwitchToFiber*(fiber: pointer) {.stdcall, discardable, dynlib: "kernel32", importc.}
  943. proc GetCurrentFiber*(): pointer {.stdcall, importc, header: "windows.h".}
  944. proc toFILETIME*(t: int64): FILETIME =
  945. ## Convert the Windows file time timestamp `t` to `FILETIME`.
  946. result = FILETIME(dwLowDateTime: cast[DWORD](t), dwHighDateTime: DWORD(t shr 32))
  947. type
  948. LPFILETIME* = ptr FILETIME
  949. proc setFileTime*(hFile: Handle, lpCreationTime: LPFILETIME,
  950. lpLastAccessTime: LPFILETIME, lpLastWriteTime: LPFILETIME): WINBOOL
  951. {.stdcall, dynlib: "kernel32", importc: "SetFileTime".}
  952. type
  953. # https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-sid_identifier_authority
  954. SID_IDENTIFIER_AUTHORITY* {.importc, header: "<windows.h>".} = object
  955. value* {.importc: "Value".}: array[6, BYTE]
  956. # https://docs.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-sid
  957. SID* {.importc, header: "<windows.h>".} = object
  958. Revision: BYTE
  959. SubAuthorityCount: BYTE
  960. IdentifierAuthority: SID_IDENTIFIER_AUTHORITY
  961. SubAuthority: ptr ptr DWORD
  962. PSID* = ptr SID
  963. const
  964. # https://docs.microsoft.com/en-us/windows/win32/secauthz/sid-components
  965. # https://github.com/mirror/mingw-w64/blob/84c950bdab7c999ace49fe8383856be77f88c4a8/mingw-w64-headers/include/winnt.h#L2994
  966. SECURITY_NT_AUTHORITY* = [BYTE(0), BYTE(0), BYTE(0), BYTE(0), BYTE(0), BYTE(5)]
  967. SECURITY_BUILTIN_DOMAIN_RID* = 32
  968. DOMAIN_ALIAS_RID_ADMINS* = 544
  969. proc allocateAndInitializeSid*(pIdentifierAuthority: ptr SID_IDENTIFIER_AUTHORITY,
  970. nSubAuthorityCount: BYTE,
  971. nSubAuthority0: DWORD,
  972. nSubAuthority1: DWORD,
  973. nSubAuthority2: DWORD,
  974. nSubAuthority3: DWORD,
  975. nSubAuthority4: DWORD,
  976. nSubAuthority5: DWORD,
  977. nSubAuthority6: DWORD,
  978. nSubAuthority7: DWORD,
  979. pSid: ptr PSID): WINBOOL
  980. {.stdcall, dynlib: "Advapi32", importc: "AllocateAndInitializeSid".}
  981. proc checkTokenMembership*(tokenHandle: Handle, sidToCheck: PSID,
  982. isMember: PBOOL): WINBOOL
  983. {.stdcall, dynlib: "Advapi32", importc: "CheckTokenMembership".}
  984. proc freeSid*(pSid: PSID): PSID
  985. {.stdcall, dynlib: "Advapi32", importc: "FreeSid".}
  986. when defined(nimHasStyleChecks):
  987. {.pop.} # {.push styleChecks: off.}