コード例 #1
0
ファイル: ecies.py プロジェクト: Parlefan/Python
def gen_ec_keypair(curve="secp112r1"):
    # this function generates EC key pair
    try:
        curve = OpenSSL.curves[curve]
        key = OpenSSL.EC_KEY_new_by_curve_name(curve)
        OpenSSL.EC_KEY_generate_key(key)
        _pubkey_x = OpenSSL.BN_new()
        _pubkey_y = OpenSSL.BN_new()
        _privkey = OpenSSL.EC_KEY_get0_private_key(key)
        _pubkey = OpenSSL.EC_KEY_get0_public_key(key)
        _group = OpenSSL.EC_KEY_get0_group(key)
        OpenSSL.EC_POINT_get_affine_coordinates_GFp(_group, _pubkey, _pubkey_x, _pubkey_y, 0)

        privkey = OpenSSL.malloc(0, OpenSSL.BN_num_bytes(_privkey))
        pubkeyx = OpenSSL.malloc(0, OpenSSL.BN_num_bytes(_pubkey_x))
        pubkeyy = OpenSSL.malloc(0, OpenSSL.BN_num_bytes(_pubkey_y))
        OpenSSL.BN_bn2bin(_privkey, privkey)
        privkey = privkey.raw
        OpenSSL.BN_bn2bin(_pubkey_x, pubkeyx)
        pubkeyx = pubkeyx.raw
        OpenSSL.BN_bn2bin(_pubkey_y, pubkeyy)
        pubkeyy = pubkeyy.raw
        # self.raw_check_key(privkey, pubkeyx, pubkeyy)

        full_privkey = pack("!H", curve) + pack("!H", len(privkey)) + privkey
        full_pubkey = pack("!H", curve) + pack("!H", len(pubkeyx)) + pubkeyx + pack("!H", len(pubkeyy)) + pubkeyy
        return full_privkey, full_pubkey

    finally:
        # release c pointers
        OpenSSL.EC_KEY_free(key)
        OpenSSL.BN_free(_pubkey_x)
        OpenSSL.BN_free(_pubkey_y)
コード例 #2
0
ファイル: ecies.py プロジェクト: Parlefan/Python
def ecdh_key(a_privkey, b_pubkey):
    # keys should be in binary format
    a_curve = int(a_privkey[0:2].encode("hex"), 16)
    b_curve = int(b_pubkey[0:2].encode("hex"), 16)
    if a_curve != b_curve:
        raise Exception("ECDH Error: Both key must have the save curve type.")

    sx = int(b_pubkey[2:4].encode("hex"), 16)
    sy = int(b_pubkey[4 + sx : sx + 6].encode("hex"), 16)
    pub_x, pub_y = b_pubkey[4 : 4 + sx], b_pubkey[6 + sx : 6 + sx + sy]

    b_key = OpenSSL.EC_KEY_new_by_curve_name(b_curve)
    _pub_x = OpenSSL.BN_bin2bn(pub_x, sx, 0)
    _pub_y = OpenSSL.BN_bin2bn(pub_y, sy, 0)
    _group = OpenSSL.EC_KEY_get0_group(b_key)
    _pubkey = OpenSSL.EC_POINT_new(_group)
    OpenSSL.EC_POINT_set_affine_coordinates_GFp(_group, _pubkey, _pub_x, _pub_y, 0)
    OpenSSL.EC_KEY_set_public_key(b_key, _pubkey)
    # OpenSSL.EC_KEY_check_key(b_key)

    s = int(a_privkey[2:4].encode("hex"), 16)
    priv = a_privkey[4 : 4 + s]
    a_key = OpenSSL.EC_KEY_new_by_curve_name(a_curve)
    _privkey = OpenSSL.BN_bin2bn(priv, len(priv), 0)
    OpenSSL.EC_KEY_set_private_key(a_key, _privkey)

    # ECDH
    OpenSSL.ECDH_set_method(a_key, OpenSSL.ECDH_OpenSSL())
    ecdh_buf = OpenSSL.malloc(0, s)  # computed buffer size should the same as key length
    ecdh_keylen = OpenSSL.ECDH_compute_key(ecdh_buf, s, _pubkey, a_key, 0)
    return ecdh_buf.raw