1#!/usr/local/bin/python
2# -*- coding: iso-8859-1 -*-
3
4# Id
5
6# Copyright (c) 2004 Kungliga Tekniska H��gskolan
7# (Royal Institute of Technology, Stockholm, Sweden).
8# All rights reserved.
9#
10# Redistribution and use in source and binary forms, with or without
11# modification, are permitted provided that the following conditions
12# are met:
13#
14# 1. Redistributions of source code must retain the above copyright
15#    notice, this list of conditions and the following disclaimer.
16#
17# 2. Redistributions in binary form must reproduce the above copyright
18#    notice, this list of conditions and the following disclaimer in the
19#    documentation and/or other materials provided with the distribution.
20#
21# 3. Neither the name of the Institute nor the names of its contributors
22#    may be used to endorse or promote products derived from this software
23#    without specific prior written permission.
24#
25# THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
26# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
28# ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
29# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
30# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
31# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
32# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
33# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
34# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
35# SUCH DAMAGE.
36
37import re
38import sys
39
40import generate
41
42if len(sys.argv) != 3:
43    print("usage: %s rfc3492.txt" % sys.argv[0])
44    sys.exit(1)
45
46f = open(sys.argv[1], 'r')
47
48examples_h = generate.Header('%s/punycode_examples.h' % sys.argv[2])
49examples_c = generate.Header('%s/punycode_examples.c' % sys.argv[2])
50
51start = False
52
53while True:
54    l = f.readline()
55    if not l:
56        break
57    if l[-2:] == "\\\n":
58        l2 = f.readline()
59        if not l2:
60            raise Exception("EOF in backslash escape")
61        l2 = re.sub('^ *', '', l2)
62        l = l[:-2] + l2
63    if start:
64        if re.match('7\.2', l):
65            start = False
66        else:
67            m = re.search('^ *\([A-Z]\) *(.*)$', l);
68            if m:
69                desc = m.group(1)
70                codes = []
71            else:
72                m = re.search('^ *([uU]+.*) *$', l)
73                if m:
74                    codes.extend(m.group(1).split(' '))
75                else:
76                    m = re.search('^ *Punycode: (.*) *$', l)
77                    if m:
78                        cases.append([codes, m.group(1), desc])
79    else:
80        if re.match('^7\.1', l):
81            start = True
82            cases = []
83
84f.close()
85
86examples_h.file.write(
87'''
88#include <krb5-types.h>
89
90#define MAX_LENGTH 40
91
92struct punycode_example {
93    size_t len;
94    uint32_t val[MAX_LENGTH];
95    const char *pc;
96    const char *description;
97};
98
99extern const struct punycode_example punycode_examples[];
100
101extern const size_t punycode_examples_size;
102''')
103
104examples_c.file.write(
105'''
106#include <stdlib.h>
107#include "punycode_examples.h"
108
109const struct punycode_example punycode_examples[] = {
110''')
111
112for x in cases:
113    [cp, pc, desc] = x
114    examples_c.file.write(
115        "  {%u, {%s}, \"%s\", \"%s\"},\n" %
116        (len(cp),
117         ",".join([re.sub('[uU]\+', '0x', x) for x in cp]),
118         pc,
119         desc))
120
121examples_c.file.write(
122'''};
123
124''')
125
126examples_c.file.write(
127    "const size_t punycode_examples_size = %u;\n\n" % len(cases))
128
129examples_h.close()
130examples_c.close()
131