diff options
Diffstat (limited to 'satp/packet.go')
-rw-r--r-- | satp/packet.go | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/satp/packet.go b/satp/packet.go index 22cf4ba..b52b0e1 100644 --- a/satp/packet.go +++ b/satp/packet.go @@ -114,7 +114,7 @@ type EncryptedPacket struct { func NewEncryptedPacket() (ep *EncryptedPacket) { ep = &EncryptedPacket{} ep.header = ep.buffer[:8:8] - ep.payload = ep.buffer[8:] + ep.payload = ep.buffer[8:8] ep.authTag = nil return } @@ -177,7 +177,11 @@ func (ep *EncryptedPacket) ReadFrom(r io.Reader) (int64, error) { return int64(n), err } +func (ep *EncryptedPacket) getPacket() (data []byte) { + return ep.buffer[:len(ep.header)+len(ep.payload)+len(ep.authTag)] +} + func (ep *EncryptedPacket) WriteTo(w io.Writer) (int64, error) { - n, err := w.Write(ep.buffer[:len(ep.header)+len(ep.payload)+len(ep.authTag)]) + n, err := w.Write(ep.getPacket()) return int64(n), err } |