1 files changed, 165 insertions, 0 deletions
diff --git a/BaseTools/Plugin/LinuxGccToolChain/LinuxGccToolChain.py b/BaseTools/Plugin/LinuxGccToolChain/LinuxGccToolChain.py
new file mode 100644
index 0000000000..e7ef3df3b6
--- /dev/null
+++ b/BaseTools/Plugin/LinuxGccToolChain/LinuxGccToolChain.py
@@ -0,0 +1,165 @@
+# @file LinuxGccToolChain.py
+# Plugin to configures paths for GCC/GCC5 ARM/AARCH64/RISCV/LOONGARCH64 Toolchain
+##
+# This plugin sets environment variables used in tools_def.template to specify the GCC compiler
+# for the requested build architecture.
+#
+# Copyright (c) Microsoft Corporation
+# Copyright (c) 2020, Hewlett Packard Enterprise Development LP. All rights reserved.<BR>
+# Copyright (c) 2022, Loongson Technology Corporation Limited. All rights reserved.<BR>
+# SPDX-License-Identifier: BSD-2-Clause-Patent
+##
+import os
+import logging
+from edk2toolext.environment.plugintypes.uefi_build_plugin import IUefiBuildPlugin
+from edk2toolext.environment import shell_environment
+
+
+class LinuxGccToolChain(IUefiBuildPlugin):
+
+    def do_post_build(self, thebuilder):
+        return 0
+
+    def do_pre_build(self, thebuilder):
+        self.Logger = logging.getLogger("LinuxGccToolChain")
+
+        #
+        # In order to keep this plugin backwards compatible with the older GCC5 toolchain tag,
+        # we support setting either the GCC5 or GCC set of env variables required
+        #
+        toolchain = "GCC"
+
+        #
+        # GCC - The non-x86 compilers need their paths set if available. To make this more stable, check for
+        # any substring of GCC in the TOOL_CHAIN_TAG to get the right compiler
+        #
+        if "GCC" in thebuilder.env.GetValue("TOOL_CHAIN_TAG"):
+            if "GCC5" in thebuilder.env.GetValue("TOOL_CHAIN_TAG"):
+                toolchain = "GCC5"
+
+            # Start with AARACH64 compiler
+            ret = self._check_aarch64(toolchain)
+            if ret != 0:
+                self.Logger.critical("Failed in check aarch64")
+                return ret
+
+            # Check arm compiler
+            ret = self._check_arm(toolchain)
+            if ret != 0:
+                self.Logger.critical("Failed in check arm")
+                return ret
+
+            # Check RISCV64 compiler
+            ret = self._check_riscv64(toolchain)
+            if ret != 0:
+                self.Logger.critical("Failed in check riscv64")
+                return ret
+
+            # Check LoongArch64 compiler
+            ret = self._check_loongarch64(toolchain)
+            if ret != 0:
+                self.Logger.critical("Failed in check loongarch64")
+                return ret
+
+        return 0
+
+    def _check_arm(self, toolchain):
+        # check to see if full path already configured
+        if shell_environment.GetEnvironment().get_shell_var(toolchain + "_ARM_PREFIX") is not None:
+            self.Logger.info(toolchain + "_ARM_PREFIX is already set.")
+
+        else:
+            # now check for install dir.  If set then set the Prefix
+            install_path = shell_environment.GetEnvironment().get_shell_var(toolchain + "_ARM_INSTALL")
+            if install_path is None:
+                return 0
+
+            # make the PREFIX to align with tools_def.txt
+            prefix = os.path.join(install_path, "bin", "arm-none-linux-gnueabihf-")
+            shell_environment.GetEnvironment().set_shell_var(toolchain + "_ARM_PREFIX", prefix)
+
+        # now confirm it exists
+        if not os.path.exists(shell_environment.GetEnvironment().get_shell_var(toolchain + "_ARM_PREFIX") + "gcc"):
+            self.Logger.error("Path for " + toolchain + "_ARM_PREFIX toolchain is invalid")
+            return -2
+
+        return 0
+
+    def _check_aarch64(self, toolchain):
+        # check to see if full path already configured
+        if shell_environment.GetEnvironment().get_shell_var(toolchain + "_AARCH64_PREFIX") is not None:
+            self.Logger.info(toolchain + "_AARCH64_PREFIX is already set.")
+
+        else:
+            # now check for install dir.  If set then set the Prefix
+            install_path = shell_environment.GetEnvironment(
+            ).get_shell_var(toolchain + "_AARCH64_INSTALL")
+            if install_path is None:
+                return 0
+
+            # make PREFIX to align with tools_def.txt
+            prefix = os.path.join(install_path, "bin", "aarch64-none-linux-gnu-")
+            shell_environment.GetEnvironment().set_shell_var(toolchain + "_AARCH64_PREFIX", prefix)
+
+        # now confirm it exists
+        if not os.path.exists(shell_environment.GetEnvironment().get_shell_var(toolchain + "_AARCH64_PREFIX") + "gcc"):
+            self.Logger.error(
+                "Path for " + toolchain + "_AARCH64_PREFIX toolchain is invalid")
+            return -2
+
+        return 0
+
+    def _check_riscv64(self, toolchain):
+        # now check for install dir.  If set then set the Prefix
+        install_path = shell_environment.GetEnvironment(
+        ).get_shell_var(toolchain + "_RISCV64_INSTALL")
+        if install_path is None:
+            return 0
+
+        # check to see if full path already configured
+        if shell_environment.GetEnvironment().get_shell_var(toolchain + "_RISCV64_PREFIX") is not None:
+            self.Logger.info(toolchain + "_RISCV64_PREFIX is already set.")
+
+        else:
+            # make PREFIX to align with tools_def.txt
+            prefix = os.path.join(install_path, "bin", "riscv64-unknown-elf-")
+            shell_environment.GetEnvironment().set_shell_var(toolchain + "_RISCV64_PREFIX", prefix)
+
+        # now confirm it exists
+        if not os.path.exists(shell_environment.GetEnvironment().get_shell_var(toolchain + "_RISCV64_PREFIX") + "gcc"):
+            self.Logger.error(
+                "Path for " + toolchain + "_RISCV64_PREFIX toolchain is invalid")
+            return -2
+
+        # Check if LD_LIBRARY_PATH is set for the libraries of RISC-V GCC toolchain
+        if shell_environment.GetEnvironment().get_shell_var("LD_LIBRARY_PATH") is not None:
+            self.Logger.info("LD_LIBRARY_PATH is already set.")
+
+        prefix = os.path.join(install_path, "lib")
+        shell_environment.GetEnvironment().set_shell_var("LD_LIBRARY_PATH", prefix)
+
+        return 0
+
+    def _check_loongarch64(self, toolchain):
+        # check to see if full path already configured
+        if shell_environment.GetEnvironment().get_shell_var(toolchain + "_LOONGARCH64_PREFIX") is not None:
+            self.Logger.info(toolchain + "_LOONGARCH64_PREFIX is already set.")
+
+        else:
+            # now check for install dir.  If set then set the Prefix
+            install_path = shell_environment.GetEnvironment(
+            ).get_shell_var(toolchain + "_LOONGARCH64_INSTALL")
+            if install_path is None:
+                return 0
+
+            # make PREFIX to align with tools_def.txt
+            prefix = os.path.join(install_path, "bin", "loongarch64-unknown-linux-gnu-")
+            shell_environment.GetEnvironment().set_shell_var(toolchain + "_LOONGARCH64_PREFIX", prefix)
+
+        # now confirm it exists
+        if not os.path.exists(shell_environment.GetEnvironment().get_shell_var(toolchain + "_LOONGARCH64_PREFIX") + "gcc"):
+            self.Logger.error(
+                "Path for " + toolchain + "_LOONGARCH64_PREFIX toolchain is invalid")
+            return -2
+
+        return 0