123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484 |
- /* <!-- 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"
- #include "RecoverableException.h"
- namespace aria2 {
- namespace bencode {
- const BDE BDE::none;
- BDE::BDE():_type(TYPE_NONE) {}
- BDE::BDE(Integer integer):_type(TYPE_INTEGER),
- _integer(new Integer(integer)) {}
- BDE::BDE(const std::string& string):_type(TYPE_STRING),
- _string(new std::string(string)) {}
- BDE::BDE(const char* cstring):_type(TYPE_STRING),
- _string(new std::string(cstring)) {}
- BDE::BDE(const char* data, size_t length):
- _type(TYPE_STRING),
- _string(new std::string(&data[0], &data[length])) {}
- BDE::BDE(const unsigned char* data, size_t length):
- _type(TYPE_STRING),
- _string(new std::string(&data[0], &data[length])) {}
- BDE BDE::dict()
- {
- BDE bde;
- bde._type = TYPE_DICT;
- bde._dict.reset(new Dict());
- return bde;
- }
- BDE BDE::list()
- {
- BDE bde;
- bde._type = TYPE_LIST;
- bde._list.reset(new List());
- return bde;
- }
- // Test for Null data
- bool BDE::isNone() const
- {
- return _type == TYPE_NONE;
- }
- // Integer Interface
- bool BDE::isInteger() const
- {
- return _type == TYPE_INTEGER;
- }
- BDE::Integer BDE::i() const
- {
- if(isInteger()) {
- return *_integer.get();
- } else {
- throw RecoverableException("Not Integer");
- }
- }
- // String Interface
- bool BDE::isString() const
- {
- return _type == TYPE_STRING;
- }
- const std::string& BDE::s() const
- {
- if(isString()) {
- return *_string.get();
- } else {
- throw RecoverableException("Not String");
- }
- }
- const unsigned char* BDE::uc() const
- {
- if(isString()) {
- return reinterpret_cast<const unsigned char*>(_string->data());
- } else {
- throw RecoverableException("Not String");
- }
- }
- // Dictionary Interface
- bool BDE::isDict() const
- {
- return _type == TYPE_DICT;
- }
- BDE& BDE::operator[](const std::string& key)
- {
- if(isDict()) {
- return (*_dict.get())[key];
- } else {
- throw RecoverableException("Not Dict");
- }
- }
- const BDE& BDE::operator[](const std::string& key) const
- {
- 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
- {
- if(isDict()) {
- return _dict->find(key) != _dict->end();
- } else {
- throw RecoverableException("Not Dict");
- }
- }
- void BDE::removeKey(const std::string& key) const
- {
- if(isDict()) {
- _dict->erase(key);
- } else {
- throw RecoverableException("Not Dict");
- }
- }
- BDE::Dict::iterator BDE::dictBegin()
- {
- if(isDict()) {
- return _dict->begin();
- } else {
- throw RecoverableException("Not Dict");
- }
- }
- BDE::Dict::const_iterator BDE::dictBegin() const
- {
- if(isDict()) {
- return _dict->begin();
- } else {
- throw RecoverableException("Not Dict");
- }
- }
- BDE::Dict::iterator BDE::dictEnd()
- {
- if(isDict()) {
- return _dict->end();
- } else {
- throw RecoverableException("Not Dict");
- }
- }
- BDE::Dict::const_iterator BDE::dictEnd() const
- {
- if(isDict()) {
- return _dict->end();
- } else {
- throw RecoverableException("Not Dict");
- }
- }
- // List Interface
- bool BDE::isList() const
- {
- return _type == TYPE_LIST;
- }
- void BDE::append(const BDE& bde)
- {
- if(isList()) {
- _list->push_back(bde);
- } else {
- throw RecoverableException("Not List");
- }
- }
- void BDE::operator<<(const BDE& bde)
- {
- if(isList()) {
- _list->push_back(bde);
- } else {
- throw RecoverableException("Not List");
- }
- }
- BDE& BDE::operator[](size_t index)
- {
- if(isList()) {
- return (*_list.get())[index];
- } else {
- throw RecoverableException("Not List");
- }
- }
- const BDE& BDE::operator[](size_t index) const
- {
- if(isList()) {
- return (*_list.get())[index];
- } else {
- throw RecoverableException("Not List");
- }
- }
- BDE::List::iterator BDE::listBegin()
- {
- if(isList()) {
- return _list->begin();
- } else {
- throw RecoverableException("Not List");
- }
- }
- BDE::List::const_iterator BDE::listBegin() const
- {
- if(isList()) {
- return _list->begin();
- } else {
- throw RecoverableException("Not List");
- }
- }
- BDE::List::iterator BDE::listEnd()
- {
- if(isList()) {
- return _list->end();
- } else {
- throw RecoverableException("Not List");
- }
- }
- BDE::List::const_iterator BDE::listEnd() const
- {
- if(isList()) {
- return _list->end();
- } else {
- throw RecoverableException("Not List");
- }
- }
- // Callable from List and Dict
- size_t BDE::size() const
- {
- 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
- {
- if(isDict()) {
- return _dict->empty();
- } else if(isList()) {
- return _list->empty();
- } else {
- throw RecoverableException("Not Dict nor List");
- }
- }
- static BDE decodeiter(std::istream& ss);
- static void checkdelim(std::istream& ss, const char delim = ':')
- {
- char d;
- if(!(ss.get(d) && d == delim)) {
- throw RecoverableException
- (StringFormat("Delimiter '%c' not found.", delim).str());
- }
- }
- static std::string decoderawstring(std::istream& ss)
- {
- size_t length;
- ss >> length;
- if(!ss) {
- throw RecoverableException("A positive 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);
- std::string str(&buf[0], &buf[length]);
- delete [] buf;
- 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());
- }
- return str;
- }
- static BDE decodestring(std::istream& ss)
- {
- return BDE(decoderawstring(ss));
- }
- static BDE decodeinteger(std::istream& ss)
- {
- 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)
- {
- 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)
- {
- 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)
- {
- 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)
- {
- return decodeiter(in);
- }
- BDE decode(const std::string& s)
- {
- if(s.empty()) {
- return BDE::none;
- }
- std::istringstream ss(s);
- return decodeiter(ss);
- }
- BDE decode(const unsigned char* data, size_t length)
- {
- return decode(std::string(&data[0], &data[length]));
- }
- BDE decodeFromFile(const std::string& filename)
- {
- std::ifstream f(filename.c_str(), std::ios::binary);
- 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)
- {
- 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)
- {
- std::ostringstream ss;
- encodeIter(ss, bde);
- return ss.str();
- }
- } // namespace bencode
- } // namespace aria2
|