Blame src/Bencode.h

Packit 8f70b4
/*
Packit 8f70b4
 * lftp - file transfer program
Packit 8f70b4
 *
Packit 8f70b4
 * Copyright (c) 1996-2012 by Alexander V. Lukyanov (lav@yars.free.net)
Packit 8f70b4
 *
Packit 8f70b4
 * This program is free software; you can redistribute it and/or modify
Packit 8f70b4
 * it under the terms of the GNU General Public License as published by
Packit 8f70b4
 * the Free Software Foundation; either version 3 of the License, or
Packit 8f70b4
 * (at your option) any later version.
Packit 8f70b4
 *
Packit 8f70b4
 * This program is distributed in the hope that it will be useful,
Packit 8f70b4
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
Packit 8f70b4
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Packit 8f70b4
 * GNU General Public License for more details.
Packit 8f70b4
 *
Packit 8f70b4
 * You should have received a copy of the GNU General Public License
Packit 8f70b4
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
Packit 8f70b4
 */
Packit 8f70b4
Packit 8f70b4
#ifndef BENCODE_H
Packit 8f70b4
#define BENCODE_H
Packit 8f70b4
Packit 8f70b4
#include "xmap.h"
Packit 8f70b4
#include "buffer.h"
Packit 8f70b4
Packit 8f70b4
class BeNode
Packit 8f70b4
{
Packit 8f70b4
   void PackDict(xstring &buf;;
Packit 8f70b4
public:
Packit 8f70b4
   enum be_type_t {
Packit 8f70b4
      BE_STR,
Packit 8f70b4
      BE_INT,
Packit 8f70b4
      BE_LIST,
Packit 8f70b4
      BE_DICT
Packit 8f70b4
   } type;
Packit 8f70b4
Packit 8f70b4
   xstring str;
Packit 8f70b4
   xstring str_lc;
Packit 8f70b4
   xarray_p<BeNode> list;
Packit 8f70b4
   xmap_p<BeNode> dict;
Packit 8f70b4
   long long num;
Packit 8f70b4
Packit 8f70b4
   static BeNode *Parse(const char *s,int len,int *rest);
Packit 8f70b4
Packit 8f70b4
   BeNode(const xstring& s);
Packit 8f70b4
   BeNode(const char *s,int l);
Packit 8f70b4
   BeNode(const char *s);
Packit 8f70b4
   BeNode(xarray_p<BeNode> *l);
Packit 8f70b4
   BeNode(xmap_p<BeNode> *d);
Packit 8f70b4
   BeNode(long long);
Packit 8f70b4
   ~BeNode();
Packit 8f70b4
Packit 8f70b4
   BeNode *lookup(const char *key) {
Packit 8f70b4
      return dict.lookup(key);
Packit 8f70b4
   }
Packit 8f70b4
   BeNode *lookup(const char *key,be_type_t t) {
Packit 8f70b4
      BeNode *n=dict.lookup(key);
Packit 8f70b4
      if(n && n->type!=t)
Packit 8f70b4
	 n=0;
Packit 8f70b4
      return n;
Packit 8f70b4
   }
Packit 8f70b4
   const xstring& lookup_str(const char *key) const {
Packit 8f70b4
      BeNode *n=dict.lookup(key);
Packit 8f70b4
      if(!n || n->type!=BE_STR)
Packit 8f70b4
	 return xstring::null;
Packit 8f70b4
      return n->str;
Packit 8f70b4
   }
Packit 8f70b4
   long long lookup_int(const char *key) const {
Packit 8f70b4
      BeNode *n=dict.lookup(key);
Packit 8f70b4
      if(!n || n->type!=BE_INT)
Packit 8f70b4
	 return 0;
Packit 8f70b4
      return n->num;
Packit 8f70b4
   }
Packit 8f70b4
Packit 8f70b4
   int ComputeLength();
Packit 8f70b4
   const xstring& Pack();
Packit 8f70b4
   void Pack(xstring &buf;;
Packit 8f70b4
   void Pack(const SMTaskRef<IOBuffer> &buf;;
Packit 8f70b4
Packit 8f70b4
   void Format(xstring &buf,int level);
Packit 8f70b4
   const char *Format();
Packit 8f70b4
   void Format1(xstring &buf;;
Packit 8f70b4
   const char *Format1();
Packit 8f70b4
Packit 8f70b4
   static const char *TypeName(be_type_t t);
Packit 8f70b4
};
Packit 8f70b4
Packit 8f70b4
#endif//BENCODE_H