From e4a15835837de48b316fb58c39ba1e3c9e616d3d Mon Sep 17 00:00:00 2001
From: saxon <saxon.milton@gmail.com>
Date: Thu, 5 Jul 2018 16:25:46 +0930
Subject: [PATCH] Working on porting rtmp write func

---
 rtmp/rtmp.go | 123 ++++++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 116 insertions(+), 7 deletions(-)

diff --git a/rtmp/rtmp.go b/rtmp/rtmp.go
index 0d15e37e..163567c9 100644
--- a/rtmp/rtmp.go
+++ b/rtmp/rtmp.go
@@ -45,6 +45,9 @@ import (
 	"errors"
 	"strconv"
 	"unsafe"
+	"log"
+
+	"bitbucket.org/ausocean/av/tools"
 )
 
 // Session provides an interface for sending flv tags over rtmp.
@@ -86,30 +89,136 @@ func (s *session) Open() error {
 	return nil
 }
 
-/*
-func (s *session) rtmpWrite(rtmp *C.RTMP, data []byte) bool {
+func AVC(str string) *C.char {
+	slice := [2]byte{str, C.sizeof(str)-1}
+	return (*C.char)(unsafe.Pointer(&slice[0]))
+}
+
+func (s *session) rtmpWrite(r *C.RTMP, buf []byte) int {
 	var pkt *C.RTMPPacket = &rtmp.m_write
 	var pend, enc *C.char
-	s2 := len(data)
+	s2 := len(buf)
 	var ret, num int
 
+	pkt.m_nChannel = 0x04
+	pkt.m_nInfoField2 = r.m_stream_id
 
+	for s2 > 0 {
+		if !pkt.m_nBytesRead {
+			if size < 11 {
+				log.Printf("size: %d\n", size)
+				log.Printf("too small \n")
+				return 0
+			}
+
+			if buf[0] == 'F' && buf[1] == 'L' && buf[2] == 'V' {
+				buf += 13
+				s2 -= 13
+			}
+
+			// pkt.m_packetType = *buf++
+			buf = buf[1:]
+			pkt.m_packetType = buf[0]
+
+			cChar := (*C.char)(unsafe.Pointer(&buf[0]))
+			pkt.m_nBodySize = C.AMF_DecodeInt24(cChar)
+
+			// C: buf+=3
+			buf = buf[3:]
+
+			cChar = (*C.char)(unsafe.Pointer(&buf[0]))
+			pkt.m_nTimeStamp = C.AMF_DecodeInt24(cChar)
+
+			// C: buf+=3
+			buf = buf[3:]
+
+			// C: pkt->m_nTimeStamp |= *buf++ << 24;
+			buf = buf[1:]
+			pkt.m_nTimeStamp |= ( buf[0] << 24 )
+
+			// C: buf+=3
+			buf = buf[3:]
+
+			s2 -= 11
+
+			if ((pkt.m_packetType == C.RTMP_PACKET_TYPE_AUDIO ||
+				pkt.m_packetType == C.RTMP_PACKET_TYPE_VIDEO) &&
+				!pkt.m_nTimeStamp) || pkt.m_packetType == C.RTMP_PACKET_TYPE_INFO {
+
+					pkt.m_headerType = C.RTMP_PACKET_SIZE_LARGE
+		      if pkt.m_packetType == C.RTMP_PACKET_TYPE_INFO {
+						pkt.m_nBodySize += 16
+					}
+			} else {
+				pkt.m_headerType = C.RTMP_PACKET_SIZE_MEDIUM
+			}
+
+			// C: if (!RTMPPacket_Alloc(pkt, pkt->m_nBodySize))
+			if !helpers.UintToBool(uint( C.RTMPPacket_Alloc(pkt, pkt.m_nBodySize))) {
+				log.Println("Failed to allocated packet")
+				return 0
+			}
+
+			// Note this is pointer arithmatic
+			enc = pkt.m_body
+			pend = enc + pkt.m_nBodySize
+
+			if pkt.m_packetType == C.RTMP_PACKET_TYPE_INFO {
+				// av_setDataFrame is a static const global in rtmp.c
+				av_setDataFram := AVC("@setDataFrame")
+				enc = C.AMF_EncodeString(enc, pend, &av_setDataFrame )
+				pkt.m_nBytesRead = enc - pkt.m_body
+			}
+		} else {
+			enc = pkt.m_body + pkt.m_nBytesRead
+		}
+
+		num = pkt.m_nBodySize - pkt.m_nBytesRead
+
+		if num > s2 {
+			num = s2
+		}
+		//	C.memcpy(enc, buf, num)
+		copy(enc,buf)
+
+		pkt.m_nBytesRead += num
+		s2 -= num
+		buf += num
+
+		if pkt.m_nBytesRead == pkt.m_nBodySize {
+			ret = C.RTMP_SendPacket(r, pkt, 0)
+			c.RTMPPacket_Free(pkt)
+			pkt.m_nBytesRead = 0
+			if !uintToBool(ret) {
+				return -1
+			}
+			// buf += 4
+			buf = buf[4:]
+			s2 -= 4
+			if s2 < 0 {
+				break
+			}
+		}
+	}
+	return size+s2
 }
-*/
+
 
 func (s *session) writeFrame(data []byte) uint {
 	if C.RTMP_IsConnected(s.rtmp) <= 0 {
 		return 1
 	}
+	/*
 	// If RTMP_Write returns less than or equal to zero then something is wrong
 	if C.RTMP_Write(s.rtmp, (*C.char)(unsafe.Pointer(&data[0])), C.int(len(data))) <= 0 {
 		return 2
 	}
-	/*
-	if rtmpWrite(s.rtmp, data) {
+	*/
+
+	if rtmpWrite(s.rtmp, data) <= 0 {
 		return 2
 	}
-	*/
+
 	return 0
 }