Harden scan robustness and tests

This commit is contained in:
ZacharyZcR
2026-06-13 07:55:37 +08:00
parent 1595c92aed
commit 15a7670ba2
100 changed files with 4483 additions and 412 deletions
+11 -1
View File
@@ -2,7 +2,11 @@
package services
import "testing"
import (
"strings"
"testing"
"unicode/utf8"
)
func TestParseZooKeeperResponse(t *testing.T) {
banner, ok := parseZooKeeperResponse([]byte("imok"))
@@ -13,4 +17,10 @@ func TestParseZooKeeperResponse(t *testing.T) {
if _, ok := parseZooKeeperResponse([]byte("hello")); ok {
t.Fatal("unexpected match for non-zookeeper response")
}
longResp := "zk_version\t" + strings.Repeat("界", 205)
banner, ok = parseZooKeeperResponse([]byte(longResp))
if !ok || !utf8.ValidString(banner) || len([]rune(banner)) != 203 {
t.Fatalf("zookeeper truncation = %q ok=%v", banner, ok)
}
}