#!/bin/bash
# SPDX-License-Identifier: BSD-3-Clause OR Apache-2.0

set -euo pipefail

# uD3TN + ION Interoperability test
# It is recommended to run this script in a Docker image based on the
# `ion-interop` Dockerfile provided with uD3TN as follows:
# $ docker run -it -v "$(pwd):/ud3tn" ud3tn-ion-interop:<tag> \
#   bash -c '/ud3tn/test/dockerfiles/prepare_for_test.sh /ud3tn /ud3tn_build "CFLAGS=-DCLA_MTCP_ION_STCP_COMPATIBILITY=1" && cd /ud3tn_build && source /ud3tn_venv/bin/activate && test/ion_interoperability/mtcp_minimal_forwarding_test/run.sh'

BP_VERSION=7
ION_SCRIPT_NAME=ipn.rc
ION_EID=ipn:1.0
UD3TN1_EID=ipn:2.0
UD3TN2_EID=ipn:3.0
SOURCE_AGENTID=1
SINK_AGENTID=1
SINK_EID=ipn:3.1
ION_SINK_EID=ipn:1.1

# This assumes you are running the command from within the "ud3tn" directory.
UD3TN_DIR="$(pwd)"
cd "$UD3TN_DIR"

exit_handler() {
    cd "$UD3TN_DIR"

    kill -TERM $UD3TN1_PID || true
    kill -TERM $UD3TN2_PID || true

    echo "Terminating ION (timeout 20s)..."
    (ionstop || true) &
    sleep 20

    echo
    echo ">>> ION LOGFILE"
    cat "ion.log" || true
    echo
    echo ">>> uD3TN1 LOGFILE"
    cat "/tmp/ud3tn1.log" || true
    echo
    echo ">>> uD3TN2 LOGFILE"
    cat "/tmp/ud3tn2.log" || true
    echo

    echo "Waiting for uD3TN to exit gracefully - if it doesn't, check for sanitizer warnings."
    wait $UD3TN1_PID
    wait $UD3TN2_PID
}

rm -f ion.log
rm -f /tmp/ion*log /tmp/ud3tn*.log

# Start first uD3TN instance (uD3TN1)
"$UD3TN_DIR/build/posix/ud3tn" -s "$UD3TN_DIR/ud3tn1.socket" -S "$UD3TN_DIR/ud3tn1.aap2.socket" -c "sqlite:ud3tn1.sqlite;mtcp:127.0.0.1,7002" -b $BP_VERSION -e "$UD3TN1_EID" -L 4 --allow-remote-config > /tmp/ud3tn1.log 2>&1 &
UD3TN1_PID=$!

# Start second uD3TN instance (uD3TN2)
"$UD3TN_DIR/build/posix/ud3tn" -s "$UD3TN_DIR/ud3tn2.socket" -S "$UD3TN_DIR/ud3tn2.aap2.socket" -c "sqlite:ud3tn2.sqlite;mtcp:127.0.0.1,7003" -b $BP_VERSION -e "$UD3TN2_EID" -L 4 --allow-remote-config > /tmp/ud3tn2.log 2>&1 &
UD3TN2_PID=$!

# Start ION instance
ulimit -n 512 # fix behavior on systems with a huge limit (e.g. if the container runtime does not change the kernel default), see: #121
ionstart -I test/ion_interoperability/mtcp_minimal_forwarding_test/$ION_SCRIPT_NAME

# UD3TN1_PID and UD3TN2_PID must be defined for this
trap exit_handler EXIT

# Configure a contact to ION in uD3TN1 which allows to reach uD3TN2
sleep 0.5
aap2-configure-link --socket "$UD3TN_DIR/ud3tn1.aap2.socket" "$ION_EID" mtcp:127.0.0.1:7001
aap2-configure-link --socket "$UD3TN_DIR/ud3tn1.aap2.socket" "$UD3TN2_EID" mtcp:127.0.0.1:7001

# Send a bundle to uD3TN1, addressed to uD3TN2
PAYLOAD="THISISTHEBUNDLEPAYLOAD"
(sleep 0.5 && aap2-send --socket "$UD3TN_DIR/ud3tn1.aap2.socket" --agentid "$SOURCE_AGENTID" "$SINK_EID" "$PAYLOAD") &

timeout -v 10 stdbuf -oL aap2-receive --socket "$UD3TN_DIR/ud3tn2.aap2.socket" --agentid "$SINK_AGENTID" --count 1 --verify-pl "$PAYLOAD" --newline -vv

# Test forwarding from ION directly
(sleep 0.5 && bpsource "$SINK_EID" "$PAYLOAD") &
timeout -v 10 stdbuf -oL aap2-receive --socket "$UD3TN_DIR/ud3tn2.aap2.socket" --agentid "$SINK_AGENTID" --count 1 --verify-pl "$PAYLOAD" --newline -vv

(sleep 0.5 && aap2-send --socket "$UD3TN_DIR/ud3tn1.aap2.socket" --agentid "$SOURCE_AGENTID" "$ION_SINK_EID" "$PAYLOAD") &
timeout -v 10 bprecvfile "$ION_SINK_EID" 1
RECEIVED_PAYLOAD="$(cat testfile1)"
echo "bprecvfile: $RECEIVED_PAYLOAD"
if [[ "$RECEIVED_PAYLOAD" != "$PAYLOAD" ]]; then
    echo "Received payload does not match: $RECEIVED_PAYLOAD"
    exit 1
fi