More for RTMP_Write

git-svn-id: svn://svn.mplayerhq.hu/rtmpdump/trunk@359 400ebc74-4327-4243-bc38-086b20814532

diff --git a/librtmp/rtmp.c b/librtmp/rtmp.c
index b3a748f2676ecfefbecc652d6c1b7b6471e70b9b..8675fb52ae4c4608b1afacdd4f6766aaa52e29d1 100644 (file)
--- a/librtmp/rtmp.c
+++ b/librtmp/rtmp.c
@@ -3599,6 +3599,7 @@ RTMP_Write(RTMP *r, char *buf, int size)
       memcpy(enc, buf, num);
       pkt->m_nBytesRead += num;
       s2 -= num;
+      buf += num;
       if (pkt->m_nBytesRead == pkt->m_nBodySize)
        {
          ret = RTMP_SendPacket(r, pkt, false);
@@ -3608,7 +3609,9 @@ RTMP_Write(RTMP *r, char *buf, int size)
            return -1;
          buf += 4;
          s2 -= 4;
+         if (s2 < 0)
+           break;
        }
     }
-  return size;
+  return size+s2;
 }