mirror of
https://github.com/ossrs/srs.git
synced 2025-03-09 15:49:59 +00:00
fix #277, refine http server refer to go http-framework. 2.0.98
This commit is contained in:
winlin
parent
9bbbaad288
commit
cbed2bbca4
9 changed files with 148 additions and 1116 deletions
|
|
@ -40,20 +40,16 @@ using namespace std;
|
|||
#include <srs_app_config.hpp>
|
||||
#include <srs_kernel_flv.hpp>
|
||||
#include <srs_kernel_utility.hpp>
|
||||
#include <srs_kernel_file.hpp>
|
||||
|
||||
#define SRS_HTTP_DEFAULT_PAGE "index.html"
|
||||
|
||||
SrsHttpRoot::SrsHttpRoot()
|
||||
{
|
||||
// TODO: FIXME: support reload vhosts.
|
||||
}
|
||||
|
||||
SrsHttpRoot::~SrsHttpRoot()
|
||||
SrsHttpServer::SrsHttpServer()
|
||||
{
|
||||
}
|
||||
|
||||
int SrsHttpRoot::initialize()
|
||||
SrsHttpServer::~SrsHttpServer()
|
||||
{
|
||||
}
|
||||
|
||||
int SrsHttpServer::initialize()
|
||||
{
|
||||
int ret = ERROR_SUCCESS;
|
||||
|
||||
|
|
@ -76,7 +72,10 @@ int SrsHttpRoot::initialize()
|
|||
std::string mount = _srs_config->get_vhost_http_mount(vhost);
|
||||
std::string dir = _srs_config->get_vhost_http_dir(vhost);
|
||||
|
||||
handlers.push_back(new SrsHttpVhost(vhost, mount, dir));
|
||||
if ((ret = mux.handle(mount, new SrsGoHttpFileServer(dir))) != ERROR_SUCCESS) {
|
||||
srs_error("http: mount dir=%s for vhost=%s failed. ret=%d", dir.c_str(), vhost.c_str(), ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
if (mount == "/") {
|
||||
default_root_exists = true;
|
||||
|
|
@ -85,299 +84,9 @@ int SrsHttpRoot::initialize()
|
|||
|
||||
if (!default_root_exists) {
|
||||
// add root
|
||||
handlers.push_back(new SrsHttpVhost(
|
||||
"__http__", "/", _srs_config->get_http_stream_dir()));
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
int SrsHttpRoot::best_match(const char* path, int length, SrsHttpHandlerMatch** ppmatch)
|
||||
{
|
||||
int ret = ERROR_SUCCESS;
|
||||
|
||||
// find the best matched child handler.
|
||||
std::vector<SrsHttpHandler*>::iterator it;
|
||||
for (it = handlers.begin(); it != handlers.end(); ++it) {
|
||||
SrsHttpHandler* h = *it;
|
||||
|
||||
// search all child handlers.
|
||||
h->best_match(path, length, ppmatch);
|
||||
}
|
||||
|
||||
// if already matched by child, return.
|
||||
if (*ppmatch) {
|
||||
return ret;
|
||||
}
|
||||
|
||||
// not matched, error.
|
||||
return ERROR_HTTP_HANDLER_MATCH_URL;
|
||||
}
|
||||
|
||||
bool SrsHttpRoot::is_handler_valid(SrsHttpMessage* /*req*/, int& status_code, std::string& reason_phrase)
|
||||
{
|
||||
status_code = SRS_CONSTS_HTTP_InternalServerError;
|
||||
reason_phrase = SRS_CONSTS_HTTP_InternalServerError_str;
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
int SrsHttpRoot::do_process_request(SrsStSocket* /*skt*/, SrsHttpMessage* /*req*/)
|
||||
{
|
||||
int ret = ERROR_SUCCESS;
|
||||
return ret;
|
||||
}
|
||||
|
||||
SrsHttpVhost::SrsHttpVhost(std::string vhost, std::string mount, std::string dir)
|
||||
{
|
||||
_vhost = vhost;
|
||||
_mount = mount;
|
||||
_dir = dir;
|
||||
}
|
||||
|
||||
SrsHttpVhost::~SrsHttpVhost()
|
||||
{
|
||||
}
|
||||
|
||||
bool SrsHttpVhost::can_handle(const char* path, int length, const char** /*pchild*/)
|
||||
{
|
||||
return srs_path_like(_mount.c_str(), path, length);
|
||||
}
|
||||
|
||||
bool SrsHttpVhost::is_handler_valid(SrsHttpMessage* req, int& status_code, std::string& reason_phrase)
|
||||
{
|
||||
std::string fullpath = get_request_file(req);
|
||||
|
||||
if (::access(fullpath.c_str(), F_OK | R_OK) < 0) {
|
||||
srs_warn("check file %s does not exists", fullpath.c_str());
|
||||
|
||||
status_code = SRS_CONSTS_HTTP_NotFound;
|
||||
reason_phrase = SRS_CONSTS_HTTP_NotFound_str;
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
int SrsHttpVhost::do_process_request(SrsStSocket* skt, SrsHttpMessage* req)
|
||||
{
|
||||
std::string fullpath = get_request_file(req);
|
||||
|
||||
// TODO: FIXME: support mp4, @see https://github.com/winlinvip/simple-rtmp-server/issues/174
|
||||
if (srs_string_ends_with(fullpath, ".ts")) {
|
||||
return response_ts_file(skt, req, fullpath);
|
||||
} else if (srs_string_ends_with(fullpath, ".flv") || srs_string_ends_with(fullpath, ".fhv")) {
|
||||
std::string start = req->query_get("start");
|
||||
if (start.empty()) {
|
||||
return response_flv_file(skt, req, fullpath);
|
||||
}
|
||||
|
||||
int offset = ::atoi(start.c_str());
|
||||
if (offset <= 0) {
|
||||
return response_flv_file(skt, req, fullpath);
|
||||
}
|
||||
|
||||
return response_flv_file2(skt, req, fullpath, offset);
|
||||
}
|
||||
|
||||
return response_regular_file(skt, req, fullpath);
|
||||
}
|
||||
|
||||
int SrsHttpVhost::response_regular_file(SrsStSocket* skt, SrsHttpMessage* req, string fullpath)
|
||||
{
|
||||
int ret = ERROR_SUCCESS;
|
||||
|
||||
SrsFileReader fs;
|
||||
|
||||
if ((ret = fs.open(fullpath)) != ERROR_SUCCESS) {
|
||||
srs_warn("open file %s failed, ret=%d", fullpath.c_str(), ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
int64_t length = fs.filesize();
|
||||
|
||||
char* buf = new char[length];
|
||||
SrsAutoFree(char, buf);
|
||||
|
||||
if ((ret = fs.read(buf, length, NULL)) != ERROR_SUCCESS) {
|
||||
srs_warn("read file %s failed, ret=%d", fullpath.c_str(), ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
std::string str;
|
||||
str.append(buf, length);
|
||||
|
||||
if (srs_string_ends_with(fullpath, ".ts")) {
|
||||
return res_mpegts(skt, req, str);
|
||||
} else if (srs_string_ends_with(fullpath, ".m3u8")) {
|
||||
return res_m3u8(skt, req, str);
|
||||
} else if (srs_string_ends_with(fullpath, ".xml")) {
|
||||
return res_xml(skt, req, str);
|
||||
} else if (srs_string_ends_with(fullpath, ".js")) {
|
||||
return res_javascript(skt, req, str);
|
||||
} else if (srs_string_ends_with(fullpath, ".json")) {
|
||||
return res_json(skt, req, str);
|
||||
} else if (srs_string_ends_with(fullpath, ".swf")) {
|
||||
return res_swf(skt, req, str);
|
||||
} else if (srs_string_ends_with(fullpath, ".css")) {
|
||||
return res_css(skt, req, str);
|
||||
} else if (srs_string_ends_with(fullpath, ".ico")) {
|
||||
return res_ico(skt, req, str);
|
||||
} else {
|
||||
return res_text(skt, req, str);
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
int SrsHttpVhost::response_flv_file(SrsStSocket* skt, SrsHttpMessage* req, string fullpath)
|
||||
{
|
||||
int ret = ERROR_SUCCESS;
|
||||
|
||||
SrsFileReader fs;
|
||||
|
||||
// TODO: FIXME: use more advance cache.
|
||||
if ((ret = fs.open(fullpath)) != ERROR_SUCCESS) {
|
||||
srs_warn("open file %s failed, ret=%d", fullpath.c_str(), ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
int64_t length = fs.filesize();
|
||||
|
||||
// write http header for ts.
|
||||
std::stringstream ss;
|
||||
|
||||
res_status_line(ss)->res_content_type_flv(ss)
|
||||
->res_content_length(ss, (int)length);
|
||||
|
||||
if (req->requires_crossdomain()) {
|
||||
res_enable_crossdomain(ss);
|
||||
}
|
||||
|
||||
res_header_eof(ss);
|
||||
|
||||
// flush http header to peer
|
||||
if ((ret = res_flush(skt, ss)) != ERROR_SUCCESS) {
|
||||
return ret;
|
||||
}
|
||||
|
||||
// write body.
|
||||
int64_t left = length;
|
||||
char* buf = req->http_ts_send_buffer();
|
||||
|
||||
while (left > 0) {
|
||||
ssize_t nread = -1;
|
||||
if ((ret = fs.read(buf, __SRS_HTTP_TS_SEND_BUFFER_SIZE, &nread)) != ERROR_SUCCESS) {
|
||||
srs_warn("read file %s failed, ret=%d", fullpath.c_str(), ret);
|
||||
break;
|
||||
}
|
||||
|
||||
left -= nread;
|
||||
if ((ret = skt->write(buf, nread, NULL)) != ERROR_SUCCESS) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
int SrsHttpVhost::response_flv_file2(SrsStSocket* skt, SrsHttpMessage* req, string fullpath, int offset)
|
||||
{
|
||||
int ret = ERROR_SUCCESS;
|
||||
|
||||
SrsFileReader fs;
|
||||
|
||||
// open flv file
|
||||
if ((ret = fs.open(fullpath)) != ERROR_SUCCESS) {
|
||||
return ret;
|
||||
}
|
||||
|
||||
if (offset > fs.filesize()) {
|
||||
ret = ERROR_HTTP_FLV_OFFSET_OVERFLOW;
|
||||
srs_warn("http flv streaming %s overflow. size=%"PRId64", offset=%d, ret=%d",
|
||||
fullpath.c_str(), fs.filesize(), offset, ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
SrsFlvVodStreamDecoder ffd;
|
||||
|
||||
// open fast decoder
|
||||
if ((ret = ffd.initialize(&fs)) != ERROR_SUCCESS) {
|
||||
return ret;
|
||||
}
|
||||
|
||||
// save header, send later.
|
||||
char flv_header[13];
|
||||
|
||||
// send flv header
|
||||
if ((ret = ffd.read_header_ext(flv_header)) != ERROR_SUCCESS) {
|
||||
return ret;
|
||||
}
|
||||
|
||||
// save sequence header, send later
|
||||
char* sh_data = NULL;
|
||||
int sh_size = 0;
|
||||
|
||||
if (true) {
|
||||
// send sequence header
|
||||
int64_t start = 0;
|
||||
if ((ret = ffd.read_sequence_header_summary(&start, &sh_size)) != ERROR_SUCCESS) {
|
||||
return ret;
|
||||
}
|
||||
if (sh_size <= 0) {
|
||||
ret = ERROR_HTTP_FLV_SEQUENCE_HEADER;
|
||||
srs_warn("http flv streaming no sequence header. size=%d, ret=%d", sh_size, ret);
|
||||
return ret;
|
||||
}
|
||||
}
|
||||
sh_data = new char[sh_size];
|
||||
SrsAutoFree(char, sh_data);
|
||||
if ((ret = fs.read(sh_data, sh_size, NULL)) != ERROR_SUCCESS) {
|
||||
return ret;
|
||||
}
|
||||
|
||||
// seek to data offset
|
||||
int64_t left = fs.filesize() - offset;
|
||||
|
||||
// write http header for ts.
|
||||
std::stringstream ss;
|
||||
|
||||
res_status_line(ss)->res_content_type_flv(ss)
|
||||
->res_content_length(ss, (int)(sizeof(flv_header) + sh_size + left));
|
||||
|
||||
if (req->requires_crossdomain()) {
|
||||
res_enable_crossdomain(ss);
|
||||
}
|
||||
|
||||
res_header_eof(ss);
|
||||
|
||||
// flush http header to peer
|
||||
if ((ret = res_flush(skt, ss)) != ERROR_SUCCESS) {
|
||||
return ret;
|
||||
}
|
||||
|
||||
if ((ret = skt->write(flv_header, sizeof(flv_header), NULL)) != ERROR_SUCCESS) {
|
||||
return ret;
|
||||
}
|
||||
if (sh_size > 0 && (ret = skt->write(sh_data, sh_size, NULL)) != ERROR_SUCCESS) {
|
||||
return ret;
|
||||
}
|
||||
|
||||
// write body.
|
||||
char* buf = req->http_ts_send_buffer();
|
||||
if ((ret = ffd.lseek(offset)) != ERROR_SUCCESS) {
|
||||
return ret;
|
||||
}
|
||||
|
||||
// send data
|
||||
while (left > 0) {
|
||||
ssize_t nread = -1;
|
||||
if ((ret = fs.read(buf, __SRS_HTTP_TS_SEND_BUFFER_SIZE, &nread)) != ERROR_SUCCESS) {
|
||||
return ret;
|
||||
}
|
||||
|
||||
left -= nread;
|
||||
if ((ret = skt->write(buf, nread, NULL)) != ERROR_SUCCESS) {
|
||||
std::string dir = _srs_config->get_http_stream_dir();
|
||||
if ((ret = mux.handle("/", new SrsGoHttpFileServer(dir))) != ERROR_SUCCESS) {
|
||||
srs_error("http: mount root dir=%s failed. ret=%d", dir.c_str(), ret);
|
||||
return ret;
|
||||
}
|
||||
}
|
||||
|
|
@ -385,103 +94,11 @@ int SrsHttpVhost::response_flv_file2(SrsStSocket* skt, SrsHttpMessage* req, stri
|
|||
return ret;
|
||||
}
|
||||
|
||||
int SrsHttpVhost::response_ts_file(SrsStSocket* skt, SrsHttpMessage* req, string fullpath)
|
||||
{
|
||||
int ret = ERROR_SUCCESS;
|
||||
|
||||
SrsFileReader fs;
|
||||
|
||||
// TODO: FIXME: use more advance cache.
|
||||
if ((ret = fs.open(fullpath)) != ERROR_SUCCESS) {
|
||||
srs_warn("open file %s failed, ret=%d", fullpath.c_str(), ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
int64_t length = fs.filesize();
|
||||
|
||||
// write http header for ts.
|
||||
std::stringstream ss;
|
||||
|
||||
res_status_line(ss)->res_content_type_mpegts(ss)
|
||||
->res_content_length(ss, (int)length);
|
||||
|
||||
if (req->requires_crossdomain()) {
|
||||
res_enable_crossdomain(ss);
|
||||
}
|
||||
|
||||
res_header_eof(ss);
|
||||
|
||||
// flush http header to peer
|
||||
if ((ret = res_flush(skt, ss)) != ERROR_SUCCESS) {
|
||||
return ret;
|
||||
}
|
||||
|
||||
// write body.
|
||||
int64_t left = length;
|
||||
char* buf = req->http_ts_send_buffer();
|
||||
|
||||
while (left > 0) {
|
||||
ssize_t nread = -1;
|
||||
if ((ret = fs.read(buf, __SRS_HTTP_TS_SEND_BUFFER_SIZE, &nread)) != ERROR_SUCCESS) {
|
||||
srs_warn("read file %s failed, ret=%d", fullpath.c_str(), ret);
|
||||
break;
|
||||
}
|
||||
|
||||
left -= nread;
|
||||
if ((ret = skt->write(buf, nread, NULL)) != ERROR_SUCCESS) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
string SrsHttpVhost::get_request_file(SrsHttpMessage* req)
|
||||
{
|
||||
std::string fullpath = _dir + "/";
|
||||
|
||||
// if root, directly use the matched url.
|
||||
if (_mount == "/") {
|
||||
// add the dir
|
||||
fullpath += req->match()->matched_url;
|
||||
// if file speicified, add the file.
|
||||
if (!req->match()->unmatched_url.empty()) {
|
||||
fullpath += "/" + req->match()->unmatched_url;
|
||||
}
|
||||
} else {
|
||||
// virtual path, ignore the virutal path.
|
||||
fullpath += req->match()->unmatched_url;
|
||||
}
|
||||
|
||||
// add default pages.
|
||||
if (srs_string_ends_with(fullpath, "/")) {
|
||||
fullpath += SRS_HTTP_DEFAULT_PAGE;
|
||||
}
|
||||
|
||||
return fullpath;
|
||||
}
|
||||
|
||||
string SrsHttpVhost::vhost()
|
||||
{
|
||||
return _vhost;
|
||||
}
|
||||
|
||||
string SrsHttpVhost::mount()
|
||||
{
|
||||
return _mount;
|
||||
}
|
||||
|
||||
string SrsHttpVhost::dir()
|
||||
{
|
||||
return _dir;
|
||||
}
|
||||
|
||||
SrsHttpConn::SrsHttpConn(SrsServer* srs_server, st_netfd_t client_stfd, SrsHttpHandler* _handler)
|
||||
: SrsConnection(srs_server, client_stfd)
|
||||
SrsHttpConn::SrsHttpConn(SrsServer* svr, st_netfd_t fd, SrsHttpServer* m)
|
||||
: SrsConnection(svr, fd)
|
||||
{
|
||||
parser = new SrsHttpParser();
|
||||
handler = _handler;
|
||||
requires_crossdomain = false;
|
||||
mux = m;
|
||||
}
|
||||
|
||||
SrsHttpConn::~SrsHttpConn()
|
||||
|
|
@ -538,7 +155,8 @@ int SrsHttpConn::do_cycle()
|
|||
SrsAutoFree(SrsHttpMessage, req);
|
||||
|
||||
// ok, handle http request.
|
||||
if ((ret = process_request(&skt, req)) != ERROR_SUCCESS) {
|
||||
SrsGoHttpResponseWriter writer(&skt);
|
||||
if ((ret = process_request(&writer, req)) != ERROR_SUCCESS) {
|
||||
return ret;
|
||||
}
|
||||
}
|
||||
|
|
@ -546,41 +164,21 @@ int SrsHttpConn::do_cycle()
|
|||
return ret;
|
||||
}
|
||||
|
||||
int SrsHttpConn::process_request(SrsStSocket* skt, SrsHttpMessage* req)
|
||||
int SrsHttpConn::process_request(ISrsGoHttpResponseWriter* w, SrsHttpMessage* r)
|
||||
{
|
||||
int ret = ERROR_SUCCESS;
|
||||
|
||||
srs_trace("HTTP %s %s, content-length=%"PRId64"",
|
||||
req->method_str().c_str(), req->url().c_str(), req->content_length());
|
||||
r->method_str().c_str(), r->url().c_str(), r->content_length());
|
||||
|
||||
// TODO: maybe need to parse the url.
|
||||
std::string url = req->path();
|
||||
|
||||
SrsHttpHandlerMatch* p = NULL;
|
||||
if ((ret = handler->best_match(url.data(), url.length(), &p)) != ERROR_SUCCESS) {
|
||||
srs_warn("failed to find the best match handler for url. ret=%d", ret);
|
||||
// use default server mux to serve http request.
|
||||
if ((ret = mux->mux.serve_http(w, r)) != ERROR_SUCCESS) {
|
||||
if (!srs_is_client_gracefully_close(ret)) {
|
||||
srs_error("serve http msg failed. ret=%d", ret);
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
// if success, p and pstart should be valid.
|
||||
srs_assert(p);
|
||||
srs_assert(p->handler);
|
||||
srs_assert(p->matched_url.length() <= url.length());
|
||||
srs_info("best match handler, matched_url=%s", p->matched_url.c_str());
|
||||
|
||||
req->set_match(p);
|
||||
req->set_requires_crossdomain(requires_crossdomain);
|
||||
|
||||
// use handler to process request.
|
||||
if ((ret = p->handler->process_request(skt, req)) != ERROR_SUCCESS) {
|
||||
srs_warn("handler failed to process http request. ret=%d", ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
if (req->requires_crossdomain()) {
|
||||
requires_crossdomain = true;
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue