mirror of
https://github.com/buger/goreplay.git
synced 2024-04-21 12:32:02 +00:00
118 lines
2.3 KiB
Go
118 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/rand"
|
|
"encoding/hex"
|
|
"strconv"
|
|
)
|
|
|
|
const (
|
|
RequestPayload = '1'
|
|
ResponsePayload = '2'
|
|
ReplayedResponsePayload = '3'
|
|
)
|
|
|
|
func randByte(len int) []byte {
|
|
b := make([]byte, len / 2)
|
|
rand.Read(b)
|
|
|
|
h := make([]byte, len)
|
|
hex.Encode(h, b)
|
|
|
|
return h
|
|
}
|
|
|
|
func uuid() []byte {
|
|
return randByte(24)
|
|
}
|
|
|
|
var payloadSeparator = "\n🐵🙈🙉\n"
|
|
|
|
func payloadScanner(data []byte, atEOF bool) (advance int, token []byte, err error) {
|
|
if atEOF && len(data) == 0 {
|
|
return 0, nil, nil
|
|
}
|
|
|
|
if i := bytes.Index(data, []byte(payloadSeparator)); i >= 0 {
|
|
// We have a full newline-terminated line.
|
|
return i + len([]byte(payloadSeparator)), data[0:i], nil
|
|
}
|
|
|
|
if atEOF {
|
|
return len(data), data, nil
|
|
}
|
|
return 0, nil, nil
|
|
}
|
|
|
|
// Timing is request start or round-trip time, depending on payloadType
|
|
func payloadHeader(payloadType byte, uuid []byte, timing int64, latency int64) (header []byte) {
|
|
var sTime, sLatency string
|
|
|
|
sTime = strconv.FormatInt(timing, 10)
|
|
if latency != -1 {
|
|
sLatency = strconv.FormatInt(latency, 10)
|
|
}
|
|
|
|
//Example:
|
|
// 3 f45590522cd1838b4a0d5c5aab80b77929dea3b3 1231\n
|
|
// `+ 1` indicates space characters or end of line
|
|
headerLen := 1 + 1 + len(uuid) + 1 + len(sTime) + 1
|
|
|
|
if latency != -1 {
|
|
headerLen += len(sLatency) + 1
|
|
}
|
|
|
|
header = make([]byte, headerLen)
|
|
header[0] = payloadType
|
|
header[1] = ' '
|
|
header[2+len(uuid)] = ' '
|
|
header[len(header)-1] = '\n'
|
|
|
|
copy(header[2:], uuid)
|
|
copy(header[3+len(uuid):], sTime)
|
|
|
|
if latency != -1 {
|
|
header[3+len(uuid)+len(sTime)] = ' '
|
|
copy(header[4+len(uuid)+len(sTime):], sLatency)
|
|
}
|
|
|
|
return header
|
|
}
|
|
|
|
func payloadBody(payload []byte) []byte {
|
|
headerSize := bytes.IndexByte(payload, '\n')
|
|
return payload[headerSize+1:]
|
|
}
|
|
|
|
func payloadMeta(payload []byte) [][]byte {
|
|
headerSize := bytes.IndexByte(payload, '\n')
|
|
if headerSize < 0 {
|
|
headerSize = 0
|
|
}
|
|
return bytes.Split(payload[:headerSize], []byte{' '})
|
|
}
|
|
|
|
func payloadID(payload []byte) []byte {
|
|
idx := bytes.IndexByte(payload[2:], ' ')
|
|
|
|
if idx == -1 {
|
|
return []byte{}
|
|
}
|
|
|
|
return payload[2: 2 + idx]
|
|
}
|
|
|
|
func isOriginPayload(payload []byte) bool {
|
|
switch payload[0] {
|
|
case RequestPayload, ResponsePayload:
|
|
return true
|
|
default:
|
|
return false
|
|
}
|
|
}
|
|
|
|
func isRequestPayload(payload []byte) bool {
|
|
return payload[0] == RequestPayload
|
|
}
|