Web alhem.net

TrackerSock.cpp

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.
*/
//#include <stdio.h>
#include <map>
#include <Parse.h>
#include "MyHandler.h"
#include "TrackerSock.h"




TrackerSock::TrackerSock(ISocketHandler& h)
:HTTPSocket(h)
{
        SetLineProtocol();
}


TrackerSock::~TrackerSock()
{
//printf("~TrackerSock\n");
}


void TrackerSock::OnFirst()
{
        if (GetMethod() == "GET")
        {
                std::string url = GetUrl();
                std::string params;
                
                {
                        Parse pa(url,"?");
                        m_uri = pa.getword();
                        params = pa.getrest();
                }
                Parse pa(params,"&");
                std::string tmp = pa.getword();
                while (tmp.size())
                {
                        {
                                Parse pa(tmp,"=");
                                std::string key = pa.getword();
                                std::string val = pa.getrest();
                                m_par[key] = val;
                        }
                        tmp = pa.getword();
                }
                m_info_hash = static_cast<MyHandler&>(Handler()).escape(m_par["info_hash"]);
                std::string ip = m_par["ip"];
                ip = ip.size() ? ip : GetRemoteAddress();
                MyHandler::PEER *p = static_cast<MyHandler&>(Handler()).reg_peer(
                        m_info_hash,
                        m_par["peer_id"],
                        ip,
                        atoi(m_par["port"].c_str()));
                if (m_par["event"] == "completed")
                {
                        p -> complete();
                }
                else
                if (m_par["event"] == "stopped")
                {
                        p -> stop();
                }
                else
                {
                        p -> update(m_par["uploaded"],m_par["downloaded"],m_par["left"]);
                }
        }
}


void TrackerSock::OnHeaderComplete()
{
        AddResponseHeader("Connection", "close");
        AddResponseHeader("Server", static_cast<MyHandler&>(Handler()).GetString("server/identity"));

        if (m_uri == "/" || m_uri == "/index.html")
        {
                SetHttpVersion("HTTP/1.0");
                SetStatus("501");
                SetStatusText("Not Implemented");
                SendResponse();
                SetCloseAndDelete();
        }
        else
        if (m_uri == "/scrape")
        {
                SetHttpVersion("HTTP/1.0");
                SetStatus("501");
                SetStatusText("Not Implemented");
                SendResponse();
                SetCloseAndDelete();
        }
        else
        if (m_uri == "/file")
        {
                SetHttpVersion("HTTP/1.0");
                SetStatus("501");
                SetStatusText("Not Implemented");
                SendResponse();
                SetCloseAndDelete();
        }
        else
        if (m_uri == "/favicon.ico")
        {
                SetHttpVersion("HTTP/1.0");
                SetStatus("501");
                SetStatusText("Not Implemented");
                SendResponse();
                SetCloseAndDelete();
        }
        else
        if (m_uri != "/announce")
        {
                SetHttpVersion("HTTP/1.0");
                SetStatus("404");
                SetStatusText("Not Found");
                SendResponse();
                SetCloseAndDelete();
                return;
        }

        SetHttpVersion("HTTP/1.0");
        SetStatus("200");
        SetStatusText("OK");
        SendResponse();

        {
                time_t t = time(NULL);
                struct tm *tp = localtime(&t);
                printf("%d-%02d-%02d %02d:%02d:%02d :: %s: %s %s %s\n",tp -> tm_year + 1900,
                        tp -> tm_mon + 1,tp -> tm_mday,
                        tp -> tm_hour,tp -> tm_min,tp -> tm_sec,
                        GetRemoteAddress().c_str(),GetMethod().c_str(),GetUrl().c_str(),GetHttpVersion().c_str());
        }

        std::string interval = "i" + static_cast<MyHandler&>(Handler()).GetString("server/reannounce_interval") + "e";
        size_t numwant = atoi(m_par["numwant"].c_str());
        size_t response_size = static_cast<MyHandler&>(Handler()).GetInt("server/response_size");

        Send("d");
        Send("8:interval");
        Send(interval);
        Send("5:peers" "l");
        static_cast<MyHandler&>(Handler()).CreatePeers(this, m_info_hash, numwant ? numwant : response_size);
        // end peers list
        Send("e");
        // end dictionary
        Send("e");

        SetCloseAndDelete();
#ifdef ENABLE_DETACH
        Detach();
#endif
}

---