mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2026-08-25 04:17:15 +00:00
Sanaei
d175050f2e
disconnectClientTemporarily still gated on the protocol list from before XrayAPI.AddUser learned hysteria, so an over-limit Hysteria2 client kept its QUIC session until the fail2ban ban aged out, while a VLESS client in the same situation was dropped at once. buildUserAccount handles hysteria and model.Client already marshals the auth field the re-add needs, so admit the protocol. wireguard stays excluded: its keepAlive marshals as a JSON number, which the string-only user-field parsing rejects after the user was already removed. Closes #6256
43 lines
1.2 KiB
Go
43 lines
1.2 KiB
Go
package job
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/mhsanaei/3x-ui/v3/internal/database/model"
|
|
"github.com/mhsanaei/3x-ui/v3/internal/logger"
|
|
)
|
|
|
|
// The protocol gate must let hysteria through: XrayAPI supports it, and the
|
|
// skip left over-limit Hysteria2 sessions alive until the fail2ban ban caught up.
|
|
func TestDisconnectClientTemporarilyAllowsHysteria(t *testing.T) {
|
|
setupIntegrationDB(t)
|
|
|
|
const email = "hy2-limit-probe"
|
|
inbound := &model.Inbound{
|
|
Id: 1,
|
|
Protocol: model.Hysteria,
|
|
Tag: "hy2-limit-probe-tag",
|
|
Settings: `{"clients":[]}`,
|
|
}
|
|
clients := []model.Client{{Email: email, Auth: "secret"}}
|
|
|
|
(&CheckClientIpJob{}).disconnectClientTemporarily(inbound, email, clients)
|
|
|
|
var unsupported, attempted bool
|
|
for _, line := range logger.GetLogs(500, "warning") {
|
|
if strings.Contains(line, "Temporary disconnect is not supported for protocol hysteria") {
|
|
unsupported = true
|
|
}
|
|
if strings.Contains(line, "Failed to remove user "+email) {
|
|
attempted = true
|
|
}
|
|
}
|
|
if unsupported {
|
|
t.Fatal("hysteria was rejected by the protocol gate")
|
|
}
|
|
if !attempted {
|
|
t.Fatal("expected a remove attempt against the Xray API for hysteria")
|
|
}
|
|
}
|