| 12
 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
 
 |  
"""M2Crypto wrapper for OpenSSL DH API.
 
Copyright (c) 1999-2003 Ng Pheng Siong. All rights reserved."""
 
from util import genparam_callback
import BIO, Err, m2
 
class DHError(Exception): pass
 
m2.dh_init(DHError)
 
class DH:
 
    """
    Object interface to the Diffie-Hellman key exchange
    protocol.
    """
 
    m2_dh_free = m2.dh_free
 
    def __init__(self, dh, _pyfree=0):
        assert m2.dh_type_check(dh)
        self.dh = dh
        self._pyfree = _pyfree
 
    def __del__(self):
        if getattr(self, '_pyfree', 0):
            self.m2_dh_free(self.dh)
 
    def __len__(self):
        assert m2.dh_type_check(self.dh), "'dh' type error"
        return m2.dh_size(self.dh)
 
    def __getattr__(self, name):
        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):
        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):
        assert m2.dh_type_check(self.dh), "'dh' type error"
        return m2.dh_check(self.dh)
 
    def gen_key(self):
        assert m2.dh_type_check(self.dh), "'dh' type error"
        m2.dh_generate_key(self.dh)   
 
    def compute_key(self, pubkey):
        assert m2.dh_type_check(self.dh), "'dh' type error"
        return m2.dh_compute_key(self.dh, pubkey)
 
    def print_params(self, bio):
        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):
    return DH(m2.dh_generate_parameters(plen, g, callback), 1)
 
 
def load_params(file):
    bio = BIO.openfile(file)
    return load_params_bio(bio)
 
 
def load_params_bio(bio):
    return DH(m2.dh_read_parameters(bio._ptr()), 1)
 
 
def set_params(p, g):
    dh = m2.dh_new()
    m2.dh_set_p(dh, p)
    m2.dh_set_g(dh, 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 | 
Partager