blob: fc9cf21013699af6146faba0c77eb8b5e3065af0 (
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
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
|
#-
# SPDX-License-Identifier: BSD-2-Clause
#
# Copyright (c) 2026 Baptiste Daroussin <bapt@FreeBSD.org>
PKG_SERVE="${PKG_SERVE:-/usr/libexec/pkg-serve}"
serve()
{
printf "$1" | "${PKG_SERVE}" "$2"
}
check_output()
{
local pattern="$1" ; shift
output=$(serve "$@")
case "$output" in
*${pattern}*)
return 0
;;
*)
echo "Expected pattern: ${pattern}"
echo "Got: ${output}"
return 1
;;
esac
}
atf_test_case greeting
greeting_head()
{
atf_set "descr" "Server sends greeting on connect"
}
greeting_body()
{
mkdir repo
check_output "ok: pkg-serve " "quit\n" repo ||
atf_fail "greeting not found"
}
atf_test_case unknown_command
unknown_command_head()
{
atf_set "descr" "Unknown commands get ko response"
}
unknown_command_body()
{
mkdir repo
check_output "ko: unknown command 'plop'" "plop\nquit\n" repo ||
atf_fail "expected ko for unknown command"
}
atf_test_case get_missing_file
get_missing_file_head()
{
atf_set "descr" "Requesting a missing file returns ko"
}
get_missing_file_body()
{
mkdir repo
check_output "ko: file not found" "get nonexistent.pkg 0\nquit\n" repo ||
atf_fail "expected file not found"
}
atf_test_case get_file
get_file_head()
{
atf_set "descr" "Requesting an existing file returns its content"
}
get_file_body()
{
mkdir repo
echo "testcontent" > repo/test.pkg
output=$(serve "get test.pkg 0\nquit\n" repo)
echo "$output" | grep -q "ok: 12" ||
atf_fail "expected ok: 12, got: ${output}"
echo "$output" | grep -q "testcontent" ||
atf_fail "expected testcontent in output"
}
atf_test_case get_file_leading_slash
get_file_leading_slash_head()
{
atf_set "descr" "Leading slash in path is stripped"
}
get_file_leading_slash_body()
{
mkdir repo
echo "testcontent" > repo/test.pkg
check_output "ok: 12" "get /test.pkg 0\nquit\n" repo ||
atf_fail "leading slash not stripped"
}
atf_test_case get_file_uptodate
get_file_uptodate_head()
{
atf_set "descr" "File with old mtime returns ok: 0"
}
get_file_uptodate_body()
{
mkdir repo
echo "testcontent" > repo/test.pkg
check_output "ok: 0" "get test.pkg 9999999999\nquit\n" repo ||
atf_fail "expected ok: 0 for up-to-date file"
}
atf_test_case get_directory
get_directory_head()
{
atf_set "descr" "Requesting a directory returns ko"
}
get_directory_body()
{
mkdir -p repo/subdir
check_output "ko: not a file" "get subdir 0\nquit\n" repo ||
atf_fail "expected not a file"
}
atf_test_case get_missing_age
get_missing_age_head()
{
atf_set "descr" "get without age argument returns error"
}
get_missing_age_body()
{
mkdir repo
check_output "ko: bad command get" "get test.pkg\nquit\n" repo ||
atf_fail "expected bad command get"
}
atf_test_case get_bad_age
get_bad_age_head()
{
atf_set "descr" "get with non-numeric age returns error"
}
get_bad_age_body()
{
mkdir repo
check_output "ko: bad number" "get test.pkg notanumber\nquit\n" repo ||
atf_fail "expected bad number"
}
atf_test_case get_empty_arg
get_empty_arg_head()
{
atf_set "descr" "get with no arguments returns error"
}
get_empty_arg_body()
{
mkdir repo
check_output "ko: bad command get" "get \nquit\n" repo ||
atf_fail "expected bad command get"
}
atf_test_case path_traversal
path_traversal_head()
{
atf_set "descr" "Path traversal with .. is rejected"
}
path_traversal_body()
{
mkdir repo
check_output "ko: file not found" \
"get ../etc/passwd 0\nquit\n" repo ||
atf_fail "path traversal not rejected"
}
atf_test_case get_subdir_file
get_subdir_file_head()
{
atf_set "descr" "Files in subdirectories are served"
}
get_subdir_file_body()
{
mkdir -p repo/sub
echo "subcontent" > repo/sub/file.pkg
output=$(serve "get sub/file.pkg 0\nquit\n" repo)
echo "$output" | grep -q "ok: 11" ||
atf_fail "expected ok: 11, got: ${output}"
echo "$output" | grep -q "subcontent" ||
atf_fail "expected subcontent in output"
}
atf_test_case multiple_gets
multiple_gets_head()
{
atf_set "descr" "Multiple get commands in one session"
}
multiple_gets_body()
{
mkdir repo
echo "aaa" > repo/a.pkg
echo "bbb" > repo/b.pkg
output=$(serve "get a.pkg 0\nget b.pkg 0\nquit\n" repo)
echo "$output" | grep -q "ok: 4" ||
atf_fail "expected ok: 4 for a.pkg"
echo "$output" | grep -q "aaa" ||
atf_fail "expected content of a.pkg"
echo "$output" | grep -q "bbb" ||
atf_fail "expected content of b.pkg"
}
atf_test_case bad_basedir
bad_basedir_head()
{
atf_set "descr" "Non-existent basedir causes exit failure"
}
bad_basedir_body()
{
atf_check -s not-exit:0 -e match:"open" \
"${PKG_SERVE}" /nonexistent/path
}
atf_init_test_cases()
{
atf_add_test_case greeting
atf_add_test_case unknown_command
atf_add_test_case get_missing_file
atf_add_test_case get_file
atf_add_test_case get_file_leading_slash
atf_add_test_case get_file_uptodate
atf_add_test_case get_directory
atf_add_test_case get_missing_age
atf_add_test_case get_bad_age
atf_add_test_case get_empty_arg
atf_add_test_case path_traversal
atf_add_test_case get_subdir_file
atf_add_test_case multiple_gets
atf_add_test_case bad_basedir
}
|
