From 51742153d0176b2b5232539d87a2c391528d69f7 Mon Sep 17 00:00:00 2001 From: Pol Henarejos Date: Sat, 4 Feb 2023 23:45:31 +0100 Subject: [PATCH] Adding checks for platform. Signed-off-by: Pol Henarejos --- tests/pk_25519_with_libgcrypt.py | 8 +++++++- tests/pk_signed_mpi_with_libgcrypt.py | 7 ++++++- 2 files changed, 13 insertions(+), 2 deletions(-) diff --git a/tests/pk_25519_with_libgcrypt.py b/tests/pk_25519_with_libgcrypt.py index 5c4638e..9791cd9 100644 --- a/tests/pk_25519_with_libgcrypt.py +++ b/tests/pk_25519_with_libgcrypt.py @@ -1,4 +1,5 @@ from cffi import FFI +import platform ffi = FFI() @@ -35,7 +36,12 @@ ffi.cdef(DEF_gcry_sexp) ffi.cdef(DEF_gcry_pk_sign, override=True) ffi.cdef(DEF_gcry_pk_verify, override=True) ffi.cdef(DEF_gcry_pk_encrypt, override=True) -libgcrypt = ffi.dlopen("libgcrypt.20.dylib") + +if (platform.system() == 'Darwin'): + libgcrypt = ffi.dlopen("libgcrypt.20.dylib") +else: + libgcrypt = ffi.dlopen("libgcrypt.so.20") + def fixup_scalar_cv25519(k): # Fixup is the responsibility for caller for Curve25519 diff --git a/tests/pk_signed_mpi_with_libgcrypt.py b/tests/pk_signed_mpi_with_libgcrypt.py index f0b1e26..3ac7c0c 100644 --- a/tests/pk_signed_mpi_with_libgcrypt.py +++ b/tests/pk_signed_mpi_with_libgcrypt.py @@ -1,4 +1,5 @@ from cffi import FFI +import platform DEF_gcry_sexp=""" typedef unsigned long long size_t; @@ -35,7 +36,11 @@ ffi.cdef(DEF_gcry_sexp) ffi.cdef(DEF_gcry_pk_sign, override=True) ffi.cdef(DEF_gcry_pk_verify, override=True) ffi.cdef(DEF_gcry_pk_encrypt, override=True) -libgcrypt = ffi.dlopen("libgcrypt.20.dylib") + +if (platform.system() == 'Darwin'): + libgcrypt = ffi.dlopen("libgcrypt.20.dylib") +else: + libgcrypt = ffi.dlopen("libgcrypt.so.20") FORMAT_DATA=b"(data(value %b))" FORMAT_SIG=b"(sig-val(ecdsa(r %b)(s %b)))"