blob: f3dadeb5161c90d2666dfd520ff9647659c18e76 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
#!/usr/bin/env bash
set -xeuo pipefail
SCRIPT_NAME=$(basename "$0")
function usage()
{
cat <<EOF
${SCRIPT_NAME}: Launch a chutney network to test arti
Usage:
${SCRIPT_NAME} [modules] : Launch chutney, and the provided modules.
Options:
-h: Print this message.
-n <network>: Name of the chutney network to launch (default: basic)
Modules:
"proxy": Launch a arti-based SOCKS proxy.
EOF
}
PROXY=no
NETWORK="basic"
while getopts "hn:" opt ; do
case "$opt" in
h) usage
exit 0
;;
n) NETWORK="$OPTARG"
;;
*) echo "Unknown option. (Run $0 -h for usage)"
exit 1
;;
esac
done
# Remove the parsed flags
shift $((OPTIND-1))
for module in "$@"; do
case "$module" in
proxy) PROXY=yes
;;
*) echo "Unrecognized module. (Run $0 -h for usage)"
;;
esac
done
target="$NETWORK"
cd "$(git rev-parse --show-toplevel)"
echo "target=$target" > tests/chutney/arti.run
# Set and fully resolve chutney bin if not already set.
: "${CHUTNEY_BIN:=$(type -P chutney)}"
if [ -z "${CHUTNEY_BIN:-}" ]; then
echo "Couldn't locate chutney bin. Ensure it's on PATH or set CHUTNEY_BIN."
echo "You can install with:"
echo "python3 -m pip install git+https://gitlab.torproject.org/tpo/core/chutney.git"
exit 1
elif [ ! -x "$CHUTNEY_BIN" ]; then
echo "CHUTNEY_BIN='$CHUTNEY_BIN' doesn't exist or isn't executable"
exit 1
else
# CHUTNEY_BIN is set; tell the user so.
echo "Using chutney at '${CHUTNEY_BIN}'"
fi
# Use consistent CHUTNEY_BIN for all steps. We export it in case we end up
# calling other tools/scripts that directly look for chutney there.
echo "export CHUTNEY_BIN=\"$CHUTNEY_BIN\"" >> tests/chutney/arti.run
# Likewise use a consistent CHUTNEY_DATA_DIR
export CHUTNEY_DATA_DIR="${CHUTNEY_DATA_DIR:-$(pwd)}"
echo "export CHUTNEY_DATA_DIR=\"$CHUTNEY_DATA_DIR\"" >> tests/chutney/arti.run
"${CHUTNEY_BIN}" configure "$target"
"${CHUTNEY_BIN}" start "$target"
CHUTNEY_START_TIME=180 "${CHUTNEY_BIN}" wait_for_bootstrap "$target"
"${CHUTNEY_BIN}" verify "$target"
if [ "$PROXY" = "no" ] ; then
exit 0
fi
ARTI_FLAGS=()
if [ "$(id -u)" = "0" ] ; then
# If we are root, then we're probably running from CI. Tell Arti
# that's okay.
ARTI_FLAGS+=("-o" "application.allow_running_as_root=true")
fi
# It's not really possible to know where the user built the arti binary.
# Was it a debug build, a release build, did they use '--target <triple>', etc?
# Try the platform triple build directory first, and then the usual build directory.
# It's possible we may choose a stale binary, but there's not much else we can
# do here other than maybe comparing timestamps.
if [ -x ./target/x86_64-unknown-linux-gnu/debug/arti ]; then
cmd=./target/x86_64-unknown-linux-gnu/debug/arti
elif [ -x ./target/debug/arti ]; then
cmd=./target/debug/arti
else
echo "Could not find a debug arti binary. You can build it with 'cargo build -p arti'."
exit 1
fi
(
set +e
"$cmd" proxy -c "${CHUTNEY_DATA_DIR}/nodes/arti.toml" -d 35353 \
"${ARTI_FLAGS[@]}" &
arti_pid=$!
echo "pid=$arti_pid" >> tests/chutney/arti.run
wait "$arti_pid"
echo "result=$?" >> tests/chutney/arti.run
) & disown
# Wait for arti to start listening (it does so "immediately", but we don't want to get there first)
# Really there should be a proper daemon startup protocol here, but arti doesn't have one yet.
for idx in $(seq 30); do
if : >&/dev/null </dev/tcp/127.0.0.1/9150 ; then
echo "Port 9150 seems open."
break
elif [ "$idx" == 30 ]; then
echo "Waited 30 seconds without result; giving up on port 9150."
exit 1
else
echo "waiting for port 9150..."
sleep 1
fi
# verify Arti did not fail early
# shellcheck disable=SC1091
source tests/chutney/arti.run
pid="${pid:?}"
if ! kill -0 "$pid"; then
result="${result:?}"
echo "Arti failed to start with code $result"
exit 1
fi
done
|
