diff --git a/configure.ac b/configure.ac index 9127e49041..b7ece219c3 100644 --- a/configure.ac +++ b/configure.ac @@ -1707,6 +1707,7 @@ gst/pnm/Makefile gst/qtmux/Makefile gst/rawparse/Makefile gst/real/Makefile +gst/rtmp/Makefile gst/rtpmux/Makefile gst/scaletempo/Makefile gst/sdp/Makefile diff --git a/gst/rtmp/Makefile.am b/gst/rtmp/Makefile.am new file mode 100644 index 0000000000..6ac507e0d3 --- /dev/null +++ b/gst/rtmp/Makefile.am @@ -0,0 +1,25 @@ +plugin_LTLIBRARIES = libgstrtmp.la + +libgstrtmp_la_SOURCES = amf.c \ + gstrtmpsrc.c \ + hashswf.c \ + log.c \ + parseurl.c \ + rtmp.c + +noinst_HEADERS = amf.h \ + bytes.h \ + dhgroups.h \ + dh.h \ + gstrtmpsrc.h \ + handshake.h \ + http.h \ + log.h \ + rtmp.h \ + rtmp_sys.h + +libgstrtmp_la_CFLAGS = $(GST_PLUGINS_BASE_CFLAGS) $(GST_CFLAGS) -DCRYPTO +libgstrtmp_la_LIBADD = $(GST_PLUGINS_BASE_LIBS) $(GST_BASE_LIBS) $(GST_LIBS) -lssl $(LIBM) +libgstrtmp_la_LDFLAGS = $(GST_PLUGIN_LDFLAGS) +libgstrtmp_la_LIBTOOLFLAGS = --tag=disable-static + diff --git a/gst/rtmp/amf.c b/gst/rtmp/amf.c new file mode 100644 index 0000000000..e7d8bcb43c --- /dev/null +++ b/gst/rtmp/amf.c @@ -0,0 +1,1127 @@ +/* + * Copyright (C) 2005-2008 Team XBMC + * http://www.xbmc.org + * Copyright (C) 2008-2009 Andrej Stepanchuk + * Copyright (C) 2009-2010 Howard Chu + * + * This file is part of librtmp. + * + * librtmp is free software; you can redistribute it and/or modify + * it under the terms of the GNU Lesser General Public License as + * published by the Free Software Foundation; either version 2.1, + * or (at your option) any later version. + * + * librtmp 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 Lesser General Public License + * along with librtmp see the file COPYING. If not, write to + * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. + * http://www.gnu.org/copyleft/lgpl.html + */ + +#include +#include +#include + +#include "rtmp_sys.h" +#include "amf.h" +#include "log.h" +#include "bytes.h" + +static const AMFObjectProperty AMFProp_Invalid = { {0, 0}, AMF_INVALID }; +static const AVal AV_empty = { 0, 0 }; + +/* Data is Big-Endian */ +unsigned short +AMF_DecodeInt16 (const char *data) +{ + unsigned char *c = (unsigned char *) data; + unsigned short val; + val = (c[0] << 8) | c[1]; + return val; +} + +unsigned int +AMF_DecodeInt24 (const char *data) +{ + unsigned char *c = (unsigned char *) data; + unsigned int val; + val = (c[0] << 16) | (c[1] << 8) | c[2]; + return val; +} + +unsigned int +AMF_DecodeInt32 (const char *data) +{ + unsigned char *c = (unsigned char *) data; + unsigned int val; + val = (c[0] << 24) | (c[1] << 16) | (c[2] << 8) | c[3]; + return val; +} + +void +AMF_DecodeString (const char *data, AVal * bv) +{ + bv->av_len = AMF_DecodeInt16 (data); + bv->av_val = (bv->av_len > 0) ? (char *) data + 2 : NULL; +} + +void +AMF_DecodeLongString (const char *data, AVal * bv) +{ + bv->av_len = AMF_DecodeInt32 (data); + bv->av_val = (bv->av_len > 0) ? (char *) data + 4 : NULL; +} + +double +AMF_DecodeNumber (const char *data) +{ + double dVal; +#if __FLOAT_WORD_ORDER == __BYTE_ORDER +#if __BYTE_ORDER == __BIG_ENDIAN + memcpy (&dVal, data, 8); +#elif __BYTE_ORDER == __LITTLE_ENDIAN + unsigned char *ci, *co; + ci = (unsigned char *) data; + co = (unsigned char *) &dVal; + co[0] = ci[7]; + co[1] = ci[6]; + co[2] = ci[5]; + co[3] = ci[4]; + co[4] = ci[3]; + co[5] = ci[2]; + co[6] = ci[1]; + co[7] = ci[0]; +#endif +#else +#if __BYTE_ORDER == __LITTLE_ENDIAN /* __FLOAT_WORD_ORER == __BIG_ENDIAN */ + unsigned char *ci, *co; + ci = (unsigned char *) data; + co = (unsigned char *) &dVal; + co[0] = ci[3]; + co[1] = ci[2]; + co[2] = ci[1]; + co[3] = ci[0]; + co[4] = ci[7]; + co[5] = ci[6]; + co[6] = ci[5]; + co[7] = ci[4]; +#else /* __BYTE_ORDER == __BIG_ENDIAN && __FLOAT_WORD_ORER == __LITTLE_ENDIAN */ + unsigned char *ci, *co; + ci = (unsigned char *) data; + co = (unsigned char *) &dVal; + co[0] = ci[4]; + co[1] = ci[5]; + co[2] = ci[6]; + co[3] = ci[7]; + co[4] = ci[0]; + co[5] = ci[1]; + co[6] = ci[2]; + co[7] = ci[3]; +#endif +#endif + return dVal; +} + +bool +AMF_DecodeBoolean (const char *data) +{ + return *data != 0; +} + +char * +AMF_EncodeInt16 (char *output, char *outend, short nVal) +{ + if (output + 2 > outend) + return NULL; + + output[1] = nVal & 0xff; + output[0] = nVal >> 8; + return output + 2; +} + +char * +AMF_EncodeInt24 (char *output, char *outend, int nVal) +{ + if (output + 3 > outend) + return NULL; + + output[2] = nVal & 0xff; + output[1] = nVal >> 8; + output[0] = nVal >> 16; + return output + 3; +} + +char * +AMF_EncodeInt32 (char *output, char *outend, int nVal) +{ + if (output + 4 > outend) + return NULL; + + output[3] = nVal & 0xff; + output[2] = nVal >> 8; + output[1] = nVal >> 16; + output[0] = nVal >> 24; + return output + 4; +} + +char * +AMF_EncodeString (char *output, char *outend, const AVal * bv) +{ + if ((bv->av_len < 65536 && output + 1 + 2 + bv->av_len > outend) || + output + 1 + 4 + bv->av_len > outend) + return NULL; + + if (bv->av_len < 65536) { + *output++ = AMF_STRING; + + output = AMF_EncodeInt16 (output, outend, bv->av_len); + } else { + *output++ = AMF_LONG_STRING; + + output = AMF_EncodeInt32 (output, outend, bv->av_len); + } + memcpy (output, bv->av_val, bv->av_len); + output += bv->av_len; + + return output; +} + +char * +AMF_EncodeNumber (char *output, char *outend, double dVal) +{ + if (output + 1 + 8 > outend) + return NULL; + + *output++ = AMF_NUMBER; /* type: Number */ + +#if __FLOAT_WORD_ORDER == __BYTE_ORDER +#if __BYTE_ORDER == __BIG_ENDIAN + memcpy (output, &dVal, 8); +#elif __BYTE_ORDER == __LITTLE_ENDIAN + { + unsigned char *ci, *co; + ci = (unsigned char *) &dVal; + co = (unsigned char *) output; + co[0] = ci[7]; + co[1] = ci[6]; + co[2] = ci[5]; + co[3] = ci[4]; + co[4] = ci[3]; + co[5] = ci[2]; + co[6] = ci[1]; + co[7] = ci[0]; + } +#endif +#else +#if __BYTE_ORDER == __LITTLE_ENDIAN /* __FLOAT_WORD_ORER == __BIG_ENDIAN */ + { + unsigned char *ci, *co; + ci = (unsigned char *) &dVal; + co = (unsigned char *) output; + co[0] = ci[3]; + co[1] = ci[2]; + co[2] = ci[1]; + co[3] = ci[0]; + co[4] = ci[7]; + co[5] = ci[6]; + co[6] = ci[5]; + co[7] = ci[4]; + } +#else /* __BYTE_ORDER == __BIG_ENDIAN && __FLOAT_WORD_ORER == __LITTLE_ENDIAN */ + { + unsigned char *ci, *co; + ci = (unsigned char *) &dVal; + co = (unsigned char *) output; + co[0] = ci[4]; + co[1] = ci[5]; + co[2] = ci[6]; + co[3] = ci[7]; + co[4] = ci[0]; + co[5] = ci[1]; + co[6] = ci[2]; + co[7] = ci[3]; + } +#endif +#endif + + return output + 8; +} + +char * +AMF_EncodeBoolean (char *output, char *outend, bool bVal) +{ + if (output + 2 > outend) + return NULL; + + *output++ = AMF_BOOLEAN; + + *output++ = bVal ? 0x01 : 0x00; + + return output; +} + +char * +AMF_EncodeNamedString (char *output, char *outend, const AVal * strName, + const AVal * strValue) +{ + if (output + 2 + strName->av_len > outend) + return NULL; + output = AMF_EncodeInt16 (output, outend, strName->av_len); + + memcpy (output, strName->av_val, strName->av_len); + output += strName->av_len; + + return AMF_EncodeString (output, outend, strValue); +} + +char * +AMF_EncodeNamedNumber (char *output, char *outend, const AVal * strName, + double dVal) +{ + if (output + 2 + strName->av_len > outend) + return NULL; + output = AMF_EncodeInt16 (output, outend, strName->av_len); + + memcpy (output, strName->av_val, strName->av_len); + output += strName->av_len; + + return AMF_EncodeNumber (output, outend, dVal); +} + +char * +AMF_EncodeNamedBoolean (char *output, char *outend, const AVal * strName, + bool bVal) +{ + if (output + 2 + strName->av_len > outend) + return NULL; + output = AMF_EncodeInt16 (output, outend, strName->av_len); + + memcpy (output, strName->av_val, strName->av_len); + output += strName->av_len; + + return AMF_EncodeBoolean (output, outend, bVal); +} + +void +AMFProp_GetName (AMFObjectProperty * prop, AVal * name) +{ + *name = prop->p_name; +} + +void +AMFProp_SetName (AMFObjectProperty * prop, AVal * name) +{ + prop->p_name = *name; +} + +AMFDataType +AMFProp_GetType (AMFObjectProperty * prop) +{ + return prop->p_type; +} + +double +AMFProp_GetNumber (AMFObjectProperty * prop) +{ + return prop->p_vu.p_number; +} + +bool +AMFProp_GetBoolean (AMFObjectProperty * prop) +{ + return prop->p_vu.p_number != 0; +} + +void +AMFProp_GetString (AMFObjectProperty * prop, AVal * str) +{ + *str = prop->p_vu.p_aval; +} + +void +AMFProp_GetObject (AMFObjectProperty * prop, AMFObject * obj) +{ + *obj = prop->p_vu.p_object; +} + +bool +AMFProp_IsValid (AMFObjectProperty * prop) +{ + return prop->p_type != AMF_INVALID; +} + +char * +AMFProp_Encode (AMFObjectProperty * prop, char *pBuffer, char *pBufEnd) +{ + if (prop->p_type == AMF_INVALID) + return NULL; + + if (prop->p_type != AMF_NULL + && pBuffer + prop->p_name.av_len + 2 + 1 >= pBufEnd) + return NULL; + + if (prop->p_type != AMF_NULL && prop->p_name.av_len) { + *pBuffer++ = prop->p_name.av_len >> 8; + *pBuffer++ = prop->p_name.av_len & 0xff; + memcpy (pBuffer, prop->p_name.av_val, prop->p_name.av_len); + pBuffer += prop->p_name.av_len; + } + + switch (prop->p_type) { + case AMF_NUMBER: + pBuffer = AMF_EncodeNumber (pBuffer, pBufEnd, prop->p_vu.p_number); + break; + + case AMF_BOOLEAN: + pBuffer = AMF_EncodeBoolean (pBuffer, pBufEnd, prop->p_vu.p_number != 0); + break; + + case AMF_STRING: + pBuffer = AMF_EncodeString (pBuffer, pBufEnd, &prop->p_vu.p_aval); + break; + + case AMF_NULL: + if (pBuffer + 1 >= pBufEnd) + return NULL; + *pBuffer++ = AMF_NULL; + break; + + case AMF_OBJECT: + pBuffer = AMF_Encode (&prop->p_vu.p_object, pBuffer, pBufEnd); + break; + + default: + RTMP_Log (RTMP_LOGERROR, "%s, invalid type. %d", __FUNCTION__, + prop->p_type); + pBuffer = NULL; + }; + + return pBuffer; +} + +#define AMF3_INTEGER_MAX 268435455 +#define AMF3_INTEGER_MIN -268435456 + +static int +AMF3ReadInteger (const char *data, int32_t * valp) +{ + int i = 0; + int32_t val = 0; + + while (i <= 2) { /* handle first 3 bytes */ + if (data[i] & 0x80) { /* byte used */ + val <<= 7; /* shift up */ + val |= (data[i] & 0x7f); /* add bits */ + i++; + } else { + break; + } + } + + if (i > 2) { /* use 4th byte, all 8bits */ + val <<= 8; + val |= data[3]; + + /* range check */ + if (val > AMF3_INTEGER_MAX) + val -= (1 << 29); + } else { /* use 7bits of last unparsed byte (0xxxxxxx) */ + val <<= 7; + val |= data[i]; + } + + *valp = val; + + return i > 2 ? 4 : i + 1; +} + +static int +AMF3ReadString (const char *data, AVal * str) +{ + int32_t ref = 0; + int len; + assert (str != 0); + + len = AMF3ReadInteger (data, &ref); + data += len; + + if ((ref & 0x1) == 0) { /* reference: 0xxx */ + uint32_t refIndex = (ref >> 1); + RTMP_Log (RTMP_LOGDEBUG, + "%s, string reference, index: %d, not supported, ignoring!", + __FUNCTION__, refIndex); + return len; + } else { + uint32_t nSize = (ref >> 1); + + str->av_val = (char *) data; + str->av_len = nSize; + + return len + nSize; + } + return len; +} + +int +AMF3Prop_Decode (AMFObjectProperty * prop, const char *pBuffer, int nSize, + bool bDecodeName) +{ + int nOriginalSize = nSize; + AMF3DataType type; + + prop->p_name.av_len = 0; + prop->p_name.av_val = NULL; + + if (nSize == 0 || !pBuffer) { + RTMP_Log (RTMP_LOGDEBUG, "empty buffer/no buffer pointer!"); + return -1; + } + + /* decode name */ + if (bDecodeName) { + AVal name; + int nRes = AMF3ReadString (pBuffer, &name); + + if (name.av_len <= 0) + return nRes; + + prop->p_name = name; + pBuffer += nRes; + nSize -= nRes; + } + + /* decode */ + type = *pBuffer++; + nSize--; + + switch (type) { + case AMF3_UNDEFINED: + case AMF3_NULL: + prop->p_type = AMF_NULL; + break; + case AMF3_FALSE: + prop->p_type = AMF_BOOLEAN; + prop->p_vu.p_number = 0.0; + break; + case AMF3_TRUE: + prop->p_type = AMF_BOOLEAN; + prop->p_vu.p_number = 1.0; + break; + case AMF3_INTEGER: + { + int32_t res = 0; + int len = AMF3ReadInteger (pBuffer, &res); + prop->p_vu.p_number = (double) res; + prop->p_type = AMF_NUMBER; + nSize -= len; + break; + } + case AMF3_DOUBLE: + if (nSize < 8) + return -1; + prop->p_vu.p_number = AMF_DecodeNumber (pBuffer); + prop->p_type = AMF_NUMBER; + nSize -= 8; + break; + case AMF3_STRING: + case AMF3_XML_DOC: + case AMF3_XML: + { + int len = AMF3ReadString (pBuffer, &prop->p_vu.p_aval); + prop->p_type = AMF_STRING; + nSize -= len; + break; + } + case AMF3_DATE: + { + int32_t res = 0; + int len = AMF3ReadInteger (pBuffer, &res); + + nSize -= len; + pBuffer += len; + + if ((res & 0x1) == 0) { /* reference */ + uint32_t nIndex = (res >> 1); + RTMP_Log (RTMP_LOGDEBUG, "AMF3_DATE reference: %d, not supported!", + nIndex); + } else { + if (nSize < 8) + return -1; + + prop->p_vu.p_number = AMF_DecodeNumber (pBuffer); + nSize -= 8; + prop->p_type = AMF_NUMBER; + } + break; + } + case AMF3_OBJECT: + { + int nRes = AMF3_Decode (&prop->p_vu.p_object, pBuffer, nSize, true); + if (nRes == -1) + return -1; + nSize -= nRes; + prop->p_type = AMF_OBJECT; + break; + } + case AMF3_ARRAY: + case AMF3_BYTE_ARRAY: + default: + RTMP_Log (RTMP_LOGDEBUG, + "%s - AMF3 unknown/unsupported datatype 0x%02x, @0x%08X", + __FUNCTION__, (unsigned char) (*pBuffer), pBuffer); + return -1; + } + + return nOriginalSize - nSize; +} + +int +AMFProp_Decode (AMFObjectProperty * prop, const char *pBuffer, int nSize, + bool bDecodeName) +{ + int nOriginalSize = nSize; + int nRes; + + prop->p_name.av_len = 0; + prop->p_name.av_val = NULL; + + if (nSize == 0 || !pBuffer) { + RTMP_Log (RTMP_LOGDEBUG, "%s: Empty buffer/no buffer pointer!", + __FUNCTION__); + return -1; + } + + if (bDecodeName && nSize < 4) { /* at least name (length + at least 1 byte) and 1 byte of data */ + RTMP_Log (RTMP_LOGDEBUG, + "%s: Not enough data for decoding with name, less than 4 bytes!", + __FUNCTION__); + return -1; + } + + if (bDecodeName) { + unsigned short nNameSize = AMF_DecodeInt16 (pBuffer); + if (nNameSize > nSize - 2) { + RTMP_Log (RTMP_LOGDEBUG, + "%s: Name size out of range: namesize (%d) > len (%d) - 2", + __FUNCTION__, nNameSize, nSize); + return -1; + } + + AMF_DecodeString (pBuffer, &prop->p_name); + nSize -= 2 + nNameSize; + pBuffer += 2 + nNameSize; + } + + if (nSize == 0) { + return -1; + } + + nSize--; + + prop->p_type = *pBuffer++; + switch (prop->p_type) { + case AMF_NUMBER: + if (nSize < 8) + return -1; + prop->p_vu.p_number = AMF_DecodeNumber (pBuffer); + nSize -= 8; + break; + case AMF_BOOLEAN: + if (nSize < 1) + return -1; + prop->p_vu.p_number = (double) AMF_DecodeBoolean (pBuffer); + nSize--; + break; + case AMF_STRING: + { + unsigned short nStringSize = AMF_DecodeInt16 (pBuffer); + + if (nSize < (long) nStringSize + 2) + return -1; + AMF_DecodeString (pBuffer, &prop->p_vu.p_aval); + nSize -= (2 + nStringSize); + break; + } + case AMF_OBJECT: + { + int nRes = AMF_Decode (&prop->p_vu.p_object, pBuffer, nSize, true); + if (nRes == -1) + return -1; + nSize -= nRes; + break; + } + case AMF_MOVIECLIP: + { + RTMP_Log (RTMP_LOGERROR, "AMF_MOVIECLIP reserved!"); + return -1; + break; + } + case AMF_NULL: + case AMF_UNDEFINED: + case AMF_UNSUPPORTED: + prop->p_type = AMF_NULL; + break; + case AMF_REFERENCE: + { + RTMP_Log (RTMP_LOGERROR, "AMF_REFERENCE not supported!"); + return -1; + break; + } + case AMF_ECMA_ARRAY: + { + nSize -= 4; + + /* next comes the rest, mixed array has a final 0x000009 mark and names, so its an object */ + nRes = AMF_Decode (&prop->p_vu.p_object, pBuffer + 4, nSize, true); + if (nRes == -1) + return -1; + nSize -= nRes; + prop->p_type = AMF_OBJECT; + break; + } + case AMF_OBJECT_END: + { + return -1; + break; + } + case AMF_STRICT_ARRAY: + { + unsigned int nArrayLen = AMF_DecodeInt32 (pBuffer); + nSize -= 4; + + nRes = AMF_DecodeArray (&prop->p_vu.p_object, pBuffer + 4, nSize, + nArrayLen, false); + if (nRes == -1) + return -1; + nSize -= nRes; + prop->p_type = AMF_OBJECT; + break; + } + case AMF_DATE: + { + RTMP_Log (RTMP_LOGDEBUG, "AMF_DATE"); + + if (nSize < 10) + return -1; + + prop->p_vu.p_number = AMF_DecodeNumber (pBuffer); + prop->p_UTCoffset = AMF_DecodeInt16 (pBuffer + 8); + + nSize -= 10; + break; + } + case AMF_LONG_STRING: + { + unsigned int nStringSize = AMF_DecodeInt32 (pBuffer); + if (nSize < (long) nStringSize + 4) + return -1; + AMF_DecodeLongString (pBuffer, &prop->p_vu.p_aval); + nSize -= (4 + nStringSize); + prop->p_type = AMF_STRING; + break; + } + case AMF_RECORDSET: + { + RTMP_Log (RTMP_LOGERROR, "AMF_RECORDSET reserved!"); + return -1; + break; + } + case AMF_XML_DOC: + { + RTMP_Log (RTMP_LOGERROR, "AMF_XML_DOC not supported!"); + return -1; + break; + } + case AMF_TYPED_OBJECT: + { + RTMP_Log (RTMP_LOGERROR, "AMF_TYPED_OBJECT not supported!"); + return -1; + break; + } + case AMF_AVMPLUS: + { + int nRes = AMF3_Decode (&prop->p_vu.p_object, pBuffer, nSize, true); + if (nRes == -1) + return -1; + nSize -= nRes; + prop->p_type = AMF_OBJECT; + break; + } + default: + RTMP_Log (RTMP_LOGDEBUG, "%s - unknown datatype 0x%02x, @0x%08X", + __FUNCTION__, prop->p_type, pBuffer - 1); + return -1; + } + + return nOriginalSize - nSize; +} + +void +AMFProp_Dump (AMFObjectProperty * prop) +{ + char strRes[256]; + char str[256]; + AVal name; + + if (prop->p_type == AMF_INVALID) { + RTMP_Log (RTMP_LOGDEBUG, "Property: INVALID"); + return; + } + + if (prop->p_type == AMF_NULL) { + RTMP_Log (RTMP_LOGDEBUG, "Property: NULL"); + return; + } + + if (prop->p_name.av_len) { + name = prop->p_name; + } else { + name.av_val = (char *) "no-name."; + name.av_len = sizeof ("no-name.") - 1; + } + if (name.av_len > 18) + name.av_len = 18; + + snprintf (strRes, 255, "Name: %18.*s, ", name.av_len, name.av_val); + + if (prop->p_type == AMF_OBJECT) { + RTMP_Log (RTMP_LOGDEBUG, "Property: <%sOBJECT>", strRes); + AMF_Dump (&prop->p_vu.p_object); + return; + } + + switch (prop->p_type) { + case AMF_NUMBER: + snprintf (str, 255, "NUMBER:\t%.2f", prop->p_vu.p_number); + break; + case AMF_BOOLEAN: + snprintf (str, 255, "BOOLEAN:\t%s", + prop->p_vu.p_number != 0.0 ? "TRUE" : "FALSE"); + break; + case AMF_STRING: + snprintf (str, 255, "STRING:\t%.*s", prop->p_vu.p_aval.av_len, + prop->p_vu.p_aval.av_val); + break; + case AMF_DATE: + snprintf (str, 255, "DATE:\ttimestamp: %.2f, UTC offset: %d", + prop->p_vu.p_number, prop->p_UTCoffset); + break; + default: + snprintf (str, 255, "INVALID TYPE 0x%02x", (unsigned char) prop->p_type); + } + + RTMP_Log (RTMP_LOGDEBUG, "Property: <%s%s>", strRes, str); +} + +void +AMFProp_Reset (AMFObjectProperty * prop) +{ + if (prop->p_type == AMF_OBJECT) + AMF_Reset (&prop->p_vu.p_object); + else { + prop->p_vu.p_aval.av_len = 0; + prop->p_vu.p_aval.av_val = NULL; + } + prop->p_type = AMF_INVALID; +} + +/* AMFObject */ + +char * +AMF_Encode (AMFObject * obj, char *pBuffer, char *pBufEnd) +{ + int i; + + if (pBuffer + 4 >= pBufEnd) + return NULL; + + *pBuffer++ = AMF_OBJECT; + + for (i = 0; i < obj->o_num; i++) { + char *res = AMFProp_Encode (&obj->o_props[i], pBuffer, pBufEnd); + if (res == NULL) { + RTMP_Log (RTMP_LOGERROR, + "AMF_Encode - failed to encode property in index %d", i); + break; + } else { + pBuffer = res; + } + } + + if (pBuffer + 3 >= pBufEnd) + return NULL; /* no room for the end marker */ + + pBuffer = AMF_EncodeInt24 (pBuffer, pBufEnd, AMF_OBJECT_END); + + return pBuffer; +} + +int +AMF_DecodeArray (AMFObject * obj, const char *pBuffer, int nSize, + int nArrayLen, bool bDecodeName) +{ + int nOriginalSize = nSize; + bool bError = false; + + obj->o_num = 0; + obj->o_props = NULL; + while (nArrayLen > 0) { + AMFObjectProperty prop; + int nRes; + nArrayLen--; + + nRes = AMFProp_Decode (&prop, pBuffer, nSize, bDecodeName); + if (nRes == -1) + bError = true; + else { + nSize -= nRes; + pBuffer += nRes; + AMF_AddProp (obj, &prop); + } + } + if (bError) + return -1; + + return nOriginalSize - nSize; +} + +int +AMF3_Decode (AMFObject * obj, const char *pBuffer, int nSize, bool bAMFData) +{ + int nOriginalSize = nSize; + int32_t ref; + int len; + + obj->o_num = 0; + obj->o_props = NULL; + if (bAMFData) { + if (*pBuffer != AMF3_OBJECT) + RTMP_Log (RTMP_LOGERROR, + "AMF3 Object encapsulated in AMF stream does not start with AMF3_OBJECT!"); + pBuffer++; + nSize--; + } + + ref = 0; + len = AMF3ReadInteger (pBuffer, &ref); + pBuffer += len; + nSize -= len; + + if ((ref & 1) == 0) { /* object reference, 0xxx */ + uint32_t objectIndex = (ref >> 1); + + RTMP_Log (RTMP_LOGDEBUG, "Object reference, index: %d", objectIndex); + } else { /* object instance */ + + int32_t classRef = (ref >> 1); + + AMF3ClassDef cd = { {0, 0} + }; + AMFObjectProperty prop; + + if ((classRef & 0x1) == 0) { /* class reference */ + uint32_t classIndex = (classRef >> 1); + RTMP_Log (RTMP_LOGDEBUG, "Class reference: %d", classIndex); + } else { + int32_t classExtRef = (classRef >> 1); + int i; + + cd.cd_externalizable = (classExtRef & 0x1) == 1; + cd.cd_dynamic = ((classExtRef >> 1) & 0x1) == 1; + + cd.cd_num = classExtRef >> 2; + + /* class name */ + + len = AMF3ReadString (pBuffer, &cd.cd_name); + nSize -= len; + pBuffer += len; + + /*std::string str = className; */ + + RTMP_Log (RTMP_LOGDEBUG, + "Class name: %s, externalizable: %d, dynamic: %d, classMembers: %d", + cd.cd_name.av_val, cd.cd_externalizable, cd.cd_dynamic, cd.cd_num); + + for (i = 0; i < cd.cd_num; i++) { + AVal memberName; + len = AMF3ReadString (pBuffer, &memberName); + RTMP_Log (RTMP_LOGDEBUG, "Member: %s", memberName.av_val); + AMF3CD_AddProp (&cd, &memberName); + nSize -= len; + pBuffer += len; + } + } + + /* add as referencable object */ + + if (cd.cd_externalizable) { + int nRes; + AVal name = AVC ("DEFAULT_ATTRIBUTE"); + + RTMP_Log (RTMP_LOGDEBUG, "Externalizable, TODO check"); + + nRes = AMF3Prop_Decode (&prop, pBuffer, nSize, false); + if (nRes == -1) + RTMP_Log (RTMP_LOGDEBUG, "%s, failed to decode AMF3 property!", + __FUNCTION__); + else { + nSize -= nRes; + pBuffer += nRes; + } + + AMFProp_SetName (&prop, &name); + AMF_AddProp (obj, &prop); + } else { + int nRes, i; + for (i = 0; i < cd.cd_num; i++) { /* non-dynamic */ + nRes = AMF3Prop_Decode (&prop, pBuffer, nSize, false); + if (nRes == -1) + RTMP_Log (RTMP_LOGDEBUG, "%s, failed to decode AMF3 property!", + __FUNCTION__); + + AMFProp_SetName (&prop, AMF3CD_GetProp (&cd, i)); + AMF_AddProp (obj, &prop); + + pBuffer += nRes; + nSize -= nRes; + } + if (cd.cd_dynamic) { + int len = 0; + + do { + nRes = AMF3Prop_Decode (&prop, pBuffer, nSize, true); + AMF_AddProp (obj, &prop); + + pBuffer += nRes; + nSize -= nRes; + + len = prop.p_name.av_len; + } + while (len > 0); + } + } + RTMP_Log (RTMP_LOGDEBUG, "class object!"); + } + return nOriginalSize - nSize; +} + +int +AMF_Decode (AMFObject * obj, const char *pBuffer, int nSize, bool bDecodeName) +{ + int nOriginalSize = nSize; + bool bError = false; /* if there is an error while decoding - try to at least find the end mark AMF_OBJECT_END */ + + obj->o_num = 0; + obj->o_props = NULL; + while (nSize > 0) { + AMFObjectProperty prop; + int nRes; + + if (nSize >= 3 && AMF_DecodeInt24 (pBuffer) == AMF_OBJECT_END) { + nSize -= 3; + bError = false; + break; + } + + if (bError) { + RTMP_Log (RTMP_LOGERROR, + "DECODING ERROR, IGNORING BYTES UNTIL NEXT KNOWN PATTERN!"); + nSize--; + pBuffer++; + continue; + } + + nRes = AMFProp_Decode (&prop, pBuffer, nSize, bDecodeName); + if (nRes == -1) + bError = true; + else { + nSize -= nRes; + pBuffer += nRes; + AMF_AddProp (obj, &prop); + } + } + + if (bError) + return -1; + + return nOriginalSize - nSize; +} + +void +AMF_AddProp (AMFObject * obj, const AMFObjectProperty * prop) +{ + if (!(obj->o_num & 0x0f)) + obj->o_props = + realloc (obj->o_props, (obj->o_num + 16) * sizeof (AMFObjectProperty)); + obj->o_props[obj->o_num++] = *prop; +} + +int +AMF_CountProp (AMFObject * obj) +{ + return obj->o_num; +} + +AMFObjectProperty * +AMF_GetProp (AMFObject * obj, const AVal * name, int nIndex) +{ + if (nIndex >= 0) { + if (nIndex <= obj->o_num) + return &obj->o_props[nIndex]; + } else { + int n; + for (n = 0; n < obj->o_num; n++) { + if (AVMATCH (&obj->o_props[n].p_name, name)) + return &obj->o_props[n]; + } + } + + return (AMFObjectProperty *) & AMFProp_Invalid; +} + +void +AMF_Dump (AMFObject * obj) +{ + int n; + RTMP_Log (RTMP_LOGDEBUG, "(object begin)"); + for (n = 0; n < obj->o_num; n++) { + AMFProp_Dump (&obj->o_props[n]); + } + RTMP_Log (RTMP_LOGDEBUG, "(object end)"); +} + +void +AMF_Reset (AMFObject * obj) +{ + int n; + for (n = 0; n < obj->o_num; n++) { + AMFProp_Reset (&obj->o_props[n]); + } + free (obj->o_props); + obj->o_props = NULL; + obj->o_num = 0; +} + + +/* AMF3ClassDefinition */ + +void +AMF3CD_AddProp (AMF3ClassDef * cd, AVal * prop) +{ + if (!(cd->cd_num & 0x0f)) + cd->cd_props = realloc (cd->cd_props, (cd->cd_num + 16) * sizeof (AVal)); + cd->cd_props[cd->cd_num++] = *prop; +} + +AVal * +AMF3CD_GetProp (AMF3ClassDef * cd, int nIndex) +{ + if (nIndex >= cd->cd_num) + return (AVal *) & AV_empty; + return &cd->cd_props[nIndex]; +} diff --git a/gst/rtmp/amf.h b/gst/rtmp/amf.h new file mode 100644 index 0000000000..82b08f8585 --- /dev/null +++ b/gst/rtmp/amf.h @@ -0,0 +1,168 @@ +#ifndef __AMF_H__ +#define __AMF_H__ +/* + * Copyright (C) 2005-2008 Team XBMC + * http://www.xbmc.org + * Copyright (C) 2008-2009 Andrej Stepanchuk + * Copyright (C) 2009-2010 Howard Chu + * + * This file is part of librtmp. + * + * librtmp is free software; you can redistribute it and/or modify + * it under the terms of the GNU Lesser General Public License as + * published by the Free Software Foundation; either version 2.1, + * or (at your option) any later version. + * + * librtmp 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 Lesser General Public License + * along with librtmp see the file COPYING. If not, write to + * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. + * http://www.gnu.org/copyleft/lgpl.html + */ + +#include + +#ifdef _XBOX + +#ifndef __cplusplus +#define bool _Bool +typedef unsigned char _Bool; +#define false 0 +#define true 1 +#endif + +#else +#include +#endif + +#ifdef __cplusplus +extern "C" +{ +#endif + + typedef enum + { AMF_NUMBER = 0, 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 */ + AMF_INVALID = 0xff + } AMFDataType; + + typedef enum + { AMF3_UNDEFINED = 0, AMF3_NULL, AMF3_FALSE, AMF3_TRUE, + AMF3_INTEGER, AMF3_DOUBLE, AMF3_STRING, AMF3_XML_DOC, AMF3_DATE, + AMF3_ARRAY, AMF3_OBJECT, AMF3_XML, AMF3_BYTE_ARRAY + } AMF3DataType; + + typedef struct AVal + { + char *av_val; + int av_len; + } AVal; +#define AVC(str) {(char *) str,sizeof(str)-1} +#define AVMATCH(a1,a2) ((a1)->av_len == (a2)->av_len && !memcmp((a1)->av_val,(a2)->av_val,(a1)->av_len)) + + struct AMFObjectProperty; + + typedef struct AMFObject + { + int o_num; + struct AMFObjectProperty *o_props; + } AMFObject; + + typedef struct AMFObjectProperty + { + AVal p_name; + AMFDataType p_type; + union + { + double p_number; + AVal p_aval; + AMFObject p_object; + } p_vu; + int16_t p_UTCoffset; + } AMFObjectProperty; + + char *AMF_EncodeString(char *output, char *outend, const AVal * str); + char *AMF_EncodeNumber(char *output, char *outend, double dVal); + char *AMF_EncodeInt16(char *output, char *outend, short nVal); + char *AMF_EncodeInt24(char *output, char *outend, int nVal); + char *AMF_EncodeInt32(char *output, char *outend, int nVal); + char *AMF_EncodeBoolean(char *output, char *outend, bool bVal); + + /* Shortcuts for AMFProp_Encode */ + char *AMF_EncodeNamedString(char *output, char *outend, const AVal * name, const AVal * value); + char *AMF_EncodeNamedNumber(char *output, char *outend, const AVal * name, double dVal); + char *AMF_EncodeNamedBoolean(char *output, char *outend, const AVal * name, bool bVal); + + unsigned short AMF_DecodeInt16(const char *data); + unsigned int AMF_DecodeInt24(const char *data); + unsigned int AMF_DecodeInt32(const char *data); + void AMF_DecodeString(const char *data, AVal * str); + void AMF_DecodeLongString(const char *data, AVal * str); + bool AMF_DecodeBoolean(const char *data); + double AMF_DecodeNumber(const char *data); + + char *AMF_Encode(AMFObject * obj, char *pBuffer, char *pBufEnd); + int AMF_Decode(AMFObject * obj, const char *pBuffer, int nSize, + bool bDecodeName); + int AMF_DecodeArray(AMFObject * obj, const char *pBuffer, int nSize, + int nArrayLen, bool bDecodeName); + int AMF3_Decode(AMFObject * obj, const char *pBuffer, int nSize, + bool bDecodeName); + void AMF_Dump(AMFObject * obj); + void AMF_Reset(AMFObject * obj); + + void AMF_AddProp(AMFObject * obj, const AMFObjectProperty * prop); + int AMF_CountProp(AMFObject * obj); + AMFObjectProperty *AMF_GetProp(AMFObject * obj, const AVal * name, + int nIndex); + + AMFDataType AMFProp_GetType(AMFObjectProperty * prop); + void AMFProp_SetNumber(AMFObjectProperty * prop, double dval); + void AMFProp_SetBoolean(AMFObjectProperty * prop, bool bflag); + void AMFProp_SetString(AMFObjectProperty * prop, AVal * str); + void AMFProp_SetObject(AMFObjectProperty * prop, AMFObject * obj); + + void AMFProp_GetName(AMFObjectProperty * prop, AVal * name); + void AMFProp_SetName(AMFObjectProperty * prop, AVal * name); + double AMFProp_GetNumber(AMFObjectProperty * prop); + bool AMFProp_GetBoolean(AMFObjectProperty * prop); + void AMFProp_GetString(AMFObjectProperty * prop, AVal * str); + void AMFProp_GetObject(AMFObjectProperty * prop, AMFObject * obj); + + bool AMFProp_IsValid(AMFObjectProperty * prop); + + char *AMFProp_Encode(AMFObjectProperty * prop, char *pBuffer, char *pBufEnd); + int AMF3Prop_Decode(AMFObjectProperty * prop, const char *pBuffer, + int nSize, bool bDecodeName); + int AMFProp_Decode(AMFObjectProperty * prop, const char *pBuffer, + int nSize, bool bDecodeName); + + void AMFProp_Dump(AMFObjectProperty * prop); + void AMFProp_Reset(AMFObjectProperty * prop); + + typedef struct AMF3ClassDef + { + AVal cd_name; + char cd_externalizable; + char cd_dynamic; + int cd_num; + AVal *cd_props; + } AMF3ClassDef; + + void AMF3CD_AddProp(AMF3ClassDef * cd, AVal * prop); + AVal *AMF3CD_GetProp(AMF3ClassDef * cd, int idx); + +#ifdef __cplusplus +} +#endif + +#endif /* __AMF_H__ */ diff --git a/gst/rtmp/bytes.h b/gst/rtmp/bytes.h new file mode 100644 index 0000000000..de412a26ff --- /dev/null +++ b/gst/rtmp/bytes.h @@ -0,0 +1,90 @@ +/* + * Copyright (C) 2005-2008 Team XBMC + * http://www.xbmc.org + * Copyright (C) 2008-2009 Andrej Stepanchuk + * Copyright (C) 2009-2010 Howard Chu + * + * This file is part of librtmp. + * + * librtmp is free software; you can redistribute it and/or modify + * it under the terms of the GNU Lesser General Public License as + * published by the Free Software Foundation; either version 2.1, + * or (at your option) any later version. + * + * librtmp 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 Lesser General Public License + * along with librtmp see the file COPYING. If not, write to + * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. + * http://www.gnu.org/copyleft/lgpl.html + */ + +#ifndef __BYTES_H__ +#define __BYTES_H__ + +#include + +#ifdef _WIN32 +/* Windows is little endian only */ +#define __LITTLE_ENDIAN 1234 +#define __BIG_ENDIAN 4321 +#define __BYTE_ORDER __LITTLE_ENDIAN +#define __FLOAT_WORD_ORDER __BYTE_ORDER + +typedef unsigned char uint8_t; + +#else /* !_WIN32 */ + +#include + +#if defined(BYTE_ORDER) && !defined(__BYTE_ORDER) +#define __BYTE_ORDER BYTE_ORDER +#endif + +#if defined(BIG_ENDIAN) && !defined(__BIG_ENDIAN) +#define __BIG_ENDIAN BIG_ENDIAN +#endif + +#if defined(LITTLE_ENDIAN) && !defined(__LITTLE_ENDIAN) +#define __LITTLE_ENDIAN LITTLE_ENDIAN +#endif + +#endif /* !_WIN32 */ + +/* define default endianness */ +#ifndef __LITTLE_ENDIAN +#define __LITTLE_ENDIAN 1234 +#endif + +#ifndef __BIG_ENDIAN +#define __BIG_ENDIAN 4321 +#endif + +#ifndef __BYTE_ORDER +#warning "Byte order not defined on your system, assuming little endian!" +#define __BYTE_ORDER __LITTLE_ENDIAN +#endif + +/* ok, we assume to have the same float word order and byte order if float word order is not defined */ +#ifndef __FLOAT_WORD_ORDER +#warning "Float word order not defined, assuming the same as byte order!" +#define __FLOAT_WORD_ORDER __BYTE_ORDER +#endif + +#if !defined(__BYTE_ORDER) || !defined(__FLOAT_WORD_ORDER) +#error "Undefined byte or float word order!" +#endif + +#if __FLOAT_WORD_ORDER != __BIG_ENDIAN && __FLOAT_WORD_ORDER != __LITTLE_ENDIAN +#error "Unknown/unsupported float word order!" +#endif + +#if __BYTE_ORDER != __BIG_ENDIAN && __BYTE_ORDER != __LITTLE_ENDIAN +#error "Unknown/unsupported byte order!" +#endif + +#endif + diff --git a/gst/rtmp/dh.h b/gst/rtmp/dh.h new file mode 100644 index 0000000000..9aea78d81b --- /dev/null +++ b/gst/rtmp/dh.h @@ -0,0 +1,333 @@ +/* RTMPDump - Diffie-Hellmann Key Exchange + * Copyright (C) 2009 Andrej Stepanchuk + * Copyright (C) 2009-2010 Howard Chu + * + * This file is part of librtmp. + * + * librtmp is free software; you can redistribute it and/or modify + * it under the terms of the GNU Lesser General Public License as + * published by the Free Software Foundation; either version 2.1, + * or (at your option) any later version. + * + * librtmp 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 Lesser General Public License + * along with librtmp see the file COPYING. If not, write to + * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. + * http://www.gnu.org/copyleft/lgpl.html + */ + +#include +#include +#include +#include +#include + +#ifdef USE_POLARSSL +#include +typedef mpi * MP_t; +#define MP_new(m) m = malloc(sizeof(mpi)); mpi_init(m, NULL) +#define MP_set_w(mpi, w) mpi_lset(mpi, w) +#define MP_cmp(u, v) mpi_cmp_mpi(u, v) +#define MP_set(u, v) mpi_copy(u, v) +#define MP_sub_w(mpi, w) mpi_sub_int(mpi, mpi, w) +#define MP_cmp_1(mpi) mpi_cmp_int(mpi, 1) +#define MP_modexp(r, y, q, p) mpi_exp_mod(r, y, q, p, NULL) +#define MP_free(mpi) mpi_free(mpi, NULL); free(mpi) +#define MP_gethex(u, hex, res) MP_new(u); res = mpi_read_string(u, 16, hex) == 0 +#define MP_bytes(u) mpi_size(u) +#define MP_setbin(u,buf,len) mpi_write_binary(u,buf,len) +#define MP_getbin(u,buf,len) MP_new(u); mpi_read_binary(u,buf,len) + +typedef struct MDH { + MP_t p; + MP_t g; + MP_t pub_key; + MP_t priv_key; + long length; + dhm_context ctx; +} MDH; + +#define MDH_new() calloc(1,sizeof(MDH)) +#define MDH_free(vp) {MDH *dh = vp; dhm_free(&dh->ctx); MP_free(dh->p); MP_free(dh->g); MP_free(dh->pub_key); MP_free(dh->priv_key); free(dh);} + +static int MDH_generate_key(MDH *dh) +{ + unsigned char out[2]; + MP_set(&dh->ctx.P, dh->p); + MP_set(&dh->ctx.G, dh->g); + dh->ctx.len = 128; + dhm_make_public(&dh->ctx, 1024, out, 1, havege_rand, &RTMP_TLS_ctx->hs); + MP_new(dh->pub_key); + MP_new(dh->priv_key); + MP_set(dh->pub_key, &dh->ctx.GX); + MP_set(dh->priv_key, &dh->ctx.X); + return 1; +} + +static int MDH_compute_key(uint8_t *secret, size_t len, MP_t pub, MDH *dh) +{ + int n = len; + MP_set(&dh->ctx.GY, pub); + dhm_calc_secret(&dh->ctx, secret, &n); + return 0; +} + +#elif defined(USE_GNUTLS) +#include +typedef gcry_mpi_t MP_t; +#define MP_new(m) m = gcry_mpi_new(1) +#define MP_set_w(mpi, w) gcry_mpi_set_ui(mpi, w) +#define MP_cmp(u, v) gcry_mpi_cmp(u, v) +#define MP_set(u, v) gcry_mpi_set(u, v) +#define MP_sub_w(mpi, w) gcry_mpi_sub_ui(mpi, mpi, w) +#define MP_cmp_1(mpi) gcry_mpi_cmp_ui(mpi, 1) +#define MP_modexp(r, y, q, p) gcry_mpi_powm(r, y, q, p) +#define MP_free(mpi) gcry_mpi_release(mpi) +#define MP_gethex(u, hex, res) res = (gcry_mpi_scan(&u, GCRYMPI_FMT_HEX, hex, 0, 0) == 0) +#define MP_bytes(u) (gcry_mpi_get_nbits(u) + 7) / 8 +#define MP_setbin(u,buf,len) gcry_mpi_print(GCRYMPI_FMT_USG,buf,len,NULL,u) +#define MP_getbin(u,buf,len) gcry_mpi_scan(&u,GCRYMPI_FMT_USG,buf,len,NULL) + +typedef struct MDH { + MP_t p; + MP_t g; + MP_t pub_key; + MP_t priv_key; + long length; +} MDH; + +#define MDH_new() calloc(1,sizeof(MDH)) +#define MDH_free(dh) do {MP_free(((MDH*)(dh))->p); MP_free(((MDH*)(dh))->g); MP_free(((MDH*)(dh))->pub_key); MP_free(((MDH*)(dh))->priv_key); free(dh);} while(0) + +extern MP_t gnutls_calc_dh_secret(MP_t *priv, MP_t g, MP_t p); +extern MP_t gnutls_calc_dh_key(MP_t y, MP_t x, MP_t p); + +#define MDH_generate_key(dh) (dh->pub_key = gnutls_calc_dh_secret(&dh->priv_key, dh->g, dh->p)) +static int MDH_compute_key(uint8_t *secret, size_t len, MP_t pub, MDH *dh) +{ + MP_t sec = gnutls_calc_dh_key(pub, dh->priv_key, dh->p); + if (sec) + { + MP_setbin(sec, secret, len); + MP_free(sec); + return 0; + } + else + return -1; +} + +#else /* USE_OPENSSL */ +#include +#include + +typedef BIGNUM * MP_t; +#define MP_new(m) m = BN_new() +#define MP_set_w(mpi, w) BN_set_word(mpi, w) +#define MP_cmp(u, v) BN_cmp(u, v) +#define MP_set(u, v) BN_copy(u, v) +#define MP_sub_w(mpi, w) BN_sub_word(mpi, w) +#define MP_cmp_1(mpi) BN_cmp(mpi, BN_value_one()) +#define MP_modexp(r, y, q, p) do {BN_CTX *ctx = BN_CTX_new(); BN_mod_exp(r, y, q, p, ctx); BN_CTX_free(ctx);} while(0) +#define MP_free(mpi) BN_free(mpi) +#define MP_gethex(u, hex, res) res = BN_hex2bn(&u, hex) +#define MP_bytes(u) BN_num_bytes(u) +#define MP_setbin(u,buf,len) BN_bn2bin(u,buf) +#define MP_getbin(u,buf,len) u = BN_bin2bn(buf,len,0) + +#define MDH DH +#define MDH_new() DH_new() +#define MDH_free(dh) DH_free(dh) +#define MDH_generate_key(dh) DH_generate_key(dh) +#define MDH_compute_key(secret, seclen, pub, dh) DH_compute_key(secret, pub, dh) + +#endif + +#include "log.h" +#include "dhgroups.h" + +/* RFC 2631, Section 2.1.5, http://www.ietf.org/rfc/rfc2631.txt */ +static bool +isValidPublicKey(MP_t y, MP_t p, MP_t q) +{ + int ret = true; + MP_t bn; + assert(y); + + MP_new(bn); + assert(bn); + + /* y must lie in [2,p-1] */ + MP_set_w(bn, 1); + if (MP_cmp(y, bn) < 0) + { + RTMP_Log(RTMP_LOGERROR, "DH public key must be at least 2"); + ret = false; + goto failed; + } + + /* bn = p-2 */ + MP_set(bn, p); + MP_sub_w(bn, 1); + if (MP_cmp(y, bn) > 0) + { + RTMP_Log(RTMP_LOGERROR, "DH public key must be at most p-2"); + ret = false; + goto failed; + } + + /* Verify with Sophie-Germain prime + * + * This is a nice test to make sure the public key position is calculated + * correctly. This test will fail in about 50% of the cases if applied to + * random data. + */ + if (q) + { + /* y must fulfill y^q mod p = 1 */ + MP_modexp(bn, y, q, p); + + if (MP_cmp_1(bn) != 0) + { + RTMP_Log(RTMP_LOGWARNING, "DH public key does not fulfill y^q mod p = 1"); + } + } + +failed: + MP_free(bn); + return ret; +} + +static MDH * +DHInit(int nKeyBits) +{ + size_t res; + MDH *dh = MDH_new(); + + if (!dh) + goto failed; + + MP_new(dh->g); + + if (!dh->g) + goto failed; + + MP_gethex(dh->p, P1024, res); /* prime P1024, see dhgroups.h */ + if (!res) + { + goto failed; + } + + MP_set_w(dh->g, 2); /* base 2 */ + + dh->length = nKeyBits; + return dh; + +failed: + if (dh) + MDH_free(dh); + + return 0; +} + +static int +DHGenerateKey(MDH *dh) +{ + size_t res = 0; + if (!dh) + return 0; + + while (!res) + { + MP_t q1 = NULL; + + if (!MDH_generate_key(dh)) + return 0; + + MP_gethex(q1, Q1024, res); + assert(res); + + res = isValidPublicKey(dh->pub_key, dh->p, q1); + if (!res) + { + MP_free(dh->pub_key); + MP_free(dh->priv_key); + dh->pub_key = dh->priv_key = 0; + } + + MP_free(q1); + } + return 1; +} + +/* fill pubkey with the public key in BIG ENDIAN order + * 00 00 00 00 00 x1 x2 x3 ..... + */ + +static int +DHGetPublicKey(MDH *dh, uint8_t *pubkey, size_t nPubkeyLen) +{ + int len; + if (!dh || !dh->pub_key) + return 0; + + len = MP_bytes(dh->pub_key); + if (len <= 0 || len > (int) nPubkeyLen) + return 0; + + memset(pubkey, 0, nPubkeyLen); + MP_setbin(dh->pub_key, pubkey + (nPubkeyLen - len), len); + return 1; +} + +#if 0 /* unused */ +static int +DHGetPrivateKey(MDH *dh, uint8_t *privkey, size_t nPrivkeyLen) +{ + if (!dh || !dh->priv_key) + return 0; + + int len = MP_bytes(dh->priv_key); + if (len <= 0 || len > (int) nPrivkeyLen) + return 0; + + memset(privkey, 0, nPrivkeyLen); + MP_setbin(dh->priv_key, privkey + (nPrivkeyLen - len), len); + return 1; +} +#endif + +/* computes the shared secret key from the private MDH value and the + * other party's public key (pubkey) + */ +static int +DHComputeSharedSecretKey(MDH *dh, uint8_t *pubkey, size_t nPubkeyLen, + uint8_t *secret) +{ + MP_t q1 = NULL, pubkeyBn = NULL; + size_t len; + int res; + + if (!dh || !secret || nPubkeyLen >= INT_MAX) + return -1; + + MP_getbin(pubkeyBn, pubkey, nPubkeyLen); + if (!pubkeyBn) + return -1; + + MP_gethex(q1, Q1024, len); + assert(len); + + if (isValidPublicKey(pubkeyBn, dh->p, q1)) + res = MDH_compute_key(secret, nPubkeyLen, pubkeyBn, dh); + else + res = -1; + + MP_free(q1); + MP_free(pubkeyBn); + + return res; +} diff --git a/gst/rtmp/dhgroups.h b/gst/rtmp/dhgroups.h new file mode 100644 index 0000000000..d5c90e1449 --- /dev/null +++ b/gst/rtmp/dhgroups.h @@ -0,0 +1,198 @@ +/* librtmp - Diffie-Hellmann Key Exchange + * Copyright (C) 2009 Andrej Stepanchuk + * + * This file is part of librtmp. + * + * librtmp is free software; you can redistribute it and/or modify + * it under the terms of the GNU Lesser General Public License as + * published by the Free Software Foundation; either version 2.1, + * or (at your option) any later version. + * + * librtmp 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 Lesser General Public License + * along with librtmp see the file COPYING. If not, write to + * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. + * http://www.gnu.org/copyleft/lgpl.html + */ + +/* from RFC 3526, see http://www.ietf.org/rfc/rfc3526.txt */ + +/* 2^768 - 2 ^704 - 1 + 2^64 * { [2^638 pi] + 149686 } */ +#define P768 \ + "FFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD1" \ + "29024E088A67CC74020BBEA63B139B22514A08798E3404DD" \ + "EF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245" \ + "E485B576625E7EC6F44C42E9A63A3620FFFFFFFFFFFFFFFF" + +/* 2^1024 - 2^960 - 1 + 2^64 * { [2^894 pi] + 129093 } */ +#define P1024 \ + "FFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD1" \ + "29024E088A67CC74020BBEA63B139B22514A08798E3404DD" \ + "EF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245" \ + "E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7ED" \ + "EE386BFB5A899FA5AE9F24117C4B1FE649286651ECE65381" \ + "FFFFFFFFFFFFFFFF" + +/* Group morder largest prime factor: */ +#define Q1024 \ + "7FFFFFFFFFFFFFFFE487ED5110B4611A62633145C06E0E68" \ + "948127044533E63A0105DF531D89CD9128A5043CC71A026E" \ + "F7CA8CD9E69D218D98158536F92F8A1BA7F09AB6B6A8E122" \ + "F242DABB312F3F637A262174D31BF6B585FFAE5B7A035BF6" \ + "F71C35FDAD44CFD2D74F9208BE258FF324943328F67329C0" \ + "FFFFFFFFFFFFFFFF" + +/* 2^1536 - 2^1472 - 1 + 2^64 * { [2^1406 pi] + 741804 } */ +#define P1536 \ + "FFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD1" \ + "29024E088A67CC74020BBEA63B139B22514A08798E3404DD" \ + "EF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245" \ + "E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7ED" \ + "EE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3D" \ + "C2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F" \ + "83655D23DCA3AD961C62F356208552BB9ED529077096966D" \ + "670C354E4ABC9804F1746C08CA237327FFFFFFFFFFFFFFFF" + +/* 2^2048 - 2^1984 - 1 + 2^64 * { [2^1918 pi] + 124476 } */ +#define P2048 \ + "FFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD1" \ + "29024E088A67CC74020BBEA63B139B22514A08798E3404DD" \ + "EF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245" \ + "E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7ED" \ + "EE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3D" \ + "C2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F" \ + "83655D23DCA3AD961C62F356208552BB9ED529077096966D" \ + "670C354E4ABC9804F1746C08CA18217C32905E462E36CE3B" \ + "E39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9" \ + "DE2BCBF6955817183995497CEA956AE515D2261898FA0510" \ + "15728E5A8AACAA68FFFFFFFFFFFFFFFF" + +/* 2^3072 - 2^3008 - 1 + 2^64 * { [2^2942 pi] + 1690314 } */ +#define P3072 \ + "FFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD1" \ + "29024E088A67CC74020BBEA63B139B22514A08798E3404DD" \ + "EF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245" \ + "E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7ED" \ + "EE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3D" \ + "C2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F" \ + "83655D23DCA3AD961C62F356208552BB9ED529077096966D" \ + "670C354E4ABC9804F1746C08CA18217C32905E462E36CE3B" \ + "E39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9" \ + "DE2BCBF6955817183995497CEA956AE515D2261898FA0510" \ + "15728E5A8AAAC42DAD33170D04507A33A85521ABDF1CBA64" \ + "ECFB850458DBEF0A8AEA71575D060C7DB3970F85A6E1E4C7" \ + "ABF5AE8CDB0933D71E8C94E04A25619DCEE3D2261AD2EE6B" \ + "F12FFA06D98A0864D87602733EC86A64521F2B18177B200C" \ + "BBE117577A615D6C770988C0BAD946E208E24FA074E5AB31" \ + "43DB5BFCE0FD108E4B82D120A93AD2CAFFFFFFFFFFFFFFFF" + +/* 2^4096 - 2^4032 - 1 + 2^64 * { [2^3966 pi] + 240904 } */ +#define P4096 \ + "FFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD1" \ + "29024E088A67CC74020BBEA63B139B22514A08798E3404DD" \ + "EF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245" \ + "E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7ED" \ + "EE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3D" \ + "C2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F" \ + "83655D23DCA3AD961C62F356208552BB9ED529077096966D" \ + "670C354E4ABC9804F1746C08CA18217C32905E462E36CE3B" \ + "E39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9" \ + "DE2BCBF6955817183995497CEA956AE515D2261898FA0510" \ + "15728E5A8AAAC42DAD33170D04507A33A85521ABDF1CBA64" \ + "ECFB850458DBEF0A8AEA71575D060C7DB3970F85A6E1E4C7" \ + "ABF5AE8CDB0933D71E8C94E04A25619DCEE3D2261AD2EE6B" \ + "F12FFA06D98A0864D87602733EC86A64521F2B18177B200C" \ + "BBE117577A615D6C770988C0BAD946E208E24FA074E5AB31" \ + "43DB5BFCE0FD108E4B82D120A92108011A723C12A787E6D7" \ + "88719A10BDBA5B2699C327186AF4E23C1A946834B6150BDA" \ + "2583E9CA2AD44CE8DBBBC2DB04DE8EF92E8EFC141FBECAA6" \ + "287C59474E6BC05D99B2964FA090C3A2233BA186515BE7ED" \ + "1F612970CEE2D7AFB81BDD762170481CD0069127D5B05AA9" \ + "93B4EA988D8FDDC186FFB7DC90A6C08F4DF435C934063199" \ + "FFFFFFFFFFFFFFFF" + +/* 2^6144 - 2^6080 - 1 + 2^64 * { [2^6014 pi] + 929484 } */ +#define P6144 \ + "FFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD1" \ + "29024E088A67CC74020BBEA63B139B22514A08798E3404DD" \ + "EF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245" \ + "E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7ED" \ + "EE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3D" \ + "C2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F" \ + "83655D23DCA3AD961C62F356208552BB9ED529077096966D" \ + "670C354E4ABC9804F1746C08CA18217C32905E462E36CE3B" \ + "E39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9" \ + "DE2BCBF6955817183995497CEA956AE515D2261898FA0510" \ + "15728E5A8AAAC42DAD33170D04507A33A85521ABDF1CBA64" \ + "ECFB850458DBEF0A8AEA71575D060C7DB3970F85A6E1E4C7" \ + "ABF5AE8CDB0933D71E8C94E04A25619DCEE3D2261AD2EE6B" \ + "F12FFA06D98A0864D87602733EC86A64521F2B18177B200C" \ + "BBE117577A615D6C770988C0BAD946E208E24FA074E5AB31" \ + "43DB5BFCE0FD108E4B82D120A92108011A723C12A787E6D7" \ + "88719A10BDBA5B2699C327186AF4E23C1A946834B6150BDA" \ + "2583E9CA2AD44CE8DBBBC2DB04DE8EF92E8EFC141FBECAA6" \ + "287C59474E6BC05D99B2964FA090C3A2233BA186515BE7ED" \ + "1F612970CEE2D7AFB81BDD762170481CD0069127D5B05AA9" \ + "93B4EA988D8FDDC186FFB7DC90A6C08F4DF435C934028492" \ + "36C3FAB4D27C7026C1D4DCB2602646DEC9751E763DBA37BD" \ + "F8FF9406AD9E530EE5DB382F413001AEB06A53ED9027D831" \ + "179727B0865A8918DA3EDBEBCF9B14ED44CE6CBACED4BB1B" \ + "DB7F1447E6CC254B332051512BD7AF426FB8F401378CD2BF" \ + "5983CA01C64B92ECF032EA15D1721D03F482D7CE6E74FEF6" \ + "D55E702F46980C82B5A84031900B1C9E59E7C97FBEC7E8F3" \ + "23A97A7E36CC88BE0F1D45B7FF585AC54BD407B22B4154AA" \ + "CC8F6D7EBF48E1D814CC5ED20F8037E0A79715EEF29BE328" \ + "06A1D58BB7C5DA76F550AA3D8A1FBFF0EB19CCB1A313D55C" \ + "DA56C9EC2EF29632387FE8D76E3C0468043E8F663F4860EE" \ + "12BF2D5B0B7474D6E694F91E6DCC4024FFFFFFFFFFFFFFFF" + +/* 2^8192 - 2^8128 - 1 + 2^64 * { [2^8062 pi] + 4743158 } */ +#define P8192 \ + "FFFFFFFFFFFFFFFFC90FDAA22168C234C4C6628B80DC1CD1" \ + "29024E088A67CC74020BBEA63B139B22514A08798E3404DD" \ + "EF9519B3CD3A431B302B0A6DF25F14374FE1356D6D51C245" \ + "E485B576625E7EC6F44C42E9A637ED6B0BFF5CB6F406B7ED" \ + "EE386BFB5A899FA5AE9F24117C4B1FE649286651ECE45B3D" \ + "C2007CB8A163BF0598DA48361C55D39A69163FA8FD24CF5F" \ + "83655D23DCA3AD961C62F356208552BB9ED529077096966D" \ + "670C354E4ABC9804F1746C08CA18217C32905E462E36CE3B" \ + "E39E772C180E86039B2783A2EC07A28FB5C55DF06F4C52C9" \ + "DE2BCBF6955817183995497CEA956AE515D2261898FA0510" \ + "15728E5A8AAAC42DAD33170D04507A33A85521ABDF1CBA64" \ + "ECFB850458DBEF0A8AEA71575D060C7DB3970F85A6E1E4C7" \ + "ABF5AE8CDB0933D71E8C94E04A25619DCEE3D2261AD2EE6B" \ + "F12FFA06D98A0864D87602733EC86A64521F2B18177B200C" \ + "BBE117577A615D6C770988C0BAD946E208E24FA074E5AB31" \ + "43DB5BFCE0FD108E4B82D120A92108011A723C12A787E6D7" \ + "88719A10BDBA5B2699C327186AF4E23C1A946834B6150BDA" \ + "2583E9CA2AD44CE8DBBBC2DB04DE8EF92E8EFC141FBECAA6" \ + "287C59474E6BC05D99B2964FA090C3A2233BA186515BE7ED" \ + "1F612970CEE2D7AFB81BDD762170481CD0069127D5B05AA9" \ + "93B4EA988D8FDDC186FFB7DC90A6C08F4DF435C934028492" \ + "36C3FAB4D27C7026C1D4DCB2602646DEC9751E763DBA37BD" \ + "F8FF9406AD9E530EE5DB382F413001AEB06A53ED9027D831" \ + "179727B0865A8918DA3EDBEBCF9B14ED44CE6CBACED4BB1B" \ + "DB7F1447E6CC254B332051512BD7AF426FB8F401378CD2BF" \ + "5983CA01C64B92ECF032EA15D1721D03F482D7CE6E74FEF6" \ + "D55E702F46980C82B5A84031900B1C9E59E7C97FBEC7E8F3" \ + "23A97A7E36CC88BE0F1D45B7FF585AC54BD407B22B4154AA" \ + "CC8F6D7EBF48E1D814CC5ED20F8037E0A79715EEF29BE328" \ + "06A1D58BB7C5DA76F550AA3D8A1FBFF0EB19CCB1A313D55C" \ + "DA56C9EC2EF29632387FE8D76E3C0468043E8F663F4860EE" \ + "12BF2D5B0B7474D6E694F91E6DBE115974A3926F12FEE5E4" \ + "38777CB6A932DF8CD8BEC4D073B931BA3BC832B68D9DD300" \ + "741FA7BF8AFC47ED2576F6936BA424663AAB639C5AE4F568" \ + "3423B4742BF1C978238F16CBE39D652DE3FDB8BEFC848AD9" \ + "22222E04A4037C0713EB57A81A23F0C73473FC646CEA306B" \ + "4BCBC8862F8385DDFA9D4B7FA2C087E879683303ED5BDD3A" \ + "062B3CF5B3A278A66D2A13F83F44F82DDF310EE074AB6A36" \ + "4597E899A0255DC164F31CC50846851DF9AB48195DED7EA1" \ + "B1D510BD7EE74D73FAF36BC31ECFA268359046F4EB879F92" \ + "4009438B481C6CD7889A002ED5EE382BC9190DA6FC026E47" \ + "9558E4475677E9AA9E3050E2765694DFC81F56E880B96E71" \ + "60C980DD98EDD3DFFFFFFFFFFFFFFFFF" + diff --git a/gst/rtmp/gstrtmpsrc.c b/gst/rtmp/gstrtmpsrc.c new file mode 100644 index 0000000000..c8aceb5fc0 --- /dev/null +++ b/gst/rtmp/gstrtmpsrc.c @@ -0,0 +1,652 @@ +/* GStreamer + * Copyright (C) 1999,2000 Erik Walthinsen + * 2000 Wim Taymans + * 2001 Bastien Nocera + * 2002 Kristian Rietveld + * 2002,2003 Colin Walters + * + * rtmpsrc.c: + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Library General Public + * License as published by the Free Software Foundation; either + * version 2 of the License, or (at your option) any later version. + * + * This library 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 + * Library General Public License for more details. + * + * You should have received a copy of the GNU Library General Public + * License along with this library; if not, write to the + * Free Software Foundation, Inc., 59 Temple Place - Suite 330, + * Boston, MA 02111-1307, USA. + */ + +/** + * SECTION:element-rtmpsrc + * + * This plugin reads data from a local or remote location specified + * by an URI. This location can be specified using any protocol supported by + * the RTMP library. Common protocols are 'file', 'http', 'ftp', or 'smb'. + * + * In case the #GstRTMPSrc:iradio-mode property is set and the + * location is a http resource, rtmpsrc will send special icecast http + * headers to the server to request additional icecast metainformation. If + * the server is not an icecast server, it will display the same behaviour + * as if the #GstRTMPSrc:iradio-mode property was not set. However, + * if the server is in fact an icecast server, rtmpsrc will output + * data with a media type of application/x-icy, in which case you will + * need to use the #GstICYDemux element as follow-up element to extract + * the icecast meta data and to determine the underlying media type. + * + * + * Example launch lines + * |[ + * gst-launch -v rtmpsrc location=file:///home/joe/foo.xyz ! fakesink + * ]| The above pipeline will simply read a local file and do nothing with the + * data read. Instead of rtmpsrc, we could just as well have used the + * filesrc element here. + * |[ + * gst-launch -v rtmpsrc location=smb://othercomputer/foo.xyz ! filesink location=/home/joe/foo.xyz + * ]| The above pipeline will copy a file from a remote host to the local file + * system using the Samba protocol. + * |[ + * gst-launch -v rtmpsrc location=http://music.foobar.com/demo.mp3 ! mad ! audioconvert ! audioresample ! alsasink + * ]| The above pipeline will read and decode and play an mp3 file from a + * web server using the http protocol. + * + */ + +#define DEFAULT_RTMP_PORT 1935 + +#ifdef HAVE_CONFIG_H +#include "config.h" +#endif + +#include + +#include "gstrtmpsrc.h" + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include +#include + +GST_DEBUG_CATEGORY_STATIC (rtmpsrc_debug); +#define GST_CAT_DEFAULT rtmpsrc_debug + +static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src", + GST_PAD_SRC, + GST_PAD_ALWAYS, + GST_STATIC_CAPS_ANY); + +enum +{ + ARG_0, + ARG_LOCATION, +}; + +static void gst_rtmp_src_base_init (gpointer g_class); +static void gst_rtmp_src_class_init (GstRTMPSrcClass * klass); +static void gst_rtmp_src_init (GstRTMPSrc * rtmpsrc); +static void gst_rtmp_src_finalize (GObject * object); +static void gst_rtmp_src_uri_handler_init (gpointer g_iface, + gpointer iface_data); + +static void gst_rtmp_src_set_property (GObject * object, guint prop_id, + const GValue * value, GParamSpec * pspec); +static void gst_rtmp_src_get_property (GObject * object, guint prop_id, + GValue * value, GParamSpec * pspec); + +static gboolean gst_rtmp_src_stop (GstBaseSrc * src); +static gboolean gst_rtmp_src_start (GstBaseSrc * src); +static gboolean gst_rtmp_src_is_seekable (GstBaseSrc * src); +#if 0 +static gboolean gst_rtmp_src_check_get_range (GstBaseSrc * src); +static gboolean gst_rtmp_src_get_size (GstBaseSrc * src, guint64 * size); +#endif +static GstFlowReturn gst_rtmp_src_create (GstBaseSrc * basesrc, + guint64 offset, guint size, GstBuffer ** buffer); +#if 0 +static gboolean gst_rtmp_src_query (GstBaseSrc * src, GstQuery * query); +#endif + +static GstElementClass *parent_class = NULL; + +static gboolean +plugin_init (GstPlugin * plugin) +{ + return gst_element_register (plugin, "rtmpsrc", GST_RANK_NONE, + GST_TYPE_RTMP_SRC); +} + +GST_PLUGIN_DEFINE (GST_VERSION_MAJOR, + GST_VERSION_MINOR, + "rtmpsrc", + "flvstreamer sources", + plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN); + +GType +gst_rtmp_src_get_type (void) +{ + static GType rtmpsrc_type = 0; + + if (!rtmpsrc_type) { + static const GTypeInfo rtmpsrc_info = { + sizeof (GstRTMPSrcClass), + gst_rtmp_src_base_init, + NULL, + (GClassInitFunc) gst_rtmp_src_class_init, + NULL, + NULL, + sizeof (GstRTMPSrc), + 0, + (GInstanceInitFunc) gst_rtmp_src_init, + }; + static const GInterfaceInfo urihandler_info = { + gst_rtmp_src_uri_handler_init, + NULL, + NULL + }; + + rtmpsrc_type = + g_type_register_static (GST_TYPE_BASE_SRC, + "GstRTMPSrc", &rtmpsrc_info, (GTypeFlags) 0); + g_type_add_interface_static (rtmpsrc_type, GST_TYPE_URI_HANDLER, + &urihandler_info); + } + return rtmpsrc_type; +} + +static void +gst_rtmp_src_base_init (gpointer g_class) +{ + GstElementClass *element_class = GST_ELEMENT_CLASS (g_class); + + gst_element_class_add_pad_template (element_class, + gst_static_pad_template_get (&srctemplate)); + + gst_element_class_set_details_simple (element_class, + "RTMP Source", + "Source/File", + "Read RTMP streams", + "Bastien Nocera \n" + "GStreamer maintainers "); + + GST_DEBUG_CATEGORY_INIT (rtmpsrc_debug, "rtmpsrc", 0, "RTMP Source"); +} + +static void +gst_rtmp_src_class_init (GstRTMPSrcClass * klass) +{ + GObjectClass *gobject_class; + GstBaseSrcClass *gstbasesrc_class; + + gobject_class = G_OBJECT_CLASS (klass); + gstbasesrc_class = GST_BASE_SRC_CLASS (klass); + + parent_class = (GstElementClass *) g_type_class_peek_parent (klass); + + gobject_class->finalize = gst_rtmp_src_finalize; + gobject_class->set_property = gst_rtmp_src_set_property; + gobject_class->get_property = gst_rtmp_src_get_property; + + /* properties */ + gst_element_class_install_std_props (GST_ELEMENT_CLASS (klass), + "location", ARG_LOCATION, G_PARAM_READWRITE, NULL); + + gstbasesrc_class->start = GST_DEBUG_FUNCPTR (gst_rtmp_src_start); + gstbasesrc_class->stop = GST_DEBUG_FUNCPTR (gst_rtmp_src_stop); +#if 0 + gstbasesrc_class->get_size = GST_DEBUG_FUNCPTR (gst_rtmp_src_get_size); +#endif + gstbasesrc_class->is_seekable = GST_DEBUG_FUNCPTR (gst_rtmp_src_is_seekable); +#if 0 + gstbasesrc_class->check_get_range = + GST_DEBUG_FUNCPTR (gst_rtmp_src_check_get_range); +#endif + gstbasesrc_class->create = GST_DEBUG_FUNCPTR (gst_rtmp_src_create); +#if 0 + gstbasesrc_class->query = GST_DEBUG_FUNCPTR (gst_rtmp_src_query); +#endif +} + +static void +gst_rtmp_src_init (GstRTMPSrc * rtmpsrc) +{ + rtmpsrc->curoffset = 0; + rtmpsrc->seekable = FALSE; +} + +static void +gst_rtmp_src_finalize (GObject * object) +{ + GstRTMPSrc *rtmpsrc = GST_RTMP_SRC (object); + + g_free (rtmpsrc->uri); + rtmpsrc->uri = NULL; + + if (rtmpsrc->rtmp) { + RTMP_Close (rtmpsrc->rtmp); + RTMP_Free (rtmpsrc->rtmp); + rtmpsrc->rtmp = NULL; + } + + G_OBJECT_CLASS (parent_class)->finalize (object); +} + +/* + * URI interface support. + */ + +static GstURIType +gst_rtmp_src_uri_get_type (void) +{ + return GST_URI_SRC; +} + +static gchar ** +gst_rtmp_src_uri_get_protocols (void) +{ + static gchar *protocols[] = { (char *) "rtmp", NULL }; + return protocols; +} + +static const gchar * +gst_rtmp_src_uri_get_uri (GstURIHandler * handler) +{ + GstRTMPSrc *src = GST_RTMP_SRC (handler); + + return src->uri; +} + +static gboolean +gst_rtmp_src_uri_set_uri (GstURIHandler * handler, const gchar * uri) +{ + GstRTMPSrc *src = GST_RTMP_SRC (handler); + + if (GST_STATE (src) == GST_STATE_PLAYING || + GST_STATE (src) == GST_STATE_PAUSED) + return FALSE; + + g_object_set (G_OBJECT (src), "location", uri, NULL); + g_message ("just set uri to %s", uri); + + return TRUE; +} + +static void +gst_rtmp_src_uri_handler_init (gpointer g_iface, gpointer iface_data) +{ + GstURIHandlerInterface *iface = (GstURIHandlerInterface *) g_iface; + + iface->get_type = gst_rtmp_src_uri_get_type; + iface->get_protocols = gst_rtmp_src_uri_get_protocols; + iface->get_uri = gst_rtmp_src_uri_get_uri; + iface->set_uri = gst_rtmp_src_uri_set_uri; +} + +static void +gst_rtmp_src_set_property (GObject * object, guint prop_id, + const GValue * value, GParamSpec * pspec) +{ + GstRTMPSrc *src; + + src = GST_RTMP_SRC (object); + + switch (prop_id) { + case ARG_LOCATION:{ + char *new_location; + /* the element must be stopped or paused in order to do this */ + if (GST_STATE (src) == GST_STATE_PLAYING || + GST_STATE (src) == GST_STATE_PAUSED) + break; + + g_free (src->uri); + src->uri = NULL; + + if (src->rtmp) { + RTMP_Close (src->rtmp); + RTMP_Free (src->rtmp); + src->rtmp = NULL; + } + + new_location = g_value_dup_string (value); + + src->rtmp = RTMP_Alloc (); + RTMP_Init (src->rtmp); + if (!RTMP_SetupURL (src->rtmp, new_location)) { + GST_ELEMENT_ERROR (src, RESOURCE, OPEN_READ, NULL, + ("Failed to setup URL '%s'", src->uri)); + g_free (new_location); + RTMP_Free (src->rtmp); + src->rtmp = NULL; + } else { + src->uri = g_value_dup_string (value); + g_message ("parsed uri '%s' properly", src->uri); + } + break; + } + default: + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} + +static void +gst_rtmp_src_get_property (GObject * object, guint prop_id, GValue * value, + GParamSpec * pspec) +{ + GstRTMPSrc *src; + + src = GST_RTMP_SRC (object); + + switch (prop_id) { + case ARG_LOCATION: + g_value_set_string (value, src->uri); + break; + default: + G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); + break; + } +} + +/* + * Read a new buffer from src->reqoffset, takes care of events + * and seeking and such. + */ +static GstFlowReturn +gst_rtmp_src_create (GstBaseSrc * basesrc, guint64 offset, guint size, + GstBuffer ** buffer) +{ + GstRTMPSrc *src; + GstBuffer *buf; + guint8 *data; + guint todo; + int read; + + src = GST_RTMP_SRC (basesrc); + + g_return_val_if_fail (src->rtmp != NULL, GST_FLOW_ERROR); + + GST_DEBUG ("now at %" G_GINT64_FORMAT ", reading from %" G_GUINT64_FORMAT + ", size %u", src->curoffset, offset, size); + + /* open if required */ + if (G_UNLIKELY (!RTMP_IsConnected (src->rtmp))) { + if (!RTMP_Connect (src->rtmp, NULL)) { + GST_ELEMENT_ERROR (src, RESOURCE, NOT_FOUND, (NULL), + ("Could not connect to RTMP stream \"%s\" for reading: %s (%d)", + src->uri, "FIXME", 0)); + return GST_FLOW_ERROR; + } + } + + /* seek if required */ + if (G_UNLIKELY (src->curoffset != offset)) { + GST_DEBUG ("need to seek"); + if (src->seekable) { +#if 0 + GST_DEBUG ("seeking to %" G_GUINT64_FORMAT, offset); + res = rtmp_seek (src->handle, RTMP_SEEK_START, offset); + if (res != RTMP_OK) + goto seek_failed; + src->curoffset = offset; +#endif + } else { + goto cannot_seek; + } + } + + buf = gst_buffer_try_new_and_alloc (size); + if (G_UNLIKELY (buf == NULL && size == 0)) { + GST_ERROR_OBJECT (src, "Failed to allocate %u bytes", size); + return GST_FLOW_ERROR; + } + + data = GST_BUFFER_DATA (buf); + + /* FIXME add FLV header first time around? */ + read = 0; + + todo = size; + while (todo > 0) { + read = RTMP_Read (src->rtmp, (char *) &data, todo); + + if (G_UNLIKELY (read == -1)) + goto eos; + + if (G_UNLIKELY (read == -2)) + goto read_failed; + + /* FIXME handle -3 ? */ + + if (read < todo) { + data = &data[read]; + todo -= read; + } else { + todo = 0; + } + GST_LOG (" got size %" G_GUINT64_FORMAT, read); + } + GST_BUFFER_OFFSET (buf) = src->curoffset; + src->curoffset += size; + + /* we're done, return the buffer */ + *buffer = buf; + +#if 0 + RTMPFileSize readbytes; + guint todo; + + + + return GST_FLOW_OK; +#endif + return GST_FLOW_OK; + +//seek_failed: + { + GST_ELEMENT_ERROR (src, RESOURCE, SEEK, (NULL), + ("Failed to seek to requested position %" G_GINT64_FORMAT ": %s", + offset, "FIXME")); + return GST_FLOW_ERROR; + } +cannot_seek: + { + GST_ELEMENT_ERROR (src, RESOURCE, SEEK, (NULL), + ("Requested seek from %" G_GINT64_FORMAT " to %" G_GINT64_FORMAT + " on non-seekable stream", src->curoffset, offset)); + return GST_FLOW_ERROR; + } +read_failed: + { + gst_buffer_unref (buf); + GST_ELEMENT_ERROR (src, RESOURCE, READ, (NULL), + ("Failed to read data: %s", "FIXME")); + return GST_FLOW_ERROR; + } +eos: + { + gst_buffer_unref (buf); + GST_DEBUG_OBJECT (src, "Reading data gave EOS"); + return GST_FLOW_UNEXPECTED; + } +} + +#if 0 +static gboolean +gst_rtmp_src_query (GstBaseSrc * basesrc, GstQuery * query) +{ + gboolean ret = FALSE; + GstRTMPSrc *src = GST_RTMP_SRC (basesrc); + + switch (GST_QUERY_TYPE (query)) { + case GST_QUERY_URI: + gst_query_set_uri (query, src->uri); + ret = TRUE; + break; + default: + ret = FALSE; + break; + } + + if (!ret) + ret = GST_BASE_SRC_CLASS (parent_class)->query (basesrc, query); + + return ret; +} +#endif +static gboolean +gst_rtmp_src_is_seekable (GstBaseSrc * basesrc) +{ + GstRTMPSrc *src; + + src = GST_RTMP_SRC (basesrc); + + return src->seekable; +} + +#if 0 +static gboolean +gst_rtmp_src_check_get_range (GstBaseSrc * basesrc) +{ + GstRTMPSrc *src; + const gchar *protocol; + + src = GST_RTMP_SRC (basesrc); + + if (src->uri == NULL) { + GST_WARNING_OBJECT (src, "no URI set yet"); + return FALSE; + } + + if (rtmp_uri_is_local (src->uri)) { + GST_LOG_OBJECT (src, "local URI (%s), assuming random access is possible", + GST_STR_NULL (src->uri_name)); + return TRUE; + } + + /* blacklist certain protocols we know won't work getrange-based */ + protocol = rtmp_uri_get_scheme (src->uri); + if (protocol == NULL) + goto undecided; + + if (strcmp (protocol, "http") == 0 || strcmp (protocol, "https") == 0) { + GST_LOG_OBJECT (src, "blacklisted protocol '%s', no random access possible" + " (URI=%s)", protocol, GST_STR_NULL (src->uri_name)); + return FALSE; + } + + /* fall through to undecided */ + +undecided: + { + /* don't know what to do, let the basesrc class decide for us */ + GST_LOG_OBJECT (src, "undecided about URI '%s', let base class handle it", + GST_STR_NULL (src->uri_name)); + + if (GST_BASE_SRC_CLASS (parent_class)->check_get_range) + return GST_BASE_SRC_CLASS (parent_class)->check_get_range (basesrc); + + return FALSE; + } +} +#endif + +#if 0 +static gboolean +gst_rtmp_src_get_size (GstBaseSrc * basesrc, guint64 * size) +{ + GstRTMPSrc *src; + RTMPFileInfo *info; + RTMPFileInfoOptions options; + RTMPResult res; + + src = GST_RTMP_SRC (basesrc); + + *size = -1; + info = rtmp_file_info_new (); + options = RTMP_FILE_INFO_DEFAULT | RTMP_FILE_INFO_FOLLOW_LINKS; + res = rtmp_get_file_info_from_handle (src->handle, info, options); + if (res == RTMP_OK) { + if ((info->valid_fields & RTMP_FILE_INFO_FIELDS_SIZE) != 0) { + *size = info->size; + GST_DEBUG_OBJECT (src, "from handle: %" G_GUINT64_FORMAT " bytes", *size); + } else if (src->own_handle && rtmp_uri_is_local (src->uri)) { + GST_DEBUG_OBJECT (src, + "file size not known, file local, trying fallback"); + res = rtmp_get_file_info_uri (src->uri, info, options); + if (res == RTMP_OK && + (info->valid_fields & RTMP_FILE_INFO_FIELDS_SIZE) != 0) { + *size = info->size; + GST_DEBUG_OBJECT (src, "from uri: %" G_GUINT64_FORMAT " bytes", *size); + } + } + } else { + GST_WARNING_OBJECT (src, "getting info failed: %s", + rtmp_result_to_string (res)); + } + rtmp_file_info_unref (info); + + if (*size == (RTMPFileSize) - 1) + return FALSE; + + GST_DEBUG_OBJECT (src, "return size %" G_GUINT64_FORMAT, *size); + + return TRUE; +} +#endif + +/* open the file, do stuff necessary to go to PAUSED state */ +static gboolean +gst_rtmp_src_start (GstBaseSrc * basesrc) +{ + GstRTMPSrc *src; + + src = GST_RTMP_SRC (basesrc); + + g_message ("start called!"); + + if (!src->uri) { + GST_ELEMENT_ERROR (src, RESOURCE, OPEN_READ, (NULL), ("No filename given")); + return FALSE; + } + + return TRUE; +} + +static gboolean +gst_rtmp_src_stop (GstBaseSrc * basesrc) +{ + GstRTMPSrc *src; + + src = GST_RTMP_SRC (basesrc); + +//FIXME you can't run RTMP_Close multiple times +// RTMP_Close (src->rtmp); + + g_message ("stop called!"); + + src->curoffset = 0; + + return TRUE; +} + +/* + * vim: sw=2 ts=8 cindent noai bs=2 + */ diff --git a/gst/rtmp/gstrtmpsrc.h b/gst/rtmp/gstrtmpsrc.h new file mode 100644 index 0000000000..c0609ccbd5 --- /dev/null +++ b/gst/rtmp/gstrtmpsrc.h @@ -0,0 +1,76 @@ +/* GStreamer + * Copyright (C) 1999,2000 Erik Walthinsen + * 2000 Wim Taymans + * 2001 Bastien Nocera + * 2002 Kristian Rietveld + * 2002,2003 Colin Walters + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Library General Public + * License as published by the Free Software Foundation; either + * version 2 of the License, or (at your option) any later version. + * + * This library 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 + * Library General Public License for more details. + * + * You should have received a copy of the GNU Library General Public + * License along with this library; if not, write to the + * Free Software Foundation, Inc., 59 Temple Place - Suite 330, + * Boston, MA 02111-1307, USA. + */ + +#ifndef __GST_RTMP_SRC_H__ +#define __GST_RTMP_SRC_H__ + +#include + +#include "rtmp.h" +#include "log.h" +#include "amf.h" + +G_BEGIN_DECLS + +#define GST_TYPE_RTMP_SRC \ + (gst_rtmp_src_get_type()) +#define GST_RTMP_SRC(obj) \ + (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_RTMP_SRC,GstRTMPSrc)) +#define GST_RTMP_SRC_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_RTMP_SRC,GstRTMPSrcClass)) +#define GST_IS_RTMP_SRC(obj) \ + (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_RTMP_SRC)) +#define GST_IS_RTMP_SRC_CLASS(klass) \ + (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_RTMP_SRC)) + +typedef struct _GstRTMPSrc GstRTMPSrc; +typedef struct _GstRTMPSrcClass GstRTMPSrcClass; + +/** + * GstRTMPSrc: + * + * Opaque data structure. + */ +struct _GstRTMPSrc +{ + GstBaseSrc basesrc; + + char *uri; + + RTMP *rtmp; + + gboolean seekable; + gint64 curoffset; +}; + +struct _GstRTMPSrcClass +{ + GstBaseSrcClass basesrc_class; +}; + +GType gst_rtmp_src_get_type (void); + +G_END_DECLS + +#endif /* __GST_RTMP_SRC_H__ */ + diff --git a/gst/rtmp/handshake.h b/gst/rtmp/handshake.h new file mode 100644 index 0000000000..c246286cec --- /dev/null +++ b/gst/rtmp/handshake.h @@ -0,0 +1,1089 @@ +/* + * Copyright (C) 2008-2009 Andrej Stepanchuk + * Copyright (C) 2009-2010 Howard Chu + * Copyright (C) 2010 2a665470ced7adb7156fcef47f8199a6371c117b8a79e399a2771e0b36384090 + * + * This file is part of librtmp. + * + * librtmp is free software; you can redistribute it and/or modify + * it under the terms of the GNU Lesser General Public License as + * published by the Free Software Foundation; either version 2.1, + * or (at your option) any later version. + * + * librtmp 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 Lesser General Public License + * along with librtmp see the file COPYING. If not, write to + * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. + * http://www.gnu.org/copyleft/lgpl.html + */ + +/* This file is #included in rtmp.c, it is not meant to be compiled alone */ + +#ifdef USE_POLARSSL +#include +#include +#ifndef SHA256_DIGEST_LENGTH +#define SHA256_DIGEST_LENGTH 32 +#endif +#define HMAC_CTX sha2_context +#define HMAC_setup(ctx, key, len) sha2_hmac_starts(&ctx, (unsigned char *)key, len, 0) +#define HMAC_crunch(ctx, buf, len) sha2_hmac_update(&ctx, buf, len) +#define HMAC_finish(ctx, dig, dlen) dlen = SHA256_DIGEST_LENGTH; sha2_hmac_finish(&ctx, dig) + +typedef arc4_context * RC4_handle; +#define RC4_setup(h) *h = malloc(sizeof(arc4_context)) +#define RC4_setkey(h,l,k) arc4_setup(h,k,l) +#define RC4_encrypt(h,l,d) arc4_crypt(h,l,(unsigned char *)d,(unsigned char *)d) +#define RC4_encrypt2(h,l,s,d) arc4_crypt(h,l,(unsigned char *)s,(unsigned char *)d) + +#elif defined(USE_GNUTLS) +#include +#ifndef SHA256_DIGEST_LENGTH +#define SHA256_DIGEST_LENGTH 32 +#endif +#define HMAC_CTX gcry_md_hd_t +#define HMAC_setup(ctx, key, len) gcry_md_open(&ctx, GCRY_MD_SHA256, GCRY_MD_FLAG_HMAC); gcry_md_setkey(ctx, key, len) +#define HMAC_crunch(ctx, buf, len) gcry_md_write(ctx, buf, len) +#define HMAC_finish(ctx, dig, dlen) dlen = SHA256_DIGEST_LENGTH; memcpy(dig, gcry_md_read(ctx, 0), dlen); gcry_md_close(ctx) + +typedef gcry_cipher_hd_t RC4_handle; +#define RC4_setup(h) gcry_cipher_open(h, GCRY_CIPHER_ARCFOUR, GCRY_CIPHER_MODE_STREAM, 0) +#define RC4_setkey(h,l,k) gcry_cipher_setkey(h,k,l) +#define RC4_encrypt(h,l,d) gcry_cipher_encrypt(h,(void *)d,l,NULL,0) +#define RC4_encrypt2(h,l,s,d) gcry_cipher_encrypt(h,(void *)d,l,(void *)s,l) + +#else /* USE_OPENSSL */ +#include +#include +#include +#if OPENSSL_VERSION_NUMBER < 0x0090800 || !defined(SHA256_DIGEST_LENGTH) +#error Your OpenSSL is too old, need 0.9.8 or newer with SHA256 +#endif +#define HMAC_setup(ctx, key, len) HMAC_CTX_init(&ctx); HMAC_Init_ex(&ctx, key, len, EVP_sha256(), 0) +#define HMAC_crunch(ctx, buf, len) HMAC_Update(&ctx, buf, len) +#define HMAC_finish(ctx, dig, dlen) HMAC_Final(&ctx, dig, &dlen); HMAC_CTX_cleanup(&ctx) + +typedef RC4_KEY * RC4_handle; +#define RC4_setup(h) *h = malloc(sizeof(RC4_KEY)) +#define RC4_setkey(h,l,k) RC4_set_key(h,l,k) +#define RC4_encrypt(h,l,d) RC4(h,l,(uint8_t *)d,(uint8_t *)d) +#define RC4_encrypt2(h,l,s,d) RC4(h,l,(uint8_t *)s,(uint8_t *)d) +#endif + +#define FP10 + +#include "dh.h" + +static const uint8_t GenuineFMSKey[] = { + 0x47, 0x65, 0x6e, 0x75, 0x69, 0x6e, 0x65, 0x20, 0x41, 0x64, 0x6f, 0x62, + 0x65, 0x20, 0x46, 0x6c, + 0x61, 0x73, 0x68, 0x20, 0x4d, 0x65, 0x64, 0x69, 0x61, 0x20, 0x53, 0x65, + 0x72, 0x76, 0x65, 0x72, + 0x20, 0x30, 0x30, 0x31, /* Genuine Adobe Flash Media Server 001 */ + + 0xf0, 0xee, 0xc2, 0x4a, 0x80, 0x68, 0xbe, 0xe8, 0x2e, 0x00, 0xd0, 0xd1, + 0x02, 0x9e, 0x7e, 0x57, 0x6e, 0xec, 0x5d, 0x2d, 0x29, 0x80, 0x6f, 0xab, + 0x93, 0xb8, 0xe6, 0x36, + 0xcf, 0xeb, 0x31, 0xae +}; /* 68 */ + +static const uint8_t GenuineFPKey[] = { + 0x47, 0x65, 0x6E, 0x75, 0x69, 0x6E, 0x65, 0x20, 0x41, 0x64, 0x6F, 0x62, + 0x65, 0x20, 0x46, 0x6C, + 0x61, 0x73, 0x68, 0x20, 0x50, 0x6C, 0x61, 0x79, 0x65, 0x72, 0x20, 0x30, + 0x30, 0x31, /* Genuine Adobe Flash Player 001 */ + 0xF0, 0xEE, + 0xC2, 0x4A, 0x80, 0x68, 0xBE, 0xE8, 0x2E, 0x00, 0xD0, 0xD1, 0x02, 0x9E, + 0x7E, 0x57, 0x6E, 0xEC, + 0x5D, 0x2D, 0x29, 0x80, 0x6F, 0xAB, 0x93, 0xB8, 0xE6, 0x36, 0xCF, 0xEB, + 0x31, 0xAE +}; /* 62 */ + +static void InitRC4Encryption + (uint8_t * secretKey, + uint8_t * pubKeyIn, + uint8_t * pubKeyOut, RC4_handle *rc4keyIn, RC4_handle *rc4keyOut) +{ + uint8_t digest[SHA256_DIGEST_LENGTH]; + unsigned int digestLen = 0; + HMAC_CTX ctx; + + RC4_setup(rc4keyIn); + RC4_setup(rc4keyOut); + + HMAC_setup(ctx, secretKey, 128); + HMAC_crunch(ctx, pubKeyIn, 128); + HMAC_finish(ctx, digest, digestLen); + + RTMP_Log(RTMP_LOGDEBUG, "RC4 Out Key: "); + RTMP_LogHex(RTMP_LOGDEBUG, digest, 16); + + RC4_setkey(*rc4keyOut, 16, digest); + + HMAC_setup(ctx, secretKey, 128); + HMAC_crunch(ctx, pubKeyOut, 128); + HMAC_finish(ctx, digest, digestLen); + + RTMP_Log(RTMP_LOGDEBUG, "RC4 In Key: "); + RTMP_LogHex(RTMP_LOGDEBUG, digest, 16); + + RC4_setkey(*rc4keyIn, 16, digest); +} + +typedef unsigned int (getoff)(uint8_t *buf, unsigned int len); + +static unsigned int +GetDHOffset2(uint8_t *handshake, unsigned int len) +{ + unsigned int offset = 0; + uint8_t *ptr = handshake + 768; + unsigned int res; + + assert(RTMP_SIG_SIZE <= len); + + offset += (*ptr); + ptr++; + offset += (*ptr); + ptr++; + offset += (*ptr); + ptr++; + offset += (*ptr); + + res = (offset % 632) + 8; + + if (res + 128 > 767) + { + RTMP_Log(RTMP_LOGERROR, + "%s: Couldn't calculate correct DH offset (got %d), exiting!", + __FUNCTION__, res); + exit(1); + } + return res; +} + +static unsigned int +GetDigestOffset2(uint8_t *handshake, unsigned int len) +{ + unsigned int offset = 0; + uint8_t *ptr = handshake + 772; + unsigned int res; + + offset += (*ptr); + ptr++; + offset += (*ptr); + ptr++; + offset += (*ptr); + ptr++; + offset += (*ptr); + + res = (offset % 728) + 776; + + if (res + 32 > 1535) + { + RTMP_Log(RTMP_LOGERROR, + "%s: Couldn't calculate correct digest offset (got %d), exiting", + __FUNCTION__, res); + exit(1); + } + return res; +} + +static unsigned int +GetDHOffset1(uint8_t *handshake, unsigned int len) +{ + unsigned int offset = 0; + uint8_t *ptr = handshake + 1532; + unsigned int res; + + assert(RTMP_SIG_SIZE <= len); + + offset += (*ptr); + ptr++; + offset += (*ptr); + ptr++; + offset += (*ptr); + ptr++; + offset += (*ptr); + + res = (offset % 632) + 772; + + if (res + 128 > 1531) + { + RTMP_Log(RTMP_LOGERROR, "%s: Couldn't calculate DH offset (got %d), exiting!", + __FUNCTION__, res); + exit(1); + } + + return res; +} + +static unsigned int +GetDigestOffset1(uint8_t *handshake, unsigned int len) +{ + unsigned int offset = 0; + uint8_t *ptr = handshake + 8; + unsigned int res; + + assert(12 <= len); + + offset += (*ptr); + ptr++; + offset += (*ptr); + ptr++; + offset += (*ptr); + ptr++; + offset += (*ptr); + + res = (offset % 728) + 12; + + if (res + 32 > 771) + { + RTMP_Log(RTMP_LOGERROR, + "%s: Couldn't calculate digest offset (got %d), exiting!", + __FUNCTION__, res); + exit(1); + } + + return res; +} + +static getoff *digoff[] = {GetDigestOffset1, GetDigestOffset2}; +static getoff *dhoff[] = {GetDHOffset1, GetDHOffset2}; + +static void +HMACsha256(const uint8_t *message, size_t messageLen, const uint8_t *key, + size_t keylen, uint8_t *digest) +{ + unsigned int digestLen; + HMAC_CTX ctx; + + HMAC_setup(ctx, key, keylen); + HMAC_crunch(ctx, message, messageLen); + HMAC_finish(ctx, digest, digestLen); + + assert(digestLen == 32); +} + +static void +CalculateDigest(unsigned int digestPos, uint8_t *handshakeMessage, + const uint8_t *key, size_t keyLen, uint8_t *digest) +{ + const int messageLen = RTMP_SIG_SIZE - SHA256_DIGEST_LENGTH; + uint8_t message[RTMP_SIG_SIZE - SHA256_DIGEST_LENGTH]; + + memcpy(message, handshakeMessage, digestPos); + memcpy(message + digestPos, + &handshakeMessage[digestPos + SHA256_DIGEST_LENGTH], + messageLen - digestPos); + + HMACsha256(message, messageLen, key, keyLen, digest); +} + +static bool +VerifyDigest(unsigned int digestPos, uint8_t *handshakeMessage, const uint8_t *key, + size_t keyLen) +{ + uint8_t calcDigest[SHA256_DIGEST_LENGTH]; + + CalculateDigest(digestPos, handshakeMessage, key, keyLen, calcDigest); + + return memcmp(&handshakeMessage[digestPos], calcDigest, + SHA256_DIGEST_LENGTH) == 0; +} + +/* handshake + * + * Type = [1 bytes] plain: 0x03, encrypted: 0x06, 0x08, 0x09 + * -------------------------------------------------------------------- [1536 bytes] + * Uptime = [4 bytes] big endian unsigned number, uptime + * Version = [4 bytes] each byte represents a version number, e.g. 9.0.124.0 + * ... + * + */ + +static const uint32_t rtmpe8_keys[16][4] = { + {0xbff034b2, 0x11d9081f, 0xccdfb795, 0x748de732}, + {0x086a5eb6, 0x1743090e, 0x6ef05ab8, 0xfe5a39e2}, + {0x7b10956f, 0x76ce0521, 0x2388a73a, 0x440149a1}, + {0xa943f317, 0xebf11bb2, 0xa691a5ee, 0x17f36339}, + {0x7a30e00a, 0xb529e22c, 0xa087aea5, 0xc0cb79ac}, + {0xbdce0c23, 0x2febdeff, 0x1cfaae16, 0x1123239d}, + {0x55dd3f7b, 0x77e7e62e, 0x9bb8c499, 0xc9481ee4}, + {0x407bb6b4, 0x71e89136, 0xa7aebf55, 0xca33b839}, + {0xfcf6bdc3, 0xb63c3697, 0x7ce4f825, 0x04d959b2}, + {0x28e091fd, 0x41954c4c, 0x7fb7db00, 0xe3a066f8}, + {0x57845b76, 0x4f251b03, 0x46d45bcd, 0xa2c30d29}, + {0x0acceef8, 0xda55b546, 0x03473452, 0x5863713b}, + {0xb82075dc, 0xa75f1fee, 0xd84268e8, 0xa72a44cc}, + {0x07cf6e9e, 0xa16d7b25, 0x9fa7ae6c, 0xd92f5629}, + {0xfeb1eae4, 0x8c8c3ce1, 0x4e0064a7, 0x6a387c2a}, + {0x893a9427, 0xcc3013a2, 0xf106385b, 0xa829f927} +}; + +/* RTMPE type 8 uses XTEA on the regular signature + * http://en.wikipedia.org/wiki/XTEA + */ +static void rtmpe8_sig(uint8_t *in, uint8_t *out, int keyid) +{ + unsigned int i, num_rounds = 32; + uint32_t v0, v1, sum=0, delta=0x9E3779B9; + uint32_t const *k; + + v0 = in[0] | (in[1] << 8) | (in[2] << 16) | (in[3] << 24); + v1 = in[4] | (in[5] << 8) | (in[6] << 16) | (in[7] << 24); + k = rtmpe8_keys[keyid]; + + for (i=0; i < num_rounds; i++) { + v0 += (((v1 << 4) ^ (v1 >> 5)) + v1) ^ (sum + k[sum & 3]); + sum += delta; + v1 += (((v0 << 4) ^ (v0 >> 5)) + v0) ^ (sum + k[(sum>>11) & 3]); + } + + out[0] = v0; v0 >>= 8; + out[1] = v0; v0 >>= 8; + out[2] = v0; v0 >>= 8; + out[3] = v0; + + out[4] = v1; v1 >>= 8; + out[5] = v1; v1 >>= 8; + out[6] = v1; v1 >>= 8; + out[7] = v1; +} + +static bool +HandShake(RTMP * r, bool FP9HandShake) +{ + int i, offalg = 0; + int dhposClient = 0; + int digestPosClient = 0; + bool encrypted = r->Link.protocol & RTMP_FEATURE_ENC; + + RC4_handle keyIn = 0; + RC4_handle keyOut = 0; + + int32_t *ip; + uint32_t uptime; + + uint8_t clientbuf[RTMP_SIG_SIZE + 4], *clientsig=clientbuf+4; + uint8_t serversig[RTMP_SIG_SIZE], client2[RTMP_SIG_SIZE], *reply; + uint8_t type; + getoff *getdh = NULL, *getdig = NULL; + + if (encrypted || r->Link.SWFSize) + FP9HandShake = true; + else + FP9HandShake = false; + + r->Link.rc4keyIn = r->Link.rc4keyOut = 0; + + if (encrypted) + { + clientsig[-1] = 0x06; /* 0x08 is RTMPE as well */ + offalg = 1; + } + else + clientsig[-1] = 0x03; + + uptime = htonl(RTMP_GetTime()); + memcpy(clientsig, &uptime, 4); + + if (FP9HandShake) + { + /* set version to at least 9.0.115.0 */ + if (encrypted) + { + clientsig[4] = 128; + clientsig[6] = 3; + } + else + { + clientsig[4] = 10; + clientsig[6] = 45; + } + clientsig[5] = 0; + clientsig[7] = 2; + + RTMP_Log(RTMP_LOGDEBUG, "%s: Client type: %02X", __FUNCTION__, clientsig[-1]); + getdig = digoff[offalg]; + getdh = dhoff[offalg]; + } + else + { + memset(&clientsig[4], 0, 4); + } + + /* generate random data */ +#ifdef _DEBUG + memset(clientsig+8, 0, RTMP_SIG_SIZE-8); +#else + ip = (int32_t *)(clientsig+8); + for (i = 2; i < RTMP_SIG_SIZE/4; i++) + *ip++ = rand(); +#endif + + /* set handshake digest */ + if (FP9HandShake) + { + if (encrypted) + { + /* generate Diffie-Hellmann parameters */ + r->Link.dh = DHInit(1024); + if (!r->Link.dh) + { + RTMP_Log(RTMP_LOGERROR, "%s: Couldn't initialize Diffie-Hellmann!", + __FUNCTION__); + return false; + } + + dhposClient = getdh(clientsig, RTMP_SIG_SIZE); + RTMP_Log(RTMP_LOGDEBUG, "%s: DH pubkey position: %d", __FUNCTION__, dhposClient); + + if (!DHGenerateKey(r->Link.dh)) + { + RTMP_Log(RTMP_LOGERROR, "%s: Couldn't generate Diffie-Hellmann public key!", + __FUNCTION__); + return false; + } + + if (!DHGetPublicKey(r->Link.dh, &clientsig[dhposClient], 128)) + { + RTMP_Log(RTMP_LOGERROR, "%s: Couldn't write public key!", __FUNCTION__); + return false; + } + } + + digestPosClient = getdig(clientsig, RTMP_SIG_SIZE); /* reuse this value in verification */ + RTMP_Log(RTMP_LOGDEBUG, "%s: Client digest offset: %d", __FUNCTION__, + digestPosClient); + + CalculateDigest(digestPosClient, clientsig, GenuineFPKey, 30, + &clientsig[digestPosClient]); + + RTMP_Log(RTMP_LOGDEBUG, "%s: Initial client digest: ", __FUNCTION__); + RTMP_LogHex(RTMP_LOGDEBUG, clientsig + digestPosClient, + SHA256_DIGEST_LENGTH); + } + +#ifdef _DEBUG + RTMP_Log(RTMP_LOGDEBUG, "Clientsig: "); + RTMP_LogHex(RTMP_LOGDEBUG, clientsig, RTMP_SIG_SIZE); +#endif + + if (!WriteN(r, (char *)clientsig-1, RTMP_SIG_SIZE + 1)) + return false; + + if (ReadN(r, (char *)&type, 1) != 1) /* 0x03 or 0x06 */ + return false; + + RTMP_Log(RTMP_LOGDEBUG, "%s: Type Answer : %02X", __FUNCTION__, type); + + if (type != clientsig[-1]) + RTMP_Log(RTMP_LOGWARNING, "%s: Type mismatch: client sent %d, server answered %d", + __FUNCTION__, clientsig[-1], type); + + if (ReadN(r, (char *)serversig, RTMP_SIG_SIZE) != RTMP_SIG_SIZE) + return false; + + /* decode server response */ + memcpy(&uptime, serversig, 4); + uptime = ntohl(uptime); + + RTMP_Log(RTMP_LOGDEBUG, "%s: Server Uptime : %d", __FUNCTION__, uptime); + RTMP_Log(RTMP_LOGDEBUG, "%s: FMS Version : %d.%d.%d.%d", __FUNCTION__, serversig[4], + serversig[5], serversig[6], serversig[7]); + + if (FP9HandShake && type == 3 && !serversig[4]) + FP9HandShake = false; + +#ifdef _DEBUG + RTMP_Log(RTMP_LOGDEBUG, "Server signature:"); + RTMP_LogHex(RTMP_LOGDEBUG, serversig, RTMP_SIG_SIZE); +#endif + + if (FP9HandShake) + { + uint8_t digestResp[SHA256_DIGEST_LENGTH]; + uint8_t *signatureResp = NULL; + + /* we have to use this signature now to find the correct algorithms for getting the digest and DH positions */ + int digestPosServer = getdig(serversig, RTMP_SIG_SIZE); + + if (!VerifyDigest(digestPosServer, serversig, GenuineFMSKey, 36)) + { + RTMP_Log(RTMP_LOGWARNING, "Trying different position for server digest!"); + offalg ^= 1; + getdig = digoff[offalg]; + getdh = dhoff[offalg]; + digestPosServer = getdig(serversig, RTMP_SIG_SIZE); + + if (!VerifyDigest(digestPosServer, serversig, GenuineFMSKey, 36)) + { + RTMP_Log(RTMP_LOGERROR, "Couldn't verify the server digest"); /* continuing anyway will probably fail */ + return false; + } + } + + /* generate SWFVerification token (SHA256 HMAC hash of decompressed SWF, key are the last 32 bytes of the server handshake) */ + if (r->Link.SWFSize) + { + const char swfVerify[] = { 0x01, 0x01 }; + char *vend = r->Link.SWFVerificationResponse+sizeof(r->Link.SWFVerificationResponse); + + memcpy(r->Link.SWFVerificationResponse, swfVerify, 2); + AMF_EncodeInt32(&r->Link.SWFVerificationResponse[2], vend, r->Link.SWFSize); + AMF_EncodeInt32(&r->Link.SWFVerificationResponse[6], vend, r->Link.SWFSize); + HMACsha256(r->Link.SWFHash, SHA256_DIGEST_LENGTH, + &serversig[RTMP_SIG_SIZE - SHA256_DIGEST_LENGTH], + SHA256_DIGEST_LENGTH, + (uint8_t *)&r->Link.SWFVerificationResponse[10]); + } + + /* do Diffie-Hellmann Key exchange for encrypted RTMP */ + if (encrypted) + { + /* compute secret key */ + uint8_t secretKey[128] = { 0 }; + int len, dhposServer; + + dhposServer = getdh(serversig, RTMP_SIG_SIZE); + RTMP_Log(RTMP_LOGDEBUG, "%s: Server DH public key offset: %d", __FUNCTION__, + dhposServer); + len = DHComputeSharedSecretKey(r->Link.dh, &serversig[dhposServer], + 128, secretKey); + if (len < 0) + { + RTMP_Log(RTMP_LOGDEBUG, "%s: Wrong secret key position!", __FUNCTION__); + return false; + } + + RTMP_Log(RTMP_LOGDEBUG, "%s: Secret key: ", __FUNCTION__); + RTMP_LogHex(RTMP_LOGDEBUG, secretKey, 128); + + InitRC4Encryption(secretKey, + (uint8_t *) & serversig[dhposServer], + (uint8_t *) & clientsig[dhposClient], + &keyIn, &keyOut); + } + + + reply = client2; +#ifdef _DEBUG + memset(reply, 0xff, RTMP_SIG_SIZE); +#else + ip = (int32_t *)reply; + for (i = 0; i < RTMP_SIG_SIZE/4; i++) + *ip++ = rand(); +#endif + /* calculate response now */ + signatureResp = reply+RTMP_SIG_SIZE-SHA256_DIGEST_LENGTH; + + HMACsha256(&serversig[digestPosServer], SHA256_DIGEST_LENGTH, + GenuineFPKey, sizeof(GenuineFPKey), digestResp); + HMACsha256(reply, RTMP_SIG_SIZE - SHA256_DIGEST_LENGTH, digestResp, + SHA256_DIGEST_LENGTH, signatureResp); + + /* some info output */ + RTMP_Log(RTMP_LOGDEBUG, + "%s: Calculated digest key from secure key and server digest: ", + __FUNCTION__); + RTMP_LogHex(RTMP_LOGDEBUG, digestResp, SHA256_DIGEST_LENGTH); + +#ifdef FP10 + if (type == 8 ) + { + uint8_t *dptr = digestResp; + uint8_t *sig = signatureResp; + /* encrypt signatureResp */ + for (i=0; iLink.rc4keyIn = keyIn; + r->Link.rc4keyOut = keyOut; + + + /* update the keystreams */ + if (r->Link.rc4keyIn) + { + RC4_encrypt(r->Link.rc4keyIn, RTMP_SIG_SIZE, (uint8_t *) buff); + } + + if (r->Link.rc4keyOut) + { + RC4_encrypt(r->Link.rc4keyOut, RTMP_SIG_SIZE, (uint8_t *) buff); + } + } + } + else + { + if (memcmp(serversig, clientsig, RTMP_SIG_SIZE) != 0) + { + RTMP_Log(RTMP_LOGWARNING, "%s: client signature does not match!", + __FUNCTION__); + } + } + + RTMP_Log(RTMP_LOGDEBUG, "%s: Handshaking finished....", __FUNCTION__); + return true; +} + +static bool +SHandShake(RTMP * r) +{ + int i, offalg = 0; + int dhposServer = 0; + int digestPosServer = 0; + RC4_handle keyIn = 0; + RC4_handle keyOut = 0; + bool FP9HandShake = false; + bool encrypted; + int32_t *ip; + + uint8_t clientsig[RTMP_SIG_SIZE]; + uint8_t serverbuf[RTMP_SIG_SIZE + 4], *serversig = serverbuf+4; + uint8_t type; + uint32_t uptime; + getoff *getdh = NULL, *getdig = NULL; + + if (ReadN(r, (char *)&type, 1) != 1) /* 0x03 or 0x06 */ + return false; + + if (ReadN(r, (char *)clientsig, RTMP_SIG_SIZE) != RTMP_SIG_SIZE) + return false; + + RTMP_Log(RTMP_LOGDEBUG, "%s: Type Requested : %02X", __FUNCTION__, type); + RTMP_LogHex(RTMP_LOGDEBUG2, clientsig, RTMP_SIG_SIZE); + + if (type == 3) + { + encrypted = false; + } + else if (type == 6 || type == 8) + { + offalg = 1; + encrypted = true; + FP9HandShake = true; + r->Link.protocol |= RTMP_FEATURE_ENC; + /* use FP10 if client is capable */ + if (clientsig[4] == 128) + type = 8; + } + else + { + RTMP_Log(RTMP_LOGERROR, "%s: Unknown version %02x", + __FUNCTION__, type); + return false; + } + + if (!FP9HandShake && clientsig[4]) + FP9HandShake = true; + + serversig[-1] = type; + + r->Link.rc4keyIn = r->Link.rc4keyOut = 0; + + uptime = htonl(RTMP_GetTime()); + memcpy(serversig, &uptime, 4); + + if (FP9HandShake) + { + /* Server version */ + serversig[4] = 3; + serversig[5] = 5; + serversig[6] = 1; + serversig[7] = 1; + + getdig = digoff[offalg]; + getdh = dhoff[offalg]; + } + else + { + memset(&serversig[4], 0, 4); + } + + /* generate random data */ +#ifdef _DEBUG + memset(serversig+8, 0, RTMP_SIG_SIZE-8); +#else + ip = (int32_t *)(serversig+8); + for (i = 2; i < RTMP_SIG_SIZE/4; i++) + *ip++ = rand(); +#endif + + /* set handshake digest */ + if (FP9HandShake) + { + if (encrypted) + { + /* generate Diffie-Hellmann parameters */ + r->Link.dh = DHInit(1024); + if (!r->Link.dh) + { + RTMP_Log(RTMP_LOGERROR, "%s: Couldn't initialize Diffie-Hellmann!", + __FUNCTION__); + return false; + } + + dhposServer = getdh(serversig, RTMP_SIG_SIZE); + RTMP_Log(RTMP_LOGDEBUG, "%s: DH pubkey position: %d", __FUNCTION__, dhposServer); + + if (!DHGenerateKey(r->Link.dh)) + { + RTMP_Log(RTMP_LOGERROR, "%s: Couldn't generate Diffie-Hellmann public key!", + __FUNCTION__); + return false; + } + + if (!DHGetPublicKey + (r->Link.dh, (uint8_t *) &serversig[dhposServer], 128)) + { + RTMP_Log(RTMP_LOGERROR, "%s: Couldn't write public key!", __FUNCTION__); + return false; + } + } + + digestPosServer = getdig(serversig, RTMP_SIG_SIZE); /* reuse this value in verification */ + RTMP_Log(RTMP_LOGDEBUG, "%s: Server digest offset: %d", __FUNCTION__, + digestPosServer); + + CalculateDigest(digestPosServer, serversig, GenuineFMSKey, 36, + &serversig[digestPosServer]); + + RTMP_Log(RTMP_LOGDEBUG, "%s: Initial server digest: ", __FUNCTION__); + RTMP_LogHex(RTMP_LOGDEBUG, serversig + digestPosServer, + SHA256_DIGEST_LENGTH); + } + + RTMP_Log(RTMP_LOGDEBUG2, "Serversig: "); + RTMP_LogHex(RTMP_LOGDEBUG2, serversig, RTMP_SIG_SIZE); + + if (!WriteN(r, (char *)serversig-1, RTMP_SIG_SIZE + 1)) + return false; + + /* decode client response */ + memcpy(&uptime, clientsig, 4); + uptime = ntohl(uptime); + + RTMP_Log(RTMP_LOGDEBUG, "%s: Client Uptime : %d", __FUNCTION__, uptime); + RTMP_Log(RTMP_LOGDEBUG, "%s: Player Version: %d.%d.%d.%d", __FUNCTION__, clientsig[4], + clientsig[5], clientsig[6], clientsig[7]); + + if (FP9HandShake) + { + uint8_t digestResp[SHA256_DIGEST_LENGTH]; + uint8_t *signatureResp = NULL; + + /* we have to use this signature now to find the correct algorithms for getting the digest and DH positions */ + int digestPosClient = getdig(clientsig, RTMP_SIG_SIZE); + + if (!VerifyDigest(digestPosClient, clientsig, GenuineFPKey, 30)) + { + RTMP_Log(RTMP_LOGWARNING, "Trying different position for client digest!"); + offalg ^= 1; + getdig = digoff[offalg]; + getdh = dhoff[offalg]; + + digestPosClient = getdig(clientsig, RTMP_SIG_SIZE); + + if (!VerifyDigest(digestPosClient, clientsig, GenuineFPKey, 30)) + { + RTMP_Log(RTMP_LOGERROR, "Couldn't verify the client digest"); /* continuing anyway will probably fail */ + return false; + } + } + + /* generate SWFVerification token (SHA256 HMAC hash of decompressed SWF, key are the last 32 bytes of the server handshake) */ + if (r->Link.SWFSize) + { + const char swfVerify[] = { 0x01, 0x01 }; + char *vend = r->Link.SWFVerificationResponse+sizeof(r->Link.SWFVerificationResponse); + + memcpy(r->Link.SWFVerificationResponse, swfVerify, 2); + AMF_EncodeInt32(&r->Link.SWFVerificationResponse[2], vend, r->Link.SWFSize); + AMF_EncodeInt32(&r->Link.SWFVerificationResponse[6], vend, r->Link.SWFSize); + HMACsha256(r->Link.SWFHash, SHA256_DIGEST_LENGTH, + &serversig[RTMP_SIG_SIZE - SHA256_DIGEST_LENGTH], + SHA256_DIGEST_LENGTH, + (uint8_t *)&r->Link.SWFVerificationResponse[10]); + } + + /* do Diffie-Hellmann Key exchange for encrypted RTMP */ + if (encrypted) + { + int dhposClient, len; + /* compute secret key */ + uint8_t secretKey[128] = { 0 }; + + dhposClient = getdh(clientsig, RTMP_SIG_SIZE); + RTMP_Log(RTMP_LOGDEBUG, "%s: Client DH public key offset: %d", __FUNCTION__, + dhposClient); + len = + DHComputeSharedSecretKey(r->Link.dh, + (uint8_t *) &clientsig[dhposClient], 128, + secretKey); + if (len < 0) + { + RTMP_Log(RTMP_LOGDEBUG, "%s: Wrong secret key position!", __FUNCTION__); + return false; + } + + RTMP_Log(RTMP_LOGDEBUG, "%s: Secret key: ", __FUNCTION__); + RTMP_LogHex(RTMP_LOGDEBUG, secretKey, 128); + + InitRC4Encryption(secretKey, + (uint8_t *) &clientsig[dhposClient], + (uint8_t *) &serversig[dhposServer], + &keyIn, &keyOut); + } + + + /* calculate response now */ + signatureResp = clientsig+RTMP_SIG_SIZE-SHA256_DIGEST_LENGTH; + + HMACsha256(&clientsig[digestPosClient], SHA256_DIGEST_LENGTH, + GenuineFMSKey, sizeof(GenuineFMSKey), digestResp); + HMACsha256(clientsig, RTMP_SIG_SIZE - SHA256_DIGEST_LENGTH, digestResp, + SHA256_DIGEST_LENGTH, signatureResp); +#ifdef FP10 + if (type == 8 ) + { + uint8_t *dptr = digestResp; + uint8_t *sig = signatureResp; + /* encrypt signatureResp */ + for (i=0; iLink.rc4keyIn = keyIn; + r->Link.rc4keyOut = keyOut; + + /* update the keystreams */ + if (r->Link.rc4keyIn) + { + RC4_encrypt(r->Link.rc4keyIn, RTMP_SIG_SIZE, (uint8_t *) buff); + } + + if (r->Link.rc4keyOut) + { + RC4_encrypt(r->Link.rc4keyOut, RTMP_SIG_SIZE, (uint8_t *) buff); + } + } + } + else + { + if (memcmp(serversig, clientsig, RTMP_SIG_SIZE) != 0) + { + RTMP_Log(RTMP_LOGWARNING, "%s: client signature does not match!", + __FUNCTION__); + } + } + + RTMP_Log(RTMP_LOGDEBUG, "%s: Handshaking finished....", __FUNCTION__); + return true; +} diff --git a/gst/rtmp/hashswf.c b/gst/rtmp/hashswf.c new file mode 100644 index 0000000000..309dcf66bb --- /dev/null +++ b/gst/rtmp/hashswf.c @@ -0,0 +1,608 @@ +/* + * Copyright (C) 2009-2010 Howard Chu + * + * This file is part of librtmp. + * + * librtmp is free software; you can redistribute it and/or modify + * it under the terms of the GNU Lesser General Public License as + * published by the Free Software Foundation; either version 2.1, + * or (at your option) any later version. + * + * librtmp 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 Lesser General Public License + * along with librtmp see the file COPYING. If not, write to + * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. + * http://www.gnu.org/copyleft/lgpl.html + */ + +#include +#include +#include +#include +#include + +#include "rtmp_sys.h" +#include "log.h" +#include "http.h" + +#ifdef CRYPTO +#ifdef USE_POLARSSL +#include +#ifndef SHA256_DIGEST_LENGTH +#define SHA256_DIGEST_LENGTH 32 +#endif +#define HMAC_CTX sha2_context +#define HMAC_setup(ctx, key, len) sha2_hmac_starts(&ctx, (unsigned char *)key, len, 0) +#define HMAC_crunch(ctx, buf, len) sha2_hmac_update(&ctx, buf, len) +#define HMAC_finish(ctx, dig, dlen) dlen = SHA256_DIGEST_LENGTH; sha2_hmac_finish(&ctx, dig) +#define HMAC_close(ctx) +#elif defined(USE_GNUTLS) +#include +#include +#ifndef SHA256_DIGEST_LENGTH +#define SHA256_DIGEST_LENGTH 32 +#endif +#define HMAC_CTX gcry_md_hd_t +#define HMAC_setup(ctx, key, len) gcry_md_open(&ctx, GCRY_MD_SHA256, GCRY_MD_FLAG_HMAC); gcry_md_setkey(ctx, key, len) +#define HMAC_crunch(ctx, buf, len) gcry_md_write(ctx, buf, len) +#define HMAC_finish(ctx, dig, dlen) dlen = SHA256_DIGEST_LENGTH; memcpy(dig, gcry_md_read(ctx, 0), dlen) +#define HMAC_close(ctx) gcry_md_close(ctx) +#else /* USE_OPENSSL */ +#include +#include +#include +#include +#define HMAC_setup(ctx, key, len) HMAC_CTX_init(&ctx); HMAC_Init_ex(&ctx, (unsigned char *)key, len, EVP_sha256(), 0) +#define HMAC_crunch(ctx, buf, len) HMAC_Update(&ctx, (unsigned char *)buf, len) +#define HMAC_finish(ctx, dig, dlen) HMAC_Final(&ctx, (unsigned char *)dig, &dlen); +#define HMAC_close(ctx) HMAC_CTX_cleanup(&ctx) +#endif + +extern void RTMP_TLS_Init (); +extern TLS_CTX RTMP_TLS_ctx; + +#endif /* CRYPTO */ + +#include + +#define AGENT "Mozilla/5.0" + +HTTPResult +HTTP_get (struct HTTP_ctx *http, const char *url, HTTP_read_callback * cb) +{ + char *host, *path; + char *p1, *p2; + char hbuf[256]; + int port = 80; +#ifdef CRYPTO + int ssl = 0; +#endif + int hlen, flen = 0; + int rc, i; + int len_known; + HTTPResult ret = HTTPRES_OK; + struct sockaddr_in sa; + RTMPSockBuf sb = { 0 }; + + http->status = -1; + + memset (&sa, 0, sizeof (struct sockaddr_in)); + sa.sin_family = AF_INET; + + /* we only handle http here */ + if (strncasecmp (url, "http", 4)) + return HTTPRES_BAD_REQUEST; + + if (url[4] == 's') { +#ifdef CRYPTO + ssl = 1; + port = 443; + if (!RTMP_TLS_ctx) + RTMP_TLS_Init (); +#else + return HTTPRES_BAD_REQUEST; +#endif + } + + p1 = strchr (url + 4, ':'); + if (!p1 || strncmp (p1, "://", 3)) + return HTTPRES_BAD_REQUEST; + + host = p1 + 3; + path = strchr (host, '/'); + hlen = path - host; + strncpy (hbuf, host, hlen); + hbuf[hlen] = '\0'; + host = hbuf; + p1 = strrchr (host, ':'); + if (p1) { + *p1++ = '\0'; + port = atoi (p1); + } + + sa.sin_addr.s_addr = inet_addr (host); + if (sa.sin_addr.s_addr == INADDR_NONE) { + struct hostent *hp = gethostbyname (host); + if (!hp || !hp->h_addr) + return HTTPRES_LOST_CONNECTION; + sa.sin_addr = *(struct in_addr *) hp->h_addr; + } + sa.sin_port = htons (port); + sb.sb_socket = socket (AF_INET, SOCK_STREAM, IPPROTO_TCP); + if (sb.sb_socket == -1) + return HTTPRES_LOST_CONNECTION; + i = sprintf (sb.sb_buf, + "GET %s HTTP/1.0\r\nUser-Agent: %s\r\nHost: %s\r\nReferrer: %.*s\r\n", + path, AGENT, host, (int) (path - url + 1), url); + if (http->date[0]) + i += sprintf (sb.sb_buf + i, "If-Modified-Since: %s\r\n", http->date); + i += sprintf (sb.sb_buf + i, "\r\n"); + + if (connect + (sb.sb_socket, (struct sockaddr *) &sa, sizeof (struct sockaddr)) < 0) { + ret = HTTPRES_LOST_CONNECTION; + goto leave; + } +#ifdef CRYPTO + if (ssl) { +#ifdef NO_SSL + RTMP_Log (RTMP_LOGERROR, "%s, No SSL/TLS support", __FUNCTION__); + ret = HTTPRES_BAD_REQUEST; + goto leave; +#else + TLS_client (RTMP_TLS_ctx, sb.sb_ssl); + TLS_setfd (sb.sb_ssl, sb.sb_socket); + if ((i = TLS_connect (sb.sb_ssl)) < 0) { + RTMP_Log (RTMP_LOGERROR, "%s, TLS_Connect failed", __FUNCTION__); + ret = HTTPRES_LOST_CONNECTION; + goto leave; + } +#endif + } +#endif + RTMPSockBuf_Send (&sb, sb.sb_buf, i); + + /* set timeout */ +#define HTTP_TIMEOUT 5 + { + SET_RCVTIMEO (tv, HTTP_TIMEOUT); + if (setsockopt + (sb.sb_socket, SOL_SOCKET, SO_RCVTIMEO, (char *) &tv, sizeof (tv))) { + RTMP_Log (RTMP_LOGERROR, "%s, Setting socket timeout to %ds failed!", + __FUNCTION__, HTTP_TIMEOUT); + } + } + + sb.sb_size = 0; + sb.sb_timedout = false; + if (RTMPSockBuf_Fill (&sb) < 1) { + ret = HTTPRES_LOST_CONNECTION; + goto leave; + } + if (strncmp (sb.sb_buf, "HTTP/1", 6)) { + ret = HTTPRES_BAD_REQUEST; + goto leave; + } + + p1 = strchr (sb.sb_buf, ' '); + rc = atoi (p1 + 1); + http->status = rc; + + if (rc >= 300) { + if (rc == 304) { + ret = HTTPRES_OK_NOT_MODIFIED; + goto leave; + } else if (rc == 404) + ret = HTTPRES_NOT_FOUND; + else if (rc >= 500) + ret = HTTPRES_SERVER_ERROR; + else if (rc >= 400) + ret = HTTPRES_BAD_REQUEST; + else + ret = HTTPRES_REDIRECTED; + } + + p1 = memchr (sb.sb_buf, '\n', sb.sb_size); + if (!p1) { + ret = HTTPRES_BAD_REQUEST; + goto leave; + } + sb.sb_start = p1 + 1; + sb.sb_size -= sb.sb_start - sb.sb_buf; + + while ((p2 = memchr (sb.sb_start, '\r', sb.sb_size))) { + if (*sb.sb_start == '\r') { + sb.sb_start += 2; + sb.sb_size -= 2; + break; + } else + if (!strncasecmp + (sb.sb_start, "Content-Length: ", sizeof ("Content-Length: ") - 1)) { + flen = atoi (sb.sb_start + sizeof ("Content-Length: ") - 1); + } else + if (!strncasecmp + (sb.sb_start, "Last-Modified: ", sizeof ("Last-Modified: ") - 1)) { + *p2 = '\0'; + strcpy (http->date, sb.sb_start + sizeof ("Last-Modified: ") - 1); + } + p2 += 2; + sb.sb_size -= p2 - sb.sb_start; + sb.sb_start = p2; + if (sb.sb_size < 1) { + if (RTMPSockBuf_Fill (&sb) < 1) { + ret = HTTPRES_LOST_CONNECTION; + goto leave; + } + } + } + + len_known = flen > 0; + while ((!len_known || flen > 0) && + (sb.sb_size > 0 || RTMPSockBuf_Fill (&sb) > 0)) { + cb (sb.sb_start, 1, sb.sb_size, http->data); + if (len_known) + flen -= sb.sb_size; + http->size += sb.sb_size; + sb.sb_size = 0; + } + + if (flen > 0) + ret = HTTPRES_LOST_CONNECTION; + +leave: + RTMPSockBuf_Close (&sb); + return ret; +} + +#ifdef CRYPTO + +#define CHUNK 16384 + +struct info +{ + z_stream *zs; + HMAC_CTX ctx; + int first; + int zlib; + int size; +}; + +static size_t +swfcrunch (void *ptr, size_t size, size_t nmemb, void *stream) +{ + struct info *i = stream; + char *p = ptr; + size_t len = size * nmemb; + + if (i->first) { + i->first = 0; + /* compressed? */ + if (!strncmp (p, "CWS", 3)) { + *p = 'F'; + i->zlib = 1; + } + HMAC_crunch (i->ctx, (unsigned char *) p, 8); + p += 8; + len -= 8; + i->size = 8; + } + + if (i->zlib) { + unsigned char out[CHUNK]; + i->zs->next_in = (unsigned char *) p; + i->zs->avail_in = len; + do { + i->zs->avail_out = CHUNK; + i->zs->next_out = out; + inflate (i->zs, Z_NO_FLUSH); + len = CHUNK - i->zs->avail_out; + i->size += len; + HMAC_crunch (i->ctx, out, len); + } + while (i->zs->avail_out == 0); + } else { + i->size += len; + HMAC_crunch (i->ctx, (unsigned char *) p, len); + } + return size * nmemb; +} + +static int tzoff; +static int tzchecked; + +#define JAN02_1980 318340800 + +static const char *monthtab[12] = { "Jan", "Feb", "Mar", + "Apr", "May", "Jun", + "Jul", "Aug", "Sep", + "Oct", "Nov", "Dec" +}; +static const char *days[] = { "Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat" }; + +/* Parse an HTTP datestamp into Unix time */ +static time_t +make_unix_time (char *s) +{ + struct tm time; + int i, ysub = 1900, fmt = 0; + char *month; + char *n; + time_t res; + + if (s[3] != ' ') { + fmt = 1; + if (s[3] != ',') + ysub = 0; + } + for (n = s; *n; ++n) + if (*n == '-' || *n == ':') + *n = ' '; + + time.tm_mon = 0; + n = strchr (s, ' '); + if (fmt) { + /* Day, DD-MMM-YYYY HH:MM:SS GMT */ + time.tm_mday = strtol (n + 1, &n, 0); + month = n + 1; + n = strchr (month, ' '); + time.tm_year = strtol (n + 1, &n, 0); + time.tm_hour = strtol (n + 1, &n, 0); + time.tm_min = strtol (n + 1, &n, 0); + time.tm_sec = strtol (n + 1, NULL, 0); + } else { + /* Unix ctime() format. Does not conform to HTTP spec. */ + /* Day MMM DD HH:MM:SS YYYY */ + month = n + 1; + n = strchr (month, ' '); + while (isspace (*n)) + n++; + time.tm_mday = strtol (n, &n, 0); + time.tm_hour = strtol (n + 1, &n, 0); + time.tm_min = strtol (n + 1, &n, 0); + time.tm_sec = strtol (n + 1, &n, 0); + time.tm_year = strtol (n + 1, NULL, 0); + } + if (time.tm_year > 100) + time.tm_year -= ysub; + + for (i = 0; i < 12; i++) + if (!strncasecmp (month, monthtab[i], 3)) { + time.tm_mon = i; + break; + } + time.tm_isdst = 0; /* daylight saving is never in effect in GMT */ + + /* this is normally the value of extern int timezone, but some + * braindead C libraries don't provide it. + */ + if (!tzchecked) { + struct tm *tc; + time_t then = JAN02_1980; + tc = localtime (&then); + tzoff = (12 - tc->tm_hour) * 3600 + tc->tm_min * 60 + tc->tm_sec; + tzchecked = 1; + } + res = mktime (&time); + /* Unfortunately, mktime() assumes the input is in local time, + * not GMT, so we have to correct it here. + */ + if (res != -1) + res += tzoff; + return res; +} + +/* Convert a Unix time to a network time string + * Weekday, DD-MMM-YYYY HH:MM:SS GMT + */ +static void +strtime (time_t * t, char *s) +{ + struct tm *tm; + + tm = gmtime ((time_t *) t); + sprintf (s, "%s, %02d %s %d %02d:%02d:%02d GMT", + days[tm->tm_wday], tm->tm_mday, monthtab[tm->tm_mon], + tm->tm_year + 1900, tm->tm_hour, tm->tm_min, tm->tm_sec); +} + +#define HEX2BIN(a) (((a)&0x40)?((a)&0xf)+9:((a)&0xf)) + +int +RTMP_HashSWF (const char *url, unsigned int *size, unsigned char *hash, int age) +{ + FILE *f = NULL; + char *path, date[64], cctim[64]; + long pos = 0; + time_t ctim = -1, cnow; + int i, got = 0, ret = 0; + unsigned int hlen; + struct info in = { 0 }; + struct HTTP_ctx http = { 0 }; + HTTPResult httpres; + z_stream zs = { 0 }; + AVal home, hpre; + + date[0] = '\0'; +#ifdef _WIN32 +#ifdef _XBOX + hpre.av_val = "Q:"; + hpre.av_len = 2; + home.av_val = "\\UserData"; +#else + hpre.av_val = getenv ("HOMEDRIVE"); + hpre.av_len = strlen (hpre.av_val); + home.av_val = getenv ("HOMEPATH"); +#endif +#define DIRSEP "\\" + +#else /* !_WIN32 */ + hpre.av_val = (char *) ""; + hpre.av_len = 0; + home.av_val = getenv ("HOME"); +#define DIRSEP "/" +#endif + if (!home.av_val) + home.av_val = (char *) "."; + home.av_len = strlen (home.av_val); + + /* SWF hash info is cached in a fixed-format file. + * url: + * ctim: HTTP datestamp of when we last checked it. + * date: HTTP datestamp of the SWF's last modification. + * size: SWF size in hex + * hash: SWF hash in hex + * + * These fields must be present in this order. All fields + * besides URL are fixed size. + */ + path = malloc (hpre.av_len + home.av_len + sizeof (DIRSEP ".swfinfo")); + sprintf (path, "%s%s" DIRSEP ".swfinfo", hpre.av_val, home.av_val); + + f = fopen (path, "r+"); + while (f) { + char buf[4096], *file, *p; + + file = strchr (url, '/'); + if (!file) + break; + file += 2; + file = strchr (file, '/'); + if (!file) + break; + file++; + hlen = file - url; + p = strrchr (file, '/'); + if (p) + file = p; + else + file--; + + while (fgets (buf, sizeof (buf), f)) { + char *r1; + + got = 0; + + if (strncmp (buf, "url: ", 5)) + continue; + if (strncmp (buf + 5, url, hlen)) + continue; + r1 = strrchr (buf, '/'); + i = strlen (r1); + r1[--i] = '\0'; + if (strncmp (r1, file, i)) + continue; + pos = ftell (f); + while (got < 4 && fgets (buf, sizeof (buf), f)) { + if (!strncmp (buf, "size: ", 6)) { + *size = strtol (buf + 6, NULL, 16); + got++; + } else if (!strncmp (buf, "hash: ", 6)) { + unsigned char *ptr = hash, *in = (unsigned char *) buf + 6; + int l = strlen ((char *) in) - 1; + for (i = 0; i < l; i += 2) + *ptr++ = (HEX2BIN (in[i]) << 4) | HEX2BIN (in[i + 1]); + got++; + } else if (!strncmp (buf, "date: ", 6)) { + buf[strlen (buf) - 1] = '\0'; + strncpy (date, buf + 6, sizeof (date)); + got++; + } else if (!strncmp (buf, "ctim: ", 6)) { + buf[strlen (buf) - 1] = '\0'; + ctim = make_unix_time (buf + 6); + got++; + } else if (!strncmp (buf, "url: ", 5)) + break; + } + break; + } + break; + } + + cnow = time (NULL); + /* If we got a cache time, see if it's young enough to use directly */ + if (age && ctim > 0) { + ctim = cnow - ctim; + ctim /= 3600 * 24; /* seconds to days */ + if (ctim < age) /* ok, it's new enough */ + goto out; + } + + in.first = 1; + HMAC_setup (in.ctx, "Genuine Adobe Flash Player 001", 30); + inflateInit (&zs); + in.zs = &zs; + + http.date = date; + http.data = ∈ + + httpres = HTTP_get (&http, url, swfcrunch); + + inflateEnd (&zs); + + if (httpres != HTTPRES_OK && httpres != HTTPRES_OK_NOT_MODIFIED) { + ret = -1; + if (httpres == HTTPRES_LOST_CONNECTION) + RTMP_Log (RTMP_LOGERROR, + "%s: connection lost while downloading swfurl %s", __FUNCTION__, url); + else if (httpres == HTTPRES_NOT_FOUND) + RTMP_Log (RTMP_LOGERROR, "%s: swfurl %s not found", __FUNCTION__, url); + else + RTMP_Log (RTMP_LOGERROR, "%s: couldn't contact swfurl %s (HTTP error %d)", + __FUNCTION__, url, http.status); + } else { + if (got && pos) + fseek (f, pos, SEEK_SET); + else { + char *q; + if (!f) + f = fopen (path, "w"); + if (!f) { + int err = errno; + RTMP_Log (RTMP_LOGERROR, + "%s: couldn't open %s for writing, errno %d (%s)", + __FUNCTION__, path, err, strerror (err)); + ret = -1; + goto out; + } + fseek (f, 0, SEEK_END); + q = strchr (url, '?'); + if (q) + i = q - url; + else + i = strlen (url); + + fprintf (f, "url: %.*s\n", i, url); + } + strtime (&cnow, cctim); + fprintf (f, "ctim: %s\n", cctim); + + if (!in.first) { + HMAC_finish (in.ctx, hash, hlen); + *size = in.size; + + fprintf (f, "date: %s\n", date); + fprintf (f, "size: %08x\n", in.size); + fprintf (f, "hash: "); + for (i = 0; i < SHA256_DIGEST_LENGTH; i++) + fprintf (f, "%02x", hash[i]); + fprintf (f, "\n"); + } + } + HMAC_close (in.ctx); +out: + free (path); + if (f) + fclose (f); + return ret; +} +#else +int +RTMP_HashSWF (const char *url, unsigned int *size, unsigned char *hash, int age) +{ + return -1; +} +#endif diff --git a/gst/rtmp/http.h b/gst/rtmp/http.h new file mode 100644 index 0000000000..be2106f8ab --- /dev/null +++ b/gst/rtmp/http.h @@ -0,0 +1,46 @@ +#ifndef __RTMP_HTTP_H__ +#define __RTMP_HTTP_H__ +/* + * Copyright (C) 2010 Howard Chu + * Copyright (C) 2010 Antti Ajanki + * + * This file is part of librtmp. + * + * librtmp is free software; you can redistribute it and/or modify + * it under the terms of the GNU Lesser General Public License as + * published by the Free Software Foundation; either version 2.1, + * or (at your option) any later version. + * + * librtmp 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 Lesser General Public License + * along with librtmp see the file COPYING. If not, write to + * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. + * http://www.gnu.org/copyleft/lgpl.html + */ + +typedef enum { + HTTPRES_OK, /* result OK */ + HTTPRES_OK_NOT_MODIFIED, /* not modified since last request */ + HTTPRES_NOT_FOUND, /* not found */ + HTTPRES_BAD_REQUEST, /* client error */ + HTTPRES_SERVER_ERROR, /* server reported an error */ + HTTPRES_REDIRECTED, /* resource has been moved */ + HTTPRES_LOST_CONNECTION /* connection lost while waiting for data */ +} HTTPResult; + +struct HTTP_ctx { + char *date; + int size; + int status; + void *data; +}; + +typedef size_t (HTTP_read_callback)(void *ptr, size_t size, size_t nmemb, void *stream); + +HTTPResult HTTP_get(struct HTTP_ctx *http, const char *url, HTTP_read_callback *cb); + +#endif diff --git a/gst/rtmp/log.c b/gst/rtmp/log.c new file mode 100644 index 0000000000..5e1c2e8380 --- /dev/null +++ b/gst/rtmp/log.c @@ -0,0 +1,233 @@ +/* + * Copyright (C) 2008-2009 Andrej Stepanchuk + * Copyright (C) 2009-2010 Howard Chu + * + * This file is part of librtmp. + * + * librtmp is free software; you can redistribute it and/or modify + * it under the terms of the GNU Lesser General Public License as + * published by the Free Software Foundation; either version 2.1, + * or (at your option) any later version. + * + * librtmp 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 Lesser General Public License + * along with librtmp see the file COPYING. If not, write to + * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. + * http://www.gnu.org/copyleft/lgpl.html + */ + +#include +#include +#include +#include +#include + +#include "rtmp_sys.h" +#include "log.h" + +#define MAX_PRINT_LEN 2048 + +RTMP_LogLevel RTMP_debuglevel = RTMP_LOGERROR; + +static int neednl; + +static FILE *fmsg; + +static RTMP_LogCallback rtmp_log_default, *cb = rtmp_log_default; + +static const char *levels[] = { + "CRIT", "ERROR", "WARNING", "INFO", + "DEBUG", "DEBUG2" +}; + +static void +rtmp_log_default (int level, const char *format, va_list vl) +{ + char str[MAX_PRINT_LEN] = ""; + + vsnprintf (str, MAX_PRINT_LEN - 1, format, vl); + + /* Filter out 'no-name' */ + if (RTMP_debuglevel < RTMP_LOGALL && strstr (str, "no-name") != NULL) + return; + + if (!fmsg) + fmsg = stderr; + + if (level <= RTMP_debuglevel) { + if (neednl) { + putc ('\n', fmsg); + neednl = 0; + } + fprintf (fmsg, "%s: %s\n", levels[level], str); +#ifdef _DEBUG + fflush (fmsg); +#endif + } +} + +void +RTMP_LogSetOutput (FILE * file) +{ + fmsg = file; +} + +void +RTMP_LogSetLevel (RTMP_LogLevel level) +{ + RTMP_debuglevel = level; +} + +void +RTMP_LogSetCallback (RTMP_LogCallback * cbp) +{ + cb = cbp; +} + +RTMP_LogLevel +RTMP_LogGetLevel (void) +{ + return RTMP_debuglevel; +} + +void +RTMP_Log (int level, const char *format, ...) +{ + va_list args; + va_start (args, format); + cb (level, format, args); + va_end (args); +} + +static const char hexdig[] = "0123456789abcdef"; + +void +RTMP_LogHex (int level, const uint8_t * data, unsigned long len) +{ + unsigned long i; + char line[50], *ptr; + + if (level > RTMP_debuglevel) + return; + + ptr = line; + + for (i = 0; i < len; i++) { + *ptr++ = hexdig[0x0f & (data[i] >> 4)]; + *ptr++ = hexdig[0x0f & data[i]]; + if ((i & 0x0f) == 0x0f) { + *ptr = '\0'; + ptr = line; + RTMP_Log (level, "%s", line); + } else { + *ptr++ = ' '; + } + } + if (i & 0x0f) { + *ptr = '\0'; + RTMP_Log (level, "%s", line); + } +} + +void +RTMP_LogHexString (int level, const uint8_t * data, unsigned long len) +{ +#define BP_OFFSET 9 +#define BP_GRAPH 60 +#define BP_LEN 80 + char line[BP_LEN]; + unsigned long i; + + if (!data || level > RTMP_debuglevel) + return; + + /* in case len is zero */ + line[0] = '\0'; + + for (i = 0; i < len; i++) { + int n = i % 16; + unsigned off; + + if (!n) { + if (i) + RTMP_Log (level, "%s", line); + memset (line, ' ', sizeof (line) - 2); + line[sizeof (line) - 2] = '\0'; + + off = i % 0x0ffffU; + + line[2] = hexdig[0x0f & (off >> 12)]; + line[3] = hexdig[0x0f & (off >> 8)]; + line[4] = hexdig[0x0f & (off >> 4)]; + line[5] = hexdig[0x0f & off]; + line[6] = ':'; + } + + off = BP_OFFSET + n * 3 + ((n >= 8) ? 1 : 0); + line[off] = hexdig[0x0f & (data[i] >> 4)]; + line[off + 1] = hexdig[0x0f & data[i]]; + + off = BP_GRAPH + n + ((n >= 8) ? 1 : 0); + + if (isprint (data[i])) { + line[BP_GRAPH + n] = data[i]; + } else { + line[BP_GRAPH + n] = '.'; + } + } + + RTMP_Log (level, "%s", line); +} + +/* These should only be used by apps, never by the library itself */ +void +RTMP_LogPrintf (const char *format, ...) +{ + char str[MAX_PRINT_LEN] = ""; + int len; + va_list args; + va_start (args, format); + len = vsnprintf (str, MAX_PRINT_LEN - 1, format, args); + va_end (args); + + if (RTMP_debuglevel == RTMP_LOGCRIT) + return; + + if (!fmsg) + fmsg = stderr; + + if (neednl) { + putc ('\n', fmsg); + neednl = 0; + } + + if (len > MAX_PRINT_LEN - 1) + len = MAX_PRINT_LEN - 1; + fprintf (fmsg, "%s", str); + if (str[len - 1] == '\n') + fflush (fmsg); +} + +void +RTMP_LogStatus (const char *format, ...) +{ + char str[MAX_PRINT_LEN] = ""; + va_list args; + va_start (args, format); + vsnprintf (str, MAX_PRINT_LEN - 1, format, args); + va_end (args); + + if (RTMP_debuglevel == RTMP_LOGCRIT) + return; + + if (!fmsg) + fmsg = stderr; + + fprintf (fmsg, "%s", str); + fflush (fmsg); + neednl = 1; +} diff --git a/gst/rtmp/log.h b/gst/rtmp/log.h new file mode 100644 index 0000000000..ebc1605f04 --- /dev/null +++ b/gst/rtmp/log.h @@ -0,0 +1,62 @@ +/* + * Copyright (C) 2008-2009 Andrej Stepanchuk + * Copyright (C) 2009-2010 Howard Chu + * + * This file is part of librtmp. + * + * librtmp is free software; you can redistribute it and/or modify + * it under the terms of the GNU Lesser General Public License as + * published by the Free Software Foundation; either version 2.1, + * or (at your option) any later version. + * + * librtmp 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 Lesser General Public License + * along with librtmp see the file COPYING. If not, write to + * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. + * http://www.gnu.org/copyleft/lgpl.html + */ + +#ifndef __RTMP_LOG_H__ +#define __RTMP_LOG_H__ + +#include +#include +#include + +#ifdef __cplusplus +extern "C" { +#endif +/* Enable this to get full debugging output */ +/* #define _DEBUG */ + +#ifdef _DEBUG +#undef NODEBUG +#endif + +typedef enum +{ RTMP_LOGCRIT=0, RTMP_LOGERROR, RTMP_LOGWARNING, RTMP_LOGINFO, + RTMP_LOGDEBUG, RTMP_LOGDEBUG2, RTMP_LOGALL +} RTMP_LogLevel; + +extern RTMP_LogLevel RTMP_debuglevel; + +typedef void (RTMP_LogCallback)(int level, const char *fmt, va_list); +void RTMP_LogSetCallback(RTMP_LogCallback *cb); +void RTMP_LogSetOutput(FILE *file); +void RTMP_LogPrintf(const char *format, ...); +void RTMP_LogStatus(const char *format, ...); +void RTMP_Log(int level, const char *format, ...); +void RTMP_LogHex(int level, const uint8_t *data, unsigned long len); +void RTMP_LogHexString(int level, const uint8_t *data, unsigned long len); +void RTMP_LogSetLevel(RTMP_LogLevel lvl); +RTMP_LogLevel RTMP_LogGetLevel(void); + +#ifdef __cplusplus +} +#endif + +#endif diff --git a/gst/rtmp/parseurl.c b/gst/rtmp/parseurl.c new file mode 100644 index 0000000000..8edf54ea1e --- /dev/null +++ b/gst/rtmp/parseurl.c @@ -0,0 +1,281 @@ +/* + * Copyright (C) 2009 Andrej Stepanchuk + * Copyright (C) 2009-2010 Howard Chu + * + * This file is part of librtmp. + * + * librtmp is free software; you can redistribute it and/or modify + * it under the terms of the GNU Lesser General Public License as + * published by the Free Software Foundation; either version 2.1, + * or (at your option) any later version. + * + * librtmp 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 Lesser General Public License + * along with librtmp see the file COPYING. If not, write to + * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. + * http://www.gnu.org/copyleft/lgpl.html + */ + +#include +#include + +#include +#include + +#include "rtmp_sys.h" +#include "log.h" + +bool +RTMP_ParseURL (const char *url, int *protocol, AVal * host, unsigned int *port, + AVal * playpath, AVal * app) +{ + char *p, *end, *col, *ques, *slash; + + RTMP_Log (RTMP_LOGDEBUG, "Parsing..."); + + *protocol = RTMP_PROTOCOL_RTMP; + *port = 0; + playpath->av_len = 0; + playpath->av_val = NULL; + app->av_len = 0; + app->av_val = NULL; + + /* Old School Parsing */ + + /* look for usual :// pattern */ + p = strstr (url, "://"); + if (!p) { + RTMP_Log (RTMP_LOGERROR, "RTMP URL: No :// in url!"); + return false; + } + { + 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; + } + } + + RTMP_Log (RTMP_LOGDEBUG, "Parsed protocol: %d", *protocol); + +parsehost: + /* let's get the hostname */ + p += 3; + + /* check for sudden death */ + if (*p == 0) { + RTMP_Log (RTMP_LOGWARNING, "No hostname in URL!"); + return false; + } + + end = p + strlen (p); + col = strchr (p, ':'); + ques = strchr (p, '?'); + slash = strchr (p, '/'); + + { + int hostlen; + if (slash) + hostlen = slash - p; + else + hostlen = end - p; + if (col && col - p < hostlen) + hostlen = col - p; + + if (hostlen < 256) { + host->av_val = p; + host->av_len = hostlen; + RTMP_Log (RTMP_LOGDEBUG, "Parsed host : %.*s", hostlen, host->av_val); + } else { + RTMP_Log (RTMP_LOGWARNING, "Hostname exceeds 255 characters!"); + } + + p += hostlen; + } + + /* get the port number if available */ + if (*p == ':') { + unsigned int p2; + p++; + p2 = atoi (p); + if (p2 > 65535) { + RTMP_Log (RTMP_LOGWARNING, "Invalid port number!"); + } else { + *port = p2; + } + } + + if (!slash) { + RTMP_Log (RTMP_LOGWARNING, "No application or playpath in URL!"); + return true; + } + p = slash + 1; + + { + /* parse application + * + * rtmp://host[:port]/app[/appinstance][/...] + * application = app[/appinstance] + */ + + char *slash2, *slash3 = NULL; + int applen, appnamelen; + + slash2 = strchr (p, '/'); + if (slash2) + slash3 = strchr (slash2 + 1, '/'); + + applen = end - p; /* ondemand, pass all parameters as app */ + appnamelen = applen; /* ondemand length */ + + if (ques && strstr (p, "slist=")) { /* whatever it is, the '?' and slist= means we need to use everything as app and parse plapath from slist= */ + appnamelen = ques - p; + } else if (strncmp (p, "ondemand/", 9) == 0) { + /* app = ondemand/foobar, only pass app=ondemand */ + applen = 8; + appnamelen = 8; + } else { /* app!=ondemand, so app is app[/appinstance] */ + if (slash3) + appnamelen = slash3 - p; + else if (slash2) + appnamelen = slash2 - p; + + applen = appnamelen; + } + + app->av_val = p; + app->av_len = applen; + RTMP_Log (RTMP_LOGDEBUG, "Parsed app : %.*s", applen, p); + + p += appnamelen; + } + + if (*p == '/') + p++; + + if (end - p) { + AVal av = { p, end - p }; + RTMP_ParsePlaypath (&av, playpath); + } + + return true; +} + +/* + * Extracts playpath from RTMP URL. playpath is the file part of the + * URL, i.e. the part that comes after rtmp://host:port/app/ + * + * Returns the stream name in a format understood by FMS. The name is + * the playpath part of the URL with formatting depending on the stream + * type: + * + * mp4 streams: prepend "mp4:", remove extension + * mp3 streams: prepend "mp3:", remove extension + * flv streams: remove extension + */ +void +RTMP_ParsePlaypath (AVal * in, AVal * out) +{ + int addMP4 = 0; + int addMP3 = 0; + int subExt = 0; + const char *playpath = in->av_val; + const char *temp, *q, *ext = NULL; + const char *ppstart = playpath; + char *streamname, *destptr, *p; + + int pplen = in->av_len; + + out->av_val = NULL; + out->av_len = 0; + + if ((*ppstart == '?') && (temp = strstr (ppstart, "slist=")) != 0) { + ppstart = temp + 6; + pplen = strlen (ppstart); + + temp = strchr (ppstart, '&'); + if (temp) { + pplen = temp - ppstart; + } + } + + q = strchr (ppstart, '?'); + if (pplen >= 4) { + if (q) + ext = q - 4; + else + ext = &ppstart[pplen - 4]; + if ((strncmp (ext, ".f4v", 4) == 0) || (strncmp (ext, ".mp4", 4) == 0)) { + addMP4 = 1; + subExt = 1; + /* Only remove .flv from rtmp URL, not slist params */ + } else if ((ppstart == playpath) && (strncmp (ext, ".flv", 4) == 0)) { + subExt = 1; + } else if (strncmp (ext, ".mp3", 4) == 0) { + addMP3 = 1; + subExt = 1; + } + } + + streamname = (char *) malloc ((pplen + 4 + 1) * sizeof (char)); + if (!streamname) + return; + + destptr = streamname; + if (addMP4) { + if (strncmp (ppstart, "mp4:", 4)) { + strcpy (destptr, "mp4:"); + destptr += 4; + } else { + subExt = 0; + } + } else if (addMP3) { + if (strncmp (ppstart, "mp3:", 4)) { + strcpy (destptr, "mp3:"); + destptr += 4; + } else { + subExt = 0; + } + } + + for (p = (char *) ppstart; pplen > 0;) { + /* skip extension */ + if (subExt && p == ext) { + p += 4; + pplen -= 4; + } + if (*p == '%') { + unsigned int c; + sscanf (p + 1, "%02x", &c); + *destptr++ = c; + pplen -= 3; + p += 3; + } else { + *destptr++ = *p++; + pplen--; + } + } + *destptr = '\0'; + + out->av_val = streamname; + out->av_len = destptr - streamname; +} diff --git a/gst/rtmp/rtmp.c b/gst/rtmp/rtmp.c new file mode 100644 index 0000000000..548464dc04 --- /dev/null +++ b/gst/rtmp/rtmp.c @@ -0,0 +1,4018 @@ +/* + * Copyright (C) 2005-2008 Team XBMC + * http://www.xbmc.org + * Copyright (C) 2008-2009 Andrej Stepanchuk + * Copyright (C) 2009-2010 Howard Chu + * + * This file is part of librtmp. + * + * librtmp is free software; you can redistribute it and/or modify + * it under the terms of the GNU Lesser General Public License as + * published by the Free Software Foundation; either version 2.1, + * or (at your option) any later version. + * + * librtmp 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 Lesser General Public License + * along with librtmp see the file COPYING. If not, write to + * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. + * http://www.gnu.org/copyleft/lgpl.html + */ + +#include +#include +#include +#include + +#include "rtmp_sys.h" +#include "log.h" + +#ifdef CRYPTO +#ifdef USE_POLARSSL +#include +#elif defined(USE_GNUTLS) +#include +#else /* USE_OPENSSL */ +#include +#include +#endif +TLS_CTX RTMP_TLS_ctx; +#endif + +#define RTMP_SIG_SIZE 1536 +#define RTMP_LARGE_HEADER_SIZE 12 + +static const int packetSize[] = { 12, 8, 4, 1 }; + +bool RTMP_ctrlC; + +const char RTMPProtocolStrings[][7] = { + "RTMP", + "RTMPT", + "RTMPE", + "RTMPTE", + "RTMPS", + "RTMPTS", + "", + "", + "RTMFP" +}; + +const char RTMPProtocolStringsLower[][7] = { + "rtmp", + "rtmpt", + "rtmpe", + "rtmpte", + "rtmps", + "rtmpts", + "", + "", + "rtmfp" +}; + +static const char *RTMPT_cmds[] = { + "open", + "send", + "idle", + "close" +}; + +typedef enum +{ + RTMPT_OPEN = 0, RTMPT_SEND, RTMPT_IDLE, RTMPT_CLOSE +} RTMPTCmd; + +static bool DumpMetaData (AMFObject * obj); +static bool HandShake (RTMP * r, bool FP9HandShake); +static bool SocksNegotiate (RTMP * r); + +static bool SendConnectPacket (RTMP * r, RTMPPacket * cp); +static bool SendCheckBW (RTMP * r); +static bool SendCheckBWResult (RTMP * r, double txn); +static bool SendDeleteStream (RTMP * r, double dStreamId); +static bool SendFCSubscribe (RTMP * r, AVal * subscribepath); +static bool SendPlay (RTMP * r); +static bool SendBytesReceived (RTMP * r); + +#if 0 /* unused */ +static bool SendBGHasStream (RTMP * r, double dId, AVal * playpath); +#endif + +static int HandleInvoke (RTMP * r, const char *body, unsigned int nBodySize); +static bool HandleMetadata (RTMP * r, char *body, unsigned int len); +static void HandleChangeChunkSize (RTMP * r, const RTMPPacket * packet); +static void HandleAudio (RTMP * r, const RTMPPacket * packet); +static void HandleVideo (RTMP * r, const RTMPPacket * packet); +static void HandleCtrl (RTMP * r, const RTMPPacket * packet); +static void HandleServerBW (RTMP * r, const RTMPPacket * packet); +static void HandleClientBW (RTMP * r, const RTMPPacket * packet); + +static int ReadN (RTMP * r, char *buffer, int n); +static bool WriteN (RTMP * r, const char *buffer, int n); + +static void DecodeTEA (AVal * key, AVal * text); + +static int HTTP_Post (RTMP * r, RTMPTCmd cmd, const char *buf, int len); +static int HTTP_read (RTMP * r, int fill); + +#ifndef _WIN32 +static int clk_tck; +#endif + +#ifdef CRYPTO +#include "handshake.h" +#endif + +uint32_t +RTMP_GetTime (void) +{ +#ifdef _DEBUG + return 0; +#elif defined(_WIN32) + return timeGetTime (); +#else + struct tms t; + if (!clk_tck) + clk_tck = sysconf (_SC_CLK_TCK); + return times (&t) * 1000 / clk_tck; +#endif +} + +void +RTMP_UserInterrupt (void) +{ + RTMP_ctrlC = true; +} + +void +RTMPPacket_Reset (RTMPPacket * p) +{ + p->m_headerType = 0; + p->m_packetType = 0; + p->m_nChannel = 0; + p->m_nTimeStamp = 0; + p->m_nInfoField2 = 0; + p->m_hasAbsTimestamp = false; + p->m_nBodySize = 0; + p->m_nBytesRead = 0; +} + +bool +RTMPPacket_Alloc (RTMPPacket * p, int nSize) +{ + char *ptr = calloc (1, nSize + RTMP_MAX_HEADER_SIZE); + if (!ptr) + return false; + p->m_body = ptr + RTMP_MAX_HEADER_SIZE; + p->m_nBytesRead = 0; + return true; +} + +void +RTMPPacket_Free (RTMPPacket * p) +{ + if (p->m_body) { + free (p->m_body - RTMP_MAX_HEADER_SIZE); + p->m_body = NULL; + } +} + +void +RTMPPacket_Dump (RTMPPacket * p) +{ + RTMP_Log (RTMP_LOGDEBUG, + "RTMP PACKET: packet type: 0x%02x. channel: 0x%02x. info 1: %d info 2: %d. Body size: %lu. body: 0x%02x", + p->m_packetType, p->m_nChannel, p->m_nTimeStamp, p->m_nInfoField2, + p->m_nBodySize, p->m_body ? (unsigned char) p->m_body[0] : 0); +} + +int +RTMP_LibVersion (void) +{ + return RTMP_LIB_VERSION; +} + +void RTMP_TLS_Init (void); + +void +RTMP_TLS_Init (void) +{ +#ifdef CRYPTO +#ifdef USE_POLARSSL + /* Do this regardless of NO_SSL, we use havege for rtmpe too */ + RTMP_TLS_ctx = calloc (1, sizeof (struct tls_ctx)); + havege_init (&RTMP_TLS_ctx->hs); +#elif defined(USE_GNUTLS) && !defined(NO_SSL) + /* Technically we need to initialize libgcrypt ourselves if + * we're not going to call gnutls_global_init(). Ignoring this + * for now. + */ + gnutls_global_init (); + RTMP_TLS_ctx = malloc (sizeof (struct tls_ctx)); + gnutls_certificate_allocate_credentials (&RTMP_TLS_ctx->cred); + gnutls_priority_init (&RTMP_TLS_ctx->prios, "NORMAL", NULL); + gnutls_certificate_set_x509_trust_file (RTMP_TLS_ctx->cred, + "ca.pem", GNUTLS_X509_FMT_PEM); +#elif !defined(NO_SSL) /* USE_OPENSSL */ + /* libcrypto doesn't need anything special */ + SSL_load_error_strings (); + SSL_library_init (); + OpenSSL_add_all_digests (); + RTMP_TLS_ctx = SSL_CTX_new (SSLv23_method ()); + SSL_CTX_set_options (RTMP_TLS_ctx, SSL_OP_ALL); + SSL_CTX_set_default_verify_paths (RTMP_TLS_ctx); +#endif +#endif +} + +RTMP * +RTMP_Alloc (void) +{ + return calloc (1, sizeof (RTMP)); +} + +void +RTMP_Free (RTMP * r) +{ + free (r); +} + +void +RTMP_Init (RTMP * r) +{ +#ifdef CRYPTO + if (!RTMP_TLS_ctx) + RTMP_TLS_Init (); +#endif + + memset (r, 0, sizeof (RTMP)); + 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; +} + +void +RTMP_EnableWrite (RTMP * r) +{ + r->Link.protocol |= RTMP_FEATURE_WRITE; +} + +double +RTMP_GetDuration (RTMP * r) +{ + return r->m_fDuration; +} + +bool +RTMP_IsConnected (RTMP * r) +{ + return r->m_sb.sb_socket != -1; +} + +bool +RTMP_IsTimedout (RTMP * r) +{ + return r->m_sb.sb_timedout; +} + +void +RTMP_SetBufferMS (RTMP * r, int size) +{ + r->m_nBufferMS = size; +} + +void +RTMP_UpdateBufferMS (RTMP * r) +{ + RTMP_SendCtrl (r, 3, r->m_stream_id, r->m_nBufferMS); +} + +#undef OSS +#ifdef _WIN32 +#define OSS "WIN" +#elif defined(__sun__) +#define OSS "SOL" +#elif defined(__APPLE__) +#define OSS "MAC" +#elif defined(__linux__) +#define OSS "LNX" +#else +#define OSS "GNU" +#endif +#define DEF_VERSTR OSS " 10,0,32,18" +static const char DEFAULT_FLASH_VER[] = DEF_VERSTR; +const AVal RTMP_DefaultFlashVer = + { (char *) DEFAULT_FLASH_VER, sizeof (DEFAULT_FLASH_VER) - 1 }; + +void +RTMP_SetupStream (RTMP * r, + int protocol, + AVal * host, + unsigned int port, + AVal * sockshost, + AVal * playpath, + AVal * tcUrl, + AVal * swfUrl, + AVal * pageUrl, + AVal * app, + AVal * auth, + AVal * swfSHA256Hash, + uint32_t swfSize, + AVal * flashVer, + AVal * subscribepath, + int dStart, int dStop, bool bLiveStream, long int timeout) +{ + RTMP_Log (RTMP_LOGDEBUG, "Protocol : %s", RTMPProtocolStrings[protocol & 7]); + RTMP_Log (RTMP_LOGDEBUG, "Hostname : %.*s", host->av_len, host->av_val); + RTMP_Log (RTMP_LOGDEBUG, "Port : %d", port); + RTMP_Log (RTMP_LOGDEBUG, "Playpath : %s", playpath->av_val); + + if (tcUrl && tcUrl->av_val) + RTMP_Log (RTMP_LOGDEBUG, "tcUrl : %s", tcUrl->av_val); + if (swfUrl && swfUrl->av_val) + RTMP_Log (RTMP_LOGDEBUG, "swfUrl : %s", swfUrl->av_val); + if (pageUrl && pageUrl->av_val) + RTMP_Log (RTMP_LOGDEBUG, "pageUrl : %s", pageUrl->av_val); + if (app && app->av_val) + RTMP_Log (RTMP_LOGDEBUG, "app : %.*s", app->av_len, app->av_val); + if (auth && auth->av_val) + RTMP_Log (RTMP_LOGDEBUG, "auth : %s", auth->av_val); + if (subscribepath && subscribepath->av_val) + RTMP_Log (RTMP_LOGDEBUG, "subscribepath : %s", subscribepath->av_val); + if (flashVer && flashVer->av_val) + RTMP_Log (RTMP_LOGDEBUG, "flashVer : %s", flashVer->av_val); + if (dStart > 0) + RTMP_Log (RTMP_LOGDEBUG, "StartTime : %d msec", dStart); + if (dStop > 0) + RTMP_Log (RTMP_LOGDEBUG, "StopTime : %d msec", dStop); + + RTMP_Log (RTMP_LOGDEBUG, "live : %s", bLiveStream ? "yes" : "no"); + RTMP_Log (RTMP_LOGDEBUG, "timeout : %d sec", timeout); + +#ifdef CRYPTO + if (swfSHA256Hash != NULL && swfSize > 0) { + memcpy (r->Link.SWFHash, swfSHA256Hash->av_val, sizeof (r->Link.SWFHash)); + r->Link.SWFSize = swfSize; + RTMP_Log (RTMP_LOGDEBUG, "SWFSHA256:"); + RTMP_LogHex (RTMP_LOGDEBUG, r->Link.SWFHash, sizeof (r->Link.SWFHash)); + RTMP_Log (RTMP_LOGDEBUG, "SWFSize : %lu", r->Link.SWFSize); + } else { + r->Link.SWFSize = 0; + } +#endif + + if (sockshost->av_len) { + const char *socksport = strchr (sockshost->av_val, ':'); + char *hostname = strdup (sockshost->av_val); + + if (socksport) + hostname[socksport - sockshost->av_val] = '\0'; + r->Link.sockshost.av_val = hostname; + r->Link.sockshost.av_len = strlen (hostname); + + r->Link.socksport = socksport ? atoi (socksport + 1) : 1080; + RTMP_Log (RTMP_LOGDEBUG, "Connecting via SOCKS proxy: %s:%d", + r->Link.sockshost.av_val, r->Link.socksport); + } else { + r->Link.sockshost.av_val = NULL; + r->Link.sockshost.av_len = 0; + r->Link.socksport = 0; + } + + if (tcUrl && tcUrl->av_len) + r->Link.tcUrl = *tcUrl; + if (swfUrl && swfUrl->av_len) + r->Link.swfUrl = *swfUrl; + if (pageUrl && pageUrl->av_len) + r->Link.pageUrl = *pageUrl; + if (app && app->av_len) + r->Link.app = *app; + if (auth && auth->av_len) { + r->Link.auth = *auth; + r->Link.lFlags |= RTMP_LF_AUTH; + } + if (flashVer && flashVer->av_len) + r->Link.flashVer = *flashVer; + else + r->Link.flashVer = RTMP_DefaultFlashVer; + if (subscribepath && subscribepath->av_len) + r->Link.subscribepath = *subscribepath; + r->Link.seekTime = dStart; + r->Link.stopTime = dStop; + if (bLiveStream) + r->Link.lFlags |= RTMP_LF_LIVE; + r->Link.timeout = timeout; + + r->Link.protocol = protocol; + r->Link.hostname = *host; + r->Link.port = port; + r->Link.playpath = *playpath; + + if (r->Link.port == 0) { + if (protocol & RTMP_FEATURE_SSL) + r->Link.port = 443; + else if (protocol & RTMP_FEATURE_HTTP) + r->Link.port = 80; + else + r->Link.port = 1935; + } +} + +enum +{ OPT_STR = 0, OPT_INT, OPT_BOOL, OPT_CONN }; +static const char *optinfo[] = { + "string", "integer", "boolean", "AMF" +}; + +#define OFF(x) offsetof(struct RTMP,x) + +static struct urlopt +{ + AVal name; + off_t off; + int otype; + int omisc; + const char *use; +} options[] = { + { + AVC ("socks"), OFF (Link.sockshost), OPT_STR, 0, + "Use the specified SOCKS proxy"}, { + AVC ("app"), OFF (Link.app), OPT_STR, 0, "Name of target app on server"}, { + AVC ("tcUrl"), OFF (Link.tcUrl), OPT_STR, 0, "URL to played stream"}, { + AVC ("pageUrl"), OFF (Link.pageUrl), OPT_STR, 0, + "URL of played media's web page"}, { + AVC ("swfUrl"), OFF (Link.swfUrl), OPT_STR, 0, "URL to player SWF file"}, { + AVC ("flashver"), OFF (Link.flashVer), OPT_STR, 0, + "Flash version string (default " DEF_VERSTR ")"}, { + AVC ("conn"), OFF (Link.extras), OPT_CONN, 0, + "Append arbitrary AMF data to Connect message"}, { + AVC ("playpath"), OFF (Link.playpath), OPT_STR, 0, + "Path to target media on server"}, { + AVC ("playlist"), OFF (Link.lFlags), OPT_BOOL, RTMP_LF_PLST, + "Set playlist before play command"}, { + AVC ("live"), OFF (Link.lFlags), OPT_BOOL, RTMP_LF_LIVE, + "Stream is live, no seeking possible"}, { + AVC ("subscribe"), OFF (Link.subscribepath), OPT_STR, 0, + "Stream to subscribe to"}, { + AVC ("token"), OFF (Link.token), OPT_STR, 0, "Key for SecureToken response"}, { + AVC ("swfVfy"), OFF (Link.lFlags), OPT_BOOL, RTMP_LF_SWFV, + "Perform SWF Verification"}, { + AVC ("swfAge"), OFF (Link.swfAge), OPT_INT, 0, + "Number of days to use cached SWF hash"}, { + AVC ("start"), OFF (Link.seekTime), OPT_INT, 0, + "Stream start position in milliseconds"}, { + AVC ("stop"), OFF (Link.stopTime), OPT_INT, 0, + "Stream stop position in milliseconds"}, { + AVC ("buffer"), OFF (m_nBufferMS), OPT_INT, 0, "Buffer time in milliseconds"}, { + AVC ("timeout"), OFF (Link.timeout), OPT_INT, 0, + "Session timeout in seconds"}, { { + NULL, 0}, 0, 0} +}; + +static const AVal truth[] = { + AVC ("1"), + AVC ("on"), + AVC ("yes"), + AVC ("true"), + {0, 0} +}; + +static void +RTMP_OptUsage (void) +{ + int i; + + RTMP_Log (RTMP_LOGERROR, "Valid RTMP options are:\n"); + for (i = 0; options[i].name.av_len; i++) { + RTMP_Log (RTMP_LOGERROR, "%10s %-7s %s\n", options[i].name.av_val, + optinfo[options[i].otype], options[i].use); + } +} + +static int +parseAMF (AMFObject * obj, AVal * av, int *depth) +{ + AMFObjectProperty prop = { {0, 0} }; + int i; + char *p, *arg = av->av_val; + + if (arg[1] == ':') { + p = (char *) arg + 2; + switch (arg[0]) { + case 'B': + prop.p_type = AMF_BOOLEAN; + prop.p_vu.p_number = atoi (p); + break; + case 'S': + prop.p_type = AMF_STRING; + prop.p_vu.p_aval.av_val = p; + prop.p_vu.p_aval.av_len = av->av_len - (p - arg); + break; + case 'N': + prop.p_type = AMF_NUMBER; + prop.p_vu.p_number = strtod (p, NULL); + break; + case 'Z': + prop.p_type = AMF_NULL; + break; + case 'O': + i = atoi (p); + if (i) { + prop.p_type = AMF_OBJECT; + } else { + (*depth)--; + return 0; + } + break; + default: + return -1; + } + } else if (arg[2] == ':' && arg[0] == 'N') { + p = strchr (arg + 3, ':'); + if (!p || !*depth) + return -1; + prop.p_name.av_val = (char *) arg + 3; + prop.p_name.av_len = p - (arg + 3); + + p++; + switch (arg[1]) { + case 'B': + prop.p_type = AMF_BOOLEAN; + prop.p_vu.p_number = atoi (p); + break; + case 'S': + prop.p_type = AMF_STRING; + prop.p_vu.p_aval.av_val = p; + prop.p_vu.p_aval.av_len = av->av_len - (p - arg); + break; + case 'N': + prop.p_type = AMF_NUMBER; + prop.p_vu.p_number = strtod (p, NULL); + break; + case 'O': + prop.p_type = AMF_OBJECT; + break; + default: + return -1; + } + } else + return -1; + + if (*depth) { + AMFObject *o2; + for (i = 0; i < *depth; i++) { + o2 = &obj->o_props[obj->o_num - 1].p_vu.p_object; + obj = o2; + } + } + AMF_AddProp (obj, &prop); + if (prop.p_type == AMF_OBJECT) + (*depth)++; + return 0; +} + +bool +RTMP_SetOpt (RTMP * r, const AVal * opt, AVal * arg) +{ + int i; + void *v; + + for (i = 0; options[i].name.av_len; i++) { + if (opt->av_len != options[i].name.av_len) + continue; + if (strcasecmp (opt->av_val, options[i].name.av_val)) + continue; + v = (char *) r + options[i].off; + switch (options[i].otype) { + case OPT_STR:{ + AVal *aptr = v; + *aptr = *arg; + } + break; + case OPT_INT:{ + long l = strtol (arg->av_val, NULL, 0); + *(int *) v = l; + } + break; + case OPT_BOOL:{ + int j, fl; + fl = *(int *) v; + for (j = 0; truth[j].av_len; j++) { + if (arg->av_len != truth[j].av_len) + continue; + if (strcasecmp (arg->av_val, truth[j].av_val)) + continue; + fl |= options[i].omisc; + break; + } + *(int *) v = fl; + } + break; + case OPT_CONN: + if (parseAMF (&r->Link.extras, arg, &r->Link.edepth)) + return false; + break; + } + break; + } + if (!options[i].name.av_len) { + RTMP_Log (RTMP_LOGERROR, "Unknown option %s", opt->av_val); + RTMP_OptUsage (); + return false; + } + + return true; +} + +bool +RTMP_SetupURL (RTMP * r, char *url) +{ + AVal opt, arg; + char *p1, *p2, *ptr = strchr (url, ' '); + bool ret; + unsigned int port = 0; + + if (ptr) + *ptr = '\0'; + + ret = RTMP_ParseURL (url, &r->Link.protocol, &r->Link.hostname, + &port, &r->Link.playpath0, &r->Link.app); + if (!ret) + return ret; + r->Link.port = port; + r->Link.playpath = r->Link.playpath0; + + while (ptr) { + *ptr++ = '\0'; + p1 = ptr; + p2 = strchr (p1, '='); + if (!p2) + break; + opt.av_val = p1; + opt.av_len = p2 - p1; + *p2++ = '\0'; + arg.av_val = p2; + ptr = strchr (p2, ' '); + if (ptr) { + *ptr = '\0'; + arg.av_len = ptr - p2; + } else { + arg.av_len = strlen (p2); + } + + /* unescape */ + port = arg.av_len; + for (p1 = p2; port > 0;) { + if (*p1 == '\\') { + unsigned int c; + if (port < 3) + return false; + sscanf (p1 + 1, "%02x", &c); + *p2++ = c; + port -= 3; + p1 += 3; + } else { + *p2++ = *p1++; + port--; + } + } + arg.av_len = p2 - arg.av_val; + + ret = RTMP_SetOpt (r, &opt, &arg); + if (!ret) + return ret; + } + + if (!r->Link.tcUrl.av_len) { + r->Link.tcUrl.av_val = url; + if (r->Link.app.av_len) + r->Link.tcUrl.av_len = r->Link.app.av_len + (r->Link.app.av_val - url); + else + r->Link.tcUrl.av_len = strlen (url); + } +#ifdef CRYPTO + if ((r->Link.lFlags & RTMP_LF_SWFV) && r->Link.swfUrl.av_len) + RTMP_HashSWF (r->Link.swfUrl.av_val, &r->Link.SWFSize, + (unsigned char *) r->Link.SWFHash, r->Link.swfAge); +#endif + + if (r->Link.port == 0) { + if (r->Link.protocol & RTMP_FEATURE_SSL) + r->Link.port = 443; + else if (r->Link.protocol & RTMP_FEATURE_HTTP) + r->Link.port = 80; + else + r->Link.port = 1935; + } + return true; +} + +static bool +add_addr_info (struct sockaddr_in *service, AVal * host, int port) +{ + char *hostname; + bool ret = true; + if (host->av_val[host->av_len]) { + hostname = malloc (host->av_len + 1); + memcpy (hostname, host->av_val, host->av_len); + hostname[host->av_len] = '\0'; + } else { + hostname = host->av_val; + } + + service->sin_addr.s_addr = inet_addr (hostname); + if (service->sin_addr.s_addr == INADDR_NONE) { + struct hostent *host = gethostbyname (hostname); + if (host == NULL || host->h_addr == NULL) { + RTMP_Log (RTMP_LOGERROR, "Problem accessing the DNS. (addr: %s)", + hostname); + ret = false; + goto finish; + } + service->sin_addr = *(struct in_addr *) host->h_addr; + } + + service->sin_port = htons (port); +finish: + if (hostname != host->av_val) + free (hostname); + return ret; +} + +bool +RTMP_Connect0 (RTMP * r, struct sockaddr * service) +{ + int on = 1; + r->m_sb.sb_timedout = false; + r->m_pausing = 0; + r->m_fDuration = 0.0; + + r->m_sb.sb_socket = socket (AF_INET, SOCK_STREAM, IPPROTO_TCP); + if (r->m_sb.sb_socket != -1) { + if (connect (r->m_sb.sb_socket, service, sizeof (struct sockaddr)) < 0) { + int err = GetSockError (); + RTMP_Log (RTMP_LOGERROR, "%s, failed to connect socket. %d (%s)", + __FUNCTION__, err, strerror (err)); + RTMP_Close (r); + return false; + } + + if (r->Link.socksport) { + RTMP_Log (RTMP_LOGDEBUG, "%s ... SOCKS negotiation", __FUNCTION__); + if (!SocksNegotiate (r)) { + RTMP_Log (RTMP_LOGERROR, "%s, SOCKS negotiation failed.", __FUNCTION__); + RTMP_Close (r); + return false; + } + } + } else { + RTMP_Log (RTMP_LOGERROR, "%s, failed to create socket. Error: %d", + __FUNCTION__, GetSockError ()); + return false; + } + + /* set timeout */ + { + SET_RCVTIMEO (tv, r->Link.timeout); + if (setsockopt + (r->m_sb.sb_socket, SOL_SOCKET, SO_RCVTIMEO, (char *) &tv, sizeof (tv))) + { + RTMP_Log (RTMP_LOGERROR, "%s, Setting socket timeout to %ds failed!", + __FUNCTION__, r->Link.timeout); + } + } + + setsockopt (r->m_sb.sb_socket, IPPROTO_TCP, TCP_NODELAY, (char *) &on, + sizeof (on)); + + return true; +} + +bool +RTMP_Connect1 (RTMP * r, RTMPPacket * cp) +{ + if (r->Link.protocol & RTMP_FEATURE_SSL) { +#if defined(CRYPTO) && !defined(NO_SSL) + TLS_client (RTMP_TLS_ctx, r->m_sb.sb_ssl); + TLS_setfd (r->m_sb.sb_ssl, r->m_sb.sb_socket); + if (TLS_connect (r->m_sb.sb_ssl) < 0) { + RTMP_Log (RTMP_LOGERROR, "%s, TLS_Connect failed", __FUNCTION__); + RTMP_Close (r); + return false; + } +#else + RTMP_Log (RTMP_LOGERROR, "%s, no SSL/TLS support", __FUNCTION__); + RTMP_Close (r); + return false; + +#endif + } + if (r->Link.protocol & RTMP_FEATURE_HTTP) { + r->m_msgCounter = 1; + r->m_clientID.av_val = NULL; + r->m_clientID.av_len = 0; + HTTP_Post (r, RTMPT_OPEN, "", 1); + HTTP_read (r, 1); + r->m_msgCounter = 0; + } + RTMP_Log (RTMP_LOGDEBUG, "%s, ... connected, handshaking", __FUNCTION__); + if (!HandShake (r, true)) { + RTMP_Log (RTMP_LOGERROR, "%s, handshake failed.", __FUNCTION__); + RTMP_Close (r); + return false; + } + RTMP_Log (RTMP_LOGDEBUG, "%s, handshaked", __FUNCTION__); + + if (!SendConnectPacket (r, cp)) { + RTMP_Log (RTMP_LOGERROR, "%s, RTMP connect failed.", __FUNCTION__); + RTMP_Close (r); + return false; + } + return true; +} + +bool +RTMP_Connect (RTMP * r, RTMPPacket * cp) +{ + struct sockaddr_in service; + if (!r->Link.hostname.av_len) + return false; + + memset (&service, 0, sizeof (struct sockaddr_in)); + service.sin_family = AF_INET; + + if (r->Link.socksport) { + /* Connect via SOCKS */ + if (!add_addr_info (&service, &r->Link.sockshost, r->Link.socksport)) + return false; + } else { + /* Connect directly */ + if (!add_addr_info (&service, &r->Link.hostname, r->Link.port)) + return false; + } + + if (!RTMP_Connect0 (r, (struct sockaddr *) &service)) + return false; + + r->m_bSendCounter = true; + + return RTMP_Connect1 (r, cp); +} + +static bool +SocksNegotiate (RTMP * r) +{ + unsigned long addr; + struct sockaddr_in service; + memset (&service, 0, sizeof (struct sockaddr_in)); + + add_addr_info (&service, &r->Link.hostname, r->Link.port); + addr = htonl (service.sin_addr.s_addr); + + { + char packet[] = { + 4, 1, /* SOCKS 4, connect */ + (r->Link.port >> 8) & 0xFF, + (r->Link.port) & 0xFF, + (char) (addr >> 24) & 0xFF, (char) (addr >> 16) & 0xFF, + (char) (addr >> 8) & 0xFF, (char) addr & 0xFF, + 0 + }; /* NULL terminate */ + + WriteN (r, packet, sizeof packet); + + if (ReadN (r, packet, 8) != 8) + return false; + + if (packet[0] == 0 && packet[1] == 90) { + return true; + } else { + RTMP_Log (RTMP_LOGERROR, "%s, SOCKS returned error code %d", packet[1]); + return false; + } + } +} + +bool +RTMP_ConnectStream (RTMP * r, int seekTime) +{ + RTMPPacket packet = { 0 }; + + /* seekTime was already set by SetupStream / SetupURL. + * This is only needed by ReconnectStream. + */ + if (seekTime > 0) + r->Link.seekTime = seekTime; + + r->m_mediaChannel = 0; + + while (!r->m_bPlaying && RTMP_IsConnected (r) && RTMP_ReadPacket (r, &packet)) { + if (RTMPPacket_IsReady (&packet)) { + if (!packet.m_nBodySize) + continue; + if ((packet.m_packetType == RTMP_PACKET_TYPE_AUDIO) || + (packet.m_packetType == RTMP_PACKET_TYPE_VIDEO) || + (packet.m_packetType == RTMP_PACKET_TYPE_INFO)) { + RTMP_Log (RTMP_LOGWARNING, + "Received FLV packet before play()! Ignoring."); + RTMPPacket_Free (&packet); + continue; + } + + RTMP_ClientPacket (r, &packet); + RTMPPacket_Free (&packet); + } + } + + return r->m_bPlaying; +} + +bool +RTMP_ReconnectStream (RTMP * r, int seekTime) +{ + RTMP_DeleteStream (r); + + RTMP_SendCreateStream (r); + + return RTMP_ConnectStream (r, seekTime); +} + +bool +RTMP_ToggleStream (RTMP * r) +{ + bool res; + + if (!r->m_pausing) { + res = RTMP_SendPause (r, true, r->m_pauseStamp); + if (!res) + return res; + + r->m_pausing = 1; + sleep (1); + } + res = RTMP_SendPause (r, false, r->m_pauseStamp); + r->m_pausing = 3; + return res; +} + +void +RTMP_DeleteStream (RTMP * r) +{ + if (r->m_stream_id < 0) + return; + + r->m_bPlaying = false; + + SendDeleteStream (r, r->m_stream_id); + r->m_stream_id = -1; +} + +int +RTMP_GetNextMediaPacket (RTMP * r, RTMPPacket * packet) +{ + int bHasMediaPacket = 0; + + while (!bHasMediaPacket && RTMP_IsConnected (r) + && RTMP_ReadPacket (r, packet)) { + if (!RTMPPacket_IsReady (packet)) { + continue; + } + + bHasMediaPacket = RTMP_ClientPacket (r, packet); + + if (!bHasMediaPacket) { + RTMPPacket_Free (packet); + } else if (r->m_pausing == 3) { + if (packet->m_nTimeStamp <= r->m_mediaStamp) { + bHasMediaPacket = 0; +#ifdef _DEBUG + RTMP_Log (RTMP_LOGDEBUG, + "Skipped type: %02X, size: %d, TS: %d ms, abs TS: %d, pause: %d ms", + packet->m_packetType, packet->m_nBodySize, + packet->m_nTimeStamp, packet->m_hasAbsTimestamp, r->m_mediaStamp); +#endif + continue; + } + r->m_pausing = 0; + } + } + + if (bHasMediaPacket) + r->m_bPlaying = true; + else if (r->m_sb.sb_timedout && !r->m_pausing) + r->m_pauseStamp = r->m_channelTimestamp[r->m_mediaChannel]; + + return bHasMediaPacket; +} + +int +RTMP_ClientPacket (RTMP * r, RTMPPacket * packet) +{ + int bHasMediaPacket = 0; + switch (packet->m_packetType) { + case 0x01: + /* chunk size */ + HandleChangeChunkSize (r, packet); + break; + + case 0x03: + /* bytes read report */ + RTMP_Log (RTMP_LOGDEBUG, "%s, received: bytes read report", __FUNCTION__); + break; + + case 0x04: + /* ctrl */ + HandleCtrl (r, packet); + break; + + case 0x05: + /* server bw */ + HandleServerBW (r, packet); + break; + + case 0x06: + /* client bw */ + HandleClientBW (r, packet); + break; + + case 0x08: + /* audio data */ + /*RTMP_Log(RTMP_LOGDEBUG, "%s, received: audio %lu bytes", __FUNCTION__, packet.m_nBodySize); */ + HandleAudio (r, packet); + bHasMediaPacket = 1; + if (!r->m_mediaChannel) + r->m_mediaChannel = packet->m_nChannel; + if (!r->m_pausing) + r->m_mediaStamp = packet->m_nTimeStamp; + break; + + case 0x09: + /* video data */ + /*RTMP_Log(RTMP_LOGDEBUG, "%s, received: video %lu bytes", __FUNCTION__, packet.m_nBodySize); */ + HandleVideo (r, packet); + bHasMediaPacket = 1; + if (!r->m_mediaChannel) + r->m_mediaChannel = packet->m_nChannel; + if (!r->m_pausing) + r->m_mediaStamp = packet->m_nTimeStamp; + break; + + case 0x0F: /* flex stream send */ + RTMP_Log (RTMP_LOGDEBUG, + "%s, flex stream send, size %lu bytes, not supported, ignoring", + __FUNCTION__, packet->m_nBodySize); + break; + + case 0x10: /* flex shared object */ + RTMP_Log (RTMP_LOGDEBUG, + "%s, flex shared object, size %lu bytes, not supported, ignoring", + __FUNCTION__, packet->m_nBodySize); + break; + + case 0x11: /* flex message */ + { + RTMP_Log (RTMP_LOGDEBUG, + "%s, flex message, size %lu bytes, not fully supported", + __FUNCTION__, packet->m_nBodySize); + /*RTMP_LogHex(packet.m_body, packet.m_nBodySize); */ + + /* some DEBUG code */ +#if 0 + RTMP_LIB_AMFObject obj; + int nRes = obj.Decode (packet.m_body + 1, packet.m_nBodySize - 1); + if (nRes < 0) { + RTMP_Log (RTMP_LOGERROR, "%s, error decoding AMF3 packet", + __FUNCTION__); + /*return; */ + } + + obj.Dump (); +#endif + + if (HandleInvoke (r, packet->m_body + 1, packet->m_nBodySize - 1) == 1) + bHasMediaPacket = 2; + break; + } + case 0x12: + /* metadata (notify) */ + RTMP_Log (RTMP_LOGDEBUG, "%s, received: notify %lu bytes", __FUNCTION__, + packet->m_nBodySize); + if (HandleMetadata (r, packet->m_body, packet->m_nBodySize)) + bHasMediaPacket = 1; + break; + + case 0x13: + RTMP_Log (RTMP_LOGDEBUG, "%s, shared object, not supported, ignoring", + __FUNCTION__); + break; + + case 0x14: + /* invoke */ + RTMP_Log (RTMP_LOGDEBUG, "%s, received: invoke %lu bytes", __FUNCTION__, + packet->m_nBodySize); + /*RTMP_LogHex(packet.m_body, packet.m_nBodySize); */ + + if (HandleInvoke (r, packet->m_body, packet->m_nBodySize) == 1) + bHasMediaPacket = 2; + break; + + case 0x16: + { + /* go through FLV packets and handle metadata packets */ + unsigned int pos = 0; + uint32_t nTimeStamp = packet->m_nTimeStamp; + + while (pos + 11 < packet->m_nBodySize) { + uint32_t dataSize = AMF_DecodeInt24 (packet->m_body + pos + 1); /* size without header (11) and prevTagSize (4) */ + + if (pos + 11 + dataSize + 4 > packet->m_nBodySize) { + RTMP_Log (RTMP_LOGWARNING, "Stream corrupt?!"); + break; + } + if (packet->m_body[pos] == 0x12) { + HandleMetadata (r, packet->m_body + pos + 11, dataSize); + } else if (packet->m_body[pos] == 8 || packet->m_body[pos] == 9) { + nTimeStamp = AMF_DecodeInt24 (packet->m_body + pos + 4); + nTimeStamp |= (packet->m_body[pos + 7] << 24); + } + pos += (11 + dataSize + 4); + } + if (!r->m_pausing) + r->m_mediaStamp = nTimeStamp; + + /* FLV tag(s) */ + /*RTMP_Log(RTMP_LOGDEBUG, "%s, received: FLV tag(s) %lu bytes", __FUNCTION__, packet.m_nBodySize); */ + bHasMediaPacket = 1; + break; + } + default: + RTMP_Log (RTMP_LOGDEBUG, "%s, unknown packet type received: 0x%02x", + __FUNCTION__, packet->m_packetType); +#ifdef _DEBUG + RTMP_LogHex (RTMP_LOGDEBUG, packet->m_body, packet->m_nBodySize); +#endif + } + + return bHasMediaPacket; +} + +#ifdef _DEBUG +extern FILE *netstackdump; +extern FILE *netstackdump_read; +#endif + +static int +ReadN (RTMP * r, char *buffer, int n) +{ + int nOriginalSize = n; + int avail; + char *ptr; + + r->m_sb.sb_timedout = false; + +#ifdef _DEBUG + memset (buffer, 0, n); +#endif + + ptr = buffer; + while (n > 0) { + int nBytes = 0, nRead; + if (r->Link.protocol & RTMP_FEATURE_HTTP) { + while (!r->m_resplen) { + if (r->m_sb.sb_size < 144) { + if (!r->m_unackd) + HTTP_Post (r, RTMPT_IDLE, "", 1); + if (RTMPSockBuf_Fill (&r->m_sb) < 1) { + if (!r->m_sb.sb_timedout) + RTMP_Close (r); + return 0; + } + } + HTTP_read (r, 0); + } + if (r->m_resplen && !r->m_sb.sb_size) + RTMPSockBuf_Fill (&r->m_sb); + avail = r->m_sb.sb_size; + if (avail > r->m_resplen) + avail = r->m_resplen; + } else { + avail = r->m_sb.sb_size; + if (avail == 0) { + if (RTMPSockBuf_Fill (&r->m_sb) < 1) { + if (!r->m_sb.sb_timedout) + RTMP_Close (r); + return 0; + } + avail = r->m_sb.sb_size; + } + } + nRead = ((n < avail) ? n : avail); + if (nRead > 0) { + memcpy (ptr, r->m_sb.sb_start, nRead); + r->m_sb.sb_start += nRead; + r->m_sb.sb_size -= nRead; + nBytes = nRead; + r->m_nBytesIn += nRead; + if (r->m_bSendCounter + && r->m_nBytesIn > r->m_nBytesInSent + r->m_nClientBW / 2) + SendBytesReceived (r); + } + /*RTMP_Log(RTMP_LOGDEBUG, "%s: %d bytes\n", __FUNCTION__, nBytes); */ +#ifdef _DEBUG + fwrite (ptr, 1, nBytes, netstackdump_read); +#endif + + if (nBytes == 0) { + RTMP_Log (RTMP_LOGDEBUG, "%s, RTMP socket closed by peer", __FUNCTION__); + /*goto again; */ + RTMP_Close (r); + break; + } + + if (r->Link.protocol & RTMP_FEATURE_HTTP) + r->m_resplen -= nBytes; + +#ifdef CRYPTO + if (r->Link.rc4keyIn) { + RC4_encrypt (r->Link.rc4keyIn, nBytes, ptr); + } +#endif + + n -= nBytes; + ptr += nBytes; + } + + return nOriginalSize - n; +} + +static bool +WriteN (RTMP * r, const char *buffer, int n) +{ + const char *ptr = buffer; +#ifdef CRYPTO + char *encrypted = 0; + char buf[RTMP_BUFFER_CACHE_SIZE]; + + if (r->Link.rc4keyOut) { + if (n > sizeof (buf)) + encrypted = (char *) malloc (n); + else + encrypted = (char *) buf; + ptr = encrypted; + RC4_encrypt2 (r->Link.rc4keyOut, n, buffer, ptr); + } +#endif + + while (n > 0) { + int nBytes; + + if (r->Link.protocol & RTMP_FEATURE_HTTP) + nBytes = HTTP_Post (r, RTMPT_SEND, ptr, n); + else + nBytes = RTMPSockBuf_Send (&r->m_sb, ptr, n); + /*RTMP_Log(RTMP_LOGDEBUG, "%s: %d\n", __FUNCTION__, nBytes); */ + + if (nBytes < 0) { + int sockerr = GetSockError (); + RTMP_Log (RTMP_LOGERROR, "%s, RTMP send error %d (%d bytes)", + __FUNCTION__, sockerr, n); + + if (sockerr == EINTR && !RTMP_ctrlC) + continue; + + RTMP_Close (r); + n = 1; + break; + } + + if (nBytes == 0) + break; + + n -= nBytes; + ptr += nBytes; + } + +#ifdef CRYPTO + if (encrypted && encrypted != buf) + free (encrypted); +#endif + + return n == 0; +} + +#define SAVC(x) static const AVal av_##x = AVC(#x) + +SAVC (app); +SAVC (connect); +SAVC (flashVer); +SAVC (swfUrl); +SAVC (pageUrl); +SAVC (tcUrl); +SAVC (fpad); +SAVC (capabilities); +SAVC (audioCodecs); +SAVC (videoCodecs); +SAVC (videoFunction); +SAVC (objectEncoding); +SAVC (secureToken); +SAVC (secureTokenResponse); +SAVC (type); +SAVC (nonprivate); + +static bool +SendConnectPacket (RTMP * r, RTMPPacket * cp) +{ + RTMPPacket packet; + char pbuf[4096], *pend = pbuf + sizeof (pbuf); + char *enc; + + if (cp) + return RTMP_SendPacket (r, cp, true); + + packet.m_nChannel = 0x03; /* control channel (invoke) */ + packet.m_headerType = RTMP_PACKET_SIZE_LARGE; + packet.m_packetType = 0x14; /* INVOKE */ + packet.m_nTimeStamp = 0; + packet.m_nInfoField2 = 0; + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + enc = packet.m_body; + enc = AMF_EncodeString (enc, pend, &av_connect); + enc = AMF_EncodeNumber (enc, pend, ++r->m_numInvokes); + *enc++ = AMF_OBJECT; + + enc = AMF_EncodeNamedString (enc, pend, &av_app, &r->Link.app); + if (!enc) + return false; + if (r->Link.protocol & RTMP_FEATURE_WRITE) { + enc = AMF_EncodeNamedString (enc, pend, &av_type, &av_nonprivate); + if (!enc) + return false; + } + if (r->Link.flashVer.av_len) { + enc = AMF_EncodeNamedString (enc, pend, &av_flashVer, &r->Link.flashVer); + if (!enc) + return false; + } + if (r->Link.swfUrl.av_len) { + enc = AMF_EncodeNamedString (enc, pend, &av_swfUrl, &r->Link.swfUrl); + if (!enc) + return false; + } + if (r->Link.tcUrl.av_len) { + enc = AMF_EncodeNamedString (enc, pend, &av_tcUrl, &r->Link.tcUrl); + if (!enc) + return false; + } + if (!(r->Link.protocol & RTMP_FEATURE_WRITE)) { + enc = AMF_EncodeNamedBoolean (enc, pend, &av_fpad, false); + if (!enc) + return false; + enc = AMF_EncodeNamedNumber (enc, pend, &av_capabilities, 15.0); + if (!enc) + return false; + enc = AMF_EncodeNamedNumber (enc, pend, &av_audioCodecs, r->m_fAudioCodecs); + if (!enc) + return false; + enc = AMF_EncodeNamedNumber (enc, pend, &av_videoCodecs, r->m_fVideoCodecs); + if (!enc) + return false; + enc = AMF_EncodeNamedNumber (enc, pend, &av_videoFunction, 1.0); + if (!enc) + return false; + if (r->Link.pageUrl.av_len) { + enc = AMF_EncodeNamedString (enc, pend, &av_pageUrl, &r->Link.pageUrl); + if (!enc) + return false; + } + } + if (r->m_fEncoding != 0.0 || r->m_bSendEncoding) { /* AMF0, AMF3 not fully supported yet */ + enc = AMF_EncodeNamedNumber (enc, pend, &av_objectEncoding, r->m_fEncoding); + if (!enc) + return false; + } + if (enc + 3 >= pend) + return false; + *enc++ = 0; + *enc++ = 0; /* end of object - 0x00 0x00 0x09 */ + *enc++ = AMF_OBJECT_END; + + /* add auth string */ + if (r->Link.auth.av_len) { + enc = AMF_EncodeBoolean (enc, pend, r->Link.lFlags & RTMP_LF_AUTH); + if (!enc) + return false; + enc = AMF_EncodeString (enc, pend, &r->Link.auth); + if (!enc) + return false; + } + if (r->Link.extras.o_num) { + int i; + for (i = 0; i < r->Link.extras.o_num; i++) { + enc = AMFProp_Encode (&r->Link.extras.o_props[i], enc, pend); + if (!enc) + return false; + } + } + packet.m_nBodySize = enc - packet.m_body; + + return RTMP_SendPacket (r, &packet, true); +} + +#if 0 /* unused */ +SAVC (bgHasStream); + +static bool +SendBGHasStream (RTMP * r, double dId, AVal * playpath) +{ + RTMPPacket packet; + char pbuf[1024], *pend = pbuf + sizeof (pbuf); + char *enc; + + packet.m_nChannel = 0x03; /* control channel (invoke) */ + packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM; + packet.m_packetType = 0x14; /* INVOKE */ + packet.m_nTimeStamp = 0; + packet.m_nInfoField2 = 0; + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + enc = packet.m_body; + enc = AMF_EncodeString (enc, pend, &av_bgHasStream); + enc = AMF_EncodeNumber (enc, pend, dId); + *enc++ = AMF_NULL; + + enc = AMF_EncodeString (enc, pend, playpath); + if (enc == NULL) + return false; + + packet.m_nBodySize = enc - packet.m_body; + + return RTMP_SendPacket (r, &packet, true); +} +#endif + +SAVC (createStream); + +bool +RTMP_SendCreateStream (RTMP * r) +{ + RTMPPacket packet; + char pbuf[256], *pend = pbuf + sizeof (pbuf); + char *enc; + + packet.m_nChannel = 0x03; /* control channel (invoke) */ + packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM; + packet.m_packetType = 0x14; /* INVOKE */ + packet.m_nTimeStamp = 0; + packet.m_nInfoField2 = 0; + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + enc = packet.m_body; + enc = AMF_EncodeString (enc, pend, &av_createStream); + enc = AMF_EncodeNumber (enc, pend, ++r->m_numInvokes); + *enc++ = AMF_NULL; /* NULL */ + + packet.m_nBodySize = enc - packet.m_body; + + return RTMP_SendPacket (r, &packet, true); +} + +SAVC (FCSubscribe); + +static bool +SendFCSubscribe (RTMP * r, AVal * subscribepath) +{ + RTMPPacket packet; + char pbuf[512], *pend = pbuf + sizeof (pbuf); + char *enc; + packet.m_nChannel = 0x03; /* control channel (invoke) */ + packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM; + packet.m_packetType = 0x14; /* INVOKE */ + packet.m_nTimeStamp = 0; + packet.m_nInfoField2 = 0; + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + RTMP_Log (RTMP_LOGDEBUG, "FCSubscribe: %s", subscribepath->av_val); + enc = packet.m_body; + enc = AMF_EncodeString (enc, pend, &av_FCSubscribe); + enc = AMF_EncodeNumber (enc, pend, ++r->m_numInvokes); + *enc++ = AMF_NULL; + enc = AMF_EncodeString (enc, pend, subscribepath); + + if (!enc) + return false; + + packet.m_nBodySize = enc - packet.m_body; + + return RTMP_SendPacket (r, &packet, true); +} + +SAVC (releaseStream); + +static bool +SendReleaseStream (RTMP * r) +{ + RTMPPacket packet; + char pbuf[1024], *pend = pbuf + sizeof (pbuf); + char *enc; + + packet.m_nChannel = 0x03; /* control channel (invoke) */ + packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM; + packet.m_packetType = 0x14; /* INVOKE */ + packet.m_nTimeStamp = 0; + packet.m_nInfoField2 = 0; + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + enc = packet.m_body; + enc = AMF_EncodeString (enc, pend, &av_releaseStream); + enc = AMF_EncodeNumber (enc, pend, ++r->m_numInvokes); + *enc++ = AMF_NULL; + enc = AMF_EncodeString (enc, pend, &r->Link.playpath); + if (!enc) + return false; + + packet.m_nBodySize = enc - packet.m_body; + + return RTMP_SendPacket (r, &packet, false); +} + +SAVC (FCPublish); + +static bool +SendFCPublish (RTMP * r) +{ + RTMPPacket packet; + char pbuf[1024], *pend = pbuf + sizeof (pbuf); + char *enc; + + packet.m_nChannel = 0x03; /* control channel (invoke) */ + packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM; + packet.m_packetType = 0x14; /* INVOKE */ + packet.m_nTimeStamp = 0; + packet.m_nInfoField2 = 0; + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + enc = packet.m_body; + enc = AMF_EncodeString (enc, pend, &av_FCPublish); + enc = AMF_EncodeNumber (enc, pend, ++r->m_numInvokes); + *enc++ = AMF_NULL; + enc = AMF_EncodeString (enc, pend, &r->Link.playpath); + if (!enc) + return false; + + packet.m_nBodySize = enc - packet.m_body; + + return RTMP_SendPacket (r, &packet, false); +} + +SAVC (FCUnpublish); + +static bool +SendFCUnpublish (RTMP * r) +{ + RTMPPacket packet; + char pbuf[1024], *pend = pbuf + sizeof (pbuf); + char *enc; + + packet.m_nChannel = 0x03; /* control channel (invoke) */ + packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM; + packet.m_packetType = 0x14; /* INVOKE */ + packet.m_nTimeStamp = 0; + packet.m_nInfoField2 = 0; + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + enc = packet.m_body; + enc = AMF_EncodeString (enc, pend, &av_FCUnpublish); + enc = AMF_EncodeNumber (enc, pend, ++r->m_numInvokes); + *enc++ = AMF_NULL; + enc = AMF_EncodeString (enc, pend, &r->Link.playpath); + if (!enc) + return false; + + packet.m_nBodySize = enc - packet.m_body; + + return RTMP_SendPacket (r, &packet, false); +} + +SAVC (publish); +SAVC (live); +SAVC (record); + +static bool +SendPublish (RTMP * r) +{ + RTMPPacket packet; + char pbuf[1024], *pend = pbuf + sizeof (pbuf); + char *enc; + + packet.m_nChannel = 0x04; /* source channel (invoke) */ + packet.m_headerType = RTMP_PACKET_SIZE_LARGE; + packet.m_packetType = 0x14; /* INVOKE */ + packet.m_nTimeStamp = 0; + packet.m_nInfoField2 = r->m_stream_id; + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + enc = packet.m_body; + enc = AMF_EncodeString (enc, pend, &av_publish); + enc = AMF_EncodeNumber (enc, pend, ++r->m_numInvokes); + *enc++ = AMF_NULL; + enc = AMF_EncodeString (enc, pend, &r->Link.playpath); + if (!enc) + return false; + + /* FIXME: should we choose live based on Link.lFlags & RTMP_LF_LIVE? */ + enc = AMF_EncodeString (enc, pend, &av_live); + if (!enc) + return false; + + packet.m_nBodySize = enc - packet.m_body; + + return RTMP_SendPacket (r, &packet, true); +} + +SAVC (deleteStream); + +static bool +SendDeleteStream (RTMP * r, double dStreamId) +{ + RTMPPacket packet; + char pbuf[256], *pend = pbuf + sizeof (pbuf); + char *enc; + + packet.m_nChannel = 0x03; /* control channel (invoke) */ + packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM; + packet.m_packetType = 0x14; /* INVOKE */ + packet.m_nTimeStamp = 0; + packet.m_nInfoField2 = 0; + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + enc = packet.m_body; + enc = AMF_EncodeString (enc, pend, &av_deleteStream); + enc = AMF_EncodeNumber (enc, pend, ++r->m_numInvokes); + *enc++ = AMF_NULL; + enc = AMF_EncodeNumber (enc, pend, dStreamId); + + packet.m_nBodySize = enc - packet.m_body; + + /* no response expected */ + return RTMP_SendPacket (r, &packet, false); +} + +SAVC (pause); + +bool +RTMP_SendPause (RTMP * r, bool DoPause, int iTime) +{ + RTMPPacket packet; + char pbuf[256], *pend = pbuf + sizeof (pbuf); + char *enc; + + packet.m_nChannel = 0x08; /* video channel */ + packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM; + packet.m_packetType = 0x14; /* invoke */ + packet.m_nTimeStamp = 0; + packet.m_nInfoField2 = 0; + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + enc = packet.m_body; + enc = AMF_EncodeString (enc, pend, &av_pause); + enc = AMF_EncodeNumber (enc, pend, ++r->m_numInvokes); + *enc++ = AMF_NULL; + enc = AMF_EncodeBoolean (enc, pend, DoPause); + enc = AMF_EncodeNumber (enc, pend, (double) iTime); + + packet.m_nBodySize = enc - packet.m_body; + + RTMP_Log (RTMP_LOGDEBUG, "%s, %d, pauseTime=%d", __FUNCTION__, DoPause, + iTime); + return RTMP_SendPacket (r, &packet, true); +} + +SAVC (seek); + +bool +RTMP_SendSeek (RTMP * r, int iTime) +{ + RTMPPacket packet; + char pbuf[256], *pend = pbuf + sizeof (pbuf); + char *enc; + + packet.m_nChannel = 0x08; /* video channel */ + packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM; + packet.m_packetType = 0x14; /* invoke */ + packet.m_nTimeStamp = 0; + packet.m_nInfoField2 = 0; + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + enc = packet.m_body; + enc = AMF_EncodeString (enc, pend, &av_seek); + enc = AMF_EncodeNumber (enc, pend, ++r->m_numInvokes); + *enc++ = AMF_NULL; + enc = AMF_EncodeNumber (enc, pend, (double) iTime); + + packet.m_nBodySize = enc - packet.m_body; + + r->m_read.flags |= RTMP_READ_SEEKING; + r->m_read.nResumeTS = 0; + + return RTMP_SendPacket (r, &packet, true); +} + +bool +RTMP_SendServerBW (RTMP * r) +{ + RTMPPacket packet; + char pbuf[256], *pend = pbuf + sizeof (pbuf); + + packet.m_nChannel = 0x02; /* control channel (invoke) */ + packet.m_headerType = RTMP_PACKET_SIZE_LARGE; + packet.m_packetType = 0x05; /* Server BW */ + packet.m_nTimeStamp = 0; + packet.m_nInfoField2 = 0; + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + packet.m_nBodySize = 4; + + AMF_EncodeInt32 (packet.m_body, pend, r->m_nServerBW); + return RTMP_SendPacket (r, &packet, false); +} + +bool +RTMP_SendClientBW (RTMP * r) +{ + RTMPPacket packet; + char pbuf[256], *pend = pbuf + sizeof (pbuf); + + packet.m_nChannel = 0x02; /* control channel (invoke) */ + packet.m_headerType = RTMP_PACKET_SIZE_LARGE; + packet.m_packetType = 0x06; /* Client BW */ + packet.m_nTimeStamp = 0; + packet.m_nInfoField2 = 0; + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + packet.m_nBodySize = 5; + + AMF_EncodeInt32 (packet.m_body, pend, r->m_nClientBW); + packet.m_body[4] = r->m_nClientBW2; + return RTMP_SendPacket (r, &packet, false); +} + +static bool +SendBytesReceived (RTMP * r) +{ + RTMPPacket packet; + char pbuf[256], *pend = pbuf + sizeof (pbuf); + + packet.m_nChannel = 0x02; /* control channel (invoke) */ + packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM; + packet.m_packetType = 0x03; /* bytes in */ + packet.m_nTimeStamp = 0; + packet.m_nInfoField2 = 0; + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + packet.m_nBodySize = 4; + + AMF_EncodeInt32 (packet.m_body, pend, r->m_nBytesIn); /* hard coded for now */ + r->m_nBytesInSent = r->m_nBytesIn; + + /*RTMP_Log(RTMP_LOGDEBUG, "Send bytes report. 0x%x (%d bytes)", (unsigned int)m_nBytesIn, m_nBytesIn); */ + return RTMP_SendPacket (r, &packet, false); +} + +SAVC (_checkbw); + +static bool +SendCheckBW (RTMP * r) +{ + RTMPPacket packet; + char pbuf[256], *pend = pbuf + sizeof (pbuf); + char *enc; + + packet.m_nChannel = 0x03; /* control channel (invoke) */ + packet.m_headerType = RTMP_PACKET_SIZE_LARGE; + packet.m_packetType = 0x14; /* INVOKE */ + packet.m_nTimeStamp = 0; /* RTMP_GetTime(); */ + packet.m_nInfoField2 = 0; + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + enc = packet.m_body; + enc = AMF_EncodeString (enc, pend, &av__checkbw); + enc = AMF_EncodeNumber (enc, pend, ++r->m_numInvokes); + *enc++ = AMF_NULL; + + packet.m_nBodySize = enc - packet.m_body; + + /* triggers _onbwcheck and eventually results in _onbwdone */ + return RTMP_SendPacket (r, &packet, false); +} + +SAVC (_result); + +static bool +SendCheckBWResult (RTMP * r, double txn) +{ + RTMPPacket packet; + char pbuf[256], *pend = pbuf + sizeof (pbuf); + char *enc; + + packet.m_nChannel = 0x03; /* control channel (invoke) */ + packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM; + packet.m_packetType = 0x14; /* INVOKE */ + packet.m_nTimeStamp = 0x16 * r->m_nBWCheckCounter; /* temp inc value. till we figure it out. */ + packet.m_nInfoField2 = 0; + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + enc = packet.m_body; + enc = AMF_EncodeString (enc, pend, &av__result); + enc = AMF_EncodeNumber (enc, pend, txn); + *enc++ = AMF_NULL; + enc = AMF_EncodeNumber (enc, pend, (double) r->m_nBWCheckCounter++); + + packet.m_nBodySize = enc - packet.m_body; + + return RTMP_SendPacket (r, &packet, false); +} + +SAVC (play); + +static bool +SendPlay (RTMP * r) +{ + RTMPPacket packet; + char pbuf[1024], *pend = pbuf + sizeof (pbuf); + char *enc; + + packet.m_nChannel = 0x08; /* we make 8 our stream channel */ + packet.m_headerType = RTMP_PACKET_SIZE_LARGE; + packet.m_packetType = 0x14; /* INVOKE */ + packet.m_nTimeStamp = 0; + packet.m_nInfoField2 = r->m_stream_id; /*0x01000000; */ + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + enc = packet.m_body; + enc = AMF_EncodeString (enc, pend, &av_play); + enc = AMF_EncodeNumber (enc, pend, ++r->m_numInvokes); + *enc++ = AMF_NULL; + + RTMP_Log (RTMP_LOGDEBUG, "%s, seekTime=%d, stopTime=%d, sending play: %s", + __FUNCTION__, r->Link.seekTime, r->Link.stopTime, + r->Link.playpath.av_val); + enc = AMF_EncodeString (enc, pend, &r->Link.playpath); + if (!enc) + return false; + + /* Optional parameters start and len. + * + * start: -2, -1, 0, positive number + * -2: looks for a live stream, then a recorded stream, + * if not found any open a live stream + * -1: plays a live stream + * >=0: plays a recorded streams from 'start' milliseconds + */ + if (r->Link.lFlags & RTMP_LF_LIVE) + enc = AMF_EncodeNumber (enc, pend, -1000.0); + else { + if (r->Link.seekTime > 0.0) + enc = AMF_EncodeNumber (enc, pend, r->Link.seekTime); /* resume from here */ + else + enc = AMF_EncodeNumber (enc, pend, 0.0); /*-2000.0);*//* recorded as default, -2000.0 is not reliable since that freezes the player if the stream is not found */ + } + if (!enc) + return false; + + /* len: -1, 0, positive number + * -1: plays live or recorded stream to the end (default) + * 0: plays a frame 'start' ms away from the beginning + * >0: plays a live or recoded stream for 'len' milliseconds + */ + /*enc += EncodeNumber(enc, -1.0); *//* len */ + if (r->Link.stopTime) { + enc = AMF_EncodeNumber (enc, pend, r->Link.stopTime - r->Link.seekTime); + if (!enc) + return false; + } + + packet.m_nBodySize = enc - packet.m_body; + + return RTMP_SendPacket (r, &packet, true); +} + +SAVC (set_playlist); +SAVC (0); + +static bool +SendPlaylist (RTMP * r) +{ + RTMPPacket packet; + char pbuf[1024], *pend = pbuf + sizeof (pbuf); + char *enc; + + packet.m_nChannel = 0x08; /* we make 8 our stream channel */ + packet.m_headerType = RTMP_PACKET_SIZE_LARGE; + packet.m_packetType = 0x14; /* INVOKE */ + packet.m_nTimeStamp = 0; + packet.m_nInfoField2 = r->m_stream_id; /*0x01000000; */ + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + enc = packet.m_body; + enc = AMF_EncodeString (enc, pend, &av_set_playlist); + enc = AMF_EncodeNumber (enc, pend, 0); + *enc++ = AMF_NULL; + *enc++ = AMF_ECMA_ARRAY; + *enc++ = 0; + *enc++ = 0; + *enc++ = 0; + *enc++ = AMF_OBJECT; + enc = AMF_EncodeNamedString (enc, pend, &av_0, &r->Link.playpath); + if (!enc) + return false; + if (enc + 3 >= pend) + return false; + *enc++ = 0; + *enc++ = 0; + *enc++ = AMF_OBJECT_END; + + packet.m_nBodySize = enc - packet.m_body; + + return RTMP_SendPacket (r, &packet, true); +} + +static bool +SendSecureTokenResponse (RTMP * r, AVal * resp) +{ + RTMPPacket packet; + char pbuf[1024], *pend = pbuf + sizeof (pbuf); + char *enc; + + packet.m_nChannel = 0x03; /* control channel (invoke) */ + packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM; + packet.m_packetType = 0x14; + packet.m_nTimeStamp = 0; + packet.m_nInfoField2 = 0; + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + enc = packet.m_body; + enc = AMF_EncodeString (enc, pend, &av_secureTokenResponse); + enc = AMF_EncodeNumber (enc, pend, 0.0); + *enc++ = AMF_NULL; + enc = AMF_EncodeString (enc, pend, resp); + if (!enc) + return false; + + packet.m_nBodySize = enc - packet.m_body; + + return RTMP_SendPacket (r, &packet, false); +} + +/* +from http://jira.red5.org/confluence/display/docs/Ping: + +Ping is the most mysterious message in RTMP and till now we haven't fully interpreted it yet. In summary, Ping message is used as a special command that are exchanged between client and server. This page aims to document all known Ping messages. Expect the list to grow. + +The type of Ping packet is 0x4 and contains two mandatory parameters and two optional parameters. The first parameter is the type of Ping and in short integer. The second parameter is the target of the ping. As Ping is always sent in Channel 2 (control channel) and the target object in RTMP header is always 0 which means the Connection object, it's necessary to put an extra parameter to indicate the exact target object the Ping is sent to. The second parameter takes this responsibility. The value has the same meaning as the target object field in RTMP header. (The second value could also be used as other purposes, like RTT Ping/Pong. It is used as the timestamp.) The third and fourth parameters are optional and could be looked upon as the parameter of the Ping packet. Below is an unexhausted list of Ping messages. + + * type 0: Clear the stream. No third and fourth parameters. The second parameter could be 0. After the connection is established, a Ping 0,0 will be sent from server to client. The message will also be sent to client on the start of Play and in response of a Seek or Pause/Resume request. This Ping tells client to re-calibrate the clock with the timestamp of the next packet server sends. + * type 1: Tell the stream to clear the playing buffer. + * type 3: Buffer time of the client. The third parameter is the buffer time in millisecond. + * type 4: Reset a stream. Used together with type 0 in the case of VOD. Often sent before type 0. + * type 6: Ping the client from server. The second parameter is the current time. + * type 7: Pong reply from client. The second parameter is the time the server sent with his ping request. + * type 26: SWFVerification request + * type 27: SWFVerification response +*/ +bool +RTMP_SendCtrl (RTMP * r, short nType, unsigned int nObject, unsigned int nTime) +{ + RTMPPacket packet; + char pbuf[256], *pend = pbuf + sizeof (pbuf); + int nSize; + char *buf; + + RTMP_Log (RTMP_LOGDEBUG, "sending ctrl. type: 0x%04x", + (unsigned short) nType); + + packet.m_nChannel = 0x02; /* control channel (ping) */ + packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM; + packet.m_packetType = 0x04; /* ctrl */ + packet.m_nTimeStamp = 0; /* RTMP_GetTime(); */ + packet.m_nInfoField2 = 0; + packet.m_hasAbsTimestamp = 0; + packet.m_body = pbuf + RTMP_MAX_HEADER_SIZE; + + switch (nType) { + case 0x03: + nSize = 10; + break; /* buffer time */ + case 0x1A: + nSize = 3; + break; /* SWF verify request */ + case 0x1B: + nSize = 44; + break; /* SWF verify response */ + default: + nSize = 6; + break; + } + + packet.m_nBodySize = nSize; + + buf = packet.m_body; + buf = AMF_EncodeInt16 (buf, pend, nType); + + if (nType == 0x1B) { +#ifdef CRYPTO + memcpy (buf, r->Link.SWFVerificationResponse, 42); + RTMP_Log (RTMP_LOGDEBUG, "Sending SWFVerification response: "); + RTMP_LogHex (RTMP_LOGDEBUG, (uint8_t *) packet.m_body, packet.m_nBodySize); +#endif + } else if (nType == 0x1A) { + *buf = nObject & 0xff; + } else { + if (nSize > 2) + buf = AMF_EncodeInt32 (buf, pend, nObject); + + if (nSize > 6) + buf = AMF_EncodeInt32 (buf, pend, nTime); + } + + return RTMP_SendPacket (r, &packet, false); +} + +static void +AV_erase (RTMP_METHOD * vals, int *num, int i, bool freeit) +{ + if (freeit) + free (vals[i].name.av_val); + (*num)--; + for (; i < *num; i++) { + vals[i] = vals[i + 1]; + } + vals[i].name.av_val = NULL; + vals[i].name.av_len = 0; + vals[i].num = 0; +} + +void +RTMP_DropRequest (RTMP * r, int i, bool freeit) +{ + AV_erase (r->m_methodCalls, &r->m_numCalls, i, freeit); +} + +static void +AV_queue (RTMP_METHOD ** vals, int *num, AVal * av, int txn) +{ + char *tmp; + if (!(*num & 0x0f)) + *vals = realloc (*vals, (*num + 16) * sizeof (RTMP_METHOD)); + tmp = malloc (av->av_len + 1); + memcpy (tmp, av->av_val, av->av_len); + tmp[av->av_len] = '\0'; + (*vals)[*num].num = txn; + (*vals)[*num].name.av_len = av->av_len; + (*vals)[(*num)++].name.av_val = tmp; +} + +static void +AV_clear (RTMP_METHOD * vals, int num) +{ + int i; + for (i = 0; i < num; i++) + free (vals[i].name.av_val); + free (vals); +} + +SAVC (onBWDone); +SAVC (onFCSubscribe); +SAVC (onFCUnsubscribe); +SAVC (_onbwcheck); +SAVC (_onbwdone); +SAVC (_error); +SAVC (close); +SAVC (code); +SAVC (level); +SAVC (onStatus); +SAVC (playlist_ready); +static const AVal av_NetStream_Failed = AVC ("NetStream.Failed"); +static const AVal av_NetStream_Play_Failed = AVC ("NetStream.Play.Failed"); +static const AVal av_NetStream_Play_StreamNotFound = +AVC ("NetStream.Play.StreamNotFound"); +static const AVal av_NetConnection_Connect_InvalidApp = +AVC ("NetConnection.Connect.InvalidApp"); +static const AVal av_NetStream_Play_Start = AVC ("NetStream.Play.Start"); +static const AVal av_NetStream_Play_Complete = AVC ("NetStream.Play.Complete"); +static const AVal av_NetStream_Play_Stop = AVC ("NetStream.Play.Stop"); +static const AVal av_NetStream_Seek_Notify = AVC ("NetStream.Seek.Notify"); +static const AVal av_NetStream_Pause_Notify = AVC ("NetStream.Pause.Notify"); +static const AVal av_NetStream_Play_UnpublishNotify = +AVC ("NetStream.Play.UnpublishNotify"); +static const AVal av_NetStream_Publish_Start = AVC ("NetStream.Publish.Start"); + +/* Returns 0 for OK/Failed/error, 1 for 'Stop or Complete' */ +static int +HandleInvoke (RTMP * r, const char *body, unsigned int nBodySize) +{ + AMFObject obj; + AVal method; + int txn; + int ret = 0, nRes; + if (body[0] != 0x02) { /* make sure it is a string method name we start with */ + RTMP_Log (RTMP_LOGWARNING, + "%s, Sanity failed. no string method in invoke packet", __FUNCTION__); + return 0; + } + + nRes = AMF_Decode (&obj, body, nBodySize, false); + if (nRes < 0) { + RTMP_Log (RTMP_LOGERROR, "%s, error decoding invoke packet", __FUNCTION__); + return 0; + } + + AMF_Dump (&obj); + AMFProp_GetString (AMF_GetProp (&obj, NULL, 0), &method); + txn = (int) AMFProp_GetNumber (AMF_GetProp (&obj, NULL, 1)); + RTMP_Log (RTMP_LOGDEBUG, "%s, server invoking <%s>", __FUNCTION__, + method.av_val); + + if (AVMATCH (&method, &av__result)) { + AVal methodInvoked = { 0 }; + int i; + + for (i = 0; i < r->m_numCalls; i++) { + if (r->m_methodCalls[i].num == txn) { + methodInvoked = r->m_methodCalls[i].name; + AV_erase (r->m_methodCalls, &r->m_numCalls, i, false); + break; + } + } + if (!methodInvoked.av_val) { + RTMP_Log (RTMP_LOGDEBUG, + "%s, received result id %d without matching request", __FUNCTION__, + txn); + goto leave; + } + + RTMP_Log (RTMP_LOGDEBUG, "%s, received result for method call <%s>", + __FUNCTION__, methodInvoked.av_val); + + if (AVMATCH (&methodInvoked, &av_connect)) { + if (r->Link.token.av_len) { + AMFObjectProperty p; + if (RTMP_FindFirstMatchingProperty (&obj, &av_secureToken, &p)) { + DecodeTEA (&r->Link.token, &p.p_vu.p_aval); + SendSecureTokenResponse (r, &p.p_vu.p_aval); + } + } + if (r->Link.protocol & RTMP_FEATURE_WRITE) { + SendReleaseStream (r); + SendFCPublish (r); + } else { + RTMP_SendServerBW (r); + RTMP_SendCtrl (r, 3, 0, 300); + } + RTMP_SendCreateStream (r); + + if (!(r->Link.protocol & RTMP_FEATURE_WRITE)) { + /* Send the FCSubscribe if live stream or if subscribepath is set */ + if (r->Link.subscribepath.av_len) + SendFCSubscribe (r, &r->Link.subscribepath); + else if (r->Link.lFlags & RTMP_LF_LIVE) + SendFCSubscribe (r, &r->Link.playpath); + } + } else if (AVMATCH (&methodInvoked, &av_createStream)) { + r->m_stream_id = (int) AMFProp_GetNumber (AMF_GetProp (&obj, NULL, 3)); + + if (r->Link.protocol & RTMP_FEATURE_WRITE) { + SendPublish (r); + } else { + if (r->Link.lFlags & RTMP_LF_PLST) + SendPlaylist (r); + SendPlay (r); + RTMP_SendCtrl (r, 3, r->m_stream_id, r->m_nBufferMS); + } + } else if (AVMATCH (&methodInvoked, &av_play) || + AVMATCH (&methodInvoked, &av_publish)) { + r->m_bPlaying = true; + } + free (methodInvoked.av_val); + } else if (AVMATCH (&method, &av_onBWDone)) { + SendCheckBW (r); + } else if (AVMATCH (&method, &av_onFCSubscribe)) { + /* SendOnFCSubscribe(); */ + } else if (AVMATCH (&method, &av_onFCUnsubscribe)) { + RTMP_Close (r); + ret = 1; + } else if (AVMATCH (&method, &av__onbwcheck)) { + SendCheckBWResult (r, txn); + } else if (AVMATCH (&method, &av__onbwdone)) { + int i; + for (i = 0; i < r->m_numCalls; i++) + if (AVMATCH (&r->m_methodCalls[i].name, &av__checkbw)) { + AV_erase (r->m_methodCalls, &r->m_numCalls, i, true); + break; + } + } else if (AVMATCH (&method, &av__error)) { + RTMP_Log (RTMP_LOGERROR, "rtmp server sent error"); + } else if (AVMATCH (&method, &av_close)) { + RTMP_Log (RTMP_LOGERROR, "rtmp server requested close"); + RTMP_Close (r); + } else if (AVMATCH (&method, &av_onStatus)) { + AMFObject obj2; + AVal code, level; + AMFProp_GetObject (AMF_GetProp (&obj, NULL, 3), &obj2); + AMFProp_GetString (AMF_GetProp (&obj2, &av_code, -1), &code); + AMFProp_GetString (AMF_GetProp (&obj2, &av_level, -1), &level); + + RTMP_Log (RTMP_LOGDEBUG, "%s, onStatus: %s", __FUNCTION__, code.av_val); + if (AVMATCH (&code, &av_NetStream_Failed) + || AVMATCH (&code, &av_NetStream_Play_Failed) + || AVMATCH (&code, &av_NetStream_Play_StreamNotFound) + || AVMATCH (&code, &av_NetConnection_Connect_InvalidApp)) { + r->m_stream_id = -1; + RTMP_Close (r); + RTMP_Log (RTMP_LOGERROR, "Closing connection: %s", code.av_val); + } + + else if (AVMATCH (&code, &av_NetStream_Play_Start)) { + int i; + r->m_bPlaying = true; + for (i = 0; i < r->m_numCalls; i++) { + if (AVMATCH (&r->m_methodCalls[i].name, &av_play)) { + AV_erase (r->m_methodCalls, &r->m_numCalls, i, true); + break; + } + } + } + + else if (AVMATCH (&code, &av_NetStream_Publish_Start)) { + int i; + r->m_bPlaying = true; + for (i = 0; i < r->m_numCalls; i++) { + if (AVMATCH (&r->m_methodCalls[i].name, &av_publish)) { + AV_erase (r->m_methodCalls, &r->m_numCalls, i, true); + break; + } + } + } + + /* Return 1 if this is a Play.Complete or Play.Stop */ + else if (AVMATCH (&code, &av_NetStream_Play_Complete) + || AVMATCH (&code, &av_NetStream_Play_Stop) + || AVMATCH (&code, &av_NetStream_Play_UnpublishNotify)) { + RTMP_Close (r); + ret = 1; + } + + else if (AVMATCH (&code, &av_NetStream_Seek_Notify)) { + r->m_read.flags &= ~RTMP_READ_SEEKING; + } + + else if (AVMATCH (&code, &av_NetStream_Pause_Notify)) { + if (r->m_pausing == 1 || r->m_pausing == 2) { + RTMP_SendPause (r, false, r->m_pauseStamp); + r->m_pausing = 3; + } + } + } else if (AVMATCH (&method, &av_playlist_ready)) { + int i; + for (i = 0; i < r->m_numCalls; i++) { + if (AVMATCH (&r->m_methodCalls[i].name, &av_set_playlist)) { + AV_erase (r->m_methodCalls, &r->m_numCalls, i, true); + break; + } + } + } else { + + } +leave: + AMF_Reset (&obj); + return ret; +} + +bool +RTMP_FindFirstMatchingProperty (AMFObject * obj, const AVal * name, + AMFObjectProperty * p) +{ + int n; + /* this is a small object search to locate the "duration" property */ + for (n = 0; n < obj->o_num; n++) { + AMFObjectProperty *prop = AMF_GetProp (obj, NULL, n); + + if (AVMATCH (&prop->p_name, name)) { + *p = *prop; + return true; + } + + if (prop->p_type == AMF_OBJECT) { + if (RTMP_FindFirstMatchingProperty (&prop->p_vu.p_object, name, p)) + return true; + } + } + return false; +} + +/* Like above, but only check if name is a prefix of property */ +static bool +RTMP_FindPrefixProperty (AMFObject * obj, const AVal * name, + AMFObjectProperty * p) +{ + int n; + for (n = 0; n < obj->o_num; n++) { + AMFObjectProperty *prop = AMF_GetProp (obj, NULL, n); + + if (prop->p_name.av_len > name->av_len && + !memcmp (prop->p_name.av_val, name->av_val, name->av_len)) { + *p = *prop; + return true; + } + + if (prop->p_type == AMF_OBJECT) { + if (RTMP_FindPrefixProperty (&prop->p_vu.p_object, name, p)) + return true; + } + } + return false; +} + +static bool +DumpMetaData (AMFObject * obj) +{ + AMFObjectProperty *prop; + int n; + for (n = 0; n < obj->o_num; n++) { + prop = AMF_GetProp (obj, NULL, n); + if (prop->p_type != AMF_OBJECT) { + char str[256] = ""; + switch (prop->p_type) { + case AMF_NUMBER: + snprintf (str, 255, "%.2f", prop->p_vu.p_number); + break; + case AMF_BOOLEAN: + snprintf (str, 255, "%s", + prop->p_vu.p_number != 0. ? "TRUE" : "FALSE"); + break; + case AMF_STRING: + snprintf (str, 255, "%.*s", prop->p_vu.p_aval.av_len, + prop->p_vu.p_aval.av_val); + break; + case AMF_DATE: + snprintf (str, 255, "timestamp:%.2f", prop->p_vu.p_number); + break; + default: + snprintf (str, 255, "INVALID TYPE 0x%02x", + (unsigned char) prop->p_type); + } + if (prop->p_name.av_len) { + /* chomp */ + if (strlen (str) >= 1 && str[strlen (str) - 1] == '\n') + str[strlen (str) - 1] = '\0'; + RTMP_Log (RTMP_LOGINFO, " %-22.*s%s", prop->p_name.av_len, + prop->p_name.av_val, str); + } + } else { + if (prop->p_name.av_len) + RTMP_Log (RTMP_LOGINFO, "%.*s:", prop->p_name.av_len, + prop->p_name.av_val); + DumpMetaData (&prop->p_vu.p_object); + } + } + return false; +} + +SAVC (onMetaData); +SAVC (duration); +SAVC (video); +SAVC (audio); + +static bool +HandleMetadata (RTMP * r, char *body, unsigned int len) +{ + /* allright we get some info here, so parse it and print it */ + /* also keep duration or filesize to make a nice progress bar */ + + AMFObject obj; + AVal metastring; + bool ret = false; + + int nRes = AMF_Decode (&obj, body, len, false); + if (nRes < 0) { + RTMP_Log (RTMP_LOGERROR, "%s, error decoding meta data packet", + __FUNCTION__); + return false; + } + + AMF_Dump (&obj); + AMFProp_GetString (AMF_GetProp (&obj, NULL, 0), &metastring); + + if (AVMATCH (&metastring, &av_onMetaData)) { + AMFObjectProperty prop; + /* Show metadata */ + RTMP_Log (RTMP_LOGINFO, "Metadata:"); + DumpMetaData (&obj); + if (RTMP_FindFirstMatchingProperty (&obj, &av_duration, &prop)) { + r->m_fDuration = prop.p_vu.p_number; + /*RTMP_Log(RTMP_LOGDEBUG, "Set duration: %.2f", m_fDuration); */ + } + /* Search for audio or video tags */ + if (RTMP_FindPrefixProperty (&obj, &av_video, &prop)) + r->m_read.dataType |= 1; + if (RTMP_FindPrefixProperty (&obj, &av_audio, &prop)) + r->m_read.dataType |= 4; + ret = true; + } + AMF_Reset (&obj); + return ret; +} + +static void +HandleChangeChunkSize (RTMP * r, const RTMPPacket * packet) +{ + if (packet->m_nBodySize >= 4) { + r->m_inChunkSize = AMF_DecodeInt32 (packet->m_body); + RTMP_Log (RTMP_LOGDEBUG, "%s, received: chunk size change to %d", + __FUNCTION__, r->m_inChunkSize); + } +} + +static void +HandleAudio (RTMP * r, const RTMPPacket * packet) +{ +} + +static void +HandleVideo (RTMP * r, const RTMPPacket * packet) +{ +} + +static void +HandleCtrl (RTMP * r, const RTMPPacket * packet) +{ + short nType = -1; + unsigned int tmp; + if (packet->m_body && packet->m_nBodySize >= 2) + nType = AMF_DecodeInt16 (packet->m_body); + RTMP_Log (RTMP_LOGDEBUG, "%s, received ctrl. type: %d, len: %d", __FUNCTION__, + nType, packet->m_nBodySize); + /*RTMP_LogHex(packet.m_body, packet.m_nBodySize); */ + + if (packet->m_nBodySize >= 6) { + switch (nType) { + case 0: + tmp = AMF_DecodeInt32 (packet->m_body + 2); + RTMP_Log (RTMP_LOGDEBUG, "%s, Stream Begin %d", __FUNCTION__, tmp); + break; + + case 1: + tmp = AMF_DecodeInt32 (packet->m_body + 2); + RTMP_Log (RTMP_LOGDEBUG, "%s, Stream EOF %d", __FUNCTION__, tmp); + if (r->m_pausing == 1) + r->m_pausing = 2; + break; + + case 2: + tmp = AMF_DecodeInt32 (packet->m_body + 2); + RTMP_Log (RTMP_LOGDEBUG, "%s, Stream Dry %d", __FUNCTION__, tmp); + break; + + case 4: + tmp = AMF_DecodeInt32 (packet->m_body + 2); + RTMP_Log (RTMP_LOGDEBUG, "%s, Stream IsRecorded %d", __FUNCTION__, tmp); + break; + + case 6: /* server ping. reply with pong. */ + tmp = AMF_DecodeInt32 (packet->m_body + 2); + RTMP_Log (RTMP_LOGDEBUG, "%s, Ping %d", __FUNCTION__, tmp); + RTMP_SendCtrl (r, 0x07, tmp, 0); + break; + + /* FMS 3.5 servers send the following two controls to let the client + * know when the server has sent a complete buffer. I.e., when the + * server has sent an amount of data equal to m_nBufferMS in duration. + * The server meters its output so that data arrives at the client + * in realtime and no faster. + * + * The rtmpdump program tries to set m_nBufferMS as large as + * possible, to force the server to send data as fast as possible. + * In practice, the server appears to cap this at about 1 hour's + * worth of data. After the server has sent a complete buffer, and + * sends this BufferEmpty message, it will wait until the play + * duration of that buffer has passed before sending a new buffer. + * The BufferReady message will be sent when the new buffer starts. + * (There is no BufferReady message for the very first buffer; + * presumably the Stream Begin message is sufficient for that + * purpose.) + * + * If the network speed is much faster than the data bitrate, then + * there may be long delays between the end of one buffer and the + * start of the next. + * + * Since usually the network allows data to be sent at + * faster than realtime, and rtmpdump wants to download the data + * as fast as possible, we use this RTMP_LF_BUFX hack: when we + * get the BufferEmpty message, we send a Pause followed by an + * Unpause. This causes the server to send the next buffer immediately + * instead of waiting for the full duration to elapse. (That's + * also the purpose of the ToggleStream function, which rtmpdump + * calls if we get a read timeout.) + * + * Media player apps don't need this hack since they are just + * going to play the data in realtime anyway. It also doesn't work + * for live streams since they obviously can only be sent in + * realtime. And it's all moot if the network speed is actually + * slower than the media bitrate. + */ + case 31: + tmp = AMF_DecodeInt32 (packet->m_body + 2); + RTMP_Log (RTMP_LOGDEBUG, "%s, Stream BufferEmpty %d", __FUNCTION__, + tmp); + if (!(r->Link.lFlags & RTMP_LF_BUFX)) + break; + if (!r->m_pausing) { + r->m_pauseStamp = r->m_channelTimestamp[r->m_mediaChannel]; + RTMP_SendPause (r, true, r->m_pauseStamp); + r->m_pausing = 1; + } else if (r->m_pausing == 2) { + RTMP_SendPause (r, false, r->m_pauseStamp); + r->m_pausing = 3; + } + break; + + case 32: + tmp = AMF_DecodeInt32 (packet->m_body + 2); + RTMP_Log (RTMP_LOGDEBUG, "%s, Stream BufferReady %d", __FUNCTION__, + tmp); + break; + + default: + tmp = AMF_DecodeInt32 (packet->m_body + 2); + RTMP_Log (RTMP_LOGDEBUG, "%s, Stream xx %d", __FUNCTION__, tmp); + break; + } + + } + + if (nType == 0x1A) { + RTMP_Log (RTMP_LOGDEBUG, "%s, SWFVerification ping received: ", + __FUNCTION__); +#ifdef CRYPTO + /*RTMP_LogHex(packet.m_body, packet.m_nBodySize); */ + + /* respond with HMAC SHA256 of decompressed SWF, key is the 30byte player key, also the last 30 bytes of the server handshake are applied */ + if (r->Link.SWFSize) { + RTMP_SendCtrl (r, 0x1B, 0, 0); + } else { + RTMP_Log (RTMP_LOGERROR, + "%s: Ignoring SWFVerification request, use --swfVfy!", __FUNCTION__); + } +#else + RTMP_Log (RTMP_LOGERROR, + "%s: Ignoring SWFVerification request, no CRYPTO support!", + __FUNCTION__); +#endif + } +} + +static void +HandleServerBW (RTMP * r, const RTMPPacket * packet) +{ + r->m_nServerBW = AMF_DecodeInt32 (packet->m_body); + RTMP_Log (RTMP_LOGDEBUG, "%s: server BW = %d", __FUNCTION__, r->m_nServerBW); +} + +static void +HandleClientBW (RTMP * r, const RTMPPacket * packet) +{ + r->m_nClientBW = AMF_DecodeInt32 (packet->m_body); + if (packet->m_nBodySize > 4) + r->m_nClientBW2 = packet->m_body[4]; + else + r->m_nClientBW2 = -1; + RTMP_Log (RTMP_LOGDEBUG, "%s: client BW = %d %d", __FUNCTION__, + r->m_nClientBW, r->m_nClientBW2); +} + +static int +DecodeInt32LE (const char *data) +{ + unsigned char *c = (unsigned char *) data; + unsigned int val; + + val = (c[3] << 24) | (c[2] << 16) | (c[1] << 8) | c[0]; + return val; +} + +static int +EncodeInt32LE (char *output, int nVal) +{ + output[0] = nVal; + nVal >>= 8; + output[1] = nVal; + nVal >>= 8; + output[2] = nVal; + nVal >>= 8; + output[3] = nVal; + return 4; +} + +bool +RTMP_ReadPacket (RTMP * r, RTMPPacket * packet) +{ + char hbuf[RTMP_MAX_HEADER_SIZE] = { 0 }, *header = hbuf; + int nSize, hSize, nToRead, nChunk; + bool didAlloc = false; + + RTMP_Log (RTMP_LOGDEBUG2, "%s: fd=%d", __FUNCTION__, r->m_sb.sb_socket); + + if (ReadN (r, hbuf, 1) == 0) { + RTMP_Log (RTMP_LOGERROR, "%s, failed to read RTMP packet header", + __FUNCTION__); + return false; + } + + packet->m_headerType = (hbuf[0] & 0xc0) >> 6; + packet->m_nChannel = (hbuf[0] & 0x3f); + header++; + if (packet->m_nChannel == 0) { + if (ReadN (r, &hbuf[1], 1) != 1) { + RTMP_Log (RTMP_LOGERROR, "%s, failed to read RTMP packet header 2nd byte", + __FUNCTION__); + return false; + } + packet->m_nChannel = (unsigned) hbuf[1]; + packet->m_nChannel += 64; + header++; + } else if (packet->m_nChannel == 1) { + int tmp; + if (ReadN (r, &hbuf[1], 2) != 2) { + RTMP_Log (RTMP_LOGERROR, "%s, failed to read RTMP packet header 3nd byte", + __FUNCTION__); + return false; + } + tmp = (((unsigned) hbuf[2]) << 8) + (unsigned) hbuf[1]; + packet->m_nChannel = tmp + 64; + RTMP_Log (RTMP_LOGDEBUG, "%s, m_nChannel: %0x", __FUNCTION__, + packet->m_nChannel); + header += 2; + } + + nSize = packetSize[packet->m_headerType]; + + if (nSize == RTMP_LARGE_HEADER_SIZE) /* if we get a full header the timestamp is absolute */ + packet->m_hasAbsTimestamp = true; + + else if (nSize < RTMP_LARGE_HEADER_SIZE) { /* using values from the last message of this channel */ + if (r->m_vecChannelsIn[packet->m_nChannel]) + memcpy (packet, r->m_vecChannelsIn[packet->m_nChannel], + sizeof (RTMPPacket)); + } + + nSize--; + + if (nSize > 0 && ReadN (r, header, nSize) != nSize) { + RTMP_Log (RTMP_LOGERROR, "%s, failed to read RTMP packet header. type: %x", + __FUNCTION__, (unsigned int) hbuf[0]); + return false; + } + + hSize = nSize + (header - hbuf); + + if (nSize >= 3) { + packet->m_nTimeStamp = AMF_DecodeInt24 (header); + + /*RTMP_Log(RTMP_LOGDEBUG, "%s, reading RTMP packet chunk on channel %x, headersz %i, timestamp %i, abs timestamp %i", __FUNCTION__, packet.m_nChannel, nSize, packet.m_nTimeStamp, packet.m_hasAbsTimestamp); */ + + if (nSize >= 6) { + packet->m_nBodySize = AMF_DecodeInt24 (header + 3); + packet->m_nBytesRead = 0; + RTMPPacket_Free (packet); + + if (nSize > 6) { + packet->m_packetType = header[6]; + + if (nSize == 11) + packet->m_nInfoField2 = DecodeInt32LE (header + 7); + } + } + if (packet->m_nTimeStamp == 0xffffff) { + if (ReadN (r, header + nSize, 4) != 4) { + RTMP_Log (RTMP_LOGERROR, "%s, failed to read extended timestamp", + __FUNCTION__); + return false; + } + packet->m_nTimeStamp = AMF_DecodeInt32 (header + nSize); + hSize += 4; + } + } + + RTMP_LogHexString (RTMP_LOGDEBUG2, (uint8_t *) hbuf, hSize); + + if (packet->m_nBodySize > 0 && packet->m_body == NULL) { + if (!RTMPPacket_Alloc (packet, packet->m_nBodySize)) { + RTMP_Log (RTMP_LOGDEBUG, "%s, failed to allocate packet", __FUNCTION__); + return false; + } + didAlloc = true; + packet->m_headerType = (hbuf[0] & 0xc0) >> 6; + } + + nToRead = packet->m_nBodySize - packet->m_nBytesRead; + nChunk = r->m_inChunkSize; + if (nToRead < nChunk) + nChunk = nToRead; + + /* Does the caller want the raw chunk? */ + if (packet->m_chunk) { + packet->m_chunk->c_headerSize = hSize; + memcpy (packet->m_chunk->c_header, hbuf, hSize); + packet->m_chunk->c_chunk = packet->m_body + packet->m_nBytesRead; + packet->m_chunk->c_chunkSize = nChunk; + } + + if (ReadN (r, packet->m_body + packet->m_nBytesRead, nChunk) != nChunk) { + RTMP_Log (RTMP_LOGERROR, "%s, failed to read RTMP packet body. len: %lu", + __FUNCTION__, packet->m_nBodySize); + return false; + } + + RTMP_LogHexString (RTMP_LOGDEBUG2, + (uint8_t *) packet->m_body + packet->m_nBytesRead, nChunk); + + packet->m_nBytesRead += nChunk; + + /* keep the packet as ref for other packets on this channel */ + if (!r->m_vecChannelsIn[packet->m_nChannel]) + r->m_vecChannelsIn[packet->m_nChannel] = malloc (sizeof (RTMPPacket)); + memcpy (r->m_vecChannelsIn[packet->m_nChannel], packet, sizeof (RTMPPacket)); + + if (RTMPPacket_IsReady (packet)) { + /* make packet's timestamp absolute */ + if (!packet->m_hasAbsTimestamp) + packet->m_nTimeStamp += r->m_channelTimestamp[packet->m_nChannel]; /* timestamps seem to be always relative!! */ + + r->m_channelTimestamp[packet->m_nChannel] = packet->m_nTimeStamp; + + /* reset the data from the stored packet. we keep the header since we may use it later if a new packet for this channel */ + /* arrives and requests to re-use some info (small packet header) */ + r->m_vecChannelsIn[packet->m_nChannel]->m_body = NULL; + r->m_vecChannelsIn[packet->m_nChannel]->m_nBytesRead = 0; + r->m_vecChannelsIn[packet->m_nChannel]->m_hasAbsTimestamp = false; /* can only be false if we reuse header */ + } else { + packet->m_body = NULL; /* so it won't be erased on free */ + } + + return true; +} + +#ifndef CRYPTO +static bool +HandShake (RTMP * r, bool FP9HandShake) +{ + int i; + uint32_t uptime, suptime; + bool bMatch; + char type; + char clientbuf[RTMP_SIG_SIZE + 1], *clientsig = clientbuf + 1; + char serversig[RTMP_SIG_SIZE]; + + clientbuf[0] = 0x03; /* not encrypted */ + + uptime = htonl (RTMP_GetTime ()); + memcpy (clientsig, &uptime, 4); + + memset (&clientsig[4], 0, 4); + +#ifdef _DEBUG + for (i = 8; i < RTMP_SIG_SIZE; i++) + clientsig[i] = 0xff; +#else + for (i = 8; i < RTMP_SIG_SIZE; i++) + clientsig[i] = (char) (rand () % 256); +#endif + + if (!WriteN (r, clientbuf, RTMP_SIG_SIZE + 1)) + return false; + + if (ReadN (r, &type, 1) != 1) /* 0x03 or 0x06 */ + return false; + + RTMP_Log (RTMP_LOGDEBUG, "%s: Type Answer : %02X", __FUNCTION__, type); + + if (type != clientbuf[0]) + RTMP_Log (RTMP_LOGWARNING, + "%s: Type mismatch: client sent %d, server answered %d", __FUNCTION__, + clientbuf[0], type); + + if (ReadN (r, serversig, RTMP_SIG_SIZE) != RTMP_SIG_SIZE) + return false; + + /* decode server response */ + + memcpy (&suptime, serversig, 4); + suptime = ntohl (suptime); + + RTMP_Log (RTMP_LOGDEBUG, "%s: Server Uptime : %d", __FUNCTION__, suptime); + RTMP_Log (RTMP_LOGDEBUG, "%s: FMS Version : %d.%d.%d.%d", __FUNCTION__, + serversig[4], serversig[5], serversig[6], serversig[7]); + + /* 2nd part of handshake */ + if (!WriteN (r, serversig, RTMP_SIG_SIZE)) + return false; + + if (ReadN (r, serversig, RTMP_SIG_SIZE) != RTMP_SIG_SIZE) + return false; + + bMatch = (memcmp (serversig, clientsig, RTMP_SIG_SIZE) == 0); + if (!bMatch) { + RTMP_Log (RTMP_LOGWARNING, "%s, client signature does not match!", + __FUNCTION__); + } + return true; +} + +static bool +SHandShake (RTMP * r) +{ + int i; + char serverbuf[RTMP_SIG_SIZE + 1], *serversig = serverbuf + 1; + char clientsig[RTMP_SIG_SIZE]; + uint32_t uptime; + bool bMatch; + + if (ReadN (r, serverbuf, 1) != 1) /* 0x03 or 0x06 */ + return false; + + RTMP_Log (RTMP_LOGDEBUG, "%s: Type Request : %02X", __FUNCTION__, + serverbuf[0]); + + if (serverbuf[0] != 3) { + RTMP_Log (RTMP_LOGERROR, "%s: Type unknown: client sent %02X", + __FUNCTION__, serverbuf[0]); + return false; + } + + uptime = htonl (RTMP_GetTime ()); + memcpy (serversig, &uptime, 4); + + memset (&serversig[4], 0, 4); +#ifdef _DEBUG + for (i = 8; i < RTMP_SIG_SIZE; i++) + serversig[i] = 0xff; +#else + for (i = 8; i < RTMP_SIG_SIZE; i++) + serversig[i] = (char) (rand () % 256); +#endif + + if (!WriteN (r, serverbuf, RTMP_SIG_SIZE + 1)) + return false; + + if (ReadN (r, clientsig, RTMP_SIG_SIZE) != RTMP_SIG_SIZE) + return false; + + /* decode client response */ + + memcpy (&uptime, clientsig, 4); + uptime = ntohl (uptime); + + RTMP_Log (RTMP_LOGDEBUG, "%s: Client Uptime : %d", __FUNCTION__, uptime); + RTMP_Log (RTMP_LOGDEBUG, "%s: Player Version: %d.%d.%d.%d", __FUNCTION__, + clientsig[4], clientsig[5], clientsig[6], clientsig[7]); + + /* 2nd part of handshake */ + if (!WriteN (r, clientsig, RTMP_SIG_SIZE)) + return false; + + if (ReadN (r, clientsig, RTMP_SIG_SIZE) != RTMP_SIG_SIZE) + return false; + + bMatch = (memcmp (serversig, clientsig, RTMP_SIG_SIZE) == 0); + if (!bMatch) { + RTMP_Log (RTMP_LOGWARNING, "%s, client signature does not match!", + __FUNCTION__); + } + return true; +} +#endif + +bool +RTMP_SendChunk (RTMP * r, RTMPChunk * chunk) +{ + bool wrote; + char hbuf[RTMP_MAX_HEADER_SIZE]; + + RTMP_Log (RTMP_LOGDEBUG2, "%s: fd=%d, size=%d", __FUNCTION__, + r->m_sb.sb_socket, chunk->c_chunkSize); + RTMP_LogHexString (RTMP_LOGDEBUG2, (uint8_t *) chunk->c_header, + chunk->c_headerSize); + if (chunk->c_chunkSize) { + char *ptr = chunk->c_chunk - chunk->c_headerSize; + RTMP_LogHexString (RTMP_LOGDEBUG2, (uint8_t *) chunk->c_chunk, + chunk->c_chunkSize); + /* save header bytes we're about to overwrite */ + memcpy (hbuf, ptr, chunk->c_headerSize); + memcpy (ptr, chunk->c_header, chunk->c_headerSize); + wrote = WriteN (r, ptr, chunk->c_headerSize + chunk->c_chunkSize); + memcpy (ptr, hbuf, chunk->c_headerSize); + } else + wrote = WriteN (r, chunk->c_header, chunk->c_headerSize); + return wrote; +} + +bool +RTMP_SendPacket (RTMP * r, RTMPPacket * packet, bool queue) +{ + const RTMPPacket *prevPacket = r->m_vecChannelsOut[packet->m_nChannel]; + uint32_t last = 0; + int nSize; + int hSize, cSize; + char *header, *hptr, *hend, hbuf[RTMP_MAX_HEADER_SIZE], c; + uint32_t t; + char *buffer, *tbuf = NULL, *toff = NULL; + int nChunkSize; + int tlen; + + if (prevPacket && 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) + packet->m_headerType = RTMP_PACKET_SIZE_MINIMUM; + last = prevPacket->m_nTimeStamp; + } + + if (packet->m_headerType > 3) { /* sanity */ + RTMP_Log (RTMP_LOGERROR, + "sanity failed!! trying to send header of type: 0x%02x.", + (unsigned char) packet->m_headerType); + return false; + } + + nSize = packetSize[packet->m_headerType]; + hSize = nSize; + cSize = 0; + t = packet->m_nTimeStamp - last; + + if (packet->m_body) { + header = packet->m_body - nSize; + hend = packet->m_body; + } else { + header = hbuf + 6; + hend = hbuf + sizeof (hbuf); + } + + if (packet->m_nChannel > 319) + cSize = 2; + else if (packet->m_nChannel > 63) + cSize = 1; + if (cSize) { + header -= cSize; + hSize += cSize; + } + + if (nSize > 1 && t >= 0xffffff) { + header -= 4; + hSize += 4; + } + + hptr = header; + c = packet->m_headerType << 6; + switch (cSize) { + case 0: + c |= packet->m_nChannel; + break; + case 1: + break; + case 2: + c |= 1; + break; + } + *hptr++ = c; + if (cSize) { + int tmp = packet->m_nChannel - 64; + *hptr++ = tmp & 0xff; + if (cSize == 2) + *hptr++ = tmp >> 8; + } + + if (nSize > 1) { + hptr = AMF_EncodeInt24 (hptr, hend, t > 0xffffff ? 0xffffff : t); + } + + if (nSize > 4) { + hptr = AMF_EncodeInt24 (hptr, hend, packet->m_nBodySize); + *hptr++ = packet->m_packetType; + } + + if (nSize > 8) + hptr += EncodeInt32LE (hptr, packet->m_nInfoField2); + + if (nSize > 1 && t >= 0xffffff) + hptr = AMF_EncodeInt32 (hptr, hend, t); + + nSize = packet->m_nBodySize; + buffer = packet->m_body; + nChunkSize = r->m_outChunkSize; + + RTMP_Log (RTMP_LOGDEBUG2, "%s: fd=%d, size=%d", __FUNCTION__, + r->m_sb.sb_socket, nSize); + /* send all chunks in one HTTP request */ + if (r->Link.protocol & RTMP_FEATURE_HTTP) { + int chunks = (nSize + nChunkSize - 1) / nChunkSize; + if (chunks > 1) { + tlen = chunks * (cSize + 1) + nSize + hSize; + tbuf = malloc (tlen); + if (!tbuf) + return false; + toff = tbuf; + } + } + while (nSize + hSize) { + int wrote; + + if (nSize < nChunkSize) + nChunkSize = nSize; + + RTMP_LogHexString (RTMP_LOGDEBUG2, (uint8_t *) header, hSize); + RTMP_LogHexString (RTMP_LOGDEBUG2, (uint8_t *) buffer, nChunkSize); + if (tbuf) { + memcpy (toff, header, nChunkSize + hSize); + toff += nChunkSize + hSize; + } else { + wrote = WriteN (r, header, nChunkSize + hSize); + if (!wrote) + return false; + } + nSize -= nChunkSize; + buffer += nChunkSize; + hSize = 0; + + if (nSize > 0) { + header = buffer - 1; + hSize = 1; + if (cSize) { + header -= cSize; + hSize += cSize; + } + *header = (0xc0 | c); + if (cSize) { + int tmp = packet->m_nChannel - 64; + header[1] = tmp & 0xff; + if (cSize == 2) + header[2] = tmp >> 8; + } + } + } + if (tbuf) { + int wrote = WriteN (r, tbuf, toff - tbuf); + free (tbuf); + tbuf = NULL; + if (!wrote) + return false; + } + + /* we invoked a remote method */ + if (packet->m_packetType == 0x14) { + AVal method; + char *ptr; + ptr = packet->m_body + 1; + AMF_DecodeString (ptr, &method); + RTMP_Log (RTMP_LOGDEBUG, "Invoking %s", method.av_val); + /* keep it in call queue till result arrives */ + if (queue) { + int txn; + ptr += 3 + method.av_len; + txn = (int) AMF_DecodeNumber (ptr); + AV_queue (&r->m_methodCalls, &r->m_numCalls, &method, txn); + } + } + + if (!r->m_vecChannelsOut[packet->m_nChannel]) + r->m_vecChannelsOut[packet->m_nChannel] = malloc (sizeof (RTMPPacket)); + memcpy (r->m_vecChannelsOut[packet->m_nChannel], packet, sizeof (RTMPPacket)); + return true; +} + +bool +RTMP_Serve (RTMP * r) +{ + return SHandShake (r); +} + +void +RTMP_Close (RTMP * r) +{ + int i; + + if (RTMP_IsConnected (r)) { + if (r->m_stream_id > 0) { + if ((r->Link.protocol & RTMP_FEATURE_WRITE)) + SendFCUnpublish (r); + i = r->m_stream_id; + r->m_stream_id = 0; + SendDeleteStream (r, i); + } + if (r->m_clientID.av_val) { + HTTP_Post (r, RTMPT_CLOSE, "", 1); + free (r->m_clientID.av_val); + r->m_clientID.av_val = NULL; + r->m_clientID.av_len = 0; + } + 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; + + free (r->m_read.buf); + r->m_read.buf = NULL; + 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; + RTMPPacket_Free (&r->m_write); + + for (i = 0; i < RTMP_CHANNELS; i++) { + if (r->m_vecChannelsIn[i]) { + RTMPPacket_Free (r->m_vecChannelsIn[i]); + free (r->m_vecChannelsIn[i]); + r->m_vecChannelsIn[i] = NULL; + } + if (r->m_vecChannelsOut[i]) { + free (r->m_vecChannelsOut[i]); + r->m_vecChannelsOut[i] = NULL; + } + } + AV_clear (r->m_methodCalls, r->m_numCalls); + r->m_methodCalls = NULL; + r->m_numCalls = 0; + r->m_numInvokes = 0; + + r->m_bPlaying = false; + r->m_sb.sb_size = 0; + + r->m_msgCounter = 0; + r->m_resplen = 0; + r->m_unackd = 0; + + free (r->Link.playpath0.av_val); + r->Link.playpath0.av_val = NULL; + +#ifdef CRYPTO + if (r->Link.dh) { + MDH_free (r->Link.dh); + r->Link.dh = NULL; + } + if (r->Link.rc4keyIn) { + free (r->Link.rc4keyIn); + r->Link.rc4keyIn = NULL; + } + if (r->Link.rc4keyOut) { + free (r->Link.rc4keyOut); + r->Link.rc4keyOut = NULL; + } +#endif +} + +int +RTMPSockBuf_Fill (RTMPSockBuf * sb) +{ + int nBytes; + + if (!sb->sb_size) + sb->sb_start = sb->sb_buf; + + while (1) { + nBytes = sizeof (sb->sb_buf) - sb->sb_size - (sb->sb_start - sb->sb_buf); +#if defined(CRYPTO) && !defined(NO_SSL) + if (sb->sb_ssl) { + nBytes = TLS_read (sb->sb_ssl, sb->sb_start + sb->sb_size, nBytes); + } else +#endif + { + nBytes = recv (sb->sb_socket, sb->sb_start + sb->sb_size, nBytes, 0); + } + if (nBytes != -1) { + sb->sb_size += nBytes; + } else { + int sockerr = GetSockError (); + RTMP_Log (RTMP_LOGDEBUG, "%s, recv returned %d. GetSockError(): %d (%s)", + __FUNCTION__, nBytes, sockerr, strerror (sockerr)); + if (sockerr == EINTR && !RTMP_ctrlC) + continue; + + if (sockerr == EWOULDBLOCK || sockerr == EAGAIN) { + sb->sb_timedout = true; + nBytes = 0; + } + } + break; + } + + return nBytes; +} + +int +RTMPSockBuf_Send (RTMPSockBuf * sb, const char *buf, int len) +{ + int rc; + +#ifdef _DEBUG + fwrite (buf, 1, len, netstackdump); +#endif + +#if defined(CRYPTO) && !defined(NO_SSL) + if (sb->sb_ssl) { + rc = TLS_write (sb->sb_ssl, buf, len); + } else +#endif + { + rc = send (sb->sb_socket, buf, len, 0); + } + return rc; +} + +int +RTMPSockBuf_Close (RTMPSockBuf * sb) +{ +#if defined(CRYPTO) && !defined(NO_SSL) + if (sb->sb_ssl) { + TLS_shutdown (sb->sb_ssl); + TLS_close (sb->sb_ssl); + sb->sb_ssl = NULL; + } +#endif + return closesocket (sb->sb_socket); +} + +#define HEX2BIN(a) (((a)&0x40)?((a)&0xf)+9:((a)&0xf)) + +static void +DecodeTEA (AVal * key, AVal * text) +{ + uint32_t *v, k[4] = { 0 }, u; + uint32_t z, y, sum = 0, e, DELTA = 0x9e3779b9; + int32_t p, q; + int i, n; + unsigned char *ptr, *out; + + /* prep key: pack 1st 16 chars into 4 LittleEndian ints */ + ptr = (unsigned char *) key->av_val; + u = 0; + n = 0; + v = k; + p = key->av_len > 16 ? 16 : key->av_len; + for (i = 0; i < p; i++) { + u |= ptr[i] << (n * 8); + if (n == 3) { + *v++ = u; + u = 0; + n = 0; + } else { + n++; + } + } + /* any trailing chars */ + if (u) + *v = u; + + /* prep text: hex2bin, multiples of 4 */ + n = (text->av_len + 7) / 8; + out = malloc (n * 8); + ptr = (unsigned char *) text->av_val; + v = (uint32_t *) out; + for (i = 0; i < n; i++) { + u = (HEX2BIN (ptr[0]) << 4) + HEX2BIN (ptr[1]); + u |= ((HEX2BIN (ptr[2]) << 4) + HEX2BIN (ptr[3])) << 8; + u |= ((HEX2BIN (ptr[4]) << 4) + HEX2BIN (ptr[5])) << 16; + u |= ((HEX2BIN (ptr[6]) << 4) + HEX2BIN (ptr[7])) << 24; + *v++ = u; + ptr += 8; + } + v = (uint32_t *) out; + + /* http://www.movable-type.co.uk/scripts/tea-block.html */ +#define MX (((z>>5)^(y<<2)) + ((y>>3)^(z<<4))) ^ ((sum^y) + (k[(p&3)^e]^z)); + z = v[n - 1]; + y = v[0]; + q = 6 + 52 / n; + sum = q * DELTA; + while (sum != 0) { + e = sum >> 2 & 3; + for (p = n - 1; p > 0; p--) + z = v[p - 1], y = v[p] -= MX; + z = v[n - 1]; + y = v[0] -= MX; + sum -= DELTA; + } + + text->av_len /= 2; + memcpy (text->av_val, out, text->av_len); + free (out); +} + +static int +HTTP_Post (RTMP * r, RTMPTCmd cmd, const char *buf, int len) +{ + char hbuf[512]; + int hlen = snprintf (hbuf, sizeof (hbuf), "POST /%s%s/%d HTTP/1.1\r\n" + "Host: %.*s:%d\r\n" + "Accept: */*\r\n" + "User-Agent: Shockwave Flash\n" + "Connection: Keep-Alive\n" + "Cache-Control: no-cache\r\n" + "Content-type: application/x-fcs\r\n" + "Content-length: %d\r\n\r\n", RTMPT_cmds[cmd], + r->m_clientID.av_val ? r->m_clientID.av_val : "", + r->m_msgCounter, r->Link.hostname.av_len, r->Link.hostname.av_val, + r->Link.port, len); + RTMPSockBuf_Send (&r->m_sb, hbuf, hlen); + hlen = RTMPSockBuf_Send (&r->m_sb, buf, len); + r->m_msgCounter++; + r->m_unackd++; + return hlen; +} + +static int +HTTP_read (RTMP * r, int fill) +{ + char *ptr; + int hlen; + + if (fill) + RTMPSockBuf_Fill (&r->m_sb); + if (r->m_sb.sb_size < 144) + return -1; + if (strncmp (r->m_sb.sb_start, "HTTP/1.1 200 ", 13)) + return -1; + ptr = strstr (r->m_sb.sb_start, "Content-Length:"); + if (!ptr) + return -1; + hlen = atoi (ptr + 16); + ptr = strstr (ptr, "\r\n\r\n"); + if (!ptr) + return -1; + ptr += 4; + r->m_sb.sb_size -= ptr - r->m_sb.sb_start; + r->m_sb.sb_start = ptr; + r->m_unackd--; + + if (!r->m_clientID.av_val) { + r->m_clientID.av_len = hlen; + r->m_clientID.av_val = malloc (hlen + 1); + if (!r->m_clientID.av_val) + return -1; + r->m_clientID.av_val[0] = '/'; + memcpy (r->m_clientID.av_val + 1, ptr, hlen - 1); + r->m_clientID.av_val[hlen] = 0; + r->m_sb.sb_size = 0; + } else { + r->m_polling = *ptr++; + r->m_resplen = hlen - 1; + r->m_sb.sb_start++; + r->m_sb.sb_size--; + } + return 0; +} + +#define MAX_IGNORED_FRAMES 50 + +/* Read from the stream until we get a media packet. + * Returns -3 if Play.Close/Stop, -2 if fatal error, -1 if no more media + * packets, 0 if ignorable error, >0 if there is a media packet + */ +static int +Read_1_Packet (RTMP * r, char *buf, unsigned int buflen) +{ + uint32_t prevTagSize = 0; + int rtnGetNextMediaPacket = 0, ret = RTMP_READ_EOF; + RTMPPacket packet = { 0 }; + bool recopy = false; + unsigned int size; + char *ptr, *pend; + uint32_t nTimeStamp = 0; + unsigned int len; + + rtnGetNextMediaPacket = RTMP_GetNextMediaPacket (r, &packet); + while (rtnGetNextMediaPacket) { + char *packetBody = packet.m_body; + unsigned int nPacketLen = packet.m_nBodySize; + + /* Return -3 if this was completed nicely with invoke message + * Play.Stop or Play.Complete + */ + if (rtnGetNextMediaPacket == 2) { + RTMP_Log (RTMP_LOGDEBUG, + "Got Play.Complete or Play.Stop from server. " + "Assuming stream is complete"); + ret = RTMP_READ_COMPLETE; + break; + } + + r->m_read.dataType |= (((packet.m_packetType == 0x08) << 2) | + (packet.m_packetType == 0x09)); + + if (packet.m_packetType == 0x09 && nPacketLen <= 5) { + RTMP_Log (RTMP_LOGDEBUG, "ignoring too small video packet: size: %d", + nPacketLen); + ret = RTMP_READ_IGNORE; + break; + } + if (packet.m_packetType == 0x08 && nPacketLen <= 1) { + RTMP_Log (RTMP_LOGDEBUG, "ignoring too small audio packet: size: %d", + nPacketLen); + ret = RTMP_READ_IGNORE; + break; + } + + if (r->m_read.flags & RTMP_READ_SEEKING) { + ret = RTMP_READ_IGNORE; + break; + } +#ifdef _DEBUG + RTMP_Log (RTMP_LOGDEBUG, "type: %02X, size: %d, TS: %d ms, abs TS: %d", + packet.m_packetType, nPacketLen, packet.m_nTimeStamp, + packet.m_hasAbsTimestamp); + if (packet.m_packetType == 0x09) + RTMP_Log (RTMP_LOGDEBUG, "frametype: %02X", (*packetBody & 0xf0)); +#endif + + if (r->m_read.flags & RTMP_READ_RESUME) { + /* check the header if we get one */ + if (packet.m_nTimeStamp == 0) { + if (r->m_read.nMetaHeaderSize > 0 && packet.m_packetType == 0x12) { + AMFObject metaObj; + int nRes = AMF_Decode (&metaObj, packetBody, nPacketLen, false); + if (nRes >= 0) { + AVal metastring; + AMFProp_GetString (AMF_GetProp (&metaObj, NULL, 0), &metastring); + + if (AVMATCH (&metastring, &av_onMetaData)) { + /* compare */ + if ((r->m_read.nMetaHeaderSize != nPacketLen) || + (memcmp + (r->m_read.metaHeader, packetBody, + r->m_read.nMetaHeaderSize) != 0)) { + ret = RTMP_READ_ERROR; + } + } + AMF_Reset (&metaObj); + if (ret == RTMP_READ_ERROR) + break; + } + } + + /* check first keyframe to make sure we got the right position + * in the stream! (the first non ignored frame) + */ + if (r->m_read.nInitialFrameSize > 0) { + /* video or audio data */ + if (packet.m_packetType == r->m_read.initialFrameType + && r->m_read.nInitialFrameSize == nPacketLen) { + /* we don't compare the sizes since the packet can + * contain several FLV packets, just make sure the + * first frame is our keyframe (which we are going + * to rewrite) + */ + if (memcmp + (r->m_read.initialFrame, packetBody, + r->m_read.nInitialFrameSize) == 0) { + RTMP_Log (RTMP_LOGDEBUG, "Checked keyframe successfully!"); + r->m_read.flags |= RTMP_READ_GOTKF; + /* ignore it! (what about audio data after it? it is + * handled by ignoring all 0ms frames, see below) + */ + ret = RTMP_READ_IGNORE; + break; + } + } + + /* hande FLV streams, even though the server resends the + * keyframe as an extra video packet it is also included + * in the first FLV stream chunk and we have to compare + * it and filter it out !! + */ + if (packet.m_packetType == 0x16) { + /* basically we have to find the keyframe with the + * correct TS being nResumeTS + */ + unsigned int pos = 0; + uint32_t ts = 0; + + while (pos + 11 < nPacketLen) { + /* size without header (11) and prevTagSize (4) */ + uint32_t dataSize = AMF_DecodeInt24 (packetBody + pos + 1); + ts = AMF_DecodeInt24 (packetBody + pos + 4); + ts |= (packetBody[pos + 7] << 24); + +#ifdef _DEBUG + RTMP_Log (RTMP_LOGDEBUG, + "keyframe search: FLV Packet: type %02X, dataSize: %d, timeStamp: %d ms", + packetBody[pos], dataSize, ts); +#endif + /* ok, is it a keyframe?: + * well doesn't work for audio! + */ + if (packetBody[pos /*6928, test 0 */ ] == + r->m_read.initialFrameType + /* && (packetBody[11]&0xf0) == 0x10 */ ) { + if (ts == r->m_read.nResumeTS) { + RTMP_Log (RTMP_LOGDEBUG, + "Found keyframe with resume-keyframe timestamp!"); + if (r->m_read.nInitialFrameSize != dataSize + || memcmp (r->m_read.initialFrame, + packetBody + pos + 11, + r->m_read.nInitialFrameSize) != 0) { + RTMP_Log (RTMP_LOGERROR, + "FLV Stream: Keyframe doesn't match!"); + ret = RTMP_READ_ERROR; + break; + } + r->m_read.flags |= RTMP_READ_GOTFLVK; + + /* skip this packet? + * check whether skippable: + */ + if (pos + 11 + dataSize + 4 > nPacketLen) { + RTMP_Log (RTMP_LOGWARNING, + "Non skipable packet since it doesn't end with chunk, stream corrupt!"); + ret = RTMP_READ_ERROR; + break; + } + packetBody += (pos + 11 + dataSize + 4); + nPacketLen -= (pos + 11 + dataSize + 4); + + goto stopKeyframeSearch; + + } else if (r->m_read.nResumeTS < ts) { + /* the timestamp ts will only increase with + * further packets, wait for seek + */ + goto stopKeyframeSearch; + } + } + pos += (11 + dataSize + 4); + } + if (ts < r->m_read.nResumeTS) { + RTMP_Log (RTMP_LOGERROR, + "First packet does not contain keyframe, all " + "timestamps are smaller than the keyframe " + "timestamp; probably the resume seek failed?"); + } + stopKeyframeSearch: + ; + if (!(r->m_read.flags & RTMP_READ_GOTFLVK)) { + RTMP_Log (RTMP_LOGERROR, + "Couldn't find the seeked keyframe in this chunk!"); + ret = RTMP_READ_IGNORE; + break; + } + } + } + } + + if (packet.m_nTimeStamp > 0 + && (r->m_read.flags & (RTMP_READ_GOTKF | RTMP_READ_GOTFLVK))) { + /* another problem is that the server can actually change from + * 09/08 video/audio packets to an FLV stream or vice versa and + * our keyframe check will prevent us from going along with the + * new stream if we resumed. + * + * in this case set the 'found keyframe' variables to true. + * We assume that if we found one keyframe somewhere and were + * already beyond TS > 0 we have written data to the output + * which means we can accept all forthcoming data including the + * change between 08/09 <-> FLV packets + */ + r->m_read.flags |= (RTMP_READ_GOTKF | RTMP_READ_GOTFLVK); + } + + /* skip till we find our keyframe + * (seeking might put us somewhere before it) + */ + if (!(r->m_read.flags & RTMP_READ_GOTKF) && packet.m_packetType != 0x16) { + RTMP_Log (RTMP_LOGWARNING, + "Stream does not start with requested frame, ignoring data... "); + r->m_read.nIgnoredFrameCounter++; + if (r->m_read.nIgnoredFrameCounter > MAX_IGNORED_FRAMES) + ret = RTMP_READ_ERROR; /* fatal error, couldn't continue stream */ + else + ret = RTMP_READ_IGNORE; + break; + } + /* ok, do the same for FLV streams */ + if (!(r->m_read.flags & RTMP_READ_GOTFLVK) && packet.m_packetType == 0x16) { + RTMP_Log (RTMP_LOGWARNING, + "Stream does not start with requested FLV frame, ignoring data... "); + r->m_read.nIgnoredFlvFrameCounter++; + if (r->m_read.nIgnoredFlvFrameCounter > MAX_IGNORED_FRAMES) + ret = RTMP_READ_ERROR; + else + ret = RTMP_READ_IGNORE; + break; + } + + /* we have to ignore the 0ms frames since these are the first + * keyframes; we've got these so don't mess around with multiple + * copies sent by the server to us! (if the keyframe is found at a + * later position there is only one copy and it will be ignored by + * the preceding if clause) + */ + if (!(r->m_read.flags & RTMP_READ_NO_IGNORE) && packet.m_packetType != 0x16) { /* exclude type 0x16 (FLV) since it can + * contain several FLV packets */ + if (packet.m_nTimeStamp == 0) { + ret = RTMP_READ_IGNORE; + break; + } else { + /* stop ignoring packets */ + r->m_read.flags |= RTMP_READ_NO_IGNORE; + } + } + } + + /* calculate packet size and allocate slop buffer if necessary */ + size = nPacketLen + + ((packet.m_packetType == 0x08 || packet.m_packetType == 0x09 + || packet.m_packetType == 0x12) ? 11 : 0) + + (packet.m_packetType != 0x16 ? 4 : 0); + + if (size + 4 > buflen) { + /* the extra 4 is for the case of an FLV stream without a last + * prevTagSize (we need extra 4 bytes to append it) */ + r->m_read.buf = malloc (size + 4); + if (r->m_read.buf == 0) { + RTMP_Log (RTMP_LOGERROR, "Couldn't allocate memory!"); + ret = RTMP_READ_ERROR; /* fatal error */ + break; + } + recopy = true; + ptr = r->m_read.buf; + } else { + ptr = buf; + } + pend = ptr + size + 4; + + /* use to return timestamp of last processed packet */ + + /* audio (0x08), video (0x09) or metadata (0x12) packets : + * construct 11 byte header then add rtmp packet's data */ + if (packet.m_packetType == 0x08 || packet.m_packetType == 0x09 + || packet.m_packetType == 0x12) { + nTimeStamp = r->m_read.nResumeTS + packet.m_nTimeStamp; + prevTagSize = 11 + nPacketLen; + + *ptr = packet.m_packetType; + ptr++; + ptr = AMF_EncodeInt24 (ptr, pend, nPacketLen); + +#if 0 + if (packet.m_packetType == 0x09) { /* video */ + + /* H264 fix: */ + if ((packetBody[0] & 0x0f) == 7) { /* CodecId = H264 */ + uint8_t packetType = *(packetBody + 1); + + uint32_t ts = AMF_DecodeInt24 (packetBody + 2); /* composition time */ + int32_t cts = (ts + 0xff800000) ^ 0xff800000; + RTMP_Log (RTMP_LOGDEBUG, "cts : %d\n", cts); + + nTimeStamp -= cts; + /* get rid of the composition time */ + CRTMP::EncodeInt24 (packetBody + 2, 0); + } + RTMP_Log (RTMP_LOGDEBUG, "VIDEO: nTimeStamp: 0x%08X (%d)\n", nTimeStamp, + nTimeStamp); + } +#endif + + ptr = AMF_EncodeInt24 (ptr, pend, nTimeStamp); + *ptr = (char) ((nTimeStamp & 0xFF000000) >> 24); + ptr++; + + /* stream id */ + ptr = AMF_EncodeInt24 (ptr, pend, 0); + } + + memcpy (ptr, packetBody, nPacketLen); + len = nPacketLen; + + /* correct tagSize and obtain timestamp if we have an FLV stream */ + if (packet.m_packetType == 0x16) { + unsigned int pos = 0; + int delta; + + /* grab first timestamp and see if it needs fixing */ + nTimeStamp = AMF_DecodeInt24 (packetBody + 4); + nTimeStamp |= (packetBody[7] << 24); + delta = packet.m_nTimeStamp - nTimeStamp; + + while (pos + 11 < nPacketLen) { + /* size without header (11) and without prevTagSize (4) */ + uint32_t dataSize = AMF_DecodeInt24 (packetBody + pos + 1); + nTimeStamp = AMF_DecodeInt24 (packetBody + pos + 4); + nTimeStamp |= (packetBody[pos + 7] << 24); + + if (delta) { + nTimeStamp += delta; + AMF_EncodeInt24 (ptr + pos + 4, pend, nTimeStamp); + ptr[pos + 7] = nTimeStamp >> 24; + } + + /* set data type */ + r->m_read.dataType |= (((*(packetBody + pos) == 0x08) << 2) | + (*(packetBody + pos) == 0x09)); + + if (pos + 11 + dataSize + 4 > nPacketLen) { + if (pos + 11 + dataSize > nPacketLen) { + RTMP_Log (RTMP_LOGERROR, + "Wrong data size (%lu), stream corrupted, aborting!", dataSize); + ret = RTMP_READ_ERROR; + break; + } + RTMP_Log (RTMP_LOGWARNING, "No tagSize found, appending!"); + + /* we have to append a last tagSize! */ + prevTagSize = dataSize + 11; + AMF_EncodeInt32 (ptr + pos + 11 + dataSize, pend, prevTagSize); + size += 4; + len += 4; + } else { + prevTagSize = AMF_DecodeInt32 (packetBody + pos + 11 + dataSize); + +#ifdef _DEBUG + RTMP_Log (RTMP_LOGDEBUG, + "FLV Packet: type %02X, dataSize: %lu, tagSize: %lu, timeStamp: %lu ms", + (unsigned char) packetBody[pos], dataSize, prevTagSize, + nTimeStamp); +#endif + + if (prevTagSize != (dataSize + 11)) { +#ifdef _DEBUG + RTMP_Log (RTMP_LOGWARNING, + "Tag and data size are not consitent, writing tag size according to dataSize+11: %d", + dataSize + 11); +#endif + + prevTagSize = dataSize + 11; + AMF_EncodeInt32 (ptr + pos + 11 + dataSize, pend, prevTagSize); + } + } + + pos += prevTagSize + 4; /*(11+dataSize+4); */ + } + } + ptr += len; + + if (packet.m_packetType != 0x16) { + /* FLV tag packets contain their own prevTagSize */ + AMF_EncodeInt32 (ptr, pend, prevTagSize); + } + + /* In non-live this nTimeStamp can contain an absolute TS. + * Update ext timestamp with this absolute offset in non-live mode + * otherwise report the relative one + */ + /* RTMP_Log(RTMP_LOGDEBUG, "type: %02X, size: %d, pktTS: %dms, TS: %dms, bLiveStream: %d", packet.m_packetType, nPacketLen, packet.m_nTimeStamp, nTimeStamp, r->Link.lFlags & RTMP_LF_LIVE); */ + r->m_read.timestamp = + (r->Link.lFlags & RTMP_LF_LIVE) ? packet.m_nTimeStamp : nTimeStamp; + + ret = size; + break; + } + + if (rtnGetNextMediaPacket) + RTMPPacket_Free (&packet); + + if (recopy) { + len = ret > buflen ? buflen : ret; + memcpy (buf, r->m_read.buf, len); + r->m_read.bufpos = r->m_read.buf + len; + r->m_read.buflen = ret - len; + } + return ret; +} + +static const char flvHeader[] = { 'F', 'L', 'V', 0x01, + 0x00, /* 0x04 == audio, 0x01 == video */ + 0x00, 0x00, 0x00, 0x09, + 0x00, 0x00, 0x00, 0x00 +}; + +#define HEADERBUF (128*1024) +int +RTMP_Read (RTMP * r, char *buf, int size) +{ + int nRead = 0, total = 0; + + /* can't continue */ +fail: + switch (r->m_read.status < 0) { + case RTMP_READ_EOF: + case RTMP_READ_COMPLETE: + return 0; + case RTMP_READ_ERROR: /* corrupted stream, resume failed */ + SetSockError (EINVAL); + return -1; + default: + break; + } + + /* first time thru */ + if (!(r->m_read.flags & RTMP_READ_HEADER)) { + if (!(r->m_read.flags & RTMP_READ_RESUME)) { + char *mybuf = malloc (HEADERBUF), *end = mybuf + HEADERBUF; + int cnt = 0; + r->m_read.buf = mybuf; + r->m_read.buflen = HEADERBUF; + + memcpy (mybuf, flvHeader, sizeof (flvHeader)); + r->m_read.buf += sizeof (flvHeader); + r->m_read.buflen -= sizeof (flvHeader); + + while (r->m_read.timestamp == 0) { + nRead = Read_1_Packet (r, r->m_read.buf, r->m_read.buflen); + if (nRead < 0) { + free (mybuf); + r->m_read.buf = NULL; + r->m_read.buflen = 0; + r->m_read.status = nRead; + goto fail; + } + /* buffer overflow, fix buffer and give up */ + if (r->m_read.buf < mybuf || r->m_read.buf > end) { + mybuf = realloc (mybuf, cnt + nRead); + memcpy (mybuf + cnt, r->m_read.buf, nRead); + r->m_read.buf = mybuf + cnt + nRead; + break; + } + cnt += nRead; + r->m_read.buf += nRead; + r->m_read.buflen -= nRead; + if (r->m_read.dataType == 5) + break; + } + mybuf[4] = r->m_read.dataType; + r->m_read.buflen = r->m_read.buf - mybuf; + r->m_read.buf = mybuf; + r->m_read.bufpos = mybuf; + } + r->m_read.flags |= RTMP_READ_HEADER; + } + + if ((r->m_read.flags & RTMP_READ_SEEKING) && r->m_read.buf) { + /* drop whatever's here */ + free (r->m_read.buf); + r->m_read.buf = NULL; + r->m_read.bufpos = NULL; + r->m_read.buflen = 0; + } + + /* If there's leftover data buffered, use it up */ + if (r->m_read.buf) { + nRead = r->m_read.buflen; + if (nRead > size) + nRead = size; + memcpy (buf, r->m_read.bufpos, nRead); + r->m_read.buflen -= nRead; + if (!r->m_read.buflen) { + free (r->m_read.buf); + r->m_read.buf = NULL; + r->m_read.bufpos = NULL; + } else { + r->m_read.bufpos += nRead; + } + buf += nRead; + total += nRead; + size -= nRead; + } + + while (size > 0 && (nRead = Read_1_Packet (r, buf, size)) >= 0) { + if (!nRead) + continue; + buf += nRead; + total += nRead; + size -= nRead; + break; + } + if (nRead < 0) + r->m_read.status = nRead; + + if (size < 0) + total += size; + return total; +} + +static const AVal av_setDataFrame = AVC ("@setDataFrame"); + +int +RTMP_Write (RTMP * r, char *buf, int size) +{ + RTMPPacket *pkt = &r->m_write; + char *pend, *enc; + int s2 = size, ret, num; + + pkt->m_nChannel = 0x04; /* source channel */ + pkt->m_nInfoField2 = r->m_stream_id; + + while (s2) { + if (!pkt->m_nBytesRead) { + if (size < 11) { + /* FLV pkt too small */ + return 0; + } + + if (buf[0] == 'F' && buf[1] == 'L' && buf[2] == 'V') { + buf += 13; + s2 -= 13; + } + + pkt->m_packetType = *buf++; + pkt->m_nBodySize = AMF_DecodeInt24 (buf); + buf += 3; + pkt->m_nTimeStamp = AMF_DecodeInt24 (buf); + buf += 3; + pkt->m_nTimeStamp |= *buf++ << 24; + buf += 3; + s2 -= 11; + + if (((pkt->m_packetType == 0x08 || pkt->m_packetType == 0x09) && + !pkt->m_nTimeStamp) || pkt->m_packetType == 0x12) { + pkt->m_headerType = RTMP_PACKET_SIZE_LARGE; + if (pkt->m_packetType == 0x12) + pkt->m_nBodySize += 16; + } else { + pkt->m_headerType = RTMP_PACKET_SIZE_MEDIUM; + } + + if (!RTMPPacket_Alloc (pkt, pkt->m_nBodySize)) { + RTMP_Log (RTMP_LOGDEBUG, "%s, failed to allocate packet", __FUNCTION__); + return false; + } + enc = pkt->m_body; + pend = enc + pkt->m_nBodySize; + if (pkt->m_packetType == 0x12) { + enc = AMF_EncodeString (enc, pend, &av_setDataFrame); + pkt->m_nBytesRead = enc - pkt->m_body; + } + } else { + enc = pkt->m_body + pkt->m_nBytesRead; + } + num = pkt->m_nBodySize - pkt->m_nBytesRead; + if (num > s2) + num = s2; + memcpy (enc, buf, num); + pkt->m_nBytesRead += num; + s2 -= num; + buf += num; + if (pkt->m_nBytesRead == pkt->m_nBodySize) { + ret = RTMP_SendPacket (r, pkt, false); + RTMPPacket_Free (pkt); + pkt->m_nBytesRead = 0; + if (!ret) + return -1; + buf += 4; + s2 -= 4; + if (s2 < 0) + break; + } + } + return size + s2; +} diff --git a/gst/rtmp/rtmp.h b/gst/rtmp/rtmp.h new file mode 100644 index 0000000000..c593cb7bb0 --- /dev/null +++ b/gst/rtmp/rtmp.h @@ -0,0 +1,336 @@ +#ifndef __RTMP_H__ +#define __RTMP_H__ +/* + * Copyright (C) 2005-2008 Team XBMC + * http://www.xbmc.org + * Copyright (C) 2008-2009 Andrej Stepanchuk + * Copyright (C) 2009-2010 Howard Chu + * + * This file is part of librtmp. + * + * librtmp is free software; you can redistribute it and/or modify + * it under the terms of the GNU Lesser General Public License as + * published by the Free Software Foundation; either version 2.1, + * or (at your option) any later version. + * + * librtmp 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 Lesser General Public License + * along with librtmp see the file COPYING. If not, write to + * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. + * http://www.gnu.org/copyleft/lgpl.html + */ + +#if !defined(NO_CRYPTO) && !defined(CRYPTO) +#define CRYPTO +#endif + +#include +#include +#include + +#include "amf.h" + +#ifdef __cplusplus +extern "C" +{ +#endif + +#define RTMP_LIB_VERSION 0x020205 /* 2.2e */ + +#define RTMP_FEATURE_HTTP 0x01 +#define RTMP_FEATURE_ENC 0x02 +#define RTMP_FEATURE_SSL 0x04 +#define RTMP_FEATURE_MFP 0x08 /* not yet supported */ +#define RTMP_FEATURE_WRITE 0x10 /* publish, not play */ +#define RTMP_FEATURE_HTTP2 0x20 /* server-side rtmpt */ + +#define RTMP_PROTOCOL_UNDEFINED -1 +#define RTMP_PROTOCOL_RTMP 0 +#define RTMP_PROTOCOL_RTMPE RTMP_FEATURE_ENC +#define RTMP_PROTOCOL_RTMPT RTMP_FEATURE_HTTP +#define RTMP_PROTOCOL_RTMPS RTMP_FEATURE_SSL +#define RTMP_PROTOCOL_RTMPTE (RTMP_FEATURE_HTTP|RTMP_FEATURE_ENC) +#define RTMP_PROTOCOL_RTMPTS (RTMP_FEATURE_HTTP|RTMP_FEATURE_SSL) +#define RTMP_PROTOCOL_RTMFP RTMP_FEATURE_MFP + +#define RTMP_DEFAULT_CHUNKSIZE 128 + +/* needs to fit largest number of bytes recv() may return */ +#define RTMP_BUFFER_CACHE_SIZE (16*1024) + +#define RTMP_CHANNELS 65600 + + extern const char RTMPProtocolStringsLower[][7]; + extern const AVal RTMP_DefaultFlashVer; + extern bool RTMP_ctrlC; + + uint32_t RTMP_GetTime(void); + +#define RTMP_PACKET_TYPE_AUDIO 0x08 +#define RTMP_PACKET_TYPE_VIDEO 0x09 +#define RTMP_PACKET_TYPE_INFO 0x12 + +#define RTMP_MAX_HEADER_SIZE 18 + +#define RTMP_PACKET_SIZE_LARGE 0 +#define RTMP_PACKET_SIZE_MEDIUM 1 +#define RTMP_PACKET_SIZE_SMALL 2 +#define RTMP_PACKET_SIZE_MINIMUM 3 + + typedef struct RTMPChunk + { + int c_headerSize; + int c_chunkSize; + char *c_chunk; + char c_header[RTMP_MAX_HEADER_SIZE]; + } RTMPChunk; + + typedef struct RTMPPacket + { + uint8_t m_headerType; + uint8_t m_packetType; + uint8_t m_hasAbsTimestamp; /* timestamp absolute or relative? */ + int m_nChannel; + uint32_t m_nTimeStamp; /* timestamp */ + int32_t m_nInfoField2; /* last 4 bytes in a long header */ + uint32_t m_nBodySize; + uint32_t m_nBytesRead; + RTMPChunk *m_chunk; + char *m_body; + } RTMPPacket; + + typedef struct RTMPSockBuf + { + int sb_socket; + int sb_size; /* number of unprocessed bytes in buffer */ + char *sb_start; /* pointer into sb_pBuffer of next byte to process */ + char sb_buf[RTMP_BUFFER_CACHE_SIZE]; /* data read from socket */ + bool sb_timedout; + void *sb_ssl; + } RTMPSockBuf; + + void RTMPPacket_Reset(RTMPPacket *p); + void RTMPPacket_Dump(RTMPPacket *p); + bool RTMPPacket_Alloc(RTMPPacket *p, int nSize); + void RTMPPacket_Free(RTMPPacket *p); + +#define RTMPPacket_IsReady(a) ((a)->m_nBytesRead == (a)->m_nBodySize) + + typedef struct RTMP_LNK + { + AVal hostname; + AVal sockshost; + + AVal playpath0; /* parsed from URL */ + AVal playpath; /* passed in explicitly */ + AVal tcUrl; + AVal swfUrl; + AVal pageUrl; + AVal app; + AVal auth; + AVal flashVer; + AVal subscribepath; + AVal token; + AMFObject extras; + int edepth; + + int seekTime; + int stopTime; + +#define RTMP_LF_AUTH 0x0001 /* using auth param */ +#define RTMP_LF_LIVE 0x0002 /* stream is live */ +#define RTMP_LF_SWFV 0x0004 /* do SWF verification */ +#define RTMP_LF_PLST 0x0008 /* send playlist before play */ +#define RTMP_LF_BUFX 0x0010 /* toggle stream on BufferEmpty msg */ + int lFlags; + + int swfAge; + + int protocol; + int timeout; /* connection timeout in seconds */ + + unsigned short socksport; + unsigned short port; + +#ifdef CRYPTO +#define RTMP_SWF_HASHLEN 32 + void *dh; /* for encryption */ + void *rc4keyIn; + void *rc4keyOut; + + uint32_t SWFSize; + uint8_t SWFHash[RTMP_SWF_HASHLEN]; + char SWFVerificationResponse[RTMP_SWF_HASHLEN+10]; +#endif + } RTMP_LNK; + + /* state for read() wrapper */ + typedef struct RTMP_READ + { + char *buf; + char *bufpos; + unsigned int buflen; + uint32_t timestamp; + uint8_t dataType; + uint8_t flags; +#define RTMP_READ_HEADER 0x01 +#define RTMP_READ_RESUME 0x02 +#define RTMP_READ_NO_IGNORE 0x04 +#define RTMP_READ_GOTKF 0x08 +#define RTMP_READ_GOTFLVK 0x10 +#define RTMP_READ_SEEKING 0x20 + int8_t status; +#define RTMP_READ_COMPLETE -3 +#define RTMP_READ_ERROR -2 +#define RTMP_READ_EOF -1 +#define RTMP_READ_IGNORE 0 + + /* if bResume == TRUE */ + uint8_t initialFrameType; + uint32_t nResumeTS; + char *metaHeader; + char *initialFrame; + uint32_t nMetaHeaderSize; + uint32_t nInitialFrameSize; + uint32_t nIgnoredFrameCounter; + uint32_t nIgnoredFlvFrameCounter; + } RTMP_READ; + + typedef struct RTMP_METHOD + { + AVal name; + int num; + } RTMP_METHOD; + + typedef struct RTMP + { + int m_inChunkSize; + int m_outChunkSize; + int m_nBWCheckCounter; + int m_nBytesIn; + int m_nBytesInSent; + int m_nBufferMS; + int m_stream_id; /* returned in _result from createStream */ + int m_mediaChannel; + uint32_t m_mediaStamp; + uint32_t m_pauseStamp; + int m_pausing; + int m_nServerBW; + int m_nClientBW; + uint8_t m_nClientBW2; + uint8_t m_bPlaying; + uint8_t m_bSendEncoding; + uint8_t m_bSendCounter; + + int m_numInvokes; + int m_numCalls; + RTMP_METHOD *m_methodCalls; /* remote method calls queue */ + + RTMPPacket *m_vecChannelsIn[RTMP_CHANNELS]; + RTMPPacket *m_vecChannelsOut[RTMP_CHANNELS]; + int m_channelTimestamp[RTMP_CHANNELS]; /* abs timestamp of last packet */ + + double m_fAudioCodecs; /* audioCodecs for the connect packet */ + double m_fVideoCodecs; /* videoCodecs for the connect packet */ + double m_fEncoding; /* AMF0 or AMF3 */ + + double m_fDuration; /* duration of stream in seconds */ + + int m_msgCounter; /* RTMPT stuff */ + int m_polling; + int m_resplen; + int m_unackd; + AVal m_clientID; + + RTMP_READ m_read; + RTMPPacket m_write; + RTMPSockBuf m_sb; + RTMP_LNK Link; + } RTMP; + + bool RTMP_ParseURL(const char *url, int *protocol, AVal *host, + unsigned int *port, AVal *playpath, AVal *app); + + void RTMP_ParsePlaypath(AVal *in, AVal *out); + void RTMP_SetBufferMS(RTMP *r, int size); + void RTMP_UpdateBufferMS(RTMP *r); + + bool RTMP_SetOpt(RTMP *r, const AVal *opt, AVal *arg); + bool RTMP_SetupURL(RTMP *r, char *url); + void RTMP_SetupStream(RTMP *r, int protocol, + AVal *hostname, + unsigned int port, + AVal *sockshost, + AVal *playpath, + AVal *tcUrl, + AVal *swfUrl, + AVal *pageUrl, + AVal *app, + AVal *auth, + AVal *swfSHA256Hash, + uint32_t swfSize, + AVal *flashVer, + AVal *subscribepath, + int dStart, + int dStop, bool bLiveStream, long int timeout); + + bool RTMP_Connect(RTMP *r, RTMPPacket *cp); + struct sockaddr; + bool RTMP_Connect0(RTMP *r, struct sockaddr *svc); + bool RTMP_Connect1(RTMP *r, RTMPPacket *cp); + bool RTMP_Serve(RTMP *r); + + bool RTMP_ReadPacket(RTMP *r, RTMPPacket *packet); + bool RTMP_SendPacket(RTMP *r, RTMPPacket *packet, bool queue); + bool RTMP_SendChunk(RTMP *r, RTMPChunk *chunk); + bool RTMP_IsConnected(RTMP *r); + bool RTMP_IsTimedout(RTMP *r); + double RTMP_GetDuration(RTMP *r); + bool RTMP_ToggleStream(RTMP *r); + + bool RTMP_ConnectStream(RTMP *r, int seekTime); + bool RTMP_ReconnectStream(RTMP *r, int seekTime); + void RTMP_DeleteStream(RTMP *r); + int RTMP_GetNextMediaPacket(RTMP *r, RTMPPacket *packet); + int RTMP_ClientPacket(RTMP *r, RTMPPacket *packet); + + void RTMP_Init(RTMP *r); + void RTMP_Close(RTMP *r); + RTMP *RTMP_Alloc(void); + void RTMP_Free(RTMP *r); + void RTMP_EnableWrite(RTMP *r); + + int RTMP_LibVersion(void); + void RTMP_UserInterrupt(void); /* user typed Ctrl-C */ + + bool RTMP_SendCtrl(RTMP *r, short nType, unsigned int nObject, + unsigned int nTime); + bool RTMP_SendPause(RTMP *r, bool DoPause, int dTime); + bool RTMP_FindFirstMatchingProperty(AMFObject *obj, const AVal *name, + AMFObjectProperty * p); + + int RTMPSockBuf_Fill(RTMPSockBuf *sb); + int RTMPSockBuf_Send(RTMPSockBuf *sb, const char *buf, int len); + int RTMPSockBuf_Close(RTMPSockBuf *sb); + + bool RTMP_SendCreateStream(RTMP *r); + bool RTMP_SendSeek(RTMP *r, int dTime); + bool RTMP_SendServerBW(RTMP *r); + bool RTMP_SendClientBW(RTMP *r); + void RTMP_DropRequest(RTMP *r, int i, bool freeit); + int RTMP_Read(RTMP *r, char *buf, int size); + int RTMP_Write(RTMP *r, char *buf, int size); + +/* hashswf.c */ + int RTMP_HashSWF(const char *url, unsigned int *size, unsigned char *hash, + int age); + +#ifdef __cplusplus +}; +#endif + +#endif diff --git a/gst/rtmp/rtmp_sys.h b/gst/rtmp/rtmp_sys.h new file mode 100644 index 0000000000..155062afa4 --- /dev/null +++ b/gst/rtmp/rtmp_sys.h @@ -0,0 +1,111 @@ +#ifndef __RTMP_SYS_H__ +#define __RTMP_SYS_H__ +/* + * Copyright (C) 2010 Howard Chu + * + * This file is part of librtmp. + * + * librtmp is free software; you can redistribute it and/or modify + * it under the terms of the GNU Lesser General Public License as + * published by the Free Software Foundation; either version 2.1, + * or (at your option) any later version. + * + * librtmp 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 Lesser General Public License + * along with librtmp see the file COPYING. If not, write to + * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. + * http://www.gnu.org/copyleft/lgpl.html + */ + +#ifdef _WIN32 + +#ifdef _XBOX +#include +#include +#define snprintf _snprintf +#define strcasecmp stricmp +#define strncasecmp strnicmp +#define vsnprintf _vsnprintf + +#else /* !_XBOX */ +#include +#include +#endif + +#define GetSockError() WSAGetLastError() +#define SetSockError(e) WSASetLastError(e) +#define setsockopt(a,b,c,d,e) (setsockopt)(a,b,c,(const char *)d,(int)e) +#define EWOULDBLOCK WSAETIMEDOUT /* we don't use nonblocking, but we do use timeouts */ +#define sleep(n) Sleep(n*1000) +#define msleep(n) Sleep(n) +#define SET_RCVTIMEO(tv,s) int tv = s*1000 +#else /* !_WIN32 */ +#include +#include +#include +#include +#include +#include +#include +#include +#define GetSockError() errno +#define SetSockError(e) errno = e +#undef closesocket +#define closesocket(s) close(s) +#define msleep(n) usleep(n*1000) +#define SET_RCVTIMEO(tv,s) struct timeval tv = {s,0} +#endif + +#include "rtmp.h" + +#ifdef USE_POLARSSL +#include +#include +#include +typedef struct tls_ctx { + havege_state hs; + ssl_session ssn; +} tls_ctx; +#define TLS_CTX tls_ctx * +#define TLS_client(ctx,s) s = malloc(sizeof(ssl_context)); ssl_init(s);\ + ssl_set_endpoint(s, SSL_IS_CLIENT); ssl_set_authmode(s, SSL_VERIFY_NONE);\ + ssl_set_rng(s, havege_rand, &ctx->hs); ssl_set_ciphers(s, ssl_default_ciphers);\ + ssl_set_session(s, 1, 600, &ctx->ssn) +#define TLS_setfd(s,fd) ssl_set_bio(s, net_recv, &fd, net_send, &fd) +#define TLS_connect(s) ssl_handshake(s) +#define TLS_read(s,b,l) ssl_read(s,(unsigned char *)b,l) +#define TLS_write(s,b,l) ssl_write(s,(unsigned char *)b,l) +#define TLS_shutdown(s) ssl_close_notify(s) +#define TLS_close(s) ssl_free(s); free(s) + +#elif defined(USE_GNUTLS) +#include +typedef struct tls_ctx { + gnutls_certificate_credentials_t cred; + gnutls_priority_t prios; +} tls_ctx; +#define TLS_CTX tls_ctx * +#define TLS_client(ctx,s) gnutls_init((gnutls_session_t *)(&s), GNUTLS_CLIENT); gnutls_priority_set(s, ctx->prios); gnutls_credentials_set(s, GNUTLS_CRD_CERTIFICATE, ctx->cred) +#define TLS_setfd(s,fd) gnutls_transport_set_ptr(s, (gnutls_transport_ptr_t)(long)fd) +#define TLS_connect(s) gnutls_handshake(s) +#define TLS_read(s,b,l) gnutls_record_recv(s,b,l) +#define TLS_write(s,b,l) gnutls_record_send(s,b,l) +#define TLS_shutdown(s) gnutls_bye(s, GNUTLS_SHUT_RDWR) +#define TLS_close(s) gnutls_deinit(s) + +#else /* USE_OPENSSL */ +#define TLS_CTX SSL_CTX * +#define TLS_client(ctx,s) s = SSL_new(ctx) +#define TLS_setfd(s,fd) SSL_set_fd(s,fd) +#define TLS_connect(s) SSL_connect(s) +#define TLS_read(s,b,l) SSL_read(s,b,l) +#define TLS_write(s,b,l) SSL_write(s,b,l) +#define TLS_shutdown(s) SSL_shutdown(s) +#define TLS_close(s) SSL_free(s) + +#endif +#endif