transmission/tests/libtransmission/web-utils-test.cc

250 lines
8.3 KiB
C++

// This file Copyright (C) 2013-2022 Mnemosyne LLC.
// It may be used under GPLv2 (SPDX: GPL-2.0-only), GPLv3 (SPDX: GPL-3.0-only),
// or any future license endorsed by Mnemosyne LLC.
// License text can be found in the licenses/ folder.
#include <array>
#include <cstddef> // size_t
#include <string_view>
#include <utility>
#include <vector>
#ifdef _WIN32
#include <windows.h>
#define setenv(key, value, unused) SetEnvironmentVariableA(key, value)
#define unsetenv(key) SetEnvironmentVariableA(key, nullptr)
#endif
#include <libtransmission/crypto-utils.h>
#include <libtransmission/web-utils.h>
#include "gtest/gtest.h"
using namespace std::literals;
using WebUtilsTest = ::testing::Test;
using namespace std::literals;
TEST_F(WebUtilsTest, urlParse)
{
auto url = "http://1"sv;
auto parsed = tr_urlParse(url);
EXPECT_TRUE(parsed);
EXPECT_EQ("http"sv, parsed->scheme);
EXPECT_EQ("1"sv, parsed->host);
EXPECT_EQ("1"sv, parsed->sitename);
EXPECT_EQ(""sv, parsed->path);
EXPECT_EQ(""sv, parsed->query);
EXPECT_EQ(""sv, parsed->fragment);
EXPECT_EQ(80, parsed->port);
url = "http://www.some-tracker.org/some/path"sv;
parsed = tr_urlParse(url);
EXPECT_TRUE(parsed);
EXPECT_EQ("http"sv, parsed->scheme);
EXPECT_EQ("www.some-tracker.org"sv, parsed->host);
EXPECT_EQ("some-tracker"sv, parsed->sitename);
EXPECT_EQ("/some/path"sv, parsed->path);
EXPECT_EQ(""sv, parsed->query);
EXPECT_EQ(""sv, parsed->fragment);
EXPECT_EQ(80, parsed->port);
url = "http://www.some-tracker.org:8080/some/path"sv;
parsed = tr_urlParse(url);
EXPECT_TRUE(parsed);
EXPECT_EQ("http"sv, parsed->scheme);
EXPECT_EQ("www.some-tracker.org"sv, parsed->host);
EXPECT_EQ("some-tracker"sv, parsed->sitename);
EXPECT_EQ("/some/path"sv, parsed->path);
EXPECT_EQ(""sv, parsed->query);
EXPECT_EQ(""sv, parsed->fragment);
EXPECT_EQ(8080, parsed->port);
url = "http://www.some-tracker.org:8080/some/path?key=val&foo=bar#fragment"sv;
parsed = tr_urlParse(url);
EXPECT_TRUE(parsed);
EXPECT_EQ("http"sv, parsed->scheme);
EXPECT_EQ("www.some-tracker.org"sv, parsed->host);
EXPECT_EQ("some-tracker"sv, parsed->sitename);
EXPECT_EQ("/some/path"sv, parsed->path);
EXPECT_EQ("key=val&foo=bar"sv, parsed->query);
EXPECT_EQ("fragment"sv, parsed->fragment);
EXPECT_EQ(8080, parsed->port);
url =
"magnet:"
"?xt=urn:btih:14ffe5dd23188fd5cb53a1d47f1289db70abf31e"
"&dn=ubuntu_12_04_1_desktop_32_bit"
"&tr=http%3A%2F%2Ftracker.publicbt.com%2Fannounce"
"&tr=udp%3A%2F%2Ftracker.publicbt.com%3A80"
"&ws=http%3A%2F%2Ftransmissionbt.com"sv;
parsed = tr_urlParse(url);
EXPECT_TRUE(parsed);
EXPECT_EQ("magnet"sv, parsed->scheme);
EXPECT_EQ(""sv, parsed->host);
EXPECT_EQ(""sv, parsed->sitename);
EXPECT_EQ(""sv, parsed->path);
EXPECT_EQ(
"xt=urn:btih:14ffe5dd23188fd5cb53a1d47f1289db70abf31e"
"&dn=ubuntu_12_04_1_desktop_32_bit"
"&tr=http%3A%2F%2Ftracker.publicbt.com%2Fannounce"
"&tr=udp%3A%2F%2Ftracker.publicbt.com%3A80"
"&ws=http%3A%2F%2Ftransmissionbt.com"sv,
parsed->query);
// test a host whose public suffix contains >1 dot
url = "https://www.example.co.uk:8080/some/path"sv;
parsed = tr_urlParse(url);
EXPECT_TRUE(parsed);
EXPECT_EQ("https"sv, parsed->scheme);
EXPECT_EQ("example"sv, parsed->sitename);
EXPECT_EQ("www.example.co.uk"sv, parsed->host);
EXPECT_EQ("/some/path"sv, parsed->path);
EXPECT_EQ(8080, parsed->port);
// test a host that lacks a subdomain
url = "http://some-tracker.co.uk/some/other/path"sv;
parsed = tr_urlParse(url);
EXPECT_TRUE(parsed);
EXPECT_EQ("http"sv, parsed->scheme);
EXPECT_EQ("some-tracker"sv, parsed->sitename);
EXPECT_EQ("some-tracker.co.uk"sv, parsed->host);
EXPECT_EQ("/some/other/path"sv, parsed->path);
EXPECT_EQ(80, parsed->port);
// test a host with an IPv4 address
url = "https://127.0.0.1:8080/some/path"sv;
parsed = tr_urlParse(url);
EXPECT_TRUE(parsed);
EXPECT_EQ("https"sv, parsed->scheme);
EXPECT_EQ("127.0.0.1"sv, parsed->sitename);
EXPECT_EQ("127.0.0.1"sv, parsed->host);
EXPECT_EQ("/some/path"sv, parsed->path);
EXPECT_EQ(8080, parsed->port);
// test a host with a bracketed IPv6 address and explicit port
url = "http://[2001:0db8:11a3:09d7:1f34:8a2e:07a0:765d]:8080/announce"sv;
parsed = tr_urlParse(url);
EXPECT_EQ("http"sv, parsed->scheme);
EXPECT_EQ("[2001:0db8:11a3:09d7:1f34:8a2e:07a0:765d]"sv, parsed->sitename);
EXPECT_EQ("[2001:0db8:11a3:09d7:1f34:8a2e:07a0:765d]"sv, parsed->host);
EXPECT_EQ("/announce"sv, parsed->path);
EXPECT_EQ(8080, parsed->port);
// test a host with a bracketed IPv6 address and implicit port
url = "http://[2001:0db8:11a3:09d7:1f34:8a2e:07a0:765d]/announce"sv;
parsed = tr_urlParse(url);
EXPECT_EQ("http"sv, parsed->scheme);
EXPECT_EQ("[2001:0db8:11a3:09d7:1f34:8a2e:07a0:765d]"sv, parsed->sitename);
EXPECT_EQ("[2001:0db8:11a3:09d7:1f34:8a2e:07a0:765d]"sv, parsed->host);
EXPECT_EQ("/announce"sv, parsed->path);
EXPECT_EQ(80, parsed->port);
// test a host with an unbracketed IPv6 address and implicit port
url = "http://2001:0db8:11a3:09d7:1f34:8a2e:07a0:765d/announce"sv;
parsed = tr_urlParse(url);
EXPECT_EQ("http"sv, parsed->scheme);
EXPECT_EQ("2001:0db8:11a3:09d7:1f34:8a2e:07a0:765d"sv, parsed->sitename);
EXPECT_EQ("2001:0db8:11a3:09d7:1f34:8a2e:07a0:765d"sv, parsed->host);
EXPECT_EQ("/announce"sv, parsed->path);
EXPECT_EQ(80, parsed->port);
}
TEST(WebUtilsTest, urlParseFuzz)
{
auto buf = std::vector<char>{};
for (size_t i = 0; i < 100000U; ++i)
{
buf.resize(tr_rand_int(1024U));
tr_rand_buffer(std::data(buf), std::size(buf));
(void)tr_urlParse({ std::data(buf), std::size(buf) });
}
}
TEST_F(WebUtilsTest, urlNextQueryPair)
{
auto constexpr Query = "a=1&b=two&c=si&d_has_no_val&e=&f&g=gee"sv;
auto const query_view = tr_url_query_view{ Query };
auto const end = std::end(query_view);
auto it = std::begin(query_view);
EXPECT_NE(end, it);
EXPECT_EQ("a"sv, it->first);
EXPECT_EQ("1"sv, it->second);
++it;
EXPECT_NE(end, it);
EXPECT_EQ("b"sv, it->first);
EXPECT_EQ("two"sv, it->second);
++it;
EXPECT_NE(end, it);
EXPECT_EQ("c"sv, it->first);
EXPECT_EQ("si"sv, it->second);
++it;
EXPECT_NE(end, it);
EXPECT_EQ("d_has_no_val"sv, it->first);
EXPECT_EQ(""sv, it->second);
++it;
EXPECT_NE(end, it);
EXPECT_EQ("e"sv, it->first);
EXPECT_EQ(""sv, it->second);
++it;
EXPECT_NE(end, it);
EXPECT_EQ("f"sv, it->first);
EXPECT_EQ(""sv, it->second);
++it;
EXPECT_NE(end, it);
EXPECT_EQ("g"sv, it->first);
EXPECT_EQ("gee"sv, it->second);
++it;
EXPECT_EQ(end, it);
}
TEST_F(WebUtilsTest, urlIsValid)
{
EXPECT_FALSE(tr_urlIsValid("hello world"sv));
EXPECT_FALSE(tr_urlIsValid("http://www.💩.com/announce/"sv));
EXPECT_TRUE(tr_urlIsValid("http://www.example.com/announce/"sv));
EXPECT_FALSE(tr_urlIsValid(""sv));
EXPECT_FALSE(tr_urlIsValid("com"sv));
EXPECT_FALSE(tr_urlIsValid("www.example.com"sv));
EXPECT_FALSE(tr_urlIsValid("://www.example.com"sv));
EXPECT_FALSE(tr_urlIsValid("zzz://www.example.com"sv)); // syntactically valid, but unsupported scheme
EXPECT_TRUE(tr_urlIsValid("https://www.example.com"sv));
EXPECT_TRUE(tr_urlIsValid("sftp://www.example.com"sv));
EXPECT_FALSE(tr_urlIsValidTracker("sftp://www.example.com"sv)); // unsupported tracker scheme
}
TEST_F(WebUtilsTest, urlPercentDecode)
{
auto constexpr Tests = std::array<std::pair<std::string_view, std::string_view>, 13>{ {
{ "%-2"sv, "%-2"sv },
{ "%6 1"sv, "%6 1"sv },
{ "%6"sv, "%6"sv },
{ "%6%a"sv, "%6%a"sv },
{ "%61 "sv, "a "sv },
{ "%61"sv, "a"sv },
{ "%61a"sv, "aa"sv },
{ "%61b"sv, "ab"sv },
{ "%6a"sv, "j"sv },
{ "%FF"sv, "\xff"sv },
{ "%FF%00%ff"sv, "\xff\x00\xff"sv },
{ "%FG"sv, "%FG"sv },
{ "http%3A%2F%2Fwww.example.com%2F~user%2F%3Ftest%3D1%26test1%3D2"sv,
"http://www.example.com/~user/?test=1&test1=2"sv },
} };
for (auto const& [encoded, decoded] : Tests)
{
EXPECT_EQ(decoded, tr_urlPercentDecode(encoded));
}
}