1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
#include "punycode.h"
#include <library/cpp/testing/unittest/registar.h>
#include <util/charset/wide.h>
namespace {
template<typename T1, typename T2>
inline bool HasSameBuffer(const T1& s1, const T2& s2) {
return s1.begin() == s2.begin();
}
}
Y_UNIT_TEST_SUITE(TPunycodeTest) {
static bool TestRaw(const TString& utf8, const TString& punycode) {
TUtf16String unicode = UTF8ToWide(utf8);
TString buf1;
TUtf16String buf2;
return HasSameBuffer(WideToPunycode(unicode, buf1), buf1) && buf1 == punycode && HasSameBuffer(PunycodeToWide(punycode, buf2), buf2) && buf2 == unicode && WideToPunycode(unicode) == punycode && PunycodeToWide(punycode) == unicode;
}
Y_UNIT_TEST(RawEncodeDecode) {
UNIT_ASSERT(TestRaw("", ""));
UNIT_ASSERT(TestRaw(" ", " -"));
UNIT_ASSERT(TestRaw("-", "--"));
UNIT_ASSERT(TestRaw("!@#$%", "!@#$%-"));
UNIT_ASSERT(TestRaw("xn-", "xn--"));
UNIT_ASSERT(TestRaw("xn--", "xn---"));
UNIT_ASSERT(TestRaw("abc", "abc-"));
UNIT_ASSERT(TestRaw("Latin123", "Latin123-"));
UNIT_ASSERT(TestRaw("München", "Mnchen-3ya"));
UNIT_ASSERT(TestRaw("bücher", "bcher-kva"));
UNIT_ASSERT(TestRaw("BüüchEr", "BchEr-kvaa"));
UNIT_ASSERT(TestRaw("президент", "d1abbgf6aiiy"));
UNIT_ASSERT(TestRaw("Президент", "r0a6bcbig1bsy"));
UNIT_ASSERT(TestRaw("ПРЕЗИДЕНТ", "g0abbgf6aiiy"));
UNIT_ASSERT(TestRaw("рф", "p1ai"));
UNIT_ASSERT(TestRaw("пример", "e1afmkfd"));
{
const wchar16 tmp[] = {0x82, 0x81, 0x80, 0};
UNIT_ASSERT(PunycodeToWide("abc") == tmp); // "abc" is still valid punycode
}
UNIT_ASSERT_EXCEPTION(PunycodeToWide(" "), TPunycodeError);
UNIT_ASSERT_EXCEPTION(PunycodeToWide("абвгд"), TPunycodeError);
UNIT_ASSERT_EXCEPTION(PunycodeToWide("-"), TPunycodeError);
{
TString longIn;
for (size_t i = 0; i < 1024; ++i)
longIn += "Qй";
TString longOut = "QQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQ-lo11fbabbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb";
UNIT_ASSERT(TestRaw(longIn, longOut));
}
}
static bool TestHostName(const TString& utf8, const TString& punycode, bool canBePunycode = false) {
TUtf16String unicode = UTF8ToWide(utf8);
TString buf1;
TUtf16String buf2;
//Cerr << "Testing " << utf8 << Endl;
return HostNameToPunycode(unicode) == punycode && HostNameToPunycode(UTF8ToWide(punycode)) == punycode // repeated encoding should give same result
&& PunycodeToHostName(punycode) == unicode && CanBePunycodeHostName(punycode) == canBePunycode;
}
static bool TestForced(const TString& bad) {
return ForceHostNameToPunycode(UTF8ToWide(bad)) == bad && ForcePunycodeToHostName(bad) == UTF8ToWide(bad);
}
Y_UNIT_TEST(HostNameEncodeDecode) {
UNIT_ASSERT(TestHostName("президент.рф", "xn--d1abbgf6aiiy.xn--p1ai", true));
UNIT_ASSERT(TestHostName("яндекс.ru", "xn--d1acpjx3f.ru", true));
UNIT_ASSERT(TestHostName("пример", "xn--e1afmkfd", true));
UNIT_ASSERT(TestHostName("ascii.test", "ascii.test"));
UNIT_ASSERT(TestHostName("", ""));
UNIT_ASSERT(TestHostName(".", "."));
UNIT_ASSERT(TestHostName("a.", "a.")); // empty root domain is ok
UNIT_ASSERT(TestHostName("a.b.c.д.e.f", "a.b.c.xn--d1a.e.f", true));
UNIT_ASSERT(TestHostName("а.б.в.г.д", "xn--80a.xn--90a.xn--b1a.xn--c1a.xn--d1a", true));
UNIT_ASSERT(TestHostName("-", "-"));
UNIT_ASSERT(TestHostName("xn--", "xn--", true));
UNIT_ASSERT(TestHostName("xn--aaa.-", "xn--aaa.-", true));
UNIT_ASSERT(TestHostName("xn--xn--d1acpjx3f.xn--ru", "xn--xn--d1acpjx3f.xn--ru", true));
{
// non-ascii
TString bad = "президент.рф";
UNIT_ASSERT_EXCEPTION(PunycodeToHostName("президент.рф"), TPunycodeError);
UNIT_ASSERT(ForcePunycodeToHostName(bad) == UTF8ToWide(bad));
}
{
// too long domain label
TString bad(500, 'a');
UNIT_ASSERT_EXCEPTION(HostNameToPunycode(UTF8ToWide(bad)), TPunycodeError);
UNIT_ASSERT(TestForced(bad)); // but can decode it
}
{
// already has ACE prefix
TString bad("xn--яндекс.xn--рф");
UNIT_ASSERT_EXCEPTION(HostNameToPunycode(UTF8ToWide(bad)), TPunycodeError);
UNIT_ASSERT(TestForced(bad));
}
{
// empty non-root domain is not allowed (?)
TString bad(".яндекс.рф");
UNIT_ASSERT_EXCEPTION(HostNameToPunycode(UTF8ToWide(bad)), TPunycodeError);
UNIT_ASSERT(TestForced(bad));
}
UNIT_ASSERT(CanBePunycodeHostName("xn--"));
UNIT_ASSERT(CanBePunycodeHostName("yandex.xn--p1ai"));
UNIT_ASSERT(CanBePunycodeHostName("xn--d1acpjx3f.xn--p1ai"));
UNIT_ASSERT(CanBePunycodeHostName("a.b.c.d.xn--e"));
UNIT_ASSERT(CanBePunycodeHostName("xn--a.b.c.xn--d.e"));
UNIT_ASSERT(!CanBePunycodeHostName("yandex.ru")); // no xn--
UNIT_ASSERT(!CanBePunycodeHostName("яндекс.рф")); // non-ascii
UNIT_ASSERT(!CanBePunycodeHostName("яндекс.xn--p1ai")); // non-ascii
UNIT_ASSERT(!CanBePunycodeHostName(""));
UNIT_ASSERT(!CanBePunycodeHostName("http://xn--a.b")); // scheme prefix is not detected here
}
}
|
