WireGold/gold/head/box.go

package head

import (
	"bytes"
	"encoding/binary"
	"encoding/hex"
	"unsafe"

	"github.com/fumiama/orbyte/pbuf"
	"github.com/sirupsen/logrus"

	"github.com/fumiama/WireGold/config"
	"github.com/fumiama/WireGold/internal/algo"
	"github.com/fumiama/WireGold/internal/bin"
)

// PreCRC64 calculate crc64 checksum without idxdatsz.
func (p *Packet) PreCRC64() (crc uint64) {
	w := bin.SelectWriter()
	if bin.IsLittleEndian {
		w.Write((*[PacketHeadNoCRCLen]byte)(
			(unsafe.Pointer)(p),
		)[:])
	} else {
		w.WriteUInt32(p.idxdatsz)
		w.WriteUInt32(uint32(p.randn))
		w.WriteUInt16((uint16(p.TTL) << 8) | uint16(p.Proto))
		w.WriteUInt16(p.SrcPort)
		w.WriteUInt16(p.DstPort)
		w.WriteUInt16(p.Offset)
		w.Write(p.src[:])
		w.Write(p.dst[:])
	}
	w.P(func(b *pbuf.Buffer) {
		crc = algo.MD5Hash8(b.Bytes()[PacketHeadPreCRCIdx:])
		if config.ShowDebugLog {
			logrus.Debugf(
				"[box] calc pre-crc64 %016x, dat %s", crc,
				hex.EncodeToString(b.Bytes()[PacketHeadPreCRCIdx:]),
			)
		}
	})
	return
}

// WriteHeaderTo write header bytes to buf
// with crc64 checksum.
func (p *Packet) WriteHeaderTo(buf *bytes.Buffer) {
	if bin.IsLittleEndian {
		buf.Write((*[PacketHeadNoCRCLen]byte)(
			(unsafe.Pointer)(p),
		)[:])
		p.md5h8rem = int64(algo.MD5Hash8(buf.Bytes()))
		binary.Write(buf, binary.LittleEndian, p.md5h8rem)
		return
	}
	w := bin.SelectWriter()
	w.WriteUInt32(p.idxdatsz)
	w.WriteUInt32(uint32(p.randn))
	w.WriteUInt16((uint16(p.TTL) << 8) | uint16(p.Proto))
	w.WriteUInt16(p.SrcPort)
	w.WriteUInt16(p.DstPort)
	w.WriteUInt16(p.Offset)
	w.Write(p.src[:])
	w.Write(p.dst[:])
	w.P(func(b *pbuf.Buffer) {
		p.md5h8rem = int64(algo.MD5Hash8(b.Bytes()))
	})
	w.WriteUInt64(uint64(p.md5h8rem))
	w.P(func(b *pbuf.Buffer) {
		buf.ReadFrom(b)
	})
}