enum
{
- STREAMING_ACCEPTING,
- STREAMING_IN_PROGRESS,
- STREAMING_STOPPING,
- STREAMING_STOPPED
+ STREAMING_ACCEPTING,
+ STREAMING_IN_PROGRESS,
+ STREAMING_STOPPING,
+ STREAMING_STOPPED
};
typedef struct
{
- int socket;
- int state;
+ int socket;
+ int state;
} STREAMING_SERVER;
-STREAMING_SERVER *httpServer = 0; // server structure pointer
+STREAMING_SERVER *httpServer = 0; // server structure pointer
STREAMING_SERVER *startStreaming(const char *address, int port);
-void stopStreaming(STREAMING_SERVER *server);
+void stopStreaming(STREAMING_SERVER * server);
typedef struct
{
- uint32_t dSeek; // seek position in resume mode, 0 otherwise
-
- char *hostname;
- int rtmpport;
- int protocol;
- bool bLiveStream; // is it a live stream? then we can't seek/resume
-
- long int timeout; // timeout connection afte 300 seconds
- uint32_t bufferTime;
-
- char *rtmpurl;
- AVal playpath;
- AVal swfUrl;
- AVal tcUrl;
- AVal pageUrl;
- AVal app;
- AVal auth;
- AVal swfHash;
- AVal flashVer;
- AVal subscribepath;
- uint32_t swfSize;
-
- uint32_t dStartOffset;
- uint32_t dStopOffset;
- uint32_t nTimeStamp;
+ uint32_t dSeek; // seek position in resume mode, 0 otherwise
+
+ char *hostname;
+ int rtmpport;
+ int protocol;
+ bool bLiveStream; // is it a live stream? then we can't seek/resume
+
+ long int timeout; // timeout connection afte 300 seconds
+ uint32_t bufferTime;
+
+ char *rtmpurl;
+ AVal playpath;
+ AVal swfUrl;
+ AVal tcUrl;
+ AVal pageUrl;
+ AVal app;
+ AVal auth;
+ AVal swfHash;
+ AVal flashVer;
+ AVal subscribepath;
+ uint32_t swfSize;
+
+ uint32_t dStartOffset;
+ uint32_t dStopOffset;
+ uint32_t nTimeStamp;
} RTMP_REQUEST;
#define STR2AVAL(av,str) av.av_val = str; av.av_len = strlen(av.av_val)
/* this request is formed from the parameters and used to initialize a new request,
* thus it is a default settings list. All settings can be overriden by specifying the
- * parameters in the GET request */
+ * parameters in the GET request. */
RTMP_REQUEST defaultRTMPRequest;
-bool ParseOption(char opt, char *arg, RTMP_REQUEST *req);
+bool ParseOption(char opt, char *arg, RTMP_REQUEST * req);
-char DEFAULT_FLASH_VER[] = "LNX 10,0,22,87";
+char DEFAULT_FLASH_VER[] = "LNX 10,0,22,87";
#ifdef _DEBUG
uint32_t debugTS = 0;
#endif
/* inplace http unescape. This is possible .. strlen(unescaped_string) <= strlen(esacped_string) */
-void http_unescape(char* data)
+void
+http_unescape(char *data)
{
- char hex[3];
- char *stp;
- int src_x=0;
- int dst_x=0;
+ char hex[3];
+ char *stp;
+ int src_x = 0;
+ int dst_x = 0;
- int length = (int)strlen(data);
- hex[2]=0;
+ int length = (int) strlen(data);
+ hex[2] = 0;
- while(src_x<length)
+ while (src_x < length)
{
- if(strncmp(data+src_x,"%",1)==0 && src_x+2<length)
- {
- //
- // Since we encountered a '%' we know this is an escaped character
- //
- hex[0] = data[src_x+1];
- hex[1] = data[src_x+2];
- data[dst_x] = (char)strtol(hex,&stp,16);
- dst_x += 1;
- src_x += 3;
- }
- else if(src_x!=dst_x)
- {
- //
- // This doesn't need to be unescaped. If we didn't unescape anything previously
- // there is no need to copy the string either
- //
- data[dst_x] = data[src_x];
- src_x += 1;
- dst_x += 1;
- }
- else
- {
- //
- // This doesn't need to be unescaped, however we need to copy the string
- //
- src_x += 1;
- dst_x += 1;
- }
+ if (strncmp(data + src_x, "%", 1) == 0 && src_x + 2 < length)
+ {
+ //
+ // Since we encountered a '%' we know this is an escaped character
+ //
+ hex[0] = data[src_x + 1];
+ hex[1] = data[src_x + 2];
+ data[dst_x] = (char) strtol(hex, &stp, 16);
+ dst_x += 1;
+ src_x += 3;
+ }
+ else if (src_x != dst_x)
+ {
+ //
+ // This doesn't need to be unescaped. If we didn't unescape anything previously
+ // there is no need to copy the string either
+ //
+ data[dst_x] = data[src_x];
+ src_x += 1;
+ dst_x += 1;
+ }
+ else
+ {
+ //
+ // This doesn't need to be unescaped, however we need to copy the string
+ //
+ src_x += 1;
+ dst_x += 1;
+ }
}
- data[dst_x]='\0';
+ data[dst_x] = '\0';
}
-int WriteHeader(
- char **buf, // target pointer, maybe preallocated
- unsigned int len // length of buffer if preallocated
- )
+int
+WriteHeader(char **buf, // target pointer, maybe preallocated
+ unsigned int len // length of buffer if preallocated
+ )
{
- char flvHeader[] = { 'F', 'L', 'V', 0x01,
- 0x05, // video + audio, we finalize later if the value is different
- 0x00, 0x00, 0x00, 0x09,
- 0x00, 0x00, 0x00, 0x00 // first prevTagSize=0
- };
-
- unsigned int size = sizeof(flvHeader);
-
- if(size > len) {
- *buf = (char *)realloc(*buf, size);
- if(*buf == 0) {
- Log(LOGERROR, "Couldn't reallocate memory!");
- return -1; // fatal error
- }
- }
- memcpy(*buf, flvHeader, sizeof(flvHeader));
- return size;
+ char flvHeader[] = { 'F', 'L', 'V', 0x01,
+ 0x05, // video + audio, we finalize later if the value is different
+ 0x00, 0x00, 0x00, 0x09,
+ 0x00, 0x00, 0x00, 0x00 // first prevTagSize=0
+ };
+
+ unsigned int size = sizeof(flvHeader);
+
+ if (size > len)
+ {
+ *buf = (char *) realloc(*buf, size);
+ if (*buf == 0)
+ {
+ Log(LOGERROR, "Couldn't reallocate memory!");
+ return -1; // fatal error
+ }
+ }
+ memcpy(*buf, flvHeader, sizeof(flvHeader));
+ return size;
}
-int WriteStream(
- RTMP *rtmp,
- char **buf, // target pointer, maybe preallocated
- unsigned int len, // length of buffer if preallocated
- uint32_t *nTimeStamp
- )
+int
+WriteStream(RTMP * rtmp, char **buf, // target pointer, maybe preallocated
+ unsigned int len, // length of buffer if preallocated
+ uint32_t * nTimeStamp)
{
- uint32_t prevTagSize = 0;
- int rtnGetNextMediaPacket = 0, ret = -1;
- RTMPPacket packet = {0};
+ uint32_t prevTagSize = 0;
+ int rtnGetNextMediaPacket = 0, ret = -1;
+ RTMPPacket packet = { 0 };
- rtnGetNextMediaPacket = RTMP_GetNextMediaPacket(rtmp, &packet);
- while(rtnGetNextMediaPacket)
+ rtnGetNextMediaPacket = RTMP_GetNextMediaPacket(rtmp, &packet);
+ while (rtnGetNextMediaPacket)
+ {
+ char *packetBody = packet.m_body;
+ unsigned int nPacketLen = packet.m_nBodySize;
+
+ // skip video info/command packets
+ if (packet.m_packetType == 0x09 &&
+ nPacketLen == 2 && ((*packetBody & 0xf0) == 0x50))
{
- char *packetBody = packet.m_body;
- unsigned int nPacketLen = packet.m_nBodySize;
-
- // skip video info/command packets
- if(packet.m_packetType == 0x09 &&
- nPacketLen == 2 &&
- ((*packetBody & 0xf0) == 0x50)) {
- ret = 0;
- break;
- }
+ ret = 0;
+ break;
+ }
- if(packet.m_packetType == 0x09 && nPacketLen <= 5) {
- Log(LOGWARNING, "ignoring too small video packet: size: %d", nPacketLen);
- ret = 0;
- break;
- }
- if(packet.m_packetType == 0x08 && nPacketLen <= 1) {
- Log(LOGWARNING, "ignoring too small audio packet: size: %d", nPacketLen);
- ret = 0;
- break;
- }
+ if (packet.m_packetType == 0x09 && nPacketLen <= 5)
+ {
+ Log(LOGWARNING, "ignoring too small video packet: size: %d",
+ nPacketLen);
+ ret = 0;
+ break;
+ }
+ if (packet.m_packetType == 0x08 && nPacketLen <= 1)
+ {
+ Log(LOGWARNING, "ignoring too small audio packet: size: %d",
+ nPacketLen);
+ ret = 0;
+ break;
+ }
#ifdef _DEBUG
- Log(LOGDEBUG, "type: %02X, size: %d, TS: %d ms", packet.m_packetType, nPacketLen, packet.m_nTimeStamp);
- if(packet.m_packetType == 0x09)
- Log(LOGDEBUG, "frametype: %02X", (*packetBody & 0xf0));
+ Log(LOGDEBUG, "type: %02X, size: %d, TS: %d ms", packet.m_packetType,
+ nPacketLen, packet.m_nTimeStamp);
+ if (packet.m_packetType == 0x09)
+ Log(LOGDEBUG, "frametype: %02X", (*packetBody & 0xf0));
#endif
- // calculate packet size and reallocate buffer if necessary
- unsigned int 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 > len) { // the extra 4 is for the case of an FLV stream without a last prevTagSize (we need extra 4 bytes to append it)
- *buf = (char *)realloc(*buf, size+4);
- if(*buf == 0) {
- Log(LOGERROR, "Couldn't reallocate memory!");
- ret = -1; // fatal error
- break;
- }
- }
- char *ptr = *buf;
+ // calculate packet size and reallocate buffer if necessary
+ unsigned int 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 > len)
+ { // the extra 4 is for the case of an FLV stream without a last prevTagSize (we need extra 4 bytes to append it)
+ *buf = (char *) realloc(*buf, size + 4);
+ if (*buf == 0)
+ {
+ Log(LOGERROR, "Couldn't reallocate memory!");
+ ret = -1; // fatal error
+ break;
+ }
+ }
+ char *ptr = *buf;
- // 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)
- {
- // set data type
- //*dataType |= (((packet.m_packetType == 0x08)<<2)|(packet.m_packetType == 0x09));
+ // 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)
+ {
+ // set data type
+ //*dataType |= (((packet.m_packetType == 0x08)<<2)|(packet.m_packetType == 0x09));
- (*nTimeStamp) = packet.m_nTimeStamp;
- prevTagSize = 11 + nPacketLen;
+ (*nTimeStamp) = packet.m_nTimeStamp;
+ prevTagSize = 11 + nPacketLen;
- *ptr = packet.m_packetType; ptr++;
- ptr += AMF_EncodeInt24(ptr, nPacketLen);
- ptr += AMF_EncodeInt24(ptr, *nTimeStamp);
- *ptr = (char)(((*nTimeStamp) & 0xFF000000) >> 24); ptr++;
+ *ptr = packet.m_packetType;
+ ptr++;
+ ptr += AMF_EncodeInt24(ptr, nPacketLen);
+ ptr += AMF_EncodeInt24(ptr, *nTimeStamp);
+ *ptr = (char) (((*nTimeStamp) & 0xFF000000) >> 24);
+ ptr++;
- // stream id
- ptr += AMF_EncodeInt24(ptr, 0);
- }
+ // stream id
+ ptr += AMF_EncodeInt24(ptr, 0);
+ }
+
+ memcpy(ptr, packetBody, nPacketLen);
+ unsigned int len = nPacketLen;
+
+ // correct tagSize and obtain timestamp if we have an FLV stream
+ if (packet.m_packetType == 0x16)
+ {
+ unsigned int pos = 0;
+
+ while (pos + 11 < nPacketLen)
+ {
+ uint32_t dataSize = AMF_DecodeInt24(packetBody + pos + 1); // size without header (11) and without prevTagSize (4)
+ *nTimeStamp = AMF_DecodeInt24(packetBody + pos + 4);
+ *nTimeStamp |= (packetBody[pos + 7] << 24);
+
+ // set data type
+ //*dataType |= (((*(packetBody+pos) == 0x08)<<2)|(*(packetBody+pos) == 0x09));
- memcpy(ptr, packetBody, nPacketLen);
- unsigned int len = nPacketLen;
-
- // correct tagSize and obtain timestamp if we have an FLV stream
- if(packet.m_packetType == 0x16)
+ if (pos + 11 + dataSize + 4 > nPacketLen)
{
- unsigned int pos=0;
-
- while(pos+11 < nPacketLen)
- {
- uint32_t dataSize = AMF_DecodeInt24(packetBody+pos+1); // size without header (11) and without prevTagSize (4)
- *nTimeStamp = AMF_DecodeInt24(packetBody+pos+4);
- *nTimeStamp |= (packetBody[pos+7]<<24);
-
- // set data type
- //*dataType |= (((*(packetBody+pos) == 0x08)<<2)|(*(packetBody+pos) == 0x09));
-
- if(pos+11+dataSize+4 > nPacketLen) {
- if(pos+11+dataSize > nPacketLen) {
- Log(LOGERROR, "Wrong data size (%lu), stream corrupted, aborting!", dataSize);
- ret = -2;
- break;
- }
- Log(LOGWARNING, "No tagSize found, appending!");
-
- // we have to append a last tagSize!
- prevTagSize = dataSize+11;
- AMF_EncodeInt32(ptr+pos+11+dataSize, prevTagSize);
- size+=4; len+=4;
- } else {
- prevTagSize = AMF_DecodeInt32(packetBody+pos+11+dataSize);
-
- #ifdef _DEBUG
- Log(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
- Log(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, prevTagSize);
- }
- }
-
- pos += prevTagSize+4;//(11+dataSize+4);
- }
+ if (pos + 11 + dataSize > nPacketLen)
+ {
+ Log(LOGERROR,
+ "Wrong data size (%lu), stream corrupted, aborting!",
+ dataSize);
+ ret = -2;
+ break;
+ }
+ Log(LOGWARNING, "No tagSize found, appending!");
+
+ // we have to append a last tagSize!
+ prevTagSize = dataSize + 11;
+ AMF_EncodeInt32(ptr + pos + 11 + dataSize, prevTagSize);
+ size += 4;
+ len += 4;
}
- ptr += len;
+ else
+ {
+ prevTagSize =
+ AMF_DecodeInt32(packetBody + pos + 11 + dataSize);
+
+#ifdef _DEBUG
+ Log(LOGDEBUG,
+ "FLV Packet: type %02X, dataSize: %lu, tagSize: %lu, timeStamp: %lu ms",
+ (unsigned char) packetBody[pos], dataSize, prevTagSize,
+ *nTimeStamp);
+#endif
- if(packet.m_packetType != 0x16) { // FLV tag packets contain their own prevTagSize
- AMF_EncodeInt32(ptr, prevTagSize);
- //ptr += 4;
+ if (prevTagSize != (dataSize + 11))
+ {
+#ifdef _DEBUG
+ Log(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, prevTagSize);
+ }
}
- // Return 0 if this was completed nicely with invoke message Play.Stop or Play.Complete
- if (rtnGetNextMediaPacket == 2) {
- Log(LOGDEBUG, "Got Play.Complete or Play.Stop from server. Assuming stream is complete");
- ret = 0;
- break;
- }
+ pos += prevTagSize + 4; //(11+dataSize+4);
+ }
+ }
+ ptr += len;
+
+ if (packet.m_packetType != 0x16)
+ { // FLV tag packets contain their own prevTagSize
+ AMF_EncodeInt32(ptr, prevTagSize);
+ //ptr += 4;
+ }
- ret = size;
- break;
+ // Return 0 if this was completed nicely with invoke message Play.Stop or Play.Complete
+ if (rtnGetNextMediaPacket == 2)
+ {
+ Log(LOGDEBUG,
+ "Got Play.Complete or Play.Stop from server. Assuming stream is complete");
+ ret = 0;
+ break;
}
- RTMPPacket_Free(&packet);
- return ret; // no more media packets
+ ret = size;
+ break;
+ }
+
+ RTMPPacket_Free(&packet);
+ return ret; // no more media packets
}
-pthread_t ThreadCreate(void *(*routine)(void *), void *args)
+pthread_t
+ThreadCreate(void *(*routine) (void *), void *args)
{
- pthread_t id = 0;
- pthread_attr_t attributes;
- int ret;
-
- pthread_attr_init(&attributes);
- pthread_attr_setdetachstate(&attributes, PTHREAD_CREATE_DETACHED);
-
- ret = pthread_create(&id, &attributes, (void *(*)(void *))routine, (void *)args);
- if(ret != 0)
- LogPrintf("%s, pthread_create failed with %d\n", __FUNCTION__, ret);
-
- return id;
+ pthread_t id = 0;
+ pthread_attr_t attributes;
+ int ret;
+
+ pthread_attr_init(&attributes);
+ pthread_attr_setdetachstate(&attributes, PTHREAD_CREATE_DETACHED);
+
+ ret =
+ pthread_create(&id, &attributes, (void *(*)(void *)) routine,
+ (void *) args);
+ if (ret != 0)
+ LogPrintf("%s, pthread_create failed with %d\n", __FUNCTION__, ret);
+
+ return id;
}
-void *controlServerThread(void *unused)
+void *
+controlServerThread(void *unused)
{
- char ich;
- while(1)
+ char ich;
+ while (1)
+ {
+ ich = getchar();
+ switch (ich)
{
- ich = getchar();
- switch(ich)
- {
- case 'q':
- LogPrintf("Exiting\n");
- stopStreaming(httpServer);
- exit(0);
- break;
- default:
- LogPrintf("Unknown command \'%c\', ignoring\n", ich);
- }
+ case 'q':
+ LogPrintf("Exiting\n");
+ stopStreaming(httpServer);
+ exit(0);
+ break;
+ default:
+ LogPrintf("Unknown command \'%c\', ignoring\n", ich);
}
- return 0;
+ }
+ return 0;
}
+
/*
ssize_t readHTTPLine(int sockfd, char *buffer, size_t length)
{
}
*/
-void processTCPrequest
-(
- STREAMING_SERVER *server, // server socket and state (our listening socket)
- int sockfd // client connection socket
-)
+void processTCPrequest(STREAMING_SERVER * server, // server socket and state (our listening socket)
+ int sockfd // client connection socket
+ )
{
- char buf[512] = {0}; // answer buffer
- char header[2048] = {0}; // request header
- char *filename = NULL; // GET request: file name //512 not enuf
- char *buffer = NULL; // stream buffer
- char *ptr = NULL; // header pointer
-
- size_t nRead = 0;
-
- char srvhead[] = "\r\nServer:HTTP-RTMP Stream Server \r\nContent-Type: Video/MPEG \r\n\r\n";
-
- server->state = STREAMING_IN_PROGRESS;
-
- RTMP rtmp = {0};
- uint32_t dSeek = 0; // can be used to start from a later point in the stream
-
- // reset RTMP options to defaults specified upon invokation of streams
- RTMP_REQUEST req;
- memcpy(&req, &defaultRTMPRequest, sizeof(RTMP_REQUEST));
-
- // timeout for http requests
- fd_set fds;
- struct timeval tv;
-
- memset(&tv, 0, sizeof(struct timeval));
- tv.tv_sec = 5;
-
- // go through request lines
- //do {
- FD_ZERO(&fds);
- FD_SET(sockfd, &fds);
-
- if(select(sockfd + 1, &fds, NULL, NULL, &tv) <= 0) {
- Log(LOGERROR, "Request timeout/select failed, ignoring request");
- goto quit;
- } else {
- nRead = read(sockfd, header, 2047);
- header[2047]='\0';
-
- Log(LOGDEBUG, "%s: header: %s", __FUNCTION__, header);
-
- if(strstr(header,"Range: bytes=") != 0)
- {
- // TODO check range starts from 0 and asking till the end.
- LogPrintf("%s, Range request not supported\n", __FUNCTION__);
- sprintf(buf,"HTTP/1.0 416 Requested Range Not Satisfiable%s", srvhead);
- send(sockfd, buf, (int) strlen(buf), 0);
- goto quit;
- }
-
- if(strncmp(header,"GET", 3) == 0 && nRead > 4) {
- filename = header+4;
-
- // filter " HTTP/..." from end of request
- char *p = filename;
- while(*p != '\0') {
- if(*p == ' ') {
- *p='\0';
- break;
- }
- p++;
- }
- }
- }
- //} while(!isHTTPRequestEOF(header, nRead));
+ char buf[512] = { 0 }; // answer buffer
+ char header[2048] = { 0 }; // request header
+ char *filename = NULL; // GET request: file name //512 not enuf
+ char *buffer = NULL; // stream buffer
+ char *ptr = NULL; // header pointer
+
+ size_t nRead = 0;
+
+ char srvhead[] =
+ "\r\nServer:HTTP-RTMP Stream Server \r\nContent-Type: Video/MPEG \r\n\r\n";
+
+ server->state = STREAMING_IN_PROGRESS;
+
+ RTMP rtmp = { 0 };
+ uint32_t dSeek = 0; // can be used to start from a later point in the stream
+
+ // reset RTMP options to defaults specified upon invokation of streams
+ RTMP_REQUEST req;
+ memcpy(&req, &defaultRTMPRequest, sizeof(RTMP_REQUEST));
+
+ // timeout for http requests
+ fd_set fds;
+ struct timeval tv;
+
+ memset(&tv, 0, sizeof(struct timeval));
+ tv.tv_sec = 5;
+
+ // go through request lines
+ //do {
+ FD_ZERO(&fds);
+ FD_SET(sockfd, &fds);
+
+ if (select(sockfd + 1, &fds, NULL, NULL, &tv) <= 0)
+ {
+ Log(LOGERROR, "Request timeout/select failed, ignoring request");
+ goto quit;
+ }
+ else
+ {
+ nRead = read(sockfd, header, 2047);
+ header[2047] = '\0';
- // if we got a filename from the GET method
- if(filename != NULL)
+ Log(LOGDEBUG, "%s: header: %s", __FUNCTION__, header);
+
+ if (strstr(header, "Range: bytes=") != 0)
{
- Log(LOGDEBUG, "%s: Request header: %s", __FUNCTION__, filename);
- if(filename[0] == '/') { // if its not empty, is it /?
- ptr = filename+1;
-
- // parse parameters
- if(*ptr == '?')
- {
- ptr++;
- int len = strlen(ptr);
-
- while(len >= 2) {
- char ich = *ptr; ptr++;
- if(*ptr != '=')
- goto filenotfound; // long parameters not (yet) supported
-
- ptr++; len-=2;
-
- // get position of the next '&'
- char *temp;
-
- unsigned int nArgLen = len;
- if((temp=strstr(ptr, "&"))!=0) {
- nArgLen = temp-ptr;
- }
-
- char *arg = (char *)malloc((nArgLen+1)*sizeof(char));
- memcpy(arg, ptr, nArgLen*sizeof(char));
- arg[nArgLen]='\0';
-
- //Log(LOGDEBUG, "%s: unescaping parameter: %s", __FUNCTION__, arg);
- http_unescape(arg);
-
- Log(LOGDEBUG, "%s: parameter: %c, arg: %s", __FUNCTION__, ich, arg);
-
- ptr += nArgLen+1;
- len -= nArgLen+1;
-
- ParseOption(ich, arg, &req);
- }
- }
- }
- else {
- goto filenotfound;
+ // TODO check range starts from 0 and asking till the end.
+ LogPrintf("%s, Range request not supported\n", __FUNCTION__);
+ sprintf(buf, "HTTP/1.0 416 Requested Range Not Satisfiable%s",
+ srvhead);
+ send(sockfd, buf, (int) strlen(buf), 0);
+ goto quit;
+ }
+
+ if (strncmp(header, "GET", 3) == 0 && nRead > 4)
+ {
+ filename = header + 4;
+
+ // filter " HTTP/..." from end of request
+ char *p = filename;
+ while (*p != '\0')
+ {
+ if (*p == ' ')
+ {
+ *p = '\0';
+ break;
}
- } else {
- LogPrintf("%s: No request header received/unsupported method\n", __FUNCTION__);
+ p++;
+ }
}
+ }
+ //} while(!isHTTPRequestEOF(header, nRead));
+
+ // if we got a filename from the GET method
+ if (filename != NULL)
+ {
+ Log(LOGDEBUG, "%s: Request header: %s", __FUNCTION__, filename);
+ if (filename[0] == '/')
+ { // if its not empty, is it /?
+ ptr = filename + 1;
+
+ // parse parameters
+ if (*ptr == '?')
+ {
+ ptr++;
+ int len = strlen(ptr);
+
+ while (len >= 2)
+ {
+ char ich = *ptr;
+ ptr++;
+ if (*ptr != '=')
+ goto filenotfound; // long parameters not (yet) supported
+
+ ptr++;
+ len -= 2;
- // do necessary checks right here to make sure the combined request of default values and GET parameters is correct
- if(req.hostname == 0) {
- Log(LOGERROR, "You must specify a hostname (--host) or url (-r \"rtmp://host[:port]/playpath\") containing a hostname");
- goto filenotfound;
- }
- if(req.playpath.av_len == 0) {
- Log(LOGERROR, "You must specify a playpath (--playpath) or url (-r \"rtmp://host[:port]/playpath\") containing a playpath");
- goto filenotfound;;
- }
-
- if(req.rtmpport == -1) {
- Log(LOGWARNING, "You haven't specified a port (--port) or rtmp url (-r), using default port 1935");
- req.rtmpport = 1935;
- }
- if(req.protocol == RTMP_PROTOCOL_UNDEFINED) {
- Log(LOGWARNING, "You haven't specified a protocol (--protocol) or rtmp url (-r), using default protocol RTMP");
- req.protocol = RTMP_PROTOCOL_RTMP;
- }
-
- if(req.flashVer.av_len == 0) {
- STR2AVAL(req.flashVer, DEFAULT_FLASH_VER);
+ // get position of the next '&'
+ char *temp;
+
+ unsigned int nArgLen = len;
+ if ((temp = strstr(ptr, "&")) != 0)
+ {
+ nArgLen = temp - ptr;
+ }
+
+ char *arg = (char *) malloc((nArgLen + 1) * sizeof(char));
+ memcpy(arg, ptr, nArgLen * sizeof(char));
+ arg[nArgLen] = '\0';
+
+ //Log(LOGDEBUG, "%s: unescaping parameter: %s", __FUNCTION__, arg);
+ http_unescape(arg);
+
+ Log(LOGDEBUG, "%s: parameter: %c, arg: %s", __FUNCTION__,
+ ich, arg);
+
+ ptr += nArgLen + 1;
+ len -= nArgLen + 1;
+
+ ParseOption(ich, arg, &req);
+ }
+ }
+ }
+ else
+ {
+ goto filenotfound;
}
+ }
+ else
+ {
+ LogPrintf("%s: No request header received/unsupported method\n",
+ __FUNCTION__);
+ }
+
+ // do necessary checks right here to make sure the combined request of default values and GET parameters is correct
+ if (req.hostname == 0)
+ {
+ Log(LOGERROR,
+ "You must specify a hostname (--host) or url (-r \"rtmp://host[:port]/playpath\") containing a hostname");
+ goto filenotfound;
+ }
+ if (req.playpath.av_len == 0)
+ {
+ Log(LOGERROR,
+ "You must specify a playpath (--playpath) or url (-r \"rtmp://host[:port]/playpath\") containing a playpath");
+ goto filenotfound;;
+ }
+
+ if (req.rtmpport == -1)
+ {
+ Log(LOGWARNING,
+ "You haven't specified a port (--port) or rtmp url (-r), using default port 1935");
+ req.rtmpport = 1935;
+ }
+ if (req.protocol == RTMP_PROTOCOL_UNDEFINED)
+ {
+ Log(LOGWARNING,
+ "You haven't specified a protocol (--protocol) or rtmp url (-r), using default protocol RTMP");
+ req.protocol = RTMP_PROTOCOL_RTMP;
+ }
+
+ if (req.flashVer.av_len == 0)
+ {
+ STR2AVAL(req.flashVer, DEFAULT_FLASH_VER);
+ }
+
+ if (req.tcUrl.av_len == 0 && req.app.av_len != 0)
+ {
+ char str[512] = { 0 };
+ snprintf(str, 511, "%s://%s/%s", RTMPProtocolStringsLower[req.protocol],
+ req.hostname, req.app.av_val);
+ req.tcUrl.av_len = strlen(str);
+ req.tcUrl.av_val = (char *) malloc(req.tcUrl.av_len + 1);
+ strcpy(req.tcUrl.av_val, str);
+ }
+
+ if (req.rtmpport == 0)
+ req.rtmpport = 1935;
- if(req.tcUrl.av_len == 0 && req.app.av_len != 0) {
- char str[512]={0};
- snprintf(str, 511, "%s://%s/%s", RTMPProtocolStringsLower[req.protocol], req.hostname, req.app.av_val);
- req.tcUrl.av_len = strlen(str);
- req.tcUrl.av_val = (char *)malloc(req.tcUrl.av_len + 1);
- strcpy(req.tcUrl.av_val, str);
- }
-
- if(req.rtmpport == 0)
- req.rtmpport = 1935;
-
- // after validation of the http request send response header
- sprintf(buf,"HTTP/1.0 200 OK%s", srvhead);
- send(sockfd, buf, (int)strlen(buf), 0);
-
- // send the packets
- buffer = (char *)calloc(PACKET_SIZE,1);
-
- // User defined seek offset
- if (req.dStartOffset > 0) {
- if (req.bLiveStream)
- Log(LOGWARNING, "Can't seek in a live stream, ignoring --seek option");
- else
- dSeek += req.dStartOffset;
+ // after validation of the http request send response header
+ sprintf(buf, "HTTP/1.0 200 OK%s", srvhead);
+ send(sockfd, buf, (int) strlen(buf), 0);
+
+ // send the packets
+ buffer = (char *) calloc(PACKET_SIZE, 1);
+
+ // User defined seek offset
+ if (req.dStartOffset > 0)
+ {
+ if (req.bLiveStream)
+ Log(LOGWARNING,
+ "Can't seek in a live stream, ignoring --seek option");
+ else
+ dSeek += req.dStartOffset;
+ }
+
+ if (dSeek != 0)
+ {
+ LogPrintf("Starting at TS: %d ms\n", req.nTimeStamp);
+ }
+
+ Log(LOGDEBUG, "Setting buffer time to: %dms", req.bufferTime);
+ RTMP_Init(&rtmp);
+ RTMP_SetBufferMS(&rtmp, req.bufferTime);
+ RTMP_SetupStream(&rtmp, req.protocol, req.hostname, req.rtmpport, NULL, // sockshost
+ &req.playpath, &req.tcUrl, &req.swfUrl, &req.pageUrl, &req.app, &req.auth, &req.swfHash, req.swfSize, &req.flashVer, &req.subscribepath, dSeek, -1, // length
+ req.bLiveStream, req.timeout);
+
+ LogPrintf("Connecting ... port: %d, app: %s\n", req.rtmpport, req.app);
+ if (!RTMP_Connect(&rtmp))
+ {
+ LogPrintf("%s, failed to connect!\n", __FUNCTION__);
+ }
+ else
+ {
+ unsigned long size = 0;
+ double percent = 0;
+ double duration = 0.0;
+
+ int nWritten = 0;
+ int nRead = 0;
+
+ // write FLV header first
+ nRead = WriteHeader(&buffer, PACKET_SIZE);
+ if (nRead > 0)
+ {
+ nWritten = send(sockfd, buffer, nRead, 0);
+ if (nWritten < 0)
+ {
+ Log(LOGERROR, "%s, sending failed, error: %d", __FUNCTION__,
+ GetSockError());
+ goto cleanup; // we are in STREAMING_IN_PROGRESS, so we'll go to STREAMING_ACCEPTING
+ }
+
+ size += nRead;
+ }
+ else
+ {
+ Log(LOGERROR, "%s: Couldn't obtain FLV header, exiting!",
+ __FUNCTION__);
+ goto cleanup;
}
- if(dSeek != 0) {
- LogPrintf("Starting at TS: %d ms\n", req.nTimeStamp);
- }
-
- Log(LOGDEBUG, "Setting buffer time to: %dms", req.bufferTime);
- RTMP_Init(&rtmp);
- RTMP_SetBufferMS(&rtmp, req.bufferTime);
- RTMP_SetupStream(&rtmp,
- req.protocol,
- req.hostname,
- req.rtmpport,
- NULL, // sockshost
- &req.playpath,
- &req.tcUrl,
- &req.swfUrl,
- &req.pageUrl,
- &req.app,
- &req.auth,
- &req.swfHash,
- req.swfSize,
- &req.flashVer,
- &req.subscribepath,
- dSeek,
- -1, // length
- req.bLiveStream,
- req.timeout);
-
- LogPrintf("Connecting ... port: %d, app: %s\n", req.rtmpport, req.app);
- if (!RTMP_Connect(&rtmp)) {
- LogPrintf("%s, failed to connect!\n", __FUNCTION__);
- }
- else
+ // get the rest of the stream
+ do
{
- unsigned long size = 0;
- double percent = 0;
- double duration = 0.0;
-
- int nWritten = 0;
- int nRead = 0;
-
- // write FLV header first
- nRead = WriteHeader(&buffer, PACKET_SIZE);
- if(nRead > 0) {
- nWritten = send(sockfd, buffer, nRead, 0);
- if(nWritten < 0) {
- Log(LOGERROR, "%s, sending failed, error: %d", __FUNCTION__, GetSockError());
- goto cleanup; // we are in STREAMING_IN_PROGRESS, so we'll go to STREAMING_ACCEPTING
- }
-
- size += nRead;
- } else {
- Log(LOGERROR, "%s: Couldn't obtain FLV header, exiting!", __FUNCTION__);
- goto cleanup;
+ nRead = WriteStream(&rtmp, &buffer, PACKET_SIZE, &req.nTimeStamp);
+
+ if (nRead > 0)
+ {
+ nWritten = send(sockfd, buffer, nRead, 0);
+ //Log(LOGDEBUG, "written: %d", nWritten);
+ if (nWritten < 0)
+ {
+ Log(LOGERROR, "%s, sending failed, error: %d", __FUNCTION__,
+ GetSockError());
+ goto cleanup; // we are in STREAMING_IN_PROGRESS, so we'll go to STREAMING_ACCEPTING
}
- // get the rest of the stream
- do
+ size += nRead;
+
+ //LogPrintf("write %dbytes (%.1f KB)\n", nRead, nRead/1024.0);
+ if (duration <= 0) // if duration unknown try to get it from the stream (onMetaData)
+ duration = RTMP_GetDuration(&rtmp);
+
+ if (duration > 0)
{
- nRead = WriteStream(&rtmp, &buffer, PACKET_SIZE, &req.nTimeStamp);
-
- if(nRead > 0)
- {
- nWritten = send(sockfd, buffer, nRead, 0);
- //Log(LOGDEBUG, "written: %d", nWritten);
- if(nWritten < 0) {
- Log(LOGERROR, "%s, sending failed, error: %d", __FUNCTION__, GetSockError());
- goto cleanup; // we are in STREAMING_IN_PROGRESS, so we'll go to STREAMING_ACCEPTING
- }
-
- size += nRead;
-
- //LogPrintf("write %dbytes (%.1f KB)\n", nRead, nRead/1024.0);
- if(duration <= 0) // if duration unknown try to get it from the stream (onMetaData)
- duration = RTMP_GetDuration(&rtmp);
-
- if(duration > 0) {
- percent = ((double)(dSeek+req.nTimeStamp)) / (duration*1000.0)*100.0;
- percent = ((double)(int)(percent*10.0))/10.0;
- LogPrintf("\r%.3f KB / %.2f sec (%.1f%%)", (double)size/1024.0, (double)(req.nTimeStamp)/1000.0, percent);
- } else {
- LogPrintf("\r%.3f KB / %.2f sec", (double)size/1024.0, (double)(req.nTimeStamp)/1000.0);
- }
- }
- #ifdef _DEBUG
- else { Log(LOGDEBUG, "zero read!"); }
- #endif
-
- // Force clean close if a specified stop offset is reached
- if (req.dStopOffset && req.nTimeStamp >= req.dStopOffset) {
- LogPrintf("\nStop offset has been reached at %.2f seconds\n", (double)req.dStopOffset/1000.0);
- nRead = 0;
- RTMP_Close(&rtmp);
- }
-
- } while(server->state == STREAMING_IN_PROGRESS && nRead > -1 && RTMP_IsConnected(&rtmp) && nWritten >= 0);
+ percent =
+ ((double) (dSeek + req.nTimeStamp)) / (duration *
+ 1000.0) * 100.0;
+ percent = ((double) (int) (percent * 10.0)) / 10.0;
+ LogPrintf("\r%.3f KB / %.2f sec (%.1f%%)",
+ (double) size / 1024.0,
+ (double) (req.nTimeStamp) / 1000.0, percent);
+ }
+ else
+ {
+ LogPrintf("\r%.3f KB / %.2f sec", (double) size / 1024.0,
+ (double) (req.nTimeStamp) / 1000.0);
+ }
+ }
+#ifdef _DEBUG
+ else
+ {
+ Log(LOGDEBUG, "zero read!");
+ }
+#endif
+
+ // Force clean close if a specified stop offset is reached
+ if (req.dStopOffset && req.nTimeStamp >= req.dStopOffset)
+ {
+ LogPrintf("\nStop offset has been reached at %.2f seconds\n",
+ (double) req.dStopOffset / 1000.0);
+ nRead = 0;
+ RTMP_Close(&rtmp);
+ }
+
}
+ while (server->state == STREAMING_IN_PROGRESS && nRead > -1
+ && RTMP_IsConnected(&rtmp) && nWritten >= 0);
+ }
cleanup:
- LogPrintf("Closing connection... ");
- RTMP_Close(&rtmp);
- LogPrintf("done!\n\n");
+ LogPrintf("Closing connection... ");
+ RTMP_Close(&rtmp);
+ LogPrintf("done!\n\n");
quit:
- if(buffer) {
- free(buffer);
- buffer = NULL;
- }
+ if (buffer)
+ {
+ free(buffer);
+ buffer = NULL;
+ }
- if(sockfd)
- close(sockfd);
+ if (sockfd)
+ close(sockfd);
- if(server->state == STREAMING_IN_PROGRESS)
- server->state = STREAMING_ACCEPTING;
+ if (server->state == STREAMING_IN_PROGRESS)
+ server->state = STREAMING_ACCEPTING;
- return;
+ return;
filenotfound:
- LogPrintf("%s, File not found, %s\n", __FUNCTION__, filename);
- sprintf(buf,"HTTP/1.0 404 File Not Found%s", srvhead);
- send(sockfd, buf, (int) strlen(buf), 0);
- goto quit;
+ LogPrintf("%s, File not found, %s\n", __FUNCTION__, filename);
+ sprintf(buf, "HTTP/1.0 404 File Not Found%s", srvhead);
+ send(sockfd, buf, (int) strlen(buf), 0);
+ goto quit;
}
-void *serverThread(STREAMING_SERVER *server)
+void *
+serverThread(STREAMING_SERVER * server)
{
- server->state = STREAMING_ACCEPTING;
+ server->state = STREAMING_ACCEPTING;
- while(server->state == STREAMING_ACCEPTING)
+ while (server->state == STREAMING_ACCEPTING)
+ {
+ struct sockaddr_in addr;
+ socklen_t addrlen = sizeof(struct sockaddr_in);
+ int sockfd =
+ accept(server->socket, (struct sockaddr *) &addr, &addrlen);
+
+ if (sockfd > 0)
{
- struct sockaddr_in addr;
- socklen_t addrlen = sizeof(struct sockaddr_in);
- int sockfd = accept(server->socket, (struct sockaddr *)&addr, &addrlen);
-
- if(sockfd > 0)
- {
- // Create a new process and transfer the control to that
- Log(LOGDEBUG, "%s: accepted connection from %s\n", __FUNCTION__, inet_ntoa(addr.sin_addr));
- processTCPrequest(server, sockfd);
- Log(LOGDEBUG, "%s: processed request\n", __FUNCTION__);
- }
- else {
- Log(LOGERROR, "%s: accept failed", __FUNCTION__);
- }
+ // Create a new process and transfer the control to that
+ Log(LOGDEBUG, "%s: accepted connection from %s\n", __FUNCTION__,
+ inet_ntoa(addr.sin_addr));
+ processTCPrequest(server, sockfd);
+ Log(LOGDEBUG, "%s: processed request\n", __FUNCTION__);
+ }
+ else
+ {
+ Log(LOGERROR, "%s: accept failed", __FUNCTION__);
}
- server->state = STREAMING_STOPPED;
- return 0;
+ }
+ server->state = STREAMING_STOPPED;
+ return 0;
}
-STREAMING_SERVER *startStreaming(const char *address, int port)
+STREAMING_SERVER *
+startStreaming(const char *address, int port)
{
- struct sockaddr_in addr;
- int sockfd;
- STREAMING_SERVER *server;
-
- sockfd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
- if(sockfd == -1) {
- Log(LOGERROR, "%s, couldn't create socket", __FUNCTION__);
- return 0;
- }
+ struct sockaddr_in addr;
+ int sockfd;
+ STREAMING_SERVER *server;
- addr.sin_family = AF_INET;
- addr.sin_addr.s_addr = inet_addr(address);//htonl(INADDR_ANY);
- addr.sin_port = htons(port);
-
- if(bind(sockfd, (struct sockaddr *)&addr, sizeof(struct sockaddr_in)) == -1) {
- Log(LOGERROR, "%s, TCP bind failed for port number: %d", __FUNCTION__, port);
- return 0;
- }
+ sockfd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+ if (sockfd == -1)
+ {
+ Log(LOGERROR, "%s, couldn't create socket", __FUNCTION__);
+ return 0;
+ }
- if(listen(sockfd, 10) == -1)
- {
- Log(LOGERROR, "%s, listen failed", __FUNCTION__);
- close(sockfd);
- return 0;
- }
+ addr.sin_family = AF_INET;
+ addr.sin_addr.s_addr = inet_addr(address); //htonl(INADDR_ANY);
+ addr.sin_port = htons(port);
+
+ if (bind(sockfd, (struct sockaddr *) &addr, sizeof(struct sockaddr_in)) ==
+ -1)
+ {
+ Log(LOGERROR, "%s, TCP bind failed for port number: %d", __FUNCTION__,
+ port);
+ return 0;
+ }
+
+ if (listen(sockfd, 10) == -1)
+ {
+ Log(LOGERROR, "%s, listen failed", __FUNCTION__);
+ close(sockfd);
+ return 0;
+ }
- server = (STREAMING_SERVER *)calloc(1, sizeof(STREAMING_SERVER));
- server->socket = sockfd;
-
- ThreadCreate((void* (*)(void*))serverThread, server);
+ server = (STREAMING_SERVER *) calloc(1, sizeof(STREAMING_SERVER));
+ server->socket = sockfd;
- return server;
+ ThreadCreate((void *(*)(void *)) serverThread, server);
+
+ return server;
}
-void stopStreaming(STREAMING_SERVER *server)
+void
+stopStreaming(STREAMING_SERVER * server)
{
- assert(server);
+ assert(server);
- if(server->state != STREAMING_STOPPED)
+ if (server->state != STREAMING_STOPPED)
+ {
+ if (server->state == STREAMING_IN_PROGRESS)
{
- if(server->state == STREAMING_IN_PROGRESS)
- {
- server->state = STREAMING_STOPPING;
+ server->state = STREAMING_STOPPING;
- // wait for streaming threads to exit
- while(server->state != STREAMING_STOPPED)
- usleep(1*1000);
- }
-
- if(close(server->socket))
- Log(LOGERROR, "%s: Failed to close listening socket, error %d", GetSockError());
-
- server->state = STREAMING_STOPPED;
+ // wait for streaming threads to exit
+ while (server->state != STREAMING_STOPPED)
+ usleep(1 * 1000);
}
+
+ if (close(server->socket))
+ Log(LOGERROR, "%s: Failed to close listening socket, error %d",
+ GetSockError());
+
+ server->state = STREAMING_STOPPED;
+ }
}
-void sigIntHandler(int sig) {
- RTMP_ctrlC = true;
- LogPrintf("Caught signal: %d, cleaning up, just a second...\n", sig);
- if(httpServer)
- stopStreaming(httpServer);
- signal(SIGINT, SIG_DFL);
+void
+sigIntHandler(int sig)
+{
+ RTMP_ctrlC = true;
+ LogPrintf("Caught signal: %d, cleaning up, just a second...\n", sig);
+ if (httpServer)
+ stopStreaming(httpServer);
+ signal(SIGINT, SIG_DFL);
}
// this will parse RTMP related options as needed
// Return values: true (option parsing ok)
// false (option not parsed/invalid)
-bool ParseOption(char opt, char *arg, RTMP_REQUEST *req)
+bool
+ParseOption(char opt, char *arg, RTMP_REQUEST * req)
{
- switch(opt)
- {
- case 'w':
- {
- int res = hex2bin(arg, &req->swfHash.av_val);
- if(!res || res!=32) {
- req->swfHash.av_val = NULL;
- Log(LOGWARNING, "Couldn't parse swf hash hex string, not heyxstring or not 32 bytes, ignoring!");
- }
- req->swfHash.av_len = 32;
- break;
- }
- case 'x':
- {
- int size = atoi(arg);
- if(size <= 0) {
- Log(LOGERROR, "SWF Size must be at least 1, ignoring\n");
- } else {
- req->swfSize = size;
- }
- break;
- }
- case 'b':
- {
- int32_t bt = atol(arg);
- if(bt < 0) {
- Log(LOGERROR, "Buffer time must be greater than zero, ignoring the specified value %d!", bt);
- } else {
- req->bufferTime = bt;
- }
- break;
- }
- case 'v':
- req->bLiveStream = true; // no seeking or resuming possible!
- break;
- case 'd':
- STR2AVAL(req->subscribepath, arg);
- break;
- case 'n':
- req->hostname = arg;
- break;
- case 'c':
- req->rtmpport = atoi(arg);
- break;
- case 'l':
- {
- int protocol = atoi(arg);
- if(protocol != RTMP_PROTOCOL_RTMP && protocol != RTMP_PROTOCOL_RTMPE) {
- Log(LOGERROR, "Unknown protocol specified: %d, using default", protocol);
- return false;
- } else {
- req->protocol = protocol;
- }
- break;
- }
- case 'y':
- STR2AVAL(req->playpath, arg);
- break;
- case 'r':
- {
- req->rtmpurl = arg;
-
- char *parsedHost = 0;
- unsigned int parsedPort = 0;
- char *parsedPlaypath = 0;
- char *parsedApp = 0;
- int parsedProtocol = RTMP_PROTOCOL_UNDEFINED;
-
- if(!ParseUrl(req->rtmpurl, &parsedProtocol, &parsedHost, &parsedPort, &parsedPlaypath, &parsedApp)) {
- Log(LOGWARNING, "Couldn't parse the specified url (%s)!", arg);
- } else {
- if(req->hostname == 0)
- req->hostname = parsedHost;
- if(req->rtmpport == -1)
- req->rtmpport = parsedPort;
- if(req->playpath.av_len == 0 && parsedPlaypath) {
- STR2AVAL(req->playpath, parsedPlaypath);
- }
- if(req->protocol == RTMP_PROTOCOL_UNDEFINED)
- req->protocol = parsedProtocol;
- if(req->app.av_len == 0 && parsedApp) {
- STR2AVAL(req->app, parsedApp);
- }
- }
- break;
- }
- case 's':
- STR2AVAL(req->swfUrl, arg);
- break;
- case 't':
- STR2AVAL(req->tcUrl, arg);
- break;
- case 'p':
- STR2AVAL(req->pageUrl, arg);
- break;
- case 'a':
- STR2AVAL(req->app, arg);
- break;
- case 'f':
- STR2AVAL(req->flashVer, arg);
- break;
- case 'u':
- STR2AVAL(req->auth, arg);
- break;
- case 'm':
- req->timeout = atoi(arg);
- break;
- case 'A':
- req->dStartOffset = atoi(arg)*1000;
- //printf("dStartOffset = %d\n", dStartOffset);
- break;
- case 'B':
- req->dStopOffset = atoi(arg)*1000;
- //printf("dStartOffset = %d\n", dStartOffset);
- break;
- case 'q':
- debuglevel = LOGCRIT;
- break;
- case 'V':
- debuglevel = LOGDEBUG;
- break;
- case 'z':
- debuglevel = LOGALL;
- break;
- default:
- LogPrintf("unknown option: %c, arg: %s\n", opt, arg);
- break;
- }
- return true;
+ switch (opt)
+ {
+ case 'w':
+ {
+ int res = hex2bin(arg, &req->swfHash.av_val);
+ if (!res || res != 32)
+ {
+ req->swfHash.av_val = NULL;
+ Log(LOGWARNING,
+ "Couldn't parse swf hash hex string, not heyxstring or not 32 bytes, ignoring!");
+ }
+ req->swfHash.av_len = 32;
+ break;
+ }
+ case 'x':
+ {
+ int size = atoi(arg);
+ if (size <= 0)
+ {
+ Log(LOGERROR, "SWF Size must be at least 1, ignoring\n");
+ }
+ else
+ {
+ req->swfSize = size;
+ }
+ break;
+ }
+ case 'b':
+ {
+ int32_t bt = atol(arg);
+ if (bt < 0)
+ {
+ Log(LOGERROR,
+ "Buffer time must be greater than zero, ignoring the specified value %d!",
+ bt);
+ }
+ else
+ {
+ req->bufferTime = bt;
+ }
+ break;
+ }
+ case 'v':
+ req->bLiveStream = true; // no seeking or resuming possible!
+ break;
+ case 'd':
+ STR2AVAL(req->subscribepath, arg);
+ break;
+ case 'n':
+ req->hostname = arg;
+ break;
+ case 'c':
+ req->rtmpport = atoi(arg);
+ break;
+ case 'l':
+ {
+ int protocol = atoi(arg);
+ if (protocol != RTMP_PROTOCOL_RTMP && protocol != RTMP_PROTOCOL_RTMPE)
+ {
+ Log(LOGERROR, "Unknown protocol specified: %d, using default",
+ protocol);
+ return false;
+ }
+ else
+ {
+ req->protocol = protocol;
+ }
+ break;
+ }
+ case 'y':
+ STR2AVAL(req->playpath, arg);
+ break;
+ case 'r':
+ {
+ req->rtmpurl = arg;
+
+ char *parsedHost = 0;
+ unsigned int parsedPort = 0;
+ char *parsedPlaypath = 0;
+ char *parsedApp = 0;
+ int parsedProtocol = RTMP_PROTOCOL_UNDEFINED;
+
+ if (!ParseUrl
+ (req->rtmpurl, &parsedProtocol, &parsedHost, &parsedPort,
+ &parsedPlaypath, &parsedApp))
+ {
+ Log(LOGWARNING, "Couldn't parse the specified url (%s)!", arg);
+ }
+ else
+ {
+ if (req->hostname == 0)
+ req->hostname = parsedHost;
+ if (req->rtmpport == -1)
+ req->rtmpport = parsedPort;
+ if (req->playpath.av_len == 0 && parsedPlaypath)
+ {
+ STR2AVAL(req->playpath, parsedPlaypath);
+ }
+ if (req->protocol == RTMP_PROTOCOL_UNDEFINED)
+ req->protocol = parsedProtocol;
+ if (req->app.av_len == 0 && parsedApp)
+ {
+ STR2AVAL(req->app, parsedApp);
+ }
+ }
+ break;
+ }
+ case 's':
+ STR2AVAL(req->swfUrl, arg);
+ break;
+ case 't':
+ STR2AVAL(req->tcUrl, arg);
+ break;
+ case 'p':
+ STR2AVAL(req->pageUrl, arg);
+ break;
+ case 'a':
+ STR2AVAL(req->app, arg);
+ break;
+ case 'f':
+ STR2AVAL(req->flashVer, arg);
+ break;
+ case 'u':
+ STR2AVAL(req->auth, arg);
+ break;
+ case 'm':
+ req->timeout = atoi(arg);
+ break;
+ case 'A':
+ req->dStartOffset = atoi(arg) * 1000;
+ //printf("dStartOffset = %d\n", dStartOffset);
+ break;
+ case 'B':
+ req->dStopOffset = atoi(arg) * 1000;
+ //printf("dStartOffset = %d\n", dStartOffset);
+ break;
+ case 'q':
+ debuglevel = LOGCRIT;
+ break;
+ case 'V':
+ debuglevel = LOGDEBUG;
+ break;
+ case 'z':
+ debuglevel = LOGALL;
+ break;
+ default:
+ LogPrintf("unknown option: %c, arg: %s\n", opt, arg);
+ break;
+ }
+ return true;
}
-int main(int argc, char **argv)
+int
+main(int argc, char **argv)
{
- int nStatus = RD_SUCCESS;
-
- // http streaming server
- char DEFAULT_HTTP_STREAMING_DEVICE[] = "0.0.0.0"; // 0.0.0.0 is any device
-
- char *httpStreamingDevice = DEFAULT_HTTP_STREAMING_DEVICE; // streaming device, default 0.0.0.0
- int nHttpStreamingPort = 80; // port
-
- LogPrintf("HTTP-RTMP Stream Server %s\n", RTMPDUMP_STREAMS_VERSION);
- LogPrintf("(c) 2009 Andrej Stepanchuk, Howard Chu; license: GPL\n\n");
-
- // init request
- memset(&defaultRTMPRequest, 0, sizeof(RTMP_REQUEST));
-
- defaultRTMPRequest.rtmpport = -1;
- defaultRTMPRequest.protocol = RTMP_PROTOCOL_UNDEFINED;
- defaultRTMPRequest.bLiveStream = false; // is it a live stream? then we can't seek/resume
-
- defaultRTMPRequest.timeout = 300; // timeout connection afte 300 seconds
- defaultRTMPRequest.bufferTime = 20*1000;
-
-
- int opt;
- struct option longopts[] = {
- {"help", 0, NULL, 'h'},
- {"host", 1, NULL, 'n'},
- {"port", 1, NULL, 'c'},
- {"protocol",1, NULL, 'l'},
- {"playpath",1, NULL, 'y'},
- {"rtmp", 1, NULL, 'r'},
- {"swfUrl", 1, NULL, 's'},
- {"tcUrl", 1, NULL, 't'},
- {"pageUrl", 1, NULL, 'p'},
- {"app", 1, NULL, 'a'},
- {"swfhash", 1, NULL, 'w'},
- {"swfsize", 1, NULL, 'x'},
- {"auth", 1, NULL, 'u'},
- {"flashVer",1, NULL, 'f'},
- {"live" ,0, NULL, 'v'},
- //{"flv", 1, NULL, 'o'},
- //{"resume", 0, NULL, 'e'},
- {"timeout", 1, NULL, 'm'},
- {"buffer", 1, NULL, 'b'},
- //{"skip", 1, NULL, 'k'},
- {"device", 1, NULL, 'D'},
- {"sport", 1, NULL, 'g'},
- {"subscribe",1,NULL, 'd'},
- {"start", 1, NULL, 'A'},
- {"stop", 1, NULL, 'B'},
- {"debug", 0, NULL, 'z'},
- {"quiet", 0, NULL, 'q'},
- {"verbose", 0, NULL, 'V'},
- {0,0,0,0}
- };
-
- signal(SIGINT, sigIntHandler);
- signal(SIGPIPE, SIG_IGN);
-
- while((opt = getopt_long(argc, argv, "hvqVzr:s:t:p:a:f:u:n:c:l:y:m:d:D:A:B:g:w:x:", longopts, NULL)) != -1) {
- switch(opt) {
- case 'h':
- LogPrintf("\nThis program dumps the media content streamed over rtmp.\n\n");
- LogPrintf("--help|-h Prints this help screen.\n");
- LogPrintf("--rtmp|-r url URL (e.g. rtmp//hotname[:port]/path)\n");
- LogPrintf("--host|-n hostname Overrides the hostname in the rtmp url\n");
- LogPrintf("--port|-c port Overrides the port in the rtmp url\n");
- LogPrintf("--protocol|-l Overrides the protocol in the rtmp url (0 - RTMP, 3 - RTMPE)\n");
- LogPrintf("--playpath|-y Overrides the playpath parsed from rtmp url\n");
- LogPrintf("--swfUrl|-s url URL to player swf file\n");
- LogPrintf("--tcUrl|-t url URL to played stream (default: \"rtmp://host[:port]/app\")\n");
- LogPrintf("--pageUrl|-p url Web URL of played programme\n");
- LogPrintf("--app|-a app Name of player used\n");
- LogPrintf("--swfhash|-w hexstring SHA256 hash of the decompressed SWF file (32 bytes)\n");
- LogPrintf("--swfsize|-x num Size of the decompressed SWF file, required for SWFVerification\n");
- LogPrintf("--auth|-u string Authentication string to be appended to the connect string\n");
- LogPrintf("--flashVer|-f string Flash version string (default: \"%s\")\n", DEFAULT_FLASH_VER);
- LogPrintf("--live|-v Get a live stream, no --resume (seeking) of live strems possible\n");
- LogPrintf("--subscribe|-d string Stream name to subscribe to (otherwise defaults to playpath if live is specifed)\n");
- LogPrintf("--timeout|-m num Timeout connection num seconds (default: %lu)\n", defaultRTMPRequest.timeout);
- LogPrintf("--start|-A num Start at num seconds into stream (not valid when using --live)\n");
- LogPrintf("--stop|-B num Stop at num seconds into stream\n");
- LogPrintf("--buffer|-b Buffer time in milliseconds (default: %lu)\n\n",
- defaultRTMPRequest.bufferTime);
-
- LogPrintf("--device|-D Streaming device ip address (default: %s)\n", DEFAULT_HTTP_STREAMING_DEVICE);
- LogPrintf("--sport|-g Streaming port (default: %d)\n\n", nHttpStreamingPort);
- LogPrintf("--quiet|-q Supresses all command output.\n");
- LogPrintf("--verbose|-x Verbose command output.\n");
- LogPrintf("--debug|-z Debug level command output.\n");
- LogPrintf("If you don't pass parameters for swfUrl, pageUrl, app or auth these propertiews will not be included in the connect ");
- LogPrintf("packet.\n\n");
- return RD_SUCCESS;
- break;
- // streaming server specific options
- case 'D':
- if(inet_addr(optarg) == INADDR_NONE) {
- Log(LOGERROR, "Invalid binding address (requested address %s), ignoring", optarg);
- } else {
- httpStreamingDevice = optarg;
- }
- break;
- case 'g':
- {
- int port = atoi(optarg);
- if(port < 0 || port > 65535) {
- Log(LOGERROR, "Streaming port out of range (requested port %d), ignoring\n", port);
- } else {
- nHttpStreamingPort = port;
- }
- break;
- }
- default:
- //LogPrintf("unknown option: %c\n", opt);
- ParseOption(opt, optarg, &defaultRTMPRequest);
- break;
- }
+ int nStatus = RD_SUCCESS;
+
+ // http streaming server
+ char DEFAULT_HTTP_STREAMING_DEVICE[] = "0.0.0.0"; // 0.0.0.0 is any device
+
+ char *httpStreamingDevice = DEFAULT_HTTP_STREAMING_DEVICE; // streaming device, default 0.0.0.0
+ int nHttpStreamingPort = 80; // port
+
+ LogPrintf("HTTP-RTMP Stream Server %s\n", RTMPDUMP_STREAMS_VERSION);
+ LogPrintf("(c) 2009 Andrej Stepanchuk, Howard Chu; license: GPL\n\n");
+
+ // init request
+ memset(&defaultRTMPRequest, 0, sizeof(RTMP_REQUEST));
+
+ defaultRTMPRequest.rtmpport = -1;
+ defaultRTMPRequest.protocol = RTMP_PROTOCOL_UNDEFINED;
+ defaultRTMPRequest.bLiveStream = false; // is it a live stream? then we can't seek/resume
+
+ defaultRTMPRequest.timeout = 300; // timeout connection afte 300 seconds
+ defaultRTMPRequest.bufferTime = 20 * 1000;
+
+
+ int opt;
+ struct option longopts[] = {
+ {"help", 0, NULL, 'h'},
+ {"host", 1, NULL, 'n'},
+ {"port", 1, NULL, 'c'},
+ {"protocol", 1, NULL, 'l'},
+ {"playpath", 1, NULL, 'y'},
+ {"rtmp", 1, NULL, 'r'},
+ {"swfUrl", 1, NULL, 's'},
+ {"tcUrl", 1, NULL, 't'},
+ {"pageUrl", 1, NULL, 'p'},
+ {"app", 1, NULL, 'a'},
+ {"swfhash", 1, NULL, 'w'},
+ {"swfsize", 1, NULL, 'x'},
+ {"auth", 1, NULL, 'u'},
+ {"flashVer", 1, NULL, 'f'},
+ {"live", 0, NULL, 'v'},
+ //{"flv", 1, NULL, 'o'},
+ //{"resume", 0, NULL, 'e'},
+ {"timeout", 1, NULL, 'm'},
+ {"buffer", 1, NULL, 'b'},
+ //{"skip", 1, NULL, 'k'},
+ {"device", 1, NULL, 'D'},
+ {"sport", 1, NULL, 'g'},
+ {"subscribe", 1, NULL, 'd'},
+ {"start", 1, NULL, 'A'},
+ {"stop", 1, NULL, 'B'},
+ {"debug", 0, NULL, 'z'},
+ {"quiet", 0, NULL, 'q'},
+ {"verbose", 0, NULL, 'V'},
+ {0, 0, 0, 0}
+ };
+
+ signal(SIGINT, sigIntHandler);
+ signal(SIGPIPE, SIG_IGN);
+
+ while ((opt =
+ getopt_long(argc, argv,
+ "hvqVzr:s:t:p:a:f:u:n:c:l:y:m:d:D:A:B:g:w:x:", longopts,
+ NULL)) != -1)
+ {
+ switch (opt)
+ {
+ case 'h':
+ LogPrintf
+ ("\nThis program dumps the media content streamed over rtmp.\n\n");
+ LogPrintf("--help|-h Prints this help screen.\n");
+ LogPrintf
+ ("--rtmp|-r url URL (e.g. rtmp//hotname[:port]/path)\n");
+ LogPrintf
+ ("--host|-n hostname Overrides the hostname in the rtmp url\n");
+ LogPrintf
+ ("--port|-c port Overrides the port in the rtmp url\n");
+ LogPrintf
+ ("--protocol|-l Overrides the protocol in the rtmp url (0 - RTMP, 3 - RTMPE)\n");
+ LogPrintf
+ ("--playpath|-y Overrides the playpath parsed from rtmp url\n");
+ LogPrintf("--swfUrl|-s url URL to player swf file\n");
+ LogPrintf
+ ("--tcUrl|-t url URL to played stream (default: \"rtmp://host[:port]/app\")\n");
+ LogPrintf("--pageUrl|-p url Web URL of played programme\n");
+ LogPrintf("--app|-a app Name of player used\n");
+ LogPrintf
+ ("--swfhash|-w hexstring SHA256 hash of the decompressed SWF file (32 bytes)\n");
+ LogPrintf
+ ("--swfsize|-x num Size of the decompressed SWF file, required for SWFVerification\n");
+ LogPrintf
+ ("--auth|-u string Authentication string to be appended to the connect string\n");
+ LogPrintf
+ ("--flashVer|-f string Flash version string (default: \"%s\")\n",
+ DEFAULT_FLASH_VER);
+ LogPrintf
+ ("--live|-v Get a live stream, no --resume (seeking) of live strems possible\n");
+ LogPrintf
+ ("--subscribe|-d string Stream name to subscribe to (otherwise defaults to playpath if live is specifed)\n");
+ LogPrintf
+ ("--timeout|-m num Timeout connection num seconds (default: %lu)\n",
+ defaultRTMPRequest.timeout);
+ LogPrintf
+ ("--start|-A num Start at num seconds into stream (not valid when using --live)\n");
+ LogPrintf
+ ("--stop|-B num Stop at num seconds into stream\n");
+ LogPrintf
+ ("--buffer|-b Buffer time in milliseconds (default: %lu)\n\n",
+ defaultRTMPRequest.bufferTime);
+
+ LogPrintf
+ ("--device|-D Streaming device ip address (default: %s)\n",
+ DEFAULT_HTTP_STREAMING_DEVICE);
+ LogPrintf
+ ("--sport|-g Streaming port (default: %d)\n\n",
+ nHttpStreamingPort);
+ LogPrintf
+ ("--quiet|-q Supresses all command output.\n");
+ LogPrintf("--verbose|-x Verbose command output.\n");
+ LogPrintf("--debug|-z Debug level command output.\n");
+ LogPrintf
+ ("If you don't pass parameters for swfUrl, pageUrl, app or auth these propertiews will not be included in the connect ");
+ LogPrintf("packet.\n\n");
+ return RD_SUCCESS;
+ break;
+ // streaming server specific options
+ case 'D':
+ if (inet_addr(optarg) == INADDR_NONE)
+ {
+ Log(LOGERROR,
+ "Invalid binding address (requested address %s), ignoring",
+ optarg);
+ }
+ else
+ {
+ httpStreamingDevice = optarg;
+ }
+ break;
+ case 'g':
+ {
+ int port = atoi(optarg);
+ if (port < 0 || port > 65535)
+ {
+ Log(LOGERROR,
+ "Streaming port out of range (requested port %d), ignoring\n",
+ port);
+ }
+ else
+ {
+ nHttpStreamingPort = port;
+ }
+ break;
+ }
+ default:
+ //LogPrintf("unknown option: %c\n", opt);
+ ParseOption(opt, optarg, &defaultRTMPRequest);
+ break;
}
+ }
#ifdef _DEBUG
- netstackdump = fopen("netstackdump", "wb");
- netstackdump_read = fopen("netstackdump_read", "wb");
+ netstackdump = fopen("netstackdump", "wb");
+ netstackdump_read = fopen("netstackdump_read", "wb");
#endif
- //InitSockets();
+ //InitSockets();
- // start text UI
- ThreadCreate(controlServerThread, 0);
+ // start text UI
+ ThreadCreate(controlServerThread, 0);
- // start http streaming
- if((httpServer=startStreaming(httpStreamingDevice, nHttpStreamingPort))==0) {
- Log(LOGERROR, "Failed to start HTTP server, exiting!");
- return RD_FAILED;
- }
- LogPrintf("Streaming on http://%s:%d\n", httpStreamingDevice, nHttpStreamingPort);
+ // start http streaming
+ if ((httpServer =
+ startStreaming(httpStreamingDevice, nHttpStreamingPort)) == 0)
+ {
+ Log(LOGERROR, "Failed to start HTTP server, exiting!");
+ return RD_FAILED;
+ }
+ LogPrintf("Streaming on http://%s:%d\n", httpStreamingDevice,
+ nHttpStreamingPort);
- while(httpServer->state != STREAMING_STOPPED) {sleep(1);}
- Log(LOGDEBUG, "Done, exiting...");
-
- //CleanupSockets();
+ while (httpServer->state != STREAMING_STOPPED)
+ {
+ sleep(1);
+ }
+ Log(LOGDEBUG, "Done, exiting...");
+
+ //CleanupSockets();
#ifdef _DEBUG
- if(netstackdump != 0)
- fclose(netstackdump);
- if(netstackdump_read != 0)
- fclose(netstackdump_read);
+ if (netstackdump != 0)
+ fclose(netstackdump);
+ if (netstackdump_read != 0)
+ fclose(netstackdump_read);
#endif
- return nStatus;
+ return nStatus;
}
-
projects / rtmpdump / commitdiff