From 37107ac08db3db5eade495ee40d915ac82da5d3b Mon Sep 17 00:00:00 2001 From: SunBK201 Date: Thu, 20 Nov 2025 13:49:06 +0800 Subject: [PATCH] refactor: optimize length handling --- src/internal/rewrite/packet.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/internal/rewrite/packet.go b/src/internal/rewrite/packet.go index 949ed6b..08e798c 100644 --- a/src/internal/rewrite/packet.go +++ b/src/internal/rewrite/packet.go @@ -52,13 +52,14 @@ func (r *Rewriter) buildReplacement(srcAddr, dstAddr string, originalUA string, }) // Adjust to the exact length needed - if len(newUA) >= n { + newUALen := len(newUA) + if newUALen >= n { return []byte(newUA[:n]) } out := make([]byte, n) copy(out, newUA) // Pad with spaces if newUA is shorter than needed - for i := len(newUA); i < n; i++ { + for i := newUALen; i < n; i++ { out[i] = ' ' }