2018-05-29 08:56:36 +03:00
|
|
|
/*
|
|
|
|
NAME
|
|
|
|
rtmp.go
|
|
|
|
|
|
|
|
DESCRIPTION
|
|
|
|
See Readme.md
|
|
|
|
|
|
|
|
AUTHOR
|
|
|
|
Saxon Nelson-Milton <saxon@ausocean.org>
|
|
|
|
Dan Kortschak <dan@ausocean.org>
|
2018-07-18 06:57:40 +03:00
|
|
|
Jake Lane <jake@ausocean.org>
|
2018-05-29 08:56:36 +03:00
|
|
|
|
|
|
|
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
|
|
|
|
along with revid in gpl.txt. If not, see http://www.gnu.org/licenses.
|
|
|
|
*/
|
|
|
|
|
|
|
|
package rtmp
|
|
|
|
|
|
|
|
/*
|
|
|
|
#cgo CFLAGS: -I/usr/local/include/librtmp
|
2018-07-19 10:13:37 +03:00
|
|
|
#cgo LDFLAGS: -L/usr/local/lib -lrtmp
|
2018-05-29 08:56:36 +03:00
|
|
|
|
|
|
|
#include <stdlib.h>
|
2018-07-12 17:17:11 +03:00
|
|
|
#include <string.h>
|
2018-05-29 08:56:36 +03:00
|
|
|
#include <rtmp.h>
|
2018-07-19 08:53:06 +03:00
|
|
|
#include <sys/socket.h>
|
2018-07-22 11:48:46 +03:00
|
|
|
#include <sys/types.h>
|
2018-07-21 02:24:16 +03:00
|
|
|
#include <netinet/in.h>
|
2018-07-22 11:48:46 +03:00
|
|
|
#include <netinet/tcp.h>
|
2018-05-29 08:56:36 +03:00
|
|
|
|
2018-07-17 13:05:25 +03:00
|
|
|
typedef enum {
|
|
|
|
RTMPT_OPEN=0, RTMPT_SEND, RTMPT_IDLE, RTMPT_CLOSE
|
|
|
|
} RTMPTCmd;
|
|
|
|
|
2018-07-21 02:24:16 +03:00
|
|
|
typedef struct sockaddr_in sockaddr_in;
|
|
|
|
typedef struct sockaddr sockaddr;
|
|
|
|
int add_addr_info(struct sockaddr_in *service, AVal *host, int port);
|
2018-05-30 09:22:33 +03:00
|
|
|
RTMP* start_session(RTMP* rtmp, char* url, uint connect_timeout);
|
2018-05-29 08:56:36 +03:00
|
|
|
int write_frame(RTMP* rtmp, char* data, uint data_length);
|
|
|
|
int end_session(RTMP* rtmp);
|
2018-07-16 06:46:40 +03:00
|
|
|
void AV_queue(RTMP_METHOD **vals, int *num, AVal *av, int txn);
|
2018-08-07 06:10:43 +03:00
|
|
|
int C_WriteN(RTMP *r, const char *buffer, int n);
|
2018-07-16 12:20:17 +03:00
|
|
|
int EncodeInt32LE(char *output, int nVal);
|
2018-07-17 13:05:25 +03:00
|
|
|
int HTTP_Post(RTMP *r, RTMPTCmd cmd, const char *buf, int len);
|
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:19:21 +03:00
|
|
|
"reflect"
|
2018-07-14 08:29:00 +03:00
|
|
|
"strconv"
|
2018-08-10 18:43:25 +03:00
|
|
|
"strings"
|
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-07-17 13:05:25 +03:00
|
|
|
const (
|
|
|
|
RTMPT_OPEN = iota
|
|
|
|
RTMPT_SEND
|
|
|
|
RTMPT_IDLE
|
|
|
|
RTMPT_CLOSE
|
|
|
|
)
|
2018-08-08 10:29:32 +03:00
|
|
|
|
2018-08-10 16:19:43 +03:00
|
|
|
// typedef enum
|
|
|
|
// amf.h +40
|
|
|
|
type C_AMFDataType int32
|
|
|
|
|
2018-07-28 10:44:02 +03:00
|
|
|
const (
|
|
|
|
AMF_NUMBER = iota
|
|
|
|
AMF_BOOLEAN
|
|
|
|
AMF_STRING
|
|
|
|
AMF_OBJECT
|
|
|
|
AMF_MOVIECLIP /* reserved, not used */
|
|
|
|
AMF_NULL
|
|
|
|
AMF_UNDEFINED
|
|
|
|
AMF_REFERENCE
|
|
|
|
AMF_ECMA_ARRAY
|
|
|
|
AMF_OBJECT_END
|
|
|
|
AMF_STRICT_ARRAY
|
|
|
|
AMF_DATE
|
|
|
|
AMF_LONG_STRING
|
|
|
|
AMF_UNSUPPORTED
|
|
|
|
AMF_RECORDSET /* reserved, not used */
|
|
|
|
AMF_XML_DOC
|
|
|
|
AMF_TYPED_OBJECT
|
|
|
|
AMF_AVMPLUS /* switch to AMF3 */
|
2018-08-10 16:19:43 +03:00
|
|
|
AMF_INVALID = 0xff
|
2018-07-28 10:44:02 +03:00
|
|
|
)
|
2018-07-17 13:05:25 +03:00
|
|
|
|
2018-07-13 18:43:17 +03:00
|
|
|
const (
|
2018-07-19 09:03:33 +03:00
|
|
|
RTMP_PACKET_TYPE_CHUNK_SIZE = 0x01
|
|
|
|
RTMP_PACKET_TYPE_BYTES_READ_REPORT = 0x03
|
|
|
|
RTMP_PACKET_TYPE_CONTROL = 0x04
|
|
|
|
RTMP_PACKET_TYPE_SERVER_BW = 0x05
|
|
|
|
RTMP_PACKET_TYPE_CLIENT_BW = 0x06
|
|
|
|
RTMP_PACKET_TYPE_AUDIO = 0x08
|
|
|
|
RTMP_PACKET_TYPE_VIDEO = 0x09
|
|
|
|
RTMP_PACKET_TYPE_FLEX_STREAM_SEND = 0x0F
|
|
|
|
RTMP_PACKET_TYPE_FLEX_SHARED_OBJECT = 0x10
|
|
|
|
RTMP_PACKET_TYPE_FLEX_MESSAGE = 0x11
|
|
|
|
RTMP_PACKET_TYPE_INFO = 0x12
|
|
|
|
RTMP_PACKET_TYPE_SHARED_OBJECT = 0x13
|
|
|
|
RTMP_PACKET_TYPE_INVOKE = 0x14
|
|
|
|
RTMP_PACKET_TYPE_FLASH_VIDEO = 0x16
|
2018-07-18 18:24:36 +03:00
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
2018-08-08 10:36:16 +03:00
|
|
|
RTMP_PACKET_SIZE_LARGE = 0
|
|
|
|
RTMP_PACKET_SIZE_MEDIUM = 1
|
|
|
|
RTMP_PACKET_SIZE_SMALL = 2
|
|
|
|
RTMP_PACKET_SIZE_MINIMUM = 3
|
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
|
|
|
RTMP_READ_HEADER = 0x01
|
|
|
|
RTMP_READ_RESUME = 0x02
|
|
|
|
RTMP_READ_NO_IGNORE = 0x04
|
|
|
|
RTMP_READ_GOTKF = 0x08
|
|
|
|
RTMP_READ_GOTFLVK = 0x10
|
|
|
|
RTMP_READ_SEEKING = 0x20
|
|
|
|
RTMP_READ_COMPLETE = -3
|
|
|
|
RTMP_READ_ERROR = -2
|
|
|
|
RTMP_READ_EOF = -1
|
|
|
|
RTMP_READ_IGNORE = 0
|
2018-07-18 18:24:36 +03:00
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
2018-08-08 10:36:16 +03:00
|
|
|
RTMP_LF_AUTH = 0x0001 /* using auth param */
|
|
|
|
RTMP_LF_LIVE = 0x0002 /* stream is live */
|
|
|
|
RTMP_LF_SWFV = 0x0004 /* do SWF verification */
|
|
|
|
RTMP_LF_PLST = 0x0008 /* send playlist before play */
|
|
|
|
RTMP_LF_BUFX = 0x0010 /* toggle stream on BufferEmpty msg */
|
|
|
|
RTMP_LF_FTCU = 0x0020 /* free tcUrl on close */
|
|
|
|
RTMP_LF_FAPU = 0x0040 /* free app on close */
|
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
|
|
|
RTMP_FEATURE_HTTP = 0x01
|
|
|
|
RTMP_FEATURE_ENC = 0x02
|
|
|
|
RTMP_FEATURE_SSL = 0x04
|
|
|
|
RTMP_FEATURE_MFP = 0x08 /* not yet supported */
|
|
|
|
RTMP_FEATURE_WRITE = 0x10 /* publish, not play */
|
|
|
|
RTMP_FEATURE_HTTP2 = 0x20 /* server-side rtmpt */
|
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
|
|
|
RTMP_PROTOCOL_UNDEFINED = -1
|
|
|
|
RTMP_PROTOCOL_RTMP = 0
|
|
|
|
RTMP_PROTOCOL_RTMPE = RTMP_FEATURE_ENC
|
|
|
|
RTMP_PROTOCOL_RTMPT = RTMP_FEATURE_HTTP
|
|
|
|
RTMP_PROTOCOL_RTMPS = RTMP_FEATURE_SSL
|
|
|
|
RTMP_PROTOCOL_RTMPTE = (RTMP_FEATURE_HTTP | RTMP_FEATURE_ENC)
|
|
|
|
RTMP_PROTOCOL_RTMPTS = (RTMP_FEATURE_HTTP | RTMP_FEATURE_SSL)
|
|
|
|
RTMP_PROTOCOL_RTMFP = RTMP_FEATURE_MFP
|
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
|
|
|
RTMP_DEFAULT_CHUNKSIZE = 128
|
|
|
|
RTMP_BUFFER_CACHE_SIZE = (16 * 1024)
|
|
|
|
RTMP_CHANNELS = 65600
|
|
|
|
RTMP_SWF_HASHLEN = 32
|
|
|
|
RTMP_SIG_SIZE = 1536
|
|
|
|
RTMP_LARGE_HEADER_SIZE = 12
|
|
|
|
RTMP_MAX_HEADER_SIZE = 18
|
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
|
|
|
minDataSize = 11
|
|
|
|
debugMode = false
|
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 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")
|
|
|
|
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")
|
|
|
|
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_Play_Seek_Notify")
|
|
|
|
av_NetStream_Pause_Notify = AVC("NetStream_Pause_Notify")
|
|
|
|
av_playlist_ready = AVC("playlist_ready")
|
|
|
|
av_set_playlist = AVC("set_playlist")
|
2018-07-26 10:36:44 +03:00
|
|
|
)
|
2018-07-16 06:30:58 +03:00
|
|
|
|
2018-07-18 05:26:32 +03:00
|
|
|
var packetSize = [...]int{12, 8, 4, 1}
|
2018-07-16 08:07:26 +03:00
|
|
|
|
2018-07-19 18:35:21 +03:00
|
|
|
var RTMPProtocolStringsLower = [...]string{
|
|
|
|
"rtmp",
|
|
|
|
"rtmpt",
|
|
|
|
"rtmpe",
|
|
|
|
"rtmpte",
|
|
|
|
"rtmps",
|
|
|
|
"rtmpts",
|
|
|
|
"",
|
|
|
|
"",
|
|
|
|
"rtmfp",
|
|
|
|
}
|
|
|
|
|
2018-08-10 15:47:01 +03:00
|
|
|
type Session interface {
|
|
|
|
Open() error
|
|
|
|
Write([]byte) (int, error)
|
|
|
|
Close() error
|
|
|
|
}
|
|
|
|
|
2018-05-29 08:56:36 +03:00
|
|
|
// session provides parameters required for an rtmp communication session.
|
2018-08-10 15:47:01 +03:00
|
|
|
type session struct {
|
2018-07-11 18:20:18 +03:00
|
|
|
rtmp *C.RTMP
|
2018-05-29 08:56:36 +03:00
|
|
|
url string
|
|
|
|
timeout uint
|
|
|
|
}
|
|
|
|
|
2018-08-07 07:24:58 +03:00
|
|
|
// typedef struct RTMP
|
|
|
|
// rtmp.h +237
|
|
|
|
type C_RTMP struct {
|
2018-08-10 10:20:53 +03:00
|
|
|
m_inChunkSize int32
|
|
|
|
m_outChunkSize int32
|
|
|
|
m_nBWCheckCounter int32
|
|
|
|
m_nBytesIn int32
|
|
|
|
m_nBytesInSent int32
|
|
|
|
m_nBufferMS int32
|
|
|
|
m_stream_id int32
|
|
|
|
m_mediaChannel int32
|
2018-07-18 18:10:36 +03:00
|
|
|
m_mediaStamp uint32
|
|
|
|
m_pauseStamp uint32
|
2018-08-10 10:20:53 +03:00
|
|
|
m_pausing int32
|
|
|
|
m_nServerBw int32
|
|
|
|
m_nClientBw int32
|
2018-07-18 18:10:36 +03:00
|
|
|
m_nClientBw2 uint8
|
|
|
|
m_bPlaying uint8
|
|
|
|
m_bSendEncoding uint8
|
|
|
|
m_bSendCounter uint8
|
2018-08-10 10:20:53 +03:00
|
|
|
m_numInvokes int32
|
|
|
|
m_numCalls int32
|
2018-08-10 16:19:43 +03:00
|
|
|
m_methodCalls *C_RTMP_METHOD
|
2018-08-10 10:20:53 +03:00
|
|
|
m_channelsAllocatedIn int32
|
|
|
|
m_channelsAllocatedOut int32
|
2018-08-10 16:19:43 +03:00
|
|
|
m_vecChannelsIn **C_RTMPPacket
|
|
|
|
m_vecChannelsOut **C_RTMPPacket
|
2018-08-10 10:20:53 +03:00
|
|
|
m_channelTimestamp *int32
|
2018-07-18 18:10:36 +03:00
|
|
|
m_fAudioCodecs float64
|
|
|
|
m_fVideoCodecs float64
|
|
|
|
m_fEncoding float64
|
|
|
|
m_fDuration float64
|
2018-08-10 10:20:53 +03:00
|
|
|
m_msgCounter int32
|
|
|
|
m_polling int32
|
|
|
|
m_resplen int32
|
|
|
|
m_unackd int32
|
2018-08-10 16:19:43 +03:00
|
|
|
m_clientID C_AVal
|
|
|
|
m_read C_RTMP_READ
|
|
|
|
m_write C_RTMPPacket
|
|
|
|
m_sb C_RTMPSockBuf
|
|
|
|
Link C_RTMP_LNK
|
2018-07-18 18:10:36 +03:00
|
|
|
}
|
|
|
|
|
2018-08-07 07:24:58 +03:00
|
|
|
// typedef struct RTMPPacket
|
|
|
|
// rtmp.h +113
|
2018-08-07 07:28:00 +03:00
|
|
|
type C_RTMPPacket struct {
|
2018-07-18 18:13:59 +03:00
|
|
|
m_headerType uint8
|
|
|
|
m_packetType uint8
|
|
|
|
m_hasAbsTimestamp uint8
|
2018-08-10 10:20:53 +03:00
|
|
|
m_nChannel int32
|
2018-07-18 18:13:59 +03:00
|
|
|
m_nTimeStamp uint32
|
|
|
|
m_nInfoField2 int32
|
|
|
|
m_nBodySize uint32
|
|
|
|
m_nBytesRead uint32
|
2018-08-10 16:19:43 +03:00
|
|
|
m_chunk *C_RTMPChunk
|
2018-07-18 18:13:59 +03:00
|
|
|
m_body *byte
|
|
|
|
}
|
|
|
|
|
2018-08-07 07:24:58 +03:00
|
|
|
// typedef struct RTMPMethod
|
|
|
|
// rtmp.h +231
|
2018-08-07 07:28:00 +03:00
|
|
|
type C_RTMP_METHOD struct {
|
2018-08-10 16:19:43 +03:00
|
|
|
name C_AVal
|
2018-08-10 10:20:53 +03:00
|
|
|
num int32
|
2018-07-18 18:15:01 +03:00
|
|
|
}
|
|
|
|
|
2018-08-10 09:40:08 +03:00
|
|
|
// typedef struct C.AVal
|
2018-08-07 07:24:58 +03:00
|
|
|
// amf.h +57
|
2018-08-07 07:28:00 +03:00
|
|
|
type C_AVal struct {
|
2018-07-18 18:16:52 +03:00
|
|
|
av_val *byte
|
2018-08-10 10:20:53 +03:00
|
|
|
av_len int32
|
2018-07-18 18:16:52 +03:00
|
|
|
}
|
|
|
|
|
2018-08-07 07:24:58 +03:00
|
|
|
// typedef struct RTMP_READ
|
|
|
|
// rtmp.h +200
|
2018-08-07 07:28:00 +03:00
|
|
|
type C_RTMP_READ struct {
|
2018-07-18 18:24:36 +03:00
|
|
|
buf *byte
|
|
|
|
bufpos *byte
|
|
|
|
buflen uint
|
|
|
|
timestamp uint32
|
|
|
|
dataType uint8
|
|
|
|
flags uint8
|
|
|
|
status int8
|
|
|
|
initialFrameType uint8
|
|
|
|
nResumeTS uint32
|
|
|
|
metaHeader *byte
|
|
|
|
initialFrame *byte
|
|
|
|
nMetaHeaderSize uint32
|
|
|
|
nInitialFrameSize uint32
|
|
|
|
nIgnoredFrameCounter uint32
|
|
|
|
nIgnoredFlvFrameCounter uint32
|
|
|
|
}
|
|
|
|
|
2018-08-07 07:24:58 +03:00
|
|
|
// typedef struct RTMP_READ
|
|
|
|
// rtmp.h +200
|
2018-08-07 07:28:00 +03:00
|
|
|
type C_RTMPSockBuf struct {
|
2018-08-10 10:20:53 +03:00
|
|
|
sb_socket int32
|
|
|
|
sb_size int32
|
2018-07-18 18:33:59 +03:00
|
|
|
sb_start *byte
|
2018-07-22 11:48:46 +03:00
|
|
|
sb_buf [RTMP_BUFFER_CACHE_SIZE]byte // port const
|
2018-08-10 10:20:53 +03:00
|
|
|
sb_timedout int32
|
2018-07-18 18:33:59 +03:00
|
|
|
sb_ssl uintptr
|
|
|
|
}
|
|
|
|
|
2018-08-07 07:24:58 +03:00
|
|
|
// typedef struct RTMPChunk
|
|
|
|
// rtmp.h +105
|
2018-08-07 07:28:00 +03:00
|
|
|
type C_RTMPChunk struct {
|
2018-08-10 10:20:53 +03:00
|
|
|
c_headerSize int32
|
|
|
|
c_chunkSize int32
|
2018-07-18 18:41:55 +03:00
|
|
|
c_chunk *byte
|
2018-07-19 09:08:10 +03:00
|
|
|
c_header [RTMP_MAX_HEADER_SIZE]byte
|
2018-07-18 18:41:55 +03:00
|
|
|
}
|
|
|
|
|
2018-08-07 07:24:58 +03:00
|
|
|
// typedef struct RTMP_LNK
|
|
|
|
// rtmp.h +144
|
2018-08-07 07:28:00 +03:00
|
|
|
type C_RTMP_LNK struct {
|
2018-08-10 16:19:43 +03:00
|
|
|
hostname C_AVal
|
|
|
|
sockshost C_AVal
|
|
|
|
playpath0 C_AVal
|
|
|
|
playpath C_AVal
|
|
|
|
tcUrl C_AVal
|
|
|
|
swfUrl C_AVal
|
|
|
|
pageUrl C_AVal
|
|
|
|
app C_AVal
|
|
|
|
auth C_AVal
|
|
|
|
flashVer C_AVal
|
|
|
|
subscribepath C_AVal
|
|
|
|
usherToken C_AVal
|
|
|
|
token C_AVal
|
|
|
|
pubUser C_AVal
|
|
|
|
pubPasswd C_AVal
|
|
|
|
extras C_AMFObject
|
2018-08-10 10:20:53 +03:00
|
|
|
edepth int32
|
|
|
|
seekTime int32
|
|
|
|
stopTime int32
|
|
|
|
lFlags int32
|
|
|
|
swfAge int32
|
|
|
|
protocol int32
|
|
|
|
timeout int32
|
|
|
|
pFlags int32
|
2018-07-31 00:46:10 +03:00
|
|
|
socksport uint16
|
|
|
|
port uint16
|
2018-07-18 18:54:47 +03:00
|
|
|
}
|
|
|
|
|
2018-08-07 07:24:58 +03:00
|
|
|
// typedef struct AMF_Object
|
2018-08-07 07:28:00 +03:00
|
|
|
// amf.h +67
|
|
|
|
type C_AMFObject struct {
|
2018-08-10 10:20:53 +03:00
|
|
|
o_num int32
|
2018-08-10 16:19:43 +03:00
|
|
|
o_props *C_AMFObjectProperty
|
|
|
|
}
|
|
|
|
|
|
|
|
// typedef struct AMFObjectProperty
|
|
|
|
// amf.h +79
|
|
|
|
type C_AMFObjectProperty struct {
|
|
|
|
p_name C_AVal
|
|
|
|
p_type C_AMFDataType
|
|
|
|
p_vu P_vu
|
|
|
|
p_UTCoffset int16
|
|
|
|
}
|
|
|
|
|
|
|
|
// typedef struct P_vu
|
|
|
|
// amf.h +73
|
|
|
|
type P_vu struct {
|
|
|
|
p_number float32
|
|
|
|
p_aval C_AVal
|
|
|
|
p_object C_AMFObject
|
2018-07-18 18:57:56 +03:00
|
|
|
}
|
|
|
|
|
2018-05-29 08:56:36 +03:00
|
|
|
// NewSession returns a new session.
|
|
|
|
func NewSession(url string, connectTimeout uint) Session {
|
2018-08-10 15:47:01 +03:00
|
|
|
return &session{
|
2018-05-29 08:56:36 +03:00
|
|
|
url: url,
|
|
|
|
timeout: connectTimeout,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-06-17 14:15:48 +03:00
|
|
|
// Open establishes an rtmp connection with the url passed into the
|
2018-05-29 08:56:36 +03:00
|
|
|
// constructor
|
2018-08-10 15:47:01 +03:00
|
|
|
func (s *session) Open() error {
|
2018-05-30 09:22:33 +03:00
|
|
|
if s.rtmp != nil {
|
2018-05-29 08:56:36 +03:00
|
|
|
return errors.New("rtmp: attempt to start already running session")
|
|
|
|
}
|
2018-06-20 07:26:40 +03:00
|
|
|
var err error
|
2018-07-19 09:35:14 +03:00
|
|
|
s.rtmp, err = startSession(s.rtmp, s.url, uint32(s.timeout))
|
2018-05-30 09:22:33 +03:00
|
|
|
if s.rtmp == nil {
|
2018-06-20 07:26:40 +03:00
|
|
|
return err
|
2018-05-29 08:56:36 +03:00
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2018-07-19 18:35:21 +03:00
|
|
|
// Close terminates the rtmp connection
|
2018-08-10 15:47:01 +03:00
|
|
|
func (s *session) Close() error {
|
2018-07-19 18:35:21 +03:00
|
|
|
if s.rtmp == nil {
|
|
|
|
return Err(3)
|
|
|
|
}
|
|
|
|
ret := endSession(s.rtmp)
|
|
|
|
s.rtmp = nil
|
|
|
|
if ret != 0 {
|
|
|
|
return Err(ret)
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Write writes a frame (flv tag) to the rtmp connection
|
2018-08-10 15:47:01 +03:00
|
|
|
func (s *session) Write(data []byte) (int, error) {
|
2018-07-19 18:35:21 +03:00
|
|
|
if s.rtmp == nil {
|
|
|
|
return 0, Err(3)
|
|
|
|
}
|
2018-07-22 17:01:19 +03:00
|
|
|
|
2018-08-10 15:50:11 +03:00
|
|
|
if C_RTMP_IsConnected(s.rtmp) == 0 {
|
|
|
|
//if C.RTMP_IsConnected(s.rtmp) == 0 {
|
2018-07-19 18:35:21 +03:00
|
|
|
return 0, Err(1)
|
|
|
|
}
|
2018-07-28 08:38:09 +03:00
|
|
|
|
2018-08-10 15:57:51 +03:00
|
|
|
if C_RTMP_Write(s.rtmp, data) == 0 {
|
|
|
|
//if C.RTMP_Write(s.rtmp, (*C.char)(unsafe.Pointer(&data[0])), C.int(len(data))) == 0 {
|
2018-07-19 18:35:21 +03:00
|
|
|
return 0, Err(2)
|
|
|
|
}
|
|
|
|
return len(data), nil
|
|
|
|
}
|
|
|
|
|
2018-08-07 07:49:11 +03:00
|
|
|
// int RTMP_IsConnected(RTMP *r);
|
|
|
|
// rtmp.c +363
|
2018-08-06 07:30:37 +03:00
|
|
|
func C_RTMP_IsConnected(r *C.RTMP) int32 {
|
2018-07-22 16:36:10 +03:00
|
|
|
if r.m_sb.sb_socket != -1 {
|
|
|
|
return 1
|
|
|
|
}
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2018-07-19 09:35:14 +03:00
|
|
|
func startSession(rtmp *C.RTMP, u string, timeout uint32) (*C.RTMP, error) {
|
|
|
|
connect_timeout := C.int(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-07-28 08:54:36 +03:00
|
|
|
//C.RTMP_Close(rtmp)
|
|
|
|
//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-07-28 08:54:36 +03:00
|
|
|
//C.RTMP_Close(rtmp)
|
|
|
|
//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-07-28 08:54:36 +03:00
|
|
|
//C.RTMP_Close(rtmp)
|
|
|
|
//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
|
|
|
|
|
|
|
// RTMP* RTMP_IsConnected();
|
|
|
|
// rtmp.c +317
|
2018-08-06 09:12:16 +03:00
|
|
|
func C_RTMP_Alloc() *C.RTMP {
|
2018-07-19 10:54:37 +03:00
|
|
|
var r C.RTMP
|
2018-08-11 07:18:21 +03:00
|
|
|
return (*C.RTMP)(C.malloc(C.size_t(unsafe.Sizeof(r))))
|
|
|
|
//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-07 05:34:04 +03:00
|
|
|
func C_RTMP_Init(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-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-07 07:49:11 +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-10 18:47:48 +03:00
|
|
|
//ret = int32(C.RTMP_ParseURL((*C.char)(unsafe.Pointer(url)), &r.Link.protocol,
|
|
|
|
// &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
|
|
|
|
|
|
|
r.Link.port = C.ushort(port)
|
|
|
|
r.Link.playpath = r.Link.playpath0
|
|
|
|
|
|
|
|
if r.Link.tcUrl.av_len == 0 {
|
2018-07-20 07:40:08 +03:00
|
|
|
r.Link.tcUrl.av_val = (*C.char)(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-07-20 08:44:39 +03:00
|
|
|
r.Link.tcUrl.av_len = C.int(int(r.Link.app.av_len) +
|
|
|
|
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-11 07:18:21 +03:00
|
|
|
r.Link.tcUrl.av_val = (*C.char)(C.malloc(C.size_t(uintptr(length))))
|
2018-07-19 18:35:21 +03:00
|
|
|
hostname := string(ptrToSlice(unsafe.Pointer(r.Link.hostname.av_val),
|
|
|
|
int(r.Link.hostname.av_len)))
|
|
|
|
|
|
|
|
app := string(ptrToSlice(unsafe.Pointer(r.Link.app.av_val),
|
|
|
|
int(r.Link.app.av_len)))
|
|
|
|
|
|
|
|
fString := fmt.Sprintf("%v://%v:%v/%v",
|
|
|
|
RTMPProtocolStringsLower[r.Link.protocol], hostname, r.Link.port, app)
|
|
|
|
|
2018-07-20 08:44:39 +03:00
|
|
|
r.Link.tcUrl.av_val = (*C.char)(bToUP(goStrToCStr(fString)))
|
|
|
|
r.Link.tcUrl.av_len = C.int(strLen(RTMPProtocolStringsLower[r.Link.protocol]) +
|
|
|
|
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-07-20 08:44:39 +03:00
|
|
|
r.Link.tcUrl.av_len = C.int(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-10 16:52:53 +03:00
|
|
|
// int RTMP_ParseURL(const char *url, int *protocol, AVal *host, unsigned int *port,
|
|
|
|
// AVal *playpath, AVal *app);
|
|
|
|
// parseurl.c +33
|
|
|
|
func C_RTMP_ParseURL(url *byte, protocol *int32, host *C.AVal, port *uint32,
|
2018-08-10 18:43:25 +03:00
|
|
|
playpath *C.AVal, app *C.AVal) int {
|
2018-08-10 16:52:53 +03:00
|
|
|
|
|
|
|
var p, end, col, ques, slash *byte
|
2018-08-10 18:43:25 +03:00
|
|
|
// TODO: use our logger here
|
2018-08-10 16:52:53 +03:00
|
|
|
// RTMP_Log(RTMP_LOGDEBUG, "Parsing...");
|
|
|
|
|
|
|
|
*protocol = RTMP_PROTOCOL_RTMP
|
|
|
|
*port = 0
|
|
|
|
playpath.av_len = 0
|
|
|
|
playpath.av_val = nil
|
|
|
|
app.av_len = 0
|
|
|
|
app.av_val = nil
|
|
|
|
|
2018-08-10 18:43:25 +03:00
|
|
|
p = strstr(url, goStrToCStr("://"))
|
|
|
|
|
|
|
|
if p == nil {
|
|
|
|
// TODO: use our logger here
|
|
|
|
log.Println("RTMP URL: No :// in url!")
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
/*
|
|
|
|
NOTE: the following code nees to be ported if we're using anything other than
|
|
|
|
rtmp!
|
|
|
|
{
|
|
|
|
int len = (int)(p-url);
|
|
|
|
|
|
|
|
if(len == 4 && strncasecmp(url, "rtmp", 4)==0)
|
|
|
|
*protocol = RTMP_PROTOCOL_RTMP;
|
|
|
|
else if(len == 5 && strncasecmp(url, "rtmpt", 5)==0)
|
|
|
|
*protocol = RTMP_PROTOCOL_RTMPT;
|
|
|
|
else if(len == 5 && strncasecmp(url, "rtmps", 5)==0)
|
|
|
|
*protocol = RTMP_PROTOCOL_RTMPS;
|
|
|
|
else if(len == 5 && strncasecmp(url, "rtmpe", 5)==0)
|
|
|
|
*protocol = RTMP_PROTOCOL_RTMPE;
|
|
|
|
else if(len == 5 && strncasecmp(url, "rtmfp", 5)==0)
|
|
|
|
*protocol = RTMP_PROTOCOL_RTMFP;
|
|
|
|
else if(len == 6 && strncasecmp(url, "rtmpte", 6)==0)
|
|
|
|
*protocol = RTMP_PROTOCOL_RTMPTE;
|
|
|
|
else if(len == 6 && strncasecmp(url, "rtmpts", 6)==0)
|
|
|
|
*protocol = RTMP_PROTOCOL_RTMPTS;
|
|
|
|
else {
|
|
|
|
RTMP_Log(RTMP_LOGWARNING, "Unknown protocol!\n");
|
|
|
|
goto parsehost;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
*/
|
|
|
|
// TODO: implement new logger here
|
|
|
|
// RTMP_Log(RTMP_LOGDEBUG, "Parsed protocol: %d", *protocol);
|
|
|
|
|
|
|
|
// Get the hostname
|
|
|
|
p = (*byte)(incBytePtr(unsafe.Pointer(p), 3))
|
|
|
|
|
|
|
|
// check for sudden death
|
|
|
|
if *p == 0 {
|
|
|
|
// TODO: use new logger here
|
|
|
|
// RTMP_Log(RTMP_LOGWARNING, "No hostname in URL!");
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
|
|
|
end = (*byte)(unsafe.Pointer(uintptr(unsafe.Pointer(p)) + uintptr(strlen(p))))
|
|
|
|
col = strchr(p, ':')
|
|
|
|
ques = strchr(p, '?')
|
|
|
|
slash = strchr(p, '/')
|
|
|
|
|
|
|
|
{
|
|
|
|
var hostlen int32
|
|
|
|
if slash != nil {
|
|
|
|
hostlen = int32(uintptr(unsafe.Pointer(slash)) - uintptr(unsafe.Pointer(p)))
|
|
|
|
} else {
|
|
|
|
hostlen = int32(uintptr(unsafe.Pointer(end)) - uintptr(unsafe.Pointer(p)))
|
|
|
|
}
|
|
|
|
if col != nil && int32(uintptr(unsafe.Pointer(col))-uintptr(unsafe.Pointer(p))) < hostlen {
|
|
|
|
hostlen = int32(uintptr(unsafe.Pointer(col)) - uintptr(unsafe.Pointer(p)))
|
|
|
|
}
|
|
|
|
|
|
|
|
if hostlen < 256 {
|
|
|
|
host.av_val = (*C.char)(unsafe.Pointer(p))
|
|
|
|
host.av_len = C.int(hostlen)
|
|
|
|
// TODO: use new logger with this
|
2018-08-12 12:07:39 +03:00
|
|
|
//RTMP_Log(RTMP_LOGDEBUG, "Parsed host : %.*s", hostlen, host.av_val);
|
2018-08-10 18:43:25 +03:00
|
|
|
} else {
|
|
|
|
// TODO: use new logger with this
|
|
|
|
// RTMP_Log(RTMP_LOGWARNING, "Hostname exceeds 255 characters!");
|
|
|
|
}
|
|
|
|
|
|
|
|
p = (*byte)(incBytePtr(unsafe.Pointer(p), int(hostlen)))
|
|
|
|
}
|
|
|
|
|
|
|
|
// get port number if available
|
|
|
|
if *p == ':' {
|
|
|
|
var p2 uint32
|
|
|
|
p = (*byte)(incBytePtr(unsafe.Pointer(p), 1))
|
|
|
|
tmp, _ := strconv.Atoi(cStrToGoStr(p))
|
|
|
|
p2 = uint32(tmp)
|
|
|
|
if p2 > 65535 {
|
|
|
|
// TODO: use new logger with this
|
|
|
|
// RTMP_Log(RTMP_LOGWARNING, "Invalid port number!");
|
|
|
|
} else {
|
|
|
|
*port = p2
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if slash == nil {
|
|
|
|
// TODO: use new logger
|
|
|
|
// RTMP_Log(RTMP_LOGWARNING, "No application or playpath in URL!");
|
|
|
|
return 1
|
|
|
|
}
|
|
|
|
|
|
|
|
p = (*byte)(incBytePtr(unsafe.Pointer(slash), 1))
|
2018-08-10 16:52:53 +03:00
|
|
|
|
2018-08-10 18:43:25 +03:00
|
|
|
{
|
|
|
|
/* parse application
|
|
|
|
*
|
|
|
|
* rtmp://host[:port]/app[/appinstance][/...]
|
|
|
|
* application = app[/appinstance]
|
|
|
|
*/
|
|
|
|
|
|
|
|
var slash2 *byte
|
|
|
|
var slash3 *byte = nil
|
|
|
|
var slash4 *byte = nil
|
|
|
|
var applen, appnamelen int32
|
|
|
|
|
|
|
|
slash2 = strchr(p, '/')
|
|
|
|
|
|
|
|
if slash2 != nil {
|
|
|
|
slash3 = strchr((*byte)(unsafe.Pointer(uintptr(unsafe.Pointer(slash2))+
|
|
|
|
uintptr(1))), '/')
|
|
|
|
}
|
|
|
|
if slash3 != nil {
|
|
|
|
slash4 = strchr((*byte)(unsafe.Pointer(uintptr(unsafe.Pointer(slash3))+
|
|
|
|
uintptr(1))), '/')
|
|
|
|
}
|
|
|
|
|
|
|
|
// ondemand, pass all parameters as app
|
|
|
|
applen = int32(uintptr(unsafe.Pointer(end)) - uintptr(unsafe.Pointer(p)))
|
|
|
|
appnamelen = applen
|
|
|
|
|
|
|
|
switch {
|
|
|
|
case ques != nil && strstr(p, goStrToCStr("slist=")) != nil:
|
|
|
|
appnamelen = int32(uintptr(unsafe.Pointer(ques)) - uintptr(unsafe.Pointer(p)))
|
|
|
|
case strings.Compare(cStrToGoStr(p)[:9], "ondemand/") == 0:
|
|
|
|
/* app = ondemand/foobar, only pass app=ondemand */
|
|
|
|
applen = 8
|
|
|
|
appnamelen = 8
|
|
|
|
default:
|
|
|
|
switch {
|
|
|
|
case slash4 != nil:
|
|
|
|
appnamelen = int32(uintptr(unsafe.Pointer(slash4)) - uintptr(
|
|
|
|
unsafe.Pointer(p)))
|
|
|
|
case slash3 != nil:
|
|
|
|
appnamelen = int32(uintptr(unsafe.Pointer(slash3)) - uintptr(
|
|
|
|
unsafe.Pointer(p)))
|
|
|
|
case slash2 != nil:
|
|
|
|
appnamelen = int32(uintptr(unsafe.Pointer(slash2)) - uintptr(
|
|
|
|
unsafe.Pointer(p)))
|
|
|
|
}
|
|
|
|
|
|
|
|
applen = appnamelen
|
|
|
|
}
|
|
|
|
|
|
|
|
app.av_val = (*C.char)(unsafe.Pointer(p))
|
|
|
|
app.av_len = C.int(applen)
|
|
|
|
// TODO: use new logging here
|
|
|
|
// RTMP_Log(RTMP_LOGDEBUG, "Parsed app : %.*s", applen, p);
|
|
|
|
|
|
|
|
p = (*byte)(incBytePtr(unsafe.Pointer(p), int(appnamelen)))
|
|
|
|
}
|
|
|
|
|
|
|
|
if *p == '/' {
|
|
|
|
p = (*byte)(incBytePtr(unsafe.Pointer(p), 1))
|
|
|
|
}
|
2018-08-10 19:30:11 +03:00
|
|
|
// NOTE: don't think we currently need this section - see 787 for this func
|
2018-08-10 18:43:25 +03:00
|
|
|
|
|
|
|
if int(uintptr(unsafe.Pointer(end))-uintptr(unsafe.Pointer(p))) != 0 {
|
|
|
|
var av C.AVal
|
|
|
|
av.av_val = (*C.char)(unsafe.Pointer(p))
|
|
|
|
av.av_len = C.int(uintptr(unsafe.Pointer(end)) - uintptr(unsafe.Pointer(p)))
|
|
|
|
// TODO: port THis
|
2018-08-10 19:44:17 +03:00
|
|
|
//C.RTMP_ParsePlaypath(&av, playpath)
|
|
|
|
C_RTMP_ParsePlaypath(&av, playpath)
|
2018-08-10 18:43:25 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
return 1
|
2018-08-10 16:52:53 +03:00
|
|
|
}
|
|
|
|
|
2018-08-10 19:30:11 +03:00
|
|
|
// void RTMP_ParsePlaypath(AVal *in, AVal *out);
|
|
|
|
// parseurl.c +201
|
|
|
|
func C_RTMP_ParsePlaypath(in, out *C.AVal) {
|
|
|
|
var addMP4 int32 = 0
|
|
|
|
var addMP3 int32 = 0
|
|
|
|
var subExt int32 = 0
|
|
|
|
playpath := in.av_val
|
|
|
|
var temp, q *byte
|
|
|
|
var ext *byte = nil
|
|
|
|
ppstart := (*byte)(unsafe.Pointer(playpath))
|
|
|
|
var streamname, destptr, p *byte
|
|
|
|
|
|
|
|
pplen := int32(in.av_len)
|
|
|
|
|
|
|
|
out.av_val = nil
|
|
|
|
out.av_len = 0
|
|
|
|
|
|
|
|
temp = strstr((*byte)(unsafe.Pointer(ppstart)), goStrToCStr("slist="))
|
|
|
|
if *ppstart == '?' && temp != nil {
|
|
|
|
ppstart = (*byte)(incBytePtr(unsafe.Pointer(temp), 6))
|
|
|
|
pplen = int32(strlen(ppstart))
|
|
|
|
|
|
|
|
temp = strchr(ppstart, '&')
|
|
|
|
|
|
|
|
if temp != nil {
|
|
|
|
pplen = int32(uintptr(unsafe.Pointer(temp)) - uintptr(unsafe.Pointer(ppstart)))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
q = strchr(ppstart, '?')
|
|
|
|
|
|
|
|
if pplen >= 4 {
|
|
|
|
if q != nil {
|
|
|
|
ext = (*byte)(decBytePtr(unsafe.Pointer(q), 4))
|
|
|
|
} else {
|
|
|
|
ext = (*byte)(indxBytePtr(unsafe.Pointer(ppstart), int(uintptr(pplen)-
|
|
|
|
uintptr(4))))
|
|
|
|
}
|
|
|
|
switch {
|
|
|
|
case strings.Compare(cStrToGoStr(ext)[:4], ".f4v") == 0 ||
|
|
|
|
strings.Compare(cStrToGoStr(ext)[:4], ".mp4") == 0:
|
|
|
|
addMP4 = 1
|
|
|
|
subExt = 1
|
|
|
|
case ppstart == (*byte)(unsafe.Pointer(playpath)) && strings.Compare(
|
|
|
|
cStrToGoStr(ext)[:4], ".flv") == 0:
|
|
|
|
subExt = 1
|
|
|
|
case strings.Compare(cStrToGoStr(ext)[:4], ".mp3") == 0:
|
|
|
|
addMP3 = 1
|
|
|
|
subExt = 1
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
streamname = (*byte)(C.malloc(C.size_t(pplen + 4 + 1)))
|
|
|
|
|
|
|
|
if streamname == nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
destptr = streamname
|
|
|
|
switch {
|
|
|
|
case addMP4 != 0:
|
|
|
|
if strings.Compare(cStrToGoStr(ppstart)[:4], "mp4") != 0 {
|
|
|
|
memmove(unsafe.Pointer(destptr), unsafe.Pointer(goStrToCStr("mp4:")),
|
|
|
|
uintptr(len("mp4:")))
|
|
|
|
destptr = (*byte)(incBytePtr(unsafe.Pointer(destptr), 4))
|
|
|
|
} else {
|
|
|
|
subExt = 0
|
|
|
|
}
|
|
|
|
case addMP3 != 0:
|
|
|
|
if strings.Compare(cStrToGoStr(ppstart)[:4], "mp3") != 0 {
|
|
|
|
memmove(unsafe.Pointer(destptr), unsafe.Pointer(goStrToCStr("mp3:")),
|
|
|
|
uintptr(len("mp4:")))
|
|
|
|
destptr = (*byte)(incBytePtr(unsafe.Pointer(destptr), 4))
|
|
|
|
} else {
|
|
|
|
subExt = 0
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
p = (*byte)(ppstart)
|
|
|
|
for pplen > 0 {
|
|
|
|
if subExt != 0 && p == ext {
|
|
|
|
p = (*byte)(incBytePtr(unsafe.Pointer(p), 4))
|
|
|
|
pplen -= 4
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
if *p == '%' {
|
|
|
|
var c uint32
|
|
|
|
fmt.Sscanf(cStrToGoStr((*byte)(incBytePtr(unsafe.Pointer(p), 1))), "%02x", &c)
|
|
|
|
*indxBytePtr(unsafe.Pointer(destptr), 0) = byte(c)
|
|
|
|
destptr = (*byte)(incBytePtr(unsafe.Pointer(destptr), 1))
|
|
|
|
pplen -= 3
|
|
|
|
p = (*byte)(incBytePtr(unsafe.Pointer(p), 3))
|
|
|
|
} else {
|
|
|
|
*indxBytePtr(unsafe.Pointer(destptr), 0) = *p
|
|
|
|
destptr = (*byte)(incBytePtr(unsafe.Pointer(destptr), 1))
|
|
|
|
p = (*byte)(incBytePtr(unsafe.Pointer(p), 1))
|
|
|
|
pplen--
|
|
|
|
}
|
|
|
|
}
|
|
|
|
*destptr = '\x00'
|
|
|
|
|
|
|
|
out.av_val = (*C.char)(unsafe.Pointer(streamname))
|
|
|
|
out.av_len = C.int(uintptr(unsafe.Pointer(destptr)) - uintptr(unsafe.Pointer(
|
|
|
|
streamname)))
|
|
|
|
}
|
|
|
|
|
2018-08-10 09:40:08 +03:00
|
|
|
// void SocksSetup(RTMP *r, C.AVal* sockshost);
|
2018-08-07 07:49:11 +03:00
|
|
|
// rtmp.c +410
|
2018-08-07 05:35:58 +03:00
|
|
|
func C_SocksSetup(r *C.RTMP, sockshost *C.AVal) {
|
2018-07-22 17:01:19 +03:00
|
|
|
if sockshost.av_len != 0 {
|
|
|
|
socksport := strchr((*byte)(unsafe.Pointer(sockshost.av_val)), ':')
|
|
|
|
hostname := strdup((*byte)(unsafe.Pointer(sockshost.av_val)))
|
|
|
|
|
2018-08-10 09:40:08 +03:00
|
|
|
if unsafe.Pointer(socksport) != nil {
|
2018-07-22 17:01:19 +03:00
|
|
|
*indxBytePtr(unsafe.Pointer(hostname),
|
|
|
|
int(uintptr(decBytePtr(unsafe.Pointer(socksport),
|
|
|
|
int(uintptr(unsafe.Pointer(sockshost.av_val))))))) = '\000'
|
|
|
|
r.Link.sockshost.av_val = (*C.char)(unsafe.Pointer(hostname))
|
|
|
|
r.Link.sockshost.av_len = C.int(strlen(hostname))
|
|
|
|
|
|
|
|
value, err := strconv.Atoi(string(ptrToSlice(unsafe.Pointer(uintptr(
|
|
|
|
unsafe.Pointer(socksport))+uintptr(1)), int(strlen((*byte)(unsafe.Pointer(
|
|
|
|
uintptr(unsafe.Pointer(socksport))+uintptr(1)))))+1)))
|
|
|
|
if err != nil {
|
2018-08-07 05:35:58 +03:00
|
|
|
log.Println("C_SocksSetup: bad string conversion!")
|
2018-07-22 17:01:19 +03:00
|
|
|
}
|
|
|
|
if uintptr(unsafe.Pointer(socksport)) == 0 {
|
|
|
|
value = 1080
|
|
|
|
}
|
|
|
|
r.Link.socksport = C.ushort(value)
|
|
|
|
}
|
2018-08-10 07:01:17 +03:00
|
|
|
} else {
|
|
|
|
r.Link.sockshost.av_val = nil
|
|
|
|
r.Link.sockshost.av_len = 0
|
|
|
|
r.Link.socksport = 0
|
2018-07-22 17:01:19 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-07-22 11:48:46 +03:00
|
|
|
/*
|
2018-07-20 14:53:57 +03:00
|
|
|
func rtmpClose(r *C.RTMP) {
|
|
|
|
closeInternal(r, 0)
|
|
|
|
}
|
|
|
|
|
|
|
|
func closeInternal(r *C.RTMP, reconnect int32) {
|
|
|
|
var i int32
|
|
|
|
|
2018-08-06 09:12:16 +03:00
|
|
|
if C_RTMP_IsConnected(r) != 0 {
|
2018-07-20 14:53:57 +03:00
|
|
|
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, C.double(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)
|
|
|
|
|
|
|
|
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-07-20 08:44:39 +03:00
|
|
|
}
|
2018-07-22 11:48:46 +03:00
|
|
|
*/
|
2018-07-20 08:44:39 +03:00
|
|
|
|
2018-08-07 07:49:11 +03:00
|
|
|
// void RTMP_EnableWrite(RTMP *r);
|
|
|
|
// rtmp.c +351
|
2018-08-07 05:37:42 +03:00
|
|
|
func C_RTMP_EnableWrite(r *C.RTMP) {
|
2018-07-20 15:01:55 +03:00
|
|
|
r.Link.protocol |= RTMP_FEATURE_WRITE
|
|
|
|
}
|
|
|
|
|
2018-08-07 07:49:11 +03:00
|
|
|
// void RTMP_SetBufferMS(RTMP *r, int size);
|
|
|
|
// rtmp.c +381
|
2018-08-07 05:38:48 +03:00
|
|
|
func C_RTMP_SetBufferMS(r *C.RTMP, size int32) {
|
2018-07-20 15:04:38 +03:00
|
|
|
r.m_nBufferMS = C.int(size)
|
|
|
|
}
|
|
|
|
|
2018-08-07 07:49:11 +03:00
|
|
|
// int RTMP_Connect(RTMP *r, RTMPPacket* cp);
|
|
|
|
// rtmp.c +1032
|
2018-08-07 05:42:11 +03:00
|
|
|
func C_RTMP_Connect(r *C.RTMP, cp *C.RTMPPacket) int {
|
2018-07-22 11:48:46 +03:00
|
|
|
// TODO: port this
|
2018-07-21 02:24:16 +03:00
|
|
|
var service C.sockaddr_in
|
|
|
|
|
|
|
|
if r.Link.hostname.av_len == 0 {
|
|
|
|
return 0
|
|
|
|
}
|
2018-08-10 07:07:35 +03:00
|
|
|
memset((*byte)(unsafe.Pointer(&service)), 0, int(unsafe.Sizeof(service)))
|
2018-07-21 02:24:16 +03:00
|
|
|
|
2018-07-22 11:48:46 +03:00
|
|
|
// TODO: port this
|
2018-07-21 02:24:16 +03:00
|
|
|
service.sin_family = C.AF_INET
|
|
|
|
|
|
|
|
if r.Link.socksport != 0 {
|
2018-07-22 11:48:46 +03:00
|
|
|
// TODO: port this
|
2018-07-21 02:24:16 +03:00
|
|
|
if C.add_addr_info(&service, &r.Link.sockshost, C.int(r.Link.socksport)) == 0 {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// connect directly
|
2018-07-22 11:48:46 +03:00
|
|
|
if C.add_addr_info(&service, (*C.AVal)(unsafe.Pointer(&r.Link.hostname)),
|
|
|
|
C.int(r.Link.port)) == 0 {
|
2018-07-21 02:24:16 +03:00
|
|
|
return 0
|
|
|
|
}
|
|
|
|
}
|
2018-07-28 09:02:38 +03:00
|
|
|
//if C.RTMP_Connect0(r, (*C.sockaddr)(unsafe.Pointer(&service))) == 0 {
|
2018-08-07 05:42:11 +03:00
|
|
|
if C_RTMP_Connect0(r, (*C.sockaddr)(unsafe.Pointer(&service))) == 0 {
|
2018-07-21 02:24:16 +03:00
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
|
|
|
r.m_bSendCounter = 1
|
|
|
|
|
2018-08-07 05:42:11 +03:00
|
|
|
return int(C_RTMP_Connect1(r, cp))
|
2018-07-28 09:04:11 +03:00
|
|
|
//return int(C.RTMP_Connect1(r, cp))
|
2018-07-21 02:24:16 +03:00
|
|
|
}
|
|
|
|
|
2018-08-07 07:49:11 +03:00
|
|
|
// int RTMP_Connect0(RTMP *r, struct sockaddr* service);
|
|
|
|
// rtmp.c +906
|
2018-08-07 05:42:11 +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
|
|
|
|
|
|
|
r.m_sb.sb_socket = C.socket(C.AF_INET, C.SOCK_STREAM, C.IPPROTO_TCP)
|
|
|
|
|
|
|
|
if r.m_sb.sb_socket != -1 {
|
2018-07-24 15:00:50 +03:00
|
|
|
if C.connect(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))
|
|
|
|
// tv := C.int(r.Link.timeout * 1000)
|
2018-07-22 11:48:46 +03:00
|
|
|
|
|
|
|
if C.setsockopt(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
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
C.setsockopt(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-10 19:54:50 +03:00
|
|
|
// int SocksNegotiate(RTMP* r);
|
|
|
|
// rtmp.c +1062
|
2018-08-11 05:10:42 +03:00
|
|
|
func C_SocksNegotiate(r *C.RTMP) int {
|
|
|
|
var addr int32
|
|
|
|
var service C.sockaddr_in
|
|
|
|
|
|
|
|
memset((*byte)(unsafe.Pointer(&service)), 0, int(unsafe.Sizeof(service)))
|
|
|
|
|
|
|
|
C.add_addr_info(&service, &r.Link.hostname, C.int(r.Link.port))
|
|
|
|
addr = int32(inet.Htonl(uint32(service.sin_addr.s_addr)))
|
|
|
|
|
|
|
|
{
|
|
|
|
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-10 19:54:50 +03:00
|
|
|
|
|
|
|
// TODO: find location in c file
|
2018-08-10 09:40:08 +03:00
|
|
|
func SET_RCVTIMEO(tv *int32, s int32) {
|
|
|
|
*tv = s * 1000
|
|
|
|
}
|
|
|
|
|
2018-08-07 07:49:11 +03:00
|
|
|
// int RTMP_Connect1(RTMP* r, RTMPPacket* cp);
|
|
|
|
// rtmp.c +978
|
2018-08-07 05:42:11 +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-07 07:49:11 +03:00
|
|
|
// int HandShake(RTMP* r, int FP9HandShake);
|
|
|
|
// rtmp.c +3744
|
2018-08-07 05:54:25 +03:00
|
|
|
func C_HandShake(r *C.RTMP, FP9HandShake int32) int {
|
2018-07-24 15:00:50 +03:00
|
|
|
var bMatch int
|
|
|
|
//uptime := uint32(0)
|
|
|
|
//suptime := uint32(0)
|
|
|
|
//typ := byte(0)
|
2018-07-22 17:44:23 +03:00
|
|
|
var uptime, suptime uint32
|
|
|
|
var typ byte
|
2018-07-24 15:00:50 +03:00
|
|
|
//clientbuf := make([]byte, RTMP_SIG_SIZE+1)
|
2018-07-22 17:44:23 +03:00
|
|
|
var clientbuf [RTMP_SIG_SIZE + 1]byte
|
2018-07-24 15:00:50 +03:00
|
|
|
clientsig := (*byte)(incBytePtr(unsafe.Pointer(&clientbuf[0]), 1))
|
|
|
|
//serversig := make([]byte, RTMP_SIG_SIZE)
|
2018-07-22 17:44:23 +03:00
|
|
|
var serversig [RTMP_SIG_SIZE]byte
|
|
|
|
|
|
|
|
clientbuf[0] = 0x03 // not encrypted
|
|
|
|
|
|
|
|
// TODO: port rtmp_getTime
|
2018-08-11 05:51:22 +03:00
|
|
|
//uptime = inet.Htonl(uint32(C.RTMP_GetTime()))
|
|
|
|
uptime = inet.Htonl(uint32(C_RTMP_GetTime()))
|
2018-07-22 17:44:23 +03:00
|
|
|
memmove(unsafe.Pointer(clientsig), unsafe.Pointer(&uptime), 4)
|
|
|
|
|
|
|
|
memset(indxBytePtr(unsafe.Pointer(clientsig), 4), 0, 4)
|
|
|
|
|
|
|
|
for i := 8; i < RTMP_SIG_SIZE; i++ {
|
|
|
|
*indxBytePtr(unsafe.Pointer(clientsig), i) = byte(rand.Intn(256))
|
|
|
|
}
|
|
|
|
|
2018-08-07 06:10:43 +03:00
|
|
|
if C_WriteN(r, unsafe.Pointer(&clientbuf[0]), RTMP_SIG_SIZE+1) == 0 {
|
2018-07-23 02:08:22 +03:00
|
|
|
return 0
|
|
|
|
}
|
2018-07-22 17:44:23 +03:00
|
|
|
|
2018-08-02 15:39:05 +03:00
|
|
|
//if C.ReadN(r, (*C.char)(unsafe.Pointer(&typ)), 1) != 1 {
|
2018-08-07 05:55:20 +03:00
|
|
|
if C_ReadN(r, (*byte)(unsafe.Pointer(&typ)), 1) != 1 {
|
2018-07-23 02:08:22 +03:00
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
|
|
|
if debugMode {
|
2018-08-07 05:54:25 +03:00
|
|
|
log.Println("C_HandShake: Type answer: %v", typ)
|
2018-07-23 02:08:22 +03:00
|
|
|
}
|
2018-07-24 15:00:50 +03:00
|
|
|
if typ != clientbuf[0] {
|
2018-08-07 05:54:25 +03:00
|
|
|
log.Println("C_HandShake: type mismatch: client sent %v, server sent: %v",
|
2018-07-24 15:00:50 +03:00
|
|
|
clientbuf[0], typ)
|
2018-07-23 02:08:22 +03:00
|
|
|
}
|
2018-08-07 05:55:20 +03:00
|
|
|
if C_ReadN(r, (*byte)(unsafe.Pointer(&serversig[0])), RTMP_SIG_SIZE) != RTMP_SIG_SIZE {
|
2018-07-29 03:03:09 +03:00
|
|
|
//if C.ReadN(r, (*C.char)(unsafe.Pointer(&serversig[0])), RTMP_SIG_SIZE) != RTMP_SIG_SIZE {
|
2018-07-23 02:08:22 +03:00
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
|
|
|
// decode server response
|
2018-07-24 15:00:50 +03:00
|
|
|
memmove(unsafe.Pointer(&suptime), unsafe.Pointer(&serversig[0]), 4)
|
2018-07-23 02:08:22 +03:00
|
|
|
suptime = inet.Ntohl(suptime)
|
|
|
|
|
|
|
|
// 2nd part of handshake
|
2018-08-07 06:10:43 +03:00
|
|
|
if C_WriteN(r, unsafe.Pointer(&serversig[0]), RTMP_SIG_SIZE) == 0 {
|
2018-07-23 02:08:22 +03:00
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2018-08-07 05:55:20 +03:00
|
|
|
if C_ReadN(r, (*byte)(unsafe.Pointer(&serversig[0])), RTMP_SIG_SIZE) != RTMP_SIG_SIZE {
|
2018-08-02 15:39:05 +03:00
|
|
|
//if C.ReadN(r, (*C.char)(unsafe.Pointer(&serversig[0])), RTMP_SIG_SIZE) != RTMP_SIG_SIZE {
|
2018-07-23 02:08:22 +03:00
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: find golang memcmp
|
|
|
|
bMatch = 0
|
2018-07-24 15:00:50 +03:00
|
|
|
if memcmp(unsafe.Pointer(&serversig[0]), unsafe.Pointer(clientsig),
|
|
|
|
RTMP_SIG_SIZE) == 0 {
|
2018-07-23 02:08:22 +03:00
|
|
|
bMatch = 1
|
|
|
|
}
|
|
|
|
|
2018-07-25 04:26:59 +03:00
|
|
|
if bMatch == 0 {
|
2018-07-23 02:08:22 +03:00
|
|
|
log.Println("Client signature does not match!")
|
|
|
|
}
|
|
|
|
return 1
|
|
|
|
}
|
2018-07-22 17:44:23 +03:00
|
|
|
|
2018-08-11 05:51:22 +03:00
|
|
|
// uint32_t RTMP_GetTime();
|
|
|
|
// rtmp.c +156
|
|
|
|
func C_RTMP_GetTime() int32 {
|
|
|
|
return int32(time.Now().UnixNano() / 1000000)
|
|
|
|
}
|
|
|
|
|
2018-08-07 07:49:11 +03:00
|
|
|
// int ReadN(RTMP* r, char* buffer, int n);
|
|
|
|
// rtmp.c +1390
|
2018-08-07 05:55:20 +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))
|
|
|
|
r.m_sb.sb_start = (*C.char)(incBytePtr(unsafe.Pointer(r.m_sb.sb_start),
|
|
|
|
nRead))
|
|
|
|
r.m_sb.sb_size -= C.int(nRead)
|
|
|
|
nBytes = nRead
|
|
|
|
r.m_nBytesIn += C.int(nRead)
|
|
|
|
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-14 11:52:37 +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-07 07:49:11 +03:00
|
|
|
// int RTMPSockBuf_Fill(RTMPSockBuf* sb);
|
|
|
|
// rtmp.c +4253
|
|
|
|
func C_RTMPSockBuf_Fill(sb *C.RTMPSockBuf) int {
|
2018-07-28 09:58:24 +03:00
|
|
|
var nBytes int
|
|
|
|
|
|
|
|
if sb.sb_size == 0 {
|
|
|
|
sb.sb_start = &sb.sb_buf[0]
|
|
|
|
}
|
|
|
|
|
|
|
|
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-28 10:14:27 +03:00
|
|
|
// TODO: figure out what to do with recv
|
2018-07-28 09:58:24 +03:00
|
|
|
nBytes = int(C.recv(sb.sb_socket, unsafe.Pointer(uintptr(unsafe.Pointer(
|
|
|
|
sb.sb_start))+uintptr(int(sb.sb_size))), C.size_t(nBytes), 0))
|
|
|
|
|
|
|
|
if nBytes != -1 {
|
|
|
|
sb.sb_size += C.int(nBytes)
|
|
|
|
} else {
|
2018-08-07 07:49:11 +03:00
|
|
|
log.Println("C_RTMPSockBuf_Fill: recv error!")
|
2018-07-28 09:58:24 +03:00
|
|
|
}
|
|
|
|
break
|
|
|
|
}
|
|
|
|
return nBytes
|
|
|
|
}
|
|
|
|
|
2018-08-07 07:49:11 +03:00
|
|
|
// int SendBytesReceived(RTMP* r);
|
|
|
|
// rtmp.c +2080
|
2018-08-07 05:57:28 +03:00
|
|
|
func C_SendBytesReceived(r *C.RTMP) int {
|
2018-07-28 10:06:01 +03:00
|
|
|
var packet C.RTMPPacket
|
|
|
|
var pbuf [256]byte
|
|
|
|
pend := (*byte)(incBytePtr(unsafe.Pointer(&pbuf[0]), 256))
|
|
|
|
|
|
|
|
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 = (*C.char)(incBytePtr(unsafe.Pointer(&pbuf[0]),
|
|
|
|
RTMP_MAX_HEADER_SIZE))
|
|
|
|
|
|
|
|
packet.m_nBodySize = 4
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
C_AMF_EncodeInt32((*byte)(unsafe.Pointer(packet.m_body)), pend, int32(r.m_nBytesIn))
|
2018-07-28 10:14:27 +03:00
|
|
|
// C.AMF_EncodeInt32(packet.m_body, (*C.char)(unsafe.Pointer(pend)), r.m_nBytesIn)
|
2018-07-28 10:06:01 +03:00
|
|
|
|
|
|
|
r.m_nBytesInSent = r.m_nBytesIn
|
|
|
|
|
2018-08-02 15:39:05 +03:00
|
|
|
//return int(C.RTMP_SendPacket(r, &packet, 0))
|
2018-08-07 06:03:40 +03:00
|
|
|
return C_RTMP_SendPacket(r, &packet, 0)
|
2018-07-28 10:06:01 +03:00
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// int SendConnectPacket(RTMP* r, RTMPPacket* cp);
|
|
|
|
// rtmp.c +1579
|
2018-08-07 05:58:06 +03:00
|
|
|
func C_SendConnectPacket(r *C.RTMP, cp *C.RTMPPacket) int {
|
2018-07-25 04:26:59 +03:00
|
|
|
var packet C.RTMPPacket
|
|
|
|
var pbuf [4096]byte
|
2018-07-27 07:36:40 +03:00
|
|
|
pend := (*byte)(unsafe.Pointer(incBytePtr(unsafe.Pointer(&pbuf[0]),
|
|
|
|
int(unsafe.Sizeof(pbuf)))))
|
2018-07-25 04:26:59 +03:00
|
|
|
var enc *byte
|
|
|
|
|
|
|
|
if cp != nil {
|
2018-08-07 06:03:40 +03:00
|
|
|
return C_RTMP_SendPacket(r, cp, 1)
|
2018-07-29 03:57:46 +03:00
|
|
|
//return int(C.RTMP_SendPacket(r, cp, 1))
|
2018-07-25 04:26:59 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
packet.m_nChannel = 0x03
|
|
|
|
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 = (*C.char)(incBytePtr(unsafe.Pointer(&pbuf[0]),
|
|
|
|
RTMP_MAX_HEADER_SIZE))
|
|
|
|
|
|
|
|
enc = (*byte)(unsafe.Pointer(packet.m_body))
|
2018-07-28 15:00:08 +03:00
|
|
|
|
2018-07-29 03:12:20 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeString((*C.char)(unsafe.Pointer(enc)),
|
|
|
|
//(*C.char)(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-07-29 03:15:14 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNumber((*C.char)(unsafe.Pointer(enc)),
|
|
|
|
//(*C.char)(unsafe.Pointer(pend)), C.double(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-07-28 15:00:08 +03:00
|
|
|
*indxBytePtr(unsafe.Pointer(enc), 0) = AMF_OBJECT
|
|
|
|
|
|
|
|
enc = (*byte)(unsafe.Pointer(incBytePtr(unsafe.Pointer(enc), 1)))
|
|
|
|
|
2018-07-29 03:47:53 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedString((*C.char)(
|
|
|
|
//unsafe.Pointer(enc)), (*C.char)(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-07-29 03:47:53 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedString((*C.char)(
|
|
|
|
//unsafe.Pointer(enc)), (*C.char)(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-07-29 03:47:53 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedString((*C.char)(
|
|
|
|
//unsafe.Pointer(enc)), (*C.char)(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-07-29 03:47:53 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedString((*C.char)(
|
|
|
|
// unsafe.Pointer(enc)), (*C.char)(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-07-29 03:47:53 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedString((*C.char)(
|
|
|
|
//unsafe.Pointer(enc)), (*C.char)(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-07-29 03:51:53 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedBoolean((*C.char)(
|
|
|
|
//unsafe.Pointer(enc)), (*C.char)(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-07-29 03:54:27 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedNumber((*C.char)(
|
|
|
|
//unsafe.Pointer(enc)), (*C.char)(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-07-29 03:54:27 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedNumber((*C.char)(
|
|
|
|
// unsafe.Pointer(enc)), (*C.char)(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-07-29 03:54:27 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedNumber((*C.char)(
|
|
|
|
//unsafe.Pointer(enc)), (*C.char)(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-07-29 03:54:27 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedNumber((*C.char)(
|
|
|
|
// unsafe.Pointer(enc)), (*C.char)(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-07-29 03:56:05 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedString((*C.char)(
|
|
|
|
//unsafe.Pointer(enc)), (*C.char)(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-07-29 03:56:05 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeNamedNumber((*C.char)(
|
|
|
|
//unsafe.Pointer(enc)), (*C.char)(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
|
|
|
|
}
|
|
|
|
|
|
|
|
*indxBytePtr(unsafe.Pointer(enc), 0) = 0
|
|
|
|
enc = (*byte)(incBytePtr(unsafe.Pointer(enc), 1))
|
|
|
|
*indxBytePtr(unsafe.Pointer(enc), 0) = 0
|
|
|
|
enc = (*byte)(incBytePtr(unsafe.Pointer(enc), 1))
|
2018-08-11 06:00:11 +03:00
|
|
|
*indxBytePtr(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-07-29 03:56:05 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeBoolean((*C.char)(
|
|
|
|
// unsafe.Pointer(enc)), (*C.char)(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-07-29 03:13:19 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMF_EncodeString((*C.char)(unsafe.Pointer(enc)),
|
|
|
|
//(*C.char)(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-07-29 15:59:53 +03:00
|
|
|
//enc = (*byte)(unsafe.Pointer(C.AMFProp_Encode((*C.AMFObjectProperty)(
|
|
|
|
//incPtr(unsafe.Pointer(&r.Link.extras.o_props), int(unsafe.Sizeof(
|
|
|
|
//r.Link.extras.o_props)), i)), (*C.char)(unsafe.Pointer(enc)), (*C.char)(
|
|
|
|
//unsafe.Pointer(pend)))))
|
2018-08-07 05:59:02 +03:00
|
|
|
enc = C_AMFPropEncode((*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-07-27 07:36:40 +03:00
|
|
|
packet.m_nBodySize = C.uint32_t(int(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-08 08:37:54 +03:00
|
|
|
// char* AMFPropEncode(AMFOBjectProperty* prop, char* pBufer, char* pBufEnd);
|
|
|
|
// amf.c +366
|
2018-08-07 05:59:02 +03:00
|
|
|
func C_AMFPropEncode(p *C.AMFObjectProperty, pBuffer *byte, pBufEnd *byte) *byte {
|
2018-07-29 15:59:53 +03:00
|
|
|
if p.p_type == AMF_INVALID {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
if p.p_type != AMF_NULL && int(uintptr(unsafe.Pointer(pBuffer)))+
|
|
|
|
int(p.p_name.av_len)+2+1 >= int(
|
|
|
|
uintptr(unsafe.Pointer(pBufEnd))) {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
if p.p_type != AMF_NULL && p.p_name.av_len != 0 {
|
|
|
|
*indxBytePtr(unsafe.Pointer(pBuffer), 0) = byte(p.p_name.av_len >> 8)
|
|
|
|
pBuffer = (*byte)(incBytePtr(unsafe.Pointer(pBuffer), 1))
|
|
|
|
*indxBytePtr(unsafe.Pointer(pBuffer), 0) = byte(p.p_name.av_len & 0xff)
|
|
|
|
pBuffer = (*byte)(incBytePtr(unsafe.Pointer(pBuffer), 1))
|
|
|
|
memmove(unsafe.Pointer(pBuffer), unsafe.Pointer(p.p_name.av_val),
|
|
|
|
uintptr(p.p_name.av_len))
|
|
|
|
pBuffer = (*byte)(incBytePtr(unsafe.Pointer(pBuffer), int(p.p_name.av_len)))
|
|
|
|
}
|
|
|
|
|
|
|
|
switch p.p_type {
|
|
|
|
case AMF_NUMBER:
|
2018-08-08 08:37:54 +03:00
|
|
|
pBuffer = C_AMF_EncodeNumber(pBuffer, pBufEnd, float64(p.p_vu.p_number))
|
2018-07-29 15:59:53 +03:00
|
|
|
case AMF_BOOLEAN:
|
|
|
|
val := 0
|
|
|
|
if p.p_vu.p_number != 0 {
|
|
|
|
val = 1
|
|
|
|
}
|
2018-08-08 08:37:54 +03:00
|
|
|
pBuffer = C_AMF_EncodeBoolean(pBuffer, pBufEnd, val)
|
2018-07-29 15:59:53 +03:00
|
|
|
case AMF_STRING:
|
2018-08-08 08:37:54 +03:00
|
|
|
pBuffer = C_AMF_EncodeString(pBuffer, pBufEnd, &p.p_vu.p_aval)
|
2018-07-29 15:59:53 +03:00
|
|
|
case AMF_NULL:
|
|
|
|
if uintptr(incBytePtr(unsafe.Pointer(pBuffer), 1)) >= uintptr(unsafe.Pointer(
|
|
|
|
pBufEnd)) {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
*(*byte)(unsafe.Pointer(pBuffer)) = AMF_NULL
|
|
|
|
pBuffer = (*byte)(incBytePtr(unsafe.Pointer(pBuffer), 1))
|
|
|
|
case AMF_OBJECT:
|
2018-08-07 06:00:14 +03:00
|
|
|
pBuffer = C_AMFEncode(&p.p_vu.p_object, pBuffer, pBufEnd)
|
2018-07-31 00:52:40 +03:00
|
|
|
//pBuffer = (*byte)(unsafe.Pointer(C.AMF_Encode(&p.p_vu.p_object, (*C.char)(
|
|
|
|
//unsafe.Pointer(pBuffer)), (*C.char)(unsafe.Pointer(pBufEnd)))))
|
2018-07-29 15:59:53 +03:00
|
|
|
case AMF_ECMA_ARRAY:
|
2018-08-07 06:00:14 +03:00
|
|
|
pBuffer = C_AMFEncodeEcmaArray(&p.p_vu.p_object, pBuffer, pBufEnd)
|
2018-07-31 01:02:46 +03:00
|
|
|
//pBuffer = (*byte)(unsafe.Pointer(C.AMF_EncodeEcmaArray(&p.p_vu.p_object, (*C.char)(unsafe.Pointer(pBuffer)), (*C.char)(unsafe.Pointer(pBufEnd)))))
|
2018-07-29 15:59:53 +03:00
|
|
|
case AMF_STRICT_ARRAY:
|
2018-08-01 01:08:35 +03:00
|
|
|
//pBuffer = (*byte)(unsafe.Pointer(C.AMF_EncodeArray(&p.p_vu.p_object, (*C.char)(unsafe.Pointer(pBuffer)), (*C.char)(unsafe.Pointer(pBufEnd)))))
|
2018-08-07 06:00:14 +03:00
|
|
|
pBuffer = C_AMFEncodeArray(&p.p_vu.p_object, pBuffer, pBufEnd)
|
2018-07-29 15:59:53 +03:00
|
|
|
default:
|
2018-08-07 05:59:02 +03:00
|
|
|
log.Println("C_AMFPropEncode: invalid type!")
|
2018-07-29 15:59:53 +03:00
|
|
|
pBuffer = nil
|
|
|
|
}
|
|
|
|
return pBuffer
|
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// char* AMF_ENCODE(AMFObject* obj, char* pBuffer, char* pBufEnd);
|
|
|
|
// amf.c +891
|
2018-08-07 06:00:14 +03:00
|
|
|
func C_AMFEncode(obj *C.AMFObject, pBuffer *byte, pBufEnd *byte) *byte {
|
2018-07-31 00:46:10 +03:00
|
|
|
if uintptr(unsafe.Pointer(pBuffer))+uintptr(4) >= uintptr(unsafe.Pointer(pBufEnd)) {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2018-07-31 01:00:34 +03:00
|
|
|
*pBuffer = AMF_OBJECT
|
2018-07-31 00:46:10 +03:00
|
|
|
pBuffer = (*byte)(incBytePtr(unsafe.Pointer(pBuffer), 1))
|
|
|
|
|
|
|
|
for i := 0; i < int(obj.o_num); i++ {
|
2018-08-07 05:59:02 +03:00
|
|
|
res := C_AMFPropEncode((*C.AMFObjectProperty)(incPtr(unsafe.Pointer(
|
2018-07-31 00:46:10 +03:00
|
|
|
obj.o_props), i, int(unsafe.Sizeof(*obj.o_props)))), pBuffer, pBufEnd)
|
|
|
|
if res == nil {
|
2018-08-07 06:00:14 +03:00
|
|
|
log.Println("C_AMFEncode: failed to encode property in index")
|
2018-07-31 00:46:10 +03:00
|
|
|
break
|
|
|
|
} else {
|
|
|
|
pBuffer = res
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if uintptr(incBytePtr(unsafe.Pointer(pBuffer), 3)) >= uintptr(unsafe.Pointer(pBufEnd)) {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2018-08-07 06:00:14 +03:00
|
|
|
pBuffer = C_AMFEncodeInt24(pBuffer, pBufEnd, int32(AMF_OBJECT_END))
|
2018-07-31 00:46:10 +03:00
|
|
|
|
|
|
|
return pBuffer
|
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// char* AMF_EncodeEcmaArray(AMFObject* obj, char* pBuffer, char* pBufEnd);
|
|
|
|
// amf.c +924
|
2018-08-07 06:00:14 +03:00
|
|
|
func C_AMFEncodeEcmaArray(obj *C.AMFObject, pBuffer *byte, pBufEnd *byte) *byte {
|
2018-07-31 01:00:34 +03:00
|
|
|
if int(uintptr(unsafe.Pointer(pBuffer)))+4 >= int(uintptr(unsafe.Pointer(pBufEnd))) {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
*pBuffer = AMF_ECMA_ARRAY
|
|
|
|
pBuffer = (*byte)(incBytePtr(unsafe.Pointer(pBuffer), 1))
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
pBuffer = C_AMF_EncodeInt32(pBuffer, pBufEnd, int32(obj.o_num))
|
2018-07-31 01:00:34 +03:00
|
|
|
|
|
|
|
for i := 0; i < int(obj.o_num); i++ {
|
2018-08-07 05:59:02 +03:00
|
|
|
res := C_AMFPropEncode((*C.AMFObjectProperty)(incPtr(unsafe.Pointer(
|
2018-07-31 01:00:34 +03:00
|
|
|
obj.o_props), i, int(unsafe.Sizeof(*obj.o_props)))), pBuffer, pBufEnd)
|
|
|
|
if res == nil {
|
2018-08-07 06:00:14 +03:00
|
|
|
log.Println("C_AMFEncodeEcmaArray: failed to encode property!")
|
2018-07-31 01:00:34 +03:00
|
|
|
break
|
|
|
|
} else {
|
|
|
|
pBuffer = res
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if int(uintptr(unsafe.Pointer(pBuffer)))+3 >= int(uintptr(unsafe.Pointer(pBufEnd))) {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2018-08-07 06:00:14 +03:00
|
|
|
pBuffer = C_AMFEncodeInt24(pBuffer, pBufEnd, AMF_OBJECT_END)
|
2018-07-31 01:00:34 +03:00
|
|
|
|
|
|
|
return pBuffer
|
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// char* AMF_EncodeArray(AMFObject* obj, char* pBuffer, char* pBufEnd);
|
|
|
|
// amf.c +959
|
2018-08-07 06:00:14 +03:00
|
|
|
func C_AMFEncodeArray(obj *C.AMFObject, pBuffer *byte, pBufEnd *byte) *byte {
|
2018-08-01 01:08:35 +03:00
|
|
|
if int(uintptr(unsafe.Pointer(pBuffer)))+4 >= int(uintptr(unsafe.Pointer(pBufEnd))) {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
*pBuffer = AMF_STRICT_ARRAY
|
|
|
|
pBuffer = (*byte)(incBytePtr(unsafe.Pointer(pBuffer), 1))
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
pBuffer = C_AMF_EncodeInt32(pBuffer, pBufEnd, int32(obj.o_num))
|
2018-08-01 01:08:35 +03:00
|
|
|
|
|
|
|
for i := 0; i < int(obj.o_num); i++ {
|
2018-08-07 05:59:02 +03:00
|
|
|
res := C_AMFPropEncode((*C.AMFObjectProperty)(incPtr(unsafe.Pointer(
|
2018-08-01 01:08:35 +03:00
|
|
|
obj.o_props), i, int(unsafe.Sizeof(*obj.o_props)))), pBuffer, pBufEnd)
|
|
|
|
if res == nil {
|
2018-08-07 06:00:14 +03:00
|
|
|
log.Println("C_AMFEncodeEcmaArray: failed to encode property!")
|
2018-08-01 01:08:35 +03:00
|
|
|
break
|
|
|
|
} else {
|
|
|
|
pBuffer = res
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return pBuffer
|
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// int RTMP_ConnectStream(RTMP* r, int seekTime);
|
|
|
|
// rtmp.c +1099
|
2018-08-07 05:42:11 +03:00
|
|
|
func C_RTMP_ConnectStream(r *C.RTMP, seekTime int32) int {
|
2018-07-22 16:31:50 +03:00
|
|
|
var packet C.RTMPPacket
|
2018-07-28 03:45:12 +03:00
|
|
|
memset((*byte)(unsafe.Pointer(&packet)), 0, int(unsafe.Sizeof(packet)))
|
2018-07-22 16:31:50 +03:00
|
|
|
|
|
|
|
if seekTime > 0 {
|
|
|
|
r.Link.seekTime = C.int(seekTime)
|
|
|
|
}
|
|
|
|
|
|
|
|
r.m_mediaChannel = 0
|
|
|
|
|
2018-07-29 15:59:53 +03:00
|
|
|
// TODO: read packet
|
2018-08-06 09:12:16 +03:00
|
|
|
for r.m_bPlaying == 0 && C_RTMP_IsConnected(r) != 0 &&
|
2018-08-11 07:18:21 +03:00
|
|
|
//C.RTMP_ReadPacket(r, &packet) != 0 {
|
2018-08-11 06:00:11 +03:00
|
|
|
C_RTMP_ReadPacket(r, &packet) != 0 {
|
2018-07-22 16:31:50 +03:00
|
|
|
|
|
|
|
// TODO: port is ready
|
2018-08-10 09:40:08 +03:00
|
|
|
if C_RTMPPacket_IsReady(&packet) != 0 {
|
2018-07-22 16:31:50 +03:00
|
|
|
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 {
|
2018-08-07 05:42:11 +03:00
|
|
|
log.Println("C_RTMP_ConnectStream: got packet before play()! Ignoring.")
|
2018-08-11 08:04:52 +03:00
|
|
|
C_RTMPPacket_Free(&packet)
|
2018-07-22 16:31:50 +03:00
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
2018-08-12 13:07:56 +03:00
|
|
|
//C.RTMP_ClientPacket(r, &packet)
|
|
|
|
C_RTMP_ClientPacket(r, &packet)
|
2018-08-11 08:04:52 +03:00
|
|
|
C_RTMPPacket_Free(&packet)
|
2018-07-22 16:31:50 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return int(r.m_bPlaying)
|
|
|
|
}
|
|
|
|
|
2018-08-12 12:07:39 +03:00
|
|
|
// int RTMP_ClientPacket()
|
|
|
|
// rtmp.c +1226
|
2018-08-12 13:07:56 +03:00
|
|
|
// NOTE cases have been commented out that are not currently used by AusOcean
|
2018-08-12 12:07:39 +03:00
|
|
|
func C_RTMP_ClientPacket(r *C.RTMP, packet *C.RTMPPacket) int32 {
|
|
|
|
var bHasMediaPacket int32 = 0
|
|
|
|
switch packet.m_packetType {
|
2018-08-12 13:07:56 +03:00
|
|
|
|
2018-08-12 12:07:39 +03:00
|
|
|
case RTMP_PACKET_TYPE_CHUNK_SIZE:
|
2018-08-12 13:07:56 +03:00
|
|
|
log.Println("RTMP_PACKET_TYPE_CHUNK_SIZE")
|
2018-08-12 12:07:39 +03:00
|
|
|
// TODO: port this
|
2018-08-12 13:17:53 +03:00
|
|
|
//C.HandleChangeChunkSize(r, packet)
|
|
|
|
C_HandleChangeChunkSize(r, packet)
|
2018-08-12 13:07:56 +03:00
|
|
|
/*
|
|
|
|
case RTMP_PACKET_TYPE_BYTES_READ_REPORT:
|
|
|
|
log.Println("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:
|
|
|
|
log.Println("RTMP_PACKET_TYPE_CONTROL")
|
|
|
|
// TODO: port this
|
|
|
|
C.HandleCtrl(r, packet)
|
|
|
|
*/
|
2018-08-12 12:07:39 +03:00
|
|
|
case RTMP_PACKET_TYPE_SERVER_BW:
|
2018-08-12 13:07:56 +03:00
|
|
|
log.Println("RTMP_PACKET_TYPE_SERVER_BW")
|
2018-08-12 12:07:39 +03:00
|
|
|
// TODO: port this
|
2018-08-12 13:17:53 +03:00
|
|
|
//C.HandleServerBW(r, packet)
|
|
|
|
C_HandlServerBW(r, packet)
|
2018-08-12 12:07:39 +03:00
|
|
|
|
|
|
|
case RTMP_PACKET_TYPE_CLIENT_BW:
|
2018-08-12 13:07:56 +03:00
|
|
|
log.Println("RTMP_PACKET_TYPE_CLIENT_BW")
|
2018-08-12 12:07:39 +03:00
|
|
|
// TODO: port this
|
2018-08-12 13:17:53 +03:00
|
|
|
//C.HandleClientBW(r, packet)
|
2018-08-12 14:15:06 +03:00
|
|
|
C_HandleClientBW(r, packet)
|
2018-08-12 13:07:56 +03:00
|
|
|
/*
|
|
|
|
case RTMP_PACKET_TYPE_AUDIO:
|
|
|
|
log.Println("RTMP_PACKET_TYPE_AUDIO")
|
|
|
|
// TODO port this
|
|
|
|
//C.HandleAudio(r, packet) NOTE this does nothing
|
|
|
|
bHasMediaPacket = 1
|
|
|
|
if r.m_mediaChannel == 0 {
|
|
|
|
r.m_mediaChannel = packet.m_nChannel
|
|
|
|
}
|
2018-08-12 12:07:39 +03:00
|
|
|
|
2018-08-12 13:07:56 +03:00
|
|
|
if r.m_pausing == 0 {
|
|
|
|
r.m_mediaStamp = packet.m_nTimeStamp
|
|
|
|
}
|
2018-08-12 12:07:39 +03:00
|
|
|
|
2018-08-12 13:07:56 +03:00
|
|
|
case RTMP_PACKET_TYPE_VIDEO:
|
|
|
|
log.Println("RTMP_PACKET_TYPE_VIDEO:")
|
|
|
|
// TODO port this
|
|
|
|
// C.HandleVideo(r, packet) NOTE this does nothing
|
|
|
|
bHasMediaPacket = 1
|
|
|
|
if r.m_mediaChannel == 0 {
|
|
|
|
r.m_mediaChannel = packet.m_nChannel
|
|
|
|
}
|
|
|
|
if r.m_pausing == 0 {
|
|
|
|
r.m_mediaStamp = packet.m_nTimeStamp
|
|
|
|
}
|
2018-08-12 12:07:39 +03:00
|
|
|
|
2018-08-12 13:07:56 +03:00
|
|
|
case RTMP_PACKET_TYPE_FLEX_MESSAGE:
|
|
|
|
log.Println("RTMP_PACKET_TYPE_FLEX_MESSAGE:")
|
|
|
|
{
|
|
|
|
// TODO use new logger here
|
|
|
|
// RTMP_Log(RTMP_LOGDEBUG,"%s, flex message, size %u bytes, not fully supported", __FUNCTION__, packet.m_nBodySize);
|
2018-08-12 12:07:39 +03:00
|
|
|
|
2018-08-12 13:07:56 +03:00
|
|
|
if C.HandleInvoke(r, (*C.char)(incBytePtr(unsafe.Pointer(packet.m_body), 1)),
|
|
|
|
C.uint(packet.m_nBodySize-1)) == 1 {
|
|
|
|
bHasMediaPacket = 2
|
|
|
|
}
|
|
|
|
}
|
|
|
|
case RTMP_PACKET_TYPE_INFO:
|
|
|
|
log.Println(" RTMP_PACKET_TYPE_INFO:")
|
|
|
|
// TODO use new logger here
|
|
|
|
//RTMP_Log(RTMP_LOGDEBUG, "%s, received: notify %u bytes", __FUNCTION__,packet.m_nBodySize);
|
|
|
|
if C.HandleMetadata(r, packet.m_body, C.uint(packet.m_nBodySize)) != 0 {
|
|
|
|
bHasMediaPacket = 1
|
|
|
|
}
|
|
|
|
*/
|
2018-08-12 12:07:39 +03:00
|
|
|
|
|
|
|
case RTMP_PACKET_TYPE_INVOKE:
|
2018-08-12 13:07:56 +03:00
|
|
|
log.Println("RTMP_PACKET_TYPE_INVOKE:")
|
2018-08-12 12:07:39 +03:00
|
|
|
// 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 {
|
2018-08-12 14:34:14 +03:00
|
|
|
//if C_HandleInvoke(r, (*byte)(unsafe.Pointer(packet.m_body), uint32(packet.m_nBodySize)) == 1 {
|
2018-08-12 12:07:39 +03:00
|
|
|
bHasMediaPacket = 2
|
|
|
|
}
|
2018-08-12 13:07:56 +03:00
|
|
|
/*
|
|
|
|
case RTMP_PACKET_TYPE_FLASH_VIDEO:
|
|
|
|
log.Println("RTMP_PACKET_TYPE_FLASH_VIDEO:")
|
|
|
|
{
|
|
|
|
var pos uint32 = 0
|
|
|
|
var nTimeStamp uint32 = uint32(packet.m_nTimeStamp)
|
|
|
|
|
|
|
|
for pos+11 < uint32(packet.m_nBodySize) {
|
|
|
|
var dataSize uint32 = C_AMF_DecodeInt24((*byte)(incBytePtr(unsafe.Pointer(
|
|
|
|
packet.m_body), int(pos+1))))
|
|
|
|
|
|
|
|
if pos+11+dataSize+4 > uint32(packet.m_nBodySize) {
|
|
|
|
// TODO use new logger here
|
|
|
|
// RTMP_Log(RTMP_LOGWARNING, "Stream corrupt?!");
|
|
|
|
break
|
|
|
|
}
|
|
|
|
|
|
|
|
switch {
|
|
|
|
case *indxBytePtr(unsafe.Pointer(packet.m_body), int(pos)) == 0x12:
|
|
|
|
C.HandleMetadata(r, (*C.char)(incBytePtr(unsafe.Pointer(packet.m_body),
|
|
|
|
int(pos+11))), C.uint(dataSize))
|
|
|
|
case *indxBytePtr(unsafe.Pointer(packet.m_body), int(pos)) == 8 ||
|
|
|
|
*indxBytePtr(unsafe.Pointer(packet.m_body), int(pos)) == 9:
|
|
|
|
nTimeStamp = C_AMF_DecodeInt24((*byte)(incBytePtr(unsafe.Pointer(
|
|
|
|
packet.m_body), int(pos+4))))
|
|
|
|
nTimeStamp |= uint32(*indxBytePtr(unsafe.Pointer(packet.m_body),
|
|
|
|
int(pos+7)) << 24)
|
|
|
|
}
|
|
|
|
pos += (11 + dataSize + 4)
|
|
|
|
}
|
|
|
|
if r.m_pausing == 0 {
|
|
|
|
r.m_mediaStamp = C.uint32_t(nTimeStamp)
|
|
|
|
}
|
|
|
|
|
|
|
|
bHasMediaPacket = 1
|
2018-08-12 12:07:39 +03:00
|
|
|
|
|
|
|
}
|
2018-08-12 13:07:56 +03:00
|
|
|
default:
|
|
|
|
log.Println("DEFAULT")
|
|
|
|
// TODO use new logger here
|
|
|
|
// RTMP_Log(RTMP_LOGDEBUG, "%s, unknown packet type received: 0x%02x", __FUNCTION__,packet.m_packetType);
|
|
|
|
*/
|
2018-08-12 12:07:39 +03:00
|
|
|
}
|
|
|
|
return bHasMediaPacket
|
|
|
|
}
|
|
|
|
|
2018-08-12 13:17:53 +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 {
|
|
|
|
// TODO port C.AMF_DeocdeInt32
|
2018-08-12 14:15:06 +03:00
|
|
|
//r.m_inChunkSize = C.int(C.AMF_DecodeInt32((*C.char)(unsafe.Pointer(packet.m_body))))
|
|
|
|
r.m_inChunkSize = C.int(C_AMF_DecodeInt32((*byte)(unsafe.Pointer(packet.m_body))))
|
2018-08-12 13:17:53 +03:00
|
|
|
// TODO use new logger here
|
2018-08-14 06:15:03 +03:00
|
|
|
// RTMP_Log(RTMP_LOGDEBUG, "%s, received: chunk size change to %d", __FUNCTION__, r.m_inChunkSize);
|
2018-08-12 13:17:53 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// void HandleServerBW(RTMP* r, const RTMPPacket* packet);
|
|
|
|
// rtmp.c +3508
|
|
|
|
func C_HandlServerBW(r *C.RTMP, packet *C.RTMPPacket) {
|
|
|
|
// TODO port AMF_DecodeInt32
|
2018-08-12 14:15:06 +03:00
|
|
|
r.m_nServerBW = C.int(C_AMF_DecodeInt32((*byte)(unsafe.Pointer(packet.m_body))))
|
|
|
|
//r.m_nServerBW = C.int(C.AMF_DecodeInt32((*C.char)(unsafe.Pointer(packet.m_body))))
|
2018-08-12 13:17:53 +03:00
|
|
|
// TODO use new logger here
|
2018-08-14 06:15:03 +03:00
|
|
|
// RTMP_Log(RTMP_LOGDEBUG, "%s: server BW = %d", __FUNCTION__, r.m_nServerBW);
|
2018-08-12 13:17:53 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
// void HandleClientBW(RTMP* r, const RTMPPacket* packet);
|
|
|
|
// rtmp.c +3515
|
|
|
|
func C_HandleClientBW(r *C.RTMP, packet *C.RTMPPacket) {
|
2018-08-12 14:15:06 +03:00
|
|
|
r.m_nClientBW = C.int(C_AMF_DecodeInt32((*byte)(unsafe.Pointer(packet.m_body))))
|
|
|
|
//r.m_nClientBW = C.int(C.AMF_DecodeInt32((*C.char)(unsafe.Pointer(packet.m_body))))
|
2018-08-12 13:17:53 +03:00
|
|
|
|
|
|
|
if packet.m_nBodySize > 4 {
|
2018-08-12 14:15:06 +03:00
|
|
|
r.m_nClientBW2 = (C.uint8_t)(*indxBytePtr(unsafe.Pointer(packet.m_body), 4))
|
2018-08-12 13:17:53 +03:00
|
|
|
} else {
|
2018-08-12 14:15:06 +03:00
|
|
|
//r.m_nClientBW2 = -1
|
|
|
|
r.m_nClientBW2 = 0
|
2018-08-12 13:17:53 +03:00
|
|
|
}
|
|
|
|
// TODO use new logger here
|
2018-08-14 06:15:03 +03:00
|
|
|
// RTMP_Log(RTMP_LOGDEBUG, "%s: client BW = %d %d", __FUNCTION__, r.m_nClientBW,
|
|
|
|
//r.m_nClientBW2);
|
2018-08-12 13:17:53 +03:00
|
|
|
}
|
|
|
|
|
2018-08-12 14:34:14 +03:00
|
|
|
// int HandleInvoke(RTMP* r, const char* body, unsigned int nBodySize);
|
|
|
|
// rtmp.c +2912
|
2018-08-14 10:06:42 +03:00
|
|
|
// TODO port AMF_Dump
|
|
|
|
// TODO port AMFPRop_GetString
|
|
|
|
// TODO port AMF_GetProp
|
|
|
|
// TODO port AMFProp_GetNumber
|
|
|
|
// TODO port RTMP_FindFirstMatchingProperty
|
|
|
|
// TODO port DecodeTEA
|
|
|
|
// TODO port SendSecureTokenResponse
|
|
|
|
// TODO port SendRealseStream
|
|
|
|
// TODO port SendFCPublish
|
|
|
|
// TODO port RTMP_SendServerBW
|
|
|
|
// TODO port RTMP_SendCtrl
|
|
|
|
// TODO port RTMP_SendCreateStream
|
|
|
|
// TODO port SendUsherToken
|
|
|
|
// TODO port SendFCSubscribe
|
|
|
|
// TODO port SendPublish
|
|
|
|
// TODO port SendPlaylist
|
|
|
|
// TODO port SendPlay
|
|
|
|
// TODO port SendCheckBw
|
|
|
|
// TODO port SendPong
|
|
|
|
// TODO port SendCheckBWResult
|
|
|
|
// TODO port AMFProp_GetObject
|
|
|
|
// TODO port AV_erase
|
2018-08-14 11:52:37 +03:00
|
|
|
// TODO port AMF_Decode
|
2018-08-12 14:34:14 +03:00
|
|
|
func C_HandleInvoke(r *C.RTMP, body *byte, nBodySize uint32) int32 {
|
2018-08-14 06:15:03 +03:00
|
|
|
var obj C.AMFObject
|
|
|
|
var method C.AVal
|
|
|
|
var txn float32
|
|
|
|
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 06:03:52 +03:00
|
|
|
|
2018-08-14 11:52:37 +03:00
|
|
|
nRes = int32(C.AMF_Decode(&obj, (*C.char)(unsafe.Pointer(body)), C.int(nBodySize), 0))
|
|
|
|
// nRes = C_AMF_Decode(&obj, body, nBodySize, 0)
|
|
|
|
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 11:52:37 +03:00
|
|
|
C.AMF_Dump(&obj)
|
|
|
|
C.AMFProp_GetString(C.AMF_GetProp(&obj, nil, 0), &method)
|
|
|
|
txn = float32(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 06:22:40 +03:00
|
|
|
var methodInvoked C.AVal
|
2018-08-14 11:52:37 +03:00
|
|
|
var i int32
|
|
|
|
for i = 0; i < int32(r.m_numCalls); i++ {
|
|
|
|
if float32((*(*C.RTMP_METHOD)(incPtr(unsafe.Pointer(r.m_methodCalls), int(i),
|
|
|
|
int(unsafe.Sizeof(*r.m_methodCalls))))).num) == txn {
|
|
|
|
methodInvoked = (*(*C.RTMP_METHOD)(incPtr(unsafe.Pointer(r.m_methodCalls),
|
|
|
|
int(i), int(unsafe.Sizeof(*r.m_methodCalls))))).name
|
|
|
|
C.AV_erase(r.m_methodCalls, &r.m_numCalls, C.int(i), 0)
|
|
|
|
break
|
2018-08-13 05:45:43 +03:00
|
|
|
}
|
2018-08-14 11:52:37 +03:00
|
|
|
}
|
|
|
|
if methodInvoked.av_val == nil {
|
2018-08-14 06:22:40 +03:00
|
|
|
// TODO use new logger here
|
|
|
|
//RTMP_Log(RTMP_LOGDEBUG, "%s, received result id %f without matching request",
|
2018-08-14 11:52:37 +03:00
|
|
|
//__FUNCTION__, txn);
|
|
|
|
goto leave
|
|
|
|
}
|
2018-08-14 06:22:40 +03:00
|
|
|
// TODO use new logger here
|
2018-08-14 11:52:37 +03:00
|
|
|
//RTMP_Log(RTMP_LOGDEBUG, "%s, received result for method call <%s>", __FUNCTION__,
|
|
|
|
//methodInvoked.av_val);
|
2018-08-14 06:03:52 +03:00
|
|
|
switch {
|
2018-08-14 11:52:37 +03:00
|
|
|
case C_AVMATCH(&methodInvoked, &av_connect) != 0:
|
|
|
|
if r.Link.token.av_len != 0 {
|
2018-08-14 06:22:40 +03:00
|
|
|
var p C.AMFObjectProperty
|
2018-08-14 11:52:37 +03:00
|
|
|
if C.RTMP_FindFirstMatchingProperty(&obj, &av_secureToken, &p) != 0 {
|
|
|
|
C.DecodeTEA(&r.Link.token, &p.p_vu.p_aval)
|
|
|
|
C.SendSecureTokenResponse(r, &p.p_vu.p_aval)
|
2018-08-14 06:03:52 +03:00
|
|
|
}
|
2018-08-14 11:52:37 +03:00
|
|
|
}
|
|
|
|
if (r.Link.protocol & RTMP_FEATURE_WRITE) != 0 {
|
|
|
|
C.SendReleaseStream(r)
|
|
|
|
C.SendFCPublish(r)
|
|
|
|
} else {
|
|
|
|
C.RTMP_SendServerBW(r)
|
|
|
|
C.RTMP_SendCtrl(r, 3, 0, 300)
|
|
|
|
}
|
|
|
|
C.RTMP_SendCreateStream(r)
|
|
|
|
|
|
|
|
if (r.Link.protocol & RTMP_FEATURE_WRITE) == 0 {
|
|
|
|
/* Authenticate on Justin.tv legacy servers before sending FCSubscribe */
|
|
|
|
if r.Link.usherToken.av_len != 0 {
|
|
|
|
C.SendUsherToken(r, &r.Link.usherToken)
|
2018-08-14 06:03:52 +03:00
|
|
|
}
|
|
|
|
/* Send the FCSubscribe if live stream or if subscribepath is set */
|
2018-08-14 11:52:37 +03:00
|
|
|
switch {
|
|
|
|
case r.Link.subscribepath.av_len != 0:
|
|
|
|
C.SendFCSubscribe(r, &r.Link.subscribepath)
|
|
|
|
case (r.Link.lFlags & RTMP_LF_LIVE) != 0:
|
|
|
|
C.SendFCSubscribe(r, &r.Link.playpath)
|
2018-08-14 06:03:52 +03:00
|
|
|
}
|
2018-08-14 11:52:37 +03:00
|
|
|
}
|
|
|
|
case C_AVMATCH(&methodInvoked, &av_createStream) != 0:
|
|
|
|
r.m_stream_id = C.int(C.AMFProp_GetNumber(C.AMF_GetProp(&obj, nil, 3)))
|
|
|
|
|
|
|
|
if (r.Link.protocol & RTMP_FEATURE_WRITE) != 0 {
|
|
|
|
C.SendPublish(r)
|
|
|
|
} else {
|
|
|
|
if (r.Link.lFlags & RTMP_LF_PLST) != 0 {
|
|
|
|
C.SendPlaylist(r)
|
2018-08-14 06:03:52 +03:00
|
|
|
}
|
2018-08-14 11:52:37 +03:00
|
|
|
C.SendPlay(r)
|
|
|
|
C.RTMP_SendCtrl(r, 3, C.uint(r.m_stream_id), C.uint(r.m_nBufferMS))
|
|
|
|
}
|
|
|
|
case C_AVMATCH(&methodInvoked, &av_play) != 0 ||
|
|
|
|
C_AVMATCH(&methodInvoked, &av_publish) != 0:
|
|
|
|
r.m_bPlaying = 1
|
2018-08-13 05:45:43 +03:00
|
|
|
}
|
2018-08-14 11:52:37 +03:00
|
|
|
//free(methodInvoked.av_val)
|
|
|
|
case C_AVMATCH(&method, &av_onBWDone) != 0:
|
|
|
|
if r.m_nBWCheckCounter == 0 {
|
|
|
|
C.SendCheckBW(r)
|
2018-08-13 05:45:43 +03:00
|
|
|
}
|
2018-08-14 11:52:37 +03:00
|
|
|
case C_AVMATCH(&method, &av_onFCSubscribe) != 0:
|
|
|
|
/* SendOnFCSubscribe(); */
|
|
|
|
case C_AVMATCH(&method, &av_onFCUnsubscribe) != 0:
|
|
|
|
C.RTMP_Close(r)
|
|
|
|
ret = 1
|
|
|
|
case C_AVMATCH(&method, &av_ping) != 0:
|
|
|
|
C.SendPong(r, C.double(txn))
|
|
|
|
case C_AVMATCH(&method, &av__onbwcheck) != 0:
|
|
|
|
C.SendCheckBWResult(r, C.double(txn))
|
|
|
|
case C_AVMATCH(&method, &av__onbwdone) != 0:
|
|
|
|
var i int32
|
|
|
|
for i = 0; i < int32(r.m_numCalls); i++ {
|
|
|
|
if C_AVMATCH(&(*(*C.RTMP_METHOD)(incPtr(unsafe.Pointer(r.m_methodCalls), int(i),
|
|
|
|
int(unsafe.Sizeof(*r.m_methodCalls))))).name, &av__checkbw) != 0 {
|
|
|
|
C.AV_erase(r.m_methodCalls, &r.m_numCalls, C.int(i), 1)
|
|
|
|
break
|
|
|
|
}
|
2018-08-14 06:03:52 +03:00
|
|
|
}
|
2018-08-14 11:52:37 +03:00
|
|
|
case C_AVMATCH(&method, &av_close) != 0:
|
2018-08-14 10:06:42 +03:00
|
|
|
// TODO use new logger
|
|
|
|
//RTMP_Log(RTMP_LOGERROR, "rtmp server requested close");
|
2018-08-14 11:52:37 +03:00
|
|
|
C.RTMP_Close(r)
|
|
|
|
case C_AVMATCH(&method, &av_onStatus) != 0:
|
2018-08-14 10:06:42 +03:00
|
|
|
var obj2 C.AMFObject
|
|
|
|
var code, level C.AVal
|
2018-08-14 11:52:37 +03:00
|
|
|
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)
|
2018-08-14 06:03:52 +03:00
|
|
|
|
2018-08-14 10:06:42 +03:00
|
|
|
// TODO use new logger
|
2018-08-14 11:52:37 +03:00
|
|
|
// RTMP_Log(RTMP_LOGDEBUG, "%s, onStatus: %s", __FUNCTION__, code.av_val);
|
2018-08-13 05:45:43 +03:00
|
|
|
switch {
|
2018-08-14 11:52:37 +03:00
|
|
|
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:
|
|
|
|
r.m_stream_id = -1
|
|
|
|
C.RTMP_Close(r)
|
2018-08-14 10:06:42 +03:00
|
|
|
// TODO use new logger
|
|
|
|
// RTMP_Log(RTMP_LOGERROR, "Closing connection: %s", code.av_val);
|
2018-08-14 11:52:37 +03:00
|
|
|
case C_AVMATCH(&code, &av_NetStream_Play_Start) != 0 ||
|
|
|
|
C_AVMATCH(&code, &av_NetStream_Play_PublishNotify) != 0:
|
|
|
|
var i int32
|
|
|
|
r.m_bPlaying = 1
|
|
|
|
for i = 0; i < int32(r.m_numCalls); i++ {
|
|
|
|
if C_AVMATCH(&(*(*C.RTMP_METHOD)(incPtr(unsafe.Pointer(r.m_methodCalls), int(i),
|
|
|
|
int(unsafe.Sizeof(*r.m_methodCalls))))).name, &av_play) != 0 {
|
|
|
|
C.AV_erase(r.m_methodCalls, &r.m_numCalls, C.int(i), 1)
|
|
|
|
break
|
2018-08-13 05:45:43 +03:00
|
|
|
}
|
2018-08-14 11:52:37 +03:00
|
|
|
}
|
|
|
|
case C_AVMATCH(&code, &av_NetStream_Publish_Start) != 0:
|
|
|
|
var i int32
|
|
|
|
r.m_bPlaying = 1
|
|
|
|
for i = 0; i < int32(r.m_numCalls); i++ {
|
|
|
|
if C_AVMATCH(&(*(*C.RTMP_METHOD)(incPtr(unsafe.Pointer(r.m_methodCalls), int(i),
|
|
|
|
int(unsafe.Sizeof(*r.m_methodCalls))))).name, &av_publish) != 0 {
|
|
|
|
C.AV_erase(r.m_methodCalls, &r.m_numCalls, C.int(i), 1)
|
|
|
|
break
|
2018-08-13 05:45:43 +03:00
|
|
|
}
|
2018-08-14 11:52:37 +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:
|
|
|
|
C.RTMP_Close(r)
|
|
|
|
ret = 1
|
|
|
|
case C_AVMATCH(&code, &av_NetStream_Seek_Notify) != 0:
|
|
|
|
// NOTE ~ has been replace by ^ - is this correct ?
|
|
|
|
r.m_read.flags = C.uint8_t(int8(r.m_read.flags) & ^RTMP_READ_SEEKING)
|
|
|
|
case C_AVMATCH(&code, &av_NetStream_Pause_Notify) != 0:
|
|
|
|
if r.m_pausing == 1 || r.m_pausing == 2 {
|
|
|
|
C.RTMP_SendPause(r, 0, C.int(r.m_pauseStamp))
|
|
|
|
r.m_pausing = 3
|
|
|
|
}
|
2018-08-13 05:45:43 +03:00
|
|
|
}
|
2018-08-14 11:52:37 +03:00
|
|
|
case C_AVMATCH(&method, &av_playlist_ready) != 0:
|
|
|
|
var i int32
|
|
|
|
for i = 0; i < int32(r.m_numCalls); i++ {
|
|
|
|
if C_AVMATCH(&(*(*C.RTMP_METHOD)(incPtr(unsafe.Pointer(r.m_methodCalls), int(i),
|
|
|
|
int(unsafe.Sizeof(*r.m_methodCalls))))).name, &av_set_playlist) != 0 {
|
|
|
|
C.AV_erase(r.m_methodCalls, &r.m_numCalls, C.int(i), 1)
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
default:
|
2018-08-14 06:03:52 +03:00
|
|
|
}
|
|
|
|
leave:
|
2018-08-14 11:52:37 +03:00
|
|
|
C.AMF_Reset(&obj)
|
|
|
|
return ret
|
2018-08-12 14:34:14 +03:00
|
|
|
}
|
2018-08-14 10:06:42 +03:00
|
|
|
|
2018-08-14 06:15:03 +03:00
|
|
|
// #define AVMATCH(a1,a2)
|
|
|
|
// amf.h +63
|
2018-08-14 11:52:37 +03:00
|
|
|
func C_AVMATCH(a1, a2 *C.AVal) int32 {
|
|
|
|
if a1.av_len == a2.av_len && memcmp(unsafe.Pointer(a1.av_val),
|
|
|
|
unsafe.Pointer(a2.av_val), int(a1.av_len)) != 0 {
|
2018-08-14 06:15:03 +03:00
|
|
|
return 1
|
|
|
|
} else {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
}
|
2018-08-13 05:45:43 +03:00
|
|
|
|
2018-08-11 08:04:52 +03:00
|
|
|
// 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-08 08:37:54 +03:00
|
|
|
// int RTMP_ReadPacket(RTMP* r, RTMPPacket* packet);
|
|
|
|
// rtmp.c +3550
|
2018-08-07 06:01:30 +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-06 07:30:37 +03:00
|
|
|
packet.m_headerType = C.uint8_t((hbuf[0] & 0xc0) >> 6)
|
|
|
|
packet.m_nChannel = C.int(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-06 07:30:37 +03:00
|
|
|
packet.m_nChannel = C.int(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-02 15:39:05 +03:00
|
|
|
packet.m_nChannel = C.int(tmp + 64)
|
|
|
|
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)
|
|
|
|
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-06 07:30:37 +03:00
|
|
|
var packetPtr *C.RTMPPacket
|
|
|
|
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-06 07:30:37 +03:00
|
|
|
C.free(unsafe.Pointer(r.m_channelTimestamp))
|
2018-08-02 15:39:05 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
if packets == nil {
|
2018-08-06 07:30:37 +03:00
|
|
|
C.free(unsafe.Pointer(r.m_vecChannelsIn))
|
2018-08-02 15:39:05 +03:00
|
|
|
}
|
|
|
|
|
2018-08-06 07:30:37 +03:00
|
|
|
r.m_channelTimestamp = (*C.int)(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))))
|
|
|
|
r.m_channelsAllocatedIn = C.int(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-06 07:30:37 +03:00
|
|
|
var tmpPacketPtr *C.RTMPPacket
|
|
|
|
if *(**C.RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsIn),
|
|
|
|
int(packet.m_nChannel), int(unsafe.Sizeof(tmpPacketPtr)))) != nil {
|
|
|
|
var tmpPacket C.RTMPPacket
|
2018-08-02 15:39:05 +03:00
|
|
|
memmove(unsafe.Pointer(packet), unsafe.Pointer(
|
2018-08-06 07:30:37 +03:00
|
|
|
*(**C.RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsIn),
|
|
|
|
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-08 08:37:54 +03:00
|
|
|
packet.m_nTimeStamp = C.uint32_t(C_AMF_DecodeInt24(header))
|
2018-08-02 15:39:05 +03:00
|
|
|
|
|
|
|
if nSize >= 6 {
|
2018-08-08 08:37:54 +03:00
|
|
|
packet.m_nBodySize = C.uint32_t(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-06 07:30:37 +03:00
|
|
|
packet.m_packetType = C.uint8_t(*indxBytePtr(unsafe.Pointer(header), 6))
|
2018-08-02 15:39:05 +03:00
|
|
|
|
|
|
|
if nSize == 11 {
|
|
|
|
// TODO: port this
|
2018-08-11 07:24:25 +03:00
|
|
|
packet.m_nInfoField2 = C.int32_t(C_DecodeInt32LE((*byte)(incBytePtr(
|
2018-08-06 07:30:37 +03:00
|
|
|
unsafe.Pointer(header), 7))))
|
2018-08-11 07:24:25 +03:00
|
|
|
//packet.m_nInfoField2 = C.int32_t(C.DecodeInt32LE((*C.char)(incBytePtr(
|
|
|
|
//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-06 07:30:37 +03:00
|
|
|
packet.m_nTimeStamp = C.uint32_t(C.AMF_DecodeInt32((*C.char)(incBytePtr(
|
|
|
|
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-06 07:30:37 +03:00
|
|
|
packet.m_headerType = C.uint8_t((hbuf[0] & 0xc0) >> 6)
|
|
|
|
}
|
|
|
|
|
|
|
|
nToRead = int32(packet.m_nBodySize - packet.m_nBytesRead)
|
|
|
|
nChunk = int32(r.m_inChunkSize)
|
|
|
|
|
|
|
|
if nToRead < nChunk {
|
|
|
|
nChunk = nToRead
|
|
|
|
}
|
|
|
|
|
|
|
|
if packet.m_chunk != nil {
|
|
|
|
packet.m_chunk.c_headerSize = C.int(hSize)
|
|
|
|
memmove(unsafe.Pointer(&packet.m_chunk.c_header[0]), unsafe.Pointer(&hbuf[0]),
|
|
|
|
uintptr(hSize))
|
|
|
|
packet.m_chunk.c_chunk = (*C.char)(incBytePtr(unsafe.Pointer(packet.m_body),
|
|
|
|
int(packet.m_nBytesRead)))
|
|
|
|
packet.m_chunk.c_chunkSize = C.int(nChunk)
|
|
|
|
}
|
|
|
|
|
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
|
|
|
}
|
|
|
|
|
|
|
|
packet.m_nBytesRead += C.uint32_t(nChunk)
|
|
|
|
|
|
|
|
var tmpPktPtr *C.RTMPPacket
|
|
|
|
// keep the packet as ref for other packets on this channel
|
|
|
|
if *(**C.RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsIn),
|
|
|
|
int(packet.m_nChannel), int(unsafe.Sizeof(tmpPktPtr)))) == nil {
|
2018-08-11 07:18:21 +03:00
|
|
|
|
2018-08-06 07:30:37 +03:00
|
|
|
var tmpPkt C.RTMPPacket
|
|
|
|
*(**C.RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsIn), int(packet.m_nChannel),
|
2018-08-11 07:18:21 +03:00
|
|
|
int(unsafe.Sizeof(tmpPktPtr)))) = (*C.RTMPPacket)(C.malloc(C.size_t(
|
|
|
|
unsafe.Sizeof(tmpPkt))))
|
2018-08-02 15:39:05 +03:00
|
|
|
}
|
2018-08-06 07:30:37 +03:00
|
|
|
memmove(unsafe.Pointer(*(**C.RTMPPacket)(incPtr(unsafe.Pointer(
|
|
|
|
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 {
|
|
|
|
(*(**C.RTMPPacket)(incPtr(unsafe.Pointer(
|
|
|
|
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-10 09:40:08 +03:00
|
|
|
packet.m_nTimeStamp += *(*C.uint32_t)(incPtr(unsafe.Pointer(r.m_channelTimestamp),
|
|
|
|
int(packet.m_nChannel), 4))
|
|
|
|
}
|
|
|
|
*(*C.uint32_t)(incPtr(unsafe.Pointer(r.m_channelTimestamp), int(packet.m_nChannel),
|
|
|
|
4)) = packet.m_nTimeStamp
|
|
|
|
|
|
|
|
(*(**C.RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsIn), int(packet.m_nChannel),
|
|
|
|
int(unsafe.Sizeof(*r.m_vecChannelsIn))))).m_body = nil
|
|
|
|
(*(**C.RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsIn), int(packet.m_nChannel),
|
|
|
|
int(unsafe.Sizeof(*r.m_vecChannelsIn))))).m_nBytesRead = 0
|
|
|
|
(*(**C.RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsIn), int(packet.m_nChannel),
|
|
|
|
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-11 07:18:21 +03:00
|
|
|
// static int DecodeInt32LE(const char* data);
|
2018-08-11 07:22:32 +03:00
|
|
|
// rtmp.c +3527
|
2018-08-11 07:29:55 +03:00
|
|
|
func C_DecodeInt32LE(data *byte) int32 {
|
2018-08-11 07:22:32 +03:00
|
|
|
var c *uint8 = (*uint8)(data)
|
2018-08-11 07:29:55 +03:00
|
|
|
return int32((*indxBytePtr(unsafe.Pointer(c), 3) << 24) |
|
2018-08-11 07:22:32 +03:00
|
|
|
(*indxBytePtr(unsafe.Pointer(c), 2) << 16) |
|
|
|
|
(*indxBytePtr(unsafe.Pointer(c), 1) << 8) |
|
|
|
|
*c)
|
|
|
|
}
|
2018-08-11 07:18:21 +03:00
|
|
|
|
2018-08-12 14:15:06 +03:00
|
|
|
// unsigned int AMF_DeocdeInt32(const char* data);
|
|
|
|
// amf.c +59
|
|
|
|
func C_AMF_DecodeInt32(data *byte) uint32 {
|
|
|
|
c := (*uint8)(data)
|
|
|
|
val := uint32(
|
|
|
|
int32(*c)<<24 |
|
|
|
|
int32(*(*uint8)(incBytePtr(unsafe.Pointer(c), 1)))<<16 |
|
|
|
|
int32(*(*uint8)(incBytePtr(unsafe.Pointer(c), 2)))<<8 |
|
|
|
|
int32(*(*uint8)(incBytePtr(unsafe.Pointer(c), 3))))
|
|
|
|
return uint32(val)
|
|
|
|
}
|
|
|
|
|
2018-08-11 07:29:55 +03:00
|
|
|
// int EncodeInt32LE(char* output, int nVal);
|
|
|
|
// rtmp.c +3537
|
|
|
|
func C_EncodeInt32LE(output *byte, nVal int32) int32 {
|
|
|
|
*output = byte(nVal)
|
|
|
|
nVal >>= 8
|
|
|
|
*indxBytePtr(unsafe.Pointer(output), 1) = byte(nVal)
|
|
|
|
nVal >>= 8
|
|
|
|
*indxBytePtr(unsafe.Pointer(output), 2) = byte(nVal)
|
|
|
|
nVal >>= 8
|
|
|
|
*indxBytePtr(unsafe.Pointer(output), 3) = byte(nVal)
|
|
|
|
return 4
|
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// #define RTMPPacket_IsReady(a)
|
|
|
|
// rtmp.h +142
|
2018-08-10 09:40:08 +03:00
|
|
|
func C_RTMPPacket_IsReady(p *C.RTMPPacket) int {
|
2018-07-22 16:31:50 +03:00
|
|
|
if p.m_nBytesRead == p.m_nBodySize {
|
|
|
|
return 1
|
|
|
|
}
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2018-07-19 09:37:31 +03:00
|
|
|
func endSession(rtmp *C.RTMP) uint32 {
|
|
|
|
if rtmp == nil {
|
|
|
|
return 3
|
|
|
|
}
|
|
|
|
|
2018-07-22 11:48:46 +03:00
|
|
|
//C.RTMP_Close(rtmp)
|
|
|
|
//C.RTMP_Free(rtmp)
|
2018-07-19 09:37:31 +03:00
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// int RTMP_Write(RTMP* r, const char* buf, int size);
|
|
|
|
// rtmp.c +5095
|
2018-08-07 06:02:23 +03:00
|
|
|
func C_RTMP_Write(r *C.RTMP, data []byte) int {
|
2018-07-14 08:19:21 +03:00
|
|
|
buf := sliceToPtr(data)
|
2018-07-14 08:22:54 +03:00
|
|
|
// TODO: port RTMPPacket
|
2018-07-12 20:03:06 +03:00
|
|
|
var pkt = &r.m_write
|
2018-07-13 18:43:17 +03:00
|
|
|
var pend, enc unsafe.Pointer
|
2018-07-12 17:17:11 +03:00
|
|
|
size := len(data)
|
2018-07-10 12:15:34 +03:00
|
|
|
s2 := size
|
2018-07-12 17:17:11 +03:00
|
|
|
var ret, num int
|
|
|
|
|
2018-07-05 09:55:46 +03:00
|
|
|
pkt.m_nChannel = 0x04
|
2018-07-16 18:38:31 +03:00
|
|
|
pkt.m_nInfoField2 = C.int32_t(r.m_stream_id)
|
2018-07-13 18:43:17 +03:00
|
|
|
for s2 != 0 {
|
2018-07-10 12:15:34 +03:00
|
|
|
if pkt.m_nBytesRead == 0 {
|
2018-07-11 07:46:19 +03:00
|
|
|
if size < minDataSize {
|
2018-07-05 09:55:46 +03:00
|
|
|
log.Printf("size: %d\n", size)
|
|
|
|
log.Printf("too small \n")
|
|
|
|
return 0
|
|
|
|
}
|
2018-07-14 08:31:06 +03:00
|
|
|
|
2018-07-18 05:26:32 +03:00
|
|
|
if *indxBytePtr(buf, 0) == 'F' && *indxBytePtr(buf, 1) == 'L' && *indxBytePtr(buf, 2) == 'V' {
|
2018-07-14 07:54:10 +03:00
|
|
|
buf = unsafe.Pointer(uintptr(buf) + uintptr(13))
|
2018-07-05 09:55:46 +03:00
|
|
|
s2 -= 13
|
|
|
|
}
|
|
|
|
|
2018-07-18 05:26:32 +03:00
|
|
|
pkt.m_packetType = C.uint8_t(*indxBytePtr(buf, 0))
|
2018-07-16 06:30:58 +03:00
|
|
|
buf = incBytePtr(buf, 1)
|
2018-08-08 08:37:54 +03:00
|
|
|
pkt.m_nBodySize = C.uint32_t(C_AMF_DecodeInt24((*byte)(buf)))
|
2018-07-16 06:30:58 +03:00
|
|
|
buf = incBytePtr(buf, 3)
|
2018-08-08 08:37:54 +03:00
|
|
|
pkt.m_nTimeStamp = C.uint32_t(C_AMF_DecodeInt24((*byte)(buf)))
|
2018-07-16 06:30:58 +03:00
|
|
|
buf = incBytePtr(buf, 3)
|
2018-07-18 05:26:32 +03:00
|
|
|
pkt.m_nTimeStamp |= C.uint32_t(*indxBytePtr(buf, 0)) << 24
|
2018-07-16 06:30:58 +03:00
|
|
|
buf = incBytePtr(buf, 4)
|
2018-07-05 09:55:46 +03:00
|
|
|
s2 -= 11
|
|
|
|
|
2018-07-12 21:07:57 +03:00
|
|
|
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 {
|
2018-07-05 09:55:46 +03:00
|
|
|
|
2018-07-12 21:07:57 +03:00
|
|
|
pkt.m_headerType = RTMP_PACKET_SIZE_LARGE
|
2018-07-11 20:20:00 +03:00
|
|
|
|
2018-07-12 21:07:57 +03:00
|
|
|
if pkt.m_packetType == RTMP_PACKET_TYPE_INFO {
|
2018-07-11 18:20:18 +03:00
|
|
|
pkt.m_nBodySize += 16
|
|
|
|
}
|
2018-07-05 09:55:46 +03:00
|
|
|
} else {
|
2018-07-12 21:07:57 +03:00
|
|
|
pkt.m_headerType = RTMP_PACKET_SIZE_MEDIUM
|
2018-07-05 09:55:46 +03:00
|
|
|
}
|
2018-07-14 08:22:54 +03:00
|
|
|
// TODO: Port this
|
2018-08-11 07:18:21 +03:00
|
|
|
if int(C_RTMPPacket_Alloc(pkt, uint32(pkt.m_nBodySize))) == 0 {
|
2018-07-12 17:17:11 +03:00
|
|
|
log.Println("Failed to allocate packet")
|
2018-07-05 09:55:46 +03:00
|
|
|
return 0
|
|
|
|
}
|
2018-06-27 21:02:16 +03:00
|
|
|
|
2018-07-13 18:43:17 +03:00
|
|
|
enc = unsafe.Pointer(pkt.m_body)
|
2018-07-16 06:30:58 +03:00
|
|
|
pend = incBytePtr(enc, int(pkt.m_nBodySize))
|
2018-07-10 13:36:49 +03:00
|
|
|
|
2018-07-12 21:07:57 +03:00
|
|
|
if pkt.m_packetType == RTMP_PACKET_TYPE_INFO {
|
2018-08-08 08:37:54 +03:00
|
|
|
enc = unsafe.Pointer(C_AMF_EncodeString((*byte)(enc), (*byte)(pend), &setDataFrame))
|
2018-07-16 18:38:31 +03:00
|
|
|
pkt.m_nBytesRead = C.uint32_t(math.Abs(float64(uintptr(enc) -
|
2018-07-12 21:07:57 +03:00
|
|
|
uintptr(unsafe.Pointer(pkt.m_body)))))
|
2018-07-05 09:55:46 +03:00
|
|
|
}
|
2018-07-12 22:51:17 +03:00
|
|
|
|
2018-07-05 09:55:46 +03:00
|
|
|
} else {
|
2018-07-16 06:30:58 +03:00
|
|
|
enc = incBytePtr(unsafe.Pointer(pkt.m_body), int(pkt.m_nBytesRead))
|
2018-07-05 09:55:46 +03:00
|
|
|
}
|
2018-07-12 17:17:11 +03:00
|
|
|
num = int(pkt.m_nBodySize - pkt.m_nBytesRead)
|
2018-07-05 09:55:46 +03:00
|
|
|
if num > s2 {
|
|
|
|
num = s2
|
|
|
|
}
|
2018-07-14 15:40:01 +03:00
|
|
|
|
2018-07-25 04:26:59 +03:00
|
|
|
memmove(enc, buf, uintptr(num))
|
2018-07-16 18:38:31 +03:00
|
|
|
pkt.m_nBytesRead += C.uint32_t(num)
|
2018-07-05 09:55:46 +03:00
|
|
|
s2 -= num
|
2018-07-16 06:30:58 +03:00
|
|
|
buf = incBytePtr(buf, num)
|
2018-07-05 09:55:46 +03:00
|
|
|
if pkt.m_nBytesRead == pkt.m_nBodySize {
|
2018-07-14 08:22:54 +03:00
|
|
|
// TODO: Port this
|
2018-08-07 06:03:40 +03:00
|
|
|
ret = C_RTMP_SendPacket(r, pkt, 0)
|
2018-07-14 08:29:00 +03:00
|
|
|
// TODO: Port this
|
2018-07-10 13:36:49 +03:00
|
|
|
C.RTMPPacket_Free(pkt)
|
2018-07-05 09:55:46 +03:00
|
|
|
pkt.m_nBytesRead = 0
|
2018-07-11 07:56:07 +03:00
|
|
|
if ret == 0 {
|
2018-07-05 09:55:46 +03:00
|
|
|
return -1
|
|
|
|
}
|
2018-07-16 06:30:58 +03:00
|
|
|
buf = incBytePtr(buf, 4)
|
2018-07-05 09:55:46 +03:00
|
|
|
s2 -= 4
|
|
|
|
if s2 < 0 {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2018-07-11 18:20:18 +03:00
|
|
|
return size + s2
|
2018-06-27 21:02:16 +03:00
|
|
|
}
|
2018-07-05 09:55:46 +03:00
|
|
|
|
2018-08-11 07:18:21 +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)(C.calloc(1, C.size_t(nSize+RTMP_MAX_HEADER_SIZE)))
|
|
|
|
|
|
|
|
if ptr == nil {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
p.m_body = (*C.char)(incBytePtr(unsafe.Pointer(ptr), RTMP_MAX_HEADER_SIZE))
|
|
|
|
p.m_nBytesRead = 0
|
|
|
|
return 1
|
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// int RTMP_SendPacket(RTMP* r, RTMPPacket* packet, int queue);
|
|
|
|
// rtmp.c +3896
|
2018-08-07 06:03:40 +03:00
|
|
|
func C_RTMP_SendPacket(r *C.RTMP, packet *C.RTMPPacket, queue int) int {
|
2018-07-15 10:48:24 +03:00
|
|
|
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-07-13 19:38:59 +03:00
|
|
|
var t int32
|
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-07-28 16:02:59 +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),
|
|
|
|
//int(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-07-28 16:02:59 +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-07-16 08:07:26 +03:00
|
|
|
r.m_vecChannelsOut = (**C.RTMPPacket)(packets)
|
2018-07-28 15:11:42 +03:00
|
|
|
|
2018-07-16 08:07:26 +03:00
|
|
|
C.memset(incPtr(unsafe.Pointer(r.m_vecChannelsOut), int(r.m_channelsAllocatedOut),
|
2018-07-18 05:26:32 +03:00
|
|
|
int(unsafe.Sizeof(packet))), 0, C.size_t(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-07-16 08:07:26 +03:00
|
|
|
r.m_channelsAllocatedOut = C.int(n)
|
2018-07-14 09:06:58 +03:00
|
|
|
}
|
2018-07-16 08:07:26 +03:00
|
|
|
prevPacket = *(**C.RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsOut),
|
|
|
|
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-07-16 08:07:26 +03:00
|
|
|
t = int32(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-07 06:00:14 +03:00
|
|
|
hptr = unsafe.Pointer(C_AMFEncodeInt24((*byte)(hptr), (*byte)(hend), res))
|
2018-07-15 09:51:20 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
if nSize > 4 {
|
2018-08-07 06:00:14 +03:00
|
|
|
hptr = unsafe.Pointer(C_AMFEncodeInt24((*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))))
|
|
|
|
//hptr = incBytePtr(hptr, int(C.EncodeInt32LE((*C.char)(hptr),
|
|
|
|
//C.int(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))
|
|
|
|
copy(ptrToSlice(toff, int(nChunkSize+hSize)), ptrToSlice(header,
|
|
|
|
int(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-07-18 05:26:32 +03:00
|
|
|
*indxBytePtr(header, 1) = byte(tmp & 0xff)
|
2018-07-15 09:51:20 +03:00
|
|
|
|
2018-07-15 10:18:38 +03:00
|
|
|
if cSize == 2 {
|
2018-07-18 05:26:32 +03:00
|
|
|
*indxBytePtr(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-07-28 07:57:30 +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-07-15 10:18:38 +03:00
|
|
|
// TODO: port C.AVal
|
2018-07-15 10:48:24 +03:00
|
|
|
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-07-29 04:06:55 +03:00
|
|
|
//C.AMF_DecodeString((*C.char)(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-07-29 04:27:06 +03:00
|
|
|
//txn = int(C.AMF_DecodeNumber((*C.char)(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-07-29 04:33:34 +03:00
|
|
|
//C.AV_queue(&r.m_methodCalls, (*C.int)(unsafe.Pointer(&r.m_numCalls)), &method,
|
|
|
|
//C.int(txn))
|
2018-07-15 10:18:38 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-07-17 11:30:24 +03:00
|
|
|
if *(**C.RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsOut),
|
|
|
|
int(packet.m_nChannel), int(unsafe.Sizeof(packet)))) == nil {
|
2018-07-16 12:20:17 +03:00
|
|
|
|
2018-07-18 05:26:32 +03:00
|
|
|
*(**C.RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsOut),
|
|
|
|
int(packet.m_nChannel), int(unsafe.Sizeof(packet)))) =
|
2018-08-11 07:18:21 +03:00
|
|
|
(*C.RTMPPacket)(C.malloc(C.size_t(unsafe.Sizeof(*packet))))
|
2018-07-15 09:51:20 +03:00
|
|
|
}
|
2018-07-14 09:06:58 +03:00
|
|
|
|
2018-07-19 10:29:03 +03:00
|
|
|
memmove(unsafe.Pointer(*(**C.RTMPPacket)(incPtr(unsafe.Pointer(r.m_vecChannelsOut),
|
|
|
|
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-08 08:37:54 +03:00
|
|
|
// int WriteN(RTMP* r, const char* buffer, int n);
|
|
|
|
// rtmp.c +1502
|
2018-08-07 06:10:43 +03:00
|
|
|
func C_WriteN(r *C.RTMP, buffer unsafe.Pointer, n int) int {
|
2018-07-15 10:48:24 +03:00
|
|
|
ptr := buffer
|
|
|
|
for n > 0 {
|
|
|
|
var nBytes int
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
nBytes = int(C_RTMPSockBuf_Send(&r.m_sb, (*byte)(ptr), int32(n)))
|
2018-07-15 10:48:24 +03:00
|
|
|
|
|
|
|
if nBytes < 0 {
|
|
|
|
if debugMode {
|
2018-08-07 06:10:43 +03:00
|
|
|
log.Println("C_WriteN, RTMP send error")
|
2018-07-15 10:48:24 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: port this
|
2018-08-11 07:18:21 +03:00
|
|
|
//C.RTMP_Close(r)
|
2018-07-15 10:48:24 +03:00
|
|
|
n = 1
|
|
|
|
break
|
|
|
|
}
|
|
|
|
|
|
|
|
if nBytes == 0 {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
|
|
|
|
n -= nBytes
|
2018-07-17 13:05:25 +03:00
|
|
|
ptr = incBytePtr(ptr, nBytes)
|
2018-07-15 10:48:24 +03:00
|
|
|
}
|
|
|
|
|
2018-07-17 13:05:25 +03:00
|
|
|
if n == 0 {
|
|
|
|
return 1
|
|
|
|
}
|
|
|
|
return 0
|
2018-07-15 10:48:24 +03:00
|
|
|
}
|
|
|
|
|
2018-07-18 06:56:23 +03:00
|
|
|
const length = 512
|
|
|
|
|
|
|
|
var RTMPT_cmds = []string{
|
|
|
|
"open",
|
|
|
|
"send",
|
|
|
|
"idle",
|
|
|
|
"close",
|
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// int RTMPSockBuf_Send(RTMPSockBuf* sb, const char* buf, int len);
|
|
|
|
// rtmp.c +4297
|
|
|
|
func C_RTMPSockBuf_Send(sb *C.RTMPSockBuf, buf *byte, l int32) int32 {
|
2018-07-19 08:53:06 +03:00
|
|
|
return int32(C.send(sb.sb_socket, unsafe.Pointer(buf), C.size_t(l), 0))
|
|
|
|
}
|
|
|
|
|
2018-08-10 09:40:08 +03:00
|
|
|
// void AV_queue(RTMP_METHOD** vals, int* num, C.AVal* av, int txn);
|
2018-08-08 08:37:54 +03:00
|
|
|
// rtmp.c +2414
|
|
|
|
func C_AV_queue(vals **C.RTMP_METHOD, num *int32, av *C.AVal, txn int32) {
|
2018-07-15 12:26:59 +03:00
|
|
|
if (*num & 0x0f) == 0 {
|
|
|
|
// TODO: work out what to do with the realloc
|
2018-07-28 16:02:59 +03:00
|
|
|
//*vals = (*C.RTMP_METHOD)(realloc(unsafe.Pointer(*vals), int((*num+16)*
|
|
|
|
//int(unsafe.Sizeof(*(*vals))))))
|
2018-07-29 14:43:40 +03:00
|
|
|
*vals = (*C.RTMP_METHOD)(C.realloc(unsafe.Pointer(*vals), C.size_t((*num+16)*
|
|
|
|
int32(unsafe.Sizeof(*(*vals))))))
|
2018-07-15 12:26:59 +03:00
|
|
|
}
|
2018-07-29 04:33:34 +03:00
|
|
|
tmp := C.malloc(C.size_t(av.av_len + 1))
|
|
|
|
//tmp := allocate(uintptr(av.av_len + 1))
|
2018-07-19 10:29:03 +03:00
|
|
|
memmove(tmp, unsafe.Pointer(av.av_val), uintptr(av.av_len))
|
2018-07-28 03:45:12 +03:00
|
|
|
*indxBytePtr(tmp, int(av.av_len)) = '\000'
|
|
|
|
|
2018-07-29 14:43:40 +03:00
|
|
|
(*(*C.RTMP_METHOD)(incPtr(unsafe.Pointer(*vals), int(*num),
|
2018-07-28 03:45:12 +03:00
|
|
|
int(unsafe.Sizeof(*(*vals)))))).num = C.int(txn)
|
2018-07-29 14:43:40 +03:00
|
|
|
(*(*C.RTMP_METHOD)(incPtr(unsafe.Pointer(*vals), int(*num),
|
2018-07-28 03:45:12 +03:00
|
|
|
int(unsafe.Sizeof(*(*vals)))))).name.av_len = av.av_len
|
2018-07-29 14:43:40 +03:00
|
|
|
(*(*C.RTMP_METHOD)(incPtr(unsafe.Pointer(*vals), int(*num),
|
2018-07-28 03:45:12 +03:00
|
|
|
int(unsafe.Sizeof(*(*vals)))))).name.av_val = (*C.char)(tmp)
|
2018-07-29 14:43:40 +03:00
|
|
|
(*num)++
|
2018-07-15 12:26:59 +03:00
|
|
|
}
|
2018-07-18 05:32:02 +03:00
|
|
|
|
2018-08-10 09:40:08 +03:00
|
|
|
// char* AMF_EncodeNamedNumber(char* output, char* outend, const C.AVal* strName, double dVal);
|
2018-08-08 08:37:54 +03:00
|
|
|
// amf.c +286
|
|
|
|
func C_AMF_EncodeNamedNumber(output *byte, outend *byte, strName *C.AVal, dVal float64) *byte {
|
2018-07-27 07:36:40 +03:00
|
|
|
if int(uintptr(unsafe.Pointer(output)))+2+int(strName.av_len) > int(uintptr(unsafe.Pointer(outend))) {
|
|
|
|
return nil
|
|
|
|
}
|
2018-08-08 08:37:54 +03:00
|
|
|
output = C_AMF_EncodeInt16(output, outend, int16(strName.av_len))
|
2018-07-29 03:54:27 +03:00
|
|
|
memmove(unsafe.Pointer(output), unsafe.Pointer(strName.av_val), uintptr(strName.av_len))
|
2018-07-27 07:36:40 +03:00
|
|
|
output = (*byte)(incBytePtr(unsafe.Pointer(output), int(strName.av_len)))
|
2018-08-08 08:37:54 +03:00
|
|
|
return C_AMF_EncodeNumber(output, outend, dVal)
|
2018-07-27 07:36:40 +03:00
|
|
|
}
|
|
|
|
|
2018-08-10 09:40:08 +03:00
|
|
|
// char* AMF_EncodeNamedBoolean(char* output, char* outend, const C.AVal* strname, int bVal);
|
2018-08-08 08:37:54 +03:00
|
|
|
// amf.c +299
|
|
|
|
func C_AMF_EncodeNamedBoolean(output *byte, outend *byte, strName *C.AVal, bVal int) *byte {
|
2018-07-27 07:36:40 +03:00
|
|
|
if int(uintptr(unsafe.Pointer(output)))+2+int(strName.av_len) > int(uintptr(unsafe.Pointer(outend))) {
|
|
|
|
return nil
|
|
|
|
}
|
2018-08-08 08:37:54 +03:00
|
|
|
output = C_AMF_EncodeInt16(output, outend, int16(strName.av_len))
|
2018-07-29 03:51:53 +03:00
|
|
|
memmove(unsafe.Pointer(output), unsafe.Pointer(strName.av_val), uintptr(strName.av_len))
|
2018-07-27 07:36:40 +03:00
|
|
|
output = (*byte)(incBytePtr(unsafe.Pointer(output), int(strName.av_len)))
|
2018-08-08 08:37:54 +03:00
|
|
|
return C_AMF_EncodeBoolean(output, outend, bVal)
|
2018-07-27 08:14:02 +03:00
|
|
|
}
|
2018-07-27 07:36:40 +03:00
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// char* AMF_EncodeBoolean(char* output, char* outend, int bVal);
|
|
|
|
// amf.c +260
|
|
|
|
func C_AMF_EncodeBoolean(output *byte, outend *byte, bVal int) *byte {
|
2018-07-27 08:14:02 +03:00
|
|
|
if int(uintptr(unsafe.Pointer(output)))+2 > int(uintptr(unsafe.Pointer(outend))) {
|
|
|
|
return nil
|
|
|
|
}
|
2018-07-29 03:51:53 +03:00
|
|
|
*(*byte)(unsafe.Pointer(output)) = AMF_BOOLEAN
|
2018-07-27 08:14:02 +03:00
|
|
|
output = (*byte)(incBytePtr(unsafe.Pointer(output), 1))
|
|
|
|
val := byte(0x01)
|
|
|
|
if bVal == 0 {
|
|
|
|
val = byte(0x00)
|
|
|
|
}
|
|
|
|
*(*byte)(unsafe.Pointer(output)) = val
|
2018-07-29 03:51:53 +03:00
|
|
|
output = (*byte)(incBytePtr(unsafe.Pointer(output), 1))
|
2018-07-27 08:14:02 +03:00
|
|
|
return output
|
2018-07-27 07:36:40 +03:00
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// char* AMF_EncodeNumber(char* output, char* outend, double dVal);
|
|
|
|
// amf.c +199
|
|
|
|
func C_AMF_EncodeNumber(output *byte, outend *byte, dVal float64) *byte {
|
2018-07-27 07:36:40 +03:00
|
|
|
if int(uintptr(unsafe.Pointer(output)))+1+8 > int(uintptr(unsafe.Pointer(outend))) {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
// TODO: port this
|
2018-08-11 07:18:21 +03:00
|
|
|
*(*byte)(unsafe.Pointer(output)) = AMF_NUMBER
|
2018-07-27 07:36:40 +03:00
|
|
|
output = (*byte)(incBytePtr(unsafe.Pointer(output), 1))
|
|
|
|
// NOTE: here we are assuming little endian for both byte order and float
|
|
|
|
// word order
|
|
|
|
var ci, co *uint8
|
|
|
|
ci = (*uint8)(unsafe.Pointer(&dVal))
|
|
|
|
co = (*uint8)(unsafe.Pointer(output))
|
|
|
|
for i := 0; i < 8; i++ {
|
|
|
|
*indxBytePtr(unsafe.Pointer(co), i) = *indxBytePtr(unsafe.Pointer(ci), 7-i)
|
|
|
|
}
|
2018-07-27 08:14:02 +03:00
|
|
|
return (*byte)(incBytePtr(unsafe.Pointer(output), 8))
|
2018-07-27 07:36:40 +03:00
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// double AMF_DecodeNumber(const char* data);
|
|
|
|
// amf.c +82
|
|
|
|
func C_AMF_DecodeNumber(data *byte) float64 {
|
2018-07-28 03:45:12 +03:00
|
|
|
var dVal float64
|
|
|
|
var ci, co *uint8
|
|
|
|
ci = (*uint8)(unsafe.Pointer(data))
|
|
|
|
co = (*uint8)(unsafe.Pointer(&dVal))
|
|
|
|
for i := 0; i < 8; i++ {
|
|
|
|
*indxBytePtr(unsafe.Pointer(co), i) = *indxBytePtr(unsafe.Pointer(ci), 7-i)
|
|
|
|
}
|
|
|
|
return dVal
|
|
|
|
}
|
|
|
|
|
2018-08-10 09:40:08 +03:00
|
|
|
// char* AMF_EncodeNamedString(char* output, char* outend, const C.AVal* strName, const C.AVal* strValue);
|
2018-08-08 08:37:54 +03:00
|
|
|
// amf.c +273
|
|
|
|
func C_AMF_EncodeNamedString(output *byte, outend *byte, strName *C.AVal, strValue *C.AVal) *byte {
|
2018-07-27 07:36:40 +03:00
|
|
|
if int(uintptr(unsafe.Pointer(output)))+2+int(strName.av_len) > int(uintptr(unsafe.Pointer(outend))) {
|
|
|
|
return nil
|
|
|
|
}
|
2018-08-08 08:37:54 +03:00
|
|
|
output = C_AMF_EncodeInt16(output, outend, int16(strName.av_len))
|
2018-07-29 03:47:53 +03:00
|
|
|
memmove(unsafe.Pointer(output), unsafe.Pointer(strName.av_val), uintptr(strName.av_len))
|
2018-07-27 07:36:40 +03:00
|
|
|
output = (*byte)(incBytePtr(unsafe.Pointer(output), int(strName.av_len)))
|
2018-08-08 08:37:54 +03:00
|
|
|
return C_AMF_EncodeString(output, outend, strValue)
|
2018-07-27 07:36:40 +03:00
|
|
|
}
|
|
|
|
|
2018-08-10 09:40:08 +03:00
|
|
|
// void AMF_DecodeString(const char* data, C.AVal* bv);
|
2018-08-08 08:37:54 +03:00
|
|
|
// amf.c +68
|
|
|
|
func C_AMF_DecodeString(data *byte, bv *C.AVal) {
|
2018-07-27 07:36:40 +03:00
|
|
|
dataPtr := unsafe.Pointer(data)
|
2018-07-29 04:25:02 +03:00
|
|
|
//bv.av_len = C.int(C.AMF_DecodeInt16((*C.char)(dataPtr)))
|
2018-08-08 08:37:54 +03:00
|
|
|
bv.av_len = C.int(C_AMF_DecodeInt16((*byte)(dataPtr)))
|
2018-07-27 07:36:40 +03:00
|
|
|
if bv.av_len > 0 {
|
|
|
|
bv.av_val = (*C.char)(incBytePtr(dataPtr, 2))
|
|
|
|
} else {
|
|
|
|
bv.av_val = nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// unsigned short AMF_DecodeInt16(const char* data);
|
|
|
|
// amf.c +41
|
|
|
|
func C_AMF_DecodeInt16(data *byte) uint16 {
|
2018-07-27 07:36:40 +03:00
|
|
|
c := unsafe.Pointer(data)
|
2018-07-29 04:25:02 +03:00
|
|
|
return uint16(*(*uint8)(c)<<8 | *(*byte)(incBytePtr(c, 1)))
|
2018-07-27 07:36:40 +03:00
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// char* AMF_EncodeInt24(char* output, char* outend, int nVal);
|
|
|
|
// amf.c +149
|
2018-08-07 06:00:14 +03:00
|
|
|
func C_AMFEncodeInt24(output *byte, outend *byte, nVal int32) *byte {
|
2018-07-27 07:36:40 +03:00
|
|
|
outputPtr := unsafe.Pointer(output)
|
|
|
|
outendPtr := unsafe.Pointer(outend)
|
|
|
|
if uintptr(outputPtr)+3 > uintptr(outendPtr) {
|
|
|
|
// length < 3
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
// Assign output[2]
|
|
|
|
third := (*byte)(incBytePtr(outputPtr, 2))
|
|
|
|
*third = (byte)(nVal & 0xff)
|
|
|
|
// Assign output[1]
|
|
|
|
second := (*byte)(incBytePtr(outputPtr, 1))
|
|
|
|
*second = (byte)(nVal >> 8)
|
|
|
|
// Assign output[0]
|
|
|
|
*output = (byte)(nVal >> 16)
|
|
|
|
return (*byte)(incBytePtr(outputPtr, 3))
|
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// unsigned int AMF_DecodeInt24(const char* data);
|
|
|
|
// amf.c +50
|
|
|
|
func C_AMF_DecodeInt24(data *byte) uint32 {
|
2018-07-27 07:36:40 +03:00
|
|
|
// TODO Understand logic and simplify
|
|
|
|
c := (*uint8)(unsafe.Pointer(data))
|
|
|
|
dst := uint32(int32(*c) << 16)
|
|
|
|
dst |= uint32(int32(*((*uint8)(unsafe.Pointer(uintptr(unsafe.Pointer(c)) +
|
|
|
|
(uintptr)(int32(1))*unsafe.Sizeof(*c))))) << 8)
|
|
|
|
dst |= uint32(int32(*((*uint8)(unsafe.Pointer(uintptr(unsafe.Pointer(c)) +
|
|
|
|
(uintptr)(int32(2))*unsafe.Sizeof(*c))))))
|
|
|
|
return dst
|
|
|
|
}
|
|
|
|
|
2018-08-10 09:40:08 +03:00
|
|
|
// char* AMF_EncodeString(char* output, char* outend, const C.AVal* bv);
|
2018-08-08 08:37:54 +03:00
|
|
|
// amf.c +174
|
|
|
|
func C_AMF_EncodeString(output *byte, outend *byte, bv *C.AVal) *byte {
|
2018-07-27 07:36:40 +03:00
|
|
|
outputPtr := unsafe.Pointer(output)
|
|
|
|
outendPtr := unsafe.Pointer(outend)
|
|
|
|
if (bv.av_len < 65536 && uintptr(incBytePtr(outputPtr, 1+2+int(bv.av_len))) >
|
|
|
|
uintptr(outendPtr)) || uintptr(incBytePtr(outputPtr, 1+4+int(bv.av_len))) >
|
|
|
|
uintptr(outendPtr) {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
if bv.av_len < 65536 {
|
2018-07-28 16:40:57 +03:00
|
|
|
*(*byte)(outputPtr) = AMF_STRING
|
2018-07-29 03:12:20 +03:00
|
|
|
outputPtr = incBytePtr(outputPtr, 1)
|
2018-08-11 07:18:21 +03:00
|
|
|
// TODO: port AMF_EncodeInt16
|
|
|
|
outputPtr = unsafe.Pointer(C_AMF_EncodeInt16((*byte)(outputPtr), (*byte)(
|
|
|
|
outendPtr), int16(bv.av_len)))
|
2018-08-08 08:37:54 +03:00
|
|
|
//outputPtr = unsafe.Pointer(C_AMF_EncodeInt16((*byte)(outputPtr),
|
2018-07-28 16:40:57 +03:00
|
|
|
//(*byte)(outendPtr), (int16)(bv.av_len)))
|
2018-07-27 07:36:40 +03:00
|
|
|
} else {
|
2018-07-28 16:40:57 +03:00
|
|
|
*(*byte)(outputPtr) = AMF_LONG_STRING
|
2018-07-29 03:12:20 +03:00
|
|
|
outputPtr = incBytePtr(outputPtr, 1)
|
2018-08-11 07:18:21 +03:00
|
|
|
outputPtr = unsafe.Pointer(C_AMF_EncodeInt32((*byte)(outputPtr), (*byte)(
|
|
|
|
outendPtr), int32(bv.av_len)))
|
2018-08-08 08:37:54 +03:00
|
|
|
//outputPtr = unsafe.Pointer(C_AMF_EncodeInt32((*byte)(outputPtr),
|
2018-07-28 16:40:57 +03:00
|
|
|
//(*byte)(outendPtr), (int32)(bv.av_len)))
|
2018-07-27 07:36:40 +03:00
|
|
|
}
|
|
|
|
memmove(unsafe.Pointer(outputPtr), unsafe.Pointer(bv.av_val), uintptr(bv.av_len))
|
|
|
|
//C.memcpy(unsafe.Pointer(outputPtr), unsafe.Pointer(bv.av_val), (C.size_t)(bv.av_len))
|
2018-07-28 16:40:57 +03:00
|
|
|
outputPtr = incBytePtr(outputPtr, int(bv.av_len))
|
2018-07-27 07:36:40 +03:00
|
|
|
return (*byte)(outputPtr)
|
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// char* AMF_EncodeInt16(char* output, char* outend, short nVal);
|
|
|
|
// amf.c +138
|
|
|
|
func C_AMF_EncodeInt16(output *byte, outend *byte, nVal int16) *byte {
|
2018-07-27 07:36:40 +03:00
|
|
|
outputPtr := unsafe.Pointer(output)
|
|
|
|
outendPtr := unsafe.Pointer(outend)
|
|
|
|
if uintptr(outputPtr)+2 > uintptr(outendPtr) {
|
|
|
|
// length < 2
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
// Assign output[1]
|
|
|
|
second := (*byte)(incBytePtr(outputPtr, 1))
|
|
|
|
*second = (byte)(nVal & 0xff)
|
|
|
|
// Assign output[0]
|
|
|
|
*output = (byte)(nVal >> 8)
|
|
|
|
return (*byte)(incBytePtr(outputPtr, 2))
|
|
|
|
}
|
|
|
|
|
2018-08-08 08:37:54 +03:00
|
|
|
// char* AMF_EncodeInt32(char* output, char* outend, int nVal);
|
|
|
|
// amf.c +161
|
|
|
|
func C_AMF_EncodeInt32(output *byte, outend *byte, nVal int32) *byte {
|
2018-07-27 07:36:40 +03:00
|
|
|
outputPtr := unsafe.Pointer(output)
|
|
|
|
outendPtr := unsafe.Pointer(outend)
|
|
|
|
if uintptr(outputPtr)+4 > uintptr(outendPtr) {
|
|
|
|
// length < 4
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
// Assign output[3]
|
|
|
|
forth := (*byte)(incBytePtr(outputPtr, 3))
|
|
|
|
*forth = (byte)(nVal & 0xff)
|
|
|
|
// Assign output[2]
|
|
|
|
third := (*byte)(incBytePtr(outputPtr, 2))
|
|
|
|
*third = (byte)(nVal >> 8)
|
|
|
|
// Assign output[1]
|
|
|
|
second := (*byte)(incBytePtr(outputPtr, 1))
|
|
|
|
*second = (byte)(nVal >> 16)
|
|
|
|
// Assign output[0]
|
|
|
|
*output = (byte)(nVal >> 24)
|
|
|
|
return (*byte)(incBytePtr(outputPtr, 4))
|
|
|
|
}
|
|
|
|
|
2018-07-28 04:35:51 +03:00
|
|
|
func realloc(ptr unsafe.Pointer, size int) unsafe.Pointer {
|
|
|
|
dest := allocate(uintptr(size))
|
2018-07-28 04:55:35 +03:00
|
|
|
if ptr != nil {
|
|
|
|
memmove(dest, ptr, uintptr(size))
|
|
|
|
}
|
2018-07-28 04:35:51 +03:00
|
|
|
return dest
|
|
|
|
}
|
|
|
|
|
|
|
|
func memmove(to, from unsafe.Pointer, n uintptr) {
|
|
|
|
copy(ptrToSlice(to, int(n)), ptrToSlice(from, int(n)))
|
|
|
|
}
|
|
|
|
|
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-07-24 15:00:50 +03:00
|
|
|
aValue := *indxBytePtr(a, i)
|
|
|
|
bValue := *indxBytePtr(b, i)
|
2018-07-23 02:08:22 +03:00
|
|
|
if aValue != bValue {
|
|
|
|
if aValue < bValue {
|
|
|
|
return -1
|
|
|
|
} else {
|
|
|
|
return 1
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2018-07-28 16:02:59 +03:00
|
|
|
func memset(ptr *byte, val int, num int) {
|
2018-07-21 02:24:16 +03:00
|
|
|
for i := 0; i < num; i++ {
|
2018-07-28 16:02:59 +03:00
|
|
|
*indxBytePtr(unsafe.Pointer(ptr), int(i)) = byte(uint8(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)
|
|
|
|
}
|
|
|
|
|
|
|
|
// wrapper for converting slice to unsafe.pointer
|
2018-07-20 08:44:39 +03:00
|
|
|
func sToUP(b []byte) unsafe.Pointer {
|
2018-07-20 07:40:08 +03:00
|
|
|
return unsafe.Pointer(&b[0])
|
|
|
|
}
|
|
|
|
|
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++ {
|
|
|
|
slice[i] = *indxBytePtr(ptr, i)
|
|
|
|
}
|
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 {
|
|
|
|
return string(ptrToSlice(unsafe.Pointer(cStr), int(strlen(cStr))))
|
|
|
|
}
|
|
|
|
|
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)
|
|
|
|
oldMem := ptrToSlice(unsafe.Pointer(str), int(length+1))
|
|
|
|
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
|
|
|
var ptr *byte
|
|
|
|
for i := 0; i < 1000; i++ {
|
2018-07-19 18:35:21 +03:00
|
|
|
ptr = indxBytePtr(unsafe.Pointer(str), i)
|
|
|
|
if *ptr == '\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 {
|
|
|
|
var ptr *byte
|
2018-07-20 08:44:39 +03:00
|
|
|
for i := 0; i < 1000; i++ {
|
2018-07-19 18:35:21 +03:00
|
|
|
ptr = indxBytePtr(unsafe.Pointer(str), i)
|
|
|
|
if *ptr == val {
|
|
|
|
return ptr
|
|
|
|
}
|
|
|
|
if *ptr == '\000' {
|
|
|
|
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-07-19 10:54:37 +03:00
|
|
|
func allocate(nOfBytes uintptr) unsafe.Pointer {
|
|
|
|
mem := make([]byte, int(nOfBytes))
|
|
|
|
return unsafe.Pointer(&mem[0])
|
|
|
|
}
|
|
|
|
|
2018-07-16 06:30:58 +03:00
|
|
|
// indxBytePtr returns a byte at the indx inc give a ptr
|
2018-07-16 12:20:17 +03:00
|
|
|
func indxBytePtr(ptr unsafe.Pointer, inc int) *byte {
|
2018-08-10 09:40:08 +03:00
|
|
|
return (*byte)(incPtr(ptr, inc, 1))
|
2018-07-16 06:30:58 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
// indxInt32Ptr returns an int32 at the indx inc given a ptr
|
2018-07-16 12:20:17 +03:00
|
|
|
func indxInt32Ptr(ptr unsafe.Pointer, inc int) *int32 {
|
2018-08-10 09:40:08 +03:00
|
|
|
return (*int32)(incPtr(ptr, inc, 4))
|
2018-07-16 06:30:58 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
// indxInt64Ptr returns an int64 at the indx inc given a ptr
|
2018-07-16 12:20:17 +03:00
|
|
|
func indxInt64Ptr(ptr unsafe.Pointer, inc int) *int64 {
|
2018-08-10 09:40:08 +03:00
|
|
|
return (*int64)(incPtr(ptr, inc, 8))
|
2018-07-16 06:30:58 +03:00
|
|
|
}
|
|
|
|
|
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
|
|
|
}
|
|
|
|
|
2018-07-16 08:07:26 +03:00
|
|
|
// incInt32Ptr returns an unsafe.Pointer to an int32 that is inc positive
|
|
|
|
// positions from the passed ptr
|
2018-07-16 06:30:58 +03:00
|
|
|
func incInt32Ptr(ptr unsafe.Pointer, inc int) unsafe.Pointer {
|
2018-08-10 09:40:08 +03:00
|
|
|
return incPtr(ptr, inc, 4)
|
2018-07-16 06:30:58 +03:00
|
|
|
}
|
|
|
|
|
2018-07-16 08:07:26 +03:00
|
|
|
// incInt64Ptr returns an unsafe.Pointer to an int64 that is inc positive
|
|
|
|
// positions from the passed ptr
|
2018-07-16 06:30:58 +03:00
|
|
|
func incInt64Ptr(ptr unsafe.Pointer, inc int) unsafe.Pointer {
|
2018-08-10 09:40:08 +03:00
|
|
|
return incPtr(ptr, inc, 8)
|
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
|
|
|
}
|
|
|
|
|
|
|
|
// decBytePtr returns an unsafe.Pointer to a int32 that is dec negative positions
|
|
|
|
// from ptr
|
|
|
|
func decInt32Ptr(ptr unsafe.Pointer, dec int) unsafe.Pointer {
|
2018-08-10 09:40:08 +03:00
|
|
|
return decPtr(ptr, dec, 4)
|
2018-07-16 08:07:26 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
// decBytePtr returns an unsafe.Pointer to a int64 that is dec negative positions
|
|
|
|
// from ptr
|
|
|
|
func decInt64Ptr(ptr unsafe.Pointer, dec int) unsafe.Pointer {
|
2018-08-10 09:40:08 +03:00
|
|
|
return decPtr(ptr, dec, 8)
|
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])
|
|
|
|
}
|
|
|
|
|
|
|
|
// ptrToSlice returns a slice given unsafe pointer and size - no allocation and
|
|
|
|
// copying is required - same data is used.
|
|
|
|
func ptrToSlice(data unsafe.Pointer, size int) []byte {
|
|
|
|
var ret []byte
|
|
|
|
shDest := (*reflect.SliceHeader)(unsafe.Pointer(&ret))
|
|
|
|
shDest.Data = uintptr(data)
|
|
|
|
shDest.Len = size
|
|
|
|
shDest.Cap = size
|
|
|
|
return ret
|
|
|
|
}
|
|
|
|
|
2018-07-18 18:24:36 +03:00
|
|
|
// C.AVal is in amf.h
|
|
|
|
// See AVC(str) {str, sizeof(str)-1} in amf.h
|
|
|
|
func AVC(str string) C.AVal {
|
|
|
|
var aval C.AVal
|
|
|
|
aval.av_val = C.CString(str)
|
|
|
|
aval.av_len = C.int(len(str))
|
|
|
|
return aval
|
|
|
|
}
|
|
|
|
|
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))
|
|
|
|
}
|