Something went wrong. Try again.
[READ-ONLY] Mirror of https://github.com/VibeDevelopers/Comet.
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432/* modules/m_cap.c * * Copyright (C) 2005 Lee Hardy <lee@leeh.co.uk> * Copyright (C) 2005 ircd-ratbox development team * Copyright (C) 2016 Ariadne Conill <ariadne@dereferenced.org> * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are * met: * * 1.Redistributions of source code must retain the above copyright notice, * this list of conditions and the following disclaimer. * 2.Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * 3.The name of the author may not be used to endorse or promote products * derived from this software without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE * POSSIBILITY OF SUCH DAMAGE. */
#include "stdinc.h"#include "class.h"#include "client.h"#include "match.h"#include "ircd.h"#include "numeric.h"#include "msg.h"#include "parse.h"#include "modules.h"#include "s_serv.h"#include "s_user.h"#include "send.h"#include "s_conf.h"#include "hash.h"#include "response.h"
static const char cap_desc[] = "Provides the commands used for client capability negotiation";
typedef int (*bqcmp)(const void *, const void *);
static void m_cap(struct MsgBuf *, struct Client *, struct Client *, int, const char **);
struct Message cap_msgtab = { "CAP", 0, 0, 0, 0, {{m_cap, 2}, {m_cap, 2}, mg_ignore, mg_ignore, mg_ignore, {m_cap, 2}}};
mapi_clist_av1 cap_clist[] = { &cap_msgtab, NULL };
DECLARE_MODULE_AV2(cap, NULL, NULL, cap_clist, NULL, NULL, NULL, NULL, cap_desc);
#define IsCapableEntry(c, e) IsClientCapable(c, 1ull << (e)->value)#define HasCapabilityFlag(c, f) (c->ownerdata != NULL && (((struct ClientCapability *)c->ownerdata)->flags & (f)) == f)
static inline intclicap_visible(struct Client *client_p, const struct CapabilityEntry *cap){ struct ClientCapability *clicap;
/* orphaned caps shouldn't be visible */ if (cap->flags & CAP_ORPHANED) return 0;
if (cap->ownerdata == NULL) return 1;
clicap = cap->ownerdata; if (clicap->visible == NULL) return (clicap->flags & CLICAP_FLAGS_INVISIBLE) == 0;
return clicap->visible(client_p);}
/* clicap_find() * Used iteratively over a buffer, extracts individual cap tokens. * * Inputs: buffer to start iterating over (NULL to iterate over existing buf) * int pointer to whether the cap token is negated * int pointer to whether we finish with success * Ouputs: Cap entry if found, NULL otherwise. */static struct CapabilityEntry *clicap_find(const char *data, int *negate, int *finished){ static char buf[BUFSIZE]; static char *p; struct CapabilityEntry *cap; char *s;
*negate = 0;
if(data) { rb_strlcpy(buf, data, sizeof(buf)); p = buf; }
if(*finished) return NULL;
/* skip any whitespace */ while(*p && IsSpace(*p)) p++;
if(EmptyString(p)) { *finished = 1; return NULL; }
if(*p == '-') { *negate = 1; p++;
/* someone sent a '-' without a parameter.. */ if(*p == '\0') return NULL; }
if((s = strchr(p, ' '))) *s++ = '\0';
if((cap = capability_find(cli_capindex, p)) != NULL) { if(s) p = s; else *finished = 1; }
return cap;}
/* clicap_generate() * Generates a list of capabilities. * * Inputs: client to send to, subcmd to send, * flags to match against: 0 to do none, -1 if client has no flags * Outputs: None */static voidclicap_generate(struct Client *source_p, const char *subcmd, int flags){ const char *str_cont = "* :"; const char *str_final = ":"; struct CapabilityEntry *entry; rb_dictionary_iter iter; bool multiline_ret; enum multiline_item_result multiline_item_ret;
multiline_ret = send_multiline_init(source_p, " ", ":%s CAP %s %s %s", me.name, EmptyString(source_p->name) ? "*" : source_p->name, subcmd, (source_p->flags & FLAGS_CLICAP_DATA) ? str_cont : str_final);
/* shortcut, nothing to do */ if (flags == -1 || !multiline_ret) { /* Currently safe to call if send_multiline_init failed. */ send_multiline_reset(); sendto_one(source_p, ":%s CAP %s %s %s", me.name, EmptyString(source_p->name) ? "*" : source_p->name, subcmd, str_final); return; }
for (int pass = 0; pass < 2; pass++) RB_DICTIONARY_FOREACH(entry, &iter, cli_capindex->cap_dict) { struct ClientCapability *clicap = entry->ownerdata; const char *data = NULL;
if (pass == 0 && !HasCapabilityFlag(entry, CLICAP_FLAGS_PRIORITY)) continue; else if (pass == 1 && HasCapabilityFlag(entry, CLICAP_FLAGS_PRIORITY)) continue;
if (!IsCapableEntry(source_p, entry) && ConfigFileEntry.hidden_caps != NULL) { size_t i; for (i = 0; ConfigFileEntry.hidden_caps[i] != NULL; i++) { if (!rb_strcasecmp(entry->cap, ConfigFileEntry.hidden_caps[i])) break; } if (ConfigFileEntry.hidden_caps[i] != NULL) continue; }
if (flags && !IsCapableEntry(source_p, entry)) continue;
if (!clicap_visible(source_p, entry)) continue;
if (source_p->flags & FLAGS_CLICAP_DATA) { if (!flags && clicap != NULL && clicap->data != NULL) data = clicap->data(source_p);
multiline_item_ret = send_multiline_item(source_p, "%s%s%s", entry->cap, data != NULL ? "=" : "", data != NULL ? data : "");
if (multiline_item_ret == MULTILINE_FAILURE) return; } else { multiline_item_ret = send_multiline_item(source_p, "%s", entry->cap);
if (multiline_item_ret != MULTILINE_SUCCESS) { send_multiline_reset(); return; } } }
send_multiline_fini(source_p, ":%s CAP %s %s %s", me.name, EmptyString(source_p->name) ? "*" : source_p->name, subcmd, str_final);}
static voidcap_ack(struct Client *source_p, const char *arg){ struct CapabilityEntry *cap; uint64_t capadd = 0, capdel = 0; int finished = 0, negate;
if(EmptyString(arg)) return;
for(cap = clicap_find(arg, &negate, &finished); cap; cap = clicap_find(NULL, &negate, &finished)) { /* sent an ACK for something they havent REQd */ if(!IsCapableEntry(source_p, cap)) continue;
if(negate) { /* dont let them ack something sticky off */ if(HasCapabilityFlag(cap, CLICAP_FLAGS_STICKY)) continue;
capdel |= (1 << cap->value); } else capadd |= (1 << cap->value); }
SetClientCap(source_p, capadd); ClearClientCap(source_p, capdel);}
static voidcap_end(struct Client *source_p, const char *arg){ if(IsRegistered(source_p)) return;
source_p->flags &= ~FLAGS_CLICAP;
if(source_p->name[0] && source_p->flags & FLAGS_SENTUSER) { register_local_user(source_p, source_p); }}
static voidcap_list(struct Client *source_p, const char *arg){ /* list of what theyre currently using */ begin_local_response_batch(); clicap_generate(source_p, "LIST", source_p->localClient->client_caps ? source_p->localClient->client_caps : -1);}
static voidcap_ls(struct Client *source_p, const char *arg){ int caps_version = 301;
if(!IsRegistered(source_p)) source_p->flags |= FLAGS_CLICAP;
if (!EmptyString(arg)) { caps_version = atoi(arg); }
if (caps_version >= 302) { source_p->flags |= FLAGS_CLICAP_DATA; SetClientCap(source_p, CLICAP_CAP_NOTIFY); }
/* list of what we support */ begin_local_response_batch(); clicap_generate(source_p, "LS", 0);}
static voidcap_req(struct Client *source_p, const char *arg){ char ack_buf[DATALEN+1]; struct CapabilityEntry *cap; uint64_t capadd = 0, capdel = 0; int finished = 0, negate; int ret; hook_data_cap_change hdata;
if(!IsRegistered(source_p)) source_p->flags |= FLAGS_CLICAP;
if(EmptyString(arg)) return;
begin_local_response_batch();
ret = snprintf(ack_buf, sizeof ack_buf, ":%s CAP %s ACK :%s", me.name, EmptyString(source_p->name)? "*" : source_p->name, arg);
if (ret < 0 || ret > DATALEN) { sendto_one(source_p, ":%s CAP %s NAK :%s", me.name, EmptyString(source_p->name) ? "*" : source_p->name, arg); return; }
for(cap = clicap_find(arg, &negate, &finished); cap; cap = clicap_find(NULL, &negate, &finished)) { if(negate) { if(HasCapabilityFlag(cap, CLICAP_FLAGS_STICKY)) { finished = 0; break; }
capdel |= (1ull << cap->value); } else { if (!clicap_visible(source_p, cap)) { finished = 0; break; }
capadd |= (1ull << cap->value); }
}
if(!finished) { sendto_one(source_p, ":%s CAP %s NAK :%s", me.name, EmptyString(source_p->name) ? "*" : source_p->name, arg); return; }
sendto_one(source_p, "%s", ack_buf);
hdata.client = source_p; hdata.oldcaps = source_p->localClient->client_caps; hdata.add = capadd; hdata.del = capdel;
SetClientCap(source_p, capadd); ClearClientCap(source_p, capdel);
call_hook(h_cap_change, &hdata);}
static struct clicap_cmd{ const char *cmd; void (*func)(struct Client *source_p, const char *arg);} clicap_cmdlist[] = { /* This list *MUST* be in alphabetical order */ { "ACK", cap_ack }, { "END", cap_end }, { "LIST", cap_list }, { "LS", cap_ls }, { "REQ", cap_req },};
static intclicap_cmd_search(const char *command, struct clicap_cmd *entry){ return irccmp(command, entry->cmd);}
static voidm_cap(struct MsgBuf *msgbuf_p, struct Client *client_p, struct Client *source_p, int parc, const char *parv[]){ struct clicap_cmd *cmd;
if (strlen(client_p->id) == 3 || (source_p->preClient && !EmptyString(source_p->preClient->id))) { exit_client(client_p, client_p, client_p, "Mixing client and server protocol"); return; }
if(!(cmd = bsearch(parv[1], clicap_cmdlist, sizeof(clicap_cmdlist) / sizeof(struct clicap_cmd), sizeof(struct clicap_cmd), (bqcmp) clicap_cmd_search))) { sendto_one(source_p, form_str(ERR_INVALIDCAPCMD), me.name, EmptyString(source_p->name) ? "*" : source_p->name, parv[1]); return; }
(cmd->func)(source_p, parv[2]);}