1 // Copyright (c) 2006-2008 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 <winsock2.h> 6 7 #include "net/base/winsock_init.h" 8 9 #include "base/logging.h" 10 #include "base/singleton.h" 11 12 namespace { 13 14 class WinsockInitSingleton { 15 public: WinsockInitSingleton()16 WinsockInitSingleton() : did_init_(false) { 17 WORD winsock_ver = MAKEWORD(2, 2); 18 WSAData wsa_data; 19 did_init_ = (WSAStartup(winsock_ver, &wsa_data) == 0); 20 if (did_init_) { 21 DCHECK(wsa_data.wVersion == winsock_ver); 22 23 // The first time WSAGetLastError is called, the delay load helper will 24 // resolve the address with GetProcAddress and fixup the import. If a 25 // third party application hooks system functions without correctly 26 // restoring the error code, it is possible that the error code will be 27 // overwritten during delay load resolution. The result of the first 28 // call may be incorrect, so make sure the function is bound and future 29 // results will be correct. 30 WSAGetLastError(); 31 } 32 } 33 ~WinsockInitSingleton()34 ~WinsockInitSingleton() { 35 if (did_init_) 36 WSACleanup(); 37 } 38 39 private: 40 bool did_init_; 41 }; 42 43 } // namespace 44 45 namespace net { 46 EnsureWinsockInit()47void EnsureWinsockInit() { 48 Singleton<WinsockInitSingleton>::get(); 49 } 50 51 } // namespace net 52