X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fcull_list.cpp;h=ddbf6acae8e7d12980c3efeb8a5f69e2418d92d2;hb=4501f0ec9ccff833e20a99ff90bd9b13a7d80fe2;hp=b06b80f7ffa1241fa184bc2fe9bbc1845aa64bd5;hpb=b477cc1c3439faa8f50a46a0c5f3dd9082007bec;p=user%2Fhenk%2Fcode%2Finspircd.git diff --git a/src/cull_list.cpp b/src/cull_list.cpp index b06b80f7f..ddbf6acae 100644 --- a/src/cull_list.cpp +++ b/src/cull_list.cpp @@ -2,128 +2,145 @@ * | Inspire Internet Relay Chat Daemon | * +------------------------------------+ * - * InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev. - * E-mail: - * - * + * InspIRCd: (C) 2002-2008 InspIRCd Development Team + * See: http://www.inspircd.org/wiki/index.php/Credits * - * Written by Craig Edwards, Craig McLure, and others. * This program is free but copyrighted software; see * the file COPYING for details. * * --------------------------------------------------- */ -using namespace std; +/* $Core: libIRCDcull_list */ -#include "inspircd_config.h" #include "inspircd.h" -#include "inspircd_io.h" -#include -#include -#include -#include -#include -#ifdef GCC3 -#include -#else -#include -#endif -#include -#include -#include -#include -#include "users.h" -#include "ctables.h" -#include "globals.h" -#include "modules.h" -#include "dynamic.h" -#include "wildcard.h" -#include "message.h" -#include "commands.h" -#include "xline.h" -#include "inspstring.h" -#include "inspircd.h" -#include "helperfuncs.h" -#include "hashcomp.h" -#include "typedefs.h" #include "cull_list.h" -extern InspIRCd* ServerInstance; - -bool CullList::IsValid(userrec* user) +CullList::CullList(InspIRCd* Instance) : ServerInstance(Instance) { - for (user_hash::iterator u = clientlist.begin(); u != clientlist.end(); u++) - { - userrec* u2 = (userrec*)*u; - if (user == u2) - return true; - } - return false; + list.clear(); } -CullItem::CullItem(userrec* u, std::string r) +void CullList::AddItem(User* user) { - this->user = u; - this->reason = r; -} + if (user->quitting) + { + ServerInstance->Logs->Log("CULLLIST",DEBUG, "*** Warning *** - You tried to quit a user (%s) twice. Did your module call QuitUser twice?", user->nick.c_str()); + return; + } -userrec* CullItem::GetUser() -{ - return this->user; + user->quitting = true; + list.push_back(user); } -std::string CullItem::GetReason() +void CullList::MakeSilent(User* user) { - return this->reason; + user->quietquit = true; + return; } -CullList::CullList() +int CullList::Apply() { - list.clear(); - exempt.clear(); -} + int n = list.size(); + int i = 0; -void CullList::AddItem(userrec* user, std::string reason) -{ - if (exempt.find(user) == exempt.end()) + while (list.size() && i++ != 100) { - CullItem item(user,reason); - list.push_back(item); - names.push_back(user->nick); - exempt[user] = 1; - } -} + std::vector::iterator a = list.begin(); -int CullList::Apply() -{ - int n = 0; - while (list.size()) - { - std::vector::iterator a = list.begin(); - userrec* u = a->GetUser(); - /* Because ServerInstance->DoOneIteration can - * take the user away from us in the middle of - * our operation, we should check to see if this - * pointer is still valid by iterating the hash. - * It's expensive, yes, but the DoOneIteration - * call stops it being horrendously bad. + User *u = (*a); + user_hash::iterator iter = ServerInstance->Users->clientlist->find(u->nick); + const std::string& preset_reason = u->GetOperQuit(); + std::string reason; + std::string oper_reason; + + reason.assign(u->quitmsg, 0, MAXQUIT - 1); + oper_reason.assign(preset_reason.empty() ? preset_reason : u->operquitmsg, 0, MAXQUIT - 1); + + if (u->registered != REG_ALL) + if (ServerInstance->Users->unregistered_count) + ServerInstance->Users->unregistered_count--; + + if (IS_LOCAL(u)) + { + if ((!u->sendq.empty()) && (!(*u->GetWriteError()))) + u->FlushWriteBuf(); + } + + if (u->registered == REG_ALL) + { + FOREACH_MOD_I(ServerInstance,I_OnUserQuit,OnUserQuit(u, reason, oper_reason)); + u->PurgeEmptyChannels(); + u->WriteCommonQuit(reason, oper_reason); + } + + FOREACH_MOD_I(ServerInstance,I_OnUserDisconnect,OnUserDisconnect(u)); + + if (IS_LOCAL(u)) + { + if (u->io) + { + try + { + u->io->OnRawSocketClose(u->GetFd()); + } + catch (CoreException& modexcept) + { + ServerInstance->Logs->Log("CULLLIST",DEBUG, "%s threw an exception: %s", modexcept.GetSource(), modexcept.GetReason()); + } + } + + ServerInstance->SE->DelFd(u); + u->CloseSocket(); + } + + /* + * this must come before the ServerInstance->SNO->WriteToSnoMaskso that it doesnt try to fill their buffer with anything + * if they were an oper with +sn +qQ. */ - if (IsValid(u)) + if (u->registered == REG_ALL) + { + if (IS_LOCAL(u)) + { + if (!u->quietquit) + { + ServerInstance->SNO->WriteToSnoMask('q',"Client exiting: %s!%s@%s [%s]", u->nick.c_str(), u->ident.c_str(), u->host.c_str(), oper_reason.c_str()); + } + } + else + { + if ((!ServerInstance->SilentULine(u->server)) && (!u->quietquit)) + { + ServerInstance->SNO->WriteToSnoMask('Q',"Client exiting on server %s: %s!%s@%s [%s]", u->server, u->nick.c_str(), u->ident.c_str(), u->host.c_str(), oper_reason.c_str()); + } + } + u->AddToWhoWas(); + } + + if (iter != ServerInstance->Users->clientlist->end()) + { + ServerInstance->Users->clientlist->erase(iter); + } + else + { + ServerInstance->Logs->Log("CULLLIST", DEBUG, "iter == clientlist->end, can't remove them from hash... problematic.."); + } + + if (IS_LOCAL(u)) { - kill_link(u,a->GetReason().c_str()); - list.erase(list.begin()); - /* So that huge numbers of quits dont block, - * we yield back to our mainloop every 15 - * iterations. - * The DoOneIteration call basically acts - * like a software threading mechanism. - */ - if (((n++) % 15) == 0) + std::vector::iterator x = find(ServerInstance->Users->local_users.begin(),ServerInstance->Users->local_users.end(),u); + if (x != ServerInstance->Users->local_users.end()) + ServerInstance->Users->local_users.erase(x); + else { - ServerInstance->DoOneIteration(false); + ServerInstance->Logs->Log("CULLLIST", DEBUG, "Failed to remove user from vector.."); } } - } - return n; + + delete u; + list.erase(list.begin()); + } + + return n; } +