From: hyc <hyc@400ebc74-4327-4243-bc38-086b20814532>
Date: Tue, 16 Mar 2010 04:47:55 +0000 (+0000)
Subject: More for RTMP_Write
X-Git-Tag: v2.4~181
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=88cdd45b864aadf99dbf31559f4de719ab2081b9;p=rtmpdump

More for RTMP_Write


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

diff --git a/librtmp/rtmp.c b/librtmp/rtmp.c
index 7dc17a8..32b1522 100644
--- a/librtmp/rtmp.c
+++ b/librtmp/rtmp.c
@@ -2498,7 +2498,6 @@ RTMP_SendPacket(RTMP *r, RTMPPacket *packet, bool queue)
       if (prevPacket->m_nInfoField2 == packet->m_nInfoField2
 	  && packet->m_headerType == RTMP_PACKET_SIZE_SMALL)
 	packet->m_headerType = RTMP_PACKET_SIZE_MINIMUM;
-
     }
 
   if (packet->m_headerType > 3)	// sanity
@@ -3530,60 +3529,82 @@ static const AVal av_setDataFrame = AVC("@setDataFrame");
 int
 RTMP_Write(RTMP *r, char *buf, int size)
 {
-  RTMPPacket packet;
+  RTMPPacket *pkt = &r->m_write;
   char *pend, *enc;
-  int s2 = size, ret;
+  int s2 = size, ret, num;
 
-  if (size < 11) {
-    /* FLV pkt too small */
-    return 0;
-  }
+  pkt->m_nChannel = 0x04;	// source channel
+  pkt->m_nInfoField2 = 0;
 
-  packet.m_nChannel = 0x04;	// source channel
-  packet.m_nInfoField2 = 0;
-
-  if (buf[0] == 'F' && buf[1] == 'L' && buf[2] == 'V')
+  while (s2)
     {
-      buf += 13;
-      s2 -= 13;
-    }
+      if (!pkt->m_nBytesRead)
+	{
+	  if (size < 11) {
+	    /* FLV pkt too small */
+	    return 0;
+	  }
 
-  packet.m_packetType = *buf++;
-  packet.m_nBodySize = AMF_DecodeInt24(buf);
-  buf += 3;
-  packet.m_nInfoField1 = AMF_DecodeInt24(buf);
-  buf += 3;
-  packet.m_nInfoField1 |= *buf++ << 24;
-  buf += 3;
-  s2 -= 11;
+	  if (buf[0] == 'F' && buf[1] == 'L' && buf[2] == 'V')
+	    {
+	      buf += 13;
+	      s2 -= 13;
+	    }
 
-  if (((packet.m_packetType == 0x08 || packet.m_packetType == 0x09) &&
-    !packet.m_nInfoField1) || packet.m_packetType == 0x12)
-    {
-      packet.m_headerType = RTMP_PACKET_SIZE_LARGE;
-      packet.m_hasAbsTimestamp = 1;
-      if (packet.m_packetType == 0x12)
-        packet.m_nBodySize += 16;
-    }
-  else
-    {
-      packet.m_headerType = RTMP_PACKET_SIZE_MEDIUM;
-      packet.m_hasAbsTimestamp = 0;
-    }
+	  pkt->m_packetType = *buf++;
+	  pkt->m_nBodySize = AMF_DecodeInt24(buf);
+	  buf += 3;
+	  pkt->m_nInfoField1 = AMF_DecodeInt24(buf);
+	  buf += 3;
+	  pkt->m_nInfoField1 |= *buf++ << 24;
+	  buf += 3;
+	  s2 -= 11;
+
+	  if (((pkt->m_packetType == 0x08 || pkt->m_packetType == 0x09) &&
+	    !pkt->m_nInfoField1) || pkt->m_packetType == 0x12)
+	    {
+	      pkt->m_headerType = RTMP_PACKET_SIZE_LARGE;
+	      if (pkt->m_packetType == 0x12)
+		pkt->m_nBodySize += 16;
+	    }
+	  else
+	    {
+	      pkt->m_headerType = RTMP_PACKET_SIZE_MEDIUM;
+	    }
 
-  if (!RTMPPacket_Alloc(&packet, packet.m_nBodySize))
-    {
-      Log(LOGDEBUG, "%s, failed to allocate packet", __FUNCTION__);
-      return false;
+	  if (!RTMPPacket_Alloc(pkt, pkt->m_nBodySize))
+	    {
+	      Log(LOGDEBUG, "%s, failed to allocate packet", __FUNCTION__);
+	      return false;
+	    }
+	  enc = pkt->m_body;
+	  pend = enc + pkt->m_nBodySize;
+	  if (pkt->m_packetType == 0x12)
+	    {
+	      enc = AMF_EncodeString(enc, pend, &av_setDataFrame);
+	      pkt->m_nBytesRead = enc - pkt->m_body;
+	    }
+	}
+      else
+	{
+	  enc = pkt->m_body + pkt->m_nBytesRead;
+	}
+      num = pkt->m_nBodySize - pkt->m_nBytesRead;
+      if (num > s2)
+	num = s2;
+      memcpy(enc, buf, num);
+      pkt->m_nBytesRead += num;
+      s2 -= num;
+      if (pkt->m_nBytesRead == pkt->m_nBodySize)
+	{
+	  ret = RTMP_SendPacket(r, pkt, false);
+	  RTMPPacket_Free(pkt);
+	  pkt->m_nBytesRead = 0;
+	  if (!ret)
+	    return -1;
+	  buf += 4;
+	  s2 -= 4;
+	}
     }
-  enc = packet.m_body;
-  pend = enc + packet.m_nBodySize;
-  if (packet.m_packetType == 0x12)
-    enc = AMF_EncodeString(enc, pend, &av_setDataFrame);
-  memcpy(enc, buf, packet.m_nBodySize);
-  ret = RTMP_SendPacket(r, &packet, false);
-  RTMPPacket_Free(&packet);
-  if (!ret)
-    return -1;
   return size;
 }
diff --git a/librtmp/rtmp.h b/librtmp/rtmp.h
index 62520a6..80d28bb 100644
--- a/librtmp/rtmp.h
+++ b/librtmp/rtmp.h
@@ -229,6 +229,7 @@ extern "C"
     AVal m_clientID;
 
     RTMP_READ m_read;
+	RTMPPacket m_write;
     RTMPSockBuf m_sb;
   } RTMP;