339 lines
9.5 KiB
Bash
Executable File
339 lines
9.5 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
#
|
|
# Start a fullnode
|
|
#
|
|
here=$(dirname "$0")
|
|
# shellcheck source=multinode-demo/common.sh
|
|
source "$here"/common.sh
|
|
|
|
# shellcheck source=scripts/oom-score-adj.sh
|
|
source "$here"/../scripts/oom-score-adj.sh
|
|
|
|
fullnode_usage() {
|
|
if [[ -n $1 ]]; then
|
|
echo "$*"
|
|
echo
|
|
fi
|
|
cat <<EOF
|
|
usage: $0 [--blockstream PATH] [--init-complete-file FILE] [--label LABEL] [--stake LAMPORTS] [--no-voting] [--rpc-port port] [rsync network path to bootstrap leader configuration] [cluster entry point]
|
|
|
|
Start a full node
|
|
|
|
--blockstream PATH - open blockstream at this unix domain socket location
|
|
--init-complete-file FILE - create this file, if it doesn't already exist, once node initialization is complete
|
|
--label LABEL - Append the given label to the fullnode configuration files, useful when running
|
|
multiple fullnodes from the same filesystem location
|
|
--stake LAMPORTS - Number of lamports to stake
|
|
--no-voting - start node without vote signer
|
|
--rpc-port port - custom RPC port for this node
|
|
|
|
EOF
|
|
exit 1
|
|
}
|
|
|
|
find_leader() {
|
|
declare leader leader_address
|
|
declare shift=0
|
|
|
|
if [[ -z $1 ]]; then
|
|
leader=$PWD # Default to local tree for rsync
|
|
leader_address=127.0.0.1:8001 # Default to local leader
|
|
elif [[ -z $2 ]]; then
|
|
leader=$1
|
|
leader_address=$leader:8001
|
|
shift=1
|
|
else
|
|
leader=$1
|
|
leader_address=$2
|
|
shift=2
|
|
fi
|
|
|
|
echo "$leader" "$leader_address" "$shift"
|
|
}
|
|
|
|
rsync_url() { # adds the 'rsync://` prefix to URLs that need it
|
|
declare url="$1"
|
|
|
|
if [[ $url =~ ^.*:.*$ ]]; then
|
|
# assume remote-shell transport when colon is present, use $url unmodified
|
|
echo "$url"
|
|
return 0
|
|
fi
|
|
|
|
if [[ -d $url ]]; then
|
|
# assume local directory if $url is a valid directory, use $url unmodified
|
|
echo "$url"
|
|
return 0
|
|
fi
|
|
|
|
# Default to rsync:// URL
|
|
echo "rsync://$url"
|
|
}
|
|
|
|
airdrop() {
|
|
declare keypair_file=$1
|
|
declare entrypoint_ip=$2
|
|
declare amount=$3
|
|
|
|
declare address
|
|
address=$($solana_wallet --keypair "$keypair_file" address)
|
|
|
|
# TODO: Until https://github.com/solana-labs/solana/issues/2355 is resolved
|
|
# a fullnode needs N lamports as its vote account gets re-created on every
|
|
# node restart, costing it lamports
|
|
declare retries=5
|
|
|
|
while ! $solana_wallet --keypair "$keypair_file" --url "http://$entrypoint_ip:8899" airdrop "$amount"; do
|
|
|
|
# TODO: Consider moving this retry logic into `solana-wallet airdrop`
|
|
# itself, currently it does not retry on "Connection refused" errors.
|
|
((retries--))
|
|
if [[ $retries -le 0 ]]; then
|
|
echo "Airdrop to $address failed."
|
|
return 1
|
|
fi
|
|
echo "Airdrop to $address failed. Remaining retries: $retries"
|
|
sleep 1
|
|
done
|
|
|
|
return 0
|
|
}
|
|
|
|
setup_vote_account() {
|
|
declare entrypoint_ip=$1
|
|
declare node_id_path=$2
|
|
declare vote_id_path=$3
|
|
declare stake=$4
|
|
|
|
declare node_id
|
|
node_id=$($solana_wallet --keypair "$node_id_path" address)
|
|
|
|
declare vote_id
|
|
vote_id=$($solana_wallet --keypair "$vote_id_path" address)
|
|
|
|
if [[ -f "$vote_id_path".configured ]]; then
|
|
echo "Vote account has already been configured"
|
|
else
|
|
airdrop "$node_id_path" "$entrypoint_ip" "$stake" || return $?
|
|
|
|
# Fund the vote account from the node, with the node as the node_id
|
|
$solana_wallet --keypair "$node_id_path" --url "http://$entrypoint_ip:8899" \
|
|
create-vote-account "$vote_id" "$node_id" $((stake - 1)) || return $?
|
|
|
|
touch "$vote_id_path".configured
|
|
fi
|
|
|
|
$solana_wallet --keypair "$node_id_path" --url "http://$entrypoint_ip:8899" \
|
|
show-vote-account "$vote_id"
|
|
return 0
|
|
}
|
|
|
|
ledger_not_setup() {
|
|
echo "Error: $*"
|
|
echo
|
|
echo "Please run: ${here}/setup.sh"
|
|
exit 1
|
|
}
|
|
|
|
default_fullnode_arg() {
|
|
declare name=$1
|
|
declare value=$2
|
|
|
|
for arg in "${extra_fullnode_args[@]}"; do
|
|
if [[ $arg = "$name" ]]; then
|
|
return
|
|
fi
|
|
done
|
|
|
|
if [[ -n $value ]]; then
|
|
extra_fullnode_args+=("$name" "$value")
|
|
else
|
|
extra_fullnode_args+=("$name")
|
|
fi
|
|
}
|
|
|
|
extra_fullnode_args=()
|
|
bootstrap_leader=false
|
|
stake=43 # number of lamports to assign as stake (plus transaction fee to setup the stake)
|
|
poll_for_new_genesis_block=0
|
|
label=
|
|
|
|
positional_args=()
|
|
while [[ -n $1 ]]; do
|
|
if [[ ${1:0:1} = - ]]; then
|
|
if [[ $1 = --label ]]; then
|
|
label="-$2"
|
|
shift 2
|
|
elif [[ $1 = --bootstrap-leader ]]; then
|
|
bootstrap_leader=true
|
|
shift
|
|
elif [[ $1 = --poll-for-new-genesis-block ]]; then
|
|
poll_for_new_genesis_block=1
|
|
shift
|
|
elif [[ $1 = --blockstream ]]; then
|
|
stake=0
|
|
extra_fullnode_args+=("$1" "$2")
|
|
shift 2
|
|
elif [[ $1 = --enable-rpc-exit ]]; then
|
|
extra_fullnode_args+=("$1")
|
|
shift
|
|
elif [[ $1 = --init-complete-file ]]; then
|
|
extra_fullnode_args+=("$1" "$2")
|
|
shift 2
|
|
elif [[ $1 = --stake ]]; then
|
|
stake="$2"
|
|
shift 2
|
|
elif [[ $1 = --no-voting ]]; then
|
|
extra_fullnode_args+=("$1")
|
|
shift
|
|
elif [[ $1 = --no-sigverify ]]; then
|
|
extra_fullnode_args+=("$1")
|
|
shift
|
|
elif [[ $1 = --rpc-port ]]; then
|
|
extra_fullnode_args+=("$1" "$2")
|
|
shift 2
|
|
elif [[ $1 = --dynamic-port-range ]]; then
|
|
extra_fullnode_args+=("$1" "$2")
|
|
shift 2
|
|
elif [[ $1 = --gossip-port ]]; then
|
|
extra_fullnode_args+=("$1" "$2")
|
|
shift 2
|
|
elif [[ $1 = -h ]]; then
|
|
fullnode_usage "$@"
|
|
else
|
|
echo "Unknown argument: $1"
|
|
exit 1
|
|
fi
|
|
else
|
|
positional_args+=("$1")
|
|
shift
|
|
fi
|
|
done
|
|
|
|
if $bootstrap_leader; then
|
|
if [[ ${#positional_args[@]} -ne 0 ]]; then
|
|
fullnode_usage "Unknown argument: ${positional_args[0]}"
|
|
fi
|
|
|
|
[[ -f "$SOLANA_CONFIG_DIR"/bootstrap-leader-id.json ]] ||
|
|
ledger_not_setup "$SOLANA_CONFIG_DIR/bootstrap-leader-id.json not found"
|
|
|
|
$solana_ledger_tool --ledger "$SOLANA_CONFIG_DIR"/bootstrap-leader-ledger verify
|
|
|
|
fullnode_id_path="$SOLANA_CONFIG_DIR"/bootstrap-leader-id.json
|
|
fullnode_vote_id_path="$SOLANA_CONFIG_DIR"/bootstrap-leader-vote-id.json
|
|
ledger_config_dir="$SOLANA_CONFIG_DIR"/bootstrap-leader-ledger
|
|
accounts_config_dir="$SOLANA_CONFIG_DIR"/bootstrap-leader-accounts
|
|
|
|
default_fullnode_arg --rpc-port 8899
|
|
default_fullnode_arg --rpc-drone-address 127.0.0.1:9900
|
|
default_fullnode_arg --gossip-port 8001
|
|
else
|
|
|
|
if [[ ${#positional_args[@]} -gt 2 ]]; then
|
|
fullnode_usage "$@"
|
|
fi
|
|
|
|
read -r leader leader_address shift < <(find_leader "${positional_args[@]}")
|
|
shift "$shift"
|
|
|
|
fullnode_id_path=$SOLANA_CONFIG_DIR/fullnode-id$label.json
|
|
fullnode_vote_id_path=$SOLANA_CONFIG_DIR/fullnode-vote-id$label.json
|
|
ledger_config_dir=$SOLANA_CONFIG_DIR/fullnode-ledger$label
|
|
accounts_config_dir=$SOLANA_CONFIG_DIR/fullnode-accounts$label
|
|
|
|
mkdir -p "$SOLANA_CONFIG_DIR"
|
|
[[ -r "$fullnode_id_path" ]] || $solana_keygen -o "$fullnode_id_path"
|
|
[[ -r "$fullnode_vote_id_path" ]] || $solana_keygen -o "$fullnode_vote_id_path"
|
|
|
|
default_fullnode_arg --entrypoint "$leader_address"
|
|
default_fullnode_arg --rpc-drone-address "${leader_address%:*}:9900"
|
|
fi
|
|
|
|
fullnode_id=$($solana_keygen pubkey "$fullnode_id_path")
|
|
fullnode_vote_id=$($solana_keygen pubkey "$fullnode_vote_id_path")
|
|
|
|
cat <<EOF
|
|
======================[ Fullnode configuration ]======================
|
|
node id: $fullnode_id
|
|
vote id: $fullnode_vote_id
|
|
ledger: $ledger_config_dir
|
|
accounts: $accounts_config_dir
|
|
======================================================================
|
|
EOF
|
|
|
|
if [[ -z $CI ]]; then # Skip in CI
|
|
# shellcheck source=scripts/tune-system.sh
|
|
source "$here"/../scripts/tune-system.sh
|
|
fi
|
|
|
|
default_fullnode_arg --identity "$fullnode_id_path"
|
|
default_fullnode_arg --voting-keypair "$fullnode_vote_id_path"
|
|
default_fullnode_arg --vote-account "$fullnode_vote_id"
|
|
default_fullnode_arg --ledger "$ledger_config_dir"
|
|
default_fullnode_arg --accounts "$accounts_config_dir"
|
|
|
|
if [[ -n $SOLANA_CUDA ]]; then
|
|
program=$solana_fullnode_cuda
|
|
else
|
|
program=$solana_fullnode
|
|
fi
|
|
|
|
set -e
|
|
secs_to_next_genesis_poll=0
|
|
PS4="$(basename "$0"): "
|
|
while true; do
|
|
if [[ ! -d "$SOLANA_RSYNC_CONFIG_DIR"/ledger ]]; then
|
|
if $bootstrap_leader; then
|
|
ledger_not_setup "$SOLANA_RSYNC_CONFIG_DIR/ledger does not exist"
|
|
fi
|
|
rsync_leader_url=$(rsync_url "$leader")
|
|
$rsync -vPr "$rsync_leader_url"/config/ledger "$SOLANA_RSYNC_CONFIG_DIR"
|
|
fi
|
|
|
|
if [[ ! -d "$ledger_config_dir" ]]; then
|
|
cp -a "$SOLANA_RSYNC_CONFIG_DIR"/ledger/ "$ledger_config_dir"
|
|
$solana_ledger_tool --ledger "$ledger_config_dir" verify
|
|
fi
|
|
|
|
trap '[[ -n $pid ]] && kill "$pid" >/dev/null 2>&1 && wait "$pid"' INT TERM ERR
|
|
|
|
if ! $bootstrap_leader && ((stake)); then
|
|
setup_vote_account "${leader_address%:*}" "$fullnode_id_path" "$fullnode_vote_id_path" "$stake"
|
|
fi
|
|
|
|
echo "$PS4$program ${extra_fullnode_args[*]}"
|
|
$program "${extra_fullnode_args[@]}" > >($fullnode_logger) 2>&1 &
|
|
pid=$!
|
|
oom_score_adj "$pid" 1000
|
|
|
|
if $bootstrap_leader; then
|
|
wait "$pid"
|
|
sleep 1
|
|
else
|
|
while true; do
|
|
if ! kill -0 "$pid"; then
|
|
wait "$pid"
|
|
exit 0
|
|
fi
|
|
|
|
sleep 1
|
|
|
|
((poll_for_new_genesis_block)) || continue
|
|
((secs_to_next_genesis_poll--)) && continue
|
|
|
|
$rsync -r "$rsync_leader_url"/config/ledger "$SOLANA_RSYNC_CONFIG_DIR" || true
|
|
diff -q "$SOLANA_RSYNC_CONFIG_DIR"/ledger/genesis.json "$ledger_config_dir"/genesis.json >/dev/null 2>&1 || break
|
|
secs_to_next_genesis_poll=60
|
|
|
|
done
|
|
|
|
echo "############## New genesis detected, restarting fullnode ##############"
|
|
kill "$pid" || true
|
|
wait "$pid" || true
|
|
rm -rf "$ledger_config_dir" "$accounts_config_dir" "$fullnode_vote_id_path".configured
|
|
sleep 60 # give the network time to come back up
|
|
fi
|
|
|
|
done
|