fix librtmp get bytes bug.

9 years ago · 53d2eb5e02
parent 588de3e7b4
commit 53d2eb5e02
1 changed files with 6 additions and 0 deletions
--- a/trunk/src/libs/srs_librtmp.cpp
+++ b/trunk/src/libs/srs_librtmp.cpp
@ -2002,6 +2002,9 @@ int64_t srs_utils_send_bytes(srs_rtmp_t rtmp)
 {
    srs_assert(rtmp != NULL);
    Context* context = (Context*)rtmp;
+    if (!context->rtmp) {
+        return 0;
+    }
    return context->rtmp->get_send_bytes();
 }

@ -2009,6 +2012,9 @@ int64_t srs_utils_recv_bytes(srs_rtmp_t rtmp)
 {
    srs_assert(rtmp != NULL);
    Context* context = (Context*)rtmp;
+    if (!context->rtmp) {
+        return 0;
+    }
    return context->rtmp->get_recv_bytes();
 }