diff --git a/extensions/Makefile.am b/extensions/Makefile.am index cf44a370..1827f911 100644 --- a/extensions/Makefile.am +++ b/extensions/Makefile.am @@ -79,6 +79,7 @@ extension_LTLIBRARIES = \ spy_trace_notice.la \ drain.la \ identify_msg.la \ + cap_realhost.la \ example_module.la if HAVE_HYPERSCAN diff --git a/extensions/cap_realhost.c b/extensions/cap_realhost.c new file mode 100644 index 00000000..1ddf96b0 --- /dev/null +++ b/extensions/cap_realhost.c @@ -0,0 +1,85 @@ +/* + * Copyright (C) 2020 Ed Kellett + * + * This program 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; either version 2 of the License, or + * (at your option) any later version. + * + * 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, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 + * USA + */ + +#include +#include +#include +#include +#include +#include +#include + +static char cap_realhost_desc[] = "Provides the solanum.chat/realhost oper-only capability"; + +static bool cap_realhost_visible(struct Client *); +static void cap_realhost_outbound_msgbuf(void *); +static void cap_realhost_umode_changed(void *); + +static unsigned CLICAP_REALHOST; + +static struct ClientCapability cap_realhost = { + .visible = cap_realhost_visible, +}; + +mapi_cap_list_av2 cap_realhost_caps[] = { + { MAPI_CAP_CLIENT, "solanum.chat/realhost", &cap_realhost, &CLICAP_REALHOST }, + { 0, NULL, NULL, NULL }, +}; + +mapi_hfn_list_av1 cap_realhost_hfnlist[] = { + { "outbound_msgbuf", cap_realhost_outbound_msgbuf, HOOK_NORMAL }, + { "umode_changed", cap_realhost_umode_changed, HOOK_MONITOR }, + { NULL, NULL, 0 }, +}; + +static bool +cap_realhost_visible(struct Client *client) +{ + return HasPrivilege(client, "cap:realhost"); +} + +static void +cap_realhost_outbound_msgbuf(void *data_) +{ + hook_data *data = data_; + struct MsgBuf *msgbuf = data->arg1; + + if (data->client == NULL) + return; + + if (!IsIPSpoof(data->client) && !EmptyString(data->client->sockhost) && strcmp(data->client->sockhost, "0")) + msgbuf_append_tag(msgbuf, "solanum.chat/ip", data->client->sockhost, CLICAP_REALHOST); + + if (!EmptyString(data->client->orighost)) + msgbuf_append_tag(msgbuf, "solanum.chat/realhost", data->client->orighost, CLICAP_REALHOST); +} + +static void +cap_realhost_umode_changed(void *data_) +{ + hook_data_umode_changed *data = data_; + + if (!MyClient(data->client)) + return; + + if (!cap_realhost_visible(data->client)) + data->client->localClient->caps &= ~CLICAP_REALHOST; +} + +DECLARE_MODULE_AV2(cap_realhost, NULL, NULL, NULL, NULL, cap_realhost_hfnlist, cap_realhost_caps, NULL, cap_realhost_desc);