move codec to kernel.

trunk/research/librtmp/srs_flv_injecter.c

@@ -50,6 +50,7 @@ int main(int argc, char** argv)
     srs_flv_t oc = NULL;
     
     // temp variables.
+    int tmp_file_size = 0;
     char* tmp_file;
     
     if (argc <= 2) {
@@ -68,20 +69,30 @@ int main(int argc, char** argv)
     in_flv_file = argv[1];
     out_flv_file = argv[2];
     
-    tmp_file = (char*)malloc(strlen(out_flv_file) + strlen(".tmp") + 1);
+    tmp_file_size = strlen(out_flv_file) + strlen(".tmp") + 1;
+    tmp_file = (char*)malloc(tmp_file_size);
+    snprintf(tmp_file, tmp_file_size, "%s.tmp", out_flv_file);
     
     trace("inject flv file keyframes to metadata.");
     trace("srs(simple-rtmp-server) client librtmp library.");
     trace("version: %d.%d.%d", srs_version_major(), srs_version_minor(), srs_version_revision());
     trace("input:  %s", in_flv_file);
     trace("output:  %s", out_flv_file);
+    trace("tmp_file:  %s", tmp_file);
 
-    if ((ret = process(in_flv_file, out_flv_file, &ic, &oc)) != 0) {
-        return ret;
-    }
+    ret = process(in_flv_file, tmp_file, &ic, &oc);
     
     srs_flv_close(ic);
     srs_flv_close(oc);
+    
+    if (ret != 0) {
+        unlink(tmp_file);
+        trace("error, remove tmp file.");
+    } else {
+        rename(tmp_file, out_flv_file);
+        trace("completed, rename to %s", out_flv_file);
+    }
+    
     free(tmp_file);
     
     return ret;
@@ -110,9 +121,86 @@ int process(const char* in_flv_file, const char* out_flv_file, srs_flv_t* pic, s
     return ret;
 }
 
+int parse_metadata(char* data, int size, srs_amf0_t* pname, srs_amf0_t* pdata)
+{
+    int ret = 0;
+    
+    int nparsed = 0;
+    *pname = srs_amf0_parse(data, size, &nparsed);
+    
+    if (*pname == NULL || nparsed >= size) {
+        trace("invalid amf0 name data.");
+        return -1;
+    }
+    
+    *pdata = srs_amf0_parse(data + nparsed, size - nparsed, &nparsed);
+    if (*pdata == NULL || nparsed > size) {
+        trace("invalid amf0 value data");
+        return -1;
+    }
+    
+    return ret;
+}
+
 int inject_flv(srs_flv_t ic, srs_flv_t oc)
 {
     int ret = 0;
     
+    // flv header
+    char header[13];
+    // packet data
+    char type;
+    u_int32_t timestamp = 0;
+    char* data = NULL;
+    int32_t size;
+    
+    // metadata
+    srs_amf0_t amf0_name = NULL;
+    srs_amf0_t amf0_data = NULL;
+    
+    // reset to generate metadata
+    srs_flv_lseek(ic, 0);
+    
+    if ((ret = srs_flv_read_header(oc, header)) != 0) {
+        return ret;
+    }
+    
+    trace("start inject flv");
+    for (;;) {
+        // tag header
+        if ((ret = srs_flv_read_tag_header(oc, &type, &size, &timestamp)) != 0) {
+            if (srs_flv_is_eof(ret)) {
+                trace("parse completed.");
+                return 0;
+            }
+            trace("flv get packet failed. ret=%d", ret);
+            return ret;
+        }
+        
+        if (size <= 0) {
+            trace("invalid size=%d", size);
+            break;
+        }
+        
+        // TODO: FIXME: mem leak when error.
+        data = (char*)malloc(size);
+        if ((ret = srs_flv_read_tag_data(oc, data, size)) != 0) {
+            return ret;
+        }
+        
+        // data tag
+        if (type == SRS_RTMP_TYPE_VIDEO) {
+        } else if (type == SRS_RTMP_TYPE_SCRIPT) {
+            if ((ret = parse_metadata(data, size, &amf0_name, &amf0_data)) != 0) {
+                return ret;
+            }
+        }
+        
+        free(data);
+    }
+    
+    srs_amf0_free(amf0_name);
+    srs_amf0_free(amf0_data);
+    
     return ret;
 }

trunk/research/librtmp/srs_flv_parser.c

@@ -233,7 +233,7 @@ int parse_flv(srs_flv_t flv)
                 trace("parse completed.");
                 return 0;
             }
-            trace("irtmp get packet failed. ret=%d", ret);
+            trace("flv get packet failed. ret=%d", ret);
             return ret;
         }