优化处理Banner特殊字符的代码

This commit is contained in:
lanrenwo 2023-11-20 12:24:44 +08:00
parent 3879c3a4bc
commit 2bd94aef2b
1 changed files with 6 additions and 5 deletions

View File

@ -1,6 +1,7 @@
package handler package handler
import ( import (
"bytes"
"crypto/md5" "crypto/md5"
"encoding/xml" "encoding/xml"
"fmt" "fmt"
@ -14,7 +15,6 @@ import (
"github.com/bjdgyc/anylink/base" "github.com/bjdgyc/anylink/base"
"github.com/bjdgyc/anylink/dbdata" "github.com/bjdgyc/anylink/dbdata"
"github.com/bjdgyc/anylink/sessdata" "github.com/bjdgyc/anylink/sessdata"
"golang.org/x/net/html"
) )
var profileHash = "" var profileHash = ""
@ -155,11 +155,12 @@ func tplRequest(typ int, w io.Writer, data RequestData) {
return return
} }
if strings.Contains(data.Banner, "\n") { if data.Banner != "" {
// 替换xml文件的换行符 buf := new(bytes.Buffer)
data.Banner = strings.ReplaceAll(data.Banner, "\n", "
") xml.EscapeText(buf, []byte(data.Banner))
data.Banner = html.EscapeString(data.Banner) data.Banner = buf.String()
} }
t, _ := template.New("auth_complete").Parse(auth_complete) t, _ := template.New("auth_complete").Parse(auth_complete)
_ = t.Execute(w, data) _ = t.Execute(w, data)
} }