Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions src/proxy/unit_tests/stub.cc
Original file line number Diff line number Diff line change
Expand Up @@ -22,3 +22,5 @@
*/

#include "proxy/IPAllow.h"

uint8_t IpAllow::subjects[IpAllow::Subject::MAX_SUBJECTS];
13 changes: 9 additions & 4 deletions tests/gold_tests/pluginTest/tsapi/test_TSVConnPPInfo.test.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,9 @@
# plaintext HTTP
tr = Test.AddTestRun()
tr.TimeOut = 10
tr.Processes.Default.Command = f"curl --haproxy-protocol --haproxy-clientip 1.2.3.4 'http://127.0.0.1:{ts.Variables.proxy_protocol_port}/httpbin/get'"
tr.Processes.Default.Command = (
f"curl --haproxy-protocol --haproxy-clientip 1.2.3.4 "
f"'http://127.0.0.1:{ts.Variables.proxy_protocol_port}/httpbin/get'")
tr.Processes.Default.ReturnCode = 0
tr.Processes.Default.StartBefore(httpbin)
tr.Processes.Default.StartBefore(Test.Processes.ts)
Expand All @@ -84,7 +86,9 @@
# HTTPS
tr = Test.AddTestRun()
tr.TimeOut = 10
tr.Processes.Default.Command = f"curl --haproxy-protocol --haproxy-clientip 5.6.7.8 -k 'https://127.0.0.1:{ts.Variables.proxy_protocol_ssl_port}/httpbin/get'"
tr.Processes.Default.Command = (
f"curl --haproxy-protocol --haproxy-clientip 5.6.7.8 -k "
f"'https://127.0.0.1:{ts.Variables.proxy_protocol_ssl_port}/httpbin/get'")
tr.Processes.Default.ReturnCode = 0
tr.Processes.Default.Streams.stdout = "test_TSVConnPPInfo_curl1.gold"
tr.StillRunningAfter = httpbin
Expand All @@ -95,7 +99,8 @@
tr.Processes.Default.ReturnCode = 0
f = tr.Disk.File(log_path)
f.Content = "test_TSVConnPPInfo_plugin_log.gold"
# curl 8.20+ intentionally uses --haproxy-clientip for both PROXY addresses so the address family matches.
f.Content += Testers.ContainsExpression(
"PP Info Received:V1,P2,T1,SRC1.2.3.4,DST127.0.0.1", "Expected information should be received")
r"PP Info Received:V1,P2,T1,SRC1\.2\.3\.4,DST(127\.0\.0\.1|1\.2\.3\.4)", "Expected information should be received")
f.Content += Testers.ContainsExpression(
"PP Info Received:V1,P2,T1,SRC5.6.7.8,DST127.0.0.1", "Expected information should be received")
r"PP Info Received:V1,P2,T1,SRC5\.6\.7\.8,DST(127\.0\.0\.1|5\.6\.7\.8)", "Expected information should be received")
12 changes: 6 additions & 6 deletions tests/gold_tests/remap/remap_acl.test.py
Original file line number Diff line number Diff line change
Expand Up @@ -166,13 +166,13 @@ def _configure_traffic_server(self) -> int:
#
tr = Test.AddTestRun("Await config reload")
p = tr.Processes.Default
p.Command = 'echo awaiting config reload'
p.Env = ts.Env
Test_remap_acl._ts_reload_counter += 1
count = Test_remap_acl._ts_reload_counter
await_config_reload = tr.Processes.Process(f'config_reload_succeeded_{count}', 'sleep 30')
await_config_reload.Ready = When.FileContains(ts.Disk.diags_log.Name, "remap.config finished loading", count)
p.StartBefore(await_config_reload)
count = 1 + Test_remap_acl._ts_reload_counter
condwait = os.path.join(Test.Variables.AtsTestToolsDir, 'condwait')
p.Command = (f"'{condwait}' 30 --contains '{ts.Disk.diags_log.Name}' "
f"'remap.config finished loading' {count}")
p.Env = ts.Env
tr.StillRunningAfter = ts

else:
record_config = {
Expand Down
8 changes: 1 addition & 7 deletions tests/gold_tests/tls/tls_flow_control.test.py
Original file line number Diff line number Diff line change
Expand Up @@ -70,13 +70,7 @@ def _configure_trafficserver(self) -> 'Process':
TestTlsFlowControl._ts_counter += 1

ts.addDefaultSSLFiles()
ts.Disk.ssl_multicert_yaml.AddLines(
"""
ssl_multicert:
- dest_ip: "*"
ssl_cert_name: server.pem
ssl_key_name: server.key
""".split("\n"))
ts.Disk.ssl_multicert_config.AddLine('dest_ip=* ssl_cert_name=server.pem ssl_key_name=server.key')
ts.Disk.remap_config.AddLine(f'map / http://127.0.0.1:{self._server.Variables.Port}')
ts.Disk.records_config.update(
{
Expand Down
8 changes: 1 addition & 7 deletions tests/gold_tests/tls/tls_record_size.test.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,13 +75,7 @@ def _configure_trafficserver(self) -> 'Process':
TestRecordSizeClamp._ts_counter += 1

ts.addDefaultSSLFiles()
ts.Disk.ssl_multicert_yaml.AddLines(
"""
ssl_multicert:
- dest_ip: "*"
ssl_cert_name: server.pem
ssl_key_name: server.key
""".split("\n"))
ts.Disk.ssl_multicert_config.AddLine('dest_ip=* ssl_cert_name=server.pem ssl_key_name=server.key')
ts.Disk.remap_config.AddLine(f'map / http://127.0.0.1:{self._server.Variables.Port}')
ts.Disk.records_config.update(
{
Expand Down
12 changes: 3 additions & 9 deletions tests/gold_tests/tls/tls_reload_under_load.test.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
'''
Existing cert/SNI reload tests reload while the server is idle. This one drives
continuous concurrent TLS handshakes and reloads ssl_multicert.yaml on top of
continuous concurrent TLS handshakes and reloads ssl_multicert.config on top of
them, stressing the SSL/BIO ownership boundary of the layered TLS VConnection.
The swapped-in certificate must take effect, every handshake must succeed, and
ATS must not crash.
Expand Down Expand Up @@ -51,13 +51,7 @@ def _configure_trafficserver(self) -> 'Process':
ts.addSSLfile("ssl/signed-bar.key")
ts.addSSLfile("ssl/signed2-bar.pem")

ts.Disk.ssl_multicert_yaml.AddLines(
"""
ssl_multicert:
- dest_ip: "*"
ssl_cert_name: signed-bar.pem
ssl_key_name: signed-bar.key
""".split("\n"))
ts.Disk.ssl_multicert_config.AddLine('dest_ip=* ssl_cert_name=signed-bar.pem ssl_key_name=signed-bar.key')
ts.Disk.records_config.update(
{
'proxy.config.ssl.server.cert.path': f'{ts.Variables.SSLDir}',
Expand All @@ -69,7 +63,7 @@ def _configure_trafficserver(self) -> 'Process':

# The reload must actually have run (otherwise the test would be vacuous).
ts.Disk.diags_log.Content = Testers.ContainsExpression(
"ssl_multicert.yaml finished loading", "the cert configuration must reload while load is in flight")
"ssl_multicert.config finished loading", "the cert configuration must reload while load is in flight")
# The reload-under-load must not crash or trip an assertion / sanitizer.
ts.Disk.traffic_out.Content = Testers.ExcludesExpression(
"received signal|failed assertion", "ATS must not crash reloading certs under load")
Expand Down
8 changes: 1 addition & 7 deletions tests/gold_tests/tls/tls_renegotiation.test.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,13 +61,7 @@ def _configure_trafficserver(self) -> 'Process':

ts.addSSLfile("ssl/server.pem")
ts.addSSLfile("ssl/server.key")
ts.Disk.ssl_multicert_yaml.AddLines(
"""
ssl_multicert:
- dest_ip: "*"
ssl_cert_name: server.pem
ssl_key_name: server.key
""".split("\n"))
ts.Disk.ssl_multicert_config.AddLine('dest_ip=* ssl_cert_name=server.pem ssl_key_name=server.key')
ts.Disk.records_config.update(
{
'proxy.config.ssl.server.cert.path': f'{ts.Variables.SSLDir}',
Expand Down
8 changes: 1 addition & 7 deletions tests/gold_tests/tls/tls_renegotiation_allowed.test.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,13 +83,7 @@ def _configure_trafficserver(self) -> 'Process':

ts.addSSLfile("ssl/server.pem")
ts.addSSLfile("ssl/server.key")
ts.Disk.ssl_multicert_yaml.AddLines(
"""
ssl_multicert:
- dest_ip: "*"
ssl_cert_name: server.pem
ssl_key_name: server.key
""".split("\n"))
ts.Disk.ssl_multicert_config.AddLine('dest_ip=* ssl_cert_name=server.pem ssl_key_name=server.key')
ts.Disk.records_config.update(
{
'proxy.config.ssl.server.cert.path': f'{ts.Variables.SSLDir}',
Expand Down
66 changes: 56 additions & 10 deletions tests/tools/condwait
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@
#
# CONDITION is the ('test' command) condition to wait for. (It may contain white space.)
# For file existence tests (-f, -e, -d), glob patterns are supported (e.g., -f /path/crash-*.log).
# Use "--contains FILE PATTERN COUNT" to wait until FILE contains PATTERN in at least COUNT lines.
#
# MAX-WAIT is the maximum number of seconds to wait for the condition. If it is omitted, it defaults to 60.
#
Expand All @@ -34,8 +35,13 @@
WAIT=60
POST_WAIT=0

usage() {
echo "usage: condwait [ MAX-WAIT [ POST-WAIT ] ] TEST-CONDITION" >&2
echo " condwait [ MAX-WAIT [ POST-WAIT ] ] --contains FILE PATTERN COUNT" >&2
}

if [[ "$1" = "" ]] ; then
echo "usage: condwait [ MAX-WAIT [ POST-WAIT ] ] TEST-CONDTION" >&2
usage
exit 1
fi

Expand All @@ -44,7 +50,7 @@ if [[ "$X" = "$1" ]] ; then
WAIT=$1
shift
if [[ "$1" = "" ]] ; then
echo "usage: condwait [ MAX-WAIT [ POST-WAIT ] ] TEST-CONDTION" >&2
usage
exit 1
fi
X=$( echo "$1" | sed 's/x/yy/g' | sed 's/[^0-9]/x/g' )
Expand All @@ -55,24 +61,64 @@ if [[ "$X" = "$1" ]] ; then
fi

if [[ "$1" = "" ]] ; then
echo "usage: condwait [ MAX-WAIT [ POST-WAIT ] ] TEST-CONDTION" >&2
usage
exit 1
fi

# Check if this is a simple file existence test (-f, -e, -d). If so, use ls -d
# which handles glob patterns properly. Otherwise, use test for the condition.
if [[ "$1" = "--contains" ]]; then
if [[ $# -ne 4 ]] || ! [[ "$4" =~ ^[0-9]+$ ]]; then
usage
exit 1
fi
fi

# Check simple file existence tests (-f, -e, -d) specially so literal paths with
# whitespace work while glob patterns remain supported.
check_condition() {
if [[ "$1" = "-f" || "$1" = "-e" || "$1" = "-d" ]] && [[ $# -eq 2 ]]; then
# Use ls -d for file existence tests to support glob patterns.
ls -d $2 >/dev/null 2>&1
if [[ "$1" = "--contains" ]]; then
[[ -f "$2" ]] || return 1
local matches
matches=$(awk -v pattern="$3" -v expected="$4" '
BEGIN {
if (expected == 0) {
print 0
found = 1
exit
}
}
index($0, pattern) {
count++
if (count >= expected) {
print count
found = 1
exit
}
}
END {
if (!found) {
print count + 0
}
}
' "$2")
(( matches >= $4 ))
elif [[ "$1" = "-f" || "$1" = "-e" || "$1" = "-d" ]] && [[ $# -eq 2 ]]; then
if [[ "$2" = *[\*\?\[]* ]]; then
# Use compgen for file existence tests to support glob patterns.
local match
while IFS= read -r match; do
test "$1" "$match" && return 0
done < <(compgen -G "$2")
return 1
fi
test "$1" "$2"
else
test $*
test "$@"
fi
}

while (( WAIT > 0 ))
do
if check_condition $*
if check_condition "$@"
then
if (( POST_WAIT > 0 ))
then
Expand Down