refine http stream server

@@ -164,58 +164,18 @@ int SrsHttpVhost::do_process_request(SrsSocket* skt, SrsHttpMessage* req)
     std::string fullpath = get_request_file(req);
     std::string fullpath = get_request_file(req);
     
     
     if (srs_string_ends_with(fullpath, ".ts")) {
     if (srs_string_ends_with(fullpath, ".ts")) {
-        // TODO: FIXME: use more advance cache.
-        // for ts video large file, use bytes to write it.
-        int fd = ::open(fullpath.c_str(), O_RDONLY);
-        if (fd < 0) {
-            ret = ERROR_HTTP_OPEN_FILE;
-            srs_warn("open file %s failed, ret=%d", fullpath.c_str(), ret);
-            return ret;
-        }
-    
-        int64_t length = (int64_t)::lseek(fd, 0, SEEK_END);
-        ::lseek(fd, 0, SEEK_SET);
-    
-        // 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;
+        return response_ts_file(skt, req, fullpath);
+    } else {
+        return response_regular_file(skt, req, fullpath);
     }
     }
     
     
-        // write body.
-        int64_t left = length;
-        char* buf = req->http_ts_send_buffer();
-        
-        while (left > 0) {
-            ssize_t nread = -1;
-            // TODO: FIXME: use st_read.
-            if ((nread = ::read(fd, buf, HTTP_TS_SEND_BUFFER_SIZE)) < 0) {
-                ::close(fd);
-                ret = ERROR_HTTP_READ_FILE;
-                srs_warn("read file %s failed, ret=%d", fullpath.c_str(), ret);
     return ret;
     return ret;
-            }
+}
 
 
-            left -= nread;
-            if ((ret = skt->write(buf, nread, NULL)) != ERROR_SUCCESS) {
-                break;
-            }
-        }
-        ::close(fd);
+int SrsHttpVhost::response_regular_file(SrsSocket* skt, SrsHttpMessage* req, string fullpath)
+{
+    int ret = ERROR_SUCCESS;
 
 
-        return ret;
-    } else {
     // TODO: FIXME: refine the file stream.
     // TODO: FIXME: refine the file stream.
     int fd = ::open(fullpath.c_str(), O_RDONLY);
     int fd = ::open(fullpath.c_str(), O_RDONLY);
     if (fd < 0) {
     if (fd < 0) {
@@ -261,8 +221,64 @@ int SrsHttpVhost::do_process_request(SrsSocket* skt, SrsHttpMessage* req)
     } else {
     } else {
         return res_text(skt, req, str);
         return res_text(skt, req, str);
     }
     }
+    
+    return ret;
+}
+
+int SrsHttpVhost::response_ts_file(SrsSocket* skt, SrsHttpMessage* req, string fullpath)
+{
+    int ret = ERROR_SUCCESS;
+    
+    // TODO: FIXME: use more advance cache.
+    // for ts video large file, use bytes to write it.
+    int fd = ::open(fullpath.c_str(), O_RDONLY);
+    if (fd < 0) {
+        ret = ERROR_HTTP_OPEN_FILE;
+        srs_warn("open file %s failed, ret=%d", fullpath.c_str(), ret);
+        return ret;
+    }
+
+    int64_t length = (int64_t)::lseek(fd, 0, SEEK_END);
+    ::lseek(fd, 0, SEEK_SET);
+
+    // 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;
+        // TODO: FIXME: use st_read.
+        if ((nread = ::read(fd, buf, HTTP_TS_SEND_BUFFER_SIZE)) < 0) {
+            ::close(fd);
+            ret = ERROR_HTTP_READ_FILE;
+            srs_warn("read file %s failed, ret=%d", fullpath.c_str(), ret);
+            return ret;
+        }
+        
+        left -= nread;
+        if ((ret = skt->write(buf, nread, NULL)) != ERROR_SUCCESS) {
+            break;
+        }
+    }
+    ::close(fd);
+    
     return ret;
     return ret;
 }
 }
 
 

@@ -70,6 +70,8 @@ protected:
     virtual bool is_handler_valid(SrsHttpMessage* req, int& status_code, std::string& reason_phrase);
     virtual bool is_handler_valid(SrsHttpMessage* req, int& status_code, std::string& reason_phrase);
     virtual int do_process_request(SrsSocket* skt, SrsHttpMessage* req);
     virtual int do_process_request(SrsSocket* skt, SrsHttpMessage* req);
 private:
 private:
+    virtual int response_regular_file(SrsSocket* skt, SrsHttpMessage* req, std::string fullpath);
+    virtual int response_ts_file(SrsSocket* skt, SrsHttpMessage* req, std::string fullpath);
     virtual std::string get_request_file(SrsHttpMessage* req);
     virtual std::string get_request_file(SrsHttpMessage* req);
 public:
 public:
     virtual std::string vhost();
     virtual std::string vhost();