~trn/gfsmux

ref: 859dacdd4619f1e458e59908d55f05fca8d691a8 gfsmux/frame.go -rw-r--r-- 1.7 KiB
859dacdd — Jeffrey H. Johnson Update dependencies 3 months ago
                                                                                
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
package gfsmux

import (
	"encoding/binary"
	"fmt"
)

const (
	// Protocol version 1: old commands

	// CmdSyn ... stream open
	CmdSyn byte = iota
	// CmdFin ... stream close, EOF
	CmdFin
	// CmdPsh ... push data
	CmdPsh
	// CmdNop ... noop
	CmdNop

	// Protocol version 2: new commands

	// CmdUpd ... notify bytes consumed by remote peer-end
	CmdUpd
)

const (
	// data size of CmdUpd, format:
	// |4B data consumed(ACK)| 4B window size(WINDOW) |
	szCmdUPD = 8
)

const (
	// initial peer window guess, a slow-start
	initialPeerWindow = 262144
)

const (
	sizeOfVer    = 1
	sizeOfCmd    = 1
	sizeOfLength = 2
	sizeOfSid    = 4
	// HeaderSize ...
	HeaderSize = sizeOfVer + sizeOfCmd + sizeOfSid + sizeOfLength
)

// Frame defines a packet from or to be multiplexed into a single Connection
type Frame struct {
	Ver  byte
	Cmd  byte
	Sid  uint32
	Data []byte
}

// NewFrame ...
func NewFrame(
	Version,
	Cmd byte,
	Sid uint32,
) Frame {
	return Frame{Ver: Version, Cmd: Cmd, Sid: Sid}
}

type rawHeader [HeaderSize]byte

func (
	h rawHeader,
) Version() byte {
	return h[0]
}

func (
	h rawHeader,
) Cmd() byte {
	return h[1]
}

func (
	h rawHeader,
) Length() uint16 {
	return binary.LittleEndian.Uint16(
		h[2:],
	)
}

func (
	h rawHeader,
) StreamID() uint32 {
	return binary.LittleEndian.Uint32(
		h[4:],
	)
}

func (
	h rawHeader,
) String() string {
	return fmt.Sprintf(
		"Version:%d Cmd:%d StreamID:%d Length:%d",
		h.Version(),
		h.Cmd(),
		h.StreamID(),
		h.Length(),
	)
}

type updHeader [szCmdUPD]byte

func (
	h updHeader,
) Consumed() uint32 {
	return binary.LittleEndian.Uint32(
		h[:],
	)
}

func (
	h updHeader,
) Window() uint32 {
	return binary.LittleEndian.Uint32(
		h[4:],
	)
}