FLV配信用パッチをマージ。(github:niwakazoider/peercast@21998fef7e24f437ef8d50e17562ba95eb5c1843)

[peercast-im/PeerCastIM.git] / core / common / flv.cpp

diff --git a/core/common/flv.cpp b/core/common/flv.cpp
new file mode 100644 (file)
index 0000000..562ef13
--- /dev/null
+++ b/core/common/flv.cpp
@@ -0,0 +1,136 @@
+// ------------------------------------------------
+// File : flv.cpp
+// Date: 14-jan-2017
+// Author: niwakazoider
+//
+// Modified by Eru @2017.1.15 for compilation error at VS2008
+//
+// (c) 2002-3 peercast.org
+// ------------------------------------------------
+// This program is free software; you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation; either version 2 of the License, or
+// (at your option) any later version.
+
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+// ------------------------------------------------
+
+#include "channel.h"
+#include "flv.h"
+#include "string.h"
+#include "stdio.h"
+#ifdef _DEBUG
+#include "chkMemoryLeak.h"
+#define DEBUG_NEW new(__FILE__, __LINE__)
+#define new DEBUG_NEW
+#endif
+
+
+// ------------------------------------------
+void FLVStream::readEnd(Stream &, Channel *)
+{
+}
+
+// ------------------------------------------
+void FLVStream::readHeader(Stream &, Channel *)
+{
+}
+
+// ------------------------------------------
+int FLVStream::readPacket(Stream &in, Channel *ch)
+{
+       bool headerUpdate = false;
+
+       if (ch->streamPos == 0) {
+               bitrate = 0;
+               FLVFileHeader header = FLVFileHeader();
+               header.read(in);
+               fileHeader = header;
+               headerUpdate = true;
+       }
+
+       FLVTag flvTag;
+       flvTag.read(in);
+       
+       switch (flvTag.type)
+       {
+               case TAG_SCRIPTDATA:
+               {
+                       AMFObject amf;
+                       MemoryStream flvmem = MemoryStream(flvTag.data, flvTag.size);
+                       if (amf.readMetaData(flvmem)) {
+                               flvmem.close();
+                               bitrate = amf.bitrate;
+                               metaData = flvTag;
+                               headerUpdate = true;
+                       }
+               }
+               case TAG_VIDEO:
+               {
+                       //AVC Header
+                       if (flvTag.data[0] == 0x17 && flvTag.data[1] == 0x00 &&
+                               flvTag.data[2] == 0x00 && flvTag.data[3] == 0x00) {
+                               avcHeader = flvTag;
+                               headerUpdate = true;
+                       }
+               }
+               case TAG_AUDIO:
+               {
+                       //AAC Header
+                       if (flvTag.data[0] == 0xaf && flvTag.data[1] == 0x00) {
+                               aacHeader = flvTag;
+                               headerUpdate = true;
+                       }
+               }
+       }
+       
+       if (headerUpdate && fileHeader.size>0) {
+               int len = fileHeader.size;
+               if (metaData.type==TAG_SCRIPTDATA) len += metaData.packetSize;
+               if (avcHeader.type == TAG_VIDEO) len += avcHeader.packetSize;
+               if (aacHeader.type == TAG_AUDIO) len += aacHeader.packetSize;
+               MemoryStream mem = MemoryStream(ch->headPack.data, len);
+               mem.write(fileHeader.data, fileHeader.size);
+               if (metaData.type == TAG_SCRIPTDATA) mem.write(metaData.packet, metaData.packetSize);
+               if (avcHeader.type == TAG_VIDEO) mem.write(avcHeader.packet, avcHeader.packetSize);
+               if (aacHeader.type == TAG_AUDIO) mem.write(aacHeader.packet, aacHeader.packetSize);
+
+               ch->info.bitrate = bitrate;
+
+               ch->headPack.type = ChanPacket::T_HEAD;
+               ch->headPack.len = mem.pos;
+               ch->headPack.pos = ch->streamPos;
+               ch->newPacket(ch->headPack);
+
+               ch->streamPos += ch->headPack.len;
+       }
+       else {
+               ChanPacket pack;
+
+               MemoryStream mem = MemoryStream(flvTag.packet, flvTag.packetSize);
+
+               int rlen = flvTag.packetSize;
+               while (rlen)
+               {
+                       int rl = rlen;
+                       if (rl > ChanMgr::MAX_METAINT)
+                               rl = ChanMgr::MAX_METAINT;
+
+                       pack.init(ChanPacket::T_DATA, pack.data, rl, ch->streamPos);
+                       mem.read(pack.data, pack.len);
+                       ch->newPacket(pack);
+                       ch->checkReadDelay(pack.len);
+                       ch->streamPos += pack.len;
+
+                       rlen -= rl;
+               }
+
+               mem.close();
+               
+       }
+       
+       return 0;
+}