feat: support DTK with Go

Signed-off-by: Fred Rolland <frolland@nvidia.com>

Author: rollandf

Golang_RHEL_Dockerfile

@@ -12,16 +12,35 @@ ARG STIG_COMPLIANT=false
 # Final clean image of precompiled driver container
 ARG D_FINAL_BASE_IMAGE=registry.access.redhat.com/ubi9/ubi:latest
 
-##################################################################
-# Stage: Minimal base image update and install common requirements
-
 # DTK base image (below example for specific kernel headers version)
 ARG D_BASE_IMAGE="registry.redhat.io/openshift4/driver-toolkit-rhel9:v4.13.0-202309112001.p0.gd719bdc.assembly.stream"
 # Standart: registry.access.redhat.com/ubi9:latest
 
 ARG D_PYTHON_VERSION="36"
 ARG D_PYTHON="python${D_PYTHON_VERSION}"
 
+##################################################################
+# Stage: build go binary for entrypoint
+FROM golang:1.24 AS go_builder
+
+# Set GOPROXY if provided
+ARG GOPROXY
+ENV GOPROXY=$GOPROXY
+
+WORKDIR /workspace
+
+COPY entrypoint/go.mod go.mod
+COPY entrypoint/go.sum go.sum
+
+RUN go mod download
+
+COPY entrypoint/ .
+
+RUN TARGETARCH=${D_ARCH} TARGETOS=linux make build
+
+##################################################################
+# Stage: Minimal base image update and install common requirements
+
 FROM $D_BASE_IMAGE AS base
 
 # Inherited global args
@@ -38,6 +57,13 @@ RUN set -x && \
 # Container functional requirements
     jq iproute kmod procps-ng udev
 
+COPY --from=go_builder /workspace/build/entrypoint /root/entrypoint
+WORKDIR /root
+ADD ./entrypoint.sh /root/entrypoint.sh
+ADD ./loader.sh /root/loader.sh
+
+ENTRYPOINT ["/root/loader.sh"]
+
 ##############################################################################################
 # Stage: Download NVIDIA driver sources and install src driver container packages requirements
 
@@ -78,7 +104,7 @@ RUN if file ${D_OFED_SRC_ARCHIVE} | grep compressed; then \
     mv ${D_OFED_SRC_ARCHIVE}/MLNX_OFED_SRC-${D_OFED_VERSION} . ; \
   fi
 
-WORKDIR /
+WORKDIR /root
 ADD ./entrypoint.sh /root/entrypoint.sh
 ADD ./dtk_nic_driver_build.sh /root/dtk_nic_driver_build.sh
 
@@ -92,7 +118,7 @@ RUN set -x && \
   fi && \
   rm -f /tmp/stig-fixer.sh
 
-ENTRYPOINT ["/root/entrypoint.sh"]
+ENTRYPOINT ["/root/loader.sh"]
 CMD ["sources"]
 
 LABEL doca-version=${D_DOCA_VERSION}
@@ -153,5 +179,5 @@ WORKDIR /
 ADD ./entrypoint.sh /root/entrypoint.sh
 ADD ./dtk_nic_driver_build.sh /root/dtk_nic_driver_build.sh
 
-ENTRYPOINT ["/root/entrypoint.sh"]
+ENTRYPOINT ["/root/loader.sh"]
 CMD ["precompiled"]

RHEL_Dockerfile

@@ -720,6 +720,29 @@ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
    See the License for the specific language governing permissions and
    limitations under the License.
 
+---
+## go-shellquote
+
+Copyright (C) 2014 Kevin Ballard
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the "Software"),
+to deal in the Software without restriction, including without limitation
+the rights to use, copy, modify, merge, publish, distribute, sublicense,
+and/or sell copies of the Software, and to permit persons to whom the
+Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included
+in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
+OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
+DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE
+OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
 ---
 ## entrypoint
 

THIRD_PARTY_NOTICES

@@ -1,6 +1,18 @@
 #!/bin/bash
 # Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 
+# Load environment variables if file exists
+if [ -f "$(dirname "$0")/dtk.env" ]; then
+    source "$(dirname "$0")/dtk.env"
+fi
+
+: ${USE_NEW_ENTRYPOINT:=false}
+
+if [ "$USE_NEW_ENTRYPOINT" = "true" ]; then
+    echo "Using Go entrypoint for DTK build"
+    exec "$(dirname "$0")/entrypoint" dtk-build
+fi
+
 : ${ENTRYPOINT_DEBUG:=false}
 : ${DTK_OCP_NIC_SHARED_DIR:=/mnt/shared-nvidia-nic-driver-toolkit}
 

