1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
|
/*
* InspIRCd -- Internet Relay Chat Daemon
*
* Copyright (C) 2010 Jackmcbarn <jackmcbarn@jackmcbarn.no-ip.org>
*
* This file is part of InspIRCd. InspIRCd 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, version 2.
*
* 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, see <http://www.gnu.org/licenses/>.
*/
#include "inspircd.h"
#include "account.h"
/* $ModDesc: Allow vhosts to be set on accounts */
static dynamic_reference<AccountProvider> account("account");
static dynamic_reference<AccountDBProvider> db("accountdb");
/** Handle /ACCTVHOST
*/
class CommandAcctvhost : public Command
{
char* hostmap;
void UpdateVhosts(irc::string acctname, std::string new_vhost)
{
for (user_hash::const_iterator i = ServerInstance->Users->clientlist->begin(); i != ServerInstance->Users->clientlist->end(); ++i)
{
if(!IS_LOCAL(i->second) || acctname != account->GetAccountName(i->second))
continue;
i->second->ChangeDisplayedHost(new_vhost.c_str());
}
}
public:
TSStringExtItem vhost;
CommandAcctvhost(Module* Creator, char* hmap) : Command(Creator,"ACCTVHOST", 1, 2), hostmap(hmap), vhost("Vhost", "", Creator)
{
flags_needed = 'o'; syntax = "<account name> [vhost]";
}
CmdResult Handle (const std::vector<std::string>& parameters, User *user)
{
AccountDBEntry* entry = db->GetAccount(parameters[0], false);
if(!entry)
{
user->WriteServ("NOTICE %s :No such account", user->nick.c_str());
return CMD_FAILURE;
}
if(parameters.size() > 1 && !parameters[1].empty())
{
if(IS_LOCAL(user))
{
size_t len = 0;
for (std::string::const_iterator x = parameters[1].begin(); x != parameters[1].end(); ++x, ++len)
{
if (!hostmap[(const unsigned char)*x])
{
user->WriteServ("NOTICE %s :Invalid characters in hostname", user->nick.c_str());
return CMD_FAILURE;
}
}
if (len > 64)
{
user->WriteServ("NOTICE %s :Host too long", user->nick.c_str());
return CMD_FAILURE;
}
vhost.set(entry, parameters[1]);
db->SendUpdate(entry, "Vhost");
ServerInstance->SNO->WriteGlobalSno('a', "%s used ACCTVHOST to set the vhost of account %s to %s", user->nick.c_str(), entry->name.c_str(), parameters[1].c_str());
user->WriteServ("NOTICE %s :Account %s vhost set to %s", user->nick.c_str(), entry->name.c_str(), parameters[1].c_str());
}
UpdateVhosts(entry->name, parameters[1]);
}
else if(IS_LOCAL(user))
{
vhost.set(entry, "");
db->SendUpdate(entry, "Vhost");
ServerInstance->SNO->WriteGlobalSno('a', "%s used ACCTVHOST to remove the vhost of account %s", user->nick.c_str(), entry->name.c_str());
user->WriteServ("NOTICE %s :Account %s vhost removed", user->nick.c_str(), entry->name.c_str());
}
return CMD_SUCCESS;
}
RouteDescriptor GetRouting(User* user, const std::vector<std::string>& parameters)
{
return ROUTE_OPT_BCAST;
}
};
class ModuleAccountVhost : public Module
{
char hostmap[256];
CommandAcctvhost cmd_acctvhost;
public:
ModuleAccountVhost() : cmd_acctvhost(this, hostmap)
{
}
void init()
{
if(!db) throw ModuleException("m_account_vhost requires that m_account be loaded");
ServerInstance->Modules->AddService(cmd_acctvhost);
ServerInstance->Modules->AddService(cmd_acctvhost.vhost);
Implementation eventlist[] = { I_OnEvent, I_OnUserConnect };
ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation));
}
void ReadConfig(ConfigReadStatus&)
{
std::string hmap = ServerInstance->Config->GetTag("hostname")->getString("charmap");
if (hmap.empty())
hmap = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz.-_/0123456789";
memset(hostmap, 0, sizeof(hostmap));
for (std::string::iterator n = hmap.begin(); n != hmap.end(); n++)
hostmap[(unsigned char)*n] = 1;
}
void OnEvent(Event& event)
{
if(event.id == "account_login"){
AccountEvent& acct_event = static_cast<AccountEvent&>(event);
if ((!IS_LOCAL(acct_event.user)) || (acct_event.user->registered != REG_ALL) || (acct_event.account.empty()))
return;
AccountDBEntry* entry = db->GetAccount(acct_event.account, false);
if(!entry)
return;
std::string* vhost = cmd_acctvhost.vhost.get_value(entry);
if(vhost && !vhost->empty())
acct_event.user->ChangeDisplayedHost(vhost->c_str());
}
}
virtual void OnUserConnect(LocalUser* user)
{
if(!account->IsRegistered(user))
return;
AccountDBEntry* entry = db->GetAccount(account->GetAccountName(user), false);
if(!entry)
return;
std::string* vhost = cmd_acctvhost.vhost.get_value(entry);
if(vhost && !vhost->empty())
user->ChangeDisplayedHost(vhost->c_str());
}
void Prioritize()
{
ServerInstance->Modules->SetPriority(this, I_ModuleInit, PRIORITY_AFTER, ServerInstance->Modules->Find("m_account.so"));
ServerInstance->Modules->SetPriority(this, I_OnUserConnect, PRIORITY_AFTER, ServerInstance->Modules->Find("m_hostchange.so"));
}
Version GetVersion()
{
return Version("Allow vhosts to be set on accounts", VF_VENDOR|VF_OPTCOMMON);
}
};
MODULE_INIT(ModuleAccountVhost)
|
