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(push) 17 #pragma warning (disable: 4748) 18 #pragma optimize( "", off ) 19 20 // Pass the important values as function arguments so that they are available 21 // in crash dumps. 22 void CheckEventWait(WSAEVENT hEvent, DWORD wait_rv, DWORD expected) { 23 if (wait_rv != expected) { 24 DWORD err = ERROR_SUCCESS; 25 if (wait_rv == WAIT_FAILED) 26 err = GetLastError(); 27 CHECK(false); // Crash. 28 } 29 } 30 31 #pragma optimize( "", on ) 32 #pragma warning(pop) 33 34 } // namespace 35 36 void AssertEventNotSignaled(WSAEVENT hEvent) { 37 DWORD wait_rv = WaitForSingleObject(hEvent, 0); 38 CheckEventWait(hEvent, wait_rv, WAIT_TIMEOUT); 39 } 40 41 bool ResetEventIfSignaled(WSAEVENT hEvent) { 42 // TODO(wtc): Remove the CHECKs after enough testing. 43 DWORD wait_rv = WaitForSingleObject(hEvent, 0); 44 if (wait_rv == WAIT_TIMEOUT) 45 return false; // The event object is not signaled. 46 CheckEventWait(hEvent, wait_rv, WAIT_OBJECT_0); 47 BOOL ok = WSAResetEvent(hEvent); 48 CHECK(ok); 49 return true; 50 } 51 52 } // namespace net 53