dtk_nic_driver_build.sh

@@ -17,11 +17,13 @@
 package main
 
 import (
+	"context"
 	"encoding/json"
 	"flag"
 	"fmt"
 	"os"
 	"os/signal"
+	"path/filepath"
 	"syscall"
 
 	"github.com/go-logr/logr"
@@ -30,10 +32,39 @@ import (
 
 	"github.com/Mellanox/doca-driver-build/entrypoint/internal/config"
 	"github.com/Mellanox/doca-driver-build/entrypoint/internal/constants"
+	"github.com/Mellanox/doca-driver-build/entrypoint/internal/dtk"
 	"github.com/Mellanox/doca-driver-build/entrypoint/internal/entrypoint"
+	"github.com/Mellanox/doca-driver-build/entrypoint/internal/utils/cmd"
 	"github.com/Mellanox/doca-driver-build/entrypoint/internal/version"
 )
 
+type ctxData struct {
+	//nolint:containedctx
+	Ctx    context.Context
+	Cancel context.CancelFunc
+}
+
+// setupSignalHandler takes a signal channel and contexts with cancel functions.
+// It starts a goroutine that cancels the first uncanceled context on receiving a signal,
+// if no uncanceled context exists, it exits the application with code 1.
+func setupSignalHandler(ch chan os.Signal, ctxs []ctxData) {
+	go func() {
+	OUT:
+		for {
+			<-ch
+			for _, ctx := range ctxs {
+				if ctx.Ctx.Err() != nil {
+					// context is already canceled, try next one
+					continue
+				}
+				ctx.Cancel()
+				continue OUT
+			}
+			os.Exit(1)
+		}
+	}()
+}
+
 func main() {
 	cfg, err := config.GetConfig()
 	if err != nil {
@@ -57,6 +88,20 @@ func main() {
 		os.Exit(1)
 	}
 	log.Info("start manager", "mode", containerMode)
+	if containerMode == constants.DriverContainerModeDtkBuild {
+		// Use a context that is canceled on signal
+		ctx, cancel := context.WithCancel(context.Background())
+		// Attach logger to context
+		ctx = logr.NewContext(ctx, log)
+		setupSignalHandler(getSignalChannel(), []ctxData{{Ctx: ctx, Cancel: cancel}})
+
+		if err := dtk.RunBuild(ctx, log, cfg, cmd.New()); err != nil {
+			log.Error(err, "DTK Build failed")
+			os.Exit(1)
+		}
+		return
+	}
+
 	if err := entrypoint.Run(getSignalChannel(), log, containerMode, cfg); err != nil {
 		log.Error(err, "Entrypoint Run failed")
 		os.Exit(1)
@@ -67,9 +112,11 @@ func getContainerMode() (string, error) {
 	flag.Parse()
 	containerMode := flag.Arg(0)
 	if flag.NArg() != 1 ||
-		(containerMode != constants.DriverContainerModePrecompiled && containerMode != string(constants.DriverContainerModeSources)) {
-		return "", fmt.Errorf("container mode argument has invalid value %s, supported values: %s, %s",
-			containerMode, constants.DriverContainerModePrecompiled, constants.DriverContainerModeSources)
+		(containerMode != constants.DriverContainerModePrecompiled &&
+			containerMode != constants.DriverContainerModeSources &&
+			containerMode != constants.DriverContainerModeDtkBuild) {
+		return "", fmt.Errorf("container mode argument has invalid value %s, supported values: %s, %s, %s",
+			containerMode, constants.DriverContainerModePrecompiled, constants.DriverContainerModeSources, constants.DriverContainerModeDtkBuild)
 	}
 	return containerMode, nil
 }
@@ -87,6 +134,11 @@ func getLogger(cfg config.Config) logr.Logger {
 	if cfg.EntrypointDebug {
 		logConfig.Level = zap.NewAtomicLevelAt(zap.DebugLevel)
 		if cfg.DebugLogFile != "" {
+			// Create directory if it doesn't exist
+			logDir := filepath.Dir(cfg.DebugLogFile)
+			if err := os.MkdirAll(logDir, 0o755); err != nil {
+				fmt.Fprintf(os.Stderr, "WARNING: failed to create log directory %s: %v\n", logDir, err)
+			}
 			logConfig.OutputPaths = append(logConfig.OutputPaths, cfg.DebugLogFile)
 			logConfig.ErrorOutputPaths = append(logConfig.ErrorOutputPaths, cfg.DebugLogFile)
 		}