1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "net/base/winsock_util.h"
6
7 #include "base/logging.h"
8 #include "net/base/net_errors.h"
9
10 namespace net {
11
12 namespace {
13
14 // Prevent the compiler from optimizing away the arguments so they appear
15 // nicely on the stack in crash dumps.
16 #pragma warning (disable: 4748)
17 #pragma optimize( "", off )
18
19 // Pass the important values as function arguments so that they are available
20 // in crash dumps.
CheckEventWait(WSAEVENT hEvent,DWORD wait_rv,DWORD expected)21 void CheckEventWait(WSAEVENT hEvent, DWORD wait_rv, DWORD expected) {
22 if (wait_rv != expected) {
23 DWORD err = ERROR_SUCCESS;
24 if (wait_rv == WAIT_FAILED)
25 err = GetLastError();
26 CHECK(false); // Crash.
27 }
28 }
29
30 #pragma optimize( "", on )
31 #pragma warning (default: 4748)
32
33 } // namespace
34
AssertEventNotSignaled(WSAEVENT hEvent)35 void AssertEventNotSignaled(WSAEVENT hEvent) {
36 DWORD wait_rv = WaitForSingleObject(hEvent, 0);
37 CheckEventWait(hEvent, wait_rv, WAIT_TIMEOUT);
38 }
39
ResetEventIfSignaled(WSAEVENT hEvent)40 bool ResetEventIfSignaled(WSAEVENT hEvent) {
41 // TODO(wtc): Remove the CHECKs after enough testing.
42 DWORD wait_rv = WaitForSingleObject(hEvent, 0);
43 if (wait_rv == WAIT_TIMEOUT)
44 return false; // The event object is not signaled.
45 CheckEventWait(hEvent, wait_rv, WAIT_OBJECT_0);
46 BOOL ok = WSAResetEvent(hEvent);
47 CHECK(ok);
48 return true;
49 }
50
51 } // namespace net
52