20 lines
589 B
Go
20 lines
589 B
Go
package agent
|
|
|
|
import "testing"
|
|
|
|
// 404 页面写入接口只接受绝对路径 + .html/.htm，避免被当成任意文件写入。
|
|
func TestValidatePagePath(t *testing.T) {
|
|
ok := []string{"/etc/frp/havline-404.html", "/var/www/404.htm"}
|
|
for _, path := range ok {
|
|
if _, err := validatePagePath(path); err != nil {
|
|
t.Errorf("%s 应当通过：%v", path, err)
|
|
}
|
|
}
|
|
bad := []string{"", "etc/frp/404.html", "/etc/frp/404.txt", "/etc/../etc/passwd.html"}
|
|
for _, path := range bad {
|
|
if _, err := validatePagePath(path); err == nil {
|
|
t.Errorf("%s 应当被拒绝", path)
|
|
}
|
|
}
|
|
}
|