The Tor Browser / file comparison
comparison: netwerk/protocol/viewsource/nsViewSourceHandler.cpp
netwerk/protocol/viewsource/nsViewSourceHandler.cpp
- changeset 2
- 7e26c7da4463
equal
deleted
inserted
replaced
| -1:000000000000 | 0:72d41f672582 |
|---|---|
| 1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ | |
| 2 /* vim:set ts=4 sw=4 sts=4 et: */ | |
| 3 /* This Source Code Form is subject to the terms of the Mozilla Public | |
| 4 * License, v. 2.0. If a copy of the MPL was not distributed with this | |
| 5 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ | |
| 6 | |
| 7 #include "nsViewSourceHandler.h" | |
| 8 #include "nsViewSourceChannel.h" | |
| 9 #include "nsNetUtil.h" | |
| 10 #include "nsSimpleNestedURI.h" | |
| 11 | |
| 12 #define VIEW_SOURCE "view-source" | |
| 13 | |
| 14 //////////////////////////////////////////////////////////////////////////////// | |
| 15 | |
| 16 NS_IMPL_ISUPPORTS(nsViewSourceHandler, nsIProtocolHandler) | |
| 17 | |
| 18 //////////////////////////////////////////////////////////////////////////////// | |
| 19 // nsIProtocolHandler methods: | |
| 20 | |
| 21 NS_IMETHODIMP | |
| 22 nsViewSourceHandler::GetScheme(nsACString &result) | |
| 23 { | |
| 24 result.AssignLiteral(VIEW_SOURCE); | |
| 25 return NS_OK; | |
| 26 } | |
| 27 | |
| 28 NS_IMETHODIMP | |
| 29 nsViewSourceHandler::GetDefaultPort(int32_t *result) | |
| 30 { | |
| 31 *result = -1; | |
| 32 return NS_OK; | |
| 33 } | |
| 34 | |
| 35 NS_IMETHODIMP | |
| 36 nsViewSourceHandler::GetProtocolFlags(uint32_t *result) | |
| 37 { | |
| 38 *result = URI_NORELATIVE | URI_NOAUTH | URI_LOADABLE_BY_ANYONE | | |
| 39 URI_NON_PERSISTABLE; | |
| 40 return NS_OK; | |
| 41 } | |
| 42 | |
| 43 NS_IMETHODIMP | |
| 44 nsViewSourceHandler::NewURI(const nsACString &aSpec, | |
| 45 const char *aCharset, | |
| 46 nsIURI *aBaseURI, | |
| 47 nsIURI **aResult) | |
| 48 { | |
| 49 *aResult = nullptr; | |
| 50 | |
| 51 // Extract inner URL and normalize to ASCII. This is done to properly | |
| 52 // support IDN in cases like "view-source:http://www.szalagavató.hu/" | |
| 53 | |
| 54 int32_t colon = aSpec.FindChar(':'); | |
| 55 if (colon == kNotFound) | |
| 56 return NS_ERROR_MALFORMED_URI; | |
| 57 | |
| 58 nsCOMPtr<nsIURI> innerURI; | |
| 59 nsresult rv = NS_NewURI(getter_AddRefs(innerURI), | |
| 60 Substring(aSpec, colon + 1), aCharset, aBaseURI); | |
| 61 if (NS_FAILED(rv)) | |
| 62 return rv; | |
| 63 | |
| 64 nsAutoCString asciiSpec; | |
| 65 rv = innerURI->GetAsciiSpec(asciiSpec); | |
| 66 if (NS_FAILED(rv)) | |
| 67 return rv; | |
| 68 | |
| 69 // put back our scheme and construct a simple-uri wrapper | |
| 70 | |
| 71 asciiSpec.Insert(VIEW_SOURCE ":", 0); | |
| 72 | |
| 73 // We can't swap() from an nsRefPtr<nsSimpleNestedURI> to an nsIURI**, | |
| 74 // sadly. | |
| 75 nsSimpleNestedURI* ourURI = new nsSimpleNestedURI(innerURI); | |
| 76 nsCOMPtr<nsIURI> uri = ourURI; | |
| 77 if (!uri) | |
| 78 return NS_ERROR_OUT_OF_MEMORY; | |
| 79 | |
| 80 rv = ourURI->SetSpec(asciiSpec); | |
| 81 if (NS_FAILED(rv)) | |
| 82 return rv; | |
| 83 | |
| 84 // Make the URI immutable so it's impossible to get it out of sync | |
| 85 // with its inner URI. | |
| 86 ourURI->SetMutable(false); | |
| 87 | |
| 88 uri.swap(*aResult); | |
| 89 return rv; | |
| 90 } | |
| 91 | |
| 92 NS_IMETHODIMP | |
| 93 nsViewSourceHandler::NewChannel(nsIURI* uri, nsIChannel* *result) | |
| 94 { | |
| 95 NS_ENSURE_ARG_POINTER(uri); | |
| 96 nsViewSourceChannel *channel = new nsViewSourceChannel(); | |
| 97 if (!channel) | |
| 98 return NS_ERROR_OUT_OF_MEMORY; | |
| 99 NS_ADDREF(channel); | |
| 100 | |
| 101 nsresult rv = channel->Init(uri); | |
| 102 if (NS_FAILED(rv)) { | |
| 103 NS_RELEASE(channel); | |
| 104 return rv; | |
| 105 } | |
| 106 | |
| 107 *result = static_cast<nsIViewSourceChannel*>(channel); | |
| 108 return NS_OK; | |
| 109 } | |
| 110 | |
| 111 nsresult | |
| 112 nsViewSourceHandler::NewSrcdocChannel(nsIURI* uri, const nsAString &srcdoc, | |
| 113 nsIURI* baseURI, nsIChannel* *result) | |
| 114 { | |
| 115 NS_ENSURE_ARG_POINTER(uri); | |
| 116 nsViewSourceChannel *channel = new nsViewSourceChannel(); | |
| 117 if (!channel) | |
| 118 return NS_ERROR_OUT_OF_MEMORY; | |
| 119 NS_ADDREF(channel); | |
| 120 | |
| 121 nsresult rv = channel->InitSrcdoc(uri, srcdoc, baseURI); | |
| 122 if (NS_FAILED(rv)) { | |
| 123 NS_RELEASE(channel); | |
| 124 return rv; | |
| 125 } | |
| 126 | |
| 127 *result = static_cast<nsIViewSourceChannel*>(channel); | |
| 128 return NS_OK; | |
| 129 } | |
| 130 | |
| 131 NS_IMETHODIMP | |
| 132 nsViewSourceHandler::AllowPort(int32_t port, const char *scheme, bool *_retval) | |
| 133 { | |
| 134 // don't override anything. | |
| 135 *_retval = false; | |
| 136 return NS_OK; | |
| 137 } | |
| 138 | |
| 139 nsViewSourceHandler::nsViewSourceHandler() | |
| 140 { | |
| 141 gInstance = this; | |
| 142 } | |
| 143 | |
| 144 nsViewSourceHandler::~nsViewSourceHandler() | |
| 145 { | |
| 146 gInstance = nullptr; | |
| 147 } | |
| 148 | |
| 149 nsViewSourceHandler* nsViewSourceHandler::gInstance = nullptr; | |
| 150 | |
| 151 nsViewSourceHandler* | |
| 152 nsViewSourceHandler::GetInstance() | |
| 153 { | |
| 154 return gInstance; | |
| 155 } |
