/
format_john_skein512.py
137 lines (105 loc) · 3.81 KB
/
format_john_skein512.py
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
127
128
129
130
131
132
133
134
135
136
137
# -*- coding: utf-8 -*-
# skein with output to john's format
# Copyright © 2015 Aleksey Cherepanov <lyosha@openwall.com>
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted.
import util_ui as U
import bytecode_main as B
import output_c as O
# import output_c_sse as O
args = U.parse_args()
c_template = 'raw2'
algo_file = 'skein512'
c_code = U.load_code_template(c_template)
size = 8
endianity = 'be'
args['args'] = { 'size': size, 'use_unroll': 1, 'two_blocks': 0 }
# O.apply_size(size)
# O.apply_endianity(endianity)
# bs_size = 64
# O.apply_bs_size(bs_size)
# Format setup
tests = r'''
//{"d578679d76b51f6a79904a3a9b58bc4fc3f6d1b6812411fcf5840570353c7fe5dbfa7b539a707abe464dfcbb691ad97b6d0da7b419e16780f68a2ed8ef250c2e", "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"},
{"8f5dd9ec798152668e35129496b029a960c9a9b88662f7f9482f110b31f9f93893ecfb25c009baad9e46737197d5630379816a886aa05526d3a70df272d96e75", "abc"},
{"bc5b4c50925519c290cc634277ae3d6257212395cba733bbad37a4af0fa06af41fca7903d06564fea7a2d3730dbdb80c1f85562dfcc070334ea4d1d9e72cba7a", ""},
{"94c2ae036dba8783d0b3f7d6cc111ff810702f5c77707999be7e1c9486ff238a7044de734293147359b4ac7e1d09cd247c351d69826b78dcddd951f0ef912713", "The quick brown fox jumps over the lazy dog"},
{"658223cb3d69b5e76e3588ca63feffba0dc2ead38a95d0650564f2a39da8e83fbb42c9d6ad9e03fbfde8a25a880357d457dbd6f74cbcb5e728979577dbce5436", "The quick brown fox jumps over the lazy dog."},
'''
vs = {
'fmt_struct_name': 'raw1_skein512',
'format_name': 'Raw-skein512',
'algo_name': 'SKEIN512',
'tag': '$skein512$',
# %% bump length
# 'plaintext_length': '47',
'plaintext_length': '80' if args['args']['two_blocks'] else '47',
'binary_form_size': '64',
'tests': tests
}
U.setup_vars(vs)
# Optimizations and code generation
interleave = 1
# B.global_vars['batch_size'] = 20
# B.global_vars['batch_size'] = 16
B.global_vars['batch_size'] = 1
reverse_num = 0
B.global_vars['interleave'] = interleave
B.global_vars['reverse_num'] = reverse_num
B.global_vars['skein'] = 1
# # Print bitslice and exit.
# d = B.thread_code( B.get_code_full(algo_file, **args),
# B.replace_state_with_const,
# [ B.unpack_const_subscripts, 'k' ],
# B.remove_assignments,
# [ B.compute_const_expressions, size ],
# B.drop_arrays,
# B.drop_print,
# [ B.compute_const_expressions, size ],
# # [ B.dump, 'before.bytecode' ],
# [ B.bitslice, size ],
# B.drop_unused,
# [ B.dump, 'skein512bs.bytecode' ]
# )
# exit(1)
d = B.thread_code( B.get_code_full(algo_file, **args),
B.replace_state_with_const,
[ B.dump, 'pure.bytecode' ],
# [ B.bitslice, 64, size ],
# [ B.unroll_cycle_const_range, 'setupW' ],
# [ B.unroll_cycle_const_range_partly, 'setupW', 16 ],
# [ B.unroll_cycle_const_range_partly, 'main', 40 ],
# [ B.unroll_cycle_const_range_partly, 'main', 2 ],
# [ B.unroll_cycle_const_range, 'main' ],
# B.remove_assignments,
[ B.compute_const_expressions, size ],
# B.drop_print,
# [ B.reduce_assignments, 'main', 'main_end' ],
[ B.dump, 'all.bytecode' ],
# B.research_reverse,
# [ B.reverse_ops, reverse_num ],
[ B.no_reverse, reverse_num ],
)
reverse = d['reverse']
code = d['code']
scalar = B.deep_copy(code)
reverse_str = B.thread_code( reverse,
[ B.dump, 'reverse.bytecode' ],
[ O.gen_to_str, '$code', args ]
)
B.global_vars['reverse'] = reverse_str
scalar_str = B.thread_code( scalar,
[ B.dump, 'scalar.bytecode' ],
[ O.gen_to_str, '$code', args ]
)
B.global_vars['scalar'] = scalar_str
B.global_vars['vectorize'] = 1
B.thread_code( code,
# B.reuse_variables,
B.vectorize_maybe,
B.use_define_for_some,
[ B.interleave, interleave ],
[ B.dump, 'code.bytecode' ],
[ O.gen, c_code, args, B.global_vars ]
)