]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/modules.cpp
8255049b177d1d037e2945bf700140d32ced2eb5
[user/henk/code/inspircd.git] / src / modules.cpp
1 /*
2  * InspIRCd -- Internet Relay Chat Daemon
3  *
4  *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
5  *   Copyright (C) 2007, 2009 Dennis Friis <peavey@inspircd.org>
6  *   Copyright (C) 2003-2008 Craig Edwards <craigedwards@brainbox.cc>
7  *   Copyright (C) 2008 Thomas Stagner <aquanight@inspircd.org>
8  *   Copyright (C) 2006-2007 Robin Burchell <robin+git@viroteck.net>
9  *   Copyright (C) 2006-2007 Oliver Lupton <oliverlupton@gmail.com>
10  *   Copyright (C) 2007 Pippijn van Steenhoven <pip88nl@gmail.com>
11  *   Copyright (C) 2003 randomdan <???@???>
12  *
13  * This file is part of InspIRCd.  InspIRCd is free software: you can
14  * redistribute it and/or modify it under the terms of the GNU General Public
15  * License as published by the Free Software Foundation, version 2.
16  *
17  * This program is distributed in the hope that it will be useful, but WITHOUT
18  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
19  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
20  * details.
21  *
22  * You should have received a copy of the GNU General Public License
23  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
24  */
25
26
27 #include <iostream>
28 #include <fstream>
29 #include "inspircd.h"
30 #include "xline.h"
31 #include "socket.h"
32 #include "socketengine.h"
33 #include "command_parse.h"
34 #include "exitcodes.h"
35
36 #ifndef _WIN32
37         #include <dirent.h>
38 #endif
39
40 static std::vector<dynamic_reference_base*>* dynrefs = NULL;
41 static bool dynref_init_complete = false;
42
43 void dynamic_reference_base::reset_all()
44 {
45         dynref_init_complete = true;
46         if (!dynrefs)
47                 return;
48         for(unsigned int i = 0; i < dynrefs->size(); i++)
49                 (*dynrefs)[i]->resolve();
50 }
51
52 // Version is a simple class for holding a modules version number
53 Version::Version(const std::string &desc, int flags) : description(desc), Flags(flags)
54 {
55 }
56
57 Version::Version(const std::string &desc, int flags, const std::string& linkdata)
58 : description(desc), Flags(flags), link_data(linkdata)
59 {
60 }
61
62 Event::Event(Module* src, const std::string &eventid) : source(src), id(eventid) { }
63
64 void Event::Send()
65 {
66         FOREACH_MOD(OnEvent, (*this));
67 }
68
69 // These declarations define the behavours of the base class Module (which does nothing at all)
70
71 Module::Module() { }
72 CullResult Module::cull()
73 {
74         return classbase::cull();
75 }
76 Module::~Module()
77 {
78 }
79
80 void Module::DetachEvent(Implementation i)
81 {
82         ServerInstance->Modules->Detach(i, this);
83 }
84
85 void            Module::ReadConfig(ConfigStatus& status) { }
86 ModResult       Module::OnSendSnotice(char &snomask, std::string &type, const std::string &message) { DetachEvent(I_OnSendSnotice); return MOD_RES_PASSTHRU; }
87 void            Module::OnUserConnect(LocalUser*) { DetachEvent(I_OnUserConnect); }
88 void            Module::OnUserQuit(User*, const std::string&, const std::string&) { DetachEvent(I_OnUserQuit); }
89 void            Module::OnUserDisconnect(LocalUser*) { DetachEvent(I_OnUserDisconnect); }
90 void            Module::OnUserJoin(Membership*, bool, bool, CUList&) { DetachEvent(I_OnUserJoin); }
91 void            Module::OnPostJoin(Membership*) { DetachEvent(I_OnPostJoin); }
92 void            Module::OnUserPart(Membership*, std::string&, CUList&) { DetachEvent(I_OnUserPart); }
93 void            Module::OnPreRehash(User*, const std::string&) { DetachEvent(I_OnPreRehash); }
94 void            Module::OnModuleRehash(User*, const std::string&) { DetachEvent(I_OnModuleRehash); }
95 ModResult       Module::OnUserPreJoin(LocalUser*, Channel*, const std::string&, std::string&, const std::string&) { DetachEvent(I_OnUserPreJoin); return MOD_RES_PASSTHRU; }
96 void            Module::OnMode(User*, User*, Channel*, const std::vector<std::string>&, const std::vector<TranslateType>&) { DetachEvent(I_OnMode); }
97 void            Module::OnOper(User*, const std::string&) { DetachEvent(I_OnOper); }
98 void            Module::OnPostOper(User*, const std::string&, const std::string &) { DetachEvent(I_OnPostOper); }
99 void            Module::OnInfo(User*) { DetachEvent(I_OnInfo); }
100 void            Module::OnWhois(User*, User*) { DetachEvent(I_OnWhois); }
101 ModResult       Module::OnUserPreInvite(User*, User*, Channel*, time_t) { DetachEvent(I_OnUserPreInvite); return MOD_RES_PASSTHRU; }
102 ModResult       Module::OnUserPreMessage(User*, void*, int, std::string&, char, CUList&, MessageType) { DetachEvent(I_OnUserPreMessage); return MOD_RES_PASSTHRU; }
103 ModResult       Module::OnUserPreNick(User*, const std::string&) { DetachEvent(I_OnUserPreNick); return MOD_RES_PASSTHRU; }
104 void            Module::OnUserPostNick(User*, const std::string&) { DetachEvent(I_OnUserPostNick); }
105 ModResult       Module::OnPreMode(User*, User*, Channel*, const std::vector<std::string>&) { DetachEvent(I_OnPreMode); return MOD_RES_PASSTHRU; }
106 void            Module::On005Numeric(std::map<std::string, std::string>&) { DetachEvent(I_On005Numeric); }
107 ModResult       Module::OnKill(User*, User*, const std::string&) { DetachEvent(I_OnKill); return MOD_RES_PASSTHRU; }
108 void            Module::OnLoadModule(Module*) { DetachEvent(I_OnLoadModule); }
109 void            Module::OnUnloadModule(Module*) { DetachEvent(I_OnUnloadModule); }
110 void            Module::OnBackgroundTimer(time_t) { DetachEvent(I_OnBackgroundTimer); }
111 ModResult       Module::OnPreCommand(std::string&, std::vector<std::string>&, LocalUser*, bool, const std::string&) { DetachEvent(I_OnPreCommand); return MOD_RES_PASSTHRU; }
112 void            Module::OnPostCommand(Command*, const std::vector<std::string>&, LocalUser*, CmdResult, const std::string&) { DetachEvent(I_OnPostCommand); }
113 void            Module::OnUserInit(LocalUser*) { DetachEvent(I_OnUserInit); }
114 ModResult       Module::OnCheckReady(LocalUser*) { DetachEvent(I_OnCheckReady); return MOD_RES_PASSTHRU; }
115 ModResult       Module::OnUserRegister(LocalUser*) { DetachEvent(I_OnUserRegister); return MOD_RES_PASSTHRU; }
116 ModResult       Module::OnUserPreKick(User*, Membership*, const std::string&) { DetachEvent(I_OnUserPreKick); return MOD_RES_PASSTHRU; }
117 void            Module::OnUserKick(User*, Membership*, const std::string&, CUList&) { DetachEvent(I_OnUserKick); }
118 ModResult       Module::OnRawMode(User*, Channel*, const char, const std::string &, bool, int) { DetachEvent(I_OnRawMode); return MOD_RES_PASSTHRU; }
119 ModResult       Module::OnCheckInvite(User*, Channel*) { DetachEvent(I_OnCheckInvite); return MOD_RES_PASSTHRU; }
120 ModResult       Module::OnCheckKey(User*, Channel*, const std::string&) { DetachEvent(I_OnCheckKey); return MOD_RES_PASSTHRU; }
121 ModResult       Module::OnCheckLimit(User*, Channel*) { DetachEvent(I_OnCheckLimit); return MOD_RES_PASSTHRU; }
122 ModResult       Module::OnCheckChannelBan(User*, Channel*) { DetachEvent(I_OnCheckChannelBan); return MOD_RES_PASSTHRU; }
123 ModResult       Module::OnCheckBan(User*, Channel*, const std::string&) { DetachEvent(I_OnCheckBan); return MOD_RES_PASSTHRU; }
124 ModResult       Module::OnExtBanCheck(User*, Channel*, char) { DetachEvent(I_OnExtBanCheck); return MOD_RES_PASSTHRU; }
125 ModResult       Module::OnStats(char, User*, string_list&) { DetachEvent(I_OnStats); return MOD_RES_PASSTHRU; }
126 ModResult       Module::OnChangeLocalUserHost(LocalUser*, const std::string&) { DetachEvent(I_OnChangeLocalUserHost); return MOD_RES_PASSTHRU; }
127 ModResult       Module::OnChangeLocalUserGECOS(LocalUser*, const std::string&) { DetachEvent(I_OnChangeLocalUserGECOS); return MOD_RES_PASSTHRU; }
128 ModResult       Module::OnPreTopicChange(User*, Channel*, const std::string&) { DetachEvent(I_OnPreTopicChange); return MOD_RES_PASSTHRU; }
129 void            Module::OnEvent(Event&) { DetachEvent(I_OnEvent); }
130 ModResult       Module::OnPassCompare(Extensible* ex, const std::string &password, const std::string &input, const std::string& hashtype) { DetachEvent(I_OnPassCompare); return MOD_RES_PASSTHRU; }
131 void            Module::OnGlobalOper(User*) { DetachEvent(I_OnGlobalOper); }
132 void            Module::OnPostConnect(User*) { DetachEvent(I_OnPostConnect); }
133 void            Module::OnUserMessage(User*, void*, int, const std::string&, char, const CUList&, MessageType) { DetachEvent(I_OnUserMessage); }
134 void            Module::OnUserInvite(User*, User*, Channel*, time_t) { DetachEvent(I_OnUserInvite); }
135 void            Module::OnPostTopicChange(User*, Channel*, const std::string&) { DetachEvent(I_OnPostTopicChange); }
136 void            Module::OnGetServerDescription(const std::string&, std::string&) { DetachEvent(I_OnGetServerDescription); }
137 void            Module::OnSyncUser(User*, Module*, void*) { DetachEvent(I_OnSyncUser); }
138 void            Module::OnSyncChannel(Channel*, Module*, void*) { DetachEvent(I_OnSyncChannel); }
139 void            Module::OnSyncNetwork(Module*, void*) { DetachEvent(I_OnSyncNetwork); }
140 void            Module::ProtoSendMode(void*, TargetTypeFlags, void*, const std::vector<std::string>&, const std::vector<TranslateType>&) { }
141 void            Module::OnDecodeMetaData(Extensible*, const std::string&, const std::string&) { DetachEvent(I_OnDecodeMetaData); }
142 void            Module::ProtoSendMetaData(void*, Extensible*, const std::string&, const std::string&) { }
143 void            Module::OnChangeHost(User*, const std::string&) { DetachEvent(I_OnChangeHost); }
144 void            Module::OnChangeName(User*, const std::string&) { DetachEvent(I_OnChangeName); }
145 void            Module::OnChangeIdent(User*, const std::string&) { DetachEvent(I_OnChangeIdent); }
146 void            Module::OnAddLine(User*, XLine*) { DetachEvent(I_OnAddLine); }
147 void            Module::OnDelLine(User*, XLine*) { DetachEvent(I_OnDelLine); }
148 void            Module::OnExpireLine(XLine*) { DetachEvent(I_OnExpireLine); }
149 void            Module::OnCleanup(int, void*) { }
150 ModResult       Module::OnChannelPreDelete(Channel*) { DetachEvent(I_OnChannelPreDelete); return MOD_RES_PASSTHRU; }
151 void            Module::OnChannelDelete(Channel*) { DetachEvent(I_OnChannelDelete); }
152 ModResult       Module::OnSetAway(User*, const std::string &) { DetachEvent(I_OnSetAway); return MOD_RES_PASSTHRU; }
153 ModResult       Module::OnWhoisLine(User*, User*, int&, std::string&) { DetachEvent(I_OnWhoisLine); return MOD_RES_PASSTHRU; }
154 void            Module::OnBuildNeighborList(User*, UserChanList&, std::map<User*,bool>&) { DetachEvent(I_OnBuildNeighborList); }
155 void            Module::OnGarbageCollect() { DetachEvent(I_OnGarbageCollect); }
156 ModResult       Module::OnSetConnectClass(LocalUser* user, ConnectClass* myclass) { DetachEvent(I_OnSetConnectClass); return MOD_RES_PASSTHRU; }
157 void            Module::OnText(User*, void*, int, const std::string&, char, CUList&) { DetachEvent(I_OnText); }
158 void            Module::OnRunTestSuite() { DetachEvent(I_OnRunTestSuite); }
159 void            Module::OnNamesListItem(User*, Membership*, std::string&, std::string&) { DetachEvent(I_OnNamesListItem); }
160 ModResult       Module::OnNumeric(User*, unsigned int, const std::string&) { DetachEvent(I_OnNumeric); return MOD_RES_PASSTHRU; }
161 void            Module::OnHookIO(StreamSocket*, ListenSocket*) { DetachEvent(I_OnHookIO); }
162 ModResult   Module::OnAcceptConnection(int, ListenSocket*, irc::sockets::sockaddrs*, irc::sockets::sockaddrs*) { DetachEvent(I_OnAcceptConnection); return MOD_RES_PASSTHRU; }
163 void            Module::OnSendWhoLine(User*, const std::vector<std::string>&, User*, std::string&) { DetachEvent(I_OnSendWhoLine); }
164 void            Module::OnSetUserIP(LocalUser*) { DetachEvent(I_OnSetUserIP); }
165
166 ServiceProvider::ServiceProvider(Module* Creator, const std::string& Name, ServiceType Type)
167         : creator(Creator), name(Name), service(Type)
168 {
169         if ((ServerInstance) && (ServerInstance->Modules->NewServices))
170                 ServerInstance->Modules->NewServices->push_back(this);
171 }
172
173 void ServiceProvider::DisableAutoRegister()
174 {
175         if ((ServerInstance) && (ServerInstance->Modules->NewServices))
176         {
177                 ModuleManager::ServiceList& list = *ServerInstance->Modules->NewServices;
178                 ModuleManager::ServiceList::iterator it = std::find(list.begin(), list.end(), this);
179                 if (it != list.end())
180                         list.erase(it);
181         }
182 }
183
184 ModuleManager::ModuleManager()
185 {
186 }
187
188 ModuleManager::~ModuleManager()
189 {
190 }
191
192 bool ModuleManager::Attach(Implementation i, Module* mod)
193 {
194         if (std::find(EventHandlers[i].begin(), EventHandlers[i].end(), mod) != EventHandlers[i].end())
195                 return false;
196
197         EventHandlers[i].push_back(mod);
198         return true;
199 }
200
201 bool ModuleManager::Detach(Implementation i, Module* mod)
202 {
203         EventHandlerIter x = std::find(EventHandlers[i].begin(), EventHandlers[i].end(), mod);
204
205         if (x == EventHandlers[i].end())
206                 return false;
207
208         EventHandlers[i].erase(x);
209         return true;
210 }
211
212 void ModuleManager::Attach(Implementation* i, Module* mod, size_t sz)
213 {
214         for (size_t n = 0; n < sz; ++n)
215                 Attach(i[n], mod);
216 }
217
218 void ModuleManager::AttachAll(Module* mod)
219 {
220         for (size_t i = I_BEGIN + 1; i != I_END; ++i)
221                 Attach((Implementation)i, mod);
222 }
223
224 void ModuleManager::DetachAll(Module* mod)
225 {
226         for (size_t n = I_BEGIN + 1; n != I_END; ++n)
227                 Detach((Implementation)n, mod);
228 }
229
230 bool ModuleManager::SetPriority(Module* mod, Priority s)
231 {
232         for (size_t n = I_BEGIN + 1; n != I_END; ++n)
233                 SetPriority(mod, (Implementation)n, s);
234
235         return true;
236 }
237
238 bool ModuleManager::SetPriority(Module* mod, Implementation i, Priority s, Module* which)
239 {
240         /** To change the priority of a module, we first find its position in the vector,
241          * then we find the position of the other modules in the vector that this module
242          * wants to be before/after. We pick off either the first or last of these depending
243          * on which they want, and we make sure our module is *at least* before or after
244          * the first or last of this subset, depending again on the type of priority.
245          */
246         size_t my_pos = 0;
247
248         /* Locate our module. This is O(n) but it only occurs on module load so we're
249          * not too bothered about it
250          */
251         for (size_t x = 0; x != EventHandlers[i].size(); ++x)
252         {
253                 if (EventHandlers[i][x] == mod)
254                 {
255                         my_pos = x;
256                         goto found_src;
257                 }
258         }
259
260         /* Eh? this module doesnt exist, probably trying to set priority on an event
261          * theyre not attached to.
262          */
263         return false;
264
265 found_src:
266         // The modules registered for a hook are called in reverse order (to allow for easier removal
267         // of list entries while looping), meaning that the Priority given to us has the exact opposite effect
268         // on the list, e.g.: PRIORITY_BEFORE will actually put 'mod' after 'which', etc.
269         size_t swap_pos = my_pos;
270         switch (s)
271         {
272                 case PRIORITY_LAST:
273                         if (prioritizationState != PRIO_STATE_FIRST)
274                                 return true;
275                         else
276                                 swap_pos = 0;
277                         break;
278                 case PRIORITY_FIRST:
279                         if (prioritizationState != PRIO_STATE_FIRST)
280                                 return true;
281                         else
282                                 swap_pos = EventHandlers[i].size() - 1;
283                         break;
284                 case PRIORITY_BEFORE:
285                 {
286                         /* Find the latest possible position, only searching AFTER our position */
287                         for (size_t x = EventHandlers[i].size() - 1; x > my_pos; --x)
288                         {
289                                 if (EventHandlers[i][x] == which)
290                                 {
291                                         swap_pos = x;
292                                         goto swap_now;
293                                 }
294                         }
295                         // didn't find it - either not loaded or we're already after
296                         return true;
297                 }
298                 /* Place this module before a set of other modules */
299                 case PRIORITY_AFTER:
300                 {
301                         for (size_t x = 0; x < my_pos; ++x)
302                         {
303                                 if (EventHandlers[i][x] == which)
304                                 {
305                                         swap_pos = x;
306                                         goto swap_now;
307                                 }
308                         }
309                         // didn't find it - either not loaded or we're already before
310                         return true;
311                 }
312         }
313
314 swap_now:
315         /* Do we need to swap? */
316         if (swap_pos != my_pos)
317         {
318                 // We are going to change positions; we'll need to run again to verify all requirements
319                 if (prioritizationState == PRIO_STATE_LAST)
320                         prioritizationState = PRIO_STATE_AGAIN;
321                 /* Suggestion from Phoenix, "shuffle" the modules to better retain call order */
322                 int incrmnt = 1;
323
324                 if (my_pos > swap_pos)
325                         incrmnt = -1;
326
327                 for (unsigned int j = my_pos; j != swap_pos; j += incrmnt)
328                 {
329                         if ((j + incrmnt > EventHandlers[i].size() - 1) || ((incrmnt == -1) && (j == 0)))
330                                 continue;
331
332                         std::swap(EventHandlers[i][j], EventHandlers[i][j+incrmnt]);
333                 }
334         }
335
336         return true;
337 }
338
339 bool ModuleManager::PrioritizeHooks()
340 {
341         /* We give every module a chance to re-prioritize when we introduce a new one,
342          * not just the one thats loading, as the new module could affect the preference
343          * of others
344          */
345         for (int tries = 0; tries < 20; tries++)
346         {
347                 prioritizationState = tries > 0 ? PRIO_STATE_LAST : PRIO_STATE_FIRST;
348                 for (std::map<std::string, Module*>::iterator n = Modules.begin(); n != Modules.end(); ++n)
349                         n->second->Prioritize();
350
351                 if (prioritizationState == PRIO_STATE_LAST)
352                         break;
353                 if (tries == 19)
354                 {
355                         ServerInstance->Logs->Log("MODULE", LOG_DEFAULT, "Hook priority dependency loop detected");
356                         return false;
357                 }
358         }
359         return true;
360 }
361
362 bool ModuleManager::CanUnload(Module* mod)
363 {
364         std::map<std::string, Module*>::iterator modfind = Modules.find(mod->ModuleSourceFile);
365
366         if ((modfind == Modules.end()) || (modfind->second != mod) || (mod->dying))
367         {
368                 LastModuleError = "Module " + mod->ModuleSourceFile + " is not loaded, cannot unload it!";
369                 ServerInstance->Logs->Log("MODULE", LOG_DEFAULT, LastModuleError);
370                 return false;
371         }
372         if (mod->GetVersion().Flags & VF_STATIC)
373         {
374                 LastModuleError = "Module " + mod->ModuleSourceFile + " not unloadable (marked static)";
375                 ServerInstance->Logs->Log("MODULE", LOG_DEFAULT, LastModuleError);
376                 return false;
377         }
378
379         mod->dying = true;
380         return true;
381 }
382
383 void ModuleManager::DoSafeUnload(Module* mod)
384 {
385         // First, notify all modules that a module is about to be unloaded, so in case
386         // they pass execution to the soon to be unloaded module, it will happen now,
387         // i.e. before we unregister the services of the module being unloaded
388         FOREACH_MOD(OnUnloadModule, (mod));
389
390         std::map<std::string, Module*>::iterator modfind = Modules.find(mod->ModuleSourceFile);
391
392         std::vector<reference<ExtensionItem> > items;
393         ServerInstance->Extensions.BeginUnregister(modfind->second, items);
394         /* Give the module a chance to tidy out all its metadata */
395         for (chan_hash::iterator c = ServerInstance->chanlist->begin(); c != ServerInstance->chanlist->end(); c++)
396         {
397                 mod->OnCleanup(TYPE_CHANNEL,c->second);
398                 c->second->doUnhookExtensions(items);
399                 const UserMembList* users = c->second->GetUsers();
400                 for(UserMembCIter mi = users->begin(); mi != users->end(); mi++)
401                         mi->second->doUnhookExtensions(items);
402         }
403         for (user_hash::iterator u = ServerInstance->Users->clientlist->begin(); u != ServerInstance->Users->clientlist->end(); u++)
404         {
405                 mod->OnCleanup(TYPE_USER,u->second);
406                 u->second->doUnhookExtensions(items);
407         }
408         for(char m='A'; m <= 'z'; m++)
409         {
410                 ModeHandler* mh;
411                 mh = ServerInstance->Modes->FindMode(m, MODETYPE_USER);
412                 if (mh && mh->creator == mod)
413                         this->DelService(*mh);
414                 mh = ServerInstance->Modes->FindMode(m, MODETYPE_CHANNEL);
415                 if (mh && mh->creator == mod)
416                         this->DelService(*mh);
417         }
418         for(std::multimap<std::string, ServiceProvider*>::iterator i = DataProviders.begin(); i != DataProviders.end(); )
419         {
420                 std::multimap<std::string, ServiceProvider*>::iterator curr = i++;
421                 if (curr->second->creator == mod)
422                         DataProviders.erase(curr);
423         }
424
425         dynamic_reference_base::reset_all();
426
427         DetachAll(mod);
428
429         Modules.erase(modfind);
430         ServerInstance->GlobalCulls.AddItem(mod);
431
432         ServerInstance->Logs->Log("MODULE", LOG_DEFAULT, "Module %s unloaded",mod->ModuleSourceFile.c_str());
433         ServerInstance->ISupport.Build();
434 }
435
436 void ModuleManager::UnloadAll()
437 {
438         /* We do this more than once, so that any service providers get a
439          * chance to be unhooked by the modules using them, but then get
440          * a chance to be removed themsleves.
441          *
442          * Note: this deliberately does NOT delete the DLLManager objects
443          */
444         for (int tries = 0; tries < 4; tries++)
445         {
446                 std::map<std::string, Module*>::iterator i = Modules.begin();
447                 while (i != Modules.end())
448                 {
449                         std::map<std::string, Module*>::iterator me = i++;
450                         if (CanUnload(me->second))
451                         {
452                                 DoSafeUnload(me->second);
453                         }
454                 }
455                 ServerInstance->GlobalCulls.Apply();
456         }
457 }
458
459 namespace
460 {
461         struct UnloadAction : public HandlerBase0<void>
462         {
463                 Module* const mod;
464                 UnloadAction(Module* m) : mod(m) {}
465                 void Call()
466                 {
467                         DLLManager* dll = mod->ModuleDLLManager;
468                         ServerInstance->Modules->DoSafeUnload(mod);
469                         ServerInstance->GlobalCulls.Apply();
470                         // In pure static mode this is always NULL
471                         delete dll;
472                         ServerInstance->GlobalCulls.AddItem(this);
473                 }
474         };
475
476         struct ReloadAction : public HandlerBase0<void>
477         {
478                 Module* const mod;
479                 HandlerBase1<void, bool>* const callback;
480                 ReloadAction(Module* m, HandlerBase1<void, bool>* c)
481                         : mod(m), callback(c) {}
482                 void Call()
483                 {
484                         DLLManager* dll = mod->ModuleDLLManager;
485                         std::string name = mod->ModuleSourceFile;
486                         ServerInstance->Modules->DoSafeUnload(mod);
487                         ServerInstance->GlobalCulls.Apply();
488                         delete dll;
489                         bool rv = ServerInstance->Modules->Load(name);
490                         if (callback)
491                                 callback->Call(rv);
492                         ServerInstance->GlobalCulls.AddItem(this);
493                 }
494         };
495 }
496
497 bool ModuleManager::Unload(Module* mod)
498 {
499         if (!CanUnload(mod))
500                 return false;
501         ServerInstance->AtomicActions.AddAction(new UnloadAction(mod));
502         return true;
503 }
504
505 void ModuleManager::Reload(Module* mod, HandlerBase1<void, bool>* callback)
506 {
507         if (CanUnload(mod))
508                 ServerInstance->AtomicActions.AddAction(new ReloadAction(mod, callback));
509         else
510                 callback->Call(false);
511 }
512
513 void ModuleManager::LoadAll()
514 {
515         std::map<std::string, ServiceList> servicemap;
516         LoadCoreModules(servicemap);
517
518         ConfigTagList tags = ServerInstance->Config->ConfTags("module");
519         for (ConfigIter i = tags.first; i != tags.second; ++i)
520         {
521                 ConfigTag* tag = i->second;
522                 std::string name = tag->getString("name");
523                 this->NewServices = &servicemap[name];
524                 std::cout << "[" << con_green << "*" << con_reset << "] Loading module:\t" << con_green << name << con_reset << std::endl;
525
526                 if (!this->Load(name, true))
527                 {
528                         ServerInstance->Logs->Log("MODULE", LOG_DEFAULT, this->LastError());
529                         std::cout << std::endl << "[" << con_red << "*" << con_reset << "] " << this->LastError() << std::endl << std::endl;
530                         ServerInstance->Exit(EXIT_STATUS_MODULE);
531                 }
532         }
533
534         ConfigStatus confstatus;
535
536         for (ModuleMap::const_iterator i = Modules.begin(); i != Modules.end(); ++i)
537         {
538                 Module* mod = i->second;
539                 try
540                 {
541                         ServerInstance->Logs->Log("MODULE", LOG_DEBUG, "Initializing %s", i->first.c_str());
542                         AttachAll(mod);
543                         AddServices(servicemap[i->first]);
544                         mod->init();
545                         mod->ReadConfig(confstatus);
546                 }
547                 catch (CoreException& modexcept)
548                 {
549                         LastModuleError = "Unable to initialize " + mod->ModuleSourceFile + ": " + modexcept.GetReason();
550                         ServerInstance->Logs->Log("MODULE", LOG_DEFAULT, LastModuleError);
551                         std::cout << std::endl << "[" << con_red << "*" << con_reset << "] " << LastModuleError << std::endl << std::endl;
552                         ServerInstance->Exit(EXIT_STATUS_MODULE);
553                 }
554         }
555
556         if (!PrioritizeHooks())
557                 ServerInstance->Exit(EXIT_STATUS_MODULE);
558 }
559
560 std::string& ModuleManager::LastError()
561 {
562         return LastModuleError;
563 }
564
565 void ModuleManager::AddServices(const ServiceList& list)
566 {
567         for (ServiceList::const_iterator i = list.begin(); i != list.end(); ++i)
568         {
569                 ServiceProvider& s = **i;
570                 AddService(s);
571         }
572 }
573
574 void ModuleManager::AddService(ServiceProvider& item)
575 {
576         switch (item.service)
577         {
578                 case SERVICE_COMMAND:
579                         if (!ServerInstance->Parser->AddCommand(static_cast<Command*>(&item)))
580                                 throw ModuleException("Command "+std::string(item.name)+" already exists.");
581                         return;
582                 case SERVICE_MODE:
583                 {
584                         ModeHandler* mh = static_cast<ModeHandler*>(&item);
585                         if (!ServerInstance->Modes->AddMode(mh))
586                                 throw ModuleException("Mode "+std::string(item.name)+" already exists.");
587                         DataProviders.insert(std::make_pair((mh->GetModeType() == MODETYPE_CHANNEL ? "mode/" : "umode/") + item.name, &item));
588                         dynamic_reference_base::reset_all();
589                         return;
590                 }
591                 case SERVICE_METADATA:
592                         if (!ServerInstance->Extensions.Register(static_cast<ExtensionItem*>(&item)))
593                                 throw ModuleException("Extension " + std::string(item.name) + " already exists.");
594                         return;
595                 case SERVICE_DATA:
596                 case SERVICE_IOHOOK:
597                 {
598                         if ((item.name.substr(0, 5) == "mode/") || (item.name.substr(0, 6) == "umode/"))
599                                 throw ModuleException("The \"mode/\" and the \"umode\" service name prefixes are reserved.");
600
601                         DataProviders.insert(std::make_pair(item.name, &item));
602                         std::string::size_type slash = item.name.find('/');
603                         if (slash != std::string::npos)
604                         {
605                                 DataProviders.insert(std::make_pair(item.name.substr(0, slash), &item));
606                                 DataProviders.insert(std::make_pair(item.name.substr(slash + 1), &item));
607                         }
608                         dynamic_reference_base::reset_all();
609                         return;
610                 }
611                 default:
612                         throw ModuleException("Cannot add unknown service type");
613         }
614 }
615
616 void ModuleManager::DelService(ServiceProvider& item)
617 {
618         switch (item.service)
619         {
620                 case SERVICE_MODE:
621                         if (!ServerInstance->Modes->DelMode(static_cast<ModeHandler*>(&item)))
622                                 throw ModuleException("Mode "+std::string(item.name)+" does not exist.");
623                         // Fall through
624                 case SERVICE_DATA:
625                 case SERVICE_IOHOOK:
626                 {
627                         for(std::multimap<std::string, ServiceProvider*>::iterator i = DataProviders.begin(); i != DataProviders.end(); )
628                         {
629                                 std::multimap<std::string, ServiceProvider*>::iterator curr = i++;
630                                 if (curr->second == &item)
631                                         DataProviders.erase(curr);
632                         }
633                         dynamic_reference_base::reset_all();
634                         return;
635                 }
636                 default:
637                         throw ModuleException("Cannot delete unknown service type");
638         }
639 }
640
641 ServiceProvider* ModuleManager::FindService(ServiceType type, const std::string& name)
642 {
643         switch (type)
644         {
645                 case SERVICE_DATA:
646                 case SERVICE_IOHOOK:
647                 {
648                         std::multimap<std::string, ServiceProvider*>::iterator i = DataProviders.find(name);
649                         if (i != DataProviders.end() && i->second->service == type)
650                                 return i->second;
651                         return NULL;
652                 }
653                 // TODO implement finding of the other types
654                 default:
655                         throw ModuleException("Cannot find unknown service type");
656         }
657 }
658
659 dynamic_reference_base::dynamic_reference_base(Module* Creator, const std::string& Name)
660         : name(Name), value(NULL), creator(Creator)
661 {
662         if (!dynrefs)
663                 dynrefs = new std::vector<dynamic_reference_base*>;
664         dynrefs->push_back(this);
665         if (dynref_init_complete)
666                 resolve();
667 }
668
669 dynamic_reference_base::~dynamic_reference_base()
670 {
671         for(unsigned int i = 0; i < dynrefs->size(); i++)
672         {
673                 if (dynrefs->at(i) == this)
674                 {
675                         unsigned int last = dynrefs->size() - 1;
676                         if (i != last)
677                                 dynrefs->at(i) = dynrefs->at(last);
678                         dynrefs->erase(dynrefs->begin() + last);
679                         if (dynrefs->empty())
680                         {
681                                 delete dynrefs;
682                                 dynrefs = NULL;
683                         }
684                         return;
685                 }
686         }
687 }
688
689 void dynamic_reference_base::SetProvider(const std::string& newname)
690 {
691         name = newname;
692         resolve();
693 }
694
695 void dynamic_reference_base::resolve()
696 {
697         std::multimap<std::string, ServiceProvider*>::iterator i = ServerInstance->Modules->DataProviders.find(name);
698         if (i != ServerInstance->Modules->DataProviders.end())
699                 value = static_cast<DataProvider*>(i->second);
700         else
701                 value = NULL;
702 }
703
704 Module* ModuleManager::Find(const std::string &name)
705 {
706         std::map<std::string, Module*>::iterator modfind = Modules.find(name);
707
708         if (modfind == Modules.end())
709                 return NULL;
710         else
711                 return modfind->second;
712 }
713
714 FileReader::FileReader(const std::string& filename)
715 {
716         Load(filename);
717 }
718
719 void FileReader::Load(const std::string& filename)
720 {
721         // If the file is stored in the file cache then we used that version instead.
722         std::string realName = ServerInstance->Config->Paths.PrependConfig(filename);
723         ConfigFileCache::iterator it = ServerInstance->Config->Files.find(realName);
724         if (it != ServerInstance->Config->Files.end())
725         {
726                 this->lines = it->second;
727         }
728         else
729         {
730                 lines.clear();
731
732                 std::ifstream stream(realName.c_str());
733                 if (!stream.is_open())
734                         throw CoreException(filename + " does not exist or is not readable!");
735
736                 std::string line;
737                 while (std::getline(stream, line))
738                 {
739                         lines.push_back(line);
740                         totalSize += line.size() + 2;
741                 }
742
743                 stream.close();
744         }
745 }
746
747 std::string FileReader::GetString()
748 {
749         std::string buffer;
750         for (file_cache::iterator it = this->lines.begin(); it != this->lines.end(); ++it)
751         {
752                 buffer.append(*it);
753                 buffer.append("\r\n");
754         }
755         return buffer;
756 }