mirror of
https://github.com/hanzokms/turn.git
synced 2026-10-30 20:50:36 +00:00
45 lines
1.0 KiB
Go
45 lines
1.0 KiB
Go
// SPDX-FileCopyrightText: 2023 The Pion community <https://pion.ly>
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package proto
|
|
|
|
import (
|
|
"encoding/binary"
|
|
|
|
"github.com/pion/stun/v3"
|
|
)
|
|
|
|
// ConnectionID represents CONNECTION-ID attribute.
|
|
//
|
|
// The CONNECTION-ID attribute uniquely identifies a peer data
|
|
// connection. It is a 32-bit unsigned integral value.
|
|
//
|
|
// RFC 6062 Section 6.2.1.
|
|
type ConnectionID uint32
|
|
|
|
const connectionIDSize = 4 // uint32: 4 bytes, 32 bits
|
|
|
|
// AddTo adds CONNECTION-ID to message.
|
|
func (c ConnectionID) AddTo(m *stun.Message) error {
|
|
v := make([]byte, lifetimeSize)
|
|
binary.BigEndian.PutUint32(v, uint32(c))
|
|
m.Add(stun.AttrConnectionID, v)
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetFrom decodes CONNECTION-ID from message.
|
|
func (c *ConnectionID) GetFrom(m *stun.Message) error {
|
|
v, err := m.Get(stun.AttrConnectionID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err = stun.CheckSize(stun.AttrConnectionID, len(v), connectionIDSize); err != nil {
|
|
return err
|
|
}
|
|
_ = v[connectionIDSize-1] // Asserting length
|
|
*(*uint32)(c) = binary.BigEndian.Uint32(v)
|
|
|
|
return nil
|
|
}
|