From 55b84c2ca44489eef2ade9eaed460a4f9068930a Mon Sep 17 00:00:00 2001 From: Ronald Cron Date: Mon, 9 Sep 2024 13:46:16 +0200 Subject: [PATCH] Add branch 3.6 detection in build_tree.py Signed-off-by: Ronald Cron --- scripts/mbedtls_framework/build_tree.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/scripts/mbedtls_framework/build_tree.py b/scripts/mbedtls_framework/build_tree.py index 47ea0fc90..8a72a06b4 100644 --- a/scripts/mbedtls_framework/build_tree.py +++ b/scripts/mbedtls_framework/build_tree.py @@ -7,6 +7,7 @@ import os import inspect +import re from typing import Optional def looks_like_tf_psa_crypto_root(path: str) -> bool: @@ -128,3 +129,13 @@ def guess_tf_psa_crypto_root(root: Optional[str] = None) -> str: return root else: raise Exception('TF-PSA-Crypto source tree not found') + +def is_mbedtls_3_6_branch(root: Optional[str] = None) -> bool: + """Whether root is the root directory of the Mbed TLS 3.6 branch """ + if root is None: + root = os.getcwd() + if not looks_like_mbedtls_root(root): + return False + with open(os.path.join(root, 'include', 'mbedtls', 'build_info.h'), 'r') as f: + pattern = re.compile("#define MBEDTLS_VERSION_NUMBER.*0x0306") + return pattern.search(f.read()) is not None