From fabcecae439c84b66e1e188e894772729c8e62bb Mon Sep 17 00:00:00 2001
From: Muaz Ahmad <m.ahmed_2000@hotmail.com>
Date: Thu, 10 Aug 2023 13:58:57 +0500
Subject: [PATCH] Refactor to export structs chnk_stream and message

---
 rtmp/chunk.go | 32 ++++++++++++++++----------------
 1 file changed, 16 insertions(+), 16 deletions(-)

diff --git a/rtmp/chunk.go b/rtmp/chunk.go
index f22c80f..870aec4 100644
--- a/rtmp/chunk.go
+++ b/rtmp/chunk.go
@@ -6,7 +6,7 @@ import (
 	"encoding/binary"
 )
 
-type chnk_stream struct {
+type ChunkStream struct {
 	timestamp uint32
 	last_msg_strm_id uint32
 	last_msg_len uint32
@@ -14,7 +14,7 @@ type chnk_stream struct {
 	timedelta uint32
 }
 
-type message struct {
+type Message struct {
 	data []byte
 	curr_bytes_read uint32
 	timestamp uint32
@@ -43,9 +43,9 @@ func new_chunk_bufs() (*chunk_bufs) {
 	return &new_chunk_bufs
 }
 
-func OpenStreamsMapInit() (map[uint32]*chnk_stream, map[uint32]*message, *chunk_bufs) {
-	open_chnkstrms := make(map[uint32]*chnk_stream)
-	open_msgs := make(map[uint32]*message)
+func OpenStreamsMapInit() (map[uint32]*ChunkStream, map[uint32]*Message, *chunk_bufs) {
+	open_chnkstrms := make(map[uint32]*ChunkStream)
+	open_msgs := make(map[uint32]*Message)
 	chunk_bufs_ptr := new_chunk_bufs()
 	return open_chnkstrms, open_msgs, chunk_bufs_ptr
 }
@@ -116,7 +116,7 @@ func read_time_extd(conn net.Conn, chunk_bufs_ptr *chunk_bufs) (time uint32, err
 	return
 }
 
-func read_message_header_0(conn net.Conn, chnk_stream_ptr *chnk_stream, msg_ptr *message, chunk_bufs_ptr *chunk_bufs) (error) {
+func read_message_header_0(conn net.Conn, chnk_stream_ptr *ChunkStream, msg_ptr *Message, chunk_bufs_ptr *chunk_bufs) (error) {
 	var extended_time bool
 	var err error
 	chnk_stream_ptr.timestamp, extended_time, err = read_time(conn, chunk_bufs_ptr)
@@ -153,7 +153,7 @@ func read_message_header_0(conn net.Conn, chnk_stream_ptr *chnk_stream, msg_ptr
 	return nil
 }
 
-func read_message_header_1(conn net.Conn, chnk_stream_ptr *chnk_stream, msg_ptr *message, chunk_bufs_ptr *chunk_bufs) (error) {
+func read_message_header_1(conn net.Conn, chnk_stream_ptr *ChunkStream, msg_ptr *Message, chunk_bufs_ptr *chunk_bufs) (error) {
 	var extended_time bool
 	var err error
 
@@ -186,7 +186,7 @@ func read_message_header_1(conn net.Conn, chnk_stream_ptr *chnk_stream, msg_ptr
 	return nil
 }
 
-func read_message_header_2(conn net.Conn, chnk_stream_ptr *chnk_stream, msg_ptr *message, chunk_bufs_ptr *chunk_bufs) (error) {
+func read_message_header_2(conn net.Conn, chnk_stream_ptr *ChunkStream, msg_ptr *Message, chunk_bufs_ptr *chunk_bufs) (error) {
 	var extended_time bool
 	var err error
 
@@ -212,7 +212,7 @@ func read_message_header_2(conn net.Conn, chnk_stream_ptr *chnk_stream, msg_ptr
 	return nil
 }
 
-func read_chunk_data(conn net.Conn, msg_ptr *message, chnk_size uint32) (error) {
+func read_chunk_data(conn net.Conn, msg_ptr *Message, chnk_size uint32) (error) {
 	bytes_left := msg_ptr.msg_len - msg_ptr.curr_bytes_read
 	var buffer_end uint32
 	if bytes_left < chnk_size {
@@ -229,11 +229,11 @@ func read_chunk_data(conn net.Conn, msg_ptr *message, chnk_size uint32) (error)
 
 }
 
-func ReadChunk(conn net.Conn, open_chnkstrms map[uint32]*chnk_stream, open_msgs map[uint32]*message, chnk_size uint32, chunk_bufs_ptr *chunk_bufs) (*message, error){
+func ReadChunk(conn net.Conn, open_chnkstrms map[uint32]*ChunkStream, open_msgs map[uint32]*Message, chnk_size uint32, chunk_bufs_ptr *chunk_bufs) (*Message, error){
 	conn.SetDeadline(time.Now().Add(10 * time.Second))
 	
-	var chnkstream_ptr *chnk_stream
-	var msg_ptr *message
+	var chnkstream_ptr *ChunkStream
+	var msg_ptr *Message
 
 	format, csid, err := read_basic_header(conn, chunk_bufs_ptr)
 	if err != nil {
@@ -242,23 +242,23 @@ func ReadChunk(conn net.Conn, open_chnkstrms map[uint32]*chnk_stream, open_msgs
 
 	switch format {
 	case 0:
-		chnkstream_ptr = new(chnk_stream)
+		chnkstream_ptr = new(ChunkStream)
 		open_chnkstrms[csid] = chnkstream_ptr
-		msg_ptr = new(message)
+		msg_ptr = new(Message)
 
 		if err := read_message_header_0(conn, chnkstream_ptr, msg_ptr, chunk_bufs_ptr); err != nil {
 			return nil, err
 		}
 	case 1:
 		chnkstream_ptr = open_chnkstrms[csid]
-		msg_ptr = new(message)
+		msg_ptr = new(Message)
 
 		if err := read_message_header_1(conn, chnkstream_ptr, msg_ptr, chunk_bufs_ptr); err != nil {
 			return nil, err
 		}
 	case 2:
 		chnkstream_ptr = open_chnkstrms[csid]
-		msg_ptr = new(message)
+		msg_ptr = new(Message)
 
 		if err := read_message_header_2(conn, chnkstream_ptr, msg_ptr, chunk_bufs_ptr); err != nil {
 			return nil, err