diff options
Diffstat (limited to 'Src/Plugins/Input/in_flv/FLVReader.cpp')
| -rw-r--r-- | Src/Plugins/Input/in_flv/FLVReader.cpp | 211 |
1 files changed, 211 insertions, 0 deletions
diff --git a/Src/Plugins/Input/in_flv/FLVReader.cpp b/Src/Plugins/Input/in_flv/FLVReader.cpp new file mode 100644 index 00000000..c756cc70 --- /dev/null +++ b/Src/Plugins/Input/in_flv/FLVReader.cpp @@ -0,0 +1,211 @@ +#include "FLVReader.h" +#include "FLVHeader.h" +#include "FLVStreamheader.h" +#include "BackgroundDownloader.h" +#include "../nu/AutoChar.h" +#include "FileProcessor.h" +#include "ProgressiveProcessor.h" +#include "StreamProcessor.h" +#include "../../..\Components\wac_network\wac_network_http_receiver_api.h" +#include <shlwapi.h> + +FLVReader::FLVReader(const wchar_t *_url) +{ + processor = 0; + url = _wcsdup(_url); + end_of_stream=false; + + killswitch=false; + DWORD threadId; + flvThread=CreateThread(NULL, NULL, ParserThreadStub, this, 0, &threadId); + SetThreadPriority(flvThread, THREAD_PRIORITY_BELOW_NORMAL); +} + +FLVReader::~FLVReader() +{ + free(url); + delete processor; +} + +uint64_t FLVReader::Seek(uint64_t position) +{ + if (processor) + return processor->Seek(position); + else + return -1; +} + +size_t FLVReader::Read(void *data, size_t bytes) +{ + if (processor) + return processor->Read(data, bytes); + else + return 0; +} + +uint64_t FLVReader::GetProcessedPosition() +{ + if (processor) + return processor->GetProcessedPosition(); + else + return 0; +} + +void FLVReader::ProcessFile() +{ + if (!processor) + return; + + while (processor->Process() == FLV_OK) + { + if (killswitch) + return ; + } +} + +int FLVReader::OnConnect(api_httpreceiver *http) +{ + if (http->content_length()) + processor = new ProgressiveProcessor; + else + processor = new StreamProcessor; + return 0; +} + +int FLVReader::OnData(void *data, size_t datalen) +{ + if (!processor) + return 1; + + bool needSleep=false; + while (datalen) + { + if (killswitch) + return 1; + + if (needSleep) + { + Sleep(10); + needSleep=false; + } + + size_t written=0; + switch (processor->Write(data, datalen, &written)) + { + case FLVPROCESSOR_WRITE_ERROR: + return 1; + case FLVPROCESSOR_WRITE_WAIT: + needSleep=true; + break; + } + + datalen -= written; + + if (written) + { + while (1) + { + if (killswitch) + return 1; + + int ret = processor->Process(); + if (ret == FLV_OK) + continue; + if (ret == FLV_NEED_MORE_DATA) + break; + if (ret == FLV_ERROR) + return 1; + } + } + } + return !!killswitch; +} + +bool FLVReader::GetFrame(size_t frameIndex, FrameData &frameData) +{ + if (processor) + return processor->GetFrame(frameIndex, frameData); + else + return false; +} + +/* +Test URLs (valid of as oct 23 2007) +http://pdl.stream.aol.com/aol/us/aolmusic/artists/astralwerks/2220s/2220s_shootyourgun_hakjfh_700_dl.flv +http://pdl.stream.aol.com/aol/us/aolmusic/sessions/2007/amberpacific/suc_amberpacific_fallbackintomylife_700_dl.flv +http://pdl.stream.aol.com/aol/us/aolmusic/sessions/2007/amberpacific/suc_amberpacific_gonesoyoung_700_dl.flv +http://pdl.stream.aol.com/aol/us/aolmusic/sessions/2007/amberpacific/suc_amberpacific_soyesterday_700_dl.flv +http://pdl.stream.aol.com/aol/us/aolmusic/sessions/2007/amberpacific/suc_amberpacific_watchingoverme_700_dl.flv + +a potential crasher: +http://pdl.stream.aol.com/aol/us/aolcomvideo/TVGuide/johncmcginley_6346/johncmcginley_6346_460_700_dl.flv + +FLV streams: +http://208.80.52.96/CBS_R20_452P_F128?ext=.flv +http://208.80.54.37/KROQFM?ext=.flv +http://208.80.52.96/CBS_R20_568P_F128?ext=.flv +*/ +DWORD FLVReader::ParserThread() +{ + if (PathIsURL(url)) + { + Downloader downloader; + downloader.Download(AutoChar(url), this); + } + else + { + processor = new FileProcessor(url); + ProcessFile(); + } + end_of_stream=true; + return 0; +} + +void FLVReader::Kill() +{ + killswitch=true; + WaitForSingleObject(flvThread, INFINITE); + CloseHandle(flvThread); +} + +void FLVReader::SignalKill() +{ + killswitch=true; +} + +bool FLVReader::IsEOF() +{ + return end_of_stream; +} + +bool FLVReader::GetPosition(int time_in_ms, size_t *frameIndex, bool needVideoKeyFrame) +{ + if (processor) + return processor->GetPosition(time_in_ms, frameIndex, needVideoKeyFrame); + else + return false; +} + +uint32_t FLVReader::GetMaxTimestamp() +{ + if (processor) + return processor->GetMaxTimestamp(); + else + return 0; +} + +bool FLVReader::IsStreaming() +{ + if (processor) + return processor->IsStreaming(); + else + return true; +} + +FLVHeader *FLVReader::GetHeader() +{ + if (processor) + return processor->GetHeader(); + else + return 0; +}
\ No newline at end of file |