1
0
mirror of https://github.com/anope/anope.git synced 2026-06-25 14:06:37 +02:00
Files
anope/modules/rpc/rpc_main.cpp
T
2025-02-13 01:54:57 +00:00

307 lines
8.2 KiB
C++

/*
*
* (C) 2010-2025 Anope Team
* Contact us at team@anope.org
*
* Please read COPYING and README for further details.
*/
#include "module.h"
#include "modules/rpc.h"
static Module *me;
class RPCIdentifyRequest final
: public IdentifyRequest
{
RPCRequest request;
HTTPReply repl; /* Request holds a reference to the HTTPReply, because we might exist long enough to invalidate it
we'll copy it here then reset the reference before we use it */
Reference<HTTPClient> client;
Reference<RPCServiceInterface> xinterface;
public:
RPCIdentifyRequest(Module *m, RPCRequest &req, HTTPClient *c, RPCServiceInterface *iface, const Anope::string &acc, const Anope::string &pass) : IdentifyRequest(m, acc, pass), request(req), repl(request.r), client(c), xinterface(iface) { }
void OnSuccess() override
{
if (!xinterface || !client)
return;
request.r = this->repl;
request.reply("result", "Success");
request.reply("account", GetAccount());
xinterface->Reply(request);
client->SendReply(&request.r);
}
void OnFail() override
{
if (!xinterface || !client)
return;
request.r = this->repl;
request.reply("error", "Invalid password");
xinterface->Reply(request);
client->SendReply(&request.r);
}
};
class MyRPCEvent final
: public RPCEvent
{
public:
bool Run(RPCServiceInterface *iface, HTTPClient *client, RPCRequest &request) override
{
if (request.name == "command")
this->DoCommand(iface, client, request);
else if (request.name == "checkAuthentication")
return this->DoCheckAuthentication(iface, client, request);
else if (request.name == "stats")
this->DoStats(iface, client, request);
else if (request.name == "channel")
this->DoChannel(iface, client, request);
else if (request.name == "user")
this->DoUser(iface, client, request);
else if (request.name == "opers")
this->DoOperType(iface, client, request);
else if (request.name == "notice")
this->DoNotice(iface, client, request);
return true;
}
private:
void DoCommand(RPCServiceInterface *iface, HTTPClient *client, RPCRequest &request)
{
Anope::string service = request.data.size() > 0 ? request.data[0] : "";
Anope::string user = request.data.size() > 1 ? request.data[1] : "";
Anope::string command = request.data.size() > 2 ? request.data[2] : "";
if (service.empty() || user.empty() || command.empty())
request.reply("error", "Invalid parameters");
else
{
BotInfo *bi = BotInfo::Find(service, true);
if (!bi)
request.reply("error", "Invalid service");
else
{
request.reply("result", "Success");
NickAlias *na = NickAlias::Find(user);
Anope::string out;
struct RPCommandReply final
: CommandReply
{
Anope::string &str;
RPCommandReply(Anope::string &s) : str(s) { }
void SendMessage(BotInfo *source, const Anope::string &msg) override
{
str += msg + "\n";
};
}
reply(out);
User *u = User::Find(user, true);
CommandSource source(user, u, na ? *na->nc : NULL, &reply, bi);
Command::Run(source, command);
if (!out.empty())
request.reply("return", out);
}
}
}
static bool DoCheckAuthentication(RPCServiceInterface *iface, HTTPClient *client, RPCRequest &request)
{
Anope::string username = request.data.size() > 0 ? request.data[0] : "";
Anope::string password = request.data.size() > 1 ? request.data[1] : "";
if (username.empty() || password.empty())
request.reply("error", "Invalid parameters");
else
{
auto *req = new RPCIdentifyRequest(me, request, client, iface, username, password);
FOREACH_MOD(OnCheckAuthentication, (NULL, req));
req->Dispatch();
return false;
}
return true;
}
static void DoStats(RPCServiceInterface *iface, HTTPClient *client, RPCRequest &request)
{
request.reply("uptime", Anope::ToString(Anope::CurTime - Anope::StartTime));
request.reply("uplinkname", Me->GetLinks().front()->GetName());
{
Anope::string buf;
for (const auto &capab : Servers::Capab)
buf += " " + capab;
if (!buf.empty())
buf.erase(buf.begin());
request.reply("uplinkcapab", buf);
}
request.reply("usercount", Anope::ToString(UserListByNick.size()));
request.reply("maxusercount", Anope::ToString(MaxUserCount));
request.reply("channelcount", Anope::ToString(ChannelList.size()));
}
static void DoChannel(RPCServiceInterface *iface, HTTPClient *client, RPCRequest &request)
{
if (request.data.empty())
return;
Channel *c = Channel::Find(request.data[0]);
request.reply("name", c ? c->name : request.data[0]);
if (c)
{
request.reply("bancount", Anope::ToString(c->HasMode("BAN")));
int count = 0;
for (auto &ban : c->GetModeList("BAN"))
request.reply("ban" + Anope::ToString(++count), ban);
request.reply("exceptcount", Anope::ToString(c->HasMode("EXCEPT")));
count = 0;
for (auto &except : c->GetModeList("EXCEPT"))
request.reply("except" + Anope::ToString(++count), except);
request.reply("invitecount", Anope::ToString(c->HasMode("INVITEOVERRIDE")));
count = 0;
for (auto &invite : c->GetModeList("INVITEOVERRIDE"))
request.reply("invite" + Anope::ToString(++count), invite);
Anope::string users;
for (Channel::ChanUserList::const_iterator it = c->users.begin(); it != c->users.end(); ++it)
{
ChanUserContainer *uc = it->second;
users += uc->status.BuildModePrefixList() + uc->user->nick + " ";
}
if (!users.empty())
{
users.erase(users.length() - 1);
request.reply("users", users);
}
if (!c->topic.empty())
request.reply("topic", c->topic);
if (!c->topic_setter.empty())
request.reply("topicsetter", c->topic_setter);
request.reply("topictime", Anope::ToString(c->topic_time));
request.reply("topicts", Anope::ToString(c->topic_ts));
}
}
static void DoUser(RPCServiceInterface *iface, HTTPClient *client, RPCRequest &request)
{
if (request.data.empty())
return;
User *u = User::Find(request.data[0]);
request.reply("nick", u ? u->nick : request.data[0]);
if (u)
{
request.reply("ident", u->GetIdent());
request.reply("vident", u->GetVIdent());
request.reply("host", u->host);
if (!u->vhost.empty())
request.reply("vhost", u->vhost);
if (!u->chost.empty())
request.reply("chost", u->chost);
request.reply("ip", u->ip.addr());
request.reply("timestamp", Anope::ToString(u->timestamp));
request.reply("signon", Anope::ToString(u->signon));
if (u->IsIdentified())
{
request.reply("account", u->Account()->display);
if (u->Account()->o)
request.reply("opertype", u->Account()->o->ot->GetName());
}
Anope::string channels;
for (User::ChanUserList::const_iterator it = u->chans.begin(); it != u->chans.end(); ++it)
{
ChanUserContainer *cc = it->second;
channels += cc->status.BuildModePrefixList() + cc->chan->name + " ";
}
if (!channels.empty())
{
channels.erase(channels.length() - 1);
request.reply("channels", channels);
}
}
}
static void DoOperType(RPCServiceInterface *iface, HTTPClient *client, RPCRequest &request)
{
for (auto *ot : Config->MyOperTypes)
{
Anope::string perms;
for (const auto &priv : ot->GetPrivs())
perms += " " + priv;
for (const auto &command : ot->GetCommands())
perms += " " + command;
request.reply(ot->GetName(), perms);
}
}
static void DoNotice(RPCServiceInterface *iface, HTTPClient *client, RPCRequest &request)
{
Anope::string from = request.data.size() > 0 ? request.data[0] : "";
Anope::string to = request.data.size() > 1 ? request.data[1] : "";
Anope::string message = request.data.size() > 2 ? request.data[2] : "";
BotInfo *bi = BotInfo::Find(from, true);
User *u = User::Find(to, true);
if (!bi || !u || message.empty())
return;
u->SendMessage(bi, message);
request.reply("result", "Success");
}
};
class ModuleRPCMain final
: public Module
{
ServiceReference<RPCServiceInterface> rpc;
MyRPCEvent stats;
public:
ModuleRPCMain(const Anope::string &modname, const Anope::string &creator) : Module(modname, creator, EXTRA | VENDOR), rpc("RPCServiceInterface", "rpc")
{
me = this;
if (!rpc)
throw ModuleException("Unable to find RPC interface, is xmlrpc loaded?");
rpc->Register(&stats);
}
~ModuleRPCMain() override
{
if (rpc)
rpc->Unregister(&stats);
}
};
MODULE_INIT(ModuleRPCMain)