test(config): add test for keyloc and header size

2026-01-25 18:17:37 +07:00
parent 9cdce24030
commit 5cd3c201f2
3 changed files with 46 additions and 1 deletions
@@ -31,7 +31,8 @@ type config struct {
 	allowedPortsEnd   uint16

 	bufferSize int
-
+	headerSize int
+	
 	pprofEnabled bool
 	pprofPort    string

@@ -73,6 +74,7 @@ func parse() (*config, error) {
 	}

 	bufferSize := parseBufferSize()
+	headerSize := parseHeaderSize()

 	pprofEnabled := getenvBool("PPROF_ENABLED", false)
 	pprofPort := getenv("PPROF_PORT", "6060")
@@ -100,6 +102,7 @@ func parse() (*config, error) {
 		allowedPortsStart: start,
 		allowedPortsEnd:   end,
 		bufferSize:        bufferSize,
+		headerSize:        headerSize,
 		pprofEnabled:      pprofEnabled,
 		pprofPort:         pprofPort,
 		mode:              mode,
@@ -161,6 +164,16 @@ func parseBufferSize() int {
 	return size
 }

+func parseHeaderSize() int {
+	raw := getenv("MAX_HEADER_SIZE", "4096")
+	size, err := strconv.Atoi(raw)
+	if err != nil || size < 4096 || size > 131072 {
+		log.Println("Invalid BUFFER_SIZE, falling back to 4096")
+		return 4096
+	}
+	return size
+}
+
 func getenv(key, def string) string {
 	if v := os.Getenv(key); v != "" {
 		return v