/** * The MIT License (MIT) * * Copyright (c) 2013-2020 Winlin * * Permission is hereby granted, free of charge, to any person obtaining a copy of * this software and associated documentation files (the "Software"), to deal in * the Software without restriction, including without limitation the rights to * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of * the Software, and to permit persons to whom the Software is furnished to do so, * subject to the following conditions: * * The above copyright notice and this permission notice shall be included in all * copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS * FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER * IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ #include #include #include #include #include #include #include #include #ifndef SRS_OSX #include #endif using namespace std; #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include // system interval in srs_utime_t, // all resolution times should be times togother, // for example, system-interval is x=1s(1000ms), // then rusage can be 3*x, for instance, 3*1=3s, // the meminfo canbe 6*x, for instance, 6*1=6s, // for performance refine, @see: https://github.com/ossrs/srs/issues/194 // @remark, recomment to 1000ms. #define SRS_SYS_CYCLE_INTERVAL (1000 * SRS_UTIME_MILLISECONDS) // update time interval: // SRS_SYS_CYCLE_INTERVAL * SRS_SYS_TIME_RESOLUTION_MS_TIMES #define SRS_SYS_TIME_RESOLUTION_MS_TIMES 1 // update rusage interval: // SRS_SYS_CYCLE_INTERVAL * SRS_SYS_RUSAGE_RESOLUTION_TIMES #define SRS_SYS_RUSAGE_RESOLUTION_TIMES 3 // update network devices info interval: // SRS_SYS_CYCLE_INTERVAL * SRS_SYS_NETWORK_RTMP_SERVER_RESOLUTION_TIMES #define SRS_SYS_NETWORK_RTMP_SERVER_RESOLUTION_TIMES 3 // update rusage interval: // SRS_SYS_CYCLE_INTERVAL * SRS_SYS_CPU_STAT_RESOLUTION_TIMES #define SRS_SYS_CPU_STAT_RESOLUTION_TIMES 3 // update the disk iops interval: // SRS_SYS_CYCLE_INTERVAL * SRS_SYS_DISK_STAT_RESOLUTION_TIMES #define SRS_SYS_DISK_STAT_RESOLUTION_TIMES 6 // update rusage interval: // SRS_SYS_CYCLE_INTERVAL * SRS_SYS_MEMINFO_RESOLUTION_TIMES #define SRS_SYS_MEMINFO_RESOLUTION_TIMES 6 // update platform info interval: // SRS_SYS_CYCLE_INTERVAL * SRS_SYS_PLATFORM_INFO_RESOLUTION_TIMES #define SRS_SYS_PLATFORM_INFO_RESOLUTION_TIMES 9 // update network devices info interval: // SRS_SYS_CYCLE_INTERVAL * SRS_SYS_NETWORK_DEVICE_RESOLUTION_TIMES #define SRS_SYS_NETWORK_DEVICE_RESOLUTION_TIMES 9 std::string srs_listener_type2string(SrsListenerType type) { switch (type) { case SrsListenerRtmpStream: return "RTMP"; case SrsListenerHttpApi: return "HTTP-API"; case SrsListenerHttpStream: return "HTTP-Server"; case SrsListenerMpegTsOverUdp: return "MPEG-TS over UDP"; case SrsListenerRtsp: return "RTSP"; case SrsListenerFlv: return "HTTP-FLV"; case SrsListenerGb28181Sip: return "GB28181-SIP over UDP"; case SrsListenerGb28181RtpMux: return "GB28181-Stream over RTP"; default: return "UNKONWN"; } } SrsListener::SrsListener(SrsServer* svr, SrsListenerType t) { port = 0; server = svr; type = t; } SrsListener::~SrsListener() { } SrsListenerType SrsListener::listen_type() { return type; } SrsBufferListener::SrsBufferListener(SrsServer* svr, SrsListenerType t) : SrsListener(svr, t) { listener = NULL; } SrsBufferListener::~SrsBufferListener() { srs_freep(listener); } srs_error_t SrsBufferListener::listen(string i, int p) { srs_error_t err = srs_success; ip = i; port = p; srs_freep(listener); listener = new SrsTcpListener(this, ip, port); if ((err = listener->listen()) != srs_success) { return srs_error_wrap(err, "buffered tcp listen"); } string v = srs_listener_type2string(type); srs_trace("%s listen at tcp://%s:%d, fd=%d", v.c_str(), ip.c_str(), port, listener->fd()); return err; } srs_error_t SrsBufferListener::on_tcp_client(srs_netfd_t stfd) { srs_error_t err = server->accept_client(type, stfd); if (err != srs_success) { srs_warn("accept client failed, err is %s", srs_error_desc(err).c_str()); srs_freep(err); } return srs_success; } SrsRtspListener::SrsRtspListener(SrsServer* svr, SrsListenerType t, SrsConfDirective* c) : SrsListener(svr, t) { listener = NULL; // the caller already ensure the type is ok, // we just assert here for unknown stream caster. srs_assert(type == SrsListenerRtsp); if (type == SrsListenerRtsp) { caster = new SrsRtspCaster(c); // TODO: FIXME: Must check error. caster->initialize(); } } SrsRtspListener::~SrsRtspListener() { srs_freep(caster); srs_freep(listener); } srs_error_t SrsRtspListener::listen(string i, int p) { srs_error_t err = srs_success; // the caller already ensure the type is ok, // we just assert here for unknown stream caster. srs_assert(type == SrsListenerRtsp); ip = i; port = p; srs_freep(listener); listener = new SrsTcpListener(this, ip, port); if ((err = listener->listen()) != srs_success) { return srs_error_wrap(err, "rtsp listen %s:%d", ip.c_str(), port); } string v = srs_listener_type2string(type); srs_trace("%s listen at tcp://%s:%d, fd=%d", v.c_str(), ip.c_str(), port, listener->fd()); return err; } srs_error_t SrsRtspListener::on_tcp_client(srs_netfd_t stfd) { srs_error_t err = caster->on_tcp_client(stfd); if (err != srs_success) { srs_warn("accept client failed, err is %s", srs_error_desc(err).c_str()); srs_freep(err); } return srs_success; } SrsHttpFlvListener::SrsHttpFlvListener(SrsServer* svr, SrsListenerType t, SrsConfDirective* c) : SrsListener(svr, t) { listener = NULL; // the caller already ensure the type is ok, // we just assert here for unknown stream caster. srs_assert(type == SrsListenerFlv); if (type == SrsListenerFlv) { caster = new SrsAppCasterFlv(c); } } SrsHttpFlvListener::~SrsHttpFlvListener() { srs_freep(caster); srs_freep(listener); } srs_error_t SrsHttpFlvListener::listen(string i, int p) { srs_error_t err = srs_success; // the caller already ensure the type is ok, // we just assert here for unknown stream caster. srs_assert(type == SrsListenerFlv); ip = i; port = p; if ((err = caster->initialize()) != srs_success) { return srs_error_wrap(err, "init caster %s:%d", ip.c_str(), port); } srs_freep(listener); listener = new SrsTcpListener(this, ip, port); if ((err = listener->listen()) != srs_success) { return srs_error_wrap(err, "listen"); } string v = srs_listener_type2string(type); srs_trace("%s listen at tcp://%s:%d, fd=%d", v.c_str(), ip.c_str(), port, listener->fd()); return err; } srs_error_t SrsHttpFlvListener::on_tcp_client(srs_netfd_t stfd) { srs_error_t err = caster->on_tcp_client(stfd); if (err != srs_success) { srs_warn("accept client failed, err is %s", srs_error_desc(err).c_str()); srs_freep(err); } return err; } SrsUdpStreamListener::SrsUdpStreamListener(SrsServer* svr, SrsListenerType t, ISrsUdpHandler* c) : SrsListener(svr, t) { listener = NULL; caster = c; } SrsUdpStreamListener::~SrsUdpStreamListener() { srs_freep(listener); } srs_error_t SrsUdpStreamListener::listen(string i, int p) { srs_error_t err = srs_success; // the caller already ensure the type is ok, // we just assert here for unknown stream caster. srs_assert(type == SrsListenerMpegTsOverUdp || type == SrsListenerGb28181Sip || type == SrsListenerGb28181RtpMux); ip = i; port = p; srs_freep(listener); listener = new SrsUdpListener(caster, ip, port); if ((err = listener->listen()) != srs_success) { return srs_error_wrap(err, "listen %s:%d", ip.c_str(), port); } // notify the handler the fd changed. if ((err = caster->on_stfd_change(listener->stfd())) != srs_success) { return srs_error_wrap(err, "notify fd change failed"); } string v = srs_listener_type2string(type); srs_trace("%s listen at udp://%s:%d, fd=%d", v.c_str(), ip.c_str(), port, listener->fd()); return err; } SrsUdpCasterListener::SrsUdpCasterListener(SrsServer* svr, SrsListenerType t, SrsConfDirective* c) : SrsUdpStreamListener(svr, t, NULL) { // the caller already ensure the type is ok, // we just assert here for unknown stream caster. srs_assert(type == SrsListenerMpegTsOverUdp); if (type == SrsListenerMpegTsOverUdp) { caster = new SrsMpegtsOverUdp(c); } } SrsUdpCasterListener::~SrsUdpCasterListener() { srs_freep(caster); } #ifdef SRS_GB28181 SrsGb28181Listener::SrsGb28181Listener(SrsServer* svr, SrsListenerType t, SrsConfDirective* c) : SrsUdpStreamListener(svr, t, NULL) { // the caller already ensure the type is ok, // we just assert here for unknown stream caster. srs_assert(type == SrsListenerGb28181Sip ||type == SrsListenerGb28181RtpMux); if (type == SrsListenerGb28181Sip) { caster = new SrsGb28181SipService(c); }else if(type == SrsListenerGb28181RtpMux){ caster = new SrsGb28181RtpMuxService(c); } } SrsGb28181Listener::~SrsGb28181Listener() { srs_freep(caster); } #endif SrsSignalManager* SrsSignalManager::instance = NULL; SrsSignalManager::SrsSignalManager(SrsServer* s) { SrsSignalManager::instance = this; server = s; sig_pipe[0] = sig_pipe[1] = -1; trd = new SrsSTCoroutine("signal", this, _srs_context->get_id()); signal_read_stfd = NULL; } SrsSignalManager::~SrsSignalManager() { srs_close_stfd(signal_read_stfd); if (sig_pipe[0] > 0) { ::close(sig_pipe[0]); } if (sig_pipe[1] > 0) { ::close(sig_pipe[1]); } srs_freep(trd); } srs_error_t SrsSignalManager::initialize() { /* Create signal pipe */ if (pipe(sig_pipe) < 0) { return srs_error_new(ERROR_SYSTEM_CREATE_PIPE, "create pipe"); } if ((signal_read_stfd = srs_netfd_open(sig_pipe[0])) == NULL) { return srs_error_new(ERROR_SYSTEM_CREATE_PIPE, "open pipe"); } return srs_success; } srs_error_t SrsSignalManager::start() { srs_error_t err = srs_success; /** * Note that if multiple processes are used (see below), * the signal pipe should be initialized after the fork(2) call * so that each process has its own private pipe. */ struct sigaction sa; /* Install sig_catcher() as a signal handler */ sa.sa_handler = SrsSignalManager::sig_catcher; sigemptyset(&sa.sa_mask); sa.sa_flags = 0; sigaction(SRS_SIGNAL_RELOAD, &sa, NULL); sa.sa_handler = SrsSignalManager::sig_catcher; sigemptyset(&sa.sa_mask); sa.sa_flags = 0; sigaction(SRS_SIGNAL_FAST_QUIT, &sa, NULL); sa.sa_handler = SrsSignalManager::sig_catcher; sigemptyset(&sa.sa_mask); sa.sa_flags = 0; sigaction(SRS_SIGNAL_GRACEFULLY_QUIT, &sa, NULL); sa.sa_handler = SrsSignalManager::sig_catcher; sigemptyset(&sa.sa_mask); sa.sa_flags = 0; sigaction(SIGINT, &sa, NULL); sa.sa_handler = SrsSignalManager::sig_catcher; sigemptyset(&sa.sa_mask); sa.sa_flags = 0; sigaction(SRS_SIGNAL_REOPEN_LOG, &sa, NULL); srs_trace("signal installed, reload=%d, reopen=%d, fast_quit=%d, grace_quit=%d", SRS_SIGNAL_RELOAD, SRS_SIGNAL_REOPEN_LOG, SRS_SIGNAL_FAST_QUIT, SRS_SIGNAL_GRACEFULLY_QUIT); if ((err = trd->start()) != srs_success) { return srs_error_wrap(err, "signal manager"); } return err; } srs_error_t SrsSignalManager::cycle() { srs_error_t err = srs_success; while (true) { if ((err = trd->pull()) != srs_success) { return srs_error_wrap(err, "signal manager"); } int signo; /* Read the next signal from the pipe */ srs_read(signal_read_stfd, &signo, sizeof(int), SRS_UTIME_NO_TIMEOUT); /* Process signal synchronously */ server->on_signal(signo); } return err; } void SrsSignalManager::sig_catcher(int signo) { int err; /* Save errno to restore it after the write() */ err = errno; /* write() is reentrant/async-safe */ int fd = SrsSignalManager::instance->sig_pipe[1]; write(fd, &signo, sizeof(int)); errno = err; } // Whether we are in docker, defined in main module. extern bool _srs_in_docker; SrsInotifyWorker::SrsInotifyWorker(SrsServer* s) { server = s; trd = new SrsSTCoroutine("inotify", this); inotify_fd = NULL; } SrsInotifyWorker::~SrsInotifyWorker() { srs_freep(trd); srs_close_stfd(inotify_fd); } srs_error_t SrsInotifyWorker::start() { srs_error_t err = srs_success; #ifndef SRS_OSX // Whether enable auto reload config. bool auto_reload = _srs_config->inotify_auto_reload(); if (!auto_reload && _srs_in_docker && _srs_config->auto_reload_for_docker()) { srs_warn("enable auto reload for docker"); auto_reload = true; } if (!auto_reload) { return err; } // Create inotify to watch config file. int fd = ::inotify_init1(IN_NONBLOCK); if (fd < 0) { return srs_error_new(ERROR_INOTIFY_CREATE, "create inotify"); } // Open as stfd to read by ST. if ((inotify_fd = srs_netfd_open(fd)) == NULL) { ::close(fd); return srs_error_new(ERROR_INOTIFY_OPENFD, "open fd=%d", fd); } if (((err = srs_fd_closeexec(fd))) != srs_success) { return srs_error_wrap(err, "closeexec fd=%d", fd); } // /* the following are legal, implemented events that user-space can watch for */ // #define IN_ACCESS 0x00000001 /* File was accessed */ // #define IN_MODIFY 0x00000002 /* File was modified */ // #define IN_ATTRIB 0x00000004 /* Metadata changed */ // #define IN_CLOSE_WRITE 0x00000008 /* Writtable file was closed */ // #define IN_CLOSE_NOWRITE 0x00000010 /* Unwrittable file closed */ // #define IN_OPEN 0x00000020 /* File was opened */ // #define IN_MOVED_FROM 0x00000040 /* File was moved from X */ // #define IN_MOVED_TO 0x00000080 /* File was moved to Y */ // #define IN_CREATE 0x00000100 /* Subfile was created */ // #define IN_DELETE 0x00000200 /* Subfile was deleted */ // #define IN_DELETE_SELF 0x00000400 /* Self was deleted */ // #define IN_MOVE_SELF 0x00000800 /* Self was moved */ // // /* the following are legal events. they are sent as needed to any watch */ // #define IN_UNMOUNT 0x00002000 /* Backing fs was unmounted */ // #define IN_Q_OVERFLOW 0x00004000 /* Event queued overflowed */ // #define IN_IGNORED 0x00008000 /* File was ignored */ // // /* helper events */ // #define IN_CLOSE (IN_CLOSE_WRITE | IN_CLOSE_NOWRITE) /* close */ // #define IN_MOVE (IN_MOVED_FROM | IN_MOVED_TO) /* moves */ // // /* special flags */ // #define IN_ONLYDIR 0x01000000 /* only watch the path if it is a directory */ // #define IN_DONT_FOLLOW 0x02000000 /* don't follow a sym link */ // #define IN_EXCL_UNLINK 0x04000000 /* exclude events on unlinked objects */ // #define IN_MASK_ADD 0x20000000 /* add to the mask of an already existing watch */ // #define IN_ISDIR 0x40000000 /* event occurred against dir */ // #define IN_ONESHOT 0x80000000 /* only send event once */ // Watch the config directory events. string config_dir = srs_path_dirname(_srs_config->config()); uint32_t mask = IN_MODIFY | IN_CREATE | IN_MOVED_TO; int watch_conf = 0; if ((watch_conf = ::inotify_add_watch(fd, config_dir.c_str(), mask)) < 0) { return srs_error_new(ERROR_INOTIFY_WATCH, "watch file=%s, fd=%d, watch=%d, mask=%#x", config_dir.c_str(), fd, watch_conf, mask); } srs_trace("auto reload watching fd=%d, watch=%d, file=%s", fd, watch_conf, config_dir.c_str()); if ((err = trd->start()) != srs_success) { return srs_error_wrap(err, "inotify"); } #endif return err; } srs_error_t SrsInotifyWorker::cycle() { srs_error_t err = srs_success; #ifndef SRS_OSX string config_path = _srs_config->config(); string config_file = srs_path_basename(config_path); string k8s_file = "..data"; while (true) { char buf[4096]; ssize_t nn = srs_read(inotify_fd, buf, (size_t)sizeof(buf), SRS_UTIME_NO_TIMEOUT); if (nn < 0) { srs_warn("inotify ignore read failed, nn=%d", (int)nn); break; } // Whether config file changed. bool do_reload = false; // Parse all inotify events. inotify_event* ie = NULL; for (char* ptr = buf; ptr < buf + nn; ptr += sizeof(inotify_event) + ie->len) { ie = (inotify_event*)ptr; if (!ie->len || !ie->name) { continue; } string name = ie->name; if ((name == k8s_file || name == config_file) && ie->mask & (IN_MODIFY|IN_CREATE|IN_MOVED_TO)) { do_reload = true; } srs_trace("inotify event wd=%d, mask=%#x, len=%d, name=%s, reload=%d", ie->wd, ie->mask, ie->len, ie->name, do_reload); } // Notify server to do reload. if (do_reload && srs_path_exists(config_path)) { server->on_signal(SRS_SIGNAL_RELOAD); } srs_usleep(3000 * SRS_UTIME_MILLISECONDS); } #endif return err; } ISrsServerCycle::ISrsServerCycle() { } ISrsServerCycle::~ISrsServerCycle() { } SrsServer::SrsServer() { signal_reload = false; signal_persistence_config = false; signal_gmc_stop = false; signal_fast_quit = false; signal_gracefully_quit = false; pid_fd = -1; signal_manager = new SrsSignalManager(this); conn_manager = new SrsResourceManager("RTMP/API"); handler = NULL; ppid = ::getppid(); // donot new object in constructor, // for some global instance is not ready now, // new these objects in initialize instead. http_api_mux = new SrsHttpServeMux(); http_server = new SrsHttpServer(this); http_heartbeat = new SrsHttpHeartbeat(); ingester = new SrsIngester(); } SrsServer::~SrsServer() { destroy(); } void SrsServer::destroy() { srs_warn("start destroy server"); dispose(); srs_freep(http_api_mux); srs_freep(http_server); srs_freep(http_heartbeat); srs_freep(ingester); if (pid_fd > 0) { ::close(pid_fd); pid_fd = -1; } srs_freep(signal_manager); srs_freep(conn_manager); #ifdef SRS_GB28181 //free global gb28181 manager srs_freep(_srs_gb28181); #endif } void SrsServer::dispose() { _srs_config->unsubscribe(this); // prevent fresh clients. close_listeners(SrsListenerRtmpStream); close_listeners(SrsListenerHttpApi); close_listeners(SrsListenerHttpStream); close_listeners(SrsListenerMpegTsOverUdp); close_listeners(SrsListenerRtsp); close_listeners(SrsListenerFlv); // Fast stop to notify FFMPEG to quit, wait for a while then fast kill. ingester->dispose(); // dispose the source for hls and dvr. _srs_sources->dispose(); // @remark don't dispose all connections, for too slow. #ifdef SRS_MEM_WATCH srs_memory_report(); #endif } void SrsServer::gracefully_dispose() { _srs_config->unsubscribe(this); // Always wait for a while to start. srs_usleep(_srs_config->get_grace_start_wait()); srs_trace("start wait for %dms", srsu2msi(_srs_config->get_grace_start_wait())); // prevent fresh clients. close_listeners(SrsListenerRtmpStream); close_listeners(SrsListenerHttpApi); close_listeners(SrsListenerHttpStream); close_listeners(SrsListenerMpegTsOverUdp); close_listeners(SrsListenerRtsp); close_listeners(SrsListenerFlv); srs_trace("listeners closed"); // Fast stop to notify FFMPEG to quit, wait for a while then fast kill. ingester->stop(); srs_trace("ingesters stopped"); // Wait for connections to quit. // While gracefully quiting, user can requires SRS to fast quit. int wait_step = 1; while (!conn_manager->empty() && !signal_fast_quit) { for (int i = 0; i < wait_step && !conn_manager->empty() && !signal_fast_quit; i++) { srs_usleep(1000 * SRS_UTIME_MILLISECONDS); } wait_step = (wait_step * 2) % 33; srs_trace("wait for %d conns to quit", (int)conn_manager->size()); } // dispose the source for hls and dvr. _srs_sources->dispose(); srs_trace("source disposed"); #ifdef SRS_MEM_WATCH srs_memory_report(); #endif srs_usleep(_srs_config->get_grace_final_wait()); srs_trace("final wait for %dms", srsu2msi(_srs_config->get_grace_final_wait())); } srs_error_t SrsServer::initialize(ISrsServerCycle* ch) { srs_error_t err = srs_success; // ensure the time is ok. srs_update_system_time(); // for the main objects(server, config, log, context), // never subscribe handler in constructor, // instead, subscribe handler in initialize method. srs_assert(_srs_config); _srs_config->subscribe(this); handler = ch; if(handler && (err = handler->initialize()) != srs_success){ return srs_error_wrap(err, "handler initialize"); } if ((err = http_api_mux->initialize()) != srs_success) { return srs_error_wrap(err, "http api initialize"); } if ((err = http_server->initialize()) != srs_success) { return srs_error_wrap(err, "http server initialize"); } return err; } srs_error_t SrsServer::initialize_st() { srs_error_t err = srs_success; // @remark, st alloc segment use mmap, which only support 32757 threads, // if need to support more, for instance, 100k threads, define the macro MALLOC_STACK. // TODO: FIXME: maybe can use "sysctl vm.max_map_count" to refine. #define __MMAP_MAX_CONNECTIONS 32756 if (_srs_config->get_max_connections() > __MMAP_MAX_CONNECTIONS) { srs_error("st mmap for stack allocation must <= %d threads, " "@see Makefile of st for MALLOC_STACK, please build st manually by " "\"make EXTRA_CFLAGS=-DMALLOC_STACK linux-debug\"", __MMAP_MAX_CONNECTIONS); return srs_error_new(ERROR_ST_EXCEED_THREADS, "%d exceed max %d threads", _srs_config->get_max_connections(), __MMAP_MAX_CONNECTIONS); } // check asprocess. bool asprocess = _srs_config->get_asprocess(); if (asprocess && ppid == 1) { return srs_error_new(ERROR_SYSTEM_ASSERT_FAILED, "ppid=%d illegal for asprocess", ppid); } srs_trace("server main cid=%s, pid=%d, ppid=%d, asprocess=%d", _srs_context->get_id().c_str(), ::getpid(), ppid, asprocess); return err; } srs_error_t SrsServer::initialize_signal() { return signal_manager->initialize(); } srs_error_t SrsServer::acquire_pid_file() { // when srs in dolphin mode, no need the pid file. if (_srs_config->is_dolphin()) { return srs_success; } std::string pid_file = _srs_config->get_pid_file(); // -rw-r--r-- // 644 int mode = S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH; int fd; // open pid file if ((fd = ::open(pid_file.c_str(), O_WRONLY | O_CREAT, mode)) == -1) { return srs_error_new(ERROR_SYSTEM_PID_ACQUIRE, "open pid file=%s", pid_file.c_str()); } // require write lock struct flock lock; lock.l_type = F_WRLCK; // F_RDLCK, F_WRLCK, F_UNLCK lock.l_start = 0; // type offset, relative to l_whence lock.l_whence = SEEK_SET; // SEEK_SET, SEEK_CUR, SEEK_END lock.l_len = 0; if (fcntl(fd, F_SETLK, &lock) == -1) { if(errno == EACCES || errno == EAGAIN) { ::close(fd); srs_error("srs is already running!"); return srs_error_new(ERROR_SYSTEM_PID_ALREADY_RUNNING, "srs is already running"); } return srs_error_new(ERROR_SYSTEM_PID_LOCK, "access to pid=%s", pid_file.c_str()); } // truncate file if (ftruncate(fd, 0) != 0) { return srs_error_new(ERROR_SYSTEM_PID_TRUNCATE_FILE, "truncate pid file=%s", pid_file.c_str()); } // write the pid string pid = srs_int2str(getpid()); if (write(fd, pid.c_str(), pid.length()) != (int)pid.length()) { return srs_error_new(ERROR_SYSTEM_PID_WRITE_FILE, "write pid=%s to file=%s", pid.c_str(), pid_file.c_str()); } // auto close when fork child process. int val; if ((val = fcntl(fd, F_GETFD, 0)) < 0) { return srs_error_new(ERROR_SYSTEM_PID_GET_FILE_INFO, "fcntl fd=%d", fd); } val |= FD_CLOEXEC; if (fcntl(fd, F_SETFD, val) < 0) { return srs_error_new(ERROR_SYSTEM_PID_SET_FILE_INFO, "lock file=%s fd=%d", pid_file.c_str(), fd); } srs_trace("write pid=%s to %s success!", pid.c_str(), pid_file.c_str()); pid_fd = fd; return srs_success; } srs_error_t SrsServer::listen() { srs_error_t err = srs_success; if ((err = listen_rtmp()) != srs_success) { return srs_error_wrap(err, "rtmp listen"); } if ((err = listen_http_api()) != srs_success) { return srs_error_wrap(err, "http api listen"); } if ((err = listen_http_stream()) != srs_success) { return srs_error_wrap(err, "http stream listen"); } if ((err = listen_stream_caster()) != srs_success) { return srs_error_wrap(err, "stream caster listen"); } if ((err = conn_manager->start()) != srs_success) { return srs_error_wrap(err, "connection manager"); } return err; } srs_error_t SrsServer::register_signal() { srs_error_t err = srs_success; if ((err = signal_manager->start()) != srs_success) { return srs_error_wrap(err, "signal manager start"); } return err; } srs_error_t SrsServer::http_handle() { srs_error_t err = srs_success; if ((err = http_api_mux->handle("/", new SrsGoApiRoot())) != srs_success) { return srs_error_wrap(err, "handle /"); } if ((err = http_api_mux->handle("/api/", new SrsGoApiApi())) != srs_success) { return srs_error_wrap(err, "handle api"); } if ((err = http_api_mux->handle("/api/v1/", new SrsGoApiV1())) != srs_success) { return srs_error_wrap(err, "handle v1"); } if ((err = http_api_mux->handle("/api/v1/versions", new SrsGoApiVersion())) != srs_success) { return srs_error_wrap(err, "handle versions"); } if ((err = http_api_mux->handle("/api/v1/summaries", new SrsGoApiSummaries())) != srs_success) { return srs_error_wrap(err, "handle summaries"); } if ((err = http_api_mux->handle("/api/v1/rusages", new SrsGoApiRusages())) != srs_success) { return srs_error_wrap(err, "handle rusages"); } if ((err = http_api_mux->handle("/api/v1/self_proc_stats", new SrsGoApiSelfProcStats())) != srs_success) { return srs_error_wrap(err, "handle self proc stats"); } if ((err = http_api_mux->handle("/api/v1/system_proc_stats", new SrsGoApiSystemProcStats())) != srs_success) { return srs_error_wrap(err, "handle system proc stats"); } if ((err = http_api_mux->handle("/api/v1/meminfos", new SrsGoApiMemInfos())) != srs_success) { return srs_error_wrap(err, "handle meminfos"); } if ((err = http_api_mux->handle("/api/v1/authors", new SrsGoApiAuthors())) != srs_success) { return srs_error_wrap(err, "handle authors"); } if ((err = http_api_mux->handle("/api/v1/features", new SrsGoApiFeatures())) != srs_success) { return srs_error_wrap(err, "handle features"); } if ((err = http_api_mux->handle("/api/v1/vhosts/", new SrsGoApiVhosts())) != srs_success) { return srs_error_wrap(err, "handle vhosts"); } if ((err = http_api_mux->handle("/api/v1/streams/", new SrsGoApiStreams())) != srs_success) { return srs_error_wrap(err, "handle streams"); } if ((err = http_api_mux->handle("/api/v1/clients/", new SrsGoApiClients())) != srs_success) { return srs_error_wrap(err, "handle clients"); } if ((err = http_api_mux->handle("/api/v1/raw", new SrsGoApiRaw(this))) != srs_success) { return srs_error_wrap(err, "handle raw"); } if ((err = http_api_mux->handle("/api/v1/clusters", new SrsGoApiClusters())) != srs_success) { return srs_error_wrap(err, "handle clusters"); } if ((err = http_api_mux->handle("/api/v1/perf", new SrsGoApiPerf())) != srs_success) { return srs_error_wrap(err, "handle perf"); } #ifdef SRS_GB28181 if ((err = http_api_mux->handle("/api/v1/gb28181", new SrsGoApiGb28181())) != srs_success) { return srs_error_wrap(err, "handle raw"); } #endif // test the request info. if ((err = http_api_mux->handle("/api/v1/tests/requests", new SrsGoApiRequests())) != srs_success) { return srs_error_wrap(err, "handle tests requests"); } // test the error code response. if ((err = http_api_mux->handle("/api/v1/tests/errors", new SrsGoApiError())) != srs_success) { return srs_error_wrap(err, "handle tests errors"); } // test the redirect mechenism. if ((err = http_api_mux->handle("/api/v1/tests/redirects", new SrsHttpRedirectHandler("/api/v1/tests/errors", SRS_CONSTS_HTTP_MovedPermanently))) != srs_success) { return srs_error_wrap(err, "handle tests redirects"); } // test the http vhost. if ((err = http_api_mux->handle("error.srs.com/api/v1/tests/errors", new SrsGoApiError())) != srs_success) { return srs_error_wrap(err, "handle tests errors for error.srs.com"); } #ifdef SRS_GPERF // The test api for get tcmalloc stats. // @see Memory Introspection in https://gperftools.github.io/gperftools/tcmalloc.html if ((err = http_api_mux->handle("/api/v1/tcmalloc", new SrsGoApiTcmalloc())) != srs_success) { return srs_error_wrap(err, "handle tests errors"); } #endif // TODO: FIXME: for console. // TODO: FIXME: support reload. std::string dir = _srs_config->get_http_stream_dir() + "/console"; if ((err = http_api_mux->handle("/console/", new SrsHttpFileServer(dir))) != srs_success) { return srs_error_wrap(err, "handle console at %s", dir.c_str()); } srs_trace("http: api mount /console to %s", dir.c_str()); return err; } srs_error_t SrsServer::ingest() { srs_error_t err = srs_success; if ((err = ingester->start()) != srs_success) { return srs_error_wrap(err, "ingest start"); } return err; } srs_error_t SrsServer::cycle() { srs_error_t err = srs_success; // Start the inotify auto reload by watching config file. SrsInotifyWorker inotify(this); if ((err = inotify.start()) != srs_success) { return srs_error_wrap(err, "start inotify"); } // Do server main cycle. err = do_cycle(); #ifdef SRS_GPERF_MC destroy(); // remark, for gmc, never invoke the exit(). srs_warn("sleep a long time for system st-threads to cleanup."); srs_usleep(3 * 1000 * 1000); srs_warn("system quit"); return err; #endif // quit normally. srs_warn("main cycle terminated, system quit normally."); // fast quit, do some essential cleanup. if (signal_fast_quit) { dispose(); // TODO: FIXME: Rename to essential_dispose. srs_trace("srs disposed"); } // gracefully quit, do carefully cleanup. if (signal_gracefully_quit) { gracefully_dispose(); srs_trace("srs gracefully quit"); } srs_trace("srs terminated"); // for valgrind to detect. srs_freep(_srs_config); srs_freep(_srs_log); exit(0); return err; } void SrsServer::on_signal(int signo) { if (signo == SRS_SIGNAL_RELOAD) { srs_trace("reload config, signo=%d", signo); signal_reload = true; return; } #ifndef SRS_GPERF_MC if (signo == SRS_SIGNAL_REOPEN_LOG) { _srs_log->reopen(); if (handler) { handler->on_logrotate(); } srs_warn("reopen log file, signo=%d", signo); return; } #endif #ifdef SRS_GPERF_MC if (signo == SRS_SIGNAL_REOPEN_LOG) { signal_gmc_stop = true; srs_warn("for gmc, the SIGUSR1 used as SIGINT, signo=%d", signo); return; } #endif if (signo == SRS_SIGNAL_PERSISTENCE_CONFIG) { signal_persistence_config = true; return; } if (signo == SIGINT) { #ifdef SRS_GPERF_MC srs_trace("gmc is on, main cycle will terminate normally, signo=%d", signo); signal_gmc_stop = true; #else #ifdef SRS_MEM_WATCH srs_memory_report(); #endif #endif } // For K8S, force to gracefully quit for gray release or canary. // @see https://github.com/ossrs/srs/issues/1595#issuecomment-587473037 if (signo == SRS_SIGNAL_FAST_QUIT && _srs_config->is_force_grace_quit()) { srs_trace("force gracefully quit, signo=%d", signo); signo = SRS_SIGNAL_GRACEFULLY_QUIT; } if ((signo == SIGINT || signo == SRS_SIGNAL_FAST_QUIT) && !signal_fast_quit) { srs_trace("sig=%d, user terminate program, fast quit", signo); signal_fast_quit = true; return; } if (signo == SRS_SIGNAL_GRACEFULLY_QUIT && !signal_gracefully_quit) { srs_trace("sig=%d, user start gracefully quit", signo); signal_gracefully_quit = true; return; } } srs_error_t SrsServer::do_cycle() { srs_error_t err = srs_success; // find the max loop int max = srs_max(0, SRS_SYS_TIME_RESOLUTION_MS_TIMES); max = srs_max(max, SRS_SYS_RUSAGE_RESOLUTION_TIMES); max = srs_max(max, SRS_SYS_CPU_STAT_RESOLUTION_TIMES); max = srs_max(max, SRS_SYS_DISK_STAT_RESOLUTION_TIMES); max = srs_max(max, SRS_SYS_MEMINFO_RESOLUTION_TIMES); max = srs_max(max, SRS_SYS_PLATFORM_INFO_RESOLUTION_TIMES); max = srs_max(max, SRS_SYS_NETWORK_DEVICE_RESOLUTION_TIMES); max = srs_max(max, SRS_SYS_NETWORK_RTMP_SERVER_RESOLUTION_TIMES); // for asprocess. bool asprocess = _srs_config->get_asprocess(); // the daemon thread, update the time cache // TODO: FIXME: use SrsHourGlass. while (true) { if (handler && (err = handler->on_cycle()) != srs_success) { return srs_error_wrap(err, "handle callback"); } // the interval in config. int heartbeat_max_resolution = (int)(_srs_config->get_heartbeat_interval() / SRS_SYS_CYCLE_INTERVAL); // dynamic fetch the max. int dynamic_max = srs_max(max, heartbeat_max_resolution); for (int i = 0; i < dynamic_max; i++) { srs_usleep(SRS_SYS_CYCLE_INTERVAL); // asprocess check. if (asprocess && ::getppid() != ppid) { return srs_error_new(ERROR_ASPROCESS_PPID, "asprocess ppid changed from %d to %d", ppid, ::getppid()); } // gracefully quit for SIGINT or SIGTERM or SIGQUIT. if (signal_fast_quit || signal_gracefully_quit) { srs_trace("cleanup for quit signal fast=%d, grace=%d", signal_fast_quit, signal_gracefully_quit); return err; } // for gperf heap checker, // @see: research/gperftools/heap-checker/heap_checker.cc // if user interrupt the program, exit to check mem leak. // but, if gperf, use reload to ensure main return normally, // because directly exit will cause core-dump. #ifdef SRS_GPERF_MC if (signal_gmc_stop) { srs_warn("gmc got singal to stop server."); return err; } #endif // do persistence config to file. if (signal_persistence_config) { signal_persistence_config = false; srs_info("get signal to persistence config to file."); if ((err = _srs_config->persistence()) != srs_success) { return srs_error_wrap(err, "config persistence to file"); } srs_trace("persistence config to file success."); } // do reload the config. if (signal_reload) { signal_reload = false; srs_info("get signal to reload the config."); if ((err = _srs_config->reload()) != srs_success) { return srs_error_wrap(err, "config reload"); } srs_trace("reload config success."); } // notice the stream sources to cycle. if ((err = _srs_sources->cycle()) != srs_success) { return srs_error_wrap(err, "source cycle"); } // update the cache time if ((i % SRS_SYS_TIME_RESOLUTION_MS_TIMES) == 0) { srs_info("update current time cache."); srs_update_system_time(); } if ((i % SRS_SYS_RUSAGE_RESOLUTION_TIMES) == 0) { srs_info("update resource info, rss."); srs_update_system_rusage(); } if ((i % SRS_SYS_CPU_STAT_RESOLUTION_TIMES) == 0) { srs_info("update cpu info, cpu usage."); srs_update_proc_stat(); } if ((i % SRS_SYS_DISK_STAT_RESOLUTION_TIMES) == 0) { srs_info("update disk info, disk iops."); srs_update_disk_stat(); } if ((i % SRS_SYS_MEMINFO_RESOLUTION_TIMES) == 0) { srs_info("update memory info, usage/free."); srs_update_meminfo(); } if ((i % SRS_SYS_PLATFORM_INFO_RESOLUTION_TIMES) == 0) { srs_info("update platform info, uptime/load."); srs_update_platform_info(); } if ((i % SRS_SYS_NETWORK_DEVICE_RESOLUTION_TIMES) == 0) { srs_info("update network devices info."); srs_update_network_devices(); } if ((i % SRS_SYS_NETWORK_RTMP_SERVER_RESOLUTION_TIMES) == 0) { srs_info("update network server kbps info."); resample_kbps(); } if (_srs_config->get_heartbeat_enabled()) { if ((i % heartbeat_max_resolution) == 0) { srs_info("do http heartbeat, for internal server to report."); http_heartbeat->heartbeat(); } } srs_info("server main thread loop"); } } return err; } srs_error_t SrsServer::listen_rtmp() { srs_error_t err = srs_success; // stream service port. std::vector ip_ports = _srs_config->get_listens(); srs_assert((int)ip_ports.size() > 0); close_listeners(SrsListenerRtmpStream); for (int i = 0; i < (int)ip_ports.size(); i++) { SrsListener* listener = new SrsBufferListener(this, SrsListenerRtmpStream); listeners.push_back(listener); int port; string ip; srs_parse_endpoint(ip_ports[i], ip, port); if ((err = listener->listen(ip, port)) != srs_success) { srs_error_wrap(err, "rtmp listen %s:%d", ip.c_str(), port); } } return err; } srs_error_t SrsServer::listen_http_api() { srs_error_t err = srs_success; close_listeners(SrsListenerHttpApi); if (_srs_config->get_http_api_enabled()) { SrsListener* listener = new SrsBufferListener(this, SrsListenerHttpApi); listeners.push_back(listener); std::string ep = _srs_config->get_http_api_listen(); std::string ip; int port; srs_parse_endpoint(ep, ip, port); if ((err = listener->listen(ip, port)) != srs_success) { return srs_error_wrap(err, "http api listen %s:%d", ip.c_str(), port); } } return err; } srs_error_t SrsServer::listen_http_stream() { srs_error_t err = srs_success; close_listeners(SrsListenerHttpStream); if (_srs_config->get_http_stream_enabled()) { SrsListener* listener = new SrsBufferListener(this, SrsListenerHttpStream); listeners.push_back(listener); std::string ep = _srs_config->get_http_stream_listen(); std::string ip; int port; srs_parse_endpoint(ep, ip, port); if ((err = listener->listen(ip, port)) != srs_success) { return srs_error_wrap(err, "http stream listen %s:%d", ip.c_str(), port); } } return err; } #ifdef SRS_GB28181 srs_error_t SrsServer::listen_gb28181_sip(SrsConfDirective* stream_caster) { srs_error_t err = srs_success; SrsListener* sip_listener = NULL; sip_listener = new SrsGb28181Listener(this, SrsListenerGb28181Sip, stream_caster); int port = _srs_config->get_stream_caster_gb28181_sip_listen(stream_caster); if (port <= 0) { return srs_error_new(ERROR_STREAM_CASTER_PORT, "invalid sip port=%d", port); } srs_assert(sip_listener != NULL); listeners.push_back(sip_listener); // TODO: support listen at <[ip:]port> if ((err = sip_listener->listen(srs_any_address_for_listener(), port)) != srs_success) { return srs_error_wrap(err, "listen at %d", port); } return err; } #endif srs_error_t SrsServer::listen_stream_caster() { srs_error_t err = srs_success; close_listeners(SrsListenerMpegTsOverUdp); std::vector::iterator it; std::vector stream_casters = _srs_config->get_stream_casters(); for (it = stream_casters.begin(); it != stream_casters.end(); ++it) { SrsConfDirective* stream_caster = *it; if (!_srs_config->get_stream_caster_enabled(stream_caster)) { continue; } SrsListener* listener = NULL; std::string caster = _srs_config->get_stream_caster_engine(stream_caster); if (srs_stream_caster_is_udp(caster)) { listener = new SrsUdpCasterListener(this, SrsListenerMpegTsOverUdp, stream_caster); } else if (srs_stream_caster_is_rtsp(caster)) { listener = new SrsRtspListener(this, SrsListenerRtsp, stream_caster); } else if (srs_stream_caster_is_flv(caster)) { listener = new SrsHttpFlvListener(this, SrsListenerFlv, stream_caster); } else if (srs_stream_caster_is_gb28181(caster)) { #ifdef SRS_GB28181 //init global gb28181 manger if (_srs_gb28181 == NULL){ _srs_gb28181 = new SrsGb28181Manger(this, stream_caster); if ((err = _srs_gb28181->initialize()) != srs_success){ return err; } } //sip listener if (_srs_config->get_stream_caster_gb28181_sip_enable(stream_caster)){ if ((err = listen_gb28181_sip(stream_caster)) != srs_success){ return err; } } //gb28181 stream listener listener = new SrsGb28181Listener(this, SrsListenerGb28181RtpMux, stream_caster); #else srs_warn("gb28181 is disabled, please enable it by: ./configure --with-gb28181"); continue; #endif } else { return srs_error_new(ERROR_STREAM_CASTER_ENGINE, "invalid caster %s", caster.c_str()); } srs_assert(listener != NULL); listeners.push_back(listener); int port = _srs_config->get_stream_caster_listen(stream_caster); if (port <= 0) { return srs_error_new(ERROR_STREAM_CASTER_PORT, "invalid port=%d", port); } // TODO: support listen at <[ip:]port> if ((err = listener->listen(srs_any_address_for_listener(), port)) != srs_success) { return srs_error_wrap(err, "listen at %d", port); } } return err; } void SrsServer::close_listeners(SrsListenerType type) { std::vector::iterator it; for (it = listeners.begin(); it != listeners.end();) { SrsListener* listener = *it; if (listener->listen_type() != type) { ++it; continue; } srs_freep(listener); it = listeners.erase(it); } } void SrsServer::resample_kbps() { SrsStatistic* stat = SrsStatistic::instance(); // collect delta from all clients. for (int i = 0; i < (int)conn_manager->size(); i++) { SrsTcpConnection* conn = dynamic_cast(conn_manager->at(i)); // add delta of connection to server kbps., // for next sample() of server kbps can get the stat. stat->kbps_add_delta(conn); } // TODO: FXME: support all other connections. // sample the kbps, get the stat. SrsKbps* kbps = stat->kbps_sample(); srs_update_rtmp_server((int)conn_manager->size(), kbps); } srs_error_t SrsServer::accept_client(SrsListenerType type, srs_netfd_t stfd) { srs_error_t err = srs_success; SrsTcpConnection* conn = NULL; if ((err = fd2conn(type, stfd, &conn)) != srs_success) { if (srs_error_code(err) == ERROR_SOCKET_GET_PEER_IP && _srs_config->empty_ip_ok()) { srs_close_stfd(stfd); srs_error_reset(err); return srs_success; } return srs_error_wrap(err, "fd2conn"); } srs_assert(conn); // directly enqueue, the cycle thread will remove the client. conn_manager->add(conn); // cycle will start process thread and when finished remove the client. // @remark never use the conn, for it maybe destroyed. if ((err = conn->start()) != srs_success) { return srs_error_wrap(err, "start conn coroutine"); } return err; } SrsHttpServeMux* SrsServer::api_server() { return http_api_mux; } srs_error_t SrsServer::fd2conn(SrsListenerType type, srs_netfd_t stfd, SrsTcpConnection** pconn) { srs_error_t err = srs_success; int fd = srs_netfd_fileno(stfd); string ip = srs_get_peer_ip(fd); int port = srs_get_peer_port(fd); // for some keep alive application, for example, the keepalived, // will send some tcp packet which we cann't got the ip, // we just ignore it. if (ip.empty()) { return srs_error_new(ERROR_SOCKET_GET_PEER_IP, "ignore empty ip, fd=%d", fd); } // check connection limitation. int max_connections = _srs_config->get_max_connections(); if (handler && (err = handler->on_accept_client(max_connections, (int)conn_manager->size())) != srs_success) { return srs_error_wrap(err, "drop client fd=%d, ip=%s:%d, max=%d, cur=%d for err: %s", fd, ip.c_str(), port, max_connections, (int)conn_manager->size(), srs_error_desc(err).c_str()); } if ((int)conn_manager->size() >= max_connections) { return srs_error_new(ERROR_EXCEED_CONNECTIONS, "drop fd=%d, ip=%s:%d, max=%d, cur=%d for exceed connection limits", fd, ip.c_str(), port, max_connections, (int)conn_manager->size()); } // avoid fd leak when fork. // @see https://github.com/ossrs/srs/issues/518 if (true) { int val; if ((val = fcntl(fd, F_GETFD, 0)) < 0) { return srs_error_new(ERROR_SYSTEM_PID_GET_FILE_INFO, "fnctl F_GETFD error! fd=%d", fd); } val |= FD_CLOEXEC; if (fcntl(fd, F_SETFD, val) < 0) { return srs_error_new(ERROR_SYSTEM_PID_SET_FILE_INFO, "fcntl F_SETFD error! fd=%d", fd); } } if (type == SrsListenerRtmpStream) { *pconn = new SrsRtmpConn(this, stfd, ip, port); } else if (type == SrsListenerHttpApi) { *pconn = new SrsHttpApi(this, stfd, http_api_mux, ip, port); } else if (type == SrsListenerHttpStream) { *pconn = new SrsResponseOnlyHttpConn(this, stfd, http_server, ip, port); } else { srs_warn("close for no service handler. fd=%d, ip=%s:%d", fd, ip.c_str(), port); srs_close_stfd(stfd); return err; } return err; } void SrsServer::remove(ISrsResource* c) { SrsTcpConnection* conn = dynamic_cast(c); SrsStatistic* stat = SrsStatistic::instance(); stat->kbps_add_delta(conn); stat->on_disconnect(conn->srs_id()); // use manager to free it async. conn_manager->remove(c); } srs_error_t SrsServer::on_reload_listen() { srs_error_t err = srs_success; if ((err = listen()) != srs_success) { return srs_error_wrap(err, "reload listen"); } return err; } srs_error_t SrsServer::on_reload_pid() { srs_error_t err = srs_success; if (pid_fd > 0) { ::close(pid_fd); pid_fd = -1; } if ((err = acquire_pid_file()) != srs_success) { return srs_error_wrap(err, "reload pid"); } return err; } srs_error_t SrsServer::on_reload_vhost_added(std::string vhost) { srs_error_t err = srs_success; if (!_srs_config->get_vhost_http_enabled(vhost)) { return err; } // TODO: FIXME: should handle the event in SrsHttpStaticServer if ((err = on_reload_vhost_http_updated()) != srs_success) { return srs_error_wrap(err, "reload vhost added"); } return err; } srs_error_t SrsServer::on_reload_vhost_removed(std::string /*vhost*/) { srs_error_t err = srs_success; // TODO: FIXME: should handle the event in SrsHttpStaticServer if ((err = on_reload_vhost_http_updated()) != srs_success) { return srs_error_wrap(err, "reload vhost removed"); } return err; } srs_error_t SrsServer::on_reload_http_api_enabled() { srs_error_t err = srs_success; if ((err = listen_http_api()) != srs_success) { return srs_error_wrap(err, "reload http_api"); } return err; } srs_error_t SrsServer::on_reload_http_api_disabled() { close_listeners(SrsListenerHttpApi); return srs_success; } srs_error_t SrsServer::on_reload_http_stream_enabled() { srs_error_t err = srs_success; if ((err = listen_http_stream()) != srs_success) { return srs_error_wrap(err, "reload http_stream enabled"); } return err; } srs_error_t SrsServer::on_reload_http_stream_disabled() { close_listeners(SrsListenerHttpStream); return srs_success; } // TODO: FIXME: rename to http_remux srs_error_t SrsServer::on_reload_http_stream_updated() { srs_error_t err = srs_success; if ((err = on_reload_http_stream_enabled()) != srs_success) { return srs_error_wrap(err, "reload http_stream updated"); } // TODO: FIXME: should handle the event in SrsHttpStaticServer if ((err = on_reload_vhost_http_updated()) != srs_success) { return srs_error_wrap(err, "reload http_stream updated"); } return err; } srs_error_t SrsServer::on_publish(SrsSource* s, SrsRequest* r) { srs_error_t err = srs_success; if ((err = http_server->http_mount(s, r)) != srs_success) { return srs_error_wrap(err, "http mount"); } SrsCoWorkers* coworkers = SrsCoWorkers::instance(); if ((err = coworkers->on_publish(s, r)) != srs_success) { return srs_error_wrap(err, "coworkers"); } return err; } void SrsServer::on_unpublish(SrsSource* s, SrsRequest* r) { http_server->http_unmount(s, r); SrsCoWorkers* coworkers = SrsCoWorkers::instance(); coworkers->on_unpublish(s, r); }