idk/go-connfilter
1
0
Fork 0
mirror of https://github.com/go-i2p/go-connfilter.git synced 2025-07-12 19:04:51 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix regex replacer

Browse Source
This commit is contained in:
eyedeekay
2025-02-04 21:34:38 -05:00
parent 1713fd7f9b
commit 0f2b889a80
1 changed files with 7 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
regexFilter.go
View File

@ -4,6 +4,7 @@ import (
"bytes"
"errors"
"net"
"regexp"
)
var ErrInvalidRegexFilter = errors.New("invalid regex filter")
@ -31,14 +32,10 @@ func (c *RegexConnFilter) ReadFilter(b []byte) ([]byte, error) {
if c.match == "" {
return b, nil
}
re := regexp.MustCompile(c.match)
s := re.ReplaceAllString(string(b), ``)
// Create a new bytes buffer
var buffer bytes.Buffer
// Write the modified data to the buffer, with regex matches removed
buffer.Write(bytes.ReplaceAll(b, []byte(c.match), []byte("")))
return buffer.Bytes(), nil
return []byte(s), nil
}
// WriteFilter creates a new RegexConnFilter that replaces occurrences of target regex with empty strings in the data read from the connection.
@ -47,14 +44,10 @@ func (c *RegexConnFilter) WriteFilter(b []byte) ([]byte, error) {
if c.match == "" {
return b, nil
}
re := regexp.MustCompile(c.match)
s := re.ReplaceAllString(string(b), ``)
// Create a new bytes buffer
var buffer bytes.Buffer
// Write the modified data to the buffer, with regex matches removed
buffer.Write(bytes.ReplaceAll(b, []byte(c.match), []byte("")))
return buffer.Bytes(), nil
return []byte(s), nil
}
// NewRegexConnFilter creates a new RegexConnFilter that replaces occurrences of target regex with empty strings in the data read from the connection.