Fix binary string regex (#1415)

This commit is contained in:
nesstord 2022-12-06 23:45:27 +07:00 committed by GitHub
parent 5e0953d778
commit 61fda0b1ca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 18 deletions

19
lib.go
View File

@ -702,8 +702,8 @@ func isNumeric(s string) (bool, int, float64) {
} }
var ( var (
bstrExp = regexp.MustCompile(`_x[a-zA-Z\d]{4}_`) bstrExp = regexp.MustCompile(`_x[a-fA-F\d]{4}_`)
bstrEscapeExp = regexp.MustCompile(`x[a-zA-Z\d]{4}_`) bstrEscapeExp = regexp.MustCompile(`x[a-fA-F\d]{4}_`)
) )
// bstrUnmarshal parses the binary basic string, this will trim escaped string // bstrUnmarshal parses the binary basic string, this will trim escaped string
@ -729,16 +729,7 @@ func bstrUnmarshal(s string) (result string) {
} }
if bstrExp.MatchString(subStr) { if bstrExp.MatchString(subStr) {
cursor = match[1] cursor = match[1]
v, err := strconv.Unquote(`"\u` + s[match[0]+2:match[1]-1] + `"`) v, _ := strconv.Unquote(`"\u` + s[match[0]+2:match[1]-1] + `"`)
if err != nil {
if l > match[1]+6 && bstrEscapeExp.MatchString(s[match[1]:match[1]+6]) {
result += subStr[:6]
cursor = match[1] + 6
continue
}
result += subStr
continue
}
result += v result += v
} }
} }
@ -769,12 +760,10 @@ func bstrMarshal(s string) (result string) {
} }
if bstrExp.MatchString(subStr) { if bstrExp.MatchString(subStr) {
cursor = match[1] cursor = match[1]
_, err := strconv.Unquote(`"\u` + s[match[0]+2:match[1]-1] + `"`) if _, err := strconv.Unquote(`"\u` + s[match[0]+2:match[1]-1] + `"`); err == nil {
if err == nil {
result += "_x005F" + subStr result += "_x005F" + subStr
continue continue
} }
result += subStr
} }
} }
if cursor < l { if cursor < l {

View File

@ -305,18 +305,19 @@ func TestBstrUnmarshal(t *testing.T) {
"*_x0008_*": "*\b*", "*_x0008_*": "*\b*",
"*_x4F60__x597D_": "*你好", "*_x4F60__x597D_": "*你好",
"*_xG000_": "*_xG000_", "*_xG000_": "*_xG000_",
"*_xG05F_x0001_*": "*_xG05F*", "*_xG05F_x0001_*": "*_xG05F\x01*",
"*_x005F__x0008_*": "*_\b*", "*_x005F__x0008_*": "*_\b*",
"*_x005F_x0001_*": "*_x0001_*", "*_x005F_x0001_*": "*_x0001_*",
"*_x005f_x005F__x0008_*": "*_x005F_\b*", "*_x005f_x005F__x0008_*": "*_x005F_\b*",
"*_x005F_x005F_xG05F_x0006_*": "*_x005F_xG05F*", "*_x005F_x005F_xG05F_x0006_*": "*_x005F_xG05F\x06*",
"*_x005F_x005F_x005F_x0006_*": "*_x005F_x0006_*", "*_x005F_x005F_x005F_x0006_*": "*_x005F_x0006_*",
"_x005F__x0008_******": "_\b******", "_x005F__x0008_******": "_\b******",
"******_x005F__x0008_": "******_\b", "******_x005F__x0008_": "******_\b",
"******_x005F__x0008_******": "******_\b******", "******_x005F__x0008_******": "******_\b******",
"_x000x_x005F_x000x_": "_x000x_x000x_",
} }
for bstr, expected := range bstrs { for bstr, expected := range bstrs {
assert.Equal(t, expected, bstrUnmarshal(bstr)) assert.Equal(t, expected, bstrUnmarshal(bstr), bstr)
} }
} }