mirror of https://github.com/40t/go-sniffer.git
87 lines
1.3 KiB
Go
87 lines
1.3 KiB
Go
package build
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"encoding/json"
|
|
"fmt"
|
|
"gopkg.in/mgo.v2/bson"
|
|
"io"
|
|
"time"
|
|
)
|
|
|
|
func GetNowStr(isClient bool) string {
|
|
var msg string
|
|
layout := "01/02 15:04:05.000000"
|
|
msg += time.Now().Format(layout)
|
|
if isClient {
|
|
msg += "| cli -> ser |"
|
|
}else{
|
|
msg += "| ser -> cli |"
|
|
}
|
|
return msg
|
|
}
|
|
|
|
func ReadInt32(r io.Reader) (n int32) {
|
|
binary.Read(r, binary.LittleEndian, &n)
|
|
return
|
|
}
|
|
|
|
func ReadInt64(r io.Reader) int64 {
|
|
var n int64
|
|
binary.Read(r, binary.LittleEndian, &n)
|
|
return n
|
|
}
|
|
|
|
func ReadString(r io.Reader) string {
|
|
|
|
var result []byte
|
|
var b = make([]byte, 1)
|
|
for {
|
|
|
|
_, err := r.Read(b)
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if b[0] == '\x00' {
|
|
break
|
|
}
|
|
|
|
result = append(result, b[0])
|
|
}
|
|
|
|
return string(result)
|
|
}
|
|
|
|
func ReadBson2Json(r io.Reader) (string) {
|
|
|
|
//read len
|
|
docLen := ReadInt32(r)
|
|
if docLen == 0 {
|
|
return ""
|
|
}
|
|
|
|
//document []byte
|
|
docBytes := make([]byte, int(docLen))
|
|
binary.LittleEndian.PutUint32(docBytes, uint32(docLen))
|
|
if _, err := io.ReadFull(r, docBytes[4:]); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
//resolve document
|
|
var bsn bson.M
|
|
err := bson.Unmarshal(docBytes, &bsn)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
//format to Json
|
|
jsonStr, err := json.Marshal(bsn)
|
|
if err != nil {
|
|
return fmt.Sprintf("{\"error\":%s}", err.Error())
|
|
}
|
|
return string(jsonStr)
|
|
}
|
|
|