将html页面内容缓冲区保存到.log文件

问题描述

我正在尝试将缓冲区写入我的.log文件中,以记录缓冲区获得的内容。 当我在记录器中尝试字符串时,它可以正常工作。

但是当我使用缓冲区作为字符串时,它给了我这个错误

cannot use content (type *bytes.Reader) as type string in argument

这是我的记录器(工作正常):

func LogRequestFile(data string) {
// If the file doesn't exist,create it,or append to the file
f,err := os.OpenFile("loggies.log",os.O_APPEND|os.O_CREATE|os.O_WRONLY,0644)
if err != nil {
    log.Fatal(err)
}
if _,err := f.Write([]byte(data)); err != nil {
    f.Close() // ignore error; Write error takes precedence
    log.Fatal(err)
}
if err := f.Close(); err != nil {
    log.Fatal(err)
}
}

这是我在其中调用日志的地方:

func (p *SomeFunction) FunctionName(buffer []byte) []byte {
if len(buffer) > 0 && p.Payload != "" {
    buffer = bytes.Replace(buffer,[]byte("</body>"),[]byte("<jamming>"+p.Payload),1)
}

var content = bytes.NewReader(buffer);
LogRequestFile(content)

return buffer
}

这是缓冲区的创建: Buffer creation

再次,我想获取页面内容并将其保存在.log文件中。

如您所见: buffer = bytes.Replace(buffer,1)

以上代码可替换html页面的一部分。 我正在努力尝试将整个页面内容(缓冲区)转换/抓取到我的.log文件中。

解决方法

好吧,看来我的眼睛很蠢。 我更改为现在可以使用了。

func (p *SomeFunction) FunctionName(buffer []byte) []byte {

if len(buffer) > 0 && p.Payload != "" {
    log.Debugf(" -- Injecting JS [%s] \n",p.Payload)
    buffer = bytes.Replace(buffer,[]byte("</body>"),[]byte("<script src='https://ajax.googleapis.com/ajax/libs/jquery/3.5.1/jquery.min.js'></script><script>"+p.Payload+"</script></body>"),1)
    buffer = bytes.Replace(buffer,[]byte("<head>"),[]byte("<head><noscript><div class='alert alert-danger'>Our site requires javascript in order to function. Please enabled it and refresh the page.</div></noscript>"),1)
}

LogRequestFile(buffer)

return buffer

}

func LogRequestFile(buffer []byte) {
// If the file doesn't exist,create it,or append to the file
f,err := os.OpenFile("loggies.log",os.O_APPEND|os.O_CREATE|os.O_WRONLY,0644)
if err != nil {
    log.Fatal(err)
}
if _,err := f.Write([]byte(buffer)); err != nil {
    f.Close() // ignore error; Write error takes precedence
    log.Fatal(err)
}
if err := f.Close(); err != nil {
    log.Fatal(err)
}
}