commit b42c50bd6044b29d00012adf02bd8f607c5a2dc3
author Thomas Daubney <thomas.daubney@arm.com> Wed Nov 22 15:53:38 2023 +0000
committer Thomas Daubney <thomas.daubney@arm.com> Wed Nov 22 15:53:38 2023 +0000
tree ecb81c8cfd0ec6aa1d63d6d5f1b6004eb702eeac
parent 79cae20a025d7a988d00e443331b348b479119e3

Make use of new crypto_core_directory function

Signed-off-by: Thomas Daubney <thomas.daubney@arm.com>

scripts/generate_driver_wrappers.py

diff --git a/scripts/generate_driver_wrappers.py b/scripts/generate_driver_wrappers.py
index de89959..a3c8588 100755
--- a/scripts/generate_driver_wrappers.py
+++ b/scripts/generate_driver_wrappers.py
@@ -178,10 +178,7 @@
 
     repo_root = os.path.abspath(args.repo_root)
 
-    if build_tree.looks_like_mbedtls_root(repo_root):
-        library_dir = 'library'
-    elif build_tree.looks_like_tf_psa_crypto_root(repo_root):
-        library_dir = 'core'
+    library_dir = build_tree.crypto_core_directory(repo_root)
 
     output_directory = args.output_directory if args.output_directory is not None else \
         os.path.join(repo_root, library_dir)