From caad2d1d4fabb93c8a6b93a27dc74fd7d3f30109 Mon Sep 17 00:00:00 2001
From: Josh Deprez <josh.deprez@gmail.com>
Date: Fri, 15 Mar 2024 16:49:53 +1100
Subject: [PATCH] RI-Req, RI-Rsp, RI-Ack

---
 aurp/aurp.go | 37 +++++++++++++++++++++++++++++++++++++
 main.go      |  1 -
 2 files changed, 37 insertions(+), 1 deletion(-)

diff --git a/aurp/aurp.go b/aurp/aurp.go
index 0382595..036d620 100644
--- a/aurp/aurp.go
+++ b/aurp/aurp.go
@@ -403,6 +403,27 @@ func parseOpenRsp(p []byte) (*OpenRspPacket, error) {
 	}, nil
 }
 
+type RIReqPacket struct {
+	*Header
+}
+
+type RIRspPacket struct {
+	*Header
+
+	RTMPData []byte
+}
+
+func (p *RIRspPacket) WriteTo(w io.Writer) (int64, error) {
+	a := acc(w)
+	a.writeTo(p.Header)
+	a.write(p.RTMPData)
+	return a.ret()
+}
+
+type RIAckPacket struct {
+	*Header
+}
+
 // ParsePacket parses the body of a UDP packet for a domain header, and then
 // based on the packet type, an AURP-Tr header, an AURP routing header, and
 // then a particular packet type.
@@ -452,6 +473,22 @@ func ParsePacket(p []byte) (Packet, error) {
 			orsp.Header = h
 			return orsp, nil
 
+		case CmdCodeRIReq:
+			return &RIReqPacket{
+				Header: h,
+			}, nil
+
+		case CmdCodeRIRsp:
+			return &RIRspPacket{
+				Header:   h,
+				RTMPData: p,
+			}, nil
+
+		case CmdCodeRIAck:
+			return &RIAckPacket{
+				Header: h,
+			}, nil
+
 		default:
 			return nil, fmt.Errorf("unknown routing packet command code %d", h.CommandCode)
 		}
diff --git a/main.go b/main.go
index 35f0e6b..92d98b7 100644
--- a/main.go
+++ b/main.go
@@ -31,6 +31,5 @@ func main() {
 			log.Printf("Failed to read packet: %v", readErr)
 			continue
 		}
-
 	}
 }