1 #!/usr/local/bin/python 2 # -*- coding: iso-8859-1 -*- 3 4 # Id 5 6 # Copyright (c) 2004 Kungliga Tekniska Hgskolan 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 37 import re 38 import sys 39 40 import generate 41 42 if len(sys.argv) != 3: 43 print("usage: %s rfc3492.txt" % sys.argv[0]) 44 sys.exit(1) 45 46 f = open(sys.argv[1], 'r') 47 48 examples_h = generate.Header('%s/punycode_examples.h' % sys.argv[2]) 49 examples_c = generate.Header('%s/punycode_examples.c' % sys.argv[2]) 50 51 start = False 52 53 while 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 84 f.close() 85 86 examples_h.file.write( 87 ''' 88 #include <krb5-types.h> 89 90 #define MAX_LENGTH 40 91 92 struct punycode_example { 93 size_t len; 94 uint32_t val[MAX_LENGTH]; 95 const char *pc; 96 const char *description; 97 }; 98 99 extern const struct punycode_example punycode_examples[]; 100 101 extern const size_t punycode_examples_size; 102 ''') 103 104 examples_c.file.write( 105 ''' 106 #include <stdlib.h> 107 #include "punycode_examples.h" 108 109 const struct punycode_example punycode_examples[] = { 110 ''') 111 112 for 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 121 examples_c.file.write( 122 '''}; 123 124 ''') 125 126 examples_c.file.write( 127 "const size_t punycode_examples_size = %u;\n\n" % len(cases)) 128 129 examples_h.close() 130 examples_c.close() 131