From a5cdcc804562143f014c6ba170fef47c45eeaf3d Mon Sep 17 00:00:00 2001
From: "Zheng, Qi" <huaiqing.zq@antgroup.com>
Date: Wed, 7 Dec 2022 17:07:04 +0800
Subject: [PATCH] [toolchain] Make occlum-go support gcc/glibc option

In default, occlum-go uses musl-gcc. Users can configure
occlum-go to use gcc like below.
" CC=gcc occlum-go build "
---
 tools/installer/deb/toolchains/golang/rules                     | 2 +-
 .../rpm/toolchains/golang/occlum-toolchains-golang.spec         | 2 +-
 tools/toolchains/golang/build.sh                                | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/tools/installer/deb/toolchains/golang/rules b/tools/installer/deb/toolchains/golang/rules
index ba978d40..e39e9894 100755
--- a/tools/installer/deb/toolchains/golang/rules
+++ b/tools/installer/deb/toolchains/golang/rules
@@ -36,7 +36,7 @@ override_dh_install:
 	rm -rf $(install_dir)/.git*
 	cat > $(install_dir)/bin/occlum-go <<EOF
 		#!/bin/bash
-		OCCLUM_GCC="\$$(which occlum-gcc)"
+		OCCLUM_GCC="\$${CC:-\$$(which occlum-gcc)}"
 		OCCLUM_GOFLAGS="-buildmode=pie \$$GOFLAGS"
 		CC=\$$OCCLUM_GCC GOFLAGS=\$$OCCLUM_GOFLAGS /opt/occlum/toolchains/golang/bin/go "\$$@"
 	EOF
diff --git a/tools/installer/rpm/toolchains/golang/occlum-toolchains-golang.spec b/tools/installer/rpm/toolchains/golang/occlum-toolchains-golang.spec
index db5beafa..426bba1b 100644
--- a/tools/installer/rpm/toolchains/golang/occlum-toolchains-golang.spec
+++ b/tools/installer/rpm/toolchains/golang/occlum-toolchains-golang.spec
@@ -49,7 +49,7 @@ mv go-go%{GO_VERSION} %{buildroot}%{INSTALL_DIR}/golang
 rm -rf %{buildroot}%{INSTALL_DIR}/golang/.git*
 cat > %{buildroot}%{INSTALL_DIR}/golang/bin/occlum-go <<EOF
 #!/bin/bash
-OCCLUM_GCC="\$(which occlum-gcc)"
+OCCLUM_GCC="\${CC:-\$(which occlum-gcc)}"
 OCCLUM_GOFLAGS="-buildmode=pie \$GOFLAGS"
 CC=\$OCCLUM_GCC GOFLAGS=\$OCCLUM_GOFLAGS %{INSTALL_DIR}/golang/bin/go "\$@"
 EOF
diff --git a/tools/toolchains/golang/build.sh b/tools/toolchains/golang/build.sh
index f518f04b..4ad509ee 100755
--- a/tools/toolchains/golang/build.sh
+++ b/tools/toolchains/golang/build.sh
@@ -28,7 +28,7 @@ mv ${BUILD_DIR} ${INSTALL_DIR}
 # Generate the wrappers for Go
 cat > ${INSTALL_DIR}/bin/occlum-go <<EOF
 #!/bin/bash
-OCCLUM_GCC="\$(which occlum-gcc)"
+OCCLUM_GCC="\${CC:-\$(which occlum-gcc)}"
 OCCLUM_GOFLAGS="-buildmode=pie \$GOFLAGS"
 CC=\$OCCLUM_GCC GOFLAGS=\$OCCLUM_GOFLAGS ${INSTALL_DIR}/bin/go "\$@"
 EOF