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
|
from __future__ import absolute_import
"""M2Crypto wrapper for OpenSSL DH API.
Copyright (c) 1999-2003 Ng Pheng Siong. All rights reserved."""
from M2Crypto import BIO, m2
from M2Crypto.util import genparam_callback
from typing import AnyStr, Callable, Optional # noqa
class DHError(Exception):
pass
m2.dh_init(DHError)
class DH(object):
"""Object interface to the Diffie-Hellman key exchange protocol.
"""
m2_dh_free = m2.dh_free
def __init__(self, dh, _pyfree=0):
# type: (bytes, int) -> None
assert m2.dh_type_check(dh)
self.dh = dh
self._pyfree = _pyfree
def __del__(self):
# type: () -> None
if getattr(self, '_pyfree', 0):
self.m2_dh_free(self.dh)
def __len__(self):
# type: () -> int
assert m2.dh_type_check(self.dh), "'dh' type error"
return int(m2.dh_size(self.dh))
def __getattr__(self, name):
# type: (str) -> bytes
if name in ('p', 'g', 'pub', 'priv'):
method = getattr(m2, 'dh_get_%s' % (name,))
assert m2.dh_type_check(self.dh), "'dh' type error"
return method(self.dh)
else:
raise AttributeError
def __setattr__(self, name, value):
# type: (str, bytes) -> bytes
if name in ('p', 'g'):
raise DHError('set (p, g) via set_params()')
elif name in ('pub', 'priv'):
raise DHError('generate (pub, priv) via gen_key()')
else:
self.__dict__[name] = value
def _ptr(self):
return self.dh
def check_params(self):
# type: () -> int
assert m2.dh_type_check(self.dh), "'dh' type error"
return m2.dh_check(self.dh)
def gen_key(self):
# type: () -> None
assert m2.dh_type_check(self.dh), "'dh' type error"
m2.dh_generate_key(self.dh)
def compute_key(self, pubkey):
# type: (bytes) -> bytes
assert m2.dh_type_check(self.dh), "'dh' type error"
return m2.dh_compute_key(self.dh, pubkey)
def print_params(self, bio):
# type: (BIO.BIO) -> int
assert m2.dh_type_check(self.dh), "'dh' type error"
return m2.dhparams_print(bio._ptr(), self.dh)
def gen_params(plen, g, callback=genparam_callback):
# type: (int, int, Optional[Callable]) -> DH
dh_parms = m2.dh_generate_parameters(plen, g, callback)
dh_obj = DH(dh_parms, 1)
return dh_obj
def load_params(file):
# type: (AnyStr) -> DH
with BIO.openfile(file) as bio:
return load_params_bio(bio)
def load_params_bio(bio):
# type: (BIO.BIO) -> DH
return DH(m2.dh_read_parameters(bio._ptr()), 1)
def set_params(p, g):
# type: (bytes, bytes) -> DH
dh = m2.dh_new()
m2.dh_set_pg(dh, p, g)
return DH(dh, 1)
# def free_params(cptr):
# m2.dh_free(cptr)
DH_GENERATOR_2 = m2.DH_GENERATOR_2
DH_GENERATOR_5 = m2.DH_GENERATOR_5
|
