// SPDX-FileCopyrightText: 2023 The Pion community <https://pion.ly>
// SPDX-License-Identifier: MIT

package rtp

import (
	
)

// Sequencer generates sequential sequence numbers for building RTP packets.
type Sequencer interface {
	NextSequenceNumber() uint16
	RollOverCount() uint64
}

// maxInitialRandomSequenceNumber is the maximum value used for the initial sequence
// number when using NewRandomSequencer().
// This uses only half the potential sequence number space to avoid issues decrypting
// SRTP when the sequence number starts near the rollover and there is packet loss.
// See https://webrtc-review.googlesource.com/c/src/+/358360
const maxInitialRandomSequenceNumber = 1<<15 - 1

// NewRandomSequencer returns a new sequencer starting from a random sequence
// number.
func () Sequencer {
	return &sequencer{
		sequenceNumber: uint16(globalMathRandomGenerator.Intn(maxInitialRandomSequenceNumber)), // nolint: gosec // G115
	}
}

// NewFixedSequencer returns a new sequencer starting from a specific
// sequence number.
func ( uint16) Sequencer {
	return &sequencer{
		sequenceNumber:  - 1, // -1 because the first sequence number prepends 1
	}
}

type sequencer struct {
	sequenceNumber uint16
	rollOverCount  uint64
	mutex          sync.Mutex
}

// NextSequenceNumber increment and returns a new sequence number for
// building RTP packets.
func ( *sequencer) () uint16 {
	.mutex.Lock()
	defer .mutex.Unlock()

	.sequenceNumber++
	if .sequenceNumber == 0 {
		.rollOverCount++
	}

	return .sequenceNumber
}

// RollOverCount returns the amount of times the 16bit sequence number
// has wrapped.
func ( *sequencer) () uint64 {
	.mutex.Lock()
	defer .mutex.Unlock()

	return .rollOverCount
}