add method to ensure buffer capacity

[ecparse.git] / CollectionReader.cpp

diff --git a/CollectionReader.cpp b/CollectionReader.cpp
index 27f9ca6..f58e5bc 100644 (file)
--- a/CollectionReader.cpp
+++ b/CollectionReader.cpp
@@ -2,13 +2,117 @@
  * © 2008 by David Roden <droden@gmail.com>
  */
 
+#include <stdlib.h>
+#include <string.h>
 #include "CollectionReader.h"
+#include "GlobalSettings.h"
 
 CollectionReader::CollectionReader(ReaderInput* readerInput) {
        this->readerInput = readerInput;
+       firstLink = true;
 }
 
 CollectionReader::~CollectionReader() {
-       delete readerInput;
+}
+
+bool CollectionReader::isLineBreakPresent() {
+       size_t indexOfLineBreak = growingBuffer.indexOf('\n');
+       GlobalSettings::isVerbose() && (indexOfLineBreak != (size_t) -1) && fprintf(stderr, "[%s:%d] found line break at %d.\n", __FILE__, __LINE__, indexOfLineBreak);
+       GlobalSettings::isVerbose() && (indexOfLineBreak == (size_t) -1) && fprintf(stderr, "[%s:%d] could not find line break.\n", __FILE__, __LINE__);
+       return indexOfLineBreak != (size_t) -1;
+}
+
+ED2KLink* CollectionReader::getNextLink() {
+       if (readerInput->isEOF() && !growingBuffer.getRemaining()) {
+               GlobalSettings::isVerbose() && fprintf(stderr, "[%s:%d] readInput EOF reached.\n", __FILE__, __LINE__);
+               return NULL;
+       }
+       if (firstLink) {
+               identifyCollectionType();
+               if (readerInput->isEOF() && !growingBuffer.getRemaining()) {
+                       GlobalSettings::isVerbose() && fprintf(stderr, "[%s:%d] readInput EOF reached.\n", __FILE__, __LINE__);
+                       return NULL;
+               }
+               firstLink = false;
+       }
+       if (isTextCollection) {
+               while (!readerInput->isEOF() && !isLineBreakPresent()) {
+                       readMoreBytes();
+               }
+               if (readerInput->isEOF() && !growingBuffer.getRemaining()) {
+                       GlobalSettings::isVerbose() && fprintf(stderr, "[%s:%d] readInput EOF reached.\n", __FILE__, __LINE__);
+                       return NULL;
+               }
+               size_t indexOfLineBreak = growingBuffer.indexOf('\n');
+               char* line;
+               if (indexOfLineBreak == (size_t) -1) {
+                       GlobalSettings::isVerbose() && fprintf(stderr, "[%s:%d] could not find line break, using remainder of file.\n", __FILE__, __LINE__);
+                       indexOfLineBreak = growingBuffer.getRemaining();
+                       line = (char*) malloc(indexOfLineBreak + 1);
+                       growingBuffer.read(line, indexOfLineBreak);
+                       line[indexOfLineBreak] = '\0';
+               } else {
+                       line = (char*) malloc(indexOfLineBreak + 1);
+                       growingBuffer.read(line, indexOfLineBreak + 1);
+                       if (line[indexOfLineBreak] == '\n') {
+                               line[indexOfLineBreak] = '\0';
+                       }
+                       if (line[indexOfLineBreak - 1] == '\r') {
+                               line[indexOfLineBreak - 1] = '\0';
+                       }
+               }
+               growingBuffer.cut();
+               GlobalSettings::isVerbose() && fprintf(stderr, "[%s:%d] got line: %s\n", __FILE__, __LINE__, line);
+               ED2KLink* ed2kLink = ED2KLink::parseED2KLink(line);
+               free(line);
+               return ed2kLink;
+       } else {
+       }
+       return NULL;
+}
+
+bool CollectionReader::ensureBufferCapacity(size_t byteCount) {
+       while (!readerInput->isEOF() && (growingBuffer.getRemaining() < byteCount)) {
+               GlobalSettings::isVerbose() && fprintf(stderr, "[%s:%d] only %d bytes remaning, need at least %d, reading more bytes.\n", __FILE__, __LINE__, growingBuffer.getRemaining(), byteCount);
+               readMoreBytes();
+       }
+       if (readerInput->isEOF() && (growingBuffer.getRemaining() < byteCount)) {
+               GlobalSettings::isVerbose() && fprintf(stderr, "[%s:%d] need %d more bytes, but file is EOF.\n", __FILE__, __LINE__, (byteCount - growingBuffer.getRemaining()));
+               return false;
+       }
+       return true;
+}
+
+void CollectionReader::readMoreBytes() {
+       char buffer[1024];
+       size_t readBytes;
+
+       readBytes = readerInput->read(buffer, 1024);
+       growingBuffer.write(buffer, readBytes);
+}
+
+void CollectionReader::identifyCollectionType() {
+       int version;
+       size_t readBytes;
+
+       readBytes = readerInput->read(&version, 4);
+       if ((readBytes < 4) || readerInput->isEOF()) {
+               return;
+       }
+       if (version == 0x01) {
+               isTextCollection = false;
+               this->version = 1;
+               GlobalSettings::isVerbose() && fprintf(stderr, "[%s:%d] identified binary collection, version 1\n", __FILE__, __LINE__);
+       } else if (version == 0x02) {
+               isTextCollection = false;
+               this->version = 2;
+               GlobalSettings::isVerbose() && fprintf(stderr, "[%s:%d] identified binary collection, version 2\n", __FILE__, __LINE__);
+       } else if (!strncmp("ed2k", (char*) &version, 4)) {
+               isTextCollection = true;
+               growingBuffer.write(&version, 4);
+               GlobalSettings::isVerbose() && fprintf(stderr, "[%s:%d] identified text collection\n", __FILE__, __LINE__);
+       } else  {
+               GlobalSettings::isVerbose() && fprintf(stderr, "[%s:%d] could not identify collection!\n", __FILE__, __LINE__);
+       }
 }