mirror of
https://github.com/librekeys/pycvc.git
synced 2026-04-14 08:44:08 -07:00
Pol Henarejos
64 lines
2.6 KiB
Python
64 lines
2.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
/*
|
|
* This file is part of the pyCVC distribution (https://github.com/polhenarejos/pycvc).
|
|
* Copyright (c) 2022 Pol Henarejos.
|
|
*
|
|
* 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, version 3.
|
|
*
|
|
* 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 cvc import oid
|
|
from cryptography.hazmat.primitives import hashes
|
|
from cryptography.hazmat.primitives.asymmetric import padding
|
|
|
|
def to_bytes(n):
|
|
if (n == 0):
|
|
return bytearray([0])
|
|
if (isinstance(n, int)):
|
|
return bytearray(n.to_bytes((n.bit_length() + 7) // 8, 'big'))
|
|
return n #Assume is already
|
|
|
|
def bcd(s):
|
|
return bytearray([int(c) for c in s])
|
|
|
|
def scheme_rsa(o):
|
|
return (o == oid.ID_TA_RSA_V1_5_SHA_1 or o == oid.ID_TA_RSA_V1_5_SHA_256 or o == oid.ID_TA_RSA_V1_5_SHA_512 or o == oid.ID_TA_RSA_PSS_SHA_1 or o == oid.ID_TA_RSA_PSS_SHA_256 or o == oid.ID_TA_RSA_PSS_SHA_512)
|
|
|
|
def scheme_ecdsa(o):
|
|
return (o == oid.ID_TA_ECDSA_SHA_1 or o == oid.ID_TA_ECDSA_SHA_224 or o == oid.ID_TA_ECDSA_SHA_256 or o == oid.ID_TA_ECDSA_SHA_384 or o == oid.ID_TA_ECDSA_SHA_512)
|
|
|
|
def scheme_eddsa(o):
|
|
return (o == oid.ID_RI_ECDH_SHA_256)
|
|
|
|
def from_bcd(c):
|
|
return ''.join([str(s) for s in c])
|
|
|
|
def get_hash_padding(scheme):
|
|
h,p = (None,None)
|
|
if (scheme == oid.ID_TA_ECDSA_SHA_1 or scheme == oid.ID_TA_RSA_PSS_SHA_1 or scheme == oid.ID_TA_RSA_V1_5_SHA_1):
|
|
h = hashes.SHA1()
|
|
elif (scheme == oid.ID_TA_ECDSA_SHA_224):
|
|
h = hashes.SHA224()
|
|
elif (scheme == oid.ID_TA_ECDSA_SHA_256 or scheme == oid.ID_TA_RSA_PSS_SHA_256 or scheme == oid.ID_TA_RSA_V1_5_SHA_256):
|
|
h = hashes.SHA256()
|
|
elif (scheme == oid.ID_TA_ECDSA_SHA_384):
|
|
h = hashes.SHA384()
|
|
elif (scheme == oid.ID_TA_ECDSA_SHA_512 or scheme == oid.ID_TA_RSA_PSS_SHA_512 or scheme == oid.ID_TA_RSA_V1_5_SHA_512):
|
|
h = hashes.SHA512()
|
|
|
|
if (scheme == oid.ID_TA_RSA_V1_5_SHA_1 or scheme == oid.ID_TA_RSA_V1_5_SHA_256 or scheme == oid.ID_TA_RSA_V1_5_SHA_512):
|
|
p = padding.PKCS1v15()
|
|
elif (scheme == oid.ID_TA_RSA_PSS_SHA_1 or scheme == oid.ID_TA_RSA_PSS_SHA_256 or scheme == oid.ID_TA_RSA_PSS_SHA_512):
|
|
p = padding.PSS(mgf=padding.MGF1(h), salt_length=padding.PSS.MAX_LENGTH)
|
|
return h,p |