15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <winsock2.h> 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/winsock_init.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/lazy_instance.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class WinsockInitSingleton { 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WinsockInitSingleton() { 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WORD winsock_ver = MAKEWORD(2, 2); 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WSAData wsa_data; 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool did_init = (WSAStartup(winsock_ver, &wsa_data) == 0); 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (did_init) { 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(wsa_data.wVersion == winsock_ver); 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The first time WSAGetLastError is called, the delay load helper will 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // resolve the address with GetProcAddress and fixup the import. If a 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // third party application hooks system functions without correctly 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // restoring the error code, it is possible that the error code will be 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // overwritten during delay load resolution. The result of the first 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // call may be incorrect, so make sure the function is bound and future 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // results will be correct. 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WSAGetLastError(); 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ~WinsockInitSingleton() { 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Don't call WSACleanup() since the worker pool threads can continue to 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // call getaddrinfo() after Winsock has shutdown, which can lead to crashes. 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static base::LazyInstance<WinsockInitSingleton> g_winsock_init_singleton = 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LAZY_INSTANCE_INITIALIZER; 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace net { 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void EnsureWinsockInit() { 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) g_winsock_init_singleton.Get(); 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace net 52