/* * Copyright (c) 2018-2020, Andreas Kling * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * 1. Redistributions of source code must retain the above copyright notice, this * list of conditions and the following disclaimer. * * 2. Redistributions in binary form must reproduce the above copyright notice, * this list of conditions and the following disclaimer in the documentation * and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ #include #include #include namespace AK { static inline bool is_valid_protocol_character(char ch) { return ch >= 'a' && ch <= 'z'; } static inline bool is_valid_hostname_character(char ch) { return ch && ch != '/' && ch != ':'; } static inline bool is_digit(char ch) { return ch >= '0' && ch <= '9'; } bool URL::parse(const StringView& string) { enum class State { InProtocol, InHostname, InPort, InPath, InQuery, InFragment, }; Vector buffer; State state { State::InProtocol }; size_t index = 0; auto peek = [&] { if (index >= string.length()) return '\0'; return string[index]; }; auto consume = [&] { if (index >= string.length()) return '\0'; return string[index++]; }; while (index < string.length()) { switch (state) { case State::InProtocol: if (is_valid_protocol_character(peek())) { buffer.append(consume()); continue; } if (consume() != ':') return false; if (consume() != '/') return false; if (consume() != '/') return false; if (buffer.is_empty()) return false; m_protocol = String::copy(buffer); if (m_protocol == "http") m_port = 80; else if (m_protocol == "https") m_port = 443; buffer.clear(); if (m_protocol == "file") state = State::InPath; else state = State::InHostname; continue; case State::InHostname: if (is_valid_hostname_character(peek())) { buffer.append(consume()); continue; } if (buffer.is_empty()) return false; m_host = String::copy(buffer); buffer.clear(); if (peek() == ':') { consume(); state = State::InPort; continue; } if (peek() == '/') { state = State::InPath; continue; } return false; case State::InPort: if (is_digit(peek())) { buffer.append(consume()); continue; } if (buffer.is_empty()) return false; { bool ok; m_port = String::copy(buffer).to_uint(ok); buffer.clear(); if (!ok) return false; } if (peek() == '/') { state = State::InPath; continue; } return false; case State::InPath: if (peek() == '?' || peek() == '#') { m_path = String::copy(buffer); buffer.clear(); state = peek() == '?' ? State::InQuery : State::InFragment; consume(); continue; } buffer.append(consume()); continue; case State::InQuery: if (peek() == '#') { m_query = String::copy(buffer); buffer.clear(); consume(); state = State::InFragment; continue; } buffer.append(consume()); continue; case State::InFragment: buffer.append(consume()); continue; } } if (state == State::InHostname) { // We're still in the hostname, so e.g "http://serenityos.org" if (buffer.is_empty()) return false; m_host = String::copy(buffer); m_path = "/"; } if (state == State::InProtocol) return false; if (state == State::InPath) m_path = String::copy(buffer); if (state == State::InQuery) m_query = String::copy(buffer); if (state == State::InFragment) m_fragment = String::copy(buffer); if (m_query.is_null()) m_query = ""; if (m_fragment.is_null()) m_fragment = ""; return true; } URL::URL(const StringView& string) { m_valid = parse(string); } String URL::to_string() const { StringBuilder builder; builder.append(m_protocol); builder.append("://"); if (protocol() != "file") { builder.append(m_host); if (!(protocol() == "http" && port() == 80) && !(protocol() == "https" && port() == 443)) { builder.append(':'); builder.append(String::number(m_port)); } } builder.append(m_path); if (!m_query.is_empty()) { builder.append('?'); builder.append(m_query); } if (!m_fragment.is_empty()) { builder.append('#'); builder.append(m_fragment); } return builder.to_string(); } URL URL::complete_url(const String& string) const { URL url(string); if (url.is_valid()) return url; if (string.starts_with("/")) { url = *this; url.set_path(string); return url; } StringBuilder builder; FileSystemPath fspath(path()); builder.append('/'); bool document_url_ends_in_slash = path()[path().length() - 1] == '/'; for (size_t i = 0; i < fspath.parts().size(); ++i) { if (i == fspath.parts().size() - 1 && !document_url_ends_in_slash) break; builder.append(fspath.parts()[i]); builder.append('/'); } builder.append(string); auto built = builder.to_string(); fspath = FileSystemPath(built); url = *this; url.set_path(fspath.string()); return url; } void URL::set_protocol(const String& protocol) { m_protocol = protocol; m_valid = compute_validity(); } void URL::set_host(const String& host) { m_host = host; m_valid = compute_validity(); } void URL::set_path(const String& path) { m_path = path; m_valid = compute_validity(); } void URL::set_query(const String& query) { m_query = query; } void URL::set_fragment(const String& fragment) { m_fragment = fragment; } bool URL::compute_validity() const { // FIXME: This is by no means complete. if (m_protocol.is_empty()) return false; if (m_protocol == "file") { if (m_path.is_empty()) return false; } else { if (m_host.is_empty()) return false; } return true; } URL URL::create_with_file_protocol(const String& path) { URL url; url.set_protocol("file"); url.set_path(path); return url; } }