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