Web alhem.net

MyHandler.h

Go to the documentation of this file.

/*
Copyright (C) 2004  grymse@alhem.net

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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
*/
#ifndef _MYHANDLER_H
#define _MYHANDLER_H

#include <map>
#include <string>
#include "MinderHandler.h"
#include <libxml/tree.h>
#include <vector>
#include <stdint.h>
#include <Utility.h>
#include "Configuration.h"
#include "key.h"
#include "MyMinionSocket.h"
#include "TrackerSock.h"


class MyHandler : public MinderHandler
{
public:
        struct PEER {
                PEER(const std::string& id,const std::string& address,port_t p)
                :escaped_peer_id(id),ip(address),port(p)
                ,uploaded(0),downloaded(0),left(0)
                ,updated(true),completed(false),stopped(false)
                ,last_contact(time(NULL))
                {
                        for (size_t i = 0; i < id.size(); i += 3)
                        {
                                int x;
                                sscanf( (char *)id.substr(i + 1,2).c_str(),"%02X",&x);
                                peer_id[i / 3] = x;
                        }
                }
                void update(const std::string& u,const std::string& d,const std::string& l) {
                        uploaded = Utility::atoi64(u);
                        downloaded = Utility::atoi64(d);
                        left = Utility::atoi64(l);
                        updated = true;
                }
                void stop() {
                        if (!stopped)
                                updated = true;
                        stopped = true;
                }
                void complete() {
                        if (!completed)
                                updated = true;
                        completed = true;
                }
                std::string escaped_peer_id;
                unsigned char peer_id[20];
                std::string ip;
                port_t port;
                uint64_t uploaded;
                uint64_t downloaded;
                uint64_t left;
                bool updated;
                bool completed;
                bool stopped;
                time_t last_contact;
        };
        typedef std::vector<PEER *> peer_v;
        typedef std::map<std::string,peer_v> peer_m;
public:
        MyHandler(const std::string& );
        ~MyHandler();

        std::string GetVersion() { 
//              std::string str = _VERSION_; return str; 
                return GetString("server/identity");
        }
        unsigned char *GetKey_m2minion() { return k1; }

        int GetInt(const std::string& );
        std::string GetString(const std::string& );
        bool GetBoolean(const std::string& );

        std::string escape(const std::string& );

        PEER *reg_peer(const std::string& hash,
                const std::string& peer_id,
                const std::string& ip,
                port_t port);

        void CreatePeers(TcpSocket *,const std::string& hash,size_t);
        void SendList();
        void CheckDropped();
        void SendInit(MyMinionSocket *);
        size_t MapSize();
        void HtmlReport(TrackerSock *);

private:
        Configuration m_config;
        peer_m m_peers;
};


#endif // _MYHANDLER_H

---