Web alhem.net

MyTcpSocket.cpp

Go to the documentation of this file.

/*
Copyright (C) 2004  Anders Hedstrom

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.
*/
#ifdef _WIN32
#pragma warning(disable:4786)
#endif
#include <string>
#include <vector>
#include <map>
#include <Parse.h>
#include <time.h>

#include "config.h"
#include "minder.h"
#include "MyHandler.h"
#include "TestSocket.h"
#include "MyTcpSocket.h"

using std::string;

#ifdef _DEBUG
#define DEB(x) x
#else
#define DEB(x)  
#endif




MyTcpSocket::MyTcpSocket(ISocketHandler& h)
:TcpSocket(h)
{
        SetLineProtocol(); // we want the OnLine callback
}


MyTcpSocket::~MyTcpSocket()
{
}


void MyTcpSocket::OnLine(const std::string& line)
{
        std::string cmd;
        std::string app;
        std::string id;
        std::string host;
        Parse pa(Utility::base64d(line),"_:");
        pa.getword(cmd); // 'Hello' / 'Goodbye'
        app = Utility::base64d(pa.getword());
        pa.getword(id);
        pa.getword(host); // host
        port_t port = pa.getvalue();
        unsigned long hid_from_host = pa.getvalue();
        long extra_info = pa.getvalue();

        if (cmd != "Hello" && cmd != "Goodbye")
        {
                SetCloseAndDelete(true);
                return;
        }

        // get ip sent to us by the minion
        ipaddr_t ip;
        Utility::u2ip(host,ip);

        // we ignore whatever ip the minion sends us
        // we only use the ip from the connection endpoint
        ipaddr_t ip2 = GetRemoteIP4();
DEB(    printf("Incoming: id '%s' %s:%d cmd '%s'\n",id.c_str(),GetRemoteAddress().c_str(),port,cmd.c_str());)
        if (cmd == "Goodbye")
        {
                static_cast<MyHandler&>(Handler()).unreg_host(app, id, ip2, port, ip);
                SetCloseAndDelete(true);
                return;
        }
        unsigned long host_id = 0;
        HOST *h = static_cast<MyHandler&>(Handler()).reg_host(app, id, ip2, port, ip, host_id, hid_from_host);
        h -> extra_information = extra_info;
        {
                std::string str;
                std::string msg;
                std::string ipstr;

                str = "You_" + id + ":" + GetRemoteAddress() + ":" + Utility::l2string(port) + ":" + Utility::l2string(host_id);
                msg = Utility::base64(str) + "\n";

                if (config[app + "/dbname"].size())
                {
                        str = "Database_" + config[app + "/dbname"];
                        msg += Utility::base64(str) + "\n";
                }

                if (config[app + "/update_host"].size() &&
                    config[app + "/update_port"].size() &&
                    config[app + "/update_url"].size() )
                {
                        str = "Update_" + config[app + "/update_host"];
                        str += ":" + config[app + "/update_port"];
                        str += ":" + config[app + "/update_url"];
                        msg += Utility::base64(str) + "\n";
                }

                static_cast<MyHandler&>(Handler()).reset_sent(app);
                for (int i = 0; i < 20; i++)
                {
                        HOST *p = static_cast<MyHandler&>(Handler()).GetRandomHOST(app);
                        if (!p -> sent)
                        {
                                Utility::l2ip(p -> ip,ipstr);
                                str = "Minion_" + p -> id + ":" + ipstr + ":" + Utility::l2string(p -> port);
                                str += ":" + Utility::l2string(p -> host_id);
                                msg += Utility::base64(str) + "\n";
                                p -> sent = true;
                        }
                }
                msg += Utility::base64("End") + "\n";
                Send(msg);
        }

        if (app == "viawww" && !h -> connectable) // try again
        {
                TestSocket *t = new TestSocket(Handler(),h -> connectable);
                t -> Open(ip2, h -> extra_information);
                t -> SetDeleteByHandler();
                Handler().Add(t);
        }
}


void MyTcpSocket::InitSSLServer()
{
        InitializeContext("comb.pem","");
}

---