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)#ifndef NET_PROXY_PROXY_CONFIG_SERVICE_WIN_H_ 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define NET_PROXY_PROXY_CONFIG_SERVICE_WIN_H_ 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <windows.h> 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <winhttp.h> 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <vector> 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/compiler_specific.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/gtest_prod_util.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/win/object_watcher.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/proxy/polling_proxy_config_service.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace net { 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Implementation of ProxyConfigService that retrieves the system proxy 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// settings. 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// It works by calling WinHttpGetIEProxyConfigForCurrentUser() to fetch the 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Internet Explorer proxy settings. 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// We use two different strategies to notice when the configuration has 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// changed: 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// (1) Watch the internet explorer settings registry keys for changes. When 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// one of the registry keys pertaining to proxy settings has changed, we 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// call WinHttpGetIEProxyConfigForCurrentUser() again to read the 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// configuration's new value. 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// (2) Do regular polling every 10 seconds during network activity to see if 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// WinHttpGetIEProxyConfigForCurrentUser() returns something different. 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Ideally strategy (1) should be sufficient to pick up all of the changes. 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// However we still do the regular polling as a precaution in case the 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// implementation details of WinHttpGetIEProxyConfigForCurrentUser() ever 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// change, or in case we got it wrong (and are not checking all possible 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// registry dependencies). 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class NET_EXPORT_PRIVATE ProxyConfigServiceWin 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : public PollingProxyConfigService, 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public base::win::ObjectWatcher::Delegate { 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ProxyConfigServiceWin(); 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~ProxyConfigServiceWin(); 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Overrides a function from PollingProxyConfigService. 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void AddObserver(Observer* observer) OVERRIDE; 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FRIEND_TEST_ALL_PREFIXES(ProxyConfigServiceWinTest, SetFromIEConfig); 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) class KeyEntry; 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) typedef std::vector<KeyEntry*> KeyEntryList; 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Registers change observers on the registry keys relating to proxy settings. 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void StartWatchingRegistryForChanges(); 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Creates a new KeyEntry and appends it to |keys_to_watch_|. If the key 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // fails to be created, it is not appended to the list and we return false. 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool AddKeyToWatchList(HKEY rootkey, const wchar_t* subkey); 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // ObjectWatcher::Delegate methods: 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This is called whenever one of the registry keys we are watching change. 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnObjectSignaled(HANDLE object) OVERRIDE; 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static void GetCurrentProxyConfig(ProxyConfig* config); 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Set |config| using the proxy configuration values of |ie_config|. 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static void SetFromIEConfig( 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ProxyConfig* config, 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const WINHTTP_CURRENT_USER_IE_PROXY_CONFIG& ie_config); 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) KeyEntryList keys_to_watch_; 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace net 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // NET_PROXY_PROXY_CONFIG_SERVICE_WIN_H_ 81