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 |