simple rtmp server代码学习 （1）
最近从事一个视频流的项目开始了对simple rtmp server源代码的学习。关于simple rtmp server项目可以参照网址
https://github.com/winlinvip/simple-rtmp-server
在此做的是对版本0.9.194的学习总结，希望和大家共同学习一起提高，如有分析不对的地方欢迎随时指正。
端口监听与初始化（一）
全局变量_srs_config存放srs.conf中的配置指令。
全局变量_srs_server SRS服务器的全局类对象。
SrsServer中的listen_rtmp
Srs_main_server.cpp run_master()函数中：
if ((ret = _srs_server->listen()) !=ERROR_SUCCESS)
{
return ret;
}
看下SrsServer中的类listen函数是如何实现的：
intSrsServer::listen()
{
int ret = ERROR_SUCCESS;
if ((ret = listen_rtmp()) != ERROR_SUCCESS){
return ret;
}
if ((ret = listen_http_api()) !=ERROR_SUCCESS) {
return ret;
}
if ((ret = listen_http_stream()) !=ERROR_SUCCESS) {
return ret;
}
return ret;
}
listen函数实现了对rtmp,http_api和http_stream的端口监听，现在进一步分析下listen_rtmp函数的实现过程，代码如下，其中红色笔者添加的注释，并且没跳注释就用数字标号，用于流程之间的传递：
intSrsServer::listen_rtmp()
{
int ret = ERROR_SUCCESS;
// streamservice port.
//_srs_config全局变量，srs.conf中的所有配置指令都在对象root成员中。(1)
std::vector
srs_assert((int)ports.size() > 0);
close_listeners(SrsListenerRtmpStream);
for (int i = 0;i < (int)ports.size(); i++) {
//对每个端口创建一个SrsListener实例，然后将实例指针放到SrsServer中的 std::vector
SrsListener* listener = newSrsListener(this, SrsListenerRtmpStream);
listeners.push_back(listener);
//对每一个port建立一个套接字，并新建一个线程。(3)
int port = ::atoi(ports[i].c_str());
if ((ret = listener->listen(port))!= ERROR_SUCCESS) {
srs_error("RTMP stream listenat port %d failed. ret=%d", port, ret);
return ret;
}
}
return ret;
}
现在看下SrsListener类中的listen函数实现：
intSrsListener::listen(int port)
{
int ret = ERROR_SUCCESS;
_port = port;
//建立一个TCP套接字
if ((fd = socket(AF_INET, SOCK_STREAM, 0))== -1) {
ret = ERROR_SOCKET_CREATE;
srs_error("create linux socketerror. ret=%d", ret);
return ret;
}
srs_verbose("create linux socketsuccess. fd=%d", fd);
int reuse_socket = 1;
if (setsockopt(fd, SOL_SOCKET,SO_REUSEADDR, &reuse_socket, sizeof(int)) == -1) {
ret = ERROR_SOCKET_SETREUSE;
srs_error("setsockopt reuse-addrerror. ret=%d", ret);
return ret;
}
srs_verbose("setsockopt reuse-addrsuccess. fd=%d", fd);
sockaddr_in addr;
addr.sin_family = AF_INET;
addr.sin_port = htons(_port);
addr.sin_addr.s_addr= INADDR_ANY;
//绑定端口port；(4)
if (bind(fd, (const sockaddr*)&addr,sizeof(sockaddr_in)) == -1) {
ret = ERROR_SOCKET_BIND;
srs_error("bind socket error.ret=%d", ret);
return ret;
}
srs_verbose("bind socket success.fd=%d", fd);
//监听端口；(5)
if (::listen(fd, SERVER_LISTEN_BACKLOG) ==-1) {
ret = ERROR_SOCKET_LISTEN;
srs_error("listen socket error. ret=%d", ret);
return ret;
}
srs_verbose("listen socket success.fd=%d", fd);
//建立一个_st_netfd_t结构体的实例；(6)
if ((stfd = st_netfd_open_socket(fd)) ==NULL){
ret = ERROR_ST_OPEN_SOCKET;
srs_error("st_netfd_open_socketopen socket failed. ret=%d", ret);
return ret;
}
srs_verbose("st open socket success.fd=%d", fd);
//SrsListener类中的有一个SrsThread类型的指针，下面标红的代码将开始一个新的线程;(7)
if ((ret = pthread->start())!= ERROR_SUCCESS) {
srs_error("st_thread_create listenthread error. ret=%d", ret);
return ret;
}
srs_verbose("create st listen threadsuccess.");
srs_trace("listen thread cid=%d,current_cid=%d, "
"listen at port=%d, type=%d, fd=%dstarted success",
pthread->cid(),_srs_context->get_id(), _port, _type, fd);
return ret;
}
现在看下SrsThread类中的start函数：
intSrsThread::start()
{
int ret = ERROR_SUCCESS;
if(tid) {
srs_info("thread alreadyrunning.");
return ret;
}
//调用了state-threads库函数st_thread_create()，将自身的成员变量thread_fun作为线程运行的主函数，this指针做为thread_fun的一个参数，也就是将SrsThread这个实例传入到thread_fun中；(8)
if((tid = st_thread_create(thread_fun,this, (_joinable? 1:0), 0)) == NULL){
ret = ERROR_ST_CREATE_CYCLE_THREAD;
srs_error("st_thread_createfailed. ret=%d", ret);
return ret;
}
// we set to loop to true for thread torun.
loop = true;
// wait for cid to ready, for parent threadto get the cid.
while (_cid < 0 && loop) {
st_usleep(10 * SRS_TIME_MILLISECONDS);
}
// now, cycle thread can run.
can_run = true;
return ret;
}
现在看下SrsThread函数thread_fun：
//此处的arg就是(8)中的this，obj->thread_cycle()即调用的SrsThread实例的thread_cycle(); (9)
void*SrsThread::thread_fun(void* arg)
{
SrsThread* obj = (SrsThread*)arg;
srs_assert(obj);
//obj是什么呢？
obj->thread_cycle();
st_thread_exit(NULL);
return NULL;
}
看下thread_cycle函数
voidSrsThread::thread_cycle()
{
int ret = ERROR_SUCCESS;
_srs_context->generate_id();
srs_info("thread cycle start");
_cid = _srs_context->get_id();
srs_assert(handler);
handler->on_thread_start();
// wait for cid to ready, for parent threadto get the cid.
while (!can_run && loop) {
st_usleep(10 * SRS_TIME_MILLISECONDS);
}
while (loop) {
if ((ret =handler->on_before_cycle()) != ERROR_SUCCESS) {
srs_warn("thread on beforecycle failed, ignored and retry, ret=%d", ret);
goto failed;
}
srs_info("thread on before cyclesuccess");
//注意标红的代码，这个handler是什么呢？这需要看下SrsThread构造函数对handler是如何初始化的。(10)
if ((ret =handler->cycle()) != ERROR_SUCCESS) {
srs_warn("thread cycle failed,ignored and retry, ret=%d", ret);
goto failed;
}
srs_info("thread cyclesuccess");
if ((ret = handler->on_end_cycle())!= ERROR_SUCCESS) {
srs_warn("thread on end cyclefailed, ignored and retry, ret=%d", ret);
goto failed;
}
srs_info("thread on end cyclesuccess");
failed:
if (!loop) {
break;
}
st_usleep(cycle_interval_us);
}
handler->on_thread_stop();
srs_info("thread cyclefinished");
}
下面的代码是SrsThread类的构造函数，从中可以看出handler值由thread_handler传递过来的，那么这个thread_handler到底是个什么东西呢？
SrsThread::SrsThread(ISrsThreadHandler*thread_handler, int64_t interval_us, bool joinable)
{
handler =thread_handler;
cycle_interval_us = interval_us;
tid = NULL;
loop = false;
_cid = -1;
_joinable = joinable;
can_run = false;
}
在注释(7)处第一次出现了thread，thread是类SrsListener一个成员，在SysListener构造函数中将listener实例自身传给SrsThread的handler，所以注释(10)中的handler->cycle也就是SrsLintener中的cycle()函数。
SrsListener::SrsListener(SrsServer*server, SrsListenerType type)
{
fd = -1;
stfd = NULL;
_port = 0;
_server = server;
_type = type;
pthread = newSrsThread(this, 0, true);//注释(11)
}
至此告一段落做个总结，图如下所示：
到最后将是对SrsListener::cycle()的调用。