Packet-byte handling, protocol struct to handle protocol states

srt/intake.go

@@ -27,7 +27,7 @@ func (intake *Intake) NewTunnel(l net.PacketConn, peer net.Addr) (*Tunnel) {
 		tunnel := new(Tunnel)
 		tunnel.socket = l
 		tunnel.peer = peer
-		tunnel.queue = make(chan *Packet, 10)
+		tunnel.queue = make(chan []byte, 10)
 		return tunnel
 	}
 	return nil
@@ -61,5 +61,7 @@ func (intake *Intake) Read() {
 		return
 	}
 	fmt.Println(string(intake.buffer[:n]))
-	tunnel.queue <- &(Packet{intake.buffer[:n]})
+	pkt := make([]byte, n)
+	copy(pkt, intake.buffer[:n])
+	tunnel.queue <- pkt
 }

srt/packet.go

@@ -6,3 +6,11 @@ import (
 type Packet struct {
 	raw []byte
 }
+
+func MarshallPacket(packet *Packet) ([]byte, error) {
+	return packet.raw, nil
+}
+
+func ParsePacket(buffer []byte) (*Packet, error) {
+	return &Packet{buffer}, nil
+}

srt/protocol.go

@@ -0,0 +1,20 @@
+package srt
+
+import (
+	"fmt"
+)
+
+const (
+	INDUCTION uint8 = iota
+	CONCLUSION
+	DATA_LOOP
+)
+
+type SRTManager struct {
+	state uint8
+}
+
+func (proto *SRTManager) Decide(packet *Packet) (*Packet, error) {
+	fmt.Println(*packet)
+	return nil, nil
+}

srt/tunnel.go

@@ -7,6 +7,38 @@ import (
 type Tunnel struct {
 	socket net.PacketConn
 	peer net.Addr
-	queue chan *Packet
+	queue chan []byte
 	broken bool
+	state *SRTManager
+}
+
+func (tunnel *Tunnel) Start() {
+	tunnel.state = new(SRTManager)
+	for {
+		packet, err := tunnel.ReadPacket()
+		if err != nil {
+			tunnel.broken = true
+		}
+		response, err := tunnel.state.Decide(packet)
+		if err != nil {
+			tunnel.broken = true
+		}
+		if response != nil {
+			tunnel.WritePacket(response)
+		}
+	}
+}
+
+func (tunnel *Tunnel) WritePacket(packet *Packet) {
+	buffer, err := MarshallPacket(packet)
+	if err != nil {
+		tunnel.broken = true
+		return
+	}
+	tunnel.socket.WriteTo(buffer, tunnel.peer)
+}
+
+func (tunnel *Tunnel) ReadPacket() (*Packet, error) {
+	packet := <- tunnel.queue
+	return ParsePacket(packet)
 }