summaryrefslogtreecommitdiffstats
path: root/tests/auc/gen_ts_55_205_test_sets/pdftxt_2_c.py
blob: 058113c01a0705d691036bd83dd661e94076d759 (plain)
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
#!/usr/bin/env python
 
# Convert test sets pasted from 3GPP TS 55.205 to C code.
 
# (C) 2016 by sysmocom s.f.m.c. GmbH <info@sysmocom.de>
#
# All Rights Reserved
#
# Author: Neels Hofmeyr <nhofmeyr@sysmocom.de>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import sys, os

script_dir = sys.path[0]

fields = (
  'Ki',
  'RAND',
  'OP',
  'OPc',
  'MIL3G-RES',
  'SRES#1',
  'SRES#2',
  'MIL3G-CK',
  'MIL3G-IK',
  'Kc',
)

test_sets_lines = []
test_set_lines = None

for line in [l.strip() for l in open(os.path.join(script_dir, 'ts55_205_test_sets.txt'), 'r')]:
  if line.startswith('Test Set'):
    if test_set_lines:
      test_sets_lines.append(test_set_lines)
    test_set_lines = []
  elif len(line) == 8:
    try:
      is_hex = int(line, 16)
      test_set_lines.append(line)
    except ValueError:
      pass

if test_set_lines:
  test_sets_lines.append(test_set_lines)

# Magic fixups for PDF-to-text uselessness
idx = (( 0, 10, 15, 19),
       ( 1, 11, 16, 20),
       ( 2, 12, 17, 21),
       ( 3, 13, 18, 22),
       ( 4, 14),
       ( 5, ),
       ( 6, ),
       ( 7, 23, 26, 28),
       ( 8, 24, 27, 29),
       ( 9, 25 ),
      )

test_sets = []
for l in test_sets_lines:
  test_sets.append( [ ''.join([l[i] for i in li]) for li in idx ] )

func_templ = open(os.path.join(script_dir, 'func_template.c'), 'r').read()

funcs = []
func_calls = []
nr = 0
for test_set in test_sets:
  nr += 1
  func_name = 'test_set_%d' % nr
  kwargs = dict(zip(fields, test_set))
  kwargs['func_name'] = func_name

  func_calls.append('\t%s();' % func_name)
  funcs.append(func_templ.format(**kwargs))

templ = open(os.path.join(script_dir, 'main_template.c')).read()

code = templ.replace('FUNCTIONS', '\n'.join(funcs)).replace('FUNCTION_CALLS', '\n'.join(func_calls))

print('''
/***** DO NOT EDIT THIS FILE -- THIS CODE IS GENERATED *****
 ***** by gen_ts_55_205_test_sets/pdftxt_2_c.py        *****/
''')
print(code)