2018-05-29 08:56:36 +03:00
|
|
|
/*
|
|
|
|
NAME
|
|
|
|
rtmp.go
|
|
|
|
|
|
|
|
DESCRIPTION
|
|
|
|
See Readme.md
|
|
|
|
|
2018-08-24 03:55:36 +03:00
|
|
|
AUTHORS
|
2018-05-29 08:56:36 +03:00
|
|
|
Saxon Nelson-Milton <saxon@ausocean.org>
|
|
|
|
Dan Kortschak <dan@ausocean.org>
|
|
|
|
|
|
|
|
LICENSE
|
|
|
|
rtmp.go is Copyright (C) 2017 the Australian Ocean Lab (AusOcean)
|
|
|
|
|
|
|
|
It is free software: you can redistribute it and/or modify them
|
|
|
|
under the terms of the GNU General Public License as published by the
|
|
|
|
Free Software Foundation, either version 3 of the License, or (at your
|
|
|
|
option) any later version.
|
|
|
|
|
|
|
|
It is distributed in the hope that it will be useful, but WITHOUT
|
|
|
|
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
|
|
FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
|
|
for more details.
|
|
|
|
|
|
|
|
You should have received a copy of the GNU General Public License
|
2018-08-30 14:01:19 +03:00
|
|
|
along with revid in gpl.txt. If not, see http://www.gnu.org/licenses.
|
2018-08-23 23:44:37 +03:00
|
|
|
|
2018-08-30 14:01:19 +03:00
|
|
|
Derived from librtmp under the GNU Lesser General Public License 2.1
|
|
|
|
Copyright (C) 2005-2008 Team XBMC http://www.xbmc.org
|
|
|
|
Copyright (C) 2008-2009 Andrej Stepanchuk
|
|
|
|
Copyright (C) 2009-2010 Howard Chu
|
2018-05-29 08:56:36 +03:00
|
|
|
*/
|
|
|
|
|
|
|
|
package rtmp
|
|
|
|
|
|
|
|
/*
|
|
|
|
#include <stdlib.h>
|
2018-07-22 11:48:46 +03:00
|
|
|
#include <netinet/tcp.h>
|
2018-08-30 14:01:19 +03:00
|
|
|
#include <arpa/inet.h>
|
|
|
|
#include <netdb.h>
|
2018-07-17 13:05:25 +03:00
|
|
|
|
2018-07-21 02:24:16 +03:00
|
|
|
typedef struct sockaddr_in sockaddr_in;
|
|
|
|
typedef struct sockaddr sockaddr;
|
2018-05-29 08:56:36 +03:00
|
|
|
*/
|
|
|
|
import "C"
|
|
|
|
|
|
|
|
import (
|
2018-07-16 06:30:58 +03:00
|
|
|
"errors"
|
2018-07-18 05:51:31 +03:00
|
|
|
"fmt"
|
2018-07-16 06:30:58 +03:00
|
|
|
"log"
|
|
|
|
"math"
|
2018-07-24 15:00:50 +03:00
|
|
|
"math/rand"
|
2018-07-14 08:29:00 +03:00
|
|
|
"strconv"
|
2018-08-11 05:51:22 +03:00
|
|
|
"time"
|
2018-07-14 08:29:00 +03:00
|
|
|
"unsafe"
|
2018-07-24 15:00:50 +03:00
|
|
|
|
|
|
|
"github.com/chamaken/cgolmnl/inet"
|
2018-05-29 08:56:36 +03:00
|
|
|
)
|
|
|
|
|
2018-08-31 04:54:42 +03:00
|
|
|
const _Gi = 1 << 30
|
|
|
|
|
2018-08-08 10:36:16 +03:00
|
|
|
const (
|
|
|
|
minDataSize = 11
|
|
|
|
debugMode = false
|
2018-08-24 00:28:22 +03:00
|
|
|
length = 512
|
2018-07-13 18:43:17 +03:00
|
|
|
)
|
|
|
|
|
2018-07-15 10:20:56 +03:00
|
|
|
// av_setDataFrame is a static const global in rtmp.c
|
2018-07-26 10:36:44 +03:00
|
|
|
var (
|
2018-08-14 20:17:40 +03:00
|
|
|
AV_empty = AVC("")
|
2018-08-14 11:52:37 +03:00
|
|
|
setDataFrame = AVC("@setDataFrame")
|
|
|
|
av_connect = AVC("connect")
|
|
|
|
av_app = AVC("app")
|
|
|
|
av_type = AVC("type")
|
|
|
|
av_nonprivate = AVC("nonprivate")
|
|
|
|
av_flashVer = AVC("flashVer")
|
|
|
|
av_swfUrl = AVC("swfUrl")
|
|
|
|
av_tcUrl = AVC("tcUrl")
|
|
|
|
av_fpad = AVC("fpad")
|
|
|
|
av_capabilities = AVC("capabilities")
|
|
|
|
av_audioCodecs = AVC("audioCodecs")
|
|
|
|
av_videoCodecs = AVC("videoCodecs")
|
|
|
|
av_videoFunction = AVC("videoFunction")
|
|
|
|
av_pageUrl = AVC("pageUrl")
|
|
|
|
av_objectEncoding = AVC("objectEncoding")
|
|
|
|
av__result = AVC("_result")
|
|
|
|
av_secureToken = AVC("secureToken")
|
|
|
|
av_createStream = AVC("createStream")
|
2018-08-14 21:30:49 +03:00
|
|
|
av_releaseStream = AVC("releaseStream")
|
2018-08-14 11:52:37 +03:00
|
|
|
av_play = AVC("play")
|
|
|
|
av_publish = AVC("publish")
|
|
|
|
av_onBWDone = AVC("onBWDone")
|
|
|
|
av_onFCSubscribe = AVC("onFCSubscribe")
|
|
|
|
av_onFCUnsubscribe = AVC("onFCUnsubscribe")
|
|
|
|
av__onbwcheck = AVC("_onbwcheck")
|
|
|
|
av__onbwdone = AVC("_onbwdone")
|
|
|
|
av_ping = AVC("ping")
|
|
|
|
av__checkbw = AVC("_checkbw")
|
|
|
|
av_close = AVC("close")
|
|
|
|
av_onStatus = AVC("onStatus")
|
|
|
|
av_code = AVC("code")
|
|
|
|
av_level = AVC("level")
|
2018-08-14 16:00:47 +03:00
|
|
|
av_NetStream_Failed = AVC("NetStream.Failed")
|
|
|
|
av_NetStream_Play_Failed = AVC("NetStream.Play.Failed")
|
|
|
|
av_NetConnection_Connect_InvalidApp = AVC("NetConnection.Connect.InvalidApp")
|
|
|
|
av_NetStream_Play_StreamNotFound = AVC("NetStream.Play.StreamNotFound")
|
|
|
|
av_NetStream_Play_Start = AVC("NetStream.Play.Start")
|
|
|
|
av_NetStream_Play_PublishNotify = AVC("NetStream.Play.PublishNotify")
|
|
|
|
av_NetStream_Publish_Start = AVC("NetStream.Publish.Start")
|
|
|
|
av_NetStream_Play_Complete = AVC("NetStream.Play.Complete")
|
|
|
|
av_NetStream_Play_Stop = AVC("NetStream.Play.Stop")
|
|
|
|
av_NetStream_Play_UnpublishNotify = AVC("NetStream.Play.UnpublishNotify")
|
|
|
|
av_NetStream_Seek_Notify = AVC("NetStream.Seek.Notify")
|
|
|
|
av_NetStream_Pause_Notify = AVC("NetStream.Pause.Notify")
|
2018-08-14 11:52:37 +03:00
|
|
|
av_playlist_ready = AVC("playlist_ready")
|
|
|
|
av_set_playlist = AVC("set_playlist")
|
2018-08-14 21:35:15 +03:00
|
|
|
av_FCPublish = AVC("FCPublish")
|
2018-08-14 21:49:18 +03:00
|
|
|
av_live = AVC("live")
|
2018-07-26 10:36:44 +03:00
|
|
|
)
|
2018-07-16 06:30:58 +03:00
|
|
|
|
2018-08-24 00:28:22 +03:00
|
|
|
var RTMPT_cmds = []string{
|
|
|
|
"open",
|
|
|
|
"send",
|
|
|
|
"idle",
|
|
|
|
"close",
|
|
|
|
}
|
|
|
|
|
2018-08-14 20:39:57 +03:00
|
|
|
var (
|
|
|
|
packetSize = [...]int{12, 8, 4, 1}
|
|
|
|
RTMPProtocolStringsLower = [...]string{
|
|
|
|
"rtmp",
|
|
|
|
"rtmpt",
|
|
|
|
"rtmpe",
|
|
|
|
"rtmpte",
|
|
|
|
"rtmps",
|
|
|
|
"rtmpts",
|
|
|
|
"",
|
|
|
|
"",
|
|
|
|
"rtmfp",
|
|
|
|
}
|
|
|
|
)
|
2018-07-19 18:35:21 +03:00
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
func startSession(rtmp *C_RTMP, u string, timeout uint32) (*C_RTMP, error) {
|
|
|
|
connect_timeout := int32(timeout)
|
2018-08-06 09:12:16 +03:00
|
|
|
rtmp = C_RTMP_Alloc()
|
2018-07-28 08:50:30 +03:00
|
|
|
//rtmp = C.RTMP_Alloc()
|
2018-08-07 05:34:04 +03:00
|
|
|
C_RTMP_Init(rtmp)
|
2018-07-28 08:52:15 +03:00
|
|
|
//C.RTMP_Init(rtmp)
|
2018-07-19 09:35:14 +03:00
|
|
|
rtmp.Link.timeout = connect_timeout
|
2018-08-07 07:49:11 +03:00
|
|
|
if C_RTMP_SetupURL(rtmp, u) == 0 {
|
2018-07-28 08:53:25 +03:00
|
|
|
// if C.RTMP_SetupURL(rtmp, C.CString(u)) == 0 {
|
2018-08-25 17:19:54 +03:00
|
|
|
C_RTMP_Close(rtmp)
|
2018-07-28 08:54:36 +03:00
|
|
|
//C.RTMP_Free(rtmp)
|
2018-07-19 09:35:14 +03:00
|
|
|
return nil, errors.New("rtmp startSession: Failed to setup URL!")
|
|
|
|
}
|
|
|
|
|
2018-08-07 05:37:42 +03:00
|
|
|
C_RTMP_EnableWrite(rtmp)
|
2018-07-28 08:55:31 +03:00
|
|
|
//C.RTMP_EnableWrite(rtmp)
|
2018-08-07 05:38:48 +03:00
|
|
|
C_RTMP_SetBufferMS(rtmp, 3600*1000)
|
2018-07-28 08:56:38 +03:00
|
|
|
//C.RTMP_SetBufferMS(rtmp, 3600*1000)
|
2018-08-07 05:42:11 +03:00
|
|
|
if C_RTMP_Connect(rtmp, nil) == 0 {
|
2018-07-28 08:59:05 +03:00
|
|
|
//if C.RTMP_Connect(rtmp, nil) == 0 {
|
2018-08-25 17:19:54 +03:00
|
|
|
C_RTMP_Close(rtmp)
|
2018-07-28 08:54:36 +03:00
|
|
|
//C.RTMP_Free(rtmp)
|
2018-07-19 09:35:14 +03:00
|
|
|
return nil, errors.New("rtmp startSession: Failed to connect!")
|
|
|
|
}
|
|
|
|
|
2018-07-22 11:48:46 +03:00
|
|
|
// TODO: port this
|
2018-08-10 16:00:22 +03:00
|
|
|
if C_RTMP_ConnectStream(rtmp, 0) == 0 {
|
|
|
|
//if C.RTMP_ConnectStream(rtmp, 0) == 0 {
|
2018-08-25 17:19:54 +03:00
|
|
|
C_RTMP_Close(rtmp)
|
2018-07-28 08:54:36 +03:00
|
|
|
//C.RTMP_Free(rtmp)
|
2018-07-19 09:35:14 +03:00
|
|
|
return nil, errors.New("rtmp startSession: Failed to connect stream!")
|
|
|
|
}
|
|
|
|
|
|
|
|
return rtmp, nil
|
|
|
|
}
|
2018-08-07 07:49:11 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
func endSession(rtmp *C_RTMP) uint32 {
|
|
|
|
if rtmp == nil {
|
|
|
|
return 3
|
|
|
|
}
|
|
|
|
|
2018-08-25 17:19:54 +03:00
|
|
|
C_RTMP_Close(rtmp)
|
2018-08-24 12:20:04 +03:00
|
|
|
//C.RTMP_Free(rtmp)
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2018-08-24 04:13:34 +03:00
|
|
|
// uint32_t RTMP_GetTime();
|
|
|
|
// rtmp.c +156
|
|
|
|
func C_RTMP_GetTime() int32 {
|
|
|
|
return int32(time.Now().UnixNano() / 1000000)
|
|
|
|
}
|
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// int RTMPPacket_Alloc(RTMPPacket* p, uint32_t nSize);
|
|
|
|
// rtmp.c +189
|
|
|
|
func C_RTMPPacket_Alloc(p *C_RTMPPacket, nSize uint32) int {
|
|
|
|
var ptr *byte
|
|
|
|
// TODO: port C.SIZE_MAX
|
|
|
|
// also work out how to deal with the overfloat
|
|
|
|
/*
|
|
|
|
if int64(nSize) > (C.SIZE_MAX - RTMP_MAX_HEADER_SIZE) {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
*/
|
|
|
|
ptr = (*byte)(calloc(1, uintptr(nSize+RTMP_MAX_HEADER_SIZE)))
|
|
|
|
|
|
|
|
if ptr == nil {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
p.m_body = (*byte)(incBytePtr(unsafe.Pointer(ptr), RTMP_MAX_HEADER_SIZE))
|
|
|
|
p.m_nBytesRead = 0
|
|
|
|
return 1
|
|
|
|
}
|
|
|
|
|
|
|
|
// void RTMPPacket_Free(RTMPPacket* p);
|
|
|
|
// rtmp.c +203
|
|
|
|
func C_RTMPPacket_Free(p *C_RTMPPacket) {
|
|
|
|
if p.m_body != nil {
|
|
|
|
//C.free(decBytePtr(unsafe.Pointer(p.m_body), RTMP_MAX_HEADER_SIZE))
|
|
|
|
p.m_body = nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-07 07:49:11 +03:00
|
|
|
// RTMP* RTMP_IsConnected();
|
|
|
|
// rtmp.c +317
|
2018-08-18 07:21:07 +03:00
|
|
|
func C_RTMP_Alloc() *C_RTMP {
|
2018-08-25 16:02:24 +03:00
|
|
|
return &C_RTMP{}
|
2018-08-18 07:21:07 +03:00
|
|
|
//return (*C_RTMP)(allocate(unsafe.Sizeof(r)))
|
2018-07-19 10:29:03 +03:00
|
|
|
}
|
2018-07-19 10:54:37 +03:00
|
|
|
|
2018-08-07 07:49:11 +03:00
|
|
|
// void RTMP_Init(RTMP *r);
|
|
|
|
// rtmp.c +329
|
2018-08-18 07:21:07 +03:00
|
|
|
func C_RTMP_Init(r *C_RTMP) {
|
2018-08-25 16:02:24 +03:00
|
|
|
*r = C_RTMP{}
|
2018-07-19 11:00:30 +03:00
|
|
|
r.m_sb.sb_socket = -1
|
|
|
|
r.m_inChunkSize = RTMP_DEFAULT_CHUNKSIZE
|
|
|
|
r.m_outChunkSize = RTMP_DEFAULT_CHUNKSIZE
|
|
|
|
r.m_nBufferMS = 30000
|
|
|
|
r.m_nClientBW = 2500000
|
|
|
|
r.m_nClientBW2 = 2
|
|
|
|
r.m_nServerBW = 2500000
|
|
|
|
r.m_fAudioCodecs = 3191.0
|
|
|
|
r.m_fVideoCodecs = 252.0
|
|
|
|
r.Link.timeout = 30
|
|
|
|
r.Link.swfAge = 30
|
|
|
|
}
|
|
|
|
|
2018-08-24 04:13:34 +03:00
|
|
|
// void RTMP_EnableWrite(RTMP *r);
|
|
|
|
// rtmp.c +351
|
|
|
|
func C_RTMP_EnableWrite(r *C_RTMP) {
|
|
|
|
r.Link.protocol |= RTMP_FEATURE_WRITE
|
|
|
|
}
|
|
|
|
|
|
|
|
// int RTMP_IsConnected(RTMP *r);
|
|
|
|
// rtmp.c +363
|
|
|
|
func C_RTMP_IsConnected(r *C_RTMP) int32 {
|
|
|
|
if r.m_sb.sb_socket != -1 {
|
|
|
|
return 1
|
|
|
|
}
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
|
|
|
// void RTMP_SetBufferMS(RTMP *r, int size);
|
|
|
|
// rtmp.c +381
|
|
|
|
func C_RTMP_SetBufferMS(r *C_RTMP, size int32) {
|
|
|
|
r.m_nBufferMS = int32(size)
|
|
|
|
}
|
|
|
|
|
|
|
|
// void SocksSetup(RTMP *r, C_AVal* sockshost);
|
|
|
|
// rtmp.c +410
|
|
|
|
func C_SocksSetup(r *C_RTMP, sockshost *C_AVal) {
|
|
|
|
if sockshost.av_len != 0 {
|
|
|
|
socksport := strchr((*byte)(unsafe.Pointer(sockshost.av_val)), ':')
|
|
|
|
hostname := strdup((*byte)(unsafe.Pointer(sockshost.av_val)))
|
|
|
|
|
|
|
|
if unsafe.Pointer(socksport) != nil {
|
2018-09-01 17:00:26 +03:00
|
|
|
(*[_Gi]byte)(unsafe.Pointer(hostname))[uintptr(decBytePtr(unsafe.Pointer(socksport),
|
|
|
|
int(uintptr(unsafe.Pointer(sockshost.av_val)))))] = '\000'
|
2018-08-24 04:13:34 +03:00
|
|
|
r.Link.sockshost.av_val = (*byte)(unsafe.Pointer(hostname))
|
|
|
|
r.Link.sockshost.av_len = int32(strlen(hostname))
|
|
|
|
|
2018-08-31 05:12:22 +03:00
|
|
|
value, err := strconv.Atoi(string((*[_Gi]byte)(unsafe.Pointer(uintptr(
|
2018-09-01 17:00:26 +03:00
|
|
|
unsafe.Pointer(socksport)) + uintptr(1)))[:strlen((*byte)(unsafe.Pointer(
|
|
|
|
uintptr(unsafe.Pointer(socksport))+uintptr(1))))+1]))
|
2018-08-24 04:13:34 +03:00
|
|
|
if err != nil {
|
|
|
|
log.Println("C_SocksSetup: bad string conversion!")
|
|
|
|
}
|
|
|
|
if uintptr(unsafe.Pointer(socksport)) == 0 {
|
|
|
|
value = 1080
|
|
|
|
}
|
|
|
|
r.Link.socksport = uint16(value)
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
r.Link.sockshost.av_val = nil
|
|
|
|
r.Link.sockshost.av_len = 0
|
|
|
|
r.Link.socksport = 0
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-07 07:49:11 +03:00
|
|
|
// int RTMP_SetupURL(RTMP *r, char* url);
|
|
|
|
// rtmp.c +757
|
2018-08-08 10:47:59 +03:00
|
|
|
// NOTE: code dealing with rtmp over http has been disregarded
|
2018-08-18 07:21:07 +03:00
|
|
|
func C_RTMP_SetupURL(r *C_RTMP, u string) int32 {
|
2018-07-19 18:35:21 +03:00
|
|
|
url := goStrToCStr(u)
|
|
|
|
|
2018-08-10 10:13:19 +03:00
|
|
|
var ret, length int32
|
2018-07-19 18:35:21 +03:00
|
|
|
var port uint32
|
|
|
|
port = 0
|
|
|
|
|
2018-08-10 10:13:19 +03:00
|
|
|
length = strlen(url)
|
2018-07-28 03:45:12 +03:00
|
|
|
// TODO: port this
|
2018-08-18 07:21:07 +03:00
|
|
|
//ret = int32(C.RTMP_ParseURL((*byte)(unsafe.Pointer(url)), &r.Link.protocol,
|
2018-08-10 18:47:48 +03:00
|
|
|
// &r.Link.hostname, (*C.uint)(&port), &r.Link.playpath0, &r.Link.app))
|
|
|
|
ret = int32(C_RTMP_ParseURL((*byte)(unsafe.Pointer(url)), (*int32)(
|
|
|
|
unsafe.Pointer(&r.Link.protocol)), &r.Link.hostname, (*uint32)(
|
|
|
|
unsafe.Pointer(&port)), &r.Link.playpath0, &r.Link.app))
|
2018-07-19 18:35:21 +03:00
|
|
|
if ret == 0 {
|
|
|
|
return ret
|
2018-07-15 21:09:36 +03:00
|
|
|
}
|
2018-07-19 18:35:21 +03:00
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
r.Link.port = uint16(port)
|
2018-07-19 18:35:21 +03:00
|
|
|
r.Link.playpath = r.Link.playpath0
|
|
|
|
|
|
|
|
if r.Link.tcUrl.av_len == 0 {
|
2018-08-18 07:21:07 +03:00
|
|
|
r.Link.tcUrl.av_val = (*byte)(unsafe.Pointer(url))
|
2018-07-19 18:35:21 +03:00
|
|
|
if r.Link.app.av_len != 0 {
|
|
|
|
if int(uintptr(unsafe.Pointer(r.Link.app.av_val))) <
|
2018-08-10 10:13:19 +03:00
|
|
|
int(uintptr(incBytePtr(unsafe.Pointer(url), int(length)))) {
|
2018-07-19 18:35:21 +03:00
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
r.Link.tcUrl.av_len = int32(int(r.Link.app.av_len) +
|
2018-07-20 08:44:39 +03:00
|
|
|
int(uintptr(decBytePtr(unsafe.Pointer(r.Link.app.av_val),
|
|
|
|
int(uintptr(unsafe.Pointer(url)))))))
|
2018-07-19 18:35:21 +03:00
|
|
|
} else {
|
2018-08-10 10:13:19 +03:00
|
|
|
length = int32(r.Link.hostname.av_len) + int32(r.Link.app.av_len) +
|
|
|
|
int32(len("rtmpte://:65535/\x00"))
|
2018-07-19 18:35:21 +03:00
|
|
|
|
2018-08-18 07:26:10 +03:00
|
|
|
r.Link.tcUrl.av_val = (*byte)(malloc(uintptr(uintptr(length))))
|
2018-09-01 17:00:26 +03:00
|
|
|
hostname := string((*[_Gi]byte)(unsafe.Pointer(r.Link.hostname.av_val))[:r.Link.hostname.av_len])
|
2018-07-19 18:35:21 +03:00
|
|
|
|
2018-09-01 17:00:26 +03:00
|
|
|
app := string((*[_Gi]byte)(unsafe.Pointer(r.Link.app.av_val))[:r.Link.app.av_len])
|
2018-07-19 18:35:21 +03:00
|
|
|
|
|
|
|
fString := fmt.Sprintf("%v://%v:%v/%v",
|
|
|
|
RTMPProtocolStringsLower[r.Link.protocol], hostname, r.Link.port, app)
|
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
r.Link.tcUrl.av_val = (*byte)(bToUP(goStrToCStr(fString)))
|
|
|
|
r.Link.tcUrl.av_len = int32(strLen(RTMPProtocolStringsLower[r.Link.protocol]) +
|
2018-07-20 08:44:39 +03:00
|
|
|
strLen(string("://")) + strLen(hostname) + strLen(string(":")) +
|
|
|
|
strLen(strconv.Itoa(int(r.Link.port))) + strLen(string("/")) + strLen(app))
|
2018-07-19 18:35:21 +03:00
|
|
|
|
|
|
|
r.Link.lFlags |= RTMP_LF_FTCU
|
|
|
|
}
|
|
|
|
} else {
|
2018-08-18 07:21:07 +03:00
|
|
|
r.Link.tcUrl.av_len = int32(strlen(url))
|
2018-07-19 18:35:21 +03:00
|
|
|
}
|
2018-07-15 21:09:36 +03:00
|
|
|
}
|
2018-07-22 17:01:19 +03:00
|
|
|
|
2018-08-07 05:35:58 +03:00
|
|
|
C_SocksSetup(r, &r.Link.sockshost)
|
2018-07-19 18:35:21 +03:00
|
|
|
|
|
|
|
if r.Link.port == 0 {
|
|
|
|
switch {
|
2018-07-20 08:44:39 +03:00
|
|
|
case (r.Link.protocol & RTMP_FEATURE_SSL) != 0:
|
2018-07-19 18:35:21 +03:00
|
|
|
r.Link.port = 433
|
2018-07-20 08:44:39 +03:00
|
|
|
case (r.Link.protocol & RTMP_FEATURE_HTTP) != 0:
|
2018-07-19 18:35:21 +03:00
|
|
|
r.Link.port = 80
|
|
|
|
default:
|
|
|
|
r.Link.port = 1935
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return 1
|
2018-07-15 21:09:36 +03:00
|
|
|
}
|
|
|
|
|
2018-08-30 14:01:19 +03:00
|
|
|
// int add_addr_info(struct sockaddr_in *service, AVal *host, int port)
|
|
|
|
// rtmp.c +869
|
|
|
|
func C_add_addr_info(service *C.sockaddr_in, host *C_AVal, port C.int) int32 {
|
|
|
|
var hostname *byte
|
|
|
|
if (*[1 << 20]byte)(unsafe.Pointer(host.av_val))[host.av_len] != 0 {
|
|
|
|
name := make([]byte, host.av_len+1)
|
|
|
|
copy(name, (*[1 << 20]byte)(unsafe.Pointer(host.av_val))[:host.av_len])
|
|
|
|
name[len(name)-1] = 0
|
|
|
|
hostname = (*byte)(unsafe.Pointer(&name[0]))
|
|
|
|
} else {
|
|
|
|
hostname = host.av_val
|
|
|
|
}
|
|
|
|
|
|
|
|
service.sin_addr.s_addr = C.inet_addr((*C.char)(unsafe.Pointer(hostname)))
|
|
|
|
if service.sin_addr.s_addr == C.INADDR_NONE {
|
|
|
|
host := C.gethostbyname((*C.char)(unsafe.Pointer(hostname)))
|
|
|
|
if host == nil || *host.h_addr_list == nil {
|
|
|
|
//RTMP_Log(RTMP_LOGERROR, "Problem accessing the DNS. (addr: %s)", hostname)
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
service.sin_addr = *(*C.struct_in_addr)(unsafe.Pointer(*host.h_addr_list))
|
|
|
|
}
|
|
|
|
|
|
|
|
service.sin_port = C.ushort(inet.Htons(uint16(port)))
|
|
|
|
return 1
|
|
|
|
}
|
|
|
|
|
2018-08-07 07:49:11 +03:00
|
|
|
// int RTMP_Connect0(RTMP *r, struct sockaddr* service);
|
|
|
|
// rtmp.c +906
|
2018-08-18 07:21:07 +03:00
|
|
|
func C_RTMP_Connect0(r *C_RTMP, service *C.sockaddr) int {
|
2018-07-22 11:48:46 +03:00
|
|
|
on := 1
|
|
|
|
r.m_sb.sb_timedout = 0
|
|
|
|
r.m_pausing = 0
|
2018-08-10 07:08:33 +03:00
|
|
|
r.m_fDuration = 0
|
2018-07-22 11:48:46 +03:00
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
r.m_sb.sb_socket = int32(C.socket(C.AF_INET, C.SOCK_STREAM, C.IPPROTO_TCP))
|
2018-07-22 11:48:46 +03:00
|
|
|
|
|
|
|
if r.m_sb.sb_socket != -1 {
|
2018-08-18 07:21:07 +03:00
|
|
|
if C.connect(C.int(r.m_sb.sb_socket), service, C.socklen_t(unsafe.Sizeof(*service))) < 0 {
|
2018-08-07 05:42:11 +03:00
|
|
|
log.Println("C_RTMP_Connect0, failed to connect socket.")
|
2018-07-22 11:48:46 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
if r.Link.socksport != 0 {
|
|
|
|
if debugMode {
|
2018-08-07 05:42:11 +03:00
|
|
|
log.Println("C_RTMP_Connect0: socks negotiation.")
|
2018-07-22 11:48:46 +03:00
|
|
|
}
|
|
|
|
|
2018-08-11 05:14:55 +03:00
|
|
|
//if C.SocksNegotiate(r) == 0 {
|
|
|
|
if C_SocksNegotiate(r) == 0 {
|
2018-08-07 05:42:11 +03:00
|
|
|
log.Println("C_RTMP_Connect0: SOCKS negotiation failed.")
|
2018-07-22 11:48:46 +03:00
|
|
|
return 0
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} else {
|
2018-08-07 05:42:11 +03:00
|
|
|
log.Println("C_RTMP_Connect0: failed to create socket.")
|
2018-07-22 11:48:46 +03:00
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
|
|
|
{
|
2018-08-10 09:40:08 +03:00
|
|
|
var tv int32
|
|
|
|
SET_RCVTIMEO(&tv, int32(r.Link.timeout))
|
2018-08-18 07:21:07 +03:00
|
|
|
// tv := int32(r.Link.timeout * 1000)
|
2018-07-22 11:48:46 +03:00
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
if C.setsockopt(C.int(r.m_sb.sb_socket), C.SOL_SOCKET, C.SO_RCVTIMEO,
|
2018-07-24 15:00:50 +03:00
|
|
|
unsafe.Pointer(&tv), C.socklen_t(unsafe.Sizeof(tv))) != 0 {
|
2018-08-07 05:42:11 +03:00
|
|
|
log.Println("C_RTMP_Connect0: Setting socket timeout failed")
|
2018-07-22 11:48:46 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
C.setsockopt(C.int(r.m_sb.sb_socket), C.IPPROTO_TCP, C.TCP_NODELAY,
|
2018-07-24 15:00:50 +03:00
|
|
|
unsafe.Pointer(&on), C.socklen_t(unsafe.Sizeof(on)))
|
2018-07-22 11:48:46 +03:00
|
|
|
|
|
|
|
return 1
|
|
|
|
}
|
|
|
|
|
2018-08-07 07:49:11 +03:00
|
|
|
// int RTMP_Connect1(RTMP* r, RTMPPacket* cp);
|
|
|
|
// rtmp.c +978
|
2018-08-18 07:21:07 +03:00
|
|
|
func C_RTMP_Connect1(r *C_RTMP, cp *C_RTMPPacket) int {
|
2018-07-22 16:31:50 +03:00
|
|
|
if debugMode {
|
|
|
|
log.Println("... connected, handshaking...")
|
|
|
|
}
|
2018-07-28 09:07:50 +03:00
|
|
|
//if C.HandShake(r, 1) == 0 {
|
2018-08-07 05:54:25 +03:00
|
|
|
if C_HandShake(r, 1) == 0 {
|
2018-08-07 05:42:11 +03:00
|
|
|
log.Println("C_RTMP_Connect1: handshake failed!")
|
2018-07-22 16:31:50 +03:00
|
|
|
return 0
|
|
|
|
}
|
|
|
|
if debugMode {
|
|
|
|
log.Println("... handshaked...")
|
|
|
|
}
|
2018-07-28 10:44:02 +03:00
|
|
|
//if C.SendConnectPacket(r, cp) == 0 {
|
2018-08-07 05:58:06 +03:00
|
|
|
if C_SendConnectPacket(r, cp) == 0 {
|
2018-07-22 16:31:50 +03:00
|
|
|
log.Println("RTMP connect failed!")
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
return 1
|
|
|
|
}
|
|
|
|
|
2018-08-24 04:13:34 +03:00
|
|
|
// int RTMP_Connect(RTMP *r, RTMPPacket* cp);
|
|
|
|
// rtmp.c +1032
|
|
|
|
func C_RTMP_Connect(r *C_RTMP, cp *C_RTMPPacket) int {
|
|
|
|
// TODO: port this
|
|
|
|
var service C.sockaddr_in
|
2018-07-22 17:44:23 +03:00
|
|
|
|
2018-08-24 04:13:34 +03:00
|
|
|
if r.Link.hostname.av_len == 0 {
|
2018-07-23 02:08:22 +03:00
|
|
|
return 0
|
|
|
|
}
|
2018-08-24 04:13:34 +03:00
|
|
|
memset((*byte)(unsafe.Pointer(&service)), 0, int(unsafe.Sizeof(service)))
|
2018-07-22 17:44:23 +03:00
|
|
|
|
2018-08-24 04:13:34 +03:00
|
|
|
// TODO: port this
|
|
|
|
service.sin_family = C.AF_INET
|
2018-07-23 02:08:22 +03:00
|
|
|
|
2018-08-24 04:13:34 +03:00
|
|
|
if r.Link.socksport != 0 {
|
2018-08-30 14:01:19 +03:00
|
|
|
if C_add_addr_info(&service, (*C_AVal)(unsafe.Pointer(&r.Link.sockshost)), C.int(r.Link.socksport)) == 0 {
|
2018-08-24 04:13:34 +03:00
|
|
|
return 0
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// connect directly
|
2018-08-30 14:01:19 +03:00
|
|
|
if C_add_addr_info(&service, (*C_AVal)(unsafe.Pointer(&r.Link.hostname)),
|
2018-08-24 04:13:34 +03:00
|
|
|
C.int(r.Link.port)) == 0 {
|
|
|
|
return 0
|
|
|
|
}
|
2018-07-23 02:08:22 +03:00
|
|
|
}
|
2018-08-24 04:13:34 +03:00
|
|
|
//if C.RTMP_Connect0(r, (*C.sockaddr)(unsafe.Pointer(&service))) == 0 {
|
|
|
|
if C_RTMP_Connect0(r, (*C.sockaddr)(unsafe.Pointer(&service))) == 0 {
|
2018-07-23 02:08:22 +03:00
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2018-08-24 04:13:34 +03:00
|
|
|
r.m_bSendCounter = 1
|
2018-07-23 02:08:22 +03:00
|
|
|
|
2018-08-24 04:13:34 +03:00
|
|
|
return int(C_RTMP_Connect1(r, cp))
|
|
|
|
//return int(C.RTMP_Connect1(r, cp))
|
|
|
|
}
|
2018-07-23 02:08:22 +03:00
|
|
|
|
2018-08-24 04:13:34 +03:00
|
|
|
// int SocksNegotiate(RTMP* r);
|
|
|
|
// rtmp.c +1062
|
|
|
|
func C_SocksNegotiate(r *C_RTMP) int {
|
|
|
|
var addr int32
|
|
|
|
var service C.sockaddr_in
|
2018-07-23 02:08:22 +03:00
|
|
|
|
2018-08-24 04:13:34 +03:00
|
|
|
memset((*byte)(unsafe.Pointer(&service)), 0, int(unsafe.Sizeof(service)))
|
2018-07-23 02:08:22 +03:00
|
|
|
|
2018-08-30 14:01:19 +03:00
|
|
|
C_add_addr_info(&service, (*C_AVal)(unsafe.Pointer(&r.Link.hostname)), C.int(r.Link.port))
|
2018-08-24 04:13:34 +03:00
|
|
|
addr = int32(inet.Htonl(uint32(service.sin_addr.s_addr)))
|
2018-07-22 17:44:23 +03:00
|
|
|
|
2018-08-24 04:13:34 +03:00
|
|
|
{
|
|
|
|
packet := []byte{
|
|
|
|
4, 1,
|
|
|
|
byte((r.Link.port >> 8) & 0xFF),
|
|
|
|
byte((r.Link.port) & 0xFF),
|
|
|
|
byte(addr>>24) & 0xFF, byte(addr>>16) & 0xFF,
|
|
|
|
byte(addr>>8) & 0xFF, byte(addr) & 0xFF,
|
|
|
|
0,
|
|
|
|
}
|
|
|
|
|
|
|
|
C_WriteN(r, unsafe.Pointer(&packet[0]), int(unsafe.Sizeof(packet)))
|
|
|
|
|
|
|
|
if C_ReadN(r, &packet[0], 8) != 8 {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
|
|
|
if packet[0] == 0 && packet[1] == 90 {
|
|
|
|
return 1
|
|
|
|
} else {
|
|
|
|
// TODO: use new logger here
|
|
|
|
log.Println("C_SocksNegotitate: SOCKS returned error code!")
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
}
|
2018-08-11 05:51:22 +03:00
|
|
|
}
|
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// int RTMP_ConnectStream(RTMP* r, int seekTime);
|
|
|
|
// rtmp.c +1099
|
|
|
|
func C_RTMP_ConnectStream(r *C_RTMP, seekTime int32) int {
|
|
|
|
var packet C_RTMPPacket
|
|
|
|
memset((*byte)(unsafe.Pointer(&packet)), 0, int(unsafe.Sizeof(packet)))
|
|
|
|
|
|
|
|
if seekTime > 0 {
|
|
|
|
r.Link.seekTime = int32(seekTime)
|
|
|
|
}
|
|
|
|
|
|
|
|
r.m_mediaChannel = 0
|
|
|
|
|
|
|
|
// TODO: read packet
|
|
|
|
for r.m_bPlaying == 0 && C_RTMP_IsConnected(r) != 0 &&
|
|
|
|
//C.RTMP_ReadPacket(r, &packet) != 0 {
|
|
|
|
C_RTMP_ReadPacket(r, &packet) != 0 {
|
|
|
|
|
|
|
|
// TODO: port is ready
|
|
|
|
if C_RTMPPacket_IsReady(&packet) != 0 {
|
|
|
|
if packet.m_nBodySize == 0 {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
if packet.m_packetType == RTMP_PACKET_TYPE_AUDIO ||
|
|
|
|
packet.m_packetType == RTMP_PACKET_TYPE_VIDEO ||
|
|
|
|
packet.m_packetType == RTMP_PACKET_TYPE_INFO {
|
|
|
|
log.Println("C_RTMP_ConnectStream: got packet before play()! Ignoring.")
|
|
|
|
C_RTMPPacket_Free(&packet)
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
//C.RTMP_ClientPacket(r, &packet)
|
|
|
|
C_RTMP_ClientPacket(r, &packet)
|
|
|
|
C_RTMPPacket_Free(&packet)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return int(r.m_bPlaying)
|
|
|
|
}
|
|
|
|
|
|
|
|
// int RTMP_ClientPacket()
|
|
|
|
// rtmp.c +1226
|
|
|
|
// NOTE cases have been commented out that are not currently used by AusOcean
|
|
|
|
func C_RTMP_ClientPacket(r *C_RTMP, packet *C_RTMPPacket) int32 {
|
|
|
|
var bHasMediaPacket int32 = 0
|
|
|
|
switch packet.m_packetType {
|
|
|
|
|
|
|
|
case RTMP_PACKET_TYPE_CHUNK_SIZE:
|
|
|
|
// TODO: port this
|
|
|
|
//C.HandleChangeChunkSize(r, packet)
|
|
|
|
C_HandleChangeChunkSize(r, packet)
|
2018-08-25 16:52:22 +03:00
|
|
|
|
|
|
|
case RTMP_PACKET_TYPE_BYTES_READ_REPORT:
|
|
|
|
// TODO: usue new logger here
|
|
|
|
//RTMP_Log(RTMP_LOGDEBUG, "%s, received: bytes read report", __FUNCTION__);
|
|
|
|
|
|
|
|
case RTMP_PACKET_TYPE_CONTROL:
|
|
|
|
panic("Unsupported packet type RTMP_PACKET_TYPE_CONTROL")
|
2018-08-24 12:20:04 +03:00
|
|
|
/*
|
2018-08-25 16:52:22 +03:00
|
|
|
log.Println("RTMP_PACKET_TYPE_CONTROL")
|
|
|
|
// TODO: port this
|
|
|
|
C.HandleCtrl(r, packet)
|
2018-08-24 12:20:04 +03:00
|
|
|
*/
|
|
|
|
case RTMP_PACKET_TYPE_SERVER_BW:
|
|
|
|
// TODO: port this
|
|
|
|
//C.HandleServerBW(r, packet)
|
|
|
|
C_HandlServerBW(r, packet)
|
|
|
|
|
|
|
|
case RTMP_PACKET_TYPE_CLIENT_BW:
|
|
|
|
// TODO: port this
|
|
|
|
//C.HandleClientBW(r, packet)
|
|
|
|
C_HandleClientBW(r, packet)
|
|
|
|
|
2018-08-25 16:52:22 +03:00
|
|
|
case RTMP_PACKET_TYPE_AUDIO:
|
|
|
|
panic("Unsupported packet type RTMP_PACKET_TYPE_AUDIO")
|
2018-08-24 12:20:04 +03:00
|
|
|
|
2018-08-25 16:52:22 +03:00
|
|
|
case RTMP_PACKET_TYPE_VIDEO:
|
|
|
|
panic("Unsupported packet type RTMP_PACKET_TYPE_VIDEO")
|
2018-08-28 12:48:13 +03:00
|
|
|
|
2018-08-25 16:52:22 +03:00
|
|
|
case RTMP_PACKET_TYPE_FLEX_MESSAGE:
|
|
|
|
panic("Unsupported packet type RTMP_PACKET_TYPE_FLEX_MESSAGE")
|
|
|
|
|
|
|
|
case RTMP_PACKET_TYPE_INFO:
|
|
|
|
panic("Unsupported packet type RTMP_PACKET_TYPE_INFO")
|
2018-08-24 12:20:04 +03:00
|
|
|
|
|
|
|
case RTMP_PACKET_TYPE_INVOKE:
|
|
|
|
log.Println("RTMP_PACKET_TYPE_INVOKE:")
|
|
|
|
// TODO use new logger here
|
|
|
|
//RTMP_Log(RTMP_LOGDEBUG, "%s, received: invoke %u bytes", __FUNCTION__,packet.m_nBodySize);
|
|
|
|
|
|
|
|
//if C.HandleInvoke(r, packet.m_body, C.uint(packet.m_nBodySize)) == 1 {
|
|
|
|
if C_HandleInvoke(r, (*byte)(unsafe.Pointer(packet.m_body)), uint32(packet.m_nBodySize)) == 1 {
|
|
|
|
log.Println("HasMediaPacket")
|
|
|
|
bHasMediaPacket = 2
|
|
|
|
}
|
2018-08-25 16:52:22 +03:00
|
|
|
|
|
|
|
case RTMP_PACKET_TYPE_FLASH_VIDEO:
|
|
|
|
panic("Unsupported packet type RTMP_PACKET_TYPE_FLASH_VIDEO")
|
2018-08-24 12:20:04 +03:00
|
|
|
|
2018-08-25 16:52:22 +03:00
|
|
|
default:
|
|
|
|
// TODO use new logger here
|
|
|
|
// RTMP_Log(RTMP_LOGDEBUG, "%s, unknown packet type received: 0x%02x", __FUNCTION__,packet.m_packetType);
|
2018-08-24 12:20:04 +03:00
|
|
|
}
|
|
|
|
return bHasMediaPacket
|
|
|
|
}
|
|
|
|
|
2018-08-07 07:49:11 +03:00
|
|
|
// int ReadN(RTMP* r, char* buffer, int n);
|
|
|
|
// rtmp.c +1390
|
2018-08-18 07:21:07 +03:00
|
|
|
func C_ReadN(r *C_RTMP, buffer *byte, n int) int {
|
2018-07-28 07:57:30 +03:00
|
|
|
nOriginalSize := n
|
|
|
|
var avail int
|
|
|
|
var ptr *byte
|
|
|
|
|
|
|
|
r.m_sb.sb_timedout = 0
|
|
|
|
|
|
|
|
ptr = buffer
|
|
|
|
|
|
|
|
for n > 0 {
|
|
|
|
nBytes := 0
|
|
|
|
var nRead int
|
2018-07-28 08:25:25 +03:00
|
|
|
|
|
|
|
avail = int(r.m_sb.sb_size)
|
|
|
|
if avail == 0 {
|
2018-08-07 07:49:11 +03:00
|
|
|
if C_RTMPSockBuf_Fill(&r.m_sb) < 1 {
|
2018-07-28 09:58:24 +03:00
|
|
|
// if C.RTMPSockBuf_Fill(&r.m_sb) < 1 {
|
2018-07-28 08:25:25 +03:00
|
|
|
if r.m_sb.sb_timedout == 0 {
|
|
|
|
return 0
|
2018-07-28 07:57:30 +03:00
|
|
|
}
|
2018-07-28 08:25:25 +03:00
|
|
|
}
|
2018-07-28 09:17:32 +03:00
|
|
|
avail = int(r.m_sb.sb_size)
|
2018-07-28 08:25:25 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
if n < avail {
|
|
|
|
nRead = n
|
|
|
|
} else {
|
|
|
|
nRead = avail
|
|
|
|
}
|
2018-07-28 07:57:30 +03:00
|
|
|
|
2018-07-28 08:25:25 +03:00
|
|
|
if nRead > 0 {
|
|
|
|
memmove(unsafe.Pointer(ptr), unsafe.Pointer(r.m_sb.sb_start), uintptr(nRead))
|
2018-08-18 07:21:07 +03:00
|
|
|
r.m_sb.sb_start = (*byte)(incBytePtr(unsafe.Pointer(r.m_sb.sb_start),
|
2018-07-28 08:25:25 +03:00
|
|
|
nRead))
|
2018-08-18 07:21:07 +03:00
|
|
|
r.m_sb.sb_size -= int32(nRead)
|
2018-07-28 08:25:25 +03:00
|
|
|
nBytes = nRead
|
2018-08-18 07:21:07 +03:00
|
|
|
r.m_nBytesIn += int32(nRead)
|
2018-07-28 08:25:25 +03:00
|
|
|
if r.m_bSendCounter != 0 && r.m_nBytesIn > (r.m_nBytesInSent+
|
|
|
|
r.m_nClientBW/10) {
|
2018-07-28 10:07:15 +03:00
|
|
|
//if C.SendBytesReceived(r) == 0 {
|
2018-08-07 05:57:28 +03:00
|
|
|
if C_SendBytesReceived(r) == 0 {
|
2018-07-28 08:25:25 +03:00
|
|
|
return 0
|
|
|
|
}
|
2018-07-28 07:57:30 +03:00
|
|
|
}
|
|
|
|
}
|
2018-07-28 08:25:25 +03:00
|
|
|
|
|
|
|
if nBytes == 0 {
|
|
|
|
log.Println("RTMP socket closed by peer")
|
2018-08-25 17:19:54 +03:00
|
|
|
C_RTMP_Close(r)
|
2018-07-28 08:25:25 +03:00
|
|
|
break
|
|
|
|
}
|
|
|
|
|
|
|
|
n -= nBytes
|
|
|
|
ptr = (*byte)(incBytePtr(unsafe.Pointer(ptr), nBytes))
|
2018-07-28 07:57:30 +03:00
|
|
|
}
|
|
|
|
|
2018-07-28 08:25:25 +03:00
|
|
|
return nOriginalSize - n
|
|
|
|
}
|
2018-07-28 07:57:30 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// int WriteN(RTMP* r, const char* buffer, int n);
|
|
|
|
// rtmp.c +1502
|
|
|
|
func C_WriteN(r *C_RTMP, buffer unsafe.Pointer, n int) int {
|
|
|
|
ptr := buffer
|
|
|
|
for n > 0 {
|
|
|
|
var nBytes int
|
2018-07-25 04:26:59 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
nBytes = int(C_RTMPSockBuf_Send(&r.m_sb, (*byte)(ptr), int32(n)))
|
2018-07-25 04:26:59 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
if nBytes < 0 {
|
|
|
|
if debugMode {
|
|
|
|
log.Println("C_WriteN, RTMP send error")
|
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: port this
|
2018-08-25 17:19:54 +03:00
|
|
|
C_RTMP_Close(r)
|
2018-08-24 12:20:04 +03:00
|
|
|
n = 1
|
|
|
|
break
|
|
|
|
}
|
|
|
|
|
|
|
|
if nBytes == 0 {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
|
|
|
|
n -= nBytes
|
|
|
|
ptr = incBytePtr(ptr, nBytes)
|
|
|
|
}
|
|
|
|
|
|
|
|
if n == 0 {
|
|
|
|
return 1
|
|
|
|
}
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
|
|
|
// int SendConnectPacket(RTMP* r, RTMPPacket* cp);
|
|
|
|
// rtmp.c +1579
|
|
|
|
func C_SendConnectPacket(r *C_RTMP, cp *C_RTMPPacket) int {
|
|
|
|
var packet C_RTMPPacket
|
|
|
|
var pbuf [4096]byte
|
|
|
|
pend := (*byte)(unsafe.Pointer(incBytePtr(unsafe.Pointer(&pbuf[0]),
|
|
|
|
int(unsafe.Sizeof(pbuf)))))
|
|
|
|
var enc *byte
|
|
|
|
|
|
|
|
if cp != nil {
|
|
|
|
return C_RTMP_SendPacket(r, cp, 1)
|
|
|
|
//return int(C.RTMP_SendPacket(r, cp, 1))
|
|
|
|
}
|
|
|
|
|
|
|
|
packet.m_nChannel = 0x03
|
|
|
|
packet.m_headerType = RTMP_PACKET_SIZE_LARGE
|
2018-07-25 04:26:59 +03:00
|
|
|
packet.m_packetType = RTMP_PACKET_TYPE_INVOKE
|
|
|
|
packet.m_nTimeStamp = 0
|
|
|
|
packet.m_nInfoField2 = 0
|
|
|
|
packet.m_hasAbsTimestamp = 0
|
2018-08-18 07:21:07 +03:00
|
|
|
packet.m_body = (*byte)(incBytePtr(unsafe.Pointer(&pbuf[0]),
|
2018-07-25 04:26:59 +03:00
|
|
|
RTMP_MAX_HEADER_SIZE))
|
|
|
|
|
|
|
|
enc = (*byte)(unsafe.Pointer(packet.m_body))
|
2018-07-28 15:00:08 +03:00
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeString((*byte)(unsafe.Pointer(enc)),
|
|
|
|
//(*byte)(unsafe.Pointer(pend)), &av_connect)))
|
2018-08-08 08:37:54 +03:00
|
|
|
enc = C_AMF_EncodeString(enc, pend, &av_connect)
|
2018-07-25 04:26:59 +03:00
|
|
|
r.m_numInvokes += 1
|
2018-08-18 07:21:07 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNumber((*byte)(unsafe.Pointer(enc)),
|
|
|
|
//(*byte)(unsafe.Pointer(pend)), float64(r.m_numInvokes))))
|
2018-08-08 08:37:54 +03:00
|
|
|
enc = C_AMF_EncodeNumber(enc, pend, float64(r.m_numInvokes))
|
2018-07-26 10:36:44 +03:00
|
|
|
|
2018-08-31 04:54:42 +03:00
|
|
|
(*[_Gi]byte)(unsafe.Pointer(enc))[0] = AMF_OBJECT
|
2018-07-28 15:00:08 +03:00
|
|
|
|
|
|
|
enc = (*byte)(unsafe.Pointer(incBytePtr(unsafe.Pointer(enc), 1)))
|
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedString((*byte)(
|
|
|
|
//unsafe.Pointer(enc)), (*byte)(unsafe.Pointer(pend)), &av_app, &r.Link.app)))
|
2018-08-08 08:37:54 +03:00
|
|
|
enc = C_AMF_EncodeNamedString(enc, pend, &av_app, &r.Link.app)
|
2018-07-26 10:36:44 +03:00
|
|
|
if enc == nil {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
if r.Link.protocol&RTMP_FEATURE_WRITE != 0 {
|
2018-08-18 07:21:07 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedString((*byte)(
|
|
|
|
//unsafe.Pointer(enc)), (*byte)(unsafe.Pointer(pend)), &av_type, &av_nonprivate)))
|
2018-08-08 08:37:54 +03:00
|
|
|
enc = C_AMF_EncodeNamedString(enc, pend, &av_type, &av_nonprivate)
|
2018-07-26 10:36:44 +03:00
|
|
|
|
|
|
|
if enc == nil {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if r.Link.flashVer.av_len != 0 {
|
2018-08-18 07:21:07 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedString((*byte)(
|
|
|
|
//unsafe.Pointer(enc)), (*byte)(unsafe.Pointer(pend)), &av_flashVer, &r.Link.flashVer)))
|
2018-08-08 08:37:54 +03:00
|
|
|
enc = C_AMF_EncodeNamedString(enc, pend, &av_flashVer, &r.Link.flashVer)
|
2018-07-26 10:36:44 +03:00
|
|
|
if enc == nil {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if r.Link.swfUrl.av_len != 0 {
|
2018-08-18 07:21:07 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedString((*byte)(
|
|
|
|
// unsafe.Pointer(enc)), (*byte)(unsafe.Pointer(pend)), &av_swfUrl, &r.Link.swfUrl)))
|
2018-08-08 08:37:54 +03:00
|
|
|
enc = C_AMF_EncodeNamedString(enc, pend, &av_swfUrl, &r.Link.swfUrl)
|
2018-07-26 10:36:44 +03:00
|
|
|
if enc == nil {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if r.Link.tcUrl.av_len != 0 {
|
2018-08-18 07:21:07 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedString((*byte)(
|
|
|
|
//unsafe.Pointer(enc)), (*byte)(unsafe.Pointer(pend)), &av_tcUrl, &r.Link.tcUrl)))
|
2018-08-08 08:37:54 +03:00
|
|
|
enc = C_AMF_EncodeNamedString(enc, pend, &av_tcUrl, &r.Link.tcUrl)
|
2018-07-26 10:36:44 +03:00
|
|
|
if enc == nil {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if r.Link.protocol&RTMP_FEATURE_WRITE == 0 {
|
2018-08-18 07:21:07 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedBoolean((*byte)(
|
|
|
|
//unsafe.Pointer(enc)), (*byte)(unsafe.Pointer(pend)), &av_fpad, 0)))
|
2018-08-08 08:37:54 +03:00
|
|
|
enc = C_AMF_EncodeNamedBoolean(enc, pend, &av_fpad, 0)
|
2018-07-26 10:36:44 +03:00
|
|
|
if enc == nil {
|
|
|
|
return 0
|
|
|
|
}
|
2018-08-18 07:21:07 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedNumber((*byte)(
|
|
|
|
//unsafe.Pointer(enc)), (*byte)(unsafe.Pointer(pend)), &av_capabilities, 15.0)))
|
2018-08-08 08:37:54 +03:00
|
|
|
enc = C_AMF_EncodeNamedNumber(enc, pend, &av_capabilities, 15.0)
|
2018-07-26 10:36:44 +03:00
|
|
|
if enc == nil {
|
|
|
|
return 0
|
|
|
|
}
|
2018-08-18 07:21:07 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedNumber((*byte)(
|
|
|
|
// unsafe.Pointer(enc)), (*byte)(unsafe.Pointer(pend)), &av_audioCodecs, r.m_fAudioCodecs)))
|
2018-08-08 08:37:54 +03:00
|
|
|
enc = C_AMF_EncodeNamedNumber(enc, pend, &av_audioCodecs, float64(r.m_fAudioCodecs))
|
2018-07-26 10:36:44 +03:00
|
|
|
if enc == nil {
|
|
|
|
return 0
|
|
|
|
}
|
2018-08-18 07:21:07 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedNumber((*byte)(
|
|
|
|
//unsafe.Pointer(enc)), (*byte)(unsafe.Pointer(pend)), &av_videoCodecs, r.m_fVideoCodecs)))
|
2018-08-08 08:37:54 +03:00
|
|
|
enc = C_AMF_EncodeNamedNumber(enc, pend, &av_videoCodecs, float64(r.m_fVideoCodecs))
|
2018-07-26 10:36:44 +03:00
|
|
|
if enc == nil {
|
|
|
|
return 0
|
|
|
|
}
|
2018-08-18 07:21:07 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedNumber((*byte)(
|
|
|
|
// unsafe.Pointer(enc)), (*byte)(unsafe.Pointer(pend)), &av_videoFunction, 1.0)))
|
2018-08-08 08:37:54 +03:00
|
|
|
enc = C_AMF_EncodeNamedNumber(enc, pend, &av_videoFunction, 1.0)
|
2018-07-26 10:36:44 +03:00
|
|
|
if enc == nil {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
if r.Link.pageUrl.av_len != 0 {
|
2018-08-18 07:21:07 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedString((*byte)(
|
|
|
|
//unsafe.Pointer(enc)), (*byte)(unsafe.Pointer(pend)), &av_pageUrl, &r.Link.pageUrl)))
|
2018-08-08 08:37:54 +03:00
|
|
|
enc = C_AMF_EncodeNamedString(enc, pend, &av_pageUrl, &r.Link.pageUrl)
|
2018-07-26 10:36:44 +03:00
|
|
|
if enc == nil {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if r.m_fEncoding != 0.0 || r.m_bSendEncoding != 0 {
|
2018-08-18 07:21:07 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedNumber((*byte)(
|
|
|
|
//unsafe.Pointer(enc)), (*byte)(unsafe.Pointer(pend)), &av_objectEncoding, r.m_fEncoding)))
|
2018-08-08 08:37:54 +03:00
|
|
|
enc = C_AMF_EncodeNamedNumber(enc, pend, &av_objectEncoding, float64(r.m_fEncoding))
|
2018-07-26 10:36:44 +03:00
|
|
|
if enc == nil {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if int(uintptr(incBytePtr(unsafe.Pointer(enc), 3))) >= int(uintptr(
|
|
|
|
unsafe.Pointer(pend))) {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2018-08-31 04:54:42 +03:00
|
|
|
(*[_Gi]byte)(unsafe.Pointer(enc))[0] = 0
|
2018-07-26 10:36:44 +03:00
|
|
|
enc = (*byte)(incBytePtr(unsafe.Pointer(enc), 1))
|
2018-08-31 04:54:42 +03:00
|
|
|
(*[_Gi]byte)(unsafe.Pointer(enc))[0] = 0
|
2018-07-26 10:36:44 +03:00
|
|
|
enc = (*byte)(incBytePtr(unsafe.Pointer(enc), 1))
|
2018-08-31 04:54:42 +03:00
|
|
|
(*[_Gi]byte)(unsafe.Pointer(enc))[0] = AMF_OBJECT_END
|
2018-07-26 10:36:44 +03:00
|
|
|
enc = (*byte)(incBytePtr(unsafe.Pointer(enc), 1))
|
|
|
|
|
|
|
|
/* add auth string */
|
|
|
|
if r.Link.auth.av_len != 0 {
|
2018-08-18 07:21:07 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeBoolean((*byte)(
|
|
|
|
// unsafe.Pointer(enc)), (*byte)(unsafe.Pointer(pend)), r.Link.lFlags&RTMP_LF_AUTH)))
|
2018-08-08 08:37:54 +03:00
|
|
|
enc = C_AMF_EncodeBoolean(enc, pend, int(r.Link.lFlags&RTMP_LF_AUTH))
|
2018-07-26 10:36:44 +03:00
|
|
|
if enc == nil {
|
|
|
|
return 0
|
|
|
|
}
|
2018-08-18 07:21:07 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeString((*byte)(unsafe.Pointer(enc)),
|
|
|
|
//(*byte)(unsafe.Pointer(pend)), &r.Link.auth)))
|
2018-08-08 08:37:54 +03:00
|
|
|
enc = C_AMF_EncodeString(enc, (*byte)(pend), &r.Link.auth)
|
2018-07-26 10:36:44 +03:00
|
|
|
if enc == nil {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if r.Link.extras.o_num != 0 {
|
|
|
|
for i := 0; i < int(r.Link.extras.o_num); i++ {
|
2018-08-18 07:21:07 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMFProp_Encode((*C_AMFObjectProperty)(
|
2018-07-29 15:59:53 +03:00
|
|
|
//incPtr(unsafe.Pointer(&r.Link.extras.o_props), int(unsafe.Sizeof(
|
2018-08-18 07:21:07 +03:00
|
|
|
//r.Link.extras.o_props)), i)), (*byte)(unsafe.Pointer(enc)), (*byte)(
|
2018-07-29 15:59:53 +03:00
|
|
|
//unsafe.Pointer(pend)))))
|
2018-08-18 07:21:07 +03:00
|
|
|
enc = C_AMF_PropEncode((*C_AMFObjectProperty)(incPtr(unsafe.Pointer(
|
2018-07-29 15:59:53 +03:00
|
|
|
&r.Link.extras.o_props), int(unsafe.Sizeof(r.Link.extras.o_props)), i)),
|
|
|
|
enc, pend)
|
2018-07-26 10:36:44 +03:00
|
|
|
|
|
|
|
if enc == nil {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
packet.m_nBodySize = uint32(uintptr(decBytePtr(unsafe.Pointer(enc),
|
|
|
|
int(uintptr(unsafe.Pointer(packet.m_body))))))
|
2018-07-26 10:36:44 +03:00
|
|
|
|
2018-07-29 03:57:46 +03:00
|
|
|
//return int(C.RTMP_SendPacket(r, &packet, 1))
|
2018-08-07 06:03:40 +03:00
|
|
|
return C_RTMP_SendPacket(r, &packet, 1)
|
2018-07-25 04:26:59 +03:00
|
|
|
}
|
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// int RTMP_SendCreateStream(RTMP* r);
|
|
|
|
// rtmp.c +1725
|
|
|
|
func C_RTMP_SendCreateStream(r *C_RTMP) int32 {
|
2018-08-24 04:13:34 +03:00
|
|
|
var packet C_RTMPPacket
|
|
|
|
var pbuf [256]byte
|
2018-08-24 12:20:04 +03:00
|
|
|
var pend *byte = (*byte)(unsafe.Pointer(uintptr(unsafe.Pointer(&pbuf[0])) +
|
|
|
|
unsafe.Sizeof(pbuf)))
|
|
|
|
var enc *byte
|
2018-08-24 04:13:34 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
packet.m_nChannel = 0x03 /* control channel (invoke) */
|
2018-08-24 04:13:34 +03:00
|
|
|
packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM
|
2018-08-24 12:20:04 +03:00
|
|
|
packet.m_packetType = RTMP_PACKET_TYPE_INVOKE
|
2018-08-24 04:13:34 +03:00
|
|
|
packet.m_nTimeStamp = 0
|
|
|
|
packet.m_nInfoField2 = 0
|
|
|
|
packet.m_hasAbsTimestamp = 0
|
|
|
|
packet.m_body = (*byte)(incBytePtr(unsafe.Pointer(&pbuf[0]),
|
2018-08-24 12:20:04 +03:00
|
|
|
int(RTMP_MAX_HEADER_SIZE)))
|
2018-08-24 04:13:34 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
enc = (*byte)(unsafe.Pointer(packet.m_body))
|
|
|
|
enc = C_AMF_EncodeString(enc, pend, &av_createStream)
|
|
|
|
r.m_numInvokes++
|
|
|
|
enc = C_AMF_EncodeNumber(enc, pend, float64(r.m_numInvokes))
|
|
|
|
*enc = AMF_NULL
|
|
|
|
enc = (*byte)(incBytePtr(unsafe.Pointer(enc), 1))
|
2018-08-24 04:13:34 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
packet.m_nBodySize = uint32(uintptr(unsafe.Pointer(enc)) - uintptr(
|
|
|
|
unsafe.Pointer(packet.m_body)))
|
2018-08-24 04:13:34 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
return int32(C_RTMP_SendPacket(r, &packet, 1))
|
2018-08-24 04:13:34 +03:00
|
|
|
}
|
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// int SendReleaseStream(RTMP* r);
|
|
|
|
// rtmp.c +1816
|
|
|
|
func C_SendReleaseStream(r *C_RTMP) int32 {
|
|
|
|
var packet C_RTMPPacket
|
|
|
|
var pbuf [1024]byte
|
|
|
|
var pend *byte = (*byte)(unsafe.Pointer(uintptr(unsafe.Pointer(&pbuf[0])) +
|
|
|
|
unsafe.Sizeof(pbuf)))
|
|
|
|
var enc *byte
|
2018-08-24 04:13:34 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
packet.m_nChannel = 0x03 /* control channel (invoke) */
|
|
|
|
packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM
|
|
|
|
packet.m_packetType = RTMP_PACKET_TYPE_INVOKE
|
|
|
|
packet.m_nTimeStamp = 0
|
|
|
|
packet.m_nInfoField2 = 0
|
|
|
|
packet.m_hasAbsTimestamp = 0
|
|
|
|
packet.m_body = (*byte)(incBytePtr(unsafe.Pointer(&pbuf[0]),
|
|
|
|
int(RTMP_MAX_HEADER_SIZE)))
|
2018-08-24 04:13:34 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
enc = (*byte)(unsafe.Pointer(packet.m_body))
|
|
|
|
enc = C_AMF_EncodeString(enc, pend, &av_releaseStream)
|
|
|
|
r.m_numInvokes++
|
|
|
|
enc = C_AMF_EncodeNumber(enc, pend, float64(r.m_numInvokes))
|
|
|
|
*enc = AMF_NULL
|
|
|
|
enc = (*byte)(incBytePtr(unsafe.Pointer(enc), 1))
|
|
|
|
enc = C_AMF_EncodeString(enc, pend, &r.Link.playpath)
|
|
|
|
if enc == nil {
|
2018-08-24 04:13:34 +03:00
|
|
|
return 0
|
|
|
|
}
|
2018-08-24 12:20:04 +03:00
|
|
|
packet.m_nBodySize = uint32(uintptr(unsafe.Pointer(enc)) - uintptr(
|
|
|
|
unsafe.Pointer(packet.m_body)))
|
2018-08-24 04:13:34 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
return int32(C_RTMP_SendPacket(r, &packet, 0))
|
|
|
|
}
|
2018-08-24 04:13:34 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// int SendFCPublish(RTMP* r);
|
|
|
|
// rtmp.c +1846
|
|
|
|
func C_SendFCPublish(r *C_RTMP) int32 {
|
|
|
|
var packet C_RTMPPacket
|
|
|
|
var pbuf [1024]byte
|
|
|
|
var pend *byte = (*byte)(unsafe.Pointer(uintptr(unsafe.Pointer(&pbuf[0])) +
|
|
|
|
unsafe.Sizeof(pbuf)))
|
|
|
|
var enc *byte
|
2018-08-24 04:13:34 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
packet.m_nChannel = 0x03 /* control channel (invoke) */
|
|
|
|
packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM
|
|
|
|
packet.m_packetType = RTMP_PACKET_TYPE_INVOKE
|
|
|
|
packet.m_nTimeStamp = 0
|
|
|
|
packet.m_nInfoField2 = 0
|
|
|
|
packet.m_hasAbsTimestamp = 0
|
|
|
|
packet.m_body = (*byte)(incBytePtr(unsafe.Pointer(&pbuf[0]),
|
|
|
|
int(RTMP_MAX_HEADER_SIZE)))
|
2018-08-24 04:13:34 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
enc = (*byte)(unsafe.Pointer(packet.m_body))
|
|
|
|
enc = C_AMF_EncodeString(enc, pend, &av_FCPublish)
|
|
|
|
r.m_numInvokes++
|
|
|
|
enc = C_AMF_EncodeNumber(enc, pend, float64(r.m_numInvokes))
|
|
|
|
*enc = AMF_NULL
|
|
|
|
enc = (*byte)(incBytePtr(unsafe.Pointer(enc), 1))
|
|
|
|
enc = C_AMF_EncodeString(enc, pend, &r.Link.playpath)
|
|
|
|
if enc == nil {
|
2018-08-24 04:13:34 +03:00
|
|
|
return 0
|
|
|
|
}
|
2018-08-24 12:20:04 +03:00
|
|
|
packet.m_nBodySize = uint32(uintptr(unsafe.Pointer(enc)) - uintptr(
|
|
|
|
unsafe.Pointer(packet.m_body)))
|
2018-08-24 04:13:34 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
return int32(C_RTMP_SendPacket(r, &packet, 0))
|
2018-08-24 04:13:34 +03:00
|
|
|
}
|
|
|
|
|
2018-08-25 17:43:08 +03:00
|
|
|
// int SendFCUnpublish(RTMP *r);
|
|
|
|
// rtmp.c +1875
|
|
|
|
func C_SendFCUnpublish(r *C_RTMP) int32 {
|
2018-08-25 17:48:28 +03:00
|
|
|
// TODO finish porting
|
2018-08-25 17:43:08 +03:00
|
|
|
/*
|
2018-08-28 12:10:10 +03:00
|
|
|
var packet C_RTMPPacket
|
|
|
|
var pbuf [1024]byte
|
|
|
|
var pend []byte = pbuf[1024:]
|
|
|
|
var enc []byte
|
|
|
|
|
|
|
|
packet.m_nChannel = 0x03
|
|
|
|
packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM
|
|
|
|
packet.m_packetType = RTMP_PACKET_TYPE_INVOKE
|
|
|
|
packet.m_nTimeStamp = 0
|
|
|
|
packet.m_nInfoField2 = 0
|
|
|
|
packet.m_hasAbsTimestamp = 0
|
|
|
|
packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE
|
|
|
|
|
|
|
|
// NOTE use of unsafe pointer will be remove here when packet.m_nBody becomes []byte
|
|
|
|
enc = []byte(unsafe.Pointer(packet.m_nBody))
|
|
|
|
enc = C_AMF_EncodeString((*byte)(unsafe.Pointer(&enc[0])),
|
|
|
|
(*byte)(unsafe.Pointer(&pend[0])), &av_FCUnpublish)
|
|
|
|
r.m_numInvokes++
|
|
|
|
enc = C_AMF_EncodeNumber(enc, pend, r->m_numInvokes);
|
|
|
|
*enc++ = AMF_NULL;
|
|
|
|
enc = AMF_EncodeString(enc, pend, &r->Link.playpath);
|
|
|
|
if (!enc)
|
|
|
|
return FALSE;
|
|
|
|
|
|
|
|
packet.m_nBodySize = enc - packet.m_body;
|
|
|
|
|
|
|
|
return RTMP_SendPacket(r, &packet, FALSE);
|
2018-08-25 17:48:28 +03:00
|
|
|
*/
|
2018-08-25 17:43:08 +03:00
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// int SendPublish(RTMP* r);
|
|
|
|
// rtmp.c +1908
|
|
|
|
func C_SendPublish(r *C_RTMP) int32 {
|
|
|
|
var packet C_RTMPPacket
|
|
|
|
var pbuf [1024]byte
|
|
|
|
var pend *byte = (*byte)(unsafe.Pointer(uintptr(unsafe.Pointer(&pbuf[0])) +
|
|
|
|
unsafe.Sizeof(pbuf)))
|
|
|
|
var enc *byte
|
2018-08-24 04:13:34 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
packet.m_nChannel = 0x04 /* source channel (invoke) */
|
|
|
|
packet.m_headerType = RTMP_PACKET_SIZE_LARGE
|
|
|
|
packet.m_packetType = RTMP_PACKET_TYPE_INVOKE
|
|
|
|
packet.m_nTimeStamp = 0
|
|
|
|
packet.m_nInfoField2 = int32(r.m_stream_id)
|
|
|
|
packet.m_hasAbsTimestamp = 0
|
|
|
|
packet.m_body = (*byte)(incBytePtr(unsafe.Pointer(&pbuf[0]),
|
|
|
|
int(RTMP_MAX_HEADER_SIZE)))
|
2018-08-24 04:13:34 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
enc = (*byte)(unsafe.Pointer(packet.m_body))
|
|
|
|
enc = C_AMF_EncodeString(enc, pend, &av_publish)
|
|
|
|
r.m_numInvokes++
|
|
|
|
enc = C_AMF_EncodeNumber(enc, pend, float64(r.m_numInvokes))
|
|
|
|
*enc = AMF_NULL
|
|
|
|
enc = (*byte)(incBytePtr(unsafe.Pointer(enc), 1))
|
|
|
|
enc = C_AMF_EncodeString(enc, pend, &r.Link.playpath)
|
2018-08-24 04:13:34 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
if enc == nil {
|
|
|
|
return 0
|
2018-08-24 04:13:34 +03:00
|
|
|
}
|
2018-07-22 16:31:50 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
enc = C_AMF_EncodeString(enc, pend, &av_live)
|
|
|
|
if enc == nil {
|
|
|
|
return 0
|
2018-07-22 16:31:50 +03:00
|
|
|
}
|
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
packet.m_nBodySize = uint32(uintptr(unsafe.Pointer(enc)) - uintptr(
|
|
|
|
unsafe.Pointer(packet.m_body)))
|
2018-07-22 16:31:50 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
return int32(C_RTMP_SendPacket(r, &packet, 1))
|
2018-07-22 16:31:50 +03:00
|
|
|
}
|
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// int SendBytesReceived(RTMP* r);
|
|
|
|
// rtmp.c +2080
|
|
|
|
func C_SendBytesReceived(r *C_RTMP) int {
|
|
|
|
var packet C_RTMPPacket
|
|
|
|
var pbuf [256]byte
|
|
|
|
pend := (*byte)(incBytePtr(unsafe.Pointer(&pbuf[0]), 256))
|
2018-08-12 12:07:39 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
packet.m_nChannel = 0x02 /* control channel (invoke) */
|
|
|
|
packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM
|
|
|
|
packet.m_packetType = RTMP_PACKET_TYPE_BYTES_READ_REPORT
|
|
|
|
packet.m_nTimeStamp = 0
|
|
|
|
packet.m_nInfoField2 = 0
|
|
|
|
packet.m_hasAbsTimestamp = 0
|
|
|
|
packet.m_body = (*byte)(incBytePtr(unsafe.Pointer(&pbuf[0]),
|
|
|
|
RTMP_MAX_HEADER_SIZE))
|
2018-08-12 12:07:39 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
packet.m_nBodySize = 4
|
2018-08-12 12:07:39 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
C_AMF_EncodeInt32((*byte)(unsafe.Pointer(packet.m_body)), pend, int32(r.m_nBytesIn))
|
|
|
|
// C.AMF_EncodeInt32(packet.m_body, (*byte)(unsafe.Pointer(pend)), r.m_nBytesIn)
|
2018-08-12 13:07:56 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
r.m_nBytesInSent = r.m_nBytesIn
|
2018-08-12 13:07:56 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
//return int(C.RTMP_SendPacket(r, &packet, 0))
|
|
|
|
return C_RTMP_SendPacket(r, &packet, 0)
|
|
|
|
}
|
2018-08-12 13:07:56 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// int SendCheckBW(RTMP* r);
|
|
|
|
// rtmp.c +2105
|
|
|
|
func C_SendCheckBW(r *C_RTMP) int32 {
|
|
|
|
var packet C_RTMPPacket
|
|
|
|
var pbuf [256]byte
|
|
|
|
var pend *byte = (*byte)(unsafe.Pointer(uintptr(unsafe.Pointer(&pbuf[0])) +
|
|
|
|
unsafe.Sizeof(pbuf)))
|
|
|
|
var enc *byte
|
2018-08-12 13:07:56 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
packet.m_nChannel = 0x03 /* control channel (invoke) */
|
|
|
|
packet.m_headerType = RTMP_PACKET_SIZE_LARGE
|
|
|
|
packet.m_packetType = RTMP_PACKET_TYPE_INVOKE
|
|
|
|
packet.m_nTimeStamp = 0
|
|
|
|
packet.m_nInfoField2 = 0
|
|
|
|
packet.m_hasAbsTimestamp = 0
|
|
|
|
packet.m_body = (*byte)(incBytePtr(unsafe.Pointer(&pbuf[0]),
|
|
|
|
int(RTMP_MAX_HEADER_SIZE)))
|
2018-08-12 12:07:39 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
enc = (*byte)(unsafe.Pointer(packet.m_body))
|
|
|
|
enc = C_AMF_EncodeString(enc, pend, &av__checkbw)
|
|
|
|
r.m_numInvokes++
|
|
|
|
enc = C_AMF_EncodeNumber(enc, pend, float64(r.m_numInvokes))
|
|
|
|
*enc = AMF_NULL
|
|
|
|
enc = (*byte)(incBytePtr(unsafe.Pointer(enc), 1))
|
2018-08-12 12:07:39 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
packet.m_nBodySize = uint32(uintptr(unsafe.Pointer(enc)) - uintptr(
|
|
|
|
unsafe.Pointer(packet.m_body)))
|
2018-08-12 13:17:53 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
return int32(C_RTMP_SendPacket(r, &packet, 0))
|
2018-08-12 13:17:53 +03:00
|
|
|
}
|
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// void AV_erase(C_RTMP_METHOD* vals, int* num, int i, int freeit);
|
|
|
|
// rtmp.c +2393
|
|
|
|
func C_AV_erase(vals *C_RTMP_METHOD, num *int32, i, freeit int32) {
|
|
|
|
if freeit != 0 {
|
|
|
|
//C.free(unsafe.Pointer((*(*C_RTMP_METHOD)(incPtr(unsafe.Pointer(vals), int(i),
|
|
|
|
//int(unsafe.Sizeof(*vals))))).name.av_val))
|
2018-08-12 13:17:53 +03:00
|
|
|
}
|
2018-08-24 12:20:04 +03:00
|
|
|
(*num)--
|
|
|
|
for ; i < *num; i++ {
|
|
|
|
*(*C_RTMP_METHOD)(incPtr(unsafe.Pointer(vals), int(i),
|
|
|
|
int(unsafe.Sizeof(*vals)))) = *(*C_RTMP_METHOD)(incPtr(
|
|
|
|
unsafe.Pointer(vals), int(i+1), int(unsafe.Sizeof(*vals))))
|
|
|
|
}
|
|
|
|
(*(*C_RTMP_METHOD)(incPtr(unsafe.Pointer(vals), int(i),
|
|
|
|
int(unsafe.Sizeof(*vals))))).name.av_val = nil
|
|
|
|
(*(*C_RTMP_METHOD)(incPtr(unsafe.Pointer(vals), int(i),
|
|
|
|
int(unsafe.Sizeof(*vals))))).name.av_len = 0
|
|
|
|
(*(*C_RTMP_METHOD)(incPtr(unsafe.Pointer(vals), int(i),
|
|
|
|
int(unsafe.Sizeof(*vals))))).num = 0
|
2018-08-12 13:17:53 +03:00
|
|
|
}
|
|
|
|
|
2018-08-24 16:00:40 +03:00
|
|
|
// void AV_queue(RTMP_METHOD** vals, int* num, C_AVal* av, int txn);
|
|
|
|
// rtmp.c +2414
|
|
|
|
func C_AV_queue(vals **C_RTMP_METHOD, num *int32, av *C_AVal, txn int32) {
|
|
|
|
if (*num & 0x0f) == 0 {
|
|
|
|
//*vals = (*C_RTMP_METHOD)(realloc(unsafe.Pointer(*vals), uint32((*num+16)*
|
|
|
|
//int32(unsafe.Sizeof(*(*vals))))))
|
|
|
|
*vals = (*C_RTMP_METHOD)(C.realloc(unsafe.Pointer(*vals), C.size_t((*num+16)*
|
|
|
|
int32(unsafe.Sizeof(*(*vals))))))
|
|
|
|
}
|
|
|
|
tmp := malloc(uintptr(av.av_len + 1))
|
|
|
|
//tmp := allocate(uintptr(av.av_len + 1))
|
|
|
|
memmove(tmp, unsafe.Pointer(av.av_val), uintptr(av.av_len))
|
2018-09-01 17:00:26 +03:00
|
|
|
(*[_Gi]byte)(tmp)[av.av_len] = '\000'
|
2018-08-24 16:00:40 +03:00
|
|
|
|
|
|
|
(*(*C_RTMP_METHOD)(incPtr(unsafe.Pointer(*vals), int(*num),
|
|
|
|
int(unsafe.Sizeof(*(*vals)))))).num = int32(txn)
|
|
|
|
(*(*C_RTMP_METHOD)(incPtr(unsafe.Pointer(*vals), int(*num),
|
|
|
|
int(unsafe.Sizeof(*(*vals)))))).name.av_len = av.av_len
|
|
|
|
(*(*C_RTMP_METHOD)(incPtr(unsafe.Pointer(*vals), int(*num),
|
|
|
|
int(unsafe.Sizeof(*(*vals)))))).name.av_val = (*byte)(tmp)
|
|
|
|
(*num)++
|
|
|
|
}
|
|
|
|
|
2018-08-12 14:34:14 +03:00
|
|
|
// int HandleInvoke(RTMP* r, const char* body, unsigned int nBodySize);
|
|
|
|
// rtmp.c +2912
|
2018-08-18 07:21:07 +03:00
|
|
|
func C_HandleInvoke(r *C_RTMP, body *byte, nBodySize uint32) int32 {
|
|
|
|
var obj C_AMFObject
|
|
|
|
var method C_AVal
|
2018-08-14 16:00:47 +03:00
|
|
|
var txn float64
|
2018-08-14 06:15:03 +03:00
|
|
|
var ret int32 = 0
|
|
|
|
var nRes int32
|
|
|
|
|
2018-08-14 11:52:37 +03:00
|
|
|
if *body != 0x02 {
|
2018-08-14 06:15:03 +03:00
|
|
|
// TODO use new logger here
|
2018-08-14 06:03:52 +03:00
|
|
|
//RTMP_Log(RTMP_LOGWARNING, "%s, Sanity failed. no string method in invoke packet",
|
2018-08-14 11:52:37 +03:00
|
|
|
//__FUNCTION__);
|
|
|
|
return 0
|
|
|
|
}
|
2018-08-14 20:06:17 +03:00
|
|
|
nRes = C_AMF_Decode(&obj, body, int32(nBodySize), 0)
|
2018-08-18 07:21:07 +03:00
|
|
|
//nRes = int32(C.AMF_Decode(&obj, (*byte)(unsafe.Pointer(body)), int32(nBodySize), 0))
|
2018-08-14 11:52:37 +03:00
|
|
|
if nRes < 0 {
|
2018-08-14 06:15:03 +03:00
|
|
|
// TODO use new logger here
|
|
|
|
//RTMP_Log(RTMP_LOGERROR, "%s, error decoding invoke packet", __FUNCTION__);
|
2018-08-14 11:52:37 +03:00
|
|
|
return 0
|
|
|
|
}
|
2018-08-14 10:06:42 +03:00
|
|
|
|
2018-08-14 20:17:40 +03:00
|
|
|
// NOTE we don't really need this ?? still functions without it
|
|
|
|
//C.AMF_Dump(&obj)
|
2018-08-14 20:39:57 +03:00
|
|
|
//C.AMFProp_GetString(C_AMF_GetProp(&obj, nil, 0), &method)
|
|
|
|
C_AMFProp_GetString(C_AMF_GetProp(&obj, nil, 0), &method)
|
2018-08-14 20:46:46 +03:00
|
|
|
txn = float64(C_AMFProp_GetNumber(C_AMF_GetProp(&obj, nil, 1)))
|
2018-08-14 06:15:03 +03:00
|
|
|
// TODO use new logger here
|
|
|
|
// RTMP_Log(RTMP_LOGDEBUG, "%s, server invoking <%s>", __FUNCTION__, method.av_val);
|
2018-08-14 06:03:52 +03:00
|
|
|
|
2018-08-14 11:52:37 +03:00
|
|
|
switch {
|
|
|
|
case C_AVMATCH(&method, &av__result) != 0:
|
2018-08-14 16:00:47 +03:00
|
|
|
{
|
2018-08-18 07:21:07 +03:00
|
|
|
var methodInvoked C_AVal
|
2018-08-14 16:00:47 +03:00
|
|
|
var i int32
|
|
|
|
for i = 0; i < int32(r.m_numCalls); i++ {
|
2018-08-18 07:21:07 +03:00
|
|
|
if float64((*(*C_RTMP_METHOD)(incPtr(unsafe.Pointer(r.m_methodCalls), int(i),
|
2018-08-14 16:00:47 +03:00
|
|
|
int(unsafe.Sizeof(*r.m_methodCalls))))).num) == txn {
|
2018-08-18 07:21:07 +03:00
|
|
|
methodInvoked = (*(*C_RTMP_METHOD)(incPtr(unsafe.Pointer(r.m_methodCalls),
|
2018-08-14 16:00:47 +03:00
|
|
|
int(i), int(unsafe.Sizeof(*r.m_methodCalls))))).name
|
2018-08-18 07:21:07 +03:00
|
|
|
//C.AV_erase(r.m_methodCalls, &r.m_numCalls, int32(i), 0)
|
2018-08-14 21:04:18 +03:00
|
|
|
C_AV_erase(r.m_methodCalls, (*int32)(&r.m_numCalls), int32(i), 0)
|
2018-08-14 16:00:47 +03:00
|
|
|
break
|
2018-08-14 06:03:52 +03:00
|
|
|
}
|
2018-08-14 11:52:37 +03:00
|
|
|
}
|
2018-08-14 16:00:47 +03:00
|
|
|
if methodInvoked.av_val == nil {
|
|
|
|
// TODO use new logger here
|
|
|
|
//RTMP_Log(RTMP_LOGDEBUG, "%s, received result id %f without matching request",
|
|
|
|
//__FUNCTION__, txn);
|
|
|
|
goto leave
|
2018-08-14 11:52:37 +03:00
|
|
|
}
|
2018-08-14 16:00:47 +03:00
|
|
|
// TODO use new logger here
|
|
|
|
//RTMP_Log(RTMP_LOGDEBUG, "%s, received result for method call <%s>", __FUNCTION__,
|
|
|
|
//methodInvoked.av_val);
|
|
|
|
switch {
|
|
|
|
case C_AVMATCH(&methodInvoked, &av_connect) != 0:
|
2018-08-28 12:10:10 +03:00
|
|
|
if r.Link.token.av_len != 0 {
|
|
|
|
panic("No support for link token")
|
2018-08-28 12:48:13 +03:00
|
|
|
|
2018-08-28 12:10:10 +03:00
|
|
|
}
|
|
|
|
if (r.Link.protocol & RTMP_FEATURE_WRITE) != 0 {
|
|
|
|
C_SendReleaseStream(r)
|
|
|
|
C_SendFCPublish(r)
|
|
|
|
} else {
|
|
|
|
panic("Link protocol has no RTMP_FEATURE_WRITE")
|
|
|
|
}
|
|
|
|
|
|
|
|
C_RTMP_SendCreateStream(r)
|
|
|
|
if (r.Link.protocol & RTMP_FEATURE_WRITE) == 0 {
|
|
|
|
panic("Link protocol has no RTMP_FEATURE_WRITE")
|
2018-08-14 06:03:52 +03:00
|
|
|
}
|
2018-08-28 12:10:10 +03:00
|
|
|
|
2018-08-14 16:00:47 +03:00
|
|
|
case C_AVMATCH(&methodInvoked, &av_createStream) != 0:
|
2018-08-28 12:10:10 +03:00
|
|
|
r.m_stream_id = int32(C_AMFProp_GetNumber(C_AMF_GetProp(&obj, nil, 3)))
|
2018-08-14 16:00:47 +03:00
|
|
|
|
2018-08-28 12:10:10 +03:00
|
|
|
if (r.Link.protocol & RTMP_FEATURE_WRITE) != 0 {
|
|
|
|
C_SendPublish(r)
|
|
|
|
} else {
|
|
|
|
panic("Link protocol has no RTMP_FEATURE_WRITE")
|
2018-08-14 06:03:52 +03:00
|
|
|
}
|
2018-08-28 12:10:10 +03:00
|
|
|
|
|
|
|
case C_AVMATCH(&methodInvoked, &av_play) != 0 ||
|
|
|
|
C_AVMATCH(&methodInvoked, &av_publish) != 0:
|
|
|
|
panic("Unsupported method av_play/av_publish")
|
2018-08-14 11:52:37 +03:00
|
|
|
}
|
2018-08-18 07:26:10 +03:00
|
|
|
//C.free(unsafe.Pointer(methodInvoked.av_val))
|
2018-08-13 05:45:43 +03:00
|
|
|
}
|
2018-08-14 11:52:37 +03:00
|
|
|
case C_AVMATCH(&method, &av_onBWDone) != 0:
|
2018-08-28 12:10:10 +03:00
|
|
|
if r.m_nBWCheckCounter == 0 {
|
|
|
|
C_SendCheckBW(r)
|
2018-08-14 16:00:47 +03:00
|
|
|
}
|
2018-08-28 12:10:10 +03:00
|
|
|
|
|
|
|
case C_AVMATCH(&method, &av_onFCUnsubscribe) != 0 || C_AVMATCH(&method, &av_onFCSubscribe) != 0:
|
|
|
|
panic("Unsupported method av_onFCUnsubscribe/av_onFCSubscribe")
|
|
|
|
|
2018-08-14 11:52:37 +03:00
|
|
|
case C_AVMATCH(&method, &av_ping) != 0:
|
2018-08-28 12:10:10 +03:00
|
|
|
panic("Unsupported method av_ping")
|
2018-08-28 12:48:13 +03:00
|
|
|
|
2018-08-14 11:52:37 +03:00
|
|
|
case C_AVMATCH(&method, &av__onbwcheck) != 0:
|
2018-08-28 12:10:10 +03:00
|
|
|
panic("Unsupported method av_onbwcheck")
|
2018-08-28 12:48:13 +03:00
|
|
|
|
2018-08-14 16:00:47 +03:00
|
|
|
case C_AVMATCH(&method, &av__onbwdone) != 0:
|
2018-08-28 12:10:10 +03:00
|
|
|
panic("Unsupported method av_onbwdone")
|
|
|
|
|
|
|
|
case C_AVMATCH(&method, &av_close) != 0:
|
|
|
|
panic("Unsupported method av_close")
|
|
|
|
|
|
|
|
case C_AVMATCH(&method, &av_onStatus) != 0:
|
|
|
|
var obj2 C_AMFObject
|
|
|
|
var code, level C_AVal
|
|
|
|
C_AMFProp_GetObject(C_AMF_GetProp(&obj, nil, 3), &obj2)
|
|
|
|
C_AMFProp_GetString(C_AMF_GetProp(&obj2, &av_code, -1), &code)
|
|
|
|
C_AMFProp_GetString(C_AMF_GetProp(&obj2, &av_level, -1), &level)
|
|
|
|
|
|
|
|
// TODO use new logger
|
|
|
|
// RTMP_Log(RTMP_LOGDEBUG, "%s, onStatus: %s", __FUNCTION__, code.av_val);
|
|
|
|
switch {
|
|
|
|
case C_AVMATCH(&code, &av_NetStream_Failed) != 0 ||
|
|
|
|
C_AVMATCH(&code, &av_NetStream_Play_Failed) != 0 ||
|
|
|
|
C_AVMATCH(&code, &av_NetStream_Play_StreamNotFound) != 0 ||
|
|
|
|
C_AVMATCH(&code, &av_NetConnection_Connect_InvalidApp) != 0:
|
|
|
|
panic("Unsupported method av_NetStream/av_NetStream_Play_Failed/av_netSTream_Play_StreamNotFound/av_netConnection_Connect_invalidApp")
|
|
|
|
|
|
|
|
case C_AVMATCH(&code, &av_NetStream_Play_Start) != 0 ||
|
|
|
|
C_AVMATCH(&code, &av_NetStream_Play_PublishNotify) != 0:
|
|
|
|
panic("Unsupported method av_NetStream_Play_Start/av_NetStream_Play_PublishNotify")
|
|
|
|
|
|
|
|
case C_AVMATCH(&code, &av_NetStream_Publish_Start) != 0:
|
|
|
|
|
2018-08-14 11:52:37 +03:00
|
|
|
var i int32
|
2018-08-28 12:10:10 +03:00
|
|
|
r.m_bPlaying = 1
|
2018-08-14 11:52:37 +03:00
|
|
|
for i = 0; i < int32(r.m_numCalls); i++ {
|
2018-08-18 07:21:07 +03:00
|
|
|
if C_AVMATCH(&(*(*C_RTMP_METHOD)(incPtr(unsafe.Pointer(r.m_methodCalls), int(i),
|
2018-08-28 12:10:10 +03:00
|
|
|
int(unsafe.Sizeof(*r.m_methodCalls))))).name, &av_publish) != 0 {
|
|
|
|
//C.AV_erase(r.m_methodCalls, &r.m_numCalls, int32(i), 1)
|
|
|
|
C_AV_erase(r.m_methodCalls, (*int32)(&r.m_numCalls), int32(i), 1)
|
2018-08-14 11:52:37 +03:00
|
|
|
break
|
2018-08-13 05:45:43 +03:00
|
|
|
}
|
2018-08-14 11:52:37 +03:00
|
|
|
}
|
2018-08-28 12:10:10 +03:00
|
|
|
|
|
|
|
case C_AVMATCH(&code, &av_NetStream_Play_Complete) != 0 ||
|
|
|
|
C_AVMATCH(&code, &av_NetStream_Play_Stop) != 0 ||
|
|
|
|
C_AVMATCH(&code, &av_NetStream_Play_UnpublishNotify) != 0:
|
|
|
|
panic("Unsupported method av_NetStream_Play_Complete/av_NetStream_Play_Stop/av_NetStream_Play_UnpublishNotify")
|
|
|
|
|
|
|
|
case C_AVMATCH(&code, &av_NetStream_Seek_Notify) != 0:
|
|
|
|
panic("Unsupported method av_netStream_Seek_Notify")
|
|
|
|
|
|
|
|
case C_AVMATCH(&code, &av_NetStream_Pause_Notify) != 0:
|
|
|
|
panic("Unsupported method av_NetStream_Pause_Notify")
|
2018-08-24 12:20:04 +03:00
|
|
|
}
|
2018-08-28 12:10:10 +03:00
|
|
|
|
|
|
|
case C_AVMATCH(&method, &av_playlist_ready) != 0:
|
|
|
|
panic("Unsupported method av_playlist_ready")
|
|
|
|
|
|
|
|
default:
|
2018-08-14 21:49:18 +03:00
|
|
|
}
|
2018-08-24 12:20:04 +03:00
|
|
|
leave:
|
|
|
|
C_AMF_Reset(&obj)
|
|
|
|
return ret
|
|
|
|
}
|
2018-08-14 21:49:18 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// void HandleChangeChunkSize(RTMP* r, const RTMPPacket* packet);
|
|
|
|
// rtmp.c +3345
|
|
|
|
func C_HandleChangeChunkSize(r *C_RTMP, packet *C_RTMPPacket) {
|
|
|
|
if packet.m_nBodySize >= 4 {
|
|
|
|
//r.m_inChunkSize = int32(C.AMF_DecodeInt32((*byte)(unsafe.Pointer(packet.m_body))))
|
|
|
|
r.m_inChunkSize = int32(C_AMF_DecodeInt32((*byte)(unsafe.Pointer(packet.m_body))))
|
|
|
|
// TODO use new logger here
|
|
|
|
// RTMP_Log(RTMP_LOGDEBUG, "%s, received: chunk size change to %d", __FUNCTION__, r.m_inChunkSize);
|
2018-08-14 21:49:18 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// void HandleServerBW(RTMP* r, const RTMPPacket* packet);
|
|
|
|
// rtmp.c +3508
|
|
|
|
func C_HandlServerBW(r *C_RTMP, packet *C_RTMPPacket) {
|
|
|
|
r.m_nServerBW = int32(C_AMF_DecodeInt32((*byte)(unsafe.Pointer(packet.m_body))))
|
|
|
|
//r.m_nServerBW = int32(C.AMF_DecodeInt32((*byte)(unsafe.Pointer(packet.m_body))))
|
|
|
|
// TODO use new logger here
|
|
|
|
// RTMP_Log(RTMP_LOGDEBUG, "%s: server BW = %d", __FUNCTION__, r.m_nServerBW);
|
2018-08-14 21:53:15 +03:00
|
|
|
}
|
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// void HandleClientBW(RTMP* r, const RTMPPacket* packet);
|
|
|
|
// rtmp.c +3515
|
|
|
|
func C_HandleClientBW(r *C_RTMP, packet *C_RTMPPacket) {
|
|
|
|
r.m_nClientBW = int32(C_AMF_DecodeInt32((*byte)(unsafe.Pointer(packet.m_body))))
|
|
|
|
//r.m_nClientBW = int32(C.AMF_DecodeInt32((*byte)(unsafe.Pointer(packet.m_body))))
|
|
|
|
|
|
|
|
if packet.m_nBodySize > 4 {
|
2018-08-31 04:54:42 +03:00
|
|
|
r.m_nClientBW2 = (uint8)((*[_Gi]byte)(unsafe.Pointer(packet.m_body))[4])
|
2018-08-14 06:15:03 +03:00
|
|
|
} else {
|
2018-08-28 12:20:51 +03:00
|
|
|
r.m_nClientBW2 = 255
|
2018-08-14 06:15:03 +03:00
|
|
|
}
|
2018-08-24 12:20:04 +03:00
|
|
|
// TODO use new logger here
|
|
|
|
// RTMP_Log(RTMP_LOGDEBUG, "%s: client BW = %d %d", __FUNCTION__, r.m_nClientBW,
|
|
|
|
//r.m_nClientBW2);
|
2018-08-14 06:15:03 +03:00
|
|
|
}
|
2018-08-13 05:45:43 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// static int DecodeInt32LE(const char* data);
|
|
|
|
// rtmp.c +3527
|
|
|
|
func C_DecodeInt32LE(data *byte) int32 {
|
|
|
|
var c *uint8 = (*uint8)(data)
|
2018-08-31 04:54:42 +03:00
|
|
|
return int32(((*[_Gi]byte)(unsafe.Pointer(c))[3] << 24) |
|
|
|
|
((*[_Gi]byte)(unsafe.Pointer(c))[2] << 16) |
|
|
|
|
((*[_Gi]byte)(unsafe.Pointer(c))[1] << 8) |
|
2018-08-24 12:20:04 +03:00
|
|
|
*c)
|
2018-08-14 21:04:18 +03:00
|
|
|
}
|
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// int EncodeInt32LE(char* output, int nVal);
|
|
|
|
// rtmp.c +3537
|
|
|
|
func C_EncodeInt32LE(output *byte, nVal int32) int32 {
|
|
|
|
*output = byte(nVal)
|
|
|
|
nVal >>= 8
|
2018-08-31 04:54:42 +03:00
|
|
|
(*[_Gi]byte)(unsafe.Pointer(output))[1] = byte(nVal)
|
2018-08-24 12:20:04 +03:00
|
|
|
nVal >>= 8
|
2018-08-31 04:54:42 +03:00
|
|
|
(*[_Gi]byte)(unsafe.Pointer(output))[2] = byte(nVal)
|
2018-08-24 12:20:04 +03:00
|
|
|
nVal >>= 8
|
2018-08-31 04:54:42 +03:00
|
|
|
(*[_Gi]byte)(unsafe.Pointer(output))[3] = byte(nVal)
|
2018-08-24 12:20:04 +03:00
|
|
|
return 4
|
2018-08-11 08:04:52 +03:00
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// int RTMP_ReadPacket(RTMP* r, RTMPPacket* packet);
|
|
|
|
// rtmp.c +3550
|
2018-08-18 07:21:07 +03:00
|
|
|
func C_RTMP_ReadPacket(r *C_RTMP, packet *C_RTMPPacket) int32 {
|
2018-08-02 15:39:05 +03:00
|
|
|
var hbuf [RTMP_MAX_HEADER_SIZE]uint8
|
2018-08-06 07:30:37 +03:00
|
|
|
memset((*byte)(&hbuf[0]), 0, RTMP_MAX_HEADER_SIZE)
|
2018-08-02 15:39:05 +03:00
|
|
|
var header *byte
|
|
|
|
header = (*byte)(unsafe.Pointer(&hbuf[0]))
|
|
|
|
var nSize, hSize, nToRead, nChunk int32
|
|
|
|
var extendedTimestamp int32
|
|
|
|
|
2018-08-07 05:55:20 +03:00
|
|
|
if C_ReadN(r, (*byte)(&hbuf[0]), 1) == 0 {
|
2018-08-07 06:01:30 +03:00
|
|
|
log.Println("C_RTMP_ReadPacket: failed to read RTMP packet header!")
|
2018-08-02 15:39:05 +03:00
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
packet.m_headerType = uint8((hbuf[0] & 0xc0) >> 6)
|
|
|
|
packet.m_nChannel = int32(hbuf[0] & 0x3f)
|
2018-08-02 15:39:05 +03:00
|
|
|
header = (*byte)(incBytePtr(unsafe.Pointer(header), 1))
|
|
|
|
|
|
|
|
switch {
|
|
|
|
case packet.m_nChannel == 0:
|
2018-08-07 05:55:20 +03:00
|
|
|
if C_ReadN(r, (*byte)(&hbuf[1]), 1) != 1 {
|
2018-08-07 06:01:30 +03:00
|
|
|
log.Println("C_RTMP_ReadPacket: failed to read rtmp packet header 2nd byte.")
|
2018-08-02 15:39:05 +03:00
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
packet.m_nChannel = int32(hbuf[1])
|
2018-08-02 15:39:05 +03:00
|
|
|
packet.m_nChannel += 64
|
|
|
|
header = (*byte)(incBytePtr(unsafe.Pointer(header), 1))
|
|
|
|
case packet.m_nChannel == 1:
|
|
|
|
var tmp int32
|
|
|
|
|
2018-08-07 05:55:20 +03:00
|
|
|
if C_ReadN(r, (*byte)(&hbuf[1]), 2) != 2 {
|
2018-08-07 06:01:30 +03:00
|
|
|
log.Println("C_RTMP_ReadPacket: failed to read RTMP packet 3rd byte")
|
2018-08-02 15:39:05 +03:00
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2018-08-06 07:30:37 +03:00
|
|
|
tmp = int32((hbuf[2] << 8) + hbuf[1])
|
2018-08-18 07:21:07 +03:00
|
|
|
packet.m_nChannel = int32(tmp + 64)
|
2018-08-02 15:39:05 +03:00
|
|
|
header = (*byte)(incBytePtr(unsafe.Pointer(header), 2))
|
|
|
|
}
|
|
|
|
|
2018-08-06 07:30:37 +03:00
|
|
|
nSize = int32(packetSize[packet.m_headerType])
|
2018-08-02 15:39:05 +03:00
|
|
|
|
|
|
|
if packet.m_nChannel >= r.m_channelsAllocatedIn {
|
2018-08-06 07:30:37 +03:00
|
|
|
var n int32 = int32(packet.m_nChannel + 10)
|
2018-08-21 12:12:13 +03:00
|
|
|
//timestamp := (*int32)(realloc(unsafe.Pointer(r.m_channelTimestamp),
|
|
|
|
//uint32(int32(unsafe.Sizeof(n))*n)))
|
|
|
|
timestamp := (*int32)(C.realloc(unsafe.Pointer(r.m_channelTimestamp),
|
|
|
|
C.size_t(int32(unsafe.Sizeof(n))*n)))
|
2018-08-11 07:18:21 +03:00
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
var packetPtr *C_RTMPPacket
|
2018-08-21 12:12:13 +03:00
|
|
|
//packets := (**C_RTMPPacket)(realloc(unsafe.Pointer(r.m_vecChannelsIn),
|
|
|
|
//uint32(int32(unsafe.Sizeof(packetPtr))*n)))
|
|
|
|
packets := (**C_RTMPPacket)(C.realloc(unsafe.Pointer(r.m_vecChannelsIn),
|
|
|
|
C.size_t(int32(unsafe.Sizeof(packetPtr))*n)))
|
2018-08-02 15:39:05 +03:00
|
|
|
if timestamp == nil {
|
2018-08-18 07:26:10 +03:00
|
|
|
//C.free(unsafe.Pointer(r.m_channelTimestamp))
|
2018-08-02 15:39:05 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
if packets == nil {
|
2018-08-18 07:26:10 +03:00
|
|
|
//C.free(unsafe.Pointer(r.m_vecChannelsIn))
|
2018-08-02 15:39:05 +03:00
|
|
|
}
|
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
r.m_channelTimestamp = (*int32)(timestamp)
|
2018-08-02 15:39:05 +03:00
|
|
|
r.m_vecChannelsIn = packets
|
|
|
|
|
|
|
|
if timestamp == nil || packets == nil {
|
2018-08-06 07:30:37 +03:00
|
|
|
r.m_channelsAllocatedIn = 0
|
2018-08-02 15:39:05 +03:00
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2018-08-06 07:30:37 +03:00
|
|
|
memset((*byte)(incPtr(unsafe.Pointer(r.m_channelTimestamp),
|
|
|
|
int(r.m_channelsAllocatedIn), int(unsafe.Sizeof(*r.m_channelTimestamp)))),
|
2018-08-10 09:40:08 +03:00
|
|
|
0, int(4*int32((n-int32(r.m_channelsAllocatedIn)))))
|
2018-08-06 07:30:37 +03:00
|
|
|
memset((*byte)(incPtr(unsafe.Pointer(r.m_vecChannelsIn), int(
|
|
|
|
r.m_channelsAllocatedIn), int(unsafe.Sizeof(*r.m_vecChannelsIn)))), 0,
|
|
|
|
int(int32(unsafe.Sizeof(*packets))*(n-int32(r.m_channelsAllocatedIn))))
|
2018-08-18 07:21:07 +03:00
|
|
|
r.m_channelsAllocatedIn = int32(n)
|
2018-08-02 15:39:05 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
switch {
|
|
|
|
case nSize == RTMP_LARGE_HEADER_SIZE:
|
|
|
|
packet.m_hasAbsTimestamp = 1
|
|
|
|
case nSize < RTMP_LARGE_HEADER_SIZE:
|
2018-08-18 07:21:07 +03:00
|
|
|
var tmpPacketPtr *C_RTMPPacket
|
|
|
|
if *(**C_RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsIn),
|
2018-08-06 07:30:37 +03:00
|
|
|
int(packet.m_nChannel), int(unsafe.Sizeof(tmpPacketPtr)))) != nil {
|
2018-08-18 07:21:07 +03:00
|
|
|
var tmpPacket C_RTMPPacket
|
2018-08-02 15:39:05 +03:00
|
|
|
memmove(unsafe.Pointer(packet), unsafe.Pointer(
|
2018-08-18 07:21:07 +03:00
|
|
|
*(**C_RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsIn),
|
2018-08-06 07:30:37 +03:00
|
|
|
int(packet.m_nChannel), int(unsafe.Sizeof(tmpPacketPtr))))),
|
|
|
|
unsafe.Sizeof(tmpPacket))
|
2018-08-02 15:39:05 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
nSize--
|
|
|
|
|
2018-08-07 05:55:20 +03:00
|
|
|
if nSize > 0 && C_ReadN(r, header, int(nSize)) != int(nSize) {
|
2018-08-07 06:01:30 +03:00
|
|
|
log.Println("C_RTMP_ReadPacket: failed to read rtmp packet header.")
|
2018-08-02 15:39:05 +03:00
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2018-08-06 07:30:37 +03:00
|
|
|
hSize = int32(uintptr(incBytePtr(decBytePtr(unsafe.Pointer(header), int(uintptr(
|
|
|
|
unsafe.Pointer(&hbuf[0])))), int(nSize))))
|
2018-08-02 15:39:05 +03:00
|
|
|
|
|
|
|
if nSize >= 3 {
|
2018-08-18 07:21:07 +03:00
|
|
|
packet.m_nTimeStamp = uint32(C_AMF_DecodeInt24(header))
|
2018-08-02 15:39:05 +03:00
|
|
|
|
|
|
|
if nSize >= 6 {
|
2018-08-18 07:21:07 +03:00
|
|
|
packet.m_nBodySize = uint32(C_AMF_DecodeInt24((*byte)(incBytePtr(
|
2018-08-06 07:30:37 +03:00
|
|
|
unsafe.Pointer(header), 3))))
|
2018-08-02 15:39:05 +03:00
|
|
|
packet.m_nBytesRead = 0
|
|
|
|
|
|
|
|
if nSize > 6 {
|
2018-08-31 04:54:42 +03:00
|
|
|
packet.m_packetType = uint8((*[_Gi]byte)(unsafe.Pointer(header))[6])
|
2018-08-02 15:39:05 +03:00
|
|
|
|
|
|
|
if nSize == 11 {
|
|
|
|
// TODO: port this
|
2018-08-18 07:21:07 +03:00
|
|
|
packet.m_nInfoField2 = int32(C_DecodeInt32LE((*byte)(incBytePtr(
|
2018-08-06 07:30:37 +03:00
|
|
|
unsafe.Pointer(header), 7))))
|
2018-08-18 07:21:07 +03:00
|
|
|
//packet.m_nInfoField2 = int32(C.DecodeInt32LE((*byte)(incBytePtr(
|
2018-08-11 07:24:25 +03:00
|
|
|
//unsafe.Pointer(header), 7))))
|
2018-08-02 15:39:05 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
extendedTimestamp = 0
|
|
|
|
if packet.m_nTimeStamp == 0xffffff {
|
|
|
|
extendedTimestamp = 1
|
|
|
|
}
|
|
|
|
|
|
|
|
if extendedTimestamp != 0 {
|
2018-08-07 05:55:20 +03:00
|
|
|
if C_ReadN(r, (*byte)(incBytePtr(unsafe.Pointer(header), int(nSize))), 4) != 4 {
|
2018-08-02 15:39:05 +03:00
|
|
|
log.Println("RTMPRead_Packet: Failed to read extended timestamp")
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
// TODO: port this
|
2018-08-18 07:21:07 +03:00
|
|
|
packet.m_nTimeStamp = uint32(C_AMF_DecodeInt32((*byte)(incBytePtr(
|
2018-08-06 07:30:37 +03:00
|
|
|
unsafe.Pointer(header), int(nSize)))))
|
2018-08-18 07:21:07 +03:00
|
|
|
//packet.m_nTimeStamp = uint32(C.AMF_DecodeInt32((*byte)(incBytePtr(
|
2018-08-15 14:13:47 +03:00
|
|
|
//unsafe.Pointer(header), int(nSize)))))
|
2018-08-02 15:39:05 +03:00
|
|
|
hSize += 4
|
|
|
|
}
|
|
|
|
|
|
|
|
if packet.m_nBodySize > 0 && packet.m_body == nil {
|
|
|
|
// TODO: port this
|
2018-08-11 07:18:21 +03:00
|
|
|
if C_RTMPPacket_Alloc(packet, uint32(packet.m_nBodySize)) == 0 {
|
2018-08-02 15:39:05 +03:00
|
|
|
log.Println("RTMPRead_Packet: failed to allocate packet")
|
|
|
|
return 0
|
|
|
|
}
|
2018-08-18 07:21:07 +03:00
|
|
|
packet.m_headerType = uint8((hbuf[0] & 0xc0) >> 6)
|
2018-08-06 07:30:37 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
nToRead = int32(packet.m_nBodySize - packet.m_nBytesRead)
|
|
|
|
nChunk = int32(r.m_inChunkSize)
|
|
|
|
|
|
|
|
if nToRead < nChunk {
|
|
|
|
nChunk = nToRead
|
|
|
|
}
|
|
|
|
|
|
|
|
if packet.m_chunk != nil {
|
2018-08-18 07:21:07 +03:00
|
|
|
packet.m_chunk.c_headerSize = int32(hSize)
|
2018-08-06 07:30:37 +03:00
|
|
|
memmove(unsafe.Pointer(&packet.m_chunk.c_header[0]), unsafe.Pointer(&hbuf[0]),
|
|
|
|
uintptr(hSize))
|
2018-08-18 07:21:07 +03:00
|
|
|
packet.m_chunk.c_chunk = (*byte)(incBytePtr(unsafe.Pointer(packet.m_body),
|
2018-08-06 07:30:37 +03:00
|
|
|
int(packet.m_nBytesRead)))
|
2018-08-18 07:21:07 +03:00
|
|
|
packet.m_chunk.c_chunkSize = int32(nChunk)
|
2018-08-06 07:30:37 +03:00
|
|
|
}
|
|
|
|
|
2018-08-07 05:55:20 +03:00
|
|
|
if C_ReadN(r, (*byte)(incBytePtr(unsafe.Pointer(packet.m_body), int(packet.m_nBytesRead))),
|
2018-08-06 07:30:37 +03:00
|
|
|
int(nChunk)) != int(nChunk) {
|
2018-08-07 06:01:30 +03:00
|
|
|
log.Println("C_RTMP_ReadPacket: failed to read RTMP packet body")
|
2018-08-10 09:40:08 +03:00
|
|
|
return 0
|
2018-08-06 07:30:37 +03:00
|
|
|
}
|
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
packet.m_nBytesRead += uint32(nChunk)
|
2018-08-06 07:30:37 +03:00
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
var tmpPktPtr *C_RTMPPacket
|
2018-08-06 07:30:37 +03:00
|
|
|
// keep the packet as ref for other packets on this channel
|
2018-08-18 07:21:07 +03:00
|
|
|
if *(**C_RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsIn),
|
2018-08-06 07:30:37 +03:00
|
|
|
int(packet.m_nChannel), int(unsafe.Sizeof(tmpPktPtr)))) == nil {
|
2018-08-11 07:18:21 +03:00
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
var tmpPkt C_RTMPPacket
|
|
|
|
*(**C_RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsIn), int(packet.m_nChannel),
|
2018-08-18 07:26:10 +03:00
|
|
|
int(unsafe.Sizeof(tmpPktPtr)))) = (*C_RTMPPacket)(malloc(uintptr(
|
2018-08-11 07:18:21 +03:00
|
|
|
unsafe.Sizeof(tmpPkt))))
|
2018-08-02 15:39:05 +03:00
|
|
|
}
|
2018-08-18 07:21:07 +03:00
|
|
|
memmove(unsafe.Pointer(*(**C_RTMPPacket)(incPtr(unsafe.Pointer(
|
2018-08-06 07:30:37 +03:00
|
|
|
r.m_vecChannelsIn), int(packet.m_nChannel), int(unsafe.Sizeof(tmpPktPtr))))),
|
2018-08-10 09:40:08 +03:00
|
|
|
unsafe.Pointer(packet), unsafe.Sizeof(tmpPktPtr))
|
2018-08-06 07:30:37 +03:00
|
|
|
|
|
|
|
if extendedTimestamp != 0 {
|
2018-08-18 07:21:07 +03:00
|
|
|
(*(**C_RTMPPacket)(incPtr(unsafe.Pointer(
|
2018-08-06 07:30:37 +03:00
|
|
|
r.m_vecChannelsIn), int(packet.m_nChannel),
|
|
|
|
int(unsafe.Sizeof(tmpPktPtr))))).m_nTimeStamp = 0xffffff
|
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: port this
|
2018-08-10 09:40:08 +03:00
|
|
|
if C_RTMPPacket_IsReady(packet) != 0 {
|
|
|
|
if packet.m_hasAbsTimestamp == 0 {
|
2018-08-06 07:30:37 +03:00
|
|
|
// timestamps seem to always be relative
|
2018-08-18 07:21:07 +03:00
|
|
|
packet.m_nTimeStamp += *(*uint32)(incPtr(unsafe.Pointer(r.m_channelTimestamp),
|
2018-08-10 09:40:08 +03:00
|
|
|
int(packet.m_nChannel), 4))
|
|
|
|
}
|
2018-08-18 07:21:07 +03:00
|
|
|
*(*uint32)(incPtr(unsafe.Pointer(r.m_channelTimestamp), int(packet.m_nChannel),
|
2018-08-10 09:40:08 +03:00
|
|
|
4)) = packet.m_nTimeStamp
|
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
(*(**C_RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsIn), int(packet.m_nChannel),
|
2018-08-10 09:40:08 +03:00
|
|
|
int(unsafe.Sizeof(*r.m_vecChannelsIn))))).m_body = nil
|
2018-08-18 07:21:07 +03:00
|
|
|
(*(**C_RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsIn), int(packet.m_nChannel),
|
2018-08-10 09:40:08 +03:00
|
|
|
int(unsafe.Sizeof(*r.m_vecChannelsIn))))).m_nBytesRead = 0
|
2018-08-18 07:21:07 +03:00
|
|
|
(*(**C_RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsIn), int(packet.m_nChannel),
|
2018-08-10 09:40:08 +03:00
|
|
|
int(unsafe.Sizeof(*r.m_vecChannelsIn))))).m_hasAbsTimestamp = 0
|
2018-08-06 07:30:37 +03:00
|
|
|
} else {
|
|
|
|
packet.m_body = nil /* so it won't be erased on free */
|
|
|
|
}
|
2018-08-10 09:40:08 +03:00
|
|
|
return 1
|
2018-08-02 15:39:05 +03:00
|
|
|
}
|
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// int HandShake(RTMP* r, int FP9HandShake);
|
|
|
|
// rtmp.c +3744
|
|
|
|
func C_HandShake(r *C_RTMP, FP9HandShake int32) int {
|
|
|
|
var bMatch int
|
|
|
|
//uptime := uint32(0)
|
|
|
|
//suptime := uint32(0)
|
|
|
|
//typ := byte(0)
|
|
|
|
var uptime, suptime uint32
|
|
|
|
var typ byte
|
|
|
|
//clientbuf := make([]byte, RTMP_SIG_SIZE+1)
|
|
|
|
var clientbuf [RTMP_SIG_SIZE + 1]byte
|
|
|
|
clientsig := (*byte)(incBytePtr(unsafe.Pointer(&clientbuf[0]), 1))
|
|
|
|
//serversig := make([]byte, RTMP_SIG_SIZE)
|
|
|
|
var serversig [RTMP_SIG_SIZE]byte
|
2018-07-05 09:55:46 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
clientbuf[0] = 0x03 // not encrypted
|
2018-07-05 09:55:46 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// TODO: port rtmp_getTime
|
|
|
|
//uptime = inet.Htonl(uint32(C.RTMP_GetTime()))
|
|
|
|
uptime = inet.Htonl(uint32(C_RTMP_GetTime()))
|
|
|
|
memmove(unsafe.Pointer(clientsig), unsafe.Pointer(&uptime), 4)
|
2018-07-11 20:20:00 +03:00
|
|
|
|
2018-08-31 05:31:34 +03:00
|
|
|
memset((*byte)(sliceToPtr((*[_Gi]byte)(unsafe.Pointer(clientsig))[4:])), 0, 4)
|
2018-06-27 21:02:16 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
for i := 8; i < RTMP_SIG_SIZE; i++ {
|
2018-08-31 04:54:42 +03:00
|
|
|
(*[_Gi]byte)(unsafe.Pointer(clientsig))[i] = byte(rand.Intn(256))
|
2018-08-24 12:20:04 +03:00
|
|
|
}
|
2018-07-10 13:36:49 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
if C_WriteN(r, unsafe.Pointer(&clientbuf[0]), RTMP_SIG_SIZE+1) == 0 {
|
|
|
|
return 0
|
|
|
|
}
|
2018-07-12 22:51:17 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
//if C.ReadN(r, (*byte)(unsafe.Pointer(&typ)), 1) != 1 {
|
|
|
|
if C_ReadN(r, (*byte)(unsafe.Pointer(&typ)), 1) != 1 {
|
|
|
|
return 0
|
|
|
|
}
|
2018-07-14 15:40:01 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
if debugMode {
|
2018-08-31 04:43:59 +03:00
|
|
|
log.Printf("C_HandShake: Type answer: %v\n", typ)
|
2018-08-24 12:20:04 +03:00
|
|
|
}
|
|
|
|
if typ != clientbuf[0] {
|
2018-08-31 04:43:59 +03:00
|
|
|
log.Printf("C_HandShake: type mismatch: client sent %v, server sent: %v\n",
|
2018-08-24 12:20:04 +03:00
|
|
|
clientbuf[0], typ)
|
|
|
|
}
|
|
|
|
if C_ReadN(r, (*byte)(unsafe.Pointer(&serversig[0])), RTMP_SIG_SIZE) != RTMP_SIG_SIZE {
|
|
|
|
//if C.ReadN(r, (*byte)(unsafe.Pointer(&serversig[0])), RTMP_SIG_SIZE) != RTMP_SIG_SIZE {
|
|
|
|
return 0
|
2018-07-05 09:55:46 +03:00
|
|
|
}
|
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// decode server response
|
|
|
|
memmove(unsafe.Pointer(&suptime), unsafe.Pointer(&serversig[0]), 4)
|
|
|
|
suptime = inet.Ntohl(suptime)
|
2018-08-11 07:18:21 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// 2nd part of handshake
|
|
|
|
if C_WriteN(r, unsafe.Pointer(&serversig[0]), RTMP_SIG_SIZE) == 0 {
|
2018-08-11 07:18:21 +03:00
|
|
|
return 0
|
|
|
|
}
|
2018-08-24 12:20:04 +03:00
|
|
|
|
|
|
|
if C_ReadN(r, (*byte)(unsafe.Pointer(&serversig[0])), RTMP_SIG_SIZE) != RTMP_SIG_SIZE {
|
|
|
|
//if C.ReadN(r, (*byte)(unsafe.Pointer(&serversig[0])), RTMP_SIG_SIZE) != RTMP_SIG_SIZE {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: find golang memcmp
|
|
|
|
bMatch = 0
|
|
|
|
if memcmp(unsafe.Pointer(&serversig[0]), unsafe.Pointer(clientsig),
|
|
|
|
RTMP_SIG_SIZE) == 0 {
|
|
|
|
bMatch = 1
|
|
|
|
}
|
|
|
|
|
|
|
|
if bMatch == 0 {
|
|
|
|
log.Println("Client signature does not match!")
|
|
|
|
}
|
2018-08-11 07:18:21 +03:00
|
|
|
return 1
|
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// int RTMP_SendPacket(RTMP* r, RTMPPacket* packet, int queue);
|
|
|
|
// rtmp.c +3896
|
2018-08-18 07:21:07 +03:00
|
|
|
func C_RTMP_SendPacket(r *C_RTMP, packet *C_RTMPPacket, queue int) int {
|
|
|
|
var prevPacket *C_RTMPPacket
|
2018-07-11 20:20:00 +03:00
|
|
|
last := 0
|
2018-07-29 03:01:19 +03:00
|
|
|
var nSize, hSize, cSize, nChunkSize int
|
2018-07-15 09:51:20 +03:00
|
|
|
var header, hptr, hend, buffer, tbuf, toff unsafe.Pointer
|
2018-07-19 09:08:10 +03:00
|
|
|
var goHbuf [RTMP_MAX_HEADER_SIZE]byte
|
2018-07-15 09:51:20 +03:00
|
|
|
var hbuf = unsafe.Pointer(&goHbuf[0])
|
|
|
|
var c byte
|
2018-08-18 09:18:49 +03:00
|
|
|
var t uint32
|
2018-07-17 11:53:31 +03:00
|
|
|
var packets unsafe.Pointer
|
2018-07-14 09:06:58 +03:00
|
|
|
|
|
|
|
if packet.m_nChannel >= r.m_channelsAllocatedOut {
|
2018-07-18 05:26:32 +03:00
|
|
|
n := int(packet.m_nChannel + 10)
|
2018-08-21 12:12:13 +03:00
|
|
|
packets = C.realloc(unsafe.Pointer(r.m_vecChannelsOut), C.size_t(
|
|
|
|
unsafe.Sizeof(packet)*uintptr(n)))
|
|
|
|
//packets = realloc(unsafe.Pointer(r.m_vecChannelsOut),
|
|
|
|
//uint32(unsafe.Sizeof(packet)*uintptr(n)))
|
2018-07-14 09:06:58 +03:00
|
|
|
|
2018-07-16 08:07:26 +03:00
|
|
|
if uintptr(packets) == uintptr(0) {
|
2018-08-18 07:26:10 +03:00
|
|
|
//C.free(unsafe.Pointer(r.m_vecChannelsOut))
|
2018-07-14 09:06:58 +03:00
|
|
|
r.m_vecChannelsOut = nil
|
|
|
|
r.m_channelsAllocatedOut = 0
|
|
|
|
return 0
|
|
|
|
}
|
2018-08-18 07:21:07 +03:00
|
|
|
r.m_vecChannelsOut = (**C_RTMPPacket)(packets)
|
2018-07-28 15:11:42 +03:00
|
|
|
|
2018-08-18 09:18:49 +03:00
|
|
|
memset((*byte)(incPtr(unsafe.Pointer(r.m_vecChannelsOut), int(r.m_channelsAllocatedOut),
|
|
|
|
int(unsafe.Sizeof(packet)))), 0, int(unsafe.Sizeof(packet)*
|
2018-07-16 08:07:26 +03:00
|
|
|
uintptr(n-int(r.m_channelsAllocatedOut))))
|
2018-07-28 16:02:59 +03:00
|
|
|
|
2018-07-28 15:11:42 +03:00
|
|
|
//memset((*byte)(incPtr(unsafe.Pointer(r.m_vecChannelsOut), int(
|
2018-07-28 16:02:59 +03:00
|
|
|
// r.m_channelsAllocatedOut), int(unsafe.Sizeof(packet)))), 0, int(
|
|
|
|
// unsafe.Sizeof(packet)*uintptr(n-int(r.m_channelsAllocatedOut))))
|
2018-08-18 07:21:07 +03:00
|
|
|
r.m_channelsAllocatedOut = int32(n)
|
2018-07-14 09:06:58 +03:00
|
|
|
}
|
2018-08-18 07:21:07 +03:00
|
|
|
prevPacket = *(**C_RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsOut),
|
2018-07-16 08:07:26 +03:00
|
|
|
int(packet.m_nChannel), int(unsafe.Sizeof(packet))))
|
2018-07-15 09:51:20 +03:00
|
|
|
|
|
|
|
if prevPacket != nil && packet.m_headerType != RTMP_PACKET_SIZE_LARGE {
|
|
|
|
// compress a bit by using the prev packet's attributes
|
|
|
|
if prevPacket.m_nBodySize == packet.m_nBodySize &&
|
|
|
|
prevPacket.m_packetType == packet.m_packetType &&
|
|
|
|
packet.m_headerType == RTMP_PACKET_SIZE_MEDIUM {
|
|
|
|
|
|
|
|
packet.m_headerType = RTMP_PACKET_SIZE_SMALL
|
|
|
|
}
|
|
|
|
|
|
|
|
if prevPacket.m_nTimeStamp == packet.m_nTimeStamp &&
|
|
|
|
packet.m_headerType == RTMP_PACKET_SIZE_SMALL {
|
2018-07-19 09:08:10 +03:00
|
|
|
packet.m_headerType = RTMP_PACKET_SIZE_MINIMUM
|
2018-07-15 09:51:20 +03:00
|
|
|
}
|
|
|
|
|
2018-07-16 08:07:26 +03:00
|
|
|
last = int(prevPacket.m_nTimeStamp)
|
2018-07-15 09:51:20 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
if packet.m_headerType > 3 {
|
|
|
|
log.Printf("Sanity failed! trying to send header of type: 0x%02x.",
|
|
|
|
packet.m_headerType)
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2018-07-17 11:30:24 +03:00
|
|
|
nSize = packetSize[int(packet.m_headerType)]
|
2018-07-15 09:51:20 +03:00
|
|
|
hSize = nSize
|
|
|
|
cSize = 0
|
2018-08-18 09:18:49 +03:00
|
|
|
t = uint32(int(packet.m_nTimeStamp) - last)
|
2018-07-15 09:51:20 +03:00
|
|
|
|
2018-07-16 08:07:26 +03:00
|
|
|
if packet.m_body != nil {
|
2018-07-16 12:20:17 +03:00
|
|
|
header = decBytePtr(unsafe.Pointer(packet.m_body), nSize)
|
|
|
|
hend = unsafe.Pointer(packet.m_body)
|
2018-07-15 09:51:20 +03:00
|
|
|
} else {
|
2018-07-18 05:26:32 +03:00
|
|
|
header = incBytePtr(hbuf, 6)
|
2018-07-15 10:18:38 +03:00
|
|
|
// TODO: be cautious about this sizeof - make sure it works how you think it
|
2018-07-15 09:51:20 +03:00
|
|
|
// does. C code used sizeof(hbuf) where hbuf is a *char
|
2018-07-19 09:08:10 +03:00
|
|
|
hend = incBytePtr(hbuf, RTMP_MAX_HEADER_SIZE)
|
2018-07-15 09:51:20 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
switch {
|
|
|
|
case packet.m_nChannel > 319:
|
|
|
|
cSize = 2
|
|
|
|
case packet.m_nChannel > 63:
|
|
|
|
cSize = 1
|
|
|
|
}
|
|
|
|
|
|
|
|
if cSize != 0 {
|
2018-07-18 05:26:32 +03:00
|
|
|
header = decBytePtr(header, cSize)
|
2018-07-16 12:20:17 +03:00
|
|
|
hSize += cSize
|
|
|
|
}
|
|
|
|
|
|
|
|
if t >= 0xffffff {
|
2018-07-18 05:26:32 +03:00
|
|
|
header = decBytePtr(header, 4)
|
2018-07-16 12:20:17 +03:00
|
|
|
hSize += 4
|
|
|
|
log.Printf("Larger timestamp than 24-bit: 0x%v", t)
|
2018-07-15 09:51:20 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
hptr = header
|
|
|
|
c = byte(packet.m_headerType) << 6
|
|
|
|
|
|
|
|
switch cSize {
|
|
|
|
case 0:
|
|
|
|
c |= byte(packet.m_nChannel)
|
|
|
|
case 1:
|
|
|
|
case 2:
|
|
|
|
c |= byte(1)
|
|
|
|
}
|
2018-07-16 12:20:17 +03:00
|
|
|
*(*byte)(hptr) = c
|
2018-07-18 05:26:32 +03:00
|
|
|
hptr = incBytePtr(hptr, 1)
|
2018-07-16 12:20:17 +03:00
|
|
|
|
|
|
|
if cSize != 0 {
|
|
|
|
tmp := packet.m_nChannel - 64
|
2018-07-18 05:26:32 +03:00
|
|
|
*(*byte)(hptr) = byte(tmp & 0xff)
|
|
|
|
hptr = incBytePtr(hptr, 1)
|
2018-07-16 12:20:17 +03:00
|
|
|
|
|
|
|
if cSize == 2 {
|
|
|
|
*(*byte)(hptr) = byte(tmp >> 8)
|
2018-07-18 05:26:32 +03:00
|
|
|
hptr = incBytePtr(hptr, 1)
|
2018-07-16 12:20:17 +03:00
|
|
|
}
|
|
|
|
}
|
2018-07-15 09:51:20 +03:00
|
|
|
|
|
|
|
if nSize > 1 {
|
|
|
|
res := t
|
|
|
|
if t > 0xffffff {
|
|
|
|
res = 0xffffff
|
|
|
|
}
|
2018-08-18 09:18:49 +03:00
|
|
|
hptr = unsafe.Pointer(C_AMF_EncodeInt24((*byte)(hptr), (*byte)(hend), int32(res)))
|
2018-07-15 09:51:20 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
if nSize > 4 {
|
2018-08-15 10:59:05 +03:00
|
|
|
hptr = unsafe.Pointer(C_AMF_EncodeInt24((*byte)(hptr), (*byte)(hend), (int32(packet.m_nBodySize))))
|
2018-07-18 05:26:32 +03:00
|
|
|
*(*byte)(hptr) = byte(packet.m_packetType)
|
|
|
|
hptr = incBytePtr(hptr, 1)
|
2018-07-15 09:51:20 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
if nSize > 8 {
|
2018-07-29 04:06:55 +03:00
|
|
|
// TODO: port this
|
2018-08-11 07:29:55 +03:00
|
|
|
hptr = incBytePtr(hptr, int(C_EncodeInt32LE((*byte)(hptr),
|
|
|
|
int32(packet.m_nInfoField2))))
|
2018-08-18 07:21:07 +03:00
|
|
|
//hptr = incBytePtr(hptr, int(C.EncodeInt32LE((*byte)(hptr),
|
|
|
|
//int32(packet.m_nInfoField2))))
|
2018-07-15 09:51:20 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
if t >= 0xffffff {
|
2018-08-08 08:37:54 +03:00
|
|
|
hptr = unsafe.Pointer(C_AMF_EncodeInt32((*byte)(hptr), (*byte)(hend), (int32)(t)))
|
2018-07-15 09:51:20 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
nSize = int(packet.m_nBodySize)
|
2018-07-16 12:20:17 +03:00
|
|
|
buffer = unsafe.Pointer(packet.m_body)
|
2018-07-15 09:51:20 +03:00
|
|
|
nChunkSize = int(r.m_outChunkSize)
|
|
|
|
|
|
|
|
if debugMode {
|
2018-08-07 06:03:40 +03:00
|
|
|
log.Printf("C_RTMP_SendPacket: fd=%v, size=%v", r.m_sb.sb_socket, nSize)
|
2018-07-15 09:51:20 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
for (nSize + hSize) != 0 {
|
|
|
|
var wrote int
|
|
|
|
|
|
|
|
if nSize < nChunkSize {
|
|
|
|
nChunkSize = nSize
|
|
|
|
}
|
|
|
|
|
|
|
|
if tbuf != nil {
|
2018-07-16 18:56:52 +03:00
|
|
|
//memmove(toff, header, uintptr(nChunkSize + hSize))
|
2018-09-01 17:00:26 +03:00
|
|
|
copy((*[_Gi]byte)(toff)[:nChunkSize+hSize], (*[_Gi]byte)(header)[:nChunkSize+hSize])
|
2018-07-18 05:26:32 +03:00
|
|
|
toff = incBytePtr(toff, nChunkSize+hSize)
|
2018-07-15 09:51:20 +03:00
|
|
|
} else {
|
|
|
|
// TODO: port this
|
2018-08-07 06:10:43 +03:00
|
|
|
wrote = int(C_WriteN(r, header, nChunkSize+hSize))
|
2018-07-15 09:51:20 +03:00
|
|
|
|
|
|
|
if wrote == 0 {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
nSize -= nChunkSize
|
2018-07-18 05:26:32 +03:00
|
|
|
buffer = incBytePtr(buffer, nChunkSize)
|
2018-07-15 09:51:20 +03:00
|
|
|
hSize = 0
|
|
|
|
|
|
|
|
if nSize > 0 {
|
2018-07-16 12:20:17 +03:00
|
|
|
header = decBytePtr(buffer, 1)
|
2018-07-15 09:51:20 +03:00
|
|
|
hSize = 1
|
|
|
|
|
|
|
|
if cSize != 0 {
|
2018-07-18 05:26:32 +03:00
|
|
|
header = decBytePtr(header, cSize)
|
2018-07-15 09:51:20 +03:00
|
|
|
hSize += cSize
|
|
|
|
}
|
|
|
|
|
|
|
|
if t >= 0xffffff {
|
2018-07-18 05:26:32 +03:00
|
|
|
header = decBytePtr(header, 4)
|
2018-07-15 09:51:20 +03:00
|
|
|
hSize += 4
|
|
|
|
}
|
|
|
|
|
2018-07-15 10:18:38 +03:00
|
|
|
*(*byte)(header) = byte(0xc0 | c)
|
2018-07-15 09:51:20 +03:00
|
|
|
|
|
|
|
if cSize != 0 {
|
2018-07-15 10:18:38 +03:00
|
|
|
tmp := int(packet.m_nChannel) - 64
|
2018-09-01 17:00:26 +03:00
|
|
|
(*[_Gi]byte)(header)[1] = byte(tmp)
|
2018-07-15 09:51:20 +03:00
|
|
|
|
2018-07-15 10:18:38 +03:00
|
|
|
if cSize == 2 {
|
2018-08-31 04:59:41 +03:00
|
|
|
(*[_Gi]byte)(header)[2] = byte(tmp >> 8)
|
2018-07-15 10:18:38 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
if t >= 0xffffff {
|
2018-07-18 05:26:32 +03:00
|
|
|
extendedTimestamp := incBytePtr(header, 1+cSize)
|
2018-08-08 08:37:54 +03:00
|
|
|
C_AMF_EncodeInt32((*byte)(extendedTimestamp),
|
2018-07-19 10:26:41 +03:00
|
|
|
(*byte)(incBytePtr(extendedTimestamp, 4)), (int32)(t))
|
2018-07-15 09:51:20 +03:00
|
|
|
}
|
|
|
|
}
|
2018-07-15 10:18:38 +03:00
|
|
|
}
|
|
|
|
|
2018-07-16 12:20:17 +03:00
|
|
|
if tbuf != nil {
|
2018-08-07 06:10:43 +03:00
|
|
|
wrote := int(C_WriteN(r, tbuf, int(uintptr(decBytePtr(toff,
|
2018-07-16 12:20:17 +03:00
|
|
|
int(uintptr(unsafe.Pointer(tbuf))))))))
|
2018-08-18 07:26:10 +03:00
|
|
|
////C.free(tbuf)
|
2018-07-15 10:18:38 +03:00
|
|
|
tbuf = nil
|
|
|
|
|
|
|
|
if wrote == 0 {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
}
|
2018-07-15 09:51:20 +03:00
|
|
|
|
2018-07-15 10:18:38 +03:00
|
|
|
// We invoked a remote method
|
|
|
|
// TODO: port the const
|
2018-07-19 09:08:10 +03:00
|
|
|
if packet.m_packetType == RTMP_PACKET_TYPE_INVOKE {
|
2018-08-18 07:21:07 +03:00
|
|
|
// TODO: port C_AVal
|
|
|
|
var method C_AVal
|
2018-07-15 10:18:38 +03:00
|
|
|
var ptr unsafe.Pointer
|
2018-07-18 05:26:32 +03:00
|
|
|
ptr = incBytePtr(unsafe.Pointer(packet.m_body), 1)
|
2018-08-18 07:21:07 +03:00
|
|
|
//C.AMF_DecodeString((*byte)(ptr), &method)
|
2018-08-08 08:37:54 +03:00
|
|
|
C_AMF_DecodeString((*byte)(ptr), &method)
|
2018-07-15 10:18:38 +03:00
|
|
|
|
|
|
|
if debugMode {
|
|
|
|
log.Printf("Invoking %v", method.av_val)
|
|
|
|
}
|
|
|
|
// keep it in call queue till result arrives
|
|
|
|
if queue != 0 {
|
|
|
|
var txn int
|
2018-07-18 05:26:32 +03:00
|
|
|
ptr = incBytePtr(ptr, 3+int(method.av_len))
|
2018-08-18 07:21:07 +03:00
|
|
|
//txn = int(C.AMF_DecodeNumber((*byte)(ptr)))
|
2018-08-08 08:37:54 +03:00
|
|
|
txn = int(C_AMF_DecodeNumber((*byte)(ptr)))
|
|
|
|
C_AV_queue(&r.m_methodCalls, (*int32)(unsafe.Pointer(&r.m_numCalls)),
|
2018-07-29 14:43:40 +03:00
|
|
|
&method, int32(txn))
|
2018-08-18 07:21:07 +03:00
|
|
|
//C.AV_queue(&r.m_methodCalls, (*int32)(unsafe.Pointer(&r.m_numCalls)), &method,
|
|
|
|
//int32(txn))
|
2018-07-15 10:18:38 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
if *(**C_RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsOut),
|
2018-07-17 11:30:24 +03:00
|
|
|
int(packet.m_nChannel), int(unsafe.Sizeof(packet)))) == nil {
|
2018-07-16 12:20:17 +03:00
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
*(**C_RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsOut),
|
2018-07-18 05:26:32 +03:00
|
|
|
int(packet.m_nChannel), int(unsafe.Sizeof(packet)))) =
|
2018-08-18 07:26:10 +03:00
|
|
|
(*C_RTMPPacket)(malloc(uintptr(unsafe.Sizeof(*packet))))
|
2018-07-15 09:51:20 +03:00
|
|
|
}
|
2018-07-14 09:06:58 +03:00
|
|
|
|
2018-08-18 07:21:07 +03:00
|
|
|
memmove(unsafe.Pointer(*(**C_RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsOut),
|
2018-07-19 10:29:03 +03:00
|
|
|
int(packet.m_nChannel), int(unsafe.Sizeof(packet))))), unsafe.Pointer(packet),
|
|
|
|
uintptr(unsafe.Sizeof(*packet)))
|
2018-07-15 10:18:38 +03:00
|
|
|
return 1
|
2018-05-29 08:56:36 +03:00
|
|
|
}
|
2018-07-15 09:51:20 +03:00
|
|
|
|
2018-08-25 17:43:08 +03:00
|
|
|
// void RTMP_Close(RTMP *r);
|
|
|
|
// rtmp.c +4168
|
|
|
|
func C_RTMP_Close(r *C_RTMP) {
|
|
|
|
C_CloseInternal(r, 0)
|
|
|
|
}
|
|
|
|
|
|
|
|
// static void CloseInternal(RTMP *r, int reconnect);
|
|
|
|
// rtmp.c +4175
|
|
|
|
func C_CloseInternal(r *C_RTMP, reconnect int32) {
|
|
|
|
// TODO: port SendFCUnpublish
|
|
|
|
// TODO: port SendDeleteStream
|
|
|
|
// TODO: port RTMPSockBuf_Close
|
|
|
|
// TODO: port AV_Clear
|
|
|
|
/*
|
|
|
|
var i int32
|
|
|
|
|
|
|
|
if C_RTMP_IsConnected(r) != 0 {
|
|
|
|
if r.m_stream_id > 0 {
|
|
|
|
i = int32(r.m_stream_id)
|
|
|
|
if r.Link.protocol&RTMP_FEATURE_WRITE != 0 {
|
|
|
|
C.SendFCUnpublish(r)
|
|
|
|
}
|
|
|
|
C.SendDeleteStream(r, float64(i))
|
|
|
|
}
|
|
|
|
C.RTMPSockBuf_Close(&r.m_sb)
|
|
|
|
}
|
|
|
|
|
|
|
|
r.m_stream_id = -1
|
|
|
|
r.m_sb.sb_socket = -1
|
|
|
|
r.m_nBWCheckCounter = 0
|
|
|
|
r.m_nBytesIn = 0
|
|
|
|
r.m_nBytesInSent = 0
|
|
|
|
|
|
|
|
if r.m_read.flags&RTMP_READ_HEADER != 0 {
|
|
|
|
//C.free(unsafe.Pointer(r.m_read.buf))
|
|
|
|
r.m_read.buf = nil
|
|
|
|
}
|
|
|
|
|
|
|
|
r.m_read.dataType = 0
|
|
|
|
r.m_read.flags = 0
|
|
|
|
r.m_read.status = 0
|
|
|
|
r.m_read.nResumeTS = 0
|
|
|
|
r.m_read.nIgnoredFrameCounter = 0
|
|
|
|
r.m_read.nIgnoredFlvFrameCounter = 0
|
|
|
|
|
|
|
|
r.m_write.m_nBytesRead = 0
|
|
|
|
C.RTMPPacket_Free(&r.m_write)
|
|
|
|
|
|
|
|
// NOTE: C frees - not using in our case
|
|
|
|
for i := 0; i < int(r.m_channelsAllocatedIn); i++ {
|
|
|
|
if *(**C_RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsIn), i,
|
|
|
|
int(unsafe.Sizeof(&r.m_write)))) != nil {
|
|
|
|
|
|
|
|
//C.RTMPPacket_Free(*(**C_RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsIn), i,
|
|
|
|
//int(unsafe.Sizeof(&r.m_write)))))
|
|
|
|
|
|
|
|
//C.free(unsafe.Pointer(*(**C_RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsIn),
|
|
|
|
//i, int(unsafe.Sizeof(&r.m_write))))))
|
|
|
|
|
|
|
|
*(**C_RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsIn),
|
|
|
|
i, int(unsafe.Sizeof(&r.m_write)))) = nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
//C.free(unsafe.Pointer(r.m_vecChannelsOut))
|
|
|
|
r.m_vecChannelsOut = nil
|
|
|
|
r.m_channelsAllocatedOut = 0
|
|
|
|
C.AV_clear(r.m_methodCalls, r.m_numCalls)
|
|
|
|
|
|
|
|
r.m_methodCalls = nil
|
|
|
|
r.m_numCalls = 0
|
|
|
|
r.m_numInvokes = 0
|
|
|
|
|
|
|
|
r.m_bPlaying = C.uchar(0)
|
|
|
|
r.m_sb.sb_size = 0
|
|
|
|
|
|
|
|
r.m_msgCounter = 0
|
|
|
|
r.m_resplen = 0
|
|
|
|
r.m_unackd = 0
|
|
|
|
|
|
|
|
if ((r.Link.lFlags & RTMP_LF_FTCU) != 0) && (reconnect == 0) {
|
|
|
|
//C.free(unsafe.Pointer(r.Link.app.av_val))
|
|
|
|
r.Link.app.av_val = nil
|
|
|
|
r.Link.lFlags ^= RTMP_LF_FAPU
|
|
|
|
}
|
|
|
|
|
|
|
|
if reconnect == 0 {
|
|
|
|
//C.free(unsafe.Pointer(r.Link.playpath0.av_val))
|
|
|
|
r.Link.playpath0.av_val = nil
|
|
|
|
}
|
|
|
|
*/
|
|
|
|
}
|
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// int RTMPSockBuf_Fill(RTMPSockBuf* sb);
|
|
|
|
// rtmp.c +4253
|
|
|
|
func C_RTMPSockBuf_Fill(sb *C_RTMPSockBuf) int {
|
|
|
|
var nBytes int
|
2018-07-15 10:48:24 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
if sb.sb_size == 0 {
|
|
|
|
sb.sb_start = &sb.sb_buf[0]
|
|
|
|
}
|
2018-07-15 10:48:24 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
for {
|
|
|
|
nBytes = int(unsafe.Sizeof(sb.sb_buf)) - 1 - int(sb.sb_size) -
|
|
|
|
int(uintptr(unsafe.Pointer(sb.sb_start))-uintptr(unsafe.Pointer(
|
|
|
|
&sb.sb_buf[0])))
|
2018-07-15 10:48:24 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// TODO: figure out what to do with recv
|
|
|
|
nBytes = int(C.recv(C.int(sb.sb_socket), unsafe.Pointer(uintptr(unsafe.Pointer(
|
|
|
|
sb.sb_start))+uintptr(int(sb.sb_size))), C.size_t(nBytes), 0))
|
2018-07-15 10:48:24 +03:00
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
if nBytes != -1 {
|
|
|
|
sb.sb_size += int32(nBytes)
|
|
|
|
} else {
|
|
|
|
log.Println("C_RTMPSockBuf_Fill: recv error!")
|
2018-07-15 10:48:24 +03:00
|
|
|
}
|
2018-08-24 12:20:04 +03:00
|
|
|
break
|
2018-07-17 13:05:25 +03:00
|
|
|
}
|
2018-08-24 12:20:04 +03:00
|
|
|
return nBytes
|
2018-07-15 10:48:24 +03:00
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// int RTMPSockBuf_Send(RTMPSockBuf* sb, const char* buf, int len);
|
|
|
|
// rtmp.c +4297
|
2018-08-18 06:24:40 +03:00
|
|
|
// TODO replace send with golang net connection send
|
2018-08-18 07:21:07 +03:00
|
|
|
func C_RTMPSockBuf_Send(sb *C_RTMPSockBuf, buf *byte, l int32) int32 {
|
|
|
|
return int32(C.send(C.int(sb.sb_socket), unsafe.Pointer(buf), C.size_t(l), 0))
|
2018-07-19 08:53:06 +03:00
|
|
|
}
|
|
|
|
|
2018-08-24 12:20:04 +03:00
|
|
|
// int RTMP_Write(RTMP* r, const char* buf, int size);
|
|
|
|
// rtmp.c +5095
|
|
|
|
func C_RTMP_Write(r *C_RTMP, data []byte) int {
|
|
|
|
buf := sliceToPtr(data)
|
|
|
|
// TODO: port RTMPPacket
|
|
|
|
var pkt = &r.m_write
|
|
|
|
var pend, enc unsafe.Pointer
|
|
|
|
size := len(data)
|
|
|
|
s2 := size
|
|
|
|
var ret, num int
|
|
|
|
|
|
|
|
pkt.m_nChannel = 0x04
|
|
|
|
pkt.m_nInfoField2 = int32(r.m_stream_id)
|
|
|
|
for s2 != 0 {
|
|
|
|
if pkt.m_nBytesRead == 0 {
|
|
|
|
if size < minDataSize {
|
|
|
|
log.Printf("size: %d\n", size)
|
|
|
|
log.Printf("too small \n")
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2018-08-31 04:59:41 +03:00
|
|
|
if (*[_Gi]byte)(buf)[0] == 'F' && (*[_Gi]byte)(buf)[1] == 'L' && (*[_Gi]byte)(buf)[2] == 'V' {
|
2018-08-24 12:20:04 +03:00
|
|
|
buf = unsafe.Pointer(uintptr(buf) + uintptr(13))
|
|
|
|
s2 -= 13
|
|
|
|
}
|
|
|
|
|
2018-08-31 04:59:41 +03:00
|
|
|
pkt.m_packetType = uint8((*[_Gi]byte)(buf)[0])
|
2018-08-24 12:20:04 +03:00
|
|
|
buf = incBytePtr(buf, 1)
|
|
|
|
pkt.m_nBodySize = uint32(C_AMF_DecodeInt24((*byte)(buf)))
|
|
|
|
buf = incBytePtr(buf, 3)
|
|
|
|
pkt.m_nTimeStamp = uint32(C_AMF_DecodeInt24((*byte)(buf)))
|
|
|
|
buf = incBytePtr(buf, 3)
|
2018-08-31 04:59:41 +03:00
|
|
|
pkt.m_nTimeStamp |= uint32((*[_Gi]byte)(buf)[0]) << 24
|
2018-08-24 12:20:04 +03:00
|
|
|
buf = incBytePtr(buf, 4)
|
|
|
|
s2 -= 11
|
|
|
|
|
|
|
|
if ((pkt.m_packetType == RTMP_PACKET_TYPE_AUDIO ||
|
|
|
|
pkt.m_packetType == RTMP_PACKET_TYPE_VIDEO) &&
|
|
|
|
pkt.m_nTimeStamp == 0) || pkt.m_packetType == RTMP_PACKET_TYPE_INFO {
|
|
|
|
|
|
|
|
pkt.m_headerType = RTMP_PACKET_SIZE_LARGE
|
|
|
|
|
|
|
|
if pkt.m_packetType == RTMP_PACKET_TYPE_INFO {
|
|
|
|
pkt.m_nBodySize += 16
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
pkt.m_headerType = RTMP_PACKET_SIZE_MEDIUM
|
|
|
|
}
|
|
|
|
// TODO: Port this
|
|
|
|
if int(C_RTMPPacket_Alloc(pkt, uint32(pkt.m_nBodySize))) == 0 {
|
|
|
|
log.Println("Failed to allocate packet")
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
|
|
|
enc = unsafe.Pointer(pkt.m_body)
|
|
|
|
pend = incBytePtr(enc, int(pkt.m_nBodySize))
|
|
|
|
|
|
|
|
if pkt.m_packetType == RTMP_PACKET_TYPE_INFO {
|
|
|
|
enc = unsafe.Pointer(C_AMF_EncodeString((*byte)(enc), (*byte)(pend), &setDataFrame))
|
|
|
|
pkt.m_nBytesRead = uint32(math.Abs(float64(uintptr(enc) -
|
|
|
|
uintptr(unsafe.Pointer(pkt.m_body)))))
|
|
|
|
}
|
|
|
|
|
|
|
|
} else {
|
|
|
|
enc = incBytePtr(unsafe.Pointer(pkt.m_body), int(pkt.m_nBytesRead))
|
|
|
|
}
|
|
|
|
num = int(pkt.m_nBodySize - pkt.m_nBytesRead)
|
|
|
|
if num > s2 {
|
|
|
|
num = s2
|
|
|
|
}
|
|
|
|
|
|
|
|
memmove(enc, buf, uintptr(num))
|
|
|
|
pkt.m_nBytesRead += uint32(num)
|
|
|
|
s2 -= num
|
|
|
|
buf = incBytePtr(buf, num)
|
|
|
|
if pkt.m_nBytesRead == pkt.m_nBodySize {
|
|
|
|
// TODO: Port this
|
|
|
|
ret = C_RTMP_SendPacket(r, pkt, 0)
|
|
|
|
// TODO: Port this
|
|
|
|
//C.RTMPPacket_Free(pkt)
|
|
|
|
C_RTMPPacket_Free(pkt)
|
|
|
|
pkt.m_nBytesRead = 0
|
|
|
|
if ret == 0 {
|
|
|
|
return -1
|
|
|
|
}
|
|
|
|
buf = incBytePtr(buf, 4)
|
|
|
|
s2 -= 4
|
|
|
|
if s2 < 0 {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return size + s2
|
2018-07-15 12:26:59 +03:00
|
|
|
}
|
2018-07-18 05:32:02 +03:00
|
|
|
|
2018-08-21 12:12:13 +03:00
|
|
|
/*
|
2018-08-18 09:18:49 +03:00
|
|
|
func realloc(ptr unsafe.Pointer, newSize uint32) unsafe.Pointer {
|
|
|
|
return unsafe.Pointer(crt.Xrealloc(crt.TLS(uintptr(unsafe.Pointer(nil))),
|
2018-08-21 11:28:04 +03:00
|
|
|
uintptr(ptr), uint32(newSize)))
|
2018-07-28 04:35:51 +03:00
|
|
|
}
|
2018-08-21 12:12:13 +03:00
|
|
|
*/
|
2018-07-28 04:35:51 +03:00
|
|
|
|
|
|
|
func memmove(to, from unsafe.Pointer, n uintptr) {
|
2018-08-18 09:18:49 +03:00
|
|
|
if to != nil && from != nil && n != 0 {
|
2018-09-01 17:00:26 +03:00
|
|
|
copy((*[_Gi]byte)(to)[:n], (*[_Gi]byte)(from)[:n])
|
2018-08-18 09:18:49 +03:00
|
|
|
}
|
2018-07-28 04:35:51 +03:00
|
|
|
}
|
|
|
|
|
2018-07-24 15:00:50 +03:00
|
|
|
// TODO: write test for this func
|
|
|
|
func memcmp(a, b unsafe.Pointer, size int) int {
|
2018-07-23 02:08:22 +03:00
|
|
|
for i := 0; i < size; i++ {
|
2018-08-31 04:59:41 +03:00
|
|
|
aValue := (*[_Gi]byte)(a)[i]
|
|
|
|
bValue := (*[_Gi]byte)(b)[i]
|
2018-07-23 02:08:22 +03:00
|
|
|
if aValue != bValue {
|
|
|
|
if aValue < bValue {
|
|
|
|
return -1
|
|
|
|
} else {
|
|
|
|
return 1
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2018-09-01 17:05:28 +03:00
|
|
|
func memset(ptr *byte, val byte, num int) {
|
2018-07-21 02:24:16 +03:00
|
|
|
for i := 0; i < num; i++ {
|
2018-09-01 17:05:28 +03:00
|
|
|
(*[_Gi]byte)(unsafe.Pointer(ptr))[i] = val
|
2018-07-21 02:24:16 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-07-20 14:53:57 +03:00
|
|
|
func strLen(str string) int {
|
|
|
|
return len(str)
|
|
|
|
}
|
|
|
|
|
2018-07-20 07:40:08 +03:00
|
|
|
// wrapper for converting byte pointer to unsafe.Pointer
|
2018-07-20 08:44:39 +03:00
|
|
|
func bToUP(b *byte) unsafe.Pointer {
|
2018-07-20 07:40:08 +03:00
|
|
|
return unsafe.Pointer(b)
|
|
|
|
}
|
|
|
|
|
2018-07-20 11:10:37 +03:00
|
|
|
// Creates a new C style string from a go string
|
2018-07-19 18:35:21 +03:00
|
|
|
func goStrToCStr(str string) *byte {
|
|
|
|
l := len(str)
|
|
|
|
slice := make([]byte, l+1)
|
|
|
|
ptr := unsafe.Pointer(&[]byte(str)[0])
|
|
|
|
for i := 0; i < l; i++ {
|
2018-08-31 04:59:41 +03:00
|
|
|
slice[i] = (*[_Gi]byte)(ptr)[i]
|
2018-07-19 18:35:21 +03:00
|
|
|
}
|
2018-08-10 18:43:25 +03:00
|
|
|
slice[l] = '\x00'
|
2018-07-20 08:44:39 +03:00
|
|
|
return &slice[0]
|
|
|
|
}
|
|
|
|
|
2018-08-10 18:43:25 +03:00
|
|
|
// TODO: need a test in rtmp_test.go
|
|
|
|
func cStrToGoStr(cStr *byte) string {
|
2018-09-01 17:00:26 +03:00
|
|
|
return string((*[_Gi]byte)(unsafe.Pointer(cStr))[:strlen(cStr)])
|
2018-08-10 18:43:25 +03:00
|
|
|
}
|
|
|
|
|
2018-07-20 08:44:39 +03:00
|
|
|
// Duplicates a string given as a byte pointer
|
|
|
|
func strdup(str *byte) *byte {
|
|
|
|
length := strlen(str)
|
|
|
|
newMem := make([]byte, length+1)
|
2018-09-01 17:00:26 +03:00
|
|
|
oldMem := (*[_Gi]byte)(unsafe.Pointer(str))[:length+1]
|
2018-07-20 08:44:39 +03:00
|
|
|
copy(newMem, oldMem)
|
|
|
|
return &newMem[0]
|
2018-07-19 18:35:21 +03:00
|
|
|
}
|
|
|
|
|
2018-07-20 08:44:39 +03:00
|
|
|
// Gets the length of the string found at str - length is number of chars
|
|
|
|
// between start and terminating null char. Returns -1 if a null char is not
|
|
|
|
// found before a count of 1000
|
2018-07-19 18:35:21 +03:00
|
|
|
func strlen(str *byte) int32 {
|
2018-07-20 08:44:39 +03:00
|
|
|
for i := 0; i < 1000; i++ {
|
2018-08-31 05:31:34 +03:00
|
|
|
if (*[_Gi]byte)(unsafe.Pointer(str))[i] == '\000' {
|
2018-07-20 08:44:39 +03:00
|
|
|
return int32(i)
|
2018-07-19 18:35:21 +03:00
|
|
|
}
|
|
|
|
}
|
2018-07-20 08:44:39 +03:00
|
|
|
return int32(-1)
|
2018-07-19 18:35:21 +03:00
|
|
|
}
|
|
|
|
|
2018-07-20 08:44:39 +03:00
|
|
|
// Returns the pointer where the first occurance of val is located in a string
|
|
|
|
// which is terminated by a null char. Returns nil if null char is not found
|
|
|
|
// before a count of 10000
|
2018-07-19 18:35:21 +03:00
|
|
|
func strchr(str *byte, val byte) *byte {
|
2018-07-20 08:44:39 +03:00
|
|
|
for i := 0; i < 1000; i++ {
|
2018-08-31 05:31:34 +03:00
|
|
|
if (*[_Gi]byte)(unsafe.Pointer(str))[i] == val {
|
|
|
|
return &(*[_Gi]byte)(unsafe.Pointer(str))[i]
|
2018-07-19 18:35:21 +03:00
|
|
|
}
|
2018-08-31 05:31:34 +03:00
|
|
|
if (*[_Gi]byte)(unsafe.Pointer(str))[i] == '\000' {
|
2018-07-19 18:35:21 +03:00
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2018-08-10 18:43:25 +03:00
|
|
|
// TODO: need a test in rtmp_test.go
|
2018-08-10 16:52:53 +03:00
|
|
|
// Porting: http://www.ai.mit.edu/projects/im/cam8/cam8/working/CAMlib/tcl/compat/strstr.c
|
|
|
|
func strstr(str *byte, substring *byte) *byte {
|
|
|
|
var a, b *byte
|
|
|
|
/* First scan quickly through the two strings looking for a
|
|
|
|
* single-character match. When it's found, then compare the
|
|
|
|
* rest of the substring.
|
|
|
|
*/
|
|
|
|
|
|
|
|
b = substring
|
|
|
|
if *b == 0 {
|
|
|
|
return str
|
|
|
|
}
|
|
|
|
|
|
|
|
for *str != 0 {
|
|
|
|
str = (*byte)(incBytePtr(unsafe.Pointer(str), 1))
|
|
|
|
if *str != *b {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
a = str
|
|
|
|
for {
|
|
|
|
if *b == 0 {
|
|
|
|
return str
|
|
|
|
}
|
|
|
|
tmp1 := a
|
|
|
|
a = (*byte)(incBytePtr(unsafe.Pointer(a), 1))
|
|
|
|
tmp2 := b
|
|
|
|
b = (*byte)(incBytePtr(unsafe.Pointer(b), 1))
|
|
|
|
if *tmp1 != *tmp2 {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
b = substring
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2018-07-20 08:44:39 +03:00
|
|
|
// Creates mem of the size noOfBytes. returns as unsafe pointer
|
2018-08-18 07:26:10 +03:00
|
|
|
func malloc(nOfBytes uintptr) unsafe.Pointer {
|
2018-07-19 10:54:37 +03:00
|
|
|
mem := make([]byte, int(nOfBytes))
|
|
|
|
return unsafe.Pointer(&mem[0])
|
|
|
|
}
|
|
|
|
|
2018-08-18 07:31:38 +03:00
|
|
|
func calloc(val byte, noOfBytes uintptr) unsafe.Pointer {
|
|
|
|
mem := malloc(noOfBytes)
|
2018-09-01 17:05:28 +03:00
|
|
|
memset((*byte)(mem), val, int(noOfBytes))
|
2018-08-18 07:31:38 +03:00
|
|
|
return mem
|
|
|
|
}
|
|
|
|
|
2018-07-16 08:07:26 +03:00
|
|
|
// incBytePtr returns an unsafe.Pointer to a byte that is inc positive positions
|
2018-07-16 06:30:58 +03:00
|
|
|
// from the passed ptr
|
|
|
|
func incBytePtr(ptr unsafe.Pointer, inc int) unsafe.Pointer {
|
2018-08-10 09:40:08 +03:00
|
|
|
return incPtr(ptr, inc, 1)
|
2018-07-16 06:30:58 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
// incPtr attempts to replicate C like pointer arithmatic functionality
|
|
|
|
func incPtr(ptr unsafe.Pointer, inc, typeSize int) unsafe.Pointer {
|
|
|
|
return unsafe.Pointer(uintptr(ptr) + uintptr(inc*typeSize))
|
|
|
|
}
|
|
|
|
|
|
|
|
// incPtr attempts to replicate C like pointer arithmatic functionality
|
|
|
|
func decPtr(ptr unsafe.Pointer, dec, typeSize int) unsafe.Pointer {
|
|
|
|
return unsafe.Pointer(uintptr(ptr) - uintptr(dec*typeSize))
|
|
|
|
}
|
|
|
|
|
2018-07-16 08:07:26 +03:00
|
|
|
// decBytePtr returns an unsafe.Pointer to a byte that is dec negative positions
|
|
|
|
// from ptr
|
|
|
|
func decBytePtr(ptr unsafe.Pointer, dec int) unsafe.Pointer {
|
2018-08-10 09:40:08 +03:00
|
|
|
return decPtr(ptr, dec, 1)
|
2018-07-16 08:07:26 +03:00
|
|
|
}
|
|
|
|
|
2018-07-15 10:20:56 +03:00
|
|
|
// sliceToPtr get's the address of the first data element and returns as unsafe
|
|
|
|
// pointer
|
|
|
|
func sliceToPtr(data []byte) unsafe.Pointer {
|
2018-08-10 10:16:39 +03:00
|
|
|
if len(data) == 0 {
|
|
|
|
return nil
|
|
|
|
}
|
2018-07-15 10:20:56 +03:00
|
|
|
return unsafe.Pointer(&data[0])
|
|
|
|
}
|
|
|
|
|
2018-06-20 07:26:40 +03:00
|
|
|
var rtmpErrs = [...]string{
|
|
|
|
1: "rtmp: not connected",
|
|
|
|
2: "rtmp: write error",
|
|
|
|
3: "rtmp: not started",
|
|
|
|
}
|
|
|
|
|
|
|
|
type Err uint
|
|
|
|
|
|
|
|
func (e Err) Error() string {
|
|
|
|
if 0 <= int(e) && int(e) < len(rtmpErrs) {
|
|
|
|
s := rtmpErrs[e]
|
|
|
|
if s != "" {
|
|
|
|
return s
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return "rtmp: " + strconv.Itoa(int(e))
|
|
|
|
}
|