1b6f97bd2d
This PR fixes a critical race condition in SRS source managers where multiple coroutines could create duplicate sources for the same stream. - **Atomic source creation**: Source lookup, creation, and pool insertion now happen atomically within lock scope - **Consistent interface**: Standardize on `ISrsRequest*` interface throughout codebase - **Handler simplification**: Remove `ISrsLiveSourceHandler*` parameter, obtain from global server instance --------- Co-authored-by: OSSRS-AI <winlinam@gmail.com>
113 lines
3.3 KiB
C++
113 lines
3.3 KiB
C++
//
|
|
// Copyright (c) 2013-2025 The SRS Authors
|
|
//
|
|
// SPDX-License-Identifier: MIT
|
|
//
|
|
|
|
#ifndef SRS_APP_HTTP_STATIC_HPP
|
|
#define SRS_APP_HTTP_STATIC_HPP
|
|
|
|
#include <srs_app_http_conn.hpp>
|
|
#include <srs_app_security.hpp>
|
|
#include <srs_core.hpp>
|
|
|
|
class ISrsFileReaderFactory;
|
|
|
|
// HLS virtual connection, build on query string ctx of hls stream.
|
|
class SrsHlsVirtualConn : public ISrsExpire
|
|
{
|
|
public:
|
|
srs_utime_t request_time;
|
|
ISrsRequest *req;
|
|
std::string ctx;
|
|
bool interrupt;
|
|
|
|
public:
|
|
SrsHlsVirtualConn();
|
|
virtual ~SrsHlsVirtualConn();
|
|
// Interface ISrsExpire.
|
|
public:
|
|
virtual void expire();
|
|
};
|
|
|
|
// Server HLS streaming.
|
|
class SrsHlsStream : public ISrsFastTimer
|
|
{
|
|
private:
|
|
// The period of validity of the ctx
|
|
std::map<std::string, SrsHlsVirtualConn *> map_ctx_info_;
|
|
|
|
public:
|
|
SrsHlsStream();
|
|
virtual ~SrsHlsStream();
|
|
|
|
public:
|
|
virtual srs_error_t serve_m3u8_ctx(ISrsHttpResponseWriter *w, ISrsHttpMessage *r, ISrsFileReaderFactory *factory, std::string fullpath, ISrsRequest *req, bool *served);
|
|
virtual void on_serve_ts_ctx(ISrsHttpResponseWriter *w, ISrsHttpMessage *r);
|
|
|
|
private:
|
|
srs_error_t serve_new_session(ISrsHttpResponseWriter *w, ISrsHttpMessage *r, ISrsRequest *req, std::string &ctx);
|
|
srs_error_t serve_exists_session(ISrsHttpResponseWriter *w, ISrsHttpMessage *r, ISrsFileReaderFactory *factory, std::string fullpath);
|
|
bool ctx_is_exist(std::string ctx);
|
|
void alive(std::string ctx, ISrsRequest *req);
|
|
srs_error_t http_hooks_on_play(ISrsRequest *req);
|
|
void http_hooks_on_stop(ISrsRequest *req);
|
|
bool is_interrupt(std::string id);
|
|
// interface ISrsFastTimer
|
|
private:
|
|
srs_error_t on_timer(srs_utime_t interval);
|
|
|
|
private:
|
|
SrsSecurity *security_;
|
|
};
|
|
|
|
// The Vod streaming, like FLV, MP4 or HLS streaming.
|
|
class SrsVodStream : public SrsHttpFileServer
|
|
{
|
|
private:
|
|
SrsHlsStream hls_;
|
|
|
|
public:
|
|
SrsVodStream(std::string root_dir);
|
|
virtual ~SrsVodStream();
|
|
|
|
protected:
|
|
// The flv vod stream supports flv?start=offset-bytes.
|
|
// For example, http://server/file.flv?start=10240
|
|
// server will write flv header and sequence header,
|
|
// then seek(10240) and response flv tag data.
|
|
virtual srs_error_t serve_flv_stream(ISrsHttpResponseWriter *w, ISrsHttpMessage *r, std::string fullpath, int64_t offset);
|
|
// Support mp4 with start and offset in query string.
|
|
virtual srs_error_t serve_mp4_stream(ISrsHttpResponseWriter *w, ISrsHttpMessage *r, std::string fullpath, int64_t start, int64_t end);
|
|
// Support HLS streaming with pseudo session id.
|
|
virtual srs_error_t serve_m3u8_ctx(ISrsHttpResponseWriter *w, ISrsHttpMessage *r, std::string fullpath);
|
|
// the ts file including: .ts .m4s init.mp4
|
|
virtual srs_error_t serve_ts_ctx(ISrsHttpResponseWriter *w, ISrsHttpMessage *r, std::string fullpath);
|
|
};
|
|
|
|
// The http static server instance,
|
|
// serve http static file and flv/mp4 vod stream.
|
|
class SrsHttpStaticServer : public ISrsReloadHandler
|
|
{
|
|
private:
|
|
SrsServer *server;
|
|
|
|
public:
|
|
SrsHttpServeMux mux;
|
|
|
|
public:
|
|
SrsHttpStaticServer(SrsServer *svr);
|
|
virtual ~SrsHttpStaticServer();
|
|
|
|
public:
|
|
virtual srs_error_t initialize();
|
|
|
|
private:
|
|
virtual srs_error_t mount_vhost(std::string vhost, std::string &pmount);
|
|
// Interface ISrsReloadHandler.
|
|
public:
|
|
virtual srs_error_t on_reload_vhost_added(std::string vhost);
|
|
};
|
|
|
|
#endif
|