mirror of
https://github.com/ossrs/srs.git
synced 2025-03-09 15:49:59 +00:00
add api framework code
This commit is contained in:
parent
0ba0c14fb8
commit
eae9b94153
3 changed files with 85 additions and 2 deletions
|
@ -25,16 +25,24 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||||
|
|
||||||
#ifdef SRS_HTTP_API
|
#ifdef SRS_HTTP_API
|
||||||
|
|
||||||
|
#include <sstream>
|
||||||
|
using namespace std;
|
||||||
|
|
||||||
#include <srs_kernel_log.hpp>
|
#include <srs_kernel_log.hpp>
|
||||||
#include <srs_kernel_error.hpp>
|
#include <srs_kernel_error.hpp>
|
||||||
|
#include <srs_app_http.hpp>
|
||||||
|
#include <srs_app_socket.hpp>
|
||||||
|
#include <srs_core_autofree.hpp>
|
||||||
|
|
||||||
SrsHttpApi::SrsHttpApi(SrsServer* srs_server, st_netfd_t client_stfd)
|
SrsHttpApi::SrsHttpApi(SrsServer* srs_server, st_netfd_t client_stfd)
|
||||||
: SrsConnection(srs_server, client_stfd)
|
: SrsConnection(srs_server, client_stfd)
|
||||||
{
|
{
|
||||||
|
parser = new SrsHttpParser();
|
||||||
}
|
}
|
||||||
|
|
||||||
SrsHttpApi::~SrsHttpApi()
|
SrsHttpApi::~SrsHttpApi()
|
||||||
{
|
{
|
||||||
|
srs_freep(parser);
|
||||||
}
|
}
|
||||||
|
|
||||||
int SrsHttpApi::do_cycle()
|
int SrsHttpApi::do_cycle()
|
||||||
|
@ -47,6 +55,73 @@ int SrsHttpApi::do_cycle()
|
||||||
}
|
}
|
||||||
srs_trace("api get peer ip success. ip=%s", ip);
|
srs_trace("api get peer ip success. ip=%s", ip);
|
||||||
|
|
||||||
|
// initialize parser
|
||||||
|
if ((ret = parser->initialize(HTTP_REQUEST)) != ERROR_SUCCESS) {
|
||||||
|
srs_error("api initialize http parser failed. ret=%d", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
// underlayer socket
|
||||||
|
SrsSocket skt(stfd);
|
||||||
|
|
||||||
|
// process http messages.
|
||||||
|
for (;;) {
|
||||||
|
SrsHttpMessage* req = NULL;
|
||||||
|
|
||||||
|
// 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;
|
||||||
|
}
|
||||||
|
|
||||||
|
int SrsHttpApi::process_request(SrsSocket* skt, SrsHttpMessage* req)
|
||||||
|
{
|
||||||
|
int ret = ERROR_SUCCESS;
|
||||||
|
|
||||||
|
if (req->method() == HTTP_OPTIONS) {
|
||||||
|
char data[] = "HTTP/1.1 200 OK" __CRLF
|
||||||
|
"Content-Length: 0"__CRLF
|
||||||
|
"Server: SRS/"RTMP_SIG_SRS_VERSION""__CRLF
|
||||||
|
"Allow: DELETE, GET, HEAD, OPTIONS, POST, PUT"__CRLF
|
||||||
|
"Access-Control-Allow-Origin: *"__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
|
||||||
|
"Content-Type: text/html;charset=utf-8"__CRLFCRLF
|
||||||
|
"";
|
||||||
|
return skt->write(data, sizeof(data), NULL);
|
||||||
|
} else {
|
||||||
|
std::string tilte = "SRS/"RTMP_SIG_SRS_VERSION;
|
||||||
|
tilte += " hello http/1.1 api~\n";
|
||||||
|
|
||||||
|
std::stringstream ss;
|
||||||
|
ss << "HTTP/1.1 200 OK " << __CRLF
|
||||||
|
<< "Content-Length: "<< tilte.length() + req->body_size() << __CRLF
|
||||||
|
<< "Server: SRS/"RTMP_SIG_SRS_VERSION"" << __CRLF
|
||||||
|
<< "Allow: DELETE, GET, HEAD, OPTIONS, POST, PUT" << __CRLF
|
||||||
|
<< "Access-Control-Allow-Origin: *" << __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
|
||||||
|
<< "Content-Type: text/html;charset=utf-8" << __CRLFCRLF
|
||||||
|
<< tilte << req->body().c_str()
|
||||||
|
<< "";
|
||||||
|
return skt->write(ss.str().c_str(), ss.str().length(), NULL);
|
||||||
|
}
|
||||||
|
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -32,16 +32,24 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||||
|
|
||||||
#ifdef SRS_HTTP_API
|
#ifdef SRS_HTTP_API
|
||||||
|
|
||||||
|
class SrsSocket;
|
||||||
|
class SrsHttpMessage;
|
||||||
|
class SrsHttpParser;
|
||||||
|
|
||||||
#include <srs_app_st.hpp>
|
#include <srs_app_st.hpp>
|
||||||
#include <srs_app_conn.hpp>
|
#include <srs_app_conn.hpp>
|
||||||
|
|
||||||
class SrsHttpApi : public SrsConnection
|
class SrsHttpApi : public SrsConnection
|
||||||
{
|
{
|
||||||
|
private:
|
||||||
|
SrsHttpParser* parser;
|
||||||
public:
|
public:
|
||||||
SrsHttpApi(SrsServer* srs_server, st_netfd_t client_stfd);
|
SrsHttpApi(SrsServer* srs_server, st_netfd_t client_stfd);
|
||||||
virtual ~SrsHttpApi();
|
virtual ~SrsHttpApi();
|
||||||
protected:
|
protected:
|
||||||
virtual int do_cycle();
|
virtual int do_cycle();
|
||||||
|
private:
|
||||||
|
virtual int process_request(SrsSocket* skt, SrsHttpMessage* req);
|
||||||
};
|
};
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -60,7 +60,7 @@ int SrsHttpConn::do_cycle()
|
||||||
|
|
||||||
// initialize parser
|
// initialize parser
|
||||||
if ((ret = parser->initialize(HTTP_REQUEST)) != ERROR_SUCCESS) {
|
if ((ret = parser->initialize(HTTP_REQUEST)) != ERROR_SUCCESS) {
|
||||||
srs_error("initialize http parser failed. ret=%d", ret);
|
srs_error("http initialize http parser failed. ret=%d", ret);
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -109,7 +109,7 @@ int SrsHttpConn::process_request(SrsSocket* skt, SrsHttpMessage* req)
|
||||||
return skt->write(data, sizeof(data), NULL);
|
return skt->write(data, sizeof(data), NULL);
|
||||||
} else {
|
} else {
|
||||||
std::string tilte = "SRS/"RTMP_SIG_SRS_VERSION;
|
std::string tilte = "SRS/"RTMP_SIG_SRS_VERSION;
|
||||||
tilte += " hello http/1.1~\n";
|
tilte += " hello http/1.1 server~\n";
|
||||||
|
|
||||||
std::stringstream ss;
|
std::stringstream ss;
|
||||||
ss << "HTTP/1.1 200 OK " << __CRLF
|
ss << "HTTP/1.1 200 OK " << __CRLF
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue