#!/bin/sh
#
# Copyright (C) 2000-2026 Kern Sibbald
# License: BSD 2-Clause; see file LICENSE-FOSS
#
# Run a simple backup of the Bacula build directory
#   then restore it.
#
TestName="source-addr-test"
JobName=backup
. scripts/functions

scripts/cleanup
scripts/copy-confs

#
# Zap out any schedule in default conf file so that
#  it doesn't start during our test
#
outf="tmp/sed_tmp"
echo "s%  Schedule =%# Schedule =%g" >${outf}
cp $scripts/bacula-dir.conf $tmp/1
sed -f ${outf} $tmp/1 >$scripts/bacula-dir.conf

PATH=$PATH:/sbin:/usr/sbin
#
# Deduce a primary IP address by examining
#  the interface defined by INTERFACE=
#
os=`uname`
INTERFACE=lo
if [ $os = 'FreeBSD' ]; then
   INTERFACE=bge0
   IP=`ifconfig ${INTERFACE} | perl -ne '/inet (.+?) / && print $1'`
elif [ $os = 'Linux' ]; then
    #  INTERFACE=eth0
    which ifconfig > /dev/null
    if [ $? -ne 0 ] ; then
        which ip > /dev/null
        if [ $? -ne 0 ] ; then
            echo "Can't find ifconfig or ip command"
            exit 0
        fi
        IP=`ip address show lo | grep -Po 'inet \K[\d.]+'`
    else
        INTERFACE=lo
        IP=`ifconfig ${INTERFACE} | perl -ne '/inet (addr:)?(.+?) / && print $2'`
    fi
fi

if [ "$IP" = '' ]; then
    echo "Can't detect ip address"
    exit 0
fi

outf="tmp/sed_tmp"
echo "s/PidDirectory/DirSourceAddress=$IP; PidDirectory/g" > $outf
echo "s/Address =/Address = \"127.0.0.1\" #/g" >> $outf 
cp $scripts/bacula-dir.conf $tmp/1
sed -f ${outf} ${cwd}/tmp/1 > $scripts/bacula-dir.conf 

# Set source address on the FD as well
echo "s/Pid Directory/FDSourceAddress=$IP; Pid Directory/g" >$outf
cp $scripts/bacula-fd.conf $tmp/2
sed -f ${outf} ${cwd}/tmp/2 > $scripts/bacula-fd.conf

DIRPORT=$BASEPORT
FDPORT=$(($BASEPORT + 1))
SDPORT=$(($BASEPORT + 2))

change_jobname BackupClient1 $JobName
start_test

$bperl -e "add_attribute('$conf/bacula-sd.conf',  'CommCompression', 'no', 'Storage')"
$bperl -e "add_attribute('$conf/bacula-dir.conf', 'CommCompression', 'no', 'Director')"
$bperl -e "add_attribute('$conf/bacula-fd.conf',  'CommCompression', 'no', 'FileDaemon')"

cat <<END_OF_DATA >${cwd}/tmp/bconcmds
@output /dev/null
messages
@$out ${cwd}/tmp/log1.out
@#label volume=TestVolume001 storage=File1 pool=File slot=1 drive=0
@#setdebug level=100 storage=File1
setbandwidth client limit="1000 kb/s"
run job=$JobName yes
@sleep 3
messages
quit
END_OF_DATA

run_bacula

print_debug "Detecting connections to SD sourced from ${IP}"
which netstat > /dev/null
if [ $? -ne 0 ] ; then
    which ss > /dev/null
    if [ $? -ne 0 ] ; then
        echo "Can't find netstat or ss command"
        exit 0
    fi
    ss -anut | grep -E "127\.0\.0\.1(.|:)$SDPORT .* ${IP}" > $tmp/log3.out
    if [ $? -ne 0 ]; then
        stat=2
        print_debug "Couldn't detect a connection to the SD sourced from SourceAddress ${IP}" 
    fi
    print_debug "and connections to FD sourced from ${IP}"
    ss -anut | grep -E "127\.0\.0\.1(.|:)$SDPORT .* ${IP}" > $tmp/log3.out
    if [ $? -ne 0 ]; then
        stat=2
        print_debug "Couldn't detect a connection to the FD sourced from SourceAddress ${IP}"
    fi
    if [ "$stat" = 2 ] ; then
        ss -anut
    fi
else
    netstat -tan | grep -E "127\.0\.0\.1(.|:)$SDPORT .* ${IP}" > $tmp/log3.out
    if [ $? -ne 0 ]; then
        stat=2
        print_debug "Couldn't detect a connection to the SD sourced from SourceAddress ${IP}" 
    fi
    print_debug "and connections to FD sourced from ${IP}"
    netstat -tan | grep -E "127\.0\.0\.1(.|:)$FDPORT .* ${IP}" >> $tmp/log3.out
    if [ $? -ne 0 ]; then
        stat=2
        print_debug "Couldn't detect a connection to the FD sourced from SourceAddress ${IP}"
    fi

    if [ "$stat" = 2 ] ; then
        netstat -tan
    fi
fi

if [ "$debug" = 1 ] ; then
    cat $tmp/log3.out
fi

cat <<END_OF_DATA >${cwd}/tmp/bconcmds
@$out $tmp/log1.out
setbandwidth client limit="1000000 kb/s"
wait
messages
@# 
@# now do a restore
@#
@$out ${cwd}/tmp/log2.out  
restore where=${cwd}/tmp/bacula-restores select all done
yes
wait
messages
quit
END_OF_DATA

run_bconsole
check_for_zombie_jobs storage=File1
stop_bacula
check_two_logs
check_restore_diff

if [ "$stat" = 2 ]; then
    print_debug `cat $tmp/log3.out`
    bstat=$stat
fi

end_test
