103 lines
2.4 KiB
Bash
Executable File
103 lines
2.4 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
|
|
# (1) Primary setup
|
|
#--------------------------------------------------------#
|
|
# (1) Get current ABS directory
|
|
ROOT=$(dirname `realpath $0`);
|
|
|
|
# (2) Check required arguments
|
|
test $# -lt 4 && echo -e "ERR: Missing arguments\n\nARGUMENTS:\n\tname\t\tThe name of the machine\n\ttarget_host\tThe hostname (or ip) of the target\n\tout_port\tThe port you want to use to send\n\tin_port\t\tThe port the peer uses to listen\n\ttrigger_port\tOptional param. Used to trigger the peer connection\n" && exit 1;
|
|
|
|
# (3) Manage optional param #
|
|
TRIGGER="";
|
|
test $# -ge 5 && TRIGGER="$5";
|
|
|
|
|
|
# (2) Secondary setup
|
|
#--------------------------------------------------------#
|
|
# (1) Reset configuration file
|
|
test -f $ROOT/config && rm $ROOT/config;
|
|
|
|
# (2) Create buffer paths
|
|
BUF_IN="/tmp/buf_in_$4";
|
|
BUF_OUT="/tmp/buf_out_$2_$3";
|
|
|
|
# (3) Flush buffers
|
|
echo -ne "" > $BUF_IN;
|
|
echo -ne "" > $BUF_OUT;
|
|
|
|
|
|
# (3) Launch INPUT server
|
|
#--------------------------------------------------------#
|
|
# (1) Start the script in background
|
|
(
|
|
|
|
infail=0;
|
|
|
|
# kill script after 10 failures
|
|
while [ $infail -lt 10 ]; do
|
|
|
|
# main command
|
|
nc -lp $4 >> $BUF_IN && infail=0 || infail=`expr $infail + 1`;
|
|
|
|
# try to kill proccess that uses port if failed and such proccess exists
|
|
if [ $infail -gt 0 ]; then
|
|
LISTEN_PID=`lsof -n -i4TCP:$4 | grep LISTEN | awk '{print $2}'`;
|
|
test -z "$LISTEN_PID" && kill -9 $LISTEN_PID;
|
|
fi;
|
|
|
|
done;
|
|
|
|
)&
|
|
|
|
# (2) Store the background proccess PID #
|
|
IN_PID=$!;
|
|
|
|
|
|
|
|
|
|
|
|
################################
|
|
# IF SET, SEND TRIGGER TO PEER #
|
|
################################
|
|
test ! -z "$TRIGGER" && bash -c "exec 3<>/dev/tcp/$2/$TRIGGER";
|
|
|
|
##################################
|
|
# WAIT 2s FOR THE PEER TO LISTEN #
|
|
##################################
|
|
sleep 2;
|
|
|
|
|
|
|
|
|
|
|
|
# (4) Launch OUTPUT server
|
|
#--------------------------------------------------------#
|
|
# (1) Launch output server
|
|
(
|
|
|
|
outfail=0;
|
|
|
|
# stop after 20 failed attempts
|
|
while [ $outfail -lt 20 ]; do
|
|
|
|
# main command
|
|
tail -f $BUF_OUT | nc $2 $3 && outfail=0 || outfail=`expr $outfail + 1`;
|
|
|
|
sleep 1; # can sleep because the buffer is never flushed
|
|
|
|
done&
|
|
|
|
)&
|
|
|
|
# (2) Store the background proccess PID #
|
|
OUT_PID=$!;
|
|
|
|
|
|
|
|
# (5) Store the config file for 'write' + 'send' to work properly
|
|
#--------------------------------------------------------#
|
|
# (1) Store config file in 'config_$NAME'
|
|
echo -e "#!/bin/bash\nBUF_IN=\"$BUF_IN\"; BUF_OUT=\"$BUF_OUT\";\nIN_PID=$IN_PID; OUT_PID=$OUT_PID;\n" > $ROOT/config_$1;
|