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