mirror of
https://github.com/pion/webrtc.git
synced 2025-11-01 03:04:06 +08:00
4fc7cfccec
We need to also store the raw including header for easier access instead of remarshaling every time
42 lines
686 B
Go
42 lines
686 B
Go
package sctp
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"math/rand"
|
|
"time"
|
|
)
|
|
|
|
type ParamStateCookie struct {
|
|
ParamHeader
|
|
Cookie []byte
|
|
}
|
|
|
|
func NewRandomStateCookie() *ParamStateCookie {
|
|
rs := rand.NewSource(time.Now().UnixNano())
|
|
r := rand.New(rs)
|
|
randCookie := make([]byte, 32)
|
|
i := 0
|
|
for i < 4 {
|
|
binary.BigEndian.PutUint64(randCookie[i*4:], r.Uint64())
|
|
i++
|
|
}
|
|
|
|
s := &ParamStateCookie{
|
|
Cookie: randCookie,
|
|
}
|
|
|
|
return s
|
|
}
|
|
|
|
func (s *ParamStateCookie) Marshal() ([]byte, error) {
|
|
s.typ = StateCookie
|
|
s.raw = s.Cookie
|
|
return s.ParamHeader.Marshal()
|
|
}
|
|
|
|
func (s *ParamStateCookie) Unmarshal(raw []byte) (Param, error) {
|
|
s.ParamHeader.Unmarshal(raw)
|
|
s.Cookie = s.raw
|
|
return s, nil
|
|
}
|