summaryrefslogtreecommitdiff
path: root/src/Xml2MetalinkProcessor.cc
diff options
context:
space:
mode:
authorTatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>2007-11-27 12:27:10 +0000
committerTatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>2007-11-27 12:27:10 +0000
commit506bc3db13f6130aca37c1df216a3e8a7967bf99 (patch)
tree09f6c0fcd703ad78e3d45eb38810582e8691b276 /src/Xml2MetalinkProcessor.cc
parent9d66150a8396d91d19bbee524243a5f371704929 (diff)
2007-11-27 Tatsuhiro Tsujikawa <tujikawa at rednoah dot com>
Rewritten to add content-type support. * src/DownloadHandler.{h, cc} * src/BtPostDownloadHandler.{h, cc} * test/BtPostDownloadHandlerTest.cc * src/MetalinkPostDownloadHandler.{h, cc} * test/MetalinkPostDownloadHandlerTest.cc * src/PostDownloadHandler.{h, cc} * src/DownloadHandlerConstants.{h, cc} * src/RequestGroup.cc * src/HttpResponseCommand.cc * src/FtpNegotiationCommand.cc * src/SingleFileDownloadContext.{h, cc} * src/RequestGroup.h * src/RequestGroupCriteria.h * src/ContentTypeRequestGroupCriteria.h Added 'mem' option value for --follow-metalink, --follow-torrent. If it is give, metalink/torrent file is not written to the disk, but just is kept in memory. Parsing is occurred on memory. * src/MetalinkHelper.{h, cc} * src/MetalinkProcessor.h * src/Xml2MetalinkProcessor.{h, cc} * test/Xml2MetalinkProcessorTest.cc * src/DownloadHandlerFactory.{h, cc} * test/DownloadHandlerFactoryTest.cc * src/PreDownloadHandler.{h, cc} * src/OptionHandlerFactory.cc * src/DefaultBtContext.{h, cc} * test/DefaultBtContextTest.cc * src/version_usage.cc * src/Metalink2RequestGroup.{h, cc} * src/RequestGroup.{h, cc} * src/a2functional.h * test/a2functionalTest.cc * src/MemoryBufferPreDownloadHandler.{h, cc} * src/OptionHandlerImpl.h * src/prefs.h * src/Util.{h, cc} * test/UtilTest.cc Keep DownloadResult rather than RequestGroup after downloads to reduce memory usage. * src/RequestGroupMan.{h, cc} * src/DownloadEngine.cc * src/BtDependency.{h, cc}: Changed the type of dependee from WeakHandle to SharedHandle because WeakHandle could be null. * src/RequestGroup.{h, cc} * src/DownloadEngineFactory.cc * src/DownloadResult.h Set totalLength after download finished * src/UnknownLengthPieceStorage.{h, cc} Keep torrent file specified in metalink in memory. * src/Metalink2RequestGroup.cc * src/BtDependency.cc * src/TrueRequestGroupCriteria.h Fixed the bug: seekg is used where seekp should be used. * src/ByteArrayDiskWriter.cc * test/ByteArraydiskWriterTest.cc
Diffstat (limited to 'src/Xml2MetalinkProcessor.cc')
-rw-r--r--src/Xml2MetalinkProcessor.cc44
1 files changed, 44 insertions, 0 deletions
diff --git a/src/Xml2MetalinkProcessor.cc b/src/Xml2MetalinkProcessor.cc
index 87c34c33..2eae4f3d 100644
--- a/src/Xml2MetalinkProcessor.cc
+++ b/src/Xml2MetalinkProcessor.cc
@@ -35,6 +35,7 @@
#include "Xml2MetalinkProcessor.h"
#include "DlAbortEx.h"
#include "Util.h"
+#include "BinaryStream.h"
#include <libxml/parser.h>
#include <libxml/xpath.h>
#include <libxml/xpathInternals.h>
@@ -62,6 +63,49 @@ MetalinkerHandle Xml2MetalinkProcessor::parseFile(const string& filename) {
if(!doc) {
throw new DlAbortEx("Cannot parse metalink file %s", filename.c_str());
}
+ return processDoc(doc);
+}
+
+MetalinkerHandle Xml2MetalinkProcessor::parseFromBinaryStream(const BinaryStreamHandle& binaryStream) {
+ release();
+ int32_t bufSize = 4096;
+ unsigned char buf[bufSize];
+
+ int32_t res = binaryStream->readData(buf, 4, 0);
+ if(res != 4) {
+ throw new DlAbortEx("Too small data for metalink parsing.");
+ }
+
+ xmlParserCtxtPtr ctx = xmlCreatePushParserCtxt(0, 0, (const char*)buf, res, 0);
+ try {
+ int64_t readOffset = res;
+ while(1) {
+ int32_t res = binaryStream->readData(buf, bufSize, readOffset);
+ if(res == 0) {
+ break;
+ }
+ if(xmlParseChunk(ctx, (const char*)buf, res, 0) != 0) {
+ throw new DlAbortEx("Cannot parse metalink file");
+ }
+ readOffset += res;
+ }
+ xmlParseChunk(ctx, (const char*)buf, 0, 1);
+ doc = ctx->myDoc;
+
+ xmlFreeParserCtxt(ctx);
+ } catch(Exception* e) {
+ xmlFreeParserCtxt(ctx);
+ throw;
+ }
+
+ if(!doc) {
+ throw new DlAbortEx("Cannot parse metalink file");
+ }
+ return processDoc(doc);
+}
+
+MetalinkerHandle Xml2MetalinkProcessor::processDoc(xmlDocPtr doc)
+{
context = xmlXPathNewContext(doc);
if(!context) {
throw new DlAbortEx("Cannot create new xpath context");