From 341b5151d959967a0c2e1a95e949f5231723bdb5 Mon Sep 17 00:00:00 2001
From: winlin <winlin@vip.126.com>
Date: Wed, 2 Apr 2014 18:07:34 +0800
Subject: [PATCH] create http handler framework

---
 trunk/auto/depends.sh                 |  4 +-
 trunk/src/app/srs_app_http.cpp        | 92 ++++++++++++++++++++++++++-
 trunk/src/app/srs_app_http.hpp        | 48 ++++++++++++++
 trunk/src/app/srs_app_http_api.cpp    | 44 ++++++++++++-
 trunk/src/app/srs_app_http_api.hpp    | 16 ++++-
 trunk/src/app/srs_app_http_conn.cpp   |  3 +-
 trunk/src/app/srs_app_http_conn.hpp   |  4 +-
 trunk/src/app/srs_app_server.cpp      | 33 +++++++++-
 trunk/src/app/srs_app_server.hpp      |  8 +++
 trunk/src/kernel/srs_kernel_error.hpp |  1 +
 10 files changed, 244 insertions(+), 9 deletions(-)
diff --git a/trunk/auto/depends.sh b/trunk/auto/depends.sh
index 4ab5da371..a36959554 100755
--- a/trunk/auto/depends.sh
+++ b/trunk/auto/depends.sh
@@ -285,7 +285,7 @@ fi
 #####################################################################################
 function write_nginx_html5()
 {
-    cat<<END >> ${html_file}
+    cat<<END > ${html_file}
 <video width="640" height="360"
         autoplay controls autobuffer 
         src="${hls_stream}"
@@ -398,7 +398,7 @@ if [ $__SRS_BUILD_NGINX = YES ]; then
         ln -sf `pwd`/research/players/nginx_index.html ${SRS_OBJS}/nginx/html/index.html
     else
         rm -f ${SRS_OBJS}/nginx/html/index.html &&
-        cat<<END >> ${SRS_OBJS}/nginx/html/index.html
+        cat<<END > ${SRS_OBJS}/nginx/html/index.html
 <!DOCTYPE html>
 <html>
 <head>
diff --git a/trunk/src/app/srs_app_http.cpp b/trunk/src/app/srs_app_http.cpp
index 28c469ad2..991ca16f8 100644
--- a/trunk/src/app/srs_app_http.cpp
+++ b/trunk/src/app/srs_app_http.cpp
@@ -32,10 +32,100 @@ using namespace std;
 #include <srs_kernel_error.hpp>
 #include <srs_kernel_log.hpp>
 #include <srs_app_socket.hpp>
+#include <srs_app_http_api.hpp>
+#include <srs_app_http_conn.hpp>
 
 #define SRS_DEFAULT_HTTP_PORT 80
 
-#define SRS_HTTP_HEADER_BUFFER        1024
+#define SRS_HTTP_HEADER_BUFFER 1024
+
+SrsHttpHandler::SrsHttpHandler()
+{
+}
+
+SrsHttpHandler::~SrsHttpHandler()
+{
+    std::vector<SrsHttpHandler*>::iterator it;
+    for (it = handlers.begin(); it != handlers.end(); ++it) {
+        SrsHttpHandler* handler = *it;
+        srs_freep(handler);
+    }
+    handlers.clear();
+}
+
+int SrsHttpHandler::initialize()
+{
+    int ret = ERROR_SUCCESS;
+    return ret;
+}
+
+bool SrsHttpHandler::can_handle(const char* /*path*/, int /*length*/)
+{
+    return false;
+}
+
+int SrsHttpHandler::process_request(SrsSocket* /*skt*/, SrsHttpMessage* /*req*/, const char* /*path*/, int /*length*/)
+{
+    int ret = ERROR_SUCCESS;
+    return ret;
+}
+
+int SrsHttpHandler::best_match(const char* path, int length, SrsHttpHandler** phandler, const char** pstart, int* plength)
+{
+    int ret = ERROR_SUCCESS;
+    
+    for (;;) {
+        // ensure cur is not NULL.
+        // ensure p not NULL and has bytes to parse.
+        if (!path || length <= 0) {
+            break;
+        }
+        
+        const char* p = NULL;
+        for (p = path + 1; p - path < length && *p != __PATH_SEP; p++) {
+        }
+        
+        // whether the handler can handler the node.
+        int node_size = p - path;
+        if (!can_handle(path, node_size)) {
+            break;
+        }
+        
+        *phandler = this;
+        *pstart = path;
+        *plength = node_size;
+        
+        std::vector<SrsHttpHandler*>::iterator it;
+        for (it = handlers.begin(); it != handlers.end(); ++it) {
+            SrsHttpHandler* handler = *it;
+            // matched, donot search.
+            if (handler->best_match(p, length - node_size, phandler, pstart, plength) == ERROR_SUCCESS) {
+                break;
+            }
+        }
+        
+        // whatever, donot loop.
+        break;
+    }
+    
+    if (*phandler == NULL) {
+        ret = ERROR_HTTP_HANDLER_MATCH_URL;
+        return ret;
+    }
+    
+    return ret;
+}
+
+SrsHttpHandler* SrsHttpHandler::create_http_api()
+{
+    return new SrsApiRoot();
+}
+
+SrsHttpHandler* SrsHttpHandler::create_http_stream()
+{
+    // TODO: FIXME: use http stream handler instead.
+    return new SrsHttpHandler();
+}
 
 SrsHttpMessage::SrsHttpMessage()
 {
diff --git a/trunk/src/app/srs_app_http.hpp b/trunk/src/app/srs_app_http.hpp
index 1bcecf277..0559c14f4 100644
--- a/trunk/src/app/srs_app_http.hpp
+++ b/trunk/src/app/srs_app_http.hpp
@@ -32,6 +32,7 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 #ifdef SRS_HTTP_PARSER
 
 #include <string>
+#include <vector>
 
 #include <http_parser.h>
 
@@ -40,6 +41,7 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 class SrsBuffer;
 class SrsRequest;
 class SrsSocket;
+class SrsHttpMessage;
 
 // http specification
 // CR             = <US-ASCII CR, carriage return (13)>
@@ -57,12 +59,58 @@ class SrsSocket;
 #define __CRLF "\r\n" // 0x0D0A
 #define __CRLFCRLF "\r\n\r\n" // 0x0D0A0D0A
 
+// linux path seprator
+#define __PATH_SEP '/'
+
+// state of message
 enum SrsHttpParseState {
     SrsHttpParseStateInit = 0, 
     SrsHttpParseStateStart, 
     SrsHttpParseStateComplete
 };
 
+/**
+* resource handler for HTTP RESTful api.
+*/
+class SrsHttpHandler
+{
+protected:
+    /**
+    * we use handler chain to process request.
+    */
+    std::vector<SrsHttpHandler*> handlers;
+public:
+    SrsHttpHandler();
+    virtual ~SrsHttpHandler();
+public:
+    /**
+    * initialize the handler.
+    */
+    virtual int initialize();
+    /**
+    * whether current handler can handle the specified path.
+    */
+    virtual bool can_handle(const char* path, int length);
+    /**
+    * use the handler to process the request.
+    */
+    virtual int process_request(SrsSocket* skt, SrsHttpMessage* req, const char* path, int length);
+public:
+    /**
+    * find the best matched handler
+    */
+    virtual int best_match(const char* path, int length, SrsHttpHandler** phandler, const char** pstart, int* plength);
+public:
+    /**
+    * create http api resource handler.
+    */
+    static SrsHttpHandler* create_http_api();
+    /**
+    * create http stream resource handler.
+    */
+    static SrsHttpHandler* create_http_stream();
+};
+
 /**
 * the http message, request or response.
 */
diff --git a/trunk/src/app/srs_app_http_api.cpp b/trunk/src/app/srs_app_http_api.cpp
index ef66725bf..0b505244b 100644
--- a/trunk/src/app/srs_app_http_api.cpp
+++ b/trunk/src/app/srs_app_http_api.cpp
@@ -34,10 +34,30 @@ using namespace std;
 #include <srs_app_socket.hpp>
 #include <srs_core_autofree.hpp>
 
-SrsHttpApi::SrsHttpApi(SrsServer* srs_server, st_netfd_t client_stfd) 
+SrsApiRoot::SrsApiRoot()
+{
+}
+
+SrsApiRoot::~SrsApiRoot()
+{
+}
+
+bool SrsApiRoot::can_handle(const char* /*path*/, int /*length*/)
+{
+    return true;
+}
+
+int SrsApiRoot::process_request(SrsSocket* /*skt*/, SrsHttpMessage* /*req*/, const char* /*path*/, int /*length*/)
+{
+    int ret = ERROR_SUCCESS;
+    return ret;
+}
+
+SrsHttpApi::SrsHttpApi(SrsServer* srs_server, st_netfd_t client_stfd, SrsHttpHandler* _handler) 
     : SrsConnection(srs_server, client_stfd)
 {
     parser = new SrsHttpParser();
+    handler = _handler;
 }
 
 SrsHttpApi::~SrsHttpApi()
@@ -93,6 +113,28 @@ int SrsHttpApi::process_request(SrsSocket* skt, SrsHttpMessage* req)
 {
     int ret = ERROR_SUCCESS;
     
+    // TODO: maybe need to parse the url.
+    std::string uri = req->url();
+    
+    int length = 0;
+    const char* start = NULL;
+    SrsHttpHandler* p = NULL;
+    if ((ret = handler->best_match(uri.data(), uri.length(), &p, &start, &length)) != ERROR_SUCCESS) {
+        srs_warn("failed to find the best match handler for url. ret=%d", ret);
+        return ret;
+    }
+    
+    // if success, p and pstart should be valid.
+    srs_assert(p);
+    srs_assert(start);
+    srs_assert(length <= (int)uri.length());
+    
+    // use handler to process request.
+    if ((ret = p->process_request(skt, req, start, length)) != ERROR_SUCCESS) {
+        srs_warn("handler failed to process http request. ret=%d", ret);
+        return ret;
+    }
+    
     if (req->method() == HTTP_OPTIONS) {
         char data[] = "HTTP/1.1 200 OK" __CRLF
             "Content-Length: 0"__CRLF
diff --git a/trunk/src/app/srs_app_http_api.hpp b/trunk/src/app/srs_app_http_api.hpp
index 771b29b0d..743a91d45 100644
--- a/trunk/src/app/srs_app_http_api.hpp
+++ b/trunk/src/app/srs_app_http_api.hpp
@@ -35,16 +35,30 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 class SrsSocket;
 class SrsHttpMessage;
 class SrsHttpParser;
+class SrsHttpHandler;
 
 #include <srs_app_st.hpp>
 #include <srs_app_conn.hpp>
+#include <srs_app_http.hpp>
+
+// for http root.
+class SrsApiRoot : public SrsHttpHandler
+{
+public:
+    SrsApiRoot();
+    virtual ~SrsApiRoot();
+public:
+    virtual bool can_handle(const char* path, int length);
+    virtual int process_request(SrsSocket* skt, SrsHttpMessage* req, const char* path, int length);
+};
 
 class SrsHttpApi : public SrsConnection
 {
 private:
     SrsHttpParser* parser;
+    SrsHttpHandler* handler;
 public:
-    SrsHttpApi(SrsServer* srs_server, st_netfd_t client_stfd);
+    SrsHttpApi(SrsServer* srs_server, st_netfd_t client_stfd, SrsHttpHandler* _handler);
     virtual ~SrsHttpApi();
 protected:
     virtual int do_cycle();
diff --git a/trunk/src/app/srs_app_http_conn.cpp b/trunk/src/app/srs_app_http_conn.cpp
index 5f26a1a71..0d8ae6aa8 100644
--- a/trunk/src/app/srs_app_http_conn.cpp
+++ b/trunk/src/app/srs_app_http_conn.cpp
@@ -37,10 +37,11 @@ using namespace std;
 
 #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, SrsHttpHandler* _handler) 
     : SrsConnection(srs_server, client_stfd)
 {
     parser = new SrsHttpParser();
+    handler = _handler;
 }
 
 SrsHttpConn::~SrsHttpConn()
diff --git a/trunk/src/app/srs_app_http_conn.hpp b/trunk/src/app/srs_app_http_conn.hpp
index c7efebf29..fc26d905e 100644
--- a/trunk/src/app/srs_app_http_conn.hpp
+++ b/trunk/src/app/srs_app_http_conn.hpp
@@ -40,13 +40,15 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 class SrsSocket;
 class SrsHttpParser;
 class SrsHttpMessage;
+class SrsHttpHandler;
 
 class SrsHttpConn : public SrsConnection
 {
 private:
     SrsHttpParser* parser;
+    SrsHttpHandler* handler;
 public:
-    SrsHttpConn(SrsServer* srs_server, st_netfd_t client_stfd);
+    SrsHttpConn(SrsServer* srs_server, st_netfd_t client_stfd, SrsHttpHandler* _handler);
     virtual ~SrsHttpConn();
 protected:
     virtual int do_cycle();
diff --git a/trunk/src/app/srs_app_server.cpp b/trunk/src/app/srs_app_server.cpp
index 57bbbb2f4..475bfd8dc 100644
--- a/trunk/src/app/srs_app_server.cpp
+++ b/trunk/src/app/srs_app_server.cpp
@@ -40,6 +40,7 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 #include <srs_kernel_utility.hpp>
 #include <srs_app_http_api.hpp>
 #include <srs_app_http_conn.hpp>
+#include <srs_app_http.hpp>
 
 #define SERVER_LISTEN_BACKLOG 512
 #define SRS_TIME_RESOLUTION_MS 500
@@ -160,6 +161,13 @@ SrsServer::SrsServer()
     
     srs_assert(_srs_config);
     _srs_config->subscribe(this);
+    
+#ifdef SRS_HTTP_API
+    http_api_handler = SrsHttpHandler::create_http_api();
+#endif
+#ifdef SRS_HTTP_SERVER
+    http_stream_handler = SrsHttpHandler::create_http_stream();
+#endif
 }
 
 SrsServer::~SrsServer()
@@ -176,11 +184,32 @@ SrsServer::~SrsServer()
     }
     
     close_listeners();
+    
+#ifdef SRS_HTTP_API
+    srs_freep(http_api_handler);
+#endif
+
+#ifdef SRS_HTTP_SERVER
+    srs_freep(http_stream_handler);
+#endif
 }
 
 int SrsServer::initialize()
 {
     int ret = ERROR_SUCCESS;
+    
+#ifdef SRS_HTTP_API
+    if ((ret = http_api_handler->initialize()) != ERROR_SUCCESS) {
+        return ret;
+    }
+#endif
+
+#ifdef SRS_HTTP_SERVER
+    if ((ret = http_stream_handler->initialize()) != ERROR_SUCCESS) {
+        return ret;
+    }
+#endif
+
     return ret;
 }
 
@@ -443,7 +472,7 @@ int SrsServer::accept_client(SrsListenerType type, st_netfd_t client_stfd)
         conn = new SrsRtmpConn(this, client_stfd);
     } else if (type == SrsListenerHttpApi) {
 #ifdef SRS_HTTP_API
-        conn = new SrsHttpApi(this, client_stfd);
+        conn = new SrsHttpApi(this, client_stfd, http_api_handler);
 #else
         srs_warn("close http client for server not support http-api");
         srs_close_stfd(client_stfd);
@@ -451,7 +480,7 @@ int SrsServer::accept_client(SrsListenerType type, st_netfd_t client_stfd)
 #endif
     } else if (type == SrsListenerHttpStream) {
 #ifdef SRS_HTTP_SERVER
-        conn = new SrsHttpConn(this, client_stfd);
+        conn = new SrsHttpConn(this, client_stfd, http_stream_handler);
 #else
         srs_warn("close http client for server not support http-server");
         srs_close_stfd(client_stfd);
diff --git a/trunk/src/app/srs_app_server.hpp b/trunk/src/app/srs_app_server.hpp
index 72d2f44b1..fa88b94d2 100644
--- a/trunk/src/app/srs_app_server.hpp
+++ b/trunk/src/app/srs_app_server.hpp
@@ -38,6 +38,7 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 
 class SrsServer;
 class SrsConnection;
+class SrsHttpHandler;
 
 // listener type for server to identify the connection,
 // that is, use different type to process the connection.
@@ -75,6 +76,13 @@ public:
 class SrsServer : public ISrsReloadHandler
 {
     friend class SrsListener;
+private:
+#ifdef SRS_HTTP_API
+    SrsHttpHandler* http_api_handler;
+#endif
+#ifdef SRS_HTTP_SERVER
+    SrsHttpHandler* http_stream_handler;
+#endif
 private:
     std::vector<SrsConnection*> conns;
     std::vector<SrsListener*> listeners;
diff --git a/trunk/src/kernel/srs_kernel_error.hpp b/trunk/src/kernel/srs_kernel_error.hpp
index 84c218f91..c2a520f39 100644
--- a/trunk/src/kernel/srs_kernel_error.hpp
+++ b/trunk/src/kernel/srs_kernel_error.hpp
@@ -157,6 +157,7 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 #define ERROR_HTTP_PARSE_URI                800
 #define ERROR_HTTP_DATA_INVLIAD             801
 #define ERROR_HTTP_PARSE_HEADER             802
+#define ERROR_HTTP_HANDLER_MATCH_URL        803
 
 // system control message, 
 // not an error, but special control logic.
