1 files changed, 469 insertions, 0 deletions
diff --git a/src/bencode.cc b/src/bencode.cc
new file mode 100644
index 00000000..60a790a2
--- /dev/null
+++ b/src/bencode.cc
@@ -0,0 +1,469 @@
+/* <!-- copyright */
+/*
+ * aria2 - The high speed download utility
+ *
+ * Copyright (C) 2006 Tatsuhiro Tsujikawa
+ *
+ * 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.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+ *
+ * In addition, as a special exception, the copyright holders give
+ * permission to link the code of portions of this program with the
+ * OpenSSL library under certain conditions as described in each
+ * individual source file, and distribute linked combinations
+ * including the two.
+ * You must obey the GNU General Public License in all respects
+ * for all of the code used other than OpenSSL.  If you modify
+ * file(s) with this exception, you may extend this exception to your
+ * version of the file(s), but you are not obligated to do so.  If you
+ * do not wish to do so, delete this exception statement from your
+ * version.  If you delete this exception statement from all source
+ * files in the program, then also delete it here.
+ */
+/* copyright --> */
+#include "bencode.h"
+
+#include <fstream>
+#include <sstream>
+
+#include "StringFormat.h"
+#include "Util.h"
+
+namespace aria2 {
+
+namespace bencode {
+
+const BDE BDE::none;
+
+BDE::BDE() throw():_type(TYPE_NONE) {}
+
+BDE::BDE(Integer integer) throw():_type(TYPE_INTEGER),
+				  _integer(new Integer(integer)) {}
+
+
+BDE::BDE(const std::string& string) throw():_type(TYPE_STRING),
+					    _string(new std::string(string)) {}
+
+BDE::BDE(const char* cstring) throw():_type(TYPE_STRING),
+				      _string(new std::string(cstring)) {}
+
+BDE::BDE(const char* data, size_t length) throw():
+  _type(TYPE_STRING),
+  _string(new std::string(&data[0], &data[length])) {}
+
+BDE::BDE(const unsigned char* data, size_t length) throw():
+  _type(TYPE_STRING),
+  _string(new std::string(&data[0], &data[length])) {}
+
+BDE BDE::dict() throw()
+{
+  BDE bde;
+  bde._type = TYPE_DICT;
+  bde._dict.reset(new Dict());
+  return bde;
+}
+
+BDE BDE::list() throw()
+{
+  BDE bde;
+  bde._type = TYPE_LIST;
+  bde._list.reset(new List());
+  return bde;
+}
+
+// Test for Null data
+bool BDE::isNone() const throw()
+{
+  return _type == TYPE_NONE;
+}
+
+// Integer Interface
+
+bool BDE::isInteger() const throw()
+{
+  return _type == TYPE_INTEGER;
+}
+
+BDE::Integer BDE::i() const throw(RecoverableException)
+{
+  if(isInteger()) {
+    return *_integer.get();
+  } else {
+    throw RecoverableException("Not Integer");
+  }
+}
+
+// String Interface
+
+bool BDE::isString() const throw()
+{
+  return _type == TYPE_STRING;
+}
+
+const std::string& BDE::s() const throw(RecoverableException)
+{
+  if(isString()) {
+    return *_string.get();
+  } else {
+    throw RecoverableException("Not String");
+  }
+}
+
+// Dictionary Interface
+
+bool BDE::isDict() const throw()
+{
+  return _type == TYPE_DICT;
+}
+
+BDE& BDE::operator[](const std::string& key) throw(RecoverableException)
+{
+  if(isDict()) {
+    return (*_dict.get())[key];
+  } else {
+    throw RecoverableException("Not Dict");
+  }
+}
+
+const BDE& BDE::operator[](const std::string& key) const
+  throw(RecoverableException)
+{
+  if(isDict()) {
+    BDE::Dict::const_iterator i = _dict->find(key);
+    if(i == _dict->end()) {
+      return none;
+    } else {
+      return (*i).second;
+    }
+  } else {
+    throw RecoverableException("Not Dict");
+  }
+}
+
+bool BDE::containsKey(const std::string& key) const throw(RecoverableException)
+{
+  if(isDict()) {
+    return _dict->find(key) != _dict->end();
+  } else {
+    throw RecoverableException("Not Dict");
+  }
+}
+
+
+BDE::Dict::iterator BDE::dictBegin() throw(RecoverableException)
+{
+  if(isDict()) {
+    return _dict->begin();
+  } else {
+    throw RecoverableException("Not Dict");
+  }
+}
+
+BDE::Dict::const_iterator BDE::dictBegin() const throw(RecoverableException)
+{
+  if(isDict()) {
+    return _dict->begin();
+  } else {
+    throw RecoverableException("Not Dict");
+  }
+}
+
+BDE::Dict::iterator BDE::dictEnd() throw(RecoverableException)
+{
+  if(isDict()) {
+    return _dict->end();
+  } else {
+    throw RecoverableException("Not Dict");
+  }
+}
+
+BDE::Dict::const_iterator BDE::dictEnd() const throw(RecoverableException)
+{
+  if(isDict()) {
+    return _dict->end();
+  } else {
+    throw RecoverableException("Not Dict");
+  }
+}
+
+// List Interface
+
+bool BDE::isList() const throw()
+{
+  return _type == TYPE_LIST;
+}
+
+void BDE::append(const BDE& bde) throw(RecoverableException)
+{
+  if(isList()) {
+    _list->push_back(bde);
+  } else {
+    throw RecoverableException("Not List");
+  }
+}
+
+void BDE::operator<<(const BDE& bde) throw(RecoverableException)
+{
+  if(isList()) {
+    _list->push_back(bde);
+  } else {
+    throw RecoverableException("Not List");
+  }
+}
+
+BDE& BDE::operator[](size_t index) throw(RecoverableException)
+{
+  if(isList()) {
+    return (*_list.get())[index];
+  } else {
+    throw RecoverableException("Not List");
+  }
+}
+
+const BDE& BDE::operator[](size_t index) const throw(RecoverableException)
+{
+  if(isList()) {
+    return (*_list.get())[index];
+  } else {
+    throw RecoverableException("Not List");
+  }
+}
+
+BDE::List::iterator BDE::listBegin() throw(RecoverableException)
+{
+  if(isList()) {
+    return _list->begin();
+  } else {
+    throw RecoverableException("Not List");
+  }
+}
+
+BDE::List::const_iterator BDE::listBegin() const throw(RecoverableException)
+{
+  if(isList()) {
+    return _list->begin();
+  } else {
+    throw RecoverableException("Not List");
+  }
+}
+
+BDE::List::iterator BDE::listEnd() throw(RecoverableException)
+{
+  if(isList()) {
+    return _list->end();
+  } else {
+    throw RecoverableException("Not List");
+  }
+}
+
+BDE::List::const_iterator BDE::listEnd() const throw(RecoverableException)
+{
+  if(isList()) {
+    return _list->end();
+  } else {
+    throw RecoverableException("Not List");
+  }
+}
+
+// Callable from List and Dict
+size_t BDE::size() const throw(RecoverableException)
+{
+  if(isDict()) {
+    return _dict->size();
+  } else if(isList()) {
+    return _list->size();
+  } else {
+    throw RecoverableException("Not Dict nor List");
+  }
+}
+
+// Callable from List and Dict
+bool BDE::empty() const throw(RecoverableException)
+{
+  if(isDict()) {
+    return _dict->empty();
+  } else if(isList()) {
+    return _list->empty();
+  } else {
+    throw RecoverableException("Not Dict nor List");
+  }
+}
+
+static BDE decodeiter(std::istream& ss) throw(RecoverableException);
+
+static void checkdelim(std::istream& ss, const char delim = ':')
+  throw(RecoverableException)
+{
+  char d;
+  if(!(ss.get(d) && d == delim)) {
+    throw RecoverableException
+      (StringFormat("Delimiter '%c' not found.", delim).str());
+  }
+}
+
+static std::string decoderawstring(std::istream& ss)
+  throw(RecoverableException)
+{
+  size_t length;
+  ss >> length;
+  if(!ss) {
+    throw RecoverableException("Integer expected but none found.");
+  }
+  // TODO check length, it must be less than or equal to INT_MAX
+  checkdelim(ss);
+  char* buf = new char[length];
+  ss.read(buf, length);
+  if(ss.gcount() != static_cast<int>(length)) {
+    throw RecoverableException
+      (StringFormat("Expected %lu bytes of data, but only %d read.",
+		    static_cast<unsigned long>(length), ss.gcount()).str());
+  }
+  std::string str(&buf[0], &buf[length]);
+  delete [] buf;
+  return str;
+}
+
+static BDE decodestring(std::istream& ss) throw(RecoverableException)
+{
+  return BDE(decoderawstring(ss));
+}
+
+static BDE decodeinteger(std::istream& ss) throw(RecoverableException)
+{
+  BDE::Integer integer;
+  ss >> integer;
+  if(!ss) {
+    throw RecoverableException("Integer expected but none found");
+  }
+  checkdelim(ss, 'e');
+  return BDE(integer);
+}
+
+static BDE decodedict(std::istream& ss) throw(RecoverableException)
+{
+  BDE dict = BDE::dict();
+  char c;
+  while(ss.get(c)) {
+    if(c == 'e') {
+      return dict;
+    } else {
+      ss.unget();
+      std::string key = decoderawstring(ss);
+      dict[key] = decodeiter(ss);
+    }
+  }
+  throw RecoverableException("Unexpected EOF in dict context. 'e' expected.");
+}
+
+static BDE decodelist(std::istream& ss) throw(RecoverableException)
+{
+  BDE list = BDE::list();
+  char c;
+  while(ss.get(c)) {
+    if(c == 'e') {
+      return list;
+    } else {
+      ss.unget();
+      list << decodeiter(ss);
+    }
+  }
+  throw RecoverableException("Unexpected EOF in list context. 'e' expected.");
+}
+
+static BDE decodeiter(std::istream& ss) throw(RecoverableException)
+{
+  char c;
+  if(!ss.get(c)) {
+    throw RecoverableException("Unexpected EOF in term context."
+			       " 'd', 'l', 'i' or digit is expected.");
+  }
+  if(c == 'd') {
+    return decodedict(ss);
+  } else if(c == 'l') {
+    return decodelist(ss);
+  } else if(c == 'i') {
+    return decodeinteger(ss);
+  } else {
+    ss.unget();
+    return decodestring(ss);
+  }
+}
+
+BDE decode(std::istream& in) throw(RecoverableException)
+{
+  return decodeiter(in);
+}
+
+BDE decode(const std::string& s) throw(RecoverableException)
+{
+  if(s.empty()) {
+    return BDE::none;
+  }
+  std::istringstream ss(s);
+
+  return decodeiter(ss);
+}
+
+BDE decode(const unsigned char* data, size_t length) throw(RecoverableException)
+{
+  return decode(std::string(&data[0], &data[length]));
+}
+
+BDE decodeFromFile(const std::string& filename) throw(RecoverableException)
+{
+  std::ifstream f(filename.c_str());
+  if(f) {
+    return decode(f);
+  } else {
+    throw RecoverableException
+      (StringFormat("Cannot open file '%s'.", filename.c_str()).str());
+  }
+}
+
+static void encodeIter(std::ostream& o, const BDE& bde) throw()
+{
+  if(bde.isInteger()) {
+    o << "i" << bde.i() << "e";
+  } else if(bde.isString()) {
+    const std::string& s = bde.s();
+    o << s.size() << ":";
+    o.write(s.data(), s.size());
+  } else if(bde.isDict()) {
+    o << "d";
+    for(BDE::Dict::const_iterator i = bde.dictBegin(); i != bde.dictEnd(); ++i){
+      const std::string& key = (*i).first;
+      o << key.size() << ":";
+      o.write(key.data(), key.size());
+      encodeIter(o, (*i).second);
+    }
+    o << "e";
+  } else if(bde.isList()) {
+    o << "l";
+    for(BDE::List::const_iterator i = bde.listBegin(); i != bde.listEnd(); ++i){
+      encodeIter(o, *i);
+    }
+    o << "e";
+  }
+}
+
+std::string encode(const BDE& bde) throw()
+{
+  std::ostringstream ss;
+  encodeIter(ss, bde);
+  return ss.str();
+}
+
+} // namespace bencode
+
+} // namespace aria2