URLDecomposition.cpp [plain text]
#include "config.h"
#include "URLDecomposition.h"
#include "SecurityOrigin.h"
namespace WebCore {
String URLDecomposition::origin() const
{
return SecurityOrigin::create(fullURL())->toString();
}
String URLDecomposition::protocol() const
{
auto fullURL = this->fullURL();
if (WTF::protocolIsJavaScript(fullURL.string()))
return "javascript:"_s;
return makeString(fullURL.protocol(), ':');
}
void URLDecomposition::setProtocol(StringView value)
{
URL copy = fullURL();
copy.setProtocol(value);
setFullURL(copy);
}
String URLDecomposition::username() const
{
return fullURL().encodedUser().toString();
}
void URLDecomposition::setUsername(StringView user)
{
auto fullURL = this->fullURL();
if (fullURL.host().isEmpty() || fullURL.cannotBeABaseURL() || fullURL.protocolIs("file"))
return;
fullURL.setUser(user);
setFullURL(fullURL);
}
String URLDecomposition::password() const
{
return fullURL().encodedPassword().toString();
}
void URLDecomposition::setPassword(StringView password)
{
auto fullURL = this->fullURL();
if (fullURL.host().isEmpty() || fullURL.cannotBeABaseURL() || fullURL.protocolIs("file"))
return;
fullURL.setPassword(password);
setFullURL(fullURL);
}
String URLDecomposition::host() const
{
return fullURL().hostAndPort();
}
static unsigned countASCIIDigits(StringView string)
{
unsigned length = string.length();
for (unsigned count = 0; count < length; ++count) {
if (!isASCIIDigit(string[count]))
return count;
}
return length;
}
void URLDecomposition::setHost(StringView value)
{
auto fullURL = this->fullURL();
if (value.isEmpty() && !fullURL.protocolIs("file"))
return;
size_t separator = value.reverseFind(':');
if (!separator)
return;
if (fullURL.cannotBeABaseURL() || !fullURL.canSetHostOrPort())
return;
size_t ipv6Separator = value.reverseFind(']');
if (separator == notFound || (ipv6Separator != notFound && ipv6Separator > separator))
fullURL.setHost(value);
else {
if (value.find(':') != separator && ipv6Separator == notFound)
return;
unsigned portLength = countASCIIDigits(value.substring(separator + 1));
if (!portLength) {
fullURL.setHost(value.substring(0, separator));
} else {
auto portNumber = parseUInt16(value.substring(separator + 1, portLength));
if (portNumber && WTF::isDefaultPortForProtocol(*portNumber, fullURL.protocol()))
fullURL.setHostAndPort(value.substring(0, separator));
else
fullURL.setHostAndPort(value.substring(0, separator + 1 + portLength));
}
}
if (fullURL.isValid())
setFullURL(fullURL);
}
String URLDecomposition::hostname() const
{
return fullURL().host().toString();
}
static StringView removeAllLeadingSolidusCharacters(StringView string)
{
unsigned i;
unsigned length = string.length();
for (i = 0; i < length; ++i) {
if (string[i] != '/')
break;
}
return string.substring(i);
}
void URLDecomposition::setHostname(StringView value)
{
auto fullURL = this->fullURL();
auto host = removeAllLeadingSolidusCharacters(value);
if (host.isEmpty() && !fullURL.protocolIs("file"))
return;
if (fullURL.cannotBeABaseURL() || !fullURL.canSetHostOrPort())
return;
fullURL.setHost(host);
if (fullURL.isValid())
setFullURL(fullURL);
}
String URLDecomposition::port() const
{
auto port = fullURL().port();
if (!port)
return emptyString();
return String::number(*port);
}
static Optional<Optional<uint16_t>> parsePort(StringView string, StringView protocol)
{
auto digitsOnly = string.left(countASCIIDigits(string));
if (digitsOnly.isEmpty())
return Optional<uint16_t> { WTF::nullopt };
auto port = parseUInt16(digitsOnly);
if (!port)
return WTF::nullopt;
if (WTF::isDefaultPortForProtocol(*port, protocol))
return Optional<uint16_t> { WTF::nullopt };
return { { *port } };
}
void URLDecomposition::setPort(StringView value)
{
auto fullURL = this->fullURL();
if (fullURL.host().isEmpty() || fullURL.cannotBeABaseURL() || fullURL.protocolIs("file") || !fullURL.canSetHostOrPort())
return;
auto port = parsePort(value, fullURL.protocol());
if (!port)
return;
fullURL.setPort(*port);
setFullURL(fullURL);
}
String URLDecomposition::pathname() const
{
return fullURL().path().toString();
}
void URLDecomposition::setPathname(StringView value)
{
auto fullURL = this->fullURL();
if (fullURL.cannotBeABaseURL() || !fullURL.canSetPathname())
return;
if (value.startsWith('/'))
fullURL.setPath(value);
else
fullURL.setPath(makeString('/', value));
setFullURL(fullURL);
}
String URLDecomposition::search() const
{
auto fullURL = this->fullURL();
return fullURL.query().isEmpty() ? emptyString() : fullURL.queryWithLeadingQuestionMark().toString();
}
void URLDecomposition::setSearch(const String& value)
{
auto fullURL = this->fullURL();
if (value.isEmpty()) {
fullURL.setQuery({ });
} else {
String newSearch = value;
fullURL.setQuery(newSearch.replaceWithLiteral('#', "%23"));
}
setFullURL(fullURL);
}
String URLDecomposition::hash() const
{
auto fullURL = this->fullURL();
return fullURL.fragmentIdentifier().isEmpty() ? emptyAtom() : fullURL.fragmentIdentifierWithLeadingNumberSign().toAtomString();
}
void URLDecomposition::setHash(StringView value)
{
auto fullURL = this->fullURL();
if (value.isEmpty())
fullURL.removeFragmentIdentifier();
else
fullURL.setFragmentIdentifier(value.startsWith('#') ? value.substring(1) : value);
setFullURL(fullURL);
}
}