| OLD | NEW |
| (Empty) |
| 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 #ifndef CHROME_FRAME_HTTP_NEGOTIATE_H_ | |
| 6 #define CHROME_FRAME_HTTP_NEGOTIATE_H_ | |
| 7 | |
| 8 #include <shdeprecated.h> | |
| 9 #include <urlmon.h> | |
| 10 | |
| 11 #include <string> | |
| 12 | |
| 13 #include "base/basictypes.h" | |
| 14 #include "base/win/scoped_comptr.h" | |
| 15 | |
| 16 // Typedefs for IHttpNegotiate methods. | |
| 17 typedef HRESULT (STDMETHODCALLTYPE* IHttpNegotiate_BeginningTransaction_Fn)( | |
| 18 IHttpNegotiate* me, LPCWSTR url, LPCWSTR headers, DWORD reserved, | |
| 19 LPWSTR* additional_headers); | |
| 20 typedef HRESULT (STDMETHODCALLTYPE* IHttpNegotiate_OnResponse_Fn)( | |
| 21 IHttpNegotiate* me, DWORD response_code, LPCWSTR response_header, | |
| 22 LPCWSTR request_header, LPWSTR* additional_request_headers); | |
| 23 | |
| 24 // Typedefs for IBindStatusCallback methods. | |
| 25 typedef HRESULT (STDMETHODCALLTYPE* IBindStatusCallback_StartBinding_Fn)( | |
| 26 IBindStatusCallback* me, DWORD reserved, IBinding *binding); | |
| 27 | |
| 28 // Typedefs for IInternetProtocolSink methods. | |
| 29 typedef HRESULT (STDMETHODCALLTYPE* IInternetProtocolSink_ReportProgress_Fn)( | |
| 30 IInternetProtocolSink* me, ULONG status_code, LPCWSTR status_text); | |
| 31 | |
| 32 // Patches methods of urlmon's IHttpNegotiate implementation for the purposes | |
| 33 // of adding to the http user agent header. | |
| 34 | |
| 35 // Also patches one of the IBindStatusCallback implementations in urlmon to pick | |
| 36 // up an IBinding during the StartBinding call. The IBinding implementor then | |
| 37 // gets a patch applied to its IInternetProtocolSink's ReportProgress method. | |
| 38 // The patched is there so that the reporting of the MIME type to the IBinding | |
| 39 // implementor can be changed if an X-Chrome-Frame HTTP header is present | |
| 40 // in the response headers. If anyone can suggest a more straightforward way of | |
| 41 // doing this, I would be eternally grateful. | |
| 42 class HttpNegotiatePatch { | |
| 43 // class is not to be instantiated atm. | |
| 44 HttpNegotiatePatch(); | |
| 45 ~HttpNegotiatePatch(); | |
| 46 | |
| 47 public: | |
| 48 static bool Initialize(); | |
| 49 static void Uninitialize(); | |
| 50 | |
| 51 // Enables and disables the User Agent header modification. It is enabled | |
| 52 // by default when the patch is installed. | |
| 53 static void set_modify_user_agent(bool value) { modify_user_agent_ = value; } | |
| 54 | |
| 55 // IHttpNegotiate patch methods | |
| 56 static STDMETHODIMP BeginningTransaction( | |
| 57 IHttpNegotiate_BeginningTransaction_Fn original, IHttpNegotiate* me, | |
| 58 LPCWSTR url, LPCWSTR headers, DWORD reserved, LPWSTR* additional_headers); | |
| 59 | |
| 60 protected: | |
| 61 static HRESULT PatchHttpNegotiate(IUnknown* to_patch); | |
| 62 | |
| 63 private: | |
| 64 static bool modify_user_agent_; | |
| 65 | |
| 66 DISALLOW_COPY_AND_ASSIGN(HttpNegotiatePatch); | |
| 67 }; | |
| 68 | |
| 69 // From the latest urlmon.h. Symbol name prepended with LOCAL_ to | |
| 70 // avoid conflict (and therefore build errors) for those building with | |
| 71 // a newer Windows SDK. | |
| 72 // TODO(robertshield): Remove this once we update our SDK version. | |
| 73 extern const int LOCAL_BINDSTATUS_SERVER_MIMETYPEAVAILABLE; | |
| 74 | |
| 75 // Scans |additional_headers| and |headers| for User-Agent header or grabs | |
| 76 // the default User-Agent if none is found. Append "chromeframe" at the end | |
| 77 // of the string. Returns the original content of |additional_headers| but | |
| 78 // with the new User-Agent header. Somewhat unusual interface is dictated | |
| 79 // because we use it in IHttpNegotiate::BeginningTransaction. | |
| 80 // The function is a public since we want to use it from | |
| 81 // UrlmonUrlRequest::BeginningTransaction for the unusual but yet possible case | |
| 82 // when |headers| contains an User-Agent string. | |
| 83 std::string AppendCFUserAgentString(LPCWSTR headers, | |
| 84 LPCWSTR additional_headers); | |
| 85 | |
| 86 // Adds or replaces the User-Agent header in a set of HTTP headers. | |
| 87 // Arguments are the same as with AppendCFUserAgentString. | |
| 88 std::string ReplaceOrAddUserAgent(LPCWSTR headers, | |
| 89 const std::string& user_agent_value); | |
| 90 | |
| 91 #endif // CHROME_FRAME_HTTP_NEGOTIATE_H_ | |
| OLD | NEW |