1/*
2 * Copyright 2022 The OpenSSL Project Authors. All Rights Reserved.
3 *
4 * Licensed under the Apache License 2.0 (the "License").  You may not use
5 * this file except in compliance with the License.  You can obtain a copy
6 * in the file LICENSE in the source distribution or at
7 * https://www.openssl.org/source/license.html
8 */
9
10#include <openssl/crypto.h>
11#include <string.h>
12
13#include "crypto/punycode.h"
14#include "internal/nelem.h"
15#include "testutil.h"
16
17
18static const struct puny_test {
19    unsigned int raw[50];
20    const char *encoded;
21} puny_cases[] = {
22    /* Test cases from RFC 3492 */
23    {   /* Arabic (Egyptian) */
24        { 0x0644, 0x064A, 0x0647, 0x0645, 0x0627, 0x0628, 0x062A, 0x0643, 0x0644,
25          0x0645, 0x0648, 0x0634, 0x0639, 0x0631, 0x0628, 0x064A, 0x061F
26        },
27        "egbpdaj6bu4bxfgehfvwxn"
28    },
29    {   /* Chinese (simplified) */
30        { 0x4ED6, 0x4EEC, 0x4E3A, 0x4EC0, 0x4E48, 0x4E0D, 0x8BF4, 0x4E2D, 0x6587
31        },
32        "ihqwcrb4cv8a8dqg056pqjye"
33    },
34    {   /* Chinese (traditional) */
35        { 0x4ED6, 0x5011, 0x7232, 0x4EC0, 0x9EBD, 0x4E0D, 0x8AAA, 0x4E2D, 0x6587
36        },
37        "ihqwctvzc91f659drss3x8bo0yb"
38    },
39    {    /* Czech: Pro<ccaron>prost<ecaron>nemluv<iacute><ccaron>esky */
40        { 0x0050, 0x0072, 0x006F, 0x010D, 0x0070, 0x0072, 0x006F, 0x0073, 0x0074,
41          0x011B, 0x006E, 0x0065, 0x006D, 0x006C, 0x0075, 0x0076, 0x00ED, 0x010D,
42          0x0065, 0x0073, 0x006B, 0x0079
43       },
44        "Proprostnemluvesky-uyb24dma41a"
45    },
46    {   /* Hebrew */
47        { 0x05DC, 0x05DE, 0x05D4, 0x05D4, 0x05DD, 0x05E4, 0x05E9, 0x05D5, 0x05D8,
48          0x05DC, 0x05D0, 0x05DE, 0x05D3, 0x05D1, 0x05E8, 0x05D9, 0x05DD, 0x05E2,
49          0x05D1, 0x05E8, 0x05D9, 0x05EA
50        },
51        "4dbcagdahymbxekheh6e0a7fei0b"
52    },
53    {   /* Hindi (Devanagari) */
54        { 0x092F, 0x0939, 0x0932, 0x094B, 0x0917, 0x0939, 0x093F, 0x0928, 0x094D,
55          0x0926, 0x0940, 0x0915, 0x094D, 0x092F, 0x094B, 0x0902, 0x0928, 0x0939,
56          0x0940, 0x0902, 0x092C, 0x094B, 0x0932, 0x0938, 0x0915, 0x0924, 0x0947,
57          0x0939, 0x0948, 0x0902
58        },
59        "i1baa7eci9glrd9b2ae1bj0hfcgg6iyaf8o0a1dig0cd"
60    },
61    {   /* Japanese (kanji and hiragana) */
62        { 0x306A, 0x305C, 0x307F, 0x3093, 0x306A, 0x65E5, 0x672C, 0x8A9E, 0x3092,
63          0x8A71, 0x3057, 0x3066, 0x304F, 0x308C, 0x306A, 0x3044, 0x306E, 0x304B
64        },
65        "n8jok5ay5dzabd5bym9f0cm5685rrjetr6pdxa"
66    },
67    {   /* Korean (Hangul syllables) */
68        { 0xC138, 0xACC4, 0xC758, 0xBAA8, 0xB4E0, 0xC0AC, 0xB78C, 0xB4E4, 0xC774,
69          0xD55C, 0xAD6D, 0xC5B4, 0xB97C, 0xC774, 0xD574, 0xD55C, 0xB2E4, 0xBA74,
70          0xC5BC, 0xB9C8, 0xB098, 0xC88B, 0xC744, 0xAE4C
71        },
72        "989aomsvi5e83db1d2a355cv1e0vak1dwrv93d5xbh15a0dt30a5jpsd879ccm6fea98c"
73    },
74    {   /* Russian (Cyrillic) */
75        { 0x043F, 0x043E, 0x0447, 0x0435, 0x043C, 0x0443, 0x0436, 0x0435, 0x043E,
76          0x043D, 0x0438, 0x043D, 0x0435, 0x0433, 0x043E, 0x0432, 0x043E, 0x0440,
77          0x044F, 0x0442, 0x043F, 0x043E, 0x0440, 0x0443, 0x0441, 0x0441, 0x043A,
78          0x0438
79        },
80        "b1abfaaepdrnnbgefbaDotcwatmq2g4l"
81    },
82    {   /* Spanish */
83        { 0x0050, 0x006F, 0x0072, 0x0071, 0x0075, 0x00E9, 0x006E, 0x006F, 0x0070,
84          0x0075, 0x0065, 0x0064, 0x0065, 0x006E, 0x0073, 0x0069, 0x006D, 0x0070,
85          0x006C, 0x0065, 0x006D, 0x0065, 0x006E, 0x0074, 0x0065, 0x0068, 0x0061,
86          0x0062, 0x006C, 0x0061, 0x0072, 0x0065, 0x006E, 0x0045, 0x0073, 0x0070,
87          0x0061, 0x00F1, 0x006F, 0x006C
88        },
89        "PorqunopuedensimplementehablarenEspaol-fmd56a"
90    },
91    {   /* Vietnamese */
92        { 0x0054, 0x1EA1, 0x0069, 0x0073, 0x0061, 0x006F, 0x0068, 0x1ECD, 0x006B,
93          0x0068, 0x00F4, 0x006E, 0x0067, 0x0074, 0x0068, 0x1EC3, 0x0063, 0x0068,
94          0x1EC9, 0x006E, 0x00F3, 0x0069, 0x0074, 0x0069, 0x1EBF, 0x006E, 0x0067,
95          0x0056, 0x0069, 0x1EC7, 0x0074
96        },
97        "TisaohkhngthchnitingVit-kjcr8268qyxafd2f1b9g"
98    },
99    {   /* Japanese: 3<nen>B<gumi><kinpachi><sensei> */
100        { 0x0033, 0x5E74, 0x0042, 0x7D44, 0x91D1, 0x516B, 0x5148, 0x751F
101        },
102        "3B-ww4c5e180e575a65lsy2b"
103    },
104    {   /* Japanese: <amuro><namie>-with-SUPER-MONKEYS */
105        { 0x5B89, 0x5BA4, 0x5948, 0x7F8E, 0x6075, 0x002D, 0x0077, 0x0069, 0x0074,
106          0x0068, 0x002D, 0x0053, 0x0055, 0x0050, 0x0045, 0x0052, 0x002D, 0x004D,
107          0x004F, 0x004E, 0x004B, 0x0045, 0x0059, 0x0053
108        },
109        "-with-SUPER-MONKEYS-pc58ag80a8qai00g7n9n"
110    },
111    {   /* Japanese: Hello-Another-Way-<sorezore><no><basho> */
112        { 0x0048, 0x0065, 0x006C, 0x006C, 0x006F, 0x002D, 0x0041, 0x006E, 0x006F,
113          0x0074, 0x0068, 0x0065, 0x0072, 0x002D, 0x0057, 0x0061, 0x0079, 0x002D,
114          0x305D, 0x308C, 0x305E, 0x308C, 0x306E, 0x5834, 0x6240
115        },
116        "Hello-Another-Way--fc4qua05auwb3674vfr0b"
117    },
118    {   /* Japanese: <hitotsu><yane><no><shita>2 */
119        { 0x3072, 0x3068, 0x3064, 0x5C4B, 0x6839, 0x306E, 0x4E0B, 0x0032
120        },
121        "2-u9tlzr9756bt3uc0v"
122    },
123    {   /* Japanese: Maji<de>Koi<suru>5<byou><mae> */
124        { 0x004D, 0x0061, 0x006A, 0x0069, 0x3067, 0x004B, 0x006F, 0x0069, 0x3059,
125          0x308B, 0x0035, 0x79D2, 0x524D
126        },
127        "MajiKoi5-783gue6qz075azm5e"
128    },
129    {   /* Japanese: <pafii>de<runba> */
130        { 0x30D1, 0x30D5, 0x30A3, 0x30FC, 0x0064, 0x0065, 0x30EB, 0x30F3, 0x30D0
131        },
132        "de-jg4avhby1noc0d"
133    },
134    {   /* Japanese: <sono><supiido><de> */
135        { 0x305D, 0x306E, 0x30B9, 0x30D4, 0x30FC, 0x30C9, 0x3067
136        },
137        "d9juau41awczczp"
138    },
139    {   /* -> $1.00 <- */
140        { 0x002D, 0x003E, 0x0020, 0x0024, 0x0031, 0x002E, 0x0030, 0x0030, 0x0020,
141          0x003C, 0x002D
142        },
143        "-> $1.00 <--"
144    }
145};
146
147static int test_punycode(int n)
148{
149    const struct puny_test *tc = puny_cases + n;
150    unsigned int buffer[50];
151    unsigned int bsize = OSSL_NELEM(buffer);
152    size_t i;
153
154    if (!TEST_true(ossl_punycode_decode(tc->encoded, strlen(tc->encoded),
155                                        buffer, &bsize)))
156        return 0;
157    for (i = 0; i < OSSL_NELEM(tc->raw); i++)
158        if (tc->raw[i] == 0)
159            break;
160    if (!TEST_mem_eq(buffer, bsize * sizeof(*buffer),
161                     tc->raw, i * sizeof(*tc->raw)))
162        return 0;
163    return 1;
164}
165
166static int test_a2ulabel(void)
167{
168    char out[50];
169    size_t outlen;
170
171    /*
172     * Test that no buffer correctly returns the true length.
173     * The punycode being passed in and parsed is malformed but we're not
174     * verifying that behaviour here.
175     */
176    if (!TEST_int_eq(ossl_a2ulabel("xn--a.b.c", NULL, &outlen), 0)
177            || !TEST_size_t_eq(outlen, 7)
178            || !TEST_int_eq(ossl_a2ulabel("xn--a.b.c", out, &outlen), 1))
179        return 0;
180    /* Test that a short input length returns the true length */
181    outlen = 1;
182    if (!TEST_int_eq(ossl_a2ulabel("xn--a.b.c", out, &outlen), 0)
183            || !TEST_size_t_eq(outlen, 7)
184            || !TEST_int_eq(ossl_a2ulabel("xn--a.b.c", out, &outlen), 1)
185            || !TEST_str_eq(out,"\xc2\x80.b.c"))
186        return 0;
187    /* Test for an off by one on the buffer size works */
188    outlen = 6;
189    if (!TEST_int_eq(ossl_a2ulabel("xn--a.b.c", out, &outlen), 0)
190            || !TEST_size_t_eq(outlen, 7)
191            || !TEST_int_eq(ossl_a2ulabel("xn--a.b.c", out, &outlen), 1)
192            || !TEST_str_eq(out,"\xc2\x80.b.c"))
193        return 0;
194    return 1;
195}
196
197static int test_puny_overrun(void)
198{
199    static const unsigned int out[] = {
200        0x0033, 0x5E74, 0x0042, 0x7D44, 0x91D1, 0x516B, 0x5148, 0x751F
201    };
202    static const char *in = "3B-ww4c5e180e575a65lsy2b";
203    unsigned int buf[OSSL_NELEM(out)];
204    unsigned int bsize = OSSL_NELEM(buf) - 1;
205
206    if (!TEST_false(ossl_punycode_decode(in, strlen(in), buf, &bsize))) {
207        if (TEST_mem_eq(buf, bsize * sizeof(*buf), out, sizeof(out)))
208            TEST_error("CRITICAL: buffer overrun detected!");
209        return 0;
210    }
211    return 1;
212}
213
214int setup_tests(void)
215{
216    ADD_ALL_TESTS(test_punycode, OSSL_NELEM(puny_cases));
217    ADD_TEST(test_a2ulabel);
218    ADD_TEST(test_puny_overrun);
219    return 1;
220}
221