#!/bin/bash
#
# SPDX-FileCopyrightText: Copyright (c) 1993-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at

# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
# This short shell script will extract all the strong "text" symbols from the
# shared library and create a new "stub" shared library with the same symbols.
# The body of these functions will be empty and therefore have no dependencies.
# This scripts uses whatever CC is defined in the user's environment.
#

set -o pipefail

# check arguments
if [ $# -ne 2 ] ; then
    echo "Usage: $(basename $0) IN_LIBFILE OUT_LIBFILE"
    exit 1
fi

IN_LIBFILE="$1"
OUT_LIBFILE="$2"

# check compiler
if [ -z "${CC}" ] ; then
    echo "Error: Environment variable 'CC' has not been defined"
    exit 1
fi

SONAME=$(readelf -d "${IN_LIBFILE}" | grep '(SONAME)' | cut -d [ -f 2 | cut -d ] -f 1)

# Detect OS from /etc/os-release instead of lsb_release (which may not be installed).
if [ -f /etc/os-release ]; then
    . /etc/os-release
    OS="${ID}-${VERSION_ID}"
else
    OS="unknown"
fi

if [ "$OS" = "ubuntu-22.04" ] ; then
    EXTRA_NM_FLAG="--without-symbol-versions"
elif [ "$OS" = "ubuntu-24.04" ] ; then
    EXTRA_NM_FLAG="--without-symbol-versions"
fi

# make stub library
# This uses the system nm in containers that compile with SCL, but the output is identical to the SCL output
if [ -z "${CC_ARGS}" ] ; then
    nm -D "${IN_LIBFILE}" ${EXTRA_NM_FLAG} | \
        awk '{if ($2 == "T") { print "void",$3,"() {}" }}' | \
        ${CC} -xc -Og -fPIC -shared -Wl,-soname=${SONAME} -Wl,--strip-all -o "${OUT_LIBFILE}" -
else
    nm -D "${IN_LIBFILE}" ${EXTRA_NM_FLAG} | \
        awk '{if ($2 == "T") { print "void",$3,"() {}" }}' | \
        ${CC} -xc -Og -fPIC -shared -Wl,-soname=${SONAME} -Wl,--strip-all -o "${OUT_LIBFILE}" "${CC_ARGS}" -
fi

exit $?