setup_workspace.sh

#!/bin/bash

set -e

function DisplayNvidiaLicense {

    # verify that curl is installed
    if  ! which curl > /dev/null  ; then
      echo "curl is not installed."
      echo "curl can be installed by 'sudo apt-get install curl'."
      exit 1
    fi

	local release;
	IFS='.' read -a release <<< "$1"
    
    RELEASE="r${release[0]}_Release_v${release[1]}.${release[2]:-0}"
    
    local URL="https://developer.download.nvidia.com/embedded/L4T/${RELEASE}/$2/Tegra_Software_License_Agreement-Tegra-Linux.txt"

    echo -e "\nPlease notice: This script will download the kernel source (from nv-tegra, NVIDIA's public git repository) which is subject to the following license:\n${URL}\n"

	local LICENSE=$(curl -Ls ${URL})

    ## display the page ##
    echo -e "${LICENSE}\n\n"

    read -t 30 -n 1 -s -r -e -p 'Press any key to continue (or wait 30 seconds..)'
    echo
}


if [[ "$1" == "-h" ]]; then
    echo "setup_workspace.sh [JetPack_version]"
    echo "setup_workspace.sh -h"
    echo "JetPack_version can be 4.6.1, 5.0.2, 5.1.2, 6.0, 6.1, 6.2, 6.2.1"
    exit 1
fi

export DEVDIR=$(cd `dirname $0` && pwd)

. $DEVDIR/scripts/setup-common "$1"
echo "Setup JetPack $1 to sources_$1"

# Display NVIDIA license
DisplayNvidiaLicense "${REVISION}" "${LICENSE}"

# Install L4T gcc if not installed
if [[ $(uname -m) == aarch64 ]]; then
    echo
    echo Native build
    echo
else
    if [[ ! -d "$DEVDIR/l4t-gcc/$JETPACK_VERSION/bin/" ]]; then
        mkdir -p $DEVDIR/l4t-gcc/$JETPACK_VERSION
        cd $DEVDIR/l4t-gcc/$JETPACK_VERSION
        if [[ "$JETPACK_VERSION" == "6.x" ]]; then
            wget --quiet --show-progress https://developer.nvidia.com/downloads/embedded/l4t/r36_release_v3.0/toolchain/aarch64--glibc--stable-2022.08-1.tar.bz2 -O aarch64--glibc--stable-final.tar.bz2
            tar xf aarch64--glibc--stable-final.tar.bz2 --strip-components 1
        elif [[ "$JETPACK_VERSION" == "5.x" ]]; then
            wget --quiet --show-progress https://developer.nvidia.com/embedded/jetson-linux/bootlin-toolchain-gcc-93 -O aarch64--glibc--stable-final.tar.gz
            tar xf aarch64--glibc--stable-final.tar.gz
        elif [[ "$JETPACK_VERSION" == "4.6.1" ]]; then
            wget --quiet --show-progress http://releases.linaro.org/components/toolchain/binaries/7.3-2018.05/aarch64-linux-gnu/gcc-linaro-7.3.1-2018.05-x86_64_aarch64-linux-gnu.tar.xz
            tar xf gcc-linaro-7.3.1-2018.05-x86_64_aarch64-linux-gnu.tar.xz --strip-components 1
        fi
    fi
    echo
fi

echo "In a case you have local changes you may reset them with ./apply_patches.sh $1 reset"
echo
# Clone L4T kernel source repo
cd $DEVDIR

# Check if local tar ball exists in /home/nvidia_sources_cache
NVIDIA_CACHE_DIR="/home/nvidia_sources_cache"
TARBALL_NAME="backup_sources_$1.tar.gz"
TARBALL_PATH="$NVIDIA_CACHE_DIR/$TARBALL_NAME"

if [[ -f "$TARBALL_PATH" ]]; then
    echo "Found local tar ball: $TARBALL_PATH"
    echo "Extracting sources from local cache instead of cloning from NVIDIA repository..."
    
    # Remove existing sources directory if it exists
    if [[ -d "sources_$1" ]]; then
        echo "Removing existing sources_$1 directory..."
        rm -rf "sources_$1"
    fi
    
    # Extract tar ball
    echo "Extracting $TARBALL_NAME..."
    tar -xzf "$TARBALL_PATH"
    
    # Check what directory was extracted and rename if necessary
    # The tar ball might contain sources_6.x instead of sources_6.0
    EXTRACTED_DIR=$(tar -tzf "$TARBALL_PATH" | head -1 | cut -d'/' -f1)
    if [[ "$EXTRACTED_DIR" != "sources_$1" ]]; then
        echo "Renaming extracted directory from $EXTRACTED_DIR to sources_$1..."
        mv "$EXTRACTED_DIR" "sources_$1"
    fi
    
    echo "Sources extracted successfully from local cache."
else
    echo "Local tar ball not found at $TARBALL_PATH"
    echo "Cloning sources from NVIDIA repository..."
    if [[ -f "./scripts/source_sync_$1.sh" ]]; then
	"./scripts/source_sync_$1.sh" -t "$L4T_VERSION" -d "sources_$1"
    elif [[ -f "./scripts/source_sync_$JETPACK_VERSION.sh" ]]; then
        ./scripts/source_sync_$JETPACK_VERSION.sh -t $L4T_VERSION -d sources_$1
    fi
fi

# copy Makefile for jp6
if [[ "$JETPACK_VERSION" == "6.x" ]]; then
    cp ./nvidia-oot/Makefile "sources_$1/"
    cp ./kernel/kernel-jammy-src/Makefile "sources_$1/kernel"
fi

# remove BUILD_NUMBER env dependency kernel vermagic
if [[ "${JETPACK_VERSION}" == "4.6.1" ]]; then
    sed -i s/'UTS_RELEASE=\$(KERNELRELEASE)-ab\$(BUILD_NUMBER)'/'UTS_RELEASE=\$(KERNELRELEASE)'/g ./sources_$1/kernel/kernel-4.9/Makefile
    sed -i 's/the-space :=/E =/g' ./sources_$1/kernel/kernel-4.9/scripts/Kbuild.include
    sed -i 's/the-space += /the-space = \$E \$E/g' ./sources_$1/kernel/kernel-4.9/scripts/Kbuild.include
fi