zeus
4 years ago
7 changed files with 265 additions and 7 deletions
@ -0,0 +1,52 @@ |
|||
#!/bin/bash |
|||
# SPDX-License-Identifier: GPL-2.0 |
|||
# |
|||
# Copyright (C) 2015-2020 Jason A. Donenfeld <Jason@zx2c4.com>. All Rights Reserved. |
|||
|
|||
exec < <(exec wg show all dump) |
|||
|
|||
printf '{' |
|||
while read -r -d $'\t' device; do |
|||
if [[ $device != "$last_device" ]]; then |
|||
[[ -z $last_device ]] && printf '\n' || printf '%s,\n' "$end" |
|||
last_device="$device" |
|||
read -r private_key public_key listen_port fwmark |
|||
printf '\t"%s": {' "$device" |
|||
delim=$'\n' |
|||
[[ $private_key == "(none)" ]] || { printf '%s\t\t"privateKey": "%s"' "$delim" "$private_key"; delim=$',\n'; } |
|||
[[ $public_key == "(none)" ]] || { printf '%s\t\t"publicKey": "%s"' "$delim" "$public_key"; delim=$',\n'; } |
|||
[[ $listen_port == "0" ]] || { printf '%s\t\t"listenPort": %u' "$delim" $(( $listen_port )); delim=$',\n'; } |
|||
[[ $fwmark == "off" ]] || { printf '%s\t\t"fwmark": %u' "$delim" $(( $fwmark )); delim=$',\n'; } |
|||
printf '%s\t\t"peers": {' "$delim"; end=$'\n\t\t}\n\t}' |
|||
delim=$'\n' |
|||
else |
|||
read -r public_key preshared_key endpoint allowed_ips latest_handshake transfer_rx transfer_tx persistent_keepalive |
|||
printf '%s\t\t\t"%s": {' "$delim" "$public_key" |
|||
delim=$'\n' |
|||
[[ $preshared_key == "(none)" ]] || { printf '%s\t\t\t\t"presharedKey": "%s"' "$delim" "$preshared_key"; delim=$',\n'; } |
|||
[[ $endpoint == "(none)" ]] || { printf '%s\t\t\t\t"endpoint": "%s"' "$delim" "$endpoint"; delim=$',\n'; } |
|||
[[ $latest_handshake == "0" ]] || { printf '%s\t\t\t\t"latestHandshake": %u' "$delim" $(( $latest_handshake )); delim=$',\n'; } |
|||
[[ $transfer_rx == "0" ]] || { printf '%s\t\t\t\t"transferRx": %u' "$delim" $(( $transfer_rx )); delim=$',\n'; } |
|||
[[ $transfer_tx == "0" ]] || { printf '%s\t\t\t\t"transferTx": %u' "$delim" $(( $transfer_tx )); delim=$',\n'; } |
|||
[[ $persistent_keepalive == "off" ]] || { printf '%s\t\t\t\t"persistentKeepalive": %u' "$delim" $(( $persistent_keepalive )); delim=$',\n'; } |
|||
printf '%s\t\t\t\t"allowedIps": [' "$delim" |
|||
delim=$'\n' |
|||
if [[ $allowed_ips != "(none)" ]]; then |
|||
old_ifs="$IFS" |
|||
IFS=, |
|||
for ip in $allowed_ips; do |
|||
printf '%s\t\t\t\t\t"%s"' "$delim" "$ip" |
|||
delim=$',\n' |
|||
done |
|||
IFS="$old_ifs" |
|||
delim=$'\n' |
|||
fi |
|||
printf '%s\t\t\t\t]' "$delim" |
|||
printf '\n\t\t\t}' |
|||
delim=$',\n' |
|||
fi |
|||
|
|||
|
|||
done |
|||
printf '%s\n' "$end" |
|||
printf '}\n' |
|||
@ -0,0 +1,127 @@ |
|||
#!/bin/bash |
|||
|
|||
|
|||
SOURCE="${BASH_SOURCE[0]}" |
|||
while [ -h "$SOURCE" ]; do # resolve $SOURCE until the file is no longer a symlink |
|||
TARGET="$(readlink "$SOURCE")" |
|||
if [[ $TARGET == /* ]]; then |
|||
#echo "SOURCE '$SOURCE' is an absolute symlink to '$TARGET'" |
|||
SOURCE="$TARGET" |
|||
else |
|||
DIR="$( dirname "$SOURCE" )" |
|||
#echo "SOURCE '$SOURCE' is a relative symlink to '$TARGET' (relative to '$DIR')" |
|||
SOURCE="$DIR/$TARGET" # if $SOURCE was a relative symlink, we need to resolve it relative to the path where the symlink file was located |
|||
fi |
|||
done |
|||
|
|||
SRPATH="$( dirname "$SOURCE" )" |
|||
SFPATH="$( cd -P "$( dirname "$SOURCE" )" && pwd )" |
|||
if [ "$SFPATH" != "$SRPATH" ]; then |
|||
RDIR=$SRPATH # relativ path directory |
|||
fi |
|||
|
|||
cwdir=$PWD |
|||
wdir=$SFPATH |
|||
|
|||
if [ "$cwdir" != "$wdir" ]; then |
|||
cd $wdir |
|||
fi |
|||
|
|||
# |
|||
#files/mongo/ |
|||
# docker-compose.yml |
|||
# .env |
|||
# logs |
|||
# mongod.conf |
|||
# package.json |
|||
# venus-stats.js |
|||
# wg-json |
|||
# |
|||
|
|||
function EPHEMERAL_PORT() { |
|||
LOW_BOUND=49152 |
|||
RANGE=16384 |
|||
while true; do |
|||
CANDIDATE=$[$LOW_BOUND + ($RANDOM % $RANGE)] |
|||
(echo "" >/dev/tcp/127.0.0.1/${CANDIDATE}) >/dev/null 2>&1 |
|||
if [ $? -ne 0 ]; then |
|||
echo $CANDIDATE |
|||
break |
|||
fi |
|||
done |
|||
} |
|||
|
|||
if [ ! -d $wdir/hybrid/venus-stats ]; then |
|||
mkdir -p $wdir/hybrid/venus-stats |
|||
fi |
|||
|
|||
cp -raf $wdir/files/mongo/wg-json $wdir/hybrid/venus-stats |
|||
cp -raf $wdir/files/mongo/mongod.conf $wdir/hybrid/venus-stats |
|||
cp -raf $wdir/files/mongo/docker-compose.yml $wdir/hybrid/venus-stats |
|||
cp -raf $wdir/files/mongo/files/mongo/venus-stats.js $wdir/hybrid/venus-stats |
|||
|
|||
|
|||
swuser=swarmlab |
|||
swpass=$(tr -dc A-Za-z0-9 </dev/urandom | head -c 13 ; echo '') |
|||
db=venus |
|||
conf=$wdir/hybrid/venus-stats/mongod.conf |
|||
port=$(EPHEMERAL_PORT) |
|||
coll=logs |
|||
ip='127.0.0.1' |
|||
|
|||
|
|||
cat << FOE > $wdir/hybrid/venus-stats/.env |
|||
#!/bin/bash |
|||
|
|||
MONGO_INITDB_ROOT_USERNAME=$swuser |
|||
MONGO_INITDB_ROOT_PASSWORD=$swpass |
|||
MONGO_INITDB_DATABASE=$db |
|||
|
|||
MONGO_VENUS_CONF=$conf |
|||
MONGO_VENUS_IP=$ip |
|||
MONGO_VENUS_PORT=$port |
|||
MONGO_VENUS_COLLECTION=$coll |
|||
FOE |
|||
|
|||
|
|||
cat << FOE > $wdir/hybrid/venus-stats/start |
|||
#!/bin/bash |
|||
|
|||
cd $wdir/hybrid/venus-stats; |
|||
|
|||
docker-compose down && |
|||
docker-compose rm && |
|||
docker-compose pull && |
|||
docker-compose build --no-cache && |
|||
docker-compose up -d --force-recreate |
|||
cd $wdir |
|||
FOE |
|||
|
|||
chmod +x $wdir/hybrid/venus-stats/start |
|||
|
|||
cat << FOE > $wdir/hybrid/venus-stats/stop |
|||
#!/bin/bash |
|||
|
|||
cd $wdir/hybrid/venus-stats; |
|||
|
|||
docker-compose down && |
|||
docker-compose rm |
|||
cd $wdir |
|||
FOE |
|||
|
|||
chmod +x $wdir/hybrid/venus-stats/stop |
|||
|
|||
|
|||
setconfig=$(jq \ |
|||
--arg key0 'mongo_user' --arg mongo_user $swuser \ |
|||
--arg key1 'mongo_pass' --arg mongo_pass $swpass \ |
|||
--arg key2 'mongo_db' --arg mongo_db $db \ |
|||
--arg key3 'mongo_ip' --arg mongo_ip $ip \ |
|||
--arg key4 'mongo_port' --arg mongo_port $port \ |
|||
--arg key5 'mongo_coll' --arg mongo_coll $coll \ |
|||
'. | .[$key0]=$mongo_user | .[$key1]=$mongo_pass | .[$key2]=$mongo_db | .[$key3]=$mongo_ip | .[$key4]=$mongo_port | .[$key5]=$mongo_coll' <<<'{}' |
|||
) |
|||
|
|||
echo $setconfig > $wdir/hybrid/venus-stats/config.json |
|||
|
|||
|
|||
Loading…
Reference in new issue