stream-server/srt/data_chain.go

package srt

import (
	"math"
	"sort"
	"io"
)

type Datum struct {
	seq_num uint32
	data []byte
	next *Datum
}

type DatumLink struct {
	queued int
	root *Datum
	end *Datum
}

type chains []*DatumLink

func (c chains) Len() (int) {
	return len(c)
}

func (c chains) Swap(i, j int) {
	c[i], c[j] = c[j], c[i]
}

func (c chains) Less(i, j int) (bool) {
	x_1 := c[i].end.seq_num
	x_2 := c[j].root.seq_num
	serial_add_limit := uint32(math.Pow(2, 30))
	if (x_1 < x_2 && x_2 - x_1 < serial_add_limit) || (x_1 > x_2 && x_1 - x_2 > serial_add_limit) {
		return true
	}
	return false
}

type DatumStorage struct {
	main *DatumLink
	offshoots chains
}

func (buffer *DatumLink) NewDatum(pkt *Packet) {
	datum := new(Datum)
	datum.seq_num = pkt.header_info.(*DataHeader).seq_num
	datum.data = pkt.cif.([]byte)

	buffer.queued += 1
	buffer.end.next = datum
	buffer.end = datum
}

func NewDatumLink(pkt *Packet) (*DatumLink) {
	buffer := new(DatumLink)
	root_datum := new(Datum)
	root_datum.seq_num = pkt.header_info.(*DataHeader).seq_num
	root_datum.data = pkt.cif.([]byte)

	buffer.root = root_datum
	buffer.end = root_datum
	buffer.queued = 1
	return buffer
}

func NewDatumStorage(packet *Packet) (*DatumStorage) {
	storage := new(DatumStorage)
	storage.main = NewDatumLink(packet)
	return storage
}

func (storage *DatumStorage) Expunge(output io.WriteCloser) (error) {
	curr_datum := storage.main.root
	seq_num_end := storage.main.end.seq_num
	for curr_datum.seq_num != seq_num_end {
		_, err := output.Write(curr_datum.data)
		if err != nil {
			return err
		}
		curr_datum = curr_datum.next
	}
	storage.main.root = curr_datum
	return nil
}

func (buffer *DatumLink) InRange(new_seq_num uint32) (bool) {
	start := buffer.root.seq_num
	end := buffer.end.seq_num
	if new_seq_num == start || new_seq_num == end {
		return true
	}
	if !((start < new_seq_num && new_seq_num - start < (1 << 30)) || (start > new_seq_num && start - new_seq_num  > (1 << 30))) {
		return false
	}
	if !((new_seq_num < end && end - new_seq_num < (1 << 30)) || (new_seq_num > end && new_seq_num - end > (1 << 30))) {
		return false
	}
	return true
}

func (storage *DatumStorage) NewDatum(pkt *Packet) {
	new_pkt_num := pkt.header_info.(*DataHeader).seq_num
	prev_num := (new_pkt_num - 1) % uint32(1 << 31)
	if storage.main.end.seq_num == prev_num {
		storage.main.NewDatum(pkt)
	} else if storage.main.InRange(new_pkt_num) {
		return
	} else {
		oldest := storage.main.root.seq_num
		if (new_pkt_num < oldest && oldest - new_pkt_num < (1 << 30)) || (new_pkt_num > oldest && new_pkt_num - oldest > (1 << 30)) {
			return
		}
		for _, v := range storage.offshoots {
			if v.end.seq_num == prev_num {
				v.NewDatum(pkt)
				return
			} else if v.InRange(new_pkt_num) {
				return
			}
		}
		new_link := NewDatumLink(pkt)
		storage.offshoots = append(storage.offshoots, new_link)
	}
}

func (buffer *DatumLink) Link(buffer_next *DatumLink) {
	buffer.end.next = buffer_next.root
	buffer.end = buffer_next.end
	buffer.queued += buffer_next.queued
}

func check_append_serial_next(buffer *DatumLink, buffer_next *DatumLink) (bool) {
	seq_1 := buffer.end.seq_num
	seq_2 := (seq_1 + 1) % uint32(math.Pow(2, 31))
	if buffer_next.root.seq_num == seq_2 {
		buffer.Link(buffer_next)
		return true
	}
	return false
}

func (storage *DatumStorage) GenNACKCIF() (*NACKCIF, bool) {
	if len(storage.offshoots) == 0 {
		return nil, false
	}
	cif := new(NACKCIF)
	init_range := new(pckts_range)
	init_range.start = (storage.main.end.seq_num + 1) % (1 << 31)
	init_range.end = (storage.offshoots[0].root.seq_num - 1) % (1 << 31)
	cif.lost_pkts = append(cif.lost_pkts, init_range)
	for i := 0; i < len(storage.offshoots) - 1; i++ {
		new_range := new(pckts_range)
		new_range.start = (storage.offshoots[i].end.seq_num + 1) % (1 << 31)
		new_range.end = (storage.offshoots[i + 1].root.seq_num - 1) % (1 << 31)
		cif.lost_pkts = append(cif.lost_pkts, new_range)
	}
	return cif, true
}

func (storage *DatumStorage) Relink() {
	sort.Sort(storage.offshoots)
	buffer := storage.main
	i := 0
	for i < len(storage.offshoots) {
		if check_append_serial_next(buffer, storage.offshoots[i]) {
			storage.offshoots = append(storage.offshoots[:i], storage.offshoots[i + 1:]...)
		} else {
			buffer = storage.offshoots[i]
			i++
		}
	}
}