mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-11-24 06:16:30 +03:00
53 lines
1.4 KiB
C++
53 lines
1.4 KiB
C++
|
// 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(push)
|
||
|
|
#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(pop)
|
||
|
|
|
||
|
|
} // 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
|