2 // Copyright (c) 1997, 1998 Etienne BERNARD
3 // Copyright (C) 2002 Clinton Ebadi
5 // This program is free software; you can redistribute it and/or modify
6 // it under the terms of the GNU General Public License as published by
7 // the Free Software Foundation; either version 2 of the License, or
10 // This program is distributed in the hope that it will be useful,
11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 // GNU General Public License for more details.
15 // You should have received a copy of the GNU General Public License
16 // along with this program; if not, write to the Free Software
17 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
26 #include <sys/types.h>
30 #include "DCCConnection.H"
31 #include "StringTokenizer.H"
32 #include "ServerConnection.H"
34 #include "UserCommands.H"
36 #define DEFAULT_NICKNAME "Bobot"
37 #define DEFAULT_USERNAME "bobot"
38 #define DEFAULT_IRCNAME "I'm a bobot++!"
39 #define DEFAULT_COMMANDCHAR '!'
40 #define DEFAULT_USERLISTFILENAME "bot.users"
41 #define DEFAULT_SHITLISTFILENAME "bot.shit"
42 #define DEFAULT_HELPFILENAME "bot.help"
43 #define DEFAULT_SCRIPTLOGFILENAME "script.log"
44 #define DEFAULT_LOGFILENAME "bot.log"
45 #define DEFAULT_LOGDIR getenv ("HOME") + String("/.bobotpp/logs/")
46 #define DEFAULT_INITFILENAME "bot.init"
48 #define DEFAULT_AUTOEXECFILENAME "bot.autoexec"
51 Bot::Bot(String filename, bool debug_on)
52 : nickName(DEFAULT_NICKNAME),
53 wantedNickName(DEFAULT_NICKNAME),
54 userName(DEFAULT_USERNAME),
55 ircName(DEFAULT_IRCNAME),
56 versionString(VERSION_STRING),
59 commandChar(DEFAULT_COMMANDCHAR),
60 configFileName(filename),
61 userListFileName(DEFAULT_USERLISTFILENAME),
62 shitListFileName(DEFAULT_SHITLISTFILENAME),
63 logFileName(DEFAULT_LOGFILENAME),
64 logs_dir (DEFAULT_LOGDIR),
65 helpFileName(DEFAULT_HELPFILENAME),
66 initFileName(DEFAULT_INITFILENAME),
68 scriptLogFileName(DEFAULT_SCRIPTLOGFILENAME),
69 autoexecFileName(DEFAULT_AUTOEXECFILENAME),
72 debug(debug_on), stop(false), sentPing(false),
73 startTime(time(NULL)), currentTime(startTime),
74 lastNickNameChange(startTime), lastChannelJoin(startTime),
75 serverConnection(0), sentUserhostID(0), receivedUserhostID(0)
78 wantedChannels.clear();
79 ignoredUserhosts.clear();
84 init_user_functions ();
86 set_log_dir (logs_dir);
87 set_log_file (logFileName);
90 channelList = new ChannelList();
91 serverList = new ServerList();
93 userList = new UserList(userListFileName);
94 shitList = new ShitList(shitListFileName);
95 todoList = new TodoList();
97 // Let's read the alias file
98 std::ifstream initFile(initFileName);
101 String temp, alias, command;
103 while (initFile >> temp, temp.length() != 0) {
105 StringTokenizer st(temp);
107 if (temp[0]=='#') continue;
108 if (st.countTokens(' ') != 2) {
109 std::cerr << "Error when reading alias file (" << initFileName
110 << ") line " << line << "...\n";
113 alias = st.nextToken().toUpper();
114 command = st.nextToken().toUpper();
116 // Does the function already exist ?
117 if (!userFunctions[alias])
119 if (userFunction *u = userFunctions[command])
120 userFunctions[alias] =
122 userFunction(u->function,
124 u->needsChannelName);
130 std::srand (std::time (0)); // srand for bot-random
132 botInterp = new BotInterp(this, logs_dir + scriptLogFileName);
133 botInterp->LoadScript(autoexecFileName);
139 // TODO: is it ok to delete iterators!?!
142 while (spyList.size() != 0) {
143 p = (*spyList.begin()).second;
144 spyList.erase(spyList.begin());
148 while (dccConnections.size() != 0) {
149 d = *dccConnections.begin();
150 dccConnections.erase(dccConnections.begin());
154 destroy_user_functions ();
157 while (wantedChannels.size() != 0) {
158 w = (*wantedChannels.begin()).second;
159 wantedChannels.erase(wantedChannels.begin());
169 delete serverConnection;
170 logLine("Stopping log.");
175 Bot::logLine(String line)
178 std::time_t current_time = time(0);
180 d = localtime(¤t_time);
181 logFile << "[" << std::setfill('0') << std::setw(2)
182 << d->tm_mday << "/" << std::setfill('0') << std::setw(2)
183 << d->tm_mon + 1 << "/"
184 << d->tm_year + 1900 << " - " << std::setfill('0') << std::setw(2)
185 << d->tm_hour << ":" << std::setfill('0') << std::setw(2)
186 << d->tm_min << ":" << std::setfill('0') << std::setw(2)
195 std::ifstream file(configFileName);
200 logLine(String("I cannot find the file ") + configFileName);
204 while (!file.eof()) {
208 if (temp.length() == 0 || temp[0] == '#') {
213 StringTokenizer st(temp);
214 String command = st.nextToken('=').trim().toUpper();
215 String parameters = st.nextToken('=').trim();
217 if (command == "NICK" || command == "NICKNAME")
218 nickName = wantedNickName = parameters;
219 else if (command == "USERNAME")
220 userName = parameters;
221 else if (command == "IRCNAME" || command == "REALNAME")
222 ircName = parameters;
223 else if (command == "CMDCHAR" || command == "COMMAND")
224 commandChar = parameters[0];
225 else if (command == "USERLIST")
226 userListFileName = parameters;
227 else if (command == "SHITLIST")
228 shitListFileName = parameters;
229 else if (command == "CHANNEL") {
230 if (parameters.indexOf(':') == -1) {
231 std::cout << "Warning. The 'channel' syntax has changed."
232 << " Please see the README file for more information."
233 << " I will use compatibility mode, but you're really"
234 << " missing something.\n";
235 StringTokenizer st2(parameters);
236 String name = st2.nextToken().toLower();
237 String key = st2.nextToken();
238 wantedChannels[name] = new wantedChannel("", "", key);
240 StringTokenizer st2(parameters);
241 String name = st2.nextToken(':').toLower();
242 String mode = st2.nextToken(':');
243 String keep = st2.nextToken(':');
244 String key = st2.nextToken(':');
245 wantedChannels[name] = new wantedChannel(mode, keep, key);
248 else if (command == "LOGFILE")
250 if (parameters != logFileName)
252 if (parameters[0] == '/')
255 set_log_file (parameters.subString (1));
258 set_log_file (parameters);
262 else if (command == "SCRIPTLOGFILE")
263 scriptLogFileName = parameters;
264 else if (command == "AUTOEXECFILE")
265 autoexecFileName = parameters;
267 else if (command == "INITFILE")
268 initFileName = parameters;
269 else if (command == "LOCALIP")
270 localIP = parameters;
271 else if (command == "SERVER") {
272 if (parameters.indexOf(' ') == -1)
273 serverList->addServer(new Server(parameters));
275 StringTokenizer st2(parameters);
276 String name = st2.nextToken();
277 int port = std::atoi(st2.nextToken());
278 serverList->addServer(new Server(name,
284 logLine(String("Syntax error in file ") + configFileName +
285 ", line " + String((long)line));
302 waitForInput(); // This is the main event loop
303 if (!serverConnection->queue->flush())
316 struct timeval timer;
318 int sock = serverConnection->getFileDescriptor();
319 int maxSocketNumber = sock;
328 for (std::list<DCCConnection *>::iterator it = dccConnections.begin();
329 it != dccConnections.end(); ++it) {
330 int s = (*it)->getFileDescriptor();
336 if (s > maxSocketNumber)
343 switch (select(maxSocketNumber + 1, &rd, NULL, NULL, &timer)) {
344 case 0: /* timeout */
348 default: /* normal */
352 if (FD_ISSET(sock, &rd))
354 if (serverConnection->handleInput())
357 std::list<DCCConnection *>::iterator it = dccConnections.begin();
358 std::list<DCCConnection *>::iterator it2;
360 while (it != dccConnections.end()) {
364 if (rd & (*it2)->getFileDescriptor()) {
366 if (FD_ISSET((*it2)->getFileDescriptor(), &rd)) {
368 if ((*it2)->handleInput()) {
370 dccConnections.erase(it2);
376 if (currentTime < std::time(NULL)) { // Actions that we do each second
377 currentTime = std::time(NULL);
378 for (std::map<String, unsigned int, std::less<String> >::iterator
379 it = ignoredUserhosts.begin();
380 it != ignoredUserhosts.end(); ++it)
381 if ((*it).second > 0)
385 while ((line = todoList->getNext()) != "") {
386 serverConnection->queue->sendChannelMode(line);
389 botInterp->RunTimers(currentTime);
393 tm *thisTime = localtime(¤tTime);
394 if (thisTime->tm_sec == 0) {
396 sprintf(s, "%2d:%2d", thisTime->tm_hour, thisTime->tm_min);
397 botInterp->RunHooks(Hook::TIMER, String(s),
398 gh_list(Utils::string2SCM(String(s)), SCM_UNDEFINED));
404 if (currentTime >= (time_t)(lastNickNameChange + Bot::NICK_CHANGE) &&
405 nickName != wantedNickName) {
406 lastNickNameChange = currentTime;
407 serverConnection->queue->sendNick(wantedNickName);
410 if (currentTime >= (std::time_t)(lastChannelJoin + Bot::CHANNEL_JOIN)) {
411 lastChannelJoin = currentTime;
412 for (std::map<String, wantedChannel *, std::less<String> >::iterator it =
413 wantedChannels.begin(); it != wantedChannels.end();
415 if (channelList->getChannel((*it).first) == 0)
416 serverConnection->queue->sendJoin((*it).first, (*it).second->key);
419 std::list<DCCConnection *>::iterator it2;
421 for (std::list<DCCConnection *>::iterator it = dccConnections.begin();
422 it != dccConnections.end(); ) {
425 if ((*it2)->autoRemove && currentTime >= (std::time_t)((*it2)->lastSpoken + Bot::DCC_DELAY)) {
427 dccConnections.erase(it2);
431 if (currentTime >= (std::time_t)(serverConnection->serverLastSpoken + Bot::PING_TIME) && !sentPing) {
432 serverConnection->queue->sendPing("Testing connection");
436 if (currentTime >= (std::time_t)(serverConnection->serverLastSpoken + Bot::TIMEOUT)) {
442 // We can change server if we will not lose op on a channel
444 Bot::canChangeServer()
449 for (std::map<String, Channel *, std::less<String> >::iterator it =
450 channelList->begin();
451 it != channelList->end(); ++it) {
452 channel = (*it).first;
453 c = channelList->getChannel(channel);
454 if (c->countOp == 1 &&
455 c->count > 1 && this->iAmOp(channel))
467 channelList->clear();
469 if (serverConnection)
470 userList->removeFirst();
472 delete serverConnection;
475 Server * s = serverList->nextServer();
477 std::cout << "No server found. Exiting..." << std::endl;
480 serverConnection = new ServerConnection(this, s, localIP);
481 if (!serverConnection->connect()) {
483 // We sleep 10 seconds, to avoid connection flood
485 delete serverConnection;
496 channelList->clear();
498 userList->removeFirst();
500 delete serverConnection;
503 new ServerConnection(this, serverList->currentServer(), localIP);
505 serverConnection->connect();
509 Bot::connect(int serverNumber)
512 channelList->clear();
514 userList->removeFirst();
516 delete serverConnection;
519 new ServerConnection(this, serverList->get(serverNumber), localIP);
521 serverConnection->connect();
525 Bot::addDCC(Person * from, unsigned long address, int port)
527 DCCConnection * d = new DCCConnection(this, from->getAddress(),
532 logLine ("DCC Connection failed from " + from->getAddress ());
535 logLine ("DCC Connection worked!");
536 dccConnections.push_back(d);
542 for (std::map<String, Channel *, std::less<String> >::iterator it = channelList->begin();
543 it != channelList->end(); ++it)
544 serverConnection->queue->sendWho((*it).first);
548 Bot::getUserhost(String channel, String nick)
555 c = channelList->getChannel(channel);
557 nick = nick.toLower();
560 if (c && c->hasNick(nick))
561 return c->getUser(nick)->userhost;
563 unsigned long num = sentUserhostID++;
565 serverConnection->queue->sendUserhost(nick);
566 userhostMap[num] = "+";
568 while (userhostMap[num] == "+") {
570 serverConnection->queue->flush();
573 // We have got our answer
574 String res = userhostMap[num];
575 userhostMap.erase(num);
581 Bot::iAmOp(String channel)
583 User * me = channelList->getChannel(channel)->getUser(nickName);
584 return (me->mode & User::OP_MODE);
588 Bot::init_user_functions ()
591 #define uf(f, l, b) new userFunction (f, l, b);
592 userFunctions["ACTION"] = uf (UserCommands::Action, User::USER, true);
593 userFunctions["ADDUSER"] = uf (UserCommands::AddUser, User::FRIEND, false);
594 userFunctions["ADDSERVER"] = uf (UserCommands::AddServer, User::FRIEND,
596 userFunctions["ADDSHIT"] = uf (UserCommands::AddShit, User::FRIEND, false);
597 userFunctions["ALIAS"] = uf (UserCommands::Alias, User::MASTER, false);
598 userFunctions["BAN"] = uf (UserCommands::Ban, User::USER, true);
599 userFunctions["BANLIST"] = uf (UserCommands::BanList, User::USER, true);
600 userFunctions["CHANNELS"] =
601 uf (UserCommands::Channels, User::FRIEND, false);
602 userFunctions["CYCLE"] = uf (UserCommands::Cycle, User::FRIEND, true);
603 userFunctions["DCCLIST"] = uf (UserCommands::DCCList, User::FRIEND, false);
604 userFunctions["DEBAN"] = uf (UserCommands::Deban, User::USER, true);
605 userFunctions["DELSERVER"] = uf (UserCommands::DelServer, User::FRIEND,
607 userFunctions["DELUSER"] = uf (UserCommands::DelUser, User::FRIEND, false);
608 userFunctions["DELSHIT"] = uf (UserCommands::DelShit, User::FRIEND, false);
609 userFunctions["DEOP"] = uf (UserCommands::Deop, User::TRUSTED_USER, true);
610 userFunctions["DIE"] = uf (UserCommands::Die, User::MASTER, false);
611 userFunctions["DO"] = uf (UserCommands::Do, User::MASTER, false);
613 userFunctions["EXECUTE"] = uf (UserCommands::Execute, User::MASTER, false);
615 userFunctions["HELP"] = uf (UserCommands::Help, User::NONE, false);
616 userFunctions["IDENT"] = uf (UserCommands::Ident, User::NONE, true);
617 userFunctions["INVITE"] = uf (UserCommands::Invite, User::USER, true);
618 userFunctions["JOIN"] = uf (UserCommands::Join, User::FRIEND, false);
619 userFunctions["KEEP"] = uf (UserCommands::Keep, User::FRIEND, true);
620 userFunctions["KICK"] = uf (UserCommands::Kick, User::USER, true);
621 userFunctions["KICKBAN"] = uf (UserCommands::KickBan, User::USER, true);
622 userFunctions["LOAD"] = uf (UserCommands::Load, User::FRIEND, false);
624 userFunctions["LOADSCRIPT"] = uf (UserCommands::LoadScript, User::MASTER,
627 userFunctions["LOCK"] = uf (UserCommands::Lock, User::FRIEND, true);
628 userFunctions["MODE"] = uf (UserCommands::Mode, User::FRIEND, true);
629 userFunctions["MSG"] = uf (UserCommands::Msg, User::USER, false);
630 userFunctions["NAMES"] = uf (UserCommands::Names, User::USER, true);
631 userFunctions["NEXTSERVER"] = uf (UserCommands::NextServer, User::FRIEND,
633 userFunctions["NICK"] = uf (UserCommands::Nick, User::FRIEND, false);
634 userFunctions["NSLOOKUP"] = uf (UserCommands::NsLookup, User::USER, false);
635 userFunctions["OP"] = uf (UserCommands::Op, User::TRUSTED_USER, true);
636 userFunctions["PART"] = uf (UserCommands::Part, User::FRIEND, true);
637 userFunctions["PASSWORD"] = uf (UserCommands::Password, User::USER, true);
638 userFunctions["RECONNECT"] =
639 uf (UserCommands::Reconnect, User::FRIEND, false);
640 userFunctions["RSPYMESSAGE"] =
641 uf (UserCommands::RSpyMessage, User::USER, false);
642 userFunctions["SAVE"] = uf (UserCommands::Save, User::FRIEND, false);
643 userFunctions["SAY"] = uf (UserCommands::Say, User::USER, true);
644 userFunctions["SERVER"] = uf (UserCommands::Server, User::FRIEND, false);
645 userFunctions["SERVERLIST"] =
646 uf (UserCommands::ServerList, User::FRIEND, false);
647 userFunctions["SETVERSION"] =
648 uf (UserCommands::SetVersion, User::MASTER, false);
649 userFunctions["SHITLIST"] =
650 uf (UserCommands::ShitList, User::FRIEND, false);
651 userFunctions["SPYLIST"] = uf (UserCommands::SpyList, User::USER, false);
652 userFunctions["SPYMESSAGE"] =
653 uf (UserCommands::SpyMessage, User::USER, false);
654 userFunctions["STATS"] = uf (UserCommands::Stats, User::FRIEND, true);
655 userFunctions["TBAN"] = uf (UserCommands::TBan, User::USER, true);
656 userFunctions["TKBAN"] = uf (UserCommands::TKBan, User::USER, true);
657 userFunctions["TOPIC"] = uf (UserCommands::Topic, User::USER, true);
658 userFunctions["UNLOCK"] = uf (UserCommands::Unlock, User::FRIEND, true);
659 userFunctions["USERLIST"] =
660 uf (UserCommands::UserList, User::FRIEND, false);
661 userFunctions["WHO"] = uf (UserCommands::Who, User::NONE, true);
662 userFunctions["WHOIS"] = uf (UserCommands::Whois, User::FRIEND, true);
668 void erase_userf (std::pair<std::string, class userFunction*> it)
675 Bot::destroy_user_functions ()
677 std::for_each (userFunctions.begin (),
678 userFunctions.end (),
680 userFunctions.erase (userFunctions.begin (),
681 userFunctions.end ());
685 Bot::set_log_file (String name)
691 logFile.open(logs_dir + logFileName, std::ios_base::out |
692 std::ios_base::ate | std::ios_base::app);
694 logFile.open(logs_dir + logFileName, ios::out | ios::ate
698 logLine("Starting log.");
702 Bot::set_log_dir (String dir)