Web alhem.net

MyMinionSocket.cpp

Go to the documentation of this file.

/*
Copyright (C) 2004  Anders Hedström (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 "MyMinionSocket.h"
#include "FinderHandler.h"
#include <Utility.h>
#ifdef _WIN32
#include <config-win.h>
#include <mysql.h>
#include <libmysqlwrapped.h>
#define strcasecmp stricmp
#else
#include <mysql/mysql.h>
#include <libmysqlwrapped.h>
#endif
#include <libfd.h>
#include "Search.h"
#include "win32/MyApp.h"


#define D2(x) { \
  FILE *fil = fopen("c:\\deb.log","at"); \
  x; \
  fclose(fil); \
}



MyMinionSocket::MyMinionSocket(SocketHandler& h)
:MinionSocket(h)
{
}


MyMinionSocket::MyMinionSocket(SocketHandler& h,const std::string& s,ipaddr_t a,port_t p)
:MinionSocket(h,s,a,p)
{
}


MyMinionSocket::~MyMinionSocket()
{
}


bool MyMinionSocket::OnVerifiedLine(const std::string& cmd,Parse& pa)
{
  FinderHandler& h = static_cast<FinderHandler&>(Handler());
  Database& db = h.GetDatabase();

  D2(fprintf(fil,"MyMinionSocket verified command: %s\n",cmd.c_str());)

  //  'post' _ typ : 64(category) : 64(tag) : 64(tag) : 64(tag) : 64(name) : data(64) : sz : sha1
  if (cmd == "post")
  {
    Query q(db);
    char sql[1000];

    std::string typ = pa.getword();
    D2(fprintf(fil," post.typ == '%s'\n",typ.c_str());)
    if (typ != "File" && typ != "URL")
    {
      return true;
    }
    std::string strCategory = Utility::base64d(pa.getword());
/*
    std::string strTag1 = Utility::base64d(pa.getword());
    std::string strTag2 = Utility::base64d(pa.getword());
    std::string strTag3 = Utility::base64d(pa.getword());
*/
    std::string strName = Utility::base64d(pa.getword());
    std::string strData = pa.getword();
    size_t sz = pa.getvalue();
    std::string strSha1 = pa.getword();
    D2(fprintf(fil,
      "category: %s\n"
      "name: %s\n"
      "data: %s\n"
      "size: %d\n"
      "sha1: %s\n", strCategory.c_str(),
      strName.c_str(), strData.c_str(), sz, strSha1.c_str());)

    sprintf(sql,"select * from category where name='%s'", q.safestr(strCategory).c_str());
    db::Category cat(&db, sql);
    cat.name = strCategory;
    cat.save();

    db::Item item(db, sz, strSha1);
    item.typ = typ;
    item.category = cat.num;
    item.name = strName;
    item.data.push_back( strData );
    item.sz = sz;
    item.sha1 = strSha1;
    item.save();

    g_app->ShowBalloonTip("Minion", cat.name + ": " + item.name);

    std::string strTag = pa.getword();
    while (strTag.size())
    {
      std::string tmp = Utility::base64d(strTag);
      sprintf(sql,"select * from tag where name='%s'", q.safestr(tmp).c_str());
      db::Tag tag(&db, sql);
      tag.name = tmp;
      tag.save();
      db::Linkitemtag x(db, item.num, tag.num);
      x.item = item.num;
      x.tag = tag.num;
      x.save();
      //
      strTag = pa.getword();
    }

  }
  else
        {
                return MinionSocket::OnVerifiedLine(cmd, pa);
        }
        return true;
}


void MyMinionSocket::OnConnect()
{
  MinionSocket::OnConnect();
  std::string remote = this->GetRemoteHostname();
  std::string remote_ip = this->GetRemoteAddress();
  g_app->ShowBalloonTip("Minion", "Connected to " + remote + " (" + remote_ip + ")");
}


void MyMinionSocket::OnAccept()
{
  MinionSocket::OnAccept();
  std::string remote = this->GetRemoteHostname();
  std::string remote_ip = this->GetRemoteAddress();
  g_app->ShowBalloonTip("Minion", "Incoming connection from " + remote + " (" + remote_ip + ")");
}

---