mirror of
https://github.com/ossrs/srs.git
synced 2025-03-09 15:49:59 +00:00
refine http framework, use http message
This commit is contained in:
parent
e70609cea3
commit
ec6d696565
7 changed files with 347 additions and 318 deletions
0
trunk/conf/console.conf
Executable file → Normal file
0
trunk/conf/console.conf
Executable file → Normal file
|
@ -27,41 +27,263 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||||
|
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
|
|
||||||
|
using namespace std;
|
||||||
|
|
||||||
#include <srs_kernel_error.hpp>
|
#include <srs_kernel_error.hpp>
|
||||||
#include <srs_kernel_log.hpp>
|
#include <srs_kernel_log.hpp>
|
||||||
#include <srs_app_socket.hpp>
|
#include <srs_app_socket.hpp>
|
||||||
|
|
||||||
#define SRS_DEFAULT_HTTP_PORT 80
|
#define SRS_DEFAULT_HTTP_PORT 80
|
||||||
|
|
||||||
|
#define SRS_HTTP_HEADER_BUFFER 1024
|
||||||
|
|
||||||
SrsHttpMessage::SrsHttpMessage()
|
SrsHttpMessage::SrsHttpMessage()
|
||||||
{
|
{
|
||||||
body = new SrsBuffer();
|
_body = new SrsBuffer();
|
||||||
state = SrsHttpParseStateInit;
|
_state = SrsHttpParseStateInit;
|
||||||
}
|
}
|
||||||
|
|
||||||
SrsHttpMessage::~SrsHttpMessage()
|
SrsHttpMessage::~SrsHttpMessage()
|
||||||
{
|
{
|
||||||
srs_freep(body);
|
srs_freep(_body);
|
||||||
}
|
}
|
||||||
|
|
||||||
void SrsHttpMessage::reset()
|
void SrsHttpMessage::reset()
|
||||||
{
|
{
|
||||||
state = SrsHttpParseStateInit;
|
_state = SrsHttpParseStateInit;
|
||||||
body->clear();
|
_body->clear();
|
||||||
url = "";
|
_url = "";
|
||||||
}
|
}
|
||||||
|
|
||||||
bool SrsHttpMessage::is_complete()
|
bool SrsHttpMessage::is_complete()
|
||||||
{
|
{
|
||||||
return state == SrsHttpParseStateComplete;
|
return _state == SrsHttpParseStateComplete;
|
||||||
|
}
|
||||||
|
|
||||||
|
u_int8_t SrsHttpMessage::method()
|
||||||
|
{
|
||||||
|
return (u_int8_t)_header.method;
|
||||||
|
}
|
||||||
|
|
||||||
|
string SrsHttpMessage::url()
|
||||||
|
{
|
||||||
|
return _url;
|
||||||
|
}
|
||||||
|
|
||||||
|
string SrsHttpMessage::body()
|
||||||
|
{
|
||||||
|
std::string b;
|
||||||
|
|
||||||
|
if (_body && !_body->empty()) {
|
||||||
|
b.append(_body->bytes(), _body->size());
|
||||||
|
}
|
||||||
|
|
||||||
|
return b;
|
||||||
|
}
|
||||||
|
|
||||||
|
int64_t SrsHttpMessage::body_size()
|
||||||
|
{
|
||||||
|
return (int64_t)_body->size();
|
||||||
|
}
|
||||||
|
|
||||||
|
int64_t SrsHttpMessage::content_length()
|
||||||
|
{
|
||||||
|
return _header.content_length;
|
||||||
|
}
|
||||||
|
|
||||||
|
void SrsHttpMessage::set_url(std::string url)
|
||||||
|
{
|
||||||
|
_url = url;
|
||||||
|
}
|
||||||
|
|
||||||
|
void SrsHttpMessage::set_state(SrsHttpParseState state)
|
||||||
|
{
|
||||||
|
_state = state;
|
||||||
|
}
|
||||||
|
|
||||||
|
void SrsHttpMessage::set_header(http_parser* header)
|
||||||
|
{
|
||||||
|
memcpy(&_header, header, sizeof(http_parser));
|
||||||
|
}
|
||||||
|
|
||||||
|
void SrsHttpMessage::append_body(const char* body, int length)
|
||||||
|
{
|
||||||
|
_body->append(body, length);
|
||||||
}
|
}
|
||||||
|
|
||||||
SrsHttpParser::SrsHttpParser()
|
SrsHttpParser::SrsHttpParser()
|
||||||
{
|
{
|
||||||
|
msg = NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
SrsHttpParser::~SrsHttpParser()
|
SrsHttpParser::~SrsHttpParser()
|
||||||
{
|
{
|
||||||
|
srs_freep(msg);
|
||||||
|
}
|
||||||
|
|
||||||
|
int SrsHttpParser::initialize(enum http_parser_type type)
|
||||||
|
{
|
||||||
|
int ret = ERROR_SUCCESS;
|
||||||
|
|
||||||
|
memset(&settings, 0, sizeof(settings));
|
||||||
|
settings.on_message_begin = on_message_begin;
|
||||||
|
settings.on_url = on_url;
|
||||||
|
settings.on_header_field = on_header_field;
|
||||||
|
settings.on_header_value = on_header_value;
|
||||||
|
settings.on_headers_complete = on_headers_complete;
|
||||||
|
settings.on_body = on_body;
|
||||||
|
settings.on_message_complete = on_message_complete;
|
||||||
|
|
||||||
|
http_parser_init(&parser, type);
|
||||||
|
// callback object ptr.
|
||||||
|
parser.data = (void*)this;
|
||||||
|
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
int SrsHttpParser::parse_message(SrsSocket* skt, SrsHttpMessage** ppmsg)
|
||||||
|
{
|
||||||
|
*ppmsg = NULL;
|
||||||
|
|
||||||
|
int ret = ERROR_SUCCESS;
|
||||||
|
|
||||||
|
// the msg must be always NULL
|
||||||
|
srs_assert(msg == NULL);
|
||||||
|
msg = new SrsHttpMessage();
|
||||||
|
|
||||||
|
// reset response header.
|
||||||
|
msg->reset();
|
||||||
|
|
||||||
|
// do parse
|
||||||
|
if ((ret = parse_message_imp(skt)) != ERROR_SUCCESS) {
|
||||||
|
if (!srs_is_client_gracefully_close(ret)) {
|
||||||
|
srs_error("parse http msg failed. ret=%d", ret);
|
||||||
|
}
|
||||||
|
srs_freep(msg);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
// parse ok, return the msg.
|
||||||
|
*ppmsg = msg;
|
||||||
|
msg = NULL;
|
||||||
|
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
int SrsHttpParser::parse_message_imp(SrsSocket* skt)
|
||||||
|
{
|
||||||
|
int ret = ERROR_SUCCESS;
|
||||||
|
|
||||||
|
// the msg should never be NULL
|
||||||
|
srs_assert(msg != NULL);
|
||||||
|
|
||||||
|
// parser header.
|
||||||
|
char buf[SRS_HTTP_HEADER_BUFFER];
|
||||||
|
for (;;) {
|
||||||
|
ssize_t nread;
|
||||||
|
if ((ret = skt->read(buf, (size_t)sizeof(buf), &nread)) != ERROR_SUCCESS) {
|
||||||
|
if (!srs_is_client_gracefully_close(ret)) {
|
||||||
|
srs_error("read body from server failed. ret=%d", ret);
|
||||||
|
}
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
ssize_t nparsed = http_parser_execute(&parser, &settings, buf, nread);
|
||||||
|
srs_info("read_size=%d, nparsed=%d", (int)nread, (int)nparsed);
|
||||||
|
|
||||||
|
// check header size.
|
||||||
|
if (msg->is_complete()) {
|
||||||
|
srs_trace("http request parsed, method=%d, url=%s, content-length=%"PRId64"",
|
||||||
|
msg->method(), msg->url().c_str(), msg->content_length());
|
||||||
|
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (nparsed != nread) {
|
||||||
|
ret = ERROR_HTTP_PARSE_HEADER;
|
||||||
|
srs_error("parse response error, parsed(%d)!=read(%d), ret=%d", (int)nparsed, (int)nread, ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
int SrsHttpParser::on_message_begin(http_parser* parser)
|
||||||
|
{
|
||||||
|
SrsHttpParser* obj = (SrsHttpParser*)parser->data;
|
||||||
|
obj->msg->set_state(SrsHttpParseStateStart);
|
||||||
|
|
||||||
|
srs_info("***MESSAGE BEGIN***");
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int SrsHttpParser::on_headers_complete(http_parser* parser)
|
||||||
|
{
|
||||||
|
SrsHttpParser* obj = (SrsHttpParser*)parser->data;
|
||||||
|
obj->msg->set_header(parser);
|
||||||
|
|
||||||
|
srs_info("***HEADERS COMPLETE***");
|
||||||
|
|
||||||
|
// see http_parser.c:1570, return 1 to skip body.
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int SrsHttpParser::on_message_complete(http_parser* parser)
|
||||||
|
{
|
||||||
|
SrsHttpParser* obj = (SrsHttpParser*)parser->data;
|
||||||
|
// save the parser when header parse completed.
|
||||||
|
obj->msg->set_state(SrsHttpParseStateComplete);
|
||||||
|
|
||||||
|
srs_info("***MESSAGE COMPLETE***\n");
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int SrsHttpParser::on_url(http_parser* parser, const char* at, size_t length)
|
||||||
|
{
|
||||||
|
SrsHttpParser* obj = (SrsHttpParser*)parser->data;
|
||||||
|
|
||||||
|
if (length > 0) {
|
||||||
|
std::string url;
|
||||||
|
|
||||||
|
url.append(at, (int)length);
|
||||||
|
|
||||||
|
obj->msg->set_url(url);
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_info("Method: %d, Url: %.*s", parser->method, (int)length, at);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int SrsHttpParser::on_header_field(http_parser* /*parser*/, const char* at, size_t length)
|
||||||
|
{
|
||||||
|
srs_info("Header field: %.*s", (int)length, at);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int SrsHttpParser::on_header_value(http_parser* /*parser*/, const char* at, size_t length)
|
||||||
|
{
|
||||||
|
srs_info("Header value: %.*s", (int)length, at);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int SrsHttpParser::on_body(http_parser* parser, const char* at, size_t length)
|
||||||
|
{
|
||||||
|
SrsHttpParser* obj = (SrsHttpParser*)parser->data;
|
||||||
|
|
||||||
|
if (length > 0) {
|
||||||
|
srs_assert(obj);
|
||||||
|
srs_assert(obj->msg);
|
||||||
|
|
||||||
|
obj->msg->append_body(at, (int)length);
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_info("Body: %.*s", (int)length, at);
|
||||||
|
|
||||||
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
SrsHttpUri::SrsHttpUri()
|
SrsHttpUri::SrsHttpUri()
|
||||||
|
|
|
@ -68,17 +68,43 @@ enum SrsHttpParseState {
|
||||||
*/
|
*/
|
||||||
class SrsHttpMessage
|
class SrsHttpMessage
|
||||||
{
|
{
|
||||||
public:
|
private:
|
||||||
std::string url;
|
/**
|
||||||
http_parser header;
|
* parsed url.
|
||||||
SrsBuffer* body;
|
*/
|
||||||
SrsHttpParseState state;
|
std::string _url;
|
||||||
|
/**
|
||||||
|
* parsed http header.
|
||||||
|
*/
|
||||||
|
http_parser _header;
|
||||||
|
/**
|
||||||
|
* body object, in bytes.
|
||||||
|
* @remark, user can get body in string by get_body().
|
||||||
|
*/
|
||||||
|
SrsBuffer* _body;
|
||||||
|
/**
|
||||||
|
* parser state
|
||||||
|
* @remark, user can use is_complete() to determine the state.
|
||||||
|
*/
|
||||||
|
SrsHttpParseState _state;
|
||||||
|
|
||||||
|
public:
|
||||||
SrsHttpMessage();
|
SrsHttpMessage();
|
||||||
virtual ~SrsHttpMessage();
|
virtual ~SrsHttpMessage();
|
||||||
|
|
||||||
|
public:
|
||||||
virtual void reset();
|
virtual void reset();
|
||||||
|
public:
|
||||||
virtual bool is_complete();
|
virtual bool is_complete();
|
||||||
|
virtual u_int8_t method();
|
||||||
|
virtual std::string url();
|
||||||
|
virtual std::string body();
|
||||||
|
virtual int64_t body_size();
|
||||||
|
virtual int64_t content_length();
|
||||||
|
virtual void set_url(std::string url);
|
||||||
|
virtual void set_state(SrsHttpParseState state);
|
||||||
|
virtual void set_header(http_parser* header);
|
||||||
|
virtual void append_body(const char* body, int length);
|
||||||
};
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -87,11 +113,39 @@ public:
|
||||||
*/
|
*/
|
||||||
class SrsHttpParser
|
class SrsHttpParser
|
||||||
{
|
{
|
||||||
|
private:
|
||||||
|
http_parser_settings settings;
|
||||||
|
http_parser parser;
|
||||||
|
SrsHttpMessage* msg;
|
||||||
public:
|
public:
|
||||||
SrsHttpParser();
|
SrsHttpParser();
|
||||||
virtual ~SrsHttpParser();
|
virtual ~SrsHttpParser();
|
||||||
public:
|
public:
|
||||||
//virtual int parse_requst(SrsHttpMessage** ppreq);
|
/**
|
||||||
|
* initialize the http parser with specified type,
|
||||||
|
* one parser can only parse request or response messages.
|
||||||
|
*/
|
||||||
|
virtual int initialize(enum http_parser_type type);
|
||||||
|
/**
|
||||||
|
* always parse a http message,
|
||||||
|
* that is, the *ppmsg always NOT-NULL when return success.
|
||||||
|
* or error and *ppmsg must be NULL.
|
||||||
|
* @remark, if success, *ppmsg always NOT-NULL, *ppmsg always is_complete().
|
||||||
|
*/
|
||||||
|
virtual int parse_message(SrsSocket* skt, SrsHttpMessage** ppmsg);
|
||||||
|
private:
|
||||||
|
/**
|
||||||
|
* parse the HTTP message to member field: msg.
|
||||||
|
*/
|
||||||
|
virtual int parse_message_imp(SrsSocket* skt);
|
||||||
|
private:
|
||||||
|
static int on_message_begin(http_parser* parser);
|
||||||
|
static int on_headers_complete(http_parser* parser);
|
||||||
|
static int on_message_complete(http_parser* parser);
|
||||||
|
static int on_url(http_parser* parser, const char* at, size_t length);
|
||||||
|
static int on_header_field(http_parser* parser, const char* at, size_t length);
|
||||||
|
static int on_header_value(http_parser* parser, const char* at, size_t length);
|
||||||
|
static int on_body(http_parser* parser, const char* at, size_t length);
|
||||||
};
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|
|
@ -33,18 +33,19 @@ using namespace std;
|
||||||
#include <srs_app_socket.hpp>
|
#include <srs_app_socket.hpp>
|
||||||
#include <srs_app_http.hpp>
|
#include <srs_app_http.hpp>
|
||||||
#include <srs_kernel_buffer.hpp>
|
#include <srs_kernel_buffer.hpp>
|
||||||
|
#include <srs_core_autofree.hpp>
|
||||||
|
|
||||||
#define SRS_HTTP_HEADER_BUFFER 1024
|
#define SRS_HTTP_HEADER_BUFFER 1024
|
||||||
|
|
||||||
SrsHttpConn::SrsHttpConn(SrsServer* srs_server, st_netfd_t client_stfd)
|
SrsHttpConn::SrsHttpConn(SrsServer* srs_server, st_netfd_t client_stfd)
|
||||||
: SrsConnection(srs_server, client_stfd)
|
: SrsConnection(srs_server, client_stfd)
|
||||||
{
|
{
|
||||||
req = new SrsHttpMessage();
|
parser = new SrsHttpParser();
|
||||||
}
|
}
|
||||||
|
|
||||||
SrsHttpConn::~SrsHttpConn()
|
SrsHttpConn::~SrsHttpConn()
|
||||||
{
|
{
|
||||||
srs_freep(req);
|
srs_freep(parser);
|
||||||
}
|
}
|
||||||
|
|
||||||
int SrsHttpConn::do_cycle()
|
int SrsHttpConn::do_cycle()
|
||||||
|
@ -56,32 +57,34 @@ int SrsHttpConn::do_cycle()
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
srs_trace("http get peer ip success. ip=%s", ip);
|
srs_trace("http get peer ip success. ip=%s", ip);
|
||||||
|
|
||||||
// setup http parser
|
|
||||||
http_parser_settings settings;
|
|
||||||
|
|
||||||
memset(&settings, 0, sizeof(settings));
|
// initialize parser
|
||||||
settings.on_message_begin = on_message_begin;
|
if ((ret = parser->initialize(HTTP_REQUEST)) != ERROR_SUCCESS) {
|
||||||
settings.on_url = on_url;
|
srs_error("initialize http parser failed. ret=%d", ret);
|
||||||
settings.on_header_field = on_header_field;
|
return ret;
|
||||||
settings.on_header_value = on_header_value;
|
}
|
||||||
settings.on_headers_complete = on_headers_complete;
|
|
||||||
settings.on_body = on_body;
|
|
||||||
settings.on_message_complete = on_message_complete;
|
|
||||||
|
|
||||||
http_parser parser;
|
|
||||||
http_parser_init(&parser, HTTP_REQUEST);
|
|
||||||
// callback object ptr.
|
|
||||||
parser.data = (void*)this;
|
|
||||||
|
|
||||||
// underlayer socket
|
// underlayer socket
|
||||||
SrsSocket skt(stfd);
|
SrsSocket skt(stfd);
|
||||||
|
|
||||||
|
// process http messages.
|
||||||
for (;;) {
|
for (;;) {
|
||||||
if ((ret = parse_request(&skt, &parser, &settings)) != ERROR_SUCCESS) {
|
SrsHttpMessage* req = NULL;
|
||||||
if (!srs_is_client_gracefully_close(ret)) {
|
|
||||||
srs_error("http client cycle failed. ret=%d", ret);
|
// get a http message
|
||||||
}
|
if ((ret = parser->parse_message(&skt, &req)) != ERROR_SUCCESS) {
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
// if SUCCESS, always NOT-NULL and completed message.
|
||||||
|
srs_assert(req);
|
||||||
|
srs_assert(req->is_complete());
|
||||||
|
|
||||||
|
// always free it in this scope.
|
||||||
|
SrsAutoFree(SrsHttpMessage, req, false);
|
||||||
|
|
||||||
|
// ok, handle http request.
|
||||||
|
if ((ret = process_request(&skt, req)) != ERROR_SUCCESS) {
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -89,50 +92,11 @@ int SrsHttpConn::do_cycle()
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
int SrsHttpConn::parse_request(SrsSocket* skt, http_parser* parser, http_parser_settings* settings)
|
int SrsHttpConn::process_request(SrsSocket* skt, SrsHttpMessage* req)
|
||||||
{
|
|
||||||
int ret = ERROR_SUCCESS;
|
|
||||||
|
|
||||||
// reset response header.
|
|
||||||
req->reset();
|
|
||||||
|
|
||||||
// parser header.
|
|
||||||
char buf[SRS_HTTP_HEADER_BUFFER];
|
|
||||||
for (;;) {
|
|
||||||
ssize_t nread;
|
|
||||||
if ((ret = skt->read(buf, (size_t)sizeof(buf), &nread)) != ERROR_SUCCESS) {
|
|
||||||
if (!srs_is_client_gracefully_close(ret)) {
|
|
||||||
srs_error("read body from server failed. ret=%d", ret);
|
|
||||||
}
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
ssize_t nparsed = http_parser_execute(parser, settings, buf, nread);
|
|
||||||
srs_info("read_size=%d, nparsed=%d", (int)nread, (int)nparsed);
|
|
||||||
|
|
||||||
// check header size.
|
|
||||||
if (req->is_complete()) {
|
|
||||||
srs_trace("http request parsed, method=%d, url=%s, content-length=%"PRId64"",
|
|
||||||
req->header.method, req->url.c_str(), req->header.content_length);
|
|
||||||
|
|
||||||
return process_request(skt);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (nparsed != nread) {
|
|
||||||
ret = ERROR_HTTP_PARSE_HEADER;
|
|
||||||
srs_error("parse response error, parsed(%d)!=read(%d), ret=%d", (int)nparsed, (int)nread, ret);
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
int SrsHttpConn::process_request(SrsSocket* skt)
|
|
||||||
{
|
{
|
||||||
int ret = ERROR_SUCCESS;
|
int ret = ERROR_SUCCESS;
|
||||||
|
|
||||||
if (req->header.method == HTTP_OPTIONS) {
|
if (req->method() == HTTP_OPTIONS) {
|
||||||
char data[] = "HTTP/1.1 200 OK" __CRLF
|
char data[] = "HTTP/1.1 200 OK" __CRLF
|
||||||
"Content-Length: 0"__CRLF
|
"Content-Length: 0"__CRLF
|
||||||
"Server: SRS/"RTMP_SIG_SRS_VERSION""__CRLF
|
"Server: SRS/"RTMP_SIG_SRS_VERSION""__CRLF
|
||||||
|
@ -149,14 +113,14 @@ int SrsHttpConn::process_request(SrsSocket* skt)
|
||||||
|
|
||||||
std::stringstream ss;
|
std::stringstream ss;
|
||||||
ss << "HTTP/1.1 200 OK " << __CRLF
|
ss << "HTTP/1.1 200 OK " << __CRLF
|
||||||
<< "Content-Length: "<< tilte.length() + req->body->size() << __CRLF
|
<< "Content-Length: "<< tilte.length() + req->body_size() << __CRLF
|
||||||
<< "Server: SRS/"RTMP_SIG_SRS_VERSION"" << __CRLF
|
<< "Server: SRS/"RTMP_SIG_SRS_VERSION"" << __CRLF
|
||||||
<< "Allow: DELETE, GET, HEAD, OPTIONS, POST, PUT" << __CRLF
|
<< "Allow: DELETE, GET, HEAD, OPTIONS, POST, PUT" << __CRLF
|
||||||
<< "Access-Control-Allow-Origin: *" << __CRLF
|
<< "Access-Control-Allow-Origin: *" << __CRLF
|
||||||
<< "Access-Control-Allow-Methods: GET, POST, HEAD, PUT, DELETE" << __CRLF
|
<< "Access-Control-Allow-Methods: GET, POST, HEAD, PUT, DELETE" << __CRLF
|
||||||
<< "Access-Control-Allow-Headers: Cache-Control,X-Proxy-Authorization,X-Requested-With,Content-Type" << __CRLF
|
<< "Access-Control-Allow-Headers: Cache-Control,X-Proxy-Authorization,X-Requested-With,Content-Type" << __CRLF
|
||||||
<< "Content-Type: text/html;charset=utf-8" << __CRLFCRLF
|
<< "Content-Type: text/html;charset=utf-8" << __CRLFCRLF
|
||||||
<< tilte << (req->body->empty()? "":req->body->bytes())
|
<< tilte << req->body().c_str()
|
||||||
<< "";
|
<< "";
|
||||||
return skt->write(ss.str().c_str(), ss.str().length(), NULL);
|
return skt->write(ss.str().c_str(), ss.str().length(), NULL);
|
||||||
}
|
}
|
||||||
|
@ -164,74 +128,4 @@ int SrsHttpConn::process_request(SrsSocket* skt)
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
int SrsHttpConn::on_message_begin(http_parser* parser)
|
|
||||||
{
|
|
||||||
SrsHttpConn* obj = (SrsHttpConn*)parser->data;
|
|
||||||
obj->req->state = SrsHttpParseStateStart;
|
|
||||||
|
|
||||||
srs_info("***MESSAGE BEGIN***");
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
int SrsHttpConn::on_headers_complete(http_parser* parser)
|
|
||||||
{
|
|
||||||
SrsHttpConn* obj = (SrsHttpConn*)parser->data;
|
|
||||||
memcpy(&obj->req->header, parser, sizeof(http_parser));
|
|
||||||
|
|
||||||
srs_info("***HEADERS COMPLETE***");
|
|
||||||
|
|
||||||
// see http_parser.c:1570, return 1 to skip body.
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
int SrsHttpConn::on_message_complete(http_parser* parser)
|
|
||||||
{
|
|
||||||
SrsHttpConn* obj = (SrsHttpConn*)parser->data;
|
|
||||||
// save the parser when header parse completed.
|
|
||||||
obj->req->state = SrsHttpParseStateComplete;
|
|
||||||
|
|
||||||
srs_info("***MESSAGE COMPLETE***\n");
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
int SrsHttpConn::on_url(http_parser* parser, const char* at, size_t length)
|
|
||||||
{
|
|
||||||
SrsHttpConn* obj = (SrsHttpConn*)parser->data;
|
|
||||||
|
|
||||||
if (length > 0) {
|
|
||||||
obj->req->url.append(at, (int)length);
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_info("Method: %d, Url: %.*s", parser->method, (int)length, at);
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
int SrsHttpConn::on_header_field(http_parser* /*parser*/, const char* at, size_t length)
|
|
||||||
{
|
|
||||||
srs_info("Header field: %.*s", (int)length, at);
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
int SrsHttpConn::on_header_value(http_parser* /*parser*/, const char* at, size_t length)
|
|
||||||
{
|
|
||||||
srs_info("Header value: %.*s", (int)length, at);
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
int SrsHttpConn::on_body(http_parser* parser, const char* at, size_t length)
|
|
||||||
{
|
|
||||||
SrsHttpConn* obj = (SrsHttpConn*)parser->data;
|
|
||||||
|
|
||||||
if (length > 0) {
|
|
||||||
obj->req->body->append(at, (int)length);
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_info("Body: %.*s", (int)length, at);
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -38,28 +38,20 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||||
#include <http_parser.h>
|
#include <http_parser.h>
|
||||||
|
|
||||||
class SrsSocket;
|
class SrsSocket;
|
||||||
|
class SrsHttpParser;
|
||||||
class SrsHttpMessage;
|
class SrsHttpMessage;
|
||||||
|
|
||||||
class SrsHttpConn : public SrsConnection
|
class SrsHttpConn : public SrsConnection
|
||||||
{
|
{
|
||||||
private:
|
private:
|
||||||
SrsHttpMessage* req;
|
SrsHttpParser* parser;
|
||||||
public:
|
public:
|
||||||
SrsHttpConn(SrsServer* srs_server, st_netfd_t client_stfd);
|
SrsHttpConn(SrsServer* srs_server, st_netfd_t client_stfd);
|
||||||
virtual ~SrsHttpConn();
|
virtual ~SrsHttpConn();
|
||||||
protected:
|
protected:
|
||||||
virtual int do_cycle();
|
virtual int do_cycle();
|
||||||
private:
|
private:
|
||||||
virtual int parse_request(SrsSocket* skt, http_parser* parser, http_parser_settings* settings);
|
virtual int process_request(SrsSocket* skt, SrsHttpMessage* req);
|
||||||
virtual int process_request(SrsSocket* skt);
|
|
||||||
private:
|
|
||||||
static int on_message_begin(http_parser* parser);
|
|
||||||
static int on_headers_complete(http_parser* parser);
|
|
||||||
static int on_message_complete(http_parser* parser);
|
|
||||||
static int on_url(http_parser* parser, const char* at, size_t length);
|
|
||||||
static int on_header_field(http_parser* parser, const char* at, size_t length);
|
|
||||||
static int on_header_value(http_parser* parser, const char* at, size_t length);
|
|
||||||
static int on_body(http_parser* parser, const char* at, size_t length);
|
|
||||||
};
|
};
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -45,17 +45,30 @@ SrsHttpClient::SrsHttpClient()
|
||||||
{
|
{
|
||||||
connected = false;
|
connected = false;
|
||||||
stfd = NULL;
|
stfd = NULL;
|
||||||
|
parser = NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
SrsHttpClient::~SrsHttpClient()
|
SrsHttpClient::~SrsHttpClient()
|
||||||
{
|
{
|
||||||
disconnect();
|
disconnect();
|
||||||
|
srs_freep(parser);
|
||||||
}
|
}
|
||||||
|
|
||||||
int SrsHttpClient::post(SrsHttpUri* uri, std::string req, std::string& res)
|
int SrsHttpClient::post(SrsHttpUri* uri, std::string req, std::string& res)
|
||||||
{
|
{
|
||||||
|
res = "";
|
||||||
|
|
||||||
int ret = ERROR_SUCCESS;
|
int ret = ERROR_SUCCESS;
|
||||||
|
|
||||||
|
if (!parser) {
|
||||||
|
parser = new SrsHttpParser();
|
||||||
|
|
||||||
|
if ((ret = parser->initialize(HTTP_RESPONSE)) != ERROR_SUCCESS) {
|
||||||
|
srs_error("initialize parser failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if ((ret = connect(uri)) != ERROR_SUCCESS) {
|
if ((ret = connect(uri)) != ERROR_SUCCESS) {
|
||||||
srs_error("http connect server failed. ret=%d", ret);
|
srs_error("http connect server failed. ret=%d", ret);
|
||||||
return ret;
|
return ret;
|
||||||
|
@ -77,8 +90,7 @@ int SrsHttpClient::post(SrsHttpUri* uri, std::string req, std::string& res)
|
||||||
SrsSocket skt(stfd);
|
SrsSocket skt(stfd);
|
||||||
|
|
||||||
std::string data = ss.str();
|
std::string data = ss.str();
|
||||||
ssize_t nwrite;
|
if ((ret = skt.write(data.c_str(), data.length(), NULL)) != ERROR_SUCCESS) {
|
||||||
if ((ret = skt.write(data.c_str(), data.length(), &nwrite)) != ERROR_SUCCESS) {
|
|
||||||
// disconnect when error.
|
// disconnect when error.
|
||||||
disconnect();
|
disconnect();
|
||||||
|
|
||||||
|
@ -86,10 +98,19 @@ int SrsHttpClient::post(SrsHttpUri* uri, std::string req, std::string& res)
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
if ((ret = parse_response(uri, &skt, &res)) != ERROR_SUCCESS) {
|
SrsHttpMessage* msg = NULL;
|
||||||
|
if ((ret = parser->parse_message(&skt, &msg)) != ERROR_SUCCESS) {
|
||||||
srs_error("parse http post response failed. ret=%d", ret);
|
srs_error("parse http post response failed. ret=%d", ret);
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
srs_assert(msg);
|
||||||
|
srs_assert(msg->is_complete());
|
||||||
|
|
||||||
|
// get response body.
|
||||||
|
if (msg->body_size() > 0) {
|
||||||
|
res = msg->body();
|
||||||
|
}
|
||||||
srs_info("parse http post response success.");
|
srs_info("parse http post response success.");
|
||||||
|
|
||||||
return ret;
|
return ret;
|
||||||
|
@ -153,152 +174,6 @@ int SrsHttpClient::connect(SrsHttpUri* uri)
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
int SrsHttpClient::parse_response(SrsHttpUri* uri, SrsSocket* skt, std::string* response)
|
|
||||||
{
|
|
||||||
int ret = ERROR_SUCCESS;
|
|
||||||
|
|
||||||
int body_received = 0;
|
|
||||||
if ((ret = parse_response_header(skt, response, body_received)) != ERROR_SUCCESS) {
|
|
||||||
srs_error("parse response header failed. ret=%d", ret);
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
if ((ret = parse_response_body(uri, skt, response, body_received)) != ERROR_SUCCESS) {
|
|
||||||
srs_error("parse response body failed. ret=%d", ret);
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_info("url %s download, body size=%"PRId64, uri->get_url(), http_header.content_length);
|
|
||||||
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
int SrsHttpClient::parse_response_header(SrsSocket* skt, std::string* response, int& body_received)
|
|
||||||
{
|
|
||||||
int ret = ERROR_SUCCESS;
|
|
||||||
|
|
||||||
http_parser_settings settings;
|
|
||||||
|
|
||||||
memset(&settings, 0, sizeof(settings));
|
|
||||||
settings.on_headers_complete = on_headers_complete;
|
|
||||||
|
|
||||||
http_parser parser;
|
|
||||||
http_parser_init(&parser, HTTP_RESPONSE);
|
|
||||||
// callback object ptr.
|
|
||||||
parser.data = (void*)this;
|
|
||||||
|
|
||||||
// reset response header.
|
|
||||||
memset(&http_header, 0, sizeof(http_header));
|
|
||||||
|
|
||||||
// parser header.
|
|
||||||
char buf[SRS_HTTP_HEADER_BUFFER];
|
|
||||||
for (;;) {
|
|
||||||
ssize_t nread;
|
|
||||||
if ((ret = skt->read(buf, (size_t)sizeof(buf), &nread)) != ERROR_SUCCESS) {
|
|
||||||
srs_error("read body from server failed. ret=%d", ret);
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
ssize_t nparsed = http_parser_execute(&parser, &settings, buf, nread);
|
|
||||||
srs_info("read_size=%d, nparsed=%d", (int)nread, (int)nparsed);
|
|
||||||
|
|
||||||
// check header size.
|
|
||||||
if (http_header.nread != 0) {
|
|
||||||
body_received = nread - nparsed;
|
|
||||||
|
|
||||||
srs_info("http header parsed, size=%d, content-length=%"PRId64", body-received=%d",
|
|
||||||
http_header.nread, http_header.content_length, body_received);
|
|
||||||
|
|
||||||
if(response != NULL && body_received > 0){
|
|
||||||
response->append(buf + nparsed, body_received);
|
|
||||||
}
|
|
||||||
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (nparsed != nread) {
|
|
||||||
ret = ERROR_HTTP_PARSE_HEADER;
|
|
||||||
srs_error("parse response error, parsed(%d)!=read(%d), ret=%d", (int)nparsed, (int)nread, ret);
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
int SrsHttpClient::parse_response_body(SrsHttpUri* uri, SrsSocket* skt, std::string* response, int body_received)
|
|
||||||
{
|
|
||||||
int ret = ERROR_SUCCESS;
|
|
||||||
|
|
||||||
srs_assert(uri != NULL);
|
|
||||||
|
|
||||||
uint64_t body_left = http_header.content_length - body_received;
|
|
||||||
|
|
||||||
if (body_left <= 0) {
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (response != NULL) {
|
|
||||||
char buf[SRS_HTTP_BODY_BUFFER];
|
|
||||||
|
|
||||||
return parse_response_body_data(
|
|
||||||
uri, skt, response, (size_t)body_left,
|
|
||||||
(const void*)buf, (size_t)SRS_HTTP_BODY_BUFFER
|
|
||||||
);
|
|
||||||
} else {
|
|
||||||
// if ignore response, use shared fast memory.
|
|
||||||
static char buf[SRS_HTTP_BODY_BUFFER];
|
|
||||||
|
|
||||||
return parse_response_body_data(
|
|
||||||
uri, skt, response, (size_t)body_left,
|
|
||||||
(const void*)buf, (size_t)SRS_HTTP_BODY_BUFFER
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
int SrsHttpClient::parse_response_body_data(SrsHttpUri* uri, SrsSocket* skt, std::string* response, size_t body_left, const void* buf, size_t size)
|
|
||||||
{
|
|
||||||
int ret = ERROR_SUCCESS;
|
|
||||||
|
|
||||||
srs_assert(uri != NULL);
|
|
||||||
|
|
||||||
while (body_left > 0) {
|
|
||||||
ssize_t nread;
|
|
||||||
int size_to_read = srs_min(size, body_left);
|
|
||||||
if ((ret = skt->read(buf, size_to_read, &nread)) != ERROR_SUCCESS) {
|
|
||||||
srs_error("read header from server failed. ret=%d", ret);
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (response != NULL && nread > 0) {
|
|
||||||
response->append((char*)buf, nread);
|
|
||||||
}
|
|
||||||
|
|
||||||
body_left -= nread;
|
|
||||||
srs_info("read url(%s) content partial %"PRId64"/%"PRId64"",
|
|
||||||
uri->get_url(), http_header.content_length - body_left, http_header.content_length);
|
|
||||||
}
|
|
||||||
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
int SrsHttpClient::on_headers_complete(http_parser* parser)
|
|
||||||
{
|
|
||||||
SrsHttpClient* obj = (SrsHttpClient*)parser->data;
|
|
||||||
obj->complete_header(parser);
|
|
||||||
|
|
||||||
// see http_parser.c:1570, return 1 to skip body.
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
void SrsHttpClient::complete_header(http_parser* parser)
|
|
||||||
{
|
|
||||||
// save the parser status when header parse completed.
|
|
||||||
memcpy(&http_header, parser, sizeof(http_header));
|
|
||||||
}
|
|
||||||
|
|
||||||
SrsHttpHooks::SrsHttpHooks()
|
SrsHttpHooks::SrsHttpHooks()
|
||||||
{
|
{
|
||||||
}
|
}
|
||||||
|
|
|
@ -36,6 +36,7 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||||
class SrsHttpUri;
|
class SrsHttpUri;
|
||||||
class SrsSocket;
|
class SrsSocket;
|
||||||
class SrsRequest;
|
class SrsRequest;
|
||||||
|
class SrsHttpParser;
|
||||||
|
|
||||||
#include <srs_app_st.hpp>
|
#include <srs_app_st.hpp>
|
||||||
|
|
||||||
|
@ -47,8 +48,7 @@ class SrsHttpClient
|
||||||
private:
|
private:
|
||||||
bool connected;
|
bool connected;
|
||||||
st_netfd_t stfd;
|
st_netfd_t stfd;
|
||||||
private:
|
SrsHttpParser* parser;
|
||||||
http_parser http_header;
|
|
||||||
public:
|
public:
|
||||||
SrsHttpClient();
|
SrsHttpClient();
|
||||||
virtual ~SrsHttpClient();
|
virtual ~SrsHttpClient();
|
||||||
|
@ -62,14 +62,6 @@ public:
|
||||||
private:
|
private:
|
||||||
virtual void disconnect();
|
virtual void disconnect();
|
||||||
virtual int connect(SrsHttpUri* uri);
|
virtual int connect(SrsHttpUri* uri);
|
||||||
private:
|
|
||||||
virtual int parse_response(SrsHttpUri* uri, SrsSocket* skt, std::string* response);
|
|
||||||
virtual int parse_response_header(SrsSocket* skt, std::string* response, int& body_received);
|
|
||||||
virtual int parse_response_body(SrsHttpUri* uri, SrsSocket* skt, std::string* response, int body_received);
|
|
||||||
virtual int parse_response_body_data(SrsHttpUri* uri, SrsSocket* skt, std::string* response, size_t body_left, const void* buf, size_t size);
|
|
||||||
private:
|
|
||||||
static int on_headers_complete(http_parser* parser);
|
|
||||||
virtual void complete_header(http_parser* parser);
|
|
||||||
};
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue