Working on porting rtmp write func

This commit is contained in:
saxon 2018-07-05 16:25:46 +09:30
parent 157ac833db
commit e4a1583583
1 changed files with 116 additions and 7 deletions

View File

@ -45,6 +45,9 @@ import (
"errors" "errors"
"strconv" "strconv"
"unsafe" "unsafe"
"log"
"bitbucket.org/ausocean/av/tools"
) )
// Session provides an interface for sending flv tags over rtmp. // Session provides an interface for sending flv tags over rtmp.
@ -86,30 +89,136 @@ func (s *session) Open() error {
return nil return nil
} }
/* func AVC(str string) *C.char {
func (s *session) rtmpWrite(rtmp *C.RTMP, data []byte) bool { 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 pkt *C.RTMPPacket = &rtmp.m_write
var pend, enc *C.char var pend, enc *C.char
s2 := len(data) s2 := len(buf)
var ret, num int 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 { func (s *session) writeFrame(data []byte) uint {
if C.RTMP_IsConnected(s.rtmp) <= 0 { if C.RTMP_IsConnected(s.rtmp) <= 0 {
return 1 return 1
} }
/*
// If RTMP_Write returns less than or equal to zero then something is wrong // 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 { if C.RTMP_Write(s.rtmp, (*C.char)(unsafe.Pointer(&data[0])), C.int(len(data))) <= 0 {
return 2 return 2
} }
/* */
if rtmpWrite(s.rtmp, data) {
if rtmpWrite(s.rtmp, data) <= 0 {
return 2 return 2
} }
*/
return 0 return 0
} }