-
Notifications
You must be signed in to change notification settings - Fork 4
/
co_toeplitz.py
130 lines (100 loc) · 4.31 KB
/
co_toeplitz.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
#!/usr/bin/env python
# vim: set fileencoding=UTF-8 filetype=python :
r"""
Confirmation module with Toeplitz matrix multiplication
AUTHORS:
- Thomas Loruenser (2013): initial version
"""
###############################################################################
# Copyright 2013, Thomas Loruenser <[email protected]>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
###############################################################################
from __future__ import print_function
from common import *
from ll import *
from staqmath import *
from random import seed, randint
import numpy as np
### globals
n = get_name()
CO_TAGSIZE_BYTE = 8
### classes
class SqCoToeplitz(SqModule):
"""
"""
def __init__(self, sock, in_q, out_q, master=True):
super(SqCoToeplitz, self).__init__(sock, in_q, out_q, master)
def run_master(self):
while True:
key_id, meta, key = self.read_key()
if key_id < 0: break
# calc final tag size
tagsize = CO_TAGSIZE_BYTE
self.debug("CO: Tag size", tagsize)
# generate toeplitz matrix
toeplitz_np = np.array(np.random.randint(0, 2, (len(key)+tagsize)*8-1), 'u1')
# send challenge to bob
self.sendall(struct.pack('<I', key_id) + toeplitz_np.tostring())
# apply hash
key_np = np.unpackbits(np.frombuffer(key, 'u1'))
tag_np = hash_toeplitz(key_np, toeplitz_np)
# receive tag
data = self.recvall()
peer_key_id = struct.unpack('<I', data[:4])[0]
peertag_np = np.frombuffer(data[4:], 'u1')
# compare tags
if (tag_np == peertag_np).all(): confirmed = True
else: confirmed = False
# send if ok
self.sendall(struct.pack('<IB', key_id, confirmed))
# enqueue key if correct
if not confirmed:
meta['error'] = 'Confirmation failed'
self.debug("CO: ERROR: Confirmation failed", key_id, meta)
else:
meta['disclosed_bits'] += tagsize * 8
self.write_key(key_id, meta, key)
def run_slave(self):
while True:
key_id, meta, key = self.read_key()
if key_id < 0: break
# receive challenge
data = self.recvall()
# extract data from packet
peer_key_id = struct.unpack('<I', data[:4])[0]
toeplitz_np = np.frombuffer(data[4:], 'u1')
if peer_key_id != key_id:
self.debug("CO: ERROR: key mismatch", key_id, peer_key_id)
raise SqError("CO: ERROR: key mismatch")
# apply hash
key_np = np.unpackbits(np.frombuffer(key, 'u1'))
tag_np = hash_toeplitz(key_np, toeplitz_np)
tagsize = tag_np.size / 8
self.debug("CO: Got tag size", tagsize)
# send tag and wait for ok
self.sendall(struct.pack('<I', key_id) + tag_np.tostring())
# wait for confimation
data = self.recvall()
peer_key_id, confirmed = struct.unpack('<IB', data)
if peer_key_id != key_id:
self.debug(n, "CO: ERROR: key mismatch", key_id, peer_key_id)
raise SqError("CO: ERROR: key mismatch")
# enqueue key if corrected
if not confirmed:
meta['error'] = 'Confirmation failed'
self.debug("CO: ERROR: Confirmation failed", key_id, meta)
else:
meta['disclosed_bits'] += tagsize * 8
self.write_key(key_id, meta, key)