1 /* Copyright (C) 2015 Wildfire Games.
2  *
3  * Permission is hereby granted, free of charge, to any person obtaining
4  * a copy of this software and associated documentation files (the
5  * "Software"), to deal in the Software without restriction, including
6  * without limitation the rights to use, copy, modify, merge, publish,
7  * distribute, sublicense, and/or sell copies of the Software, and to
8  * permit persons to whom the Software is furnished to do so, subject to
9  * the following conditions:
10  *
11  * The above copyright notice and this permission notice shall be included
12  * in all copies or substantial portions of the Software.
13  *
14  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
15  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
16  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
17  * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
18  * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
19  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
20  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
21  */
22 
23 #include "precompiled.h"
24 #include "lib/sysdep/os/win/wiocp.h"
25 
26 #include "lib/file/file.h"	// ERR::IO
27 #include "lib/sysdep/os/win/wutil.h"
28 
29 
AttachToCompletionPort(HANDLE hFile,HANDLE & hIOCP,ULONG_PTR key,DWORD numConcurrentThreads)30 void AttachToCompletionPort(HANDLE hFile, HANDLE& hIOCP, ULONG_PTR key, DWORD numConcurrentThreads)
31 {
32 	WinScopedPreserveLastError s;	// CreateIoCompletionPort
33 
34 	// (when called for the first time, ends up creating hIOCP)
35 	hIOCP = CreateIoCompletionPort(hFile, hIOCP, key, numConcurrentThreads);
36 	ENSURE(wutil_IsValidHandle(hIOCP));
37 }
38 
39 
PollCompletionPort(HANDLE hIOCP,DWORD timeout,DWORD & bytesTransferred,ULONG_PTR & key,OVERLAPPED * & ovl)40 Status PollCompletionPort(HANDLE hIOCP, DWORD timeout, DWORD& bytesTransferred, ULONG_PTR& key, OVERLAPPED*& ovl)
41 {
42 	if(hIOCP == 0)
43 		return ERR::INVALID_HANDLE;	// NOWARN (happens if called before the first Attach)
44 
45 	WinScopedPreserveLastError s;
46 
47 	bytesTransferred = 0;
48 	key = 0;
49 	ovl = 0;
50 	if(GetQueuedCompletionStatus(hIOCP, &bytesTransferred, &key, &ovl, timeout))
51 		return INFO::OK;
52 	const Status ret = StatusFromWin();
53 	if(ret == ERR::AGAIN || ret == ERR::ABORTED)	// avoid polluting last error
54 		SetLastError(0);
55 	return ret;	// NOWARN (let caller decide what to do)
56 }
57