Harden scan robustness and tests

This commit is contained in:
ZacharyZcR
2026-06-14 22:23:48 +08:00
parent 5ad914a1bb
commit c49c23c7f0
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)
}
}