150 lines
3.9 KiB
C++
150 lines
3.9 KiB
C++
#define DEBUG
|
|
#include <iostream>
|
|
#include <cstdlib>
|
|
#include <cstdio>
|
|
#include <cmath>
|
|
#include <unistd.h>
|
|
#include <signal.h>
|
|
#include <sys/types.h>
|
|
#include <sys/wait.h>
|
|
#include <sys/epoll.h>
|
|
|
|
//for connection to server
|
|
#include "../sockets/SocketW.h"
|
|
bool ready4data = false;//set to true when streaming starts
|
|
bool inited = false;
|
|
bool stopparsing = false;
|
|
timeval lastrec;
|
|
|
|
int CONN_fd = 0;
|
|
FILE * CONN = 0;
|
|
#include "parsechunks.cpp" //chunkstream parsing
|
|
#include "handshake.cpp" //handshaking
|
|
#include "../util/flv_sock.cpp" //FLV parsing with SocketW
|
|
#include "../util/ddv_socket.cpp" //DDVTech Socket wrapper
|
|
|
|
int main(){
|
|
|
|
int server_socket = DDV_Listen(1935);
|
|
int status;
|
|
while (server_socket > 0){
|
|
waitpid((pid_t)-1, &status, WNOHANG);
|
|
CONN_fd = DDV_Accept(server_socket);
|
|
CONN = fdopen(CONN_fd, "r+");
|
|
pid_t myid = fork();
|
|
if (myid == 0){
|
|
break;
|
|
}else{
|
|
printf("Spawned new process %i for incoming client\n", (int)myid);
|
|
}
|
|
}
|
|
if (server_socket <= 0){
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
unsigned int ts;
|
|
unsigned int fts = 0;
|
|
unsigned int ftst;
|
|
SWUnixSocket ss;
|
|
|
|
//first timestamp set
|
|
firsttime = getNowMS();
|
|
|
|
#ifdef DEBUG
|
|
fprintf(stderr, "Doing handshake...\n");
|
|
#endif
|
|
if (doHandshake()){
|
|
#ifdef DEBUG
|
|
fprintf(stderr, "Handshake succcess!\n");
|
|
#endif
|
|
}else{
|
|
#ifdef DEBUG
|
|
fprintf(stderr, "Handshake fail!\n");
|
|
#endif
|
|
return 0;
|
|
}
|
|
#ifdef DEBUG
|
|
fprintf(stderr, "Starting processing...\n");
|
|
#endif
|
|
|
|
|
|
int retval;
|
|
int poller = epoll_create(1);
|
|
struct epoll_event ev;
|
|
ev.events = EPOLLIN;
|
|
ev.data.fd = CONN_fd;
|
|
epoll_ctl(poller, EPOLL_CTL_ADD, CONN_fd, &ev);
|
|
struct epoll_event events[1];
|
|
|
|
|
|
|
|
|
|
while (!ferror(CONN) && !feof(CONN)){
|
|
//only parse input if available or not yet init'ed
|
|
//rightnow = getNowMS();
|
|
retval = epoll_wait(poller, events, 1, 100);
|
|
if (retval > 0){
|
|
parseChunk();
|
|
fflush(CONN);
|
|
}
|
|
if (ready4data){
|
|
if (!inited){
|
|
//we are ready, connect the socket!
|
|
if (!ss.connect(streamname.c_str())){
|
|
#ifdef DEBUG
|
|
fprintf(stderr, "Could not connect to server!\n");
|
|
#endif
|
|
return 0;
|
|
}
|
|
FLV_Readheader(ss);//read the header, we don't want it
|
|
#ifdef DEBUG
|
|
fprintf(stderr, "Header read, starting to send video data...\n");
|
|
#endif
|
|
inited = true;
|
|
}
|
|
//only send data if previous data has been ACK'ed...
|
|
if (snd_cnt - snd_window_at < snd_window_size){
|
|
if (FLV_GetPacket(ss)){//able to read a full packet?
|
|
ts = FLVbuffer[7] * 256*256*256;
|
|
ts += FLVbuffer[4] * 256*256;
|
|
ts += FLVbuffer[5] * 256;
|
|
ts += FLVbuffer[6];
|
|
if (ts != 0){
|
|
if (fts == 0){fts = ts;ftst = getNowMS();}
|
|
ts -= fts;
|
|
FLVbuffer[7] = ts / (256*256*256);
|
|
FLVbuffer[4] = ts / (256*256);
|
|
FLVbuffer[5] = ts / 256;
|
|
FLVbuffer[6] = ts % 256;
|
|
ts += ftst;
|
|
}else{
|
|
ftst = getNowMS();
|
|
FLVbuffer[7] = ftst / (256*256*256);
|
|
FLVbuffer[4] = ftst / (256*256);
|
|
FLVbuffer[5] = ftst / 256;
|
|
FLVbuffer[6] = ftst % 256;
|
|
}
|
|
SendMedia((unsigned char)FLVbuffer[0], (unsigned char *)FLVbuffer+11, FLV_len-15, ts);
|
|
FLV_Dump();//dump packet and get ready for next
|
|
}
|
|
if ((SWBerr != SWBaseSocket::ok) && (SWBerr != SWBaseSocket::notReady)){
|
|
#ifdef DEBUG
|
|
fprintf(stderr, "No more data! :-( (%s)\n", SWBerr.get_error().c_str());
|
|
#endif
|
|
return 0;//no more input possible! Fail immediately.
|
|
}
|
|
}
|
|
}
|
|
//send ACK if we received a whole window
|
|
if (rec_cnt - rec_window_at > rec_window_size){
|
|
rec_window_at = rec_cnt;
|
|
SendCTL(3, rec_cnt);//send ack (msg 3)
|
|
}
|
|
}
|
|
#ifdef DEBUG
|
|
fprintf(stderr, "User disconnected.\n");
|
|
#endif
|
|
return 0;
|
|
}//main
|