mirror of
https://github.com/matrix-construct/construct
synced 2024-11-29 10:12:39 +01:00
61b517ca3c
* To benefit from the precompiled-header (PCH) it MUST provide "the first C token." Advantages: Never worry about the include stack again. Remember, this means one less thing for random module developers, community people learning C++, and new developers to deal with. It should reduce the learning curve and barrier for participation. Disadvantages: Makes overall compilation a bit slower, especially without any additional work to improve it again. There are several opportunities, places where the PCH is probably being ignored, etc that can be addressed.
238 lines
5.1 KiB
C++
238 lines
5.1 KiB
C++
/*
|
|
* charybdis - an advanced ircd.
|
|
* Copyright (c) 2016 William Pitcock <nenolod@dereferenced.org>.
|
|
*
|
|
* Permission to use, copy, modify, and/or distribute this software for any
|
|
* purpose with or without fee is hereby granted, provided that the above
|
|
* copyright notice and this permission notice is present in all copies.
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
namespace ircd {
|
|
|
|
/*
|
|
* parse a message into a MsgBuf.
|
|
* returns 0 on success, 1 on error.
|
|
*/
|
|
int
|
|
msgbuf_parse(struct MsgBuf *msgbuf, char *line)
|
|
{
|
|
char *ch;
|
|
char *parv[MAXPARA];
|
|
size_t n_para;
|
|
|
|
/* skip any leading spaces */
|
|
for (ch = line; *ch && *ch == ' '; ch++)
|
|
;
|
|
|
|
msgbuf_init(msgbuf);
|
|
|
|
if (*ch == '@')
|
|
{
|
|
char *t = ch + 1;
|
|
|
|
ch = strchr(ch, ' ');
|
|
if (ch != NULL)
|
|
{
|
|
while (1)
|
|
{
|
|
char *next = strchr(t, ';');
|
|
char *eq = strchr(t, '=');
|
|
|
|
if (next != NULL)
|
|
{
|
|
*next = '\0';
|
|
|
|
if (eq > next)
|
|
eq = NULL;
|
|
}
|
|
|
|
if (eq != NULL)
|
|
*eq++ = '\0';
|
|
|
|
if (*t && *t != ' ')
|
|
msgbuf_append_tag(msgbuf, t, eq, 0);
|
|
else
|
|
break;
|
|
|
|
if (next != NULL)
|
|
t = next + 1;
|
|
else
|
|
break;
|
|
}
|
|
|
|
*ch++ = '\0';
|
|
}
|
|
}
|
|
|
|
/* skip any whitespace between tags and origin */
|
|
for (; *ch && *ch == ' '; ch++)
|
|
;
|
|
|
|
if (*ch == ':')
|
|
{
|
|
ch++;
|
|
msgbuf->origin = ch;
|
|
|
|
char *end = strchr(ch, ' ');
|
|
if (end == NULL)
|
|
return 1;
|
|
|
|
*end = '\0';
|
|
|
|
for (ch = end + 1; *ch && *ch == ' '; ch++)
|
|
;
|
|
}
|
|
|
|
if (*ch == '\0')
|
|
return 1;
|
|
|
|
n_para = rb_string_to_array(ch, parv, MAXPARA);
|
|
if (n_para == 0)
|
|
return 1;
|
|
|
|
msgbuf->cmd = parv[0];
|
|
for (size_t i = 0; i < n_para; i++)
|
|
msgbuf_append_para(msgbuf, parv[i]);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
msgbuf_has_matching_tags(struct MsgBuf *msgbuf, unsigned int capmask)
|
|
{
|
|
for (size_t i = 0; i < msgbuf->n_tags; i++)
|
|
{
|
|
if ((msgbuf->tags[i].capmask & capmask) != 0)
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void
|
|
msgbuf_unparse_tags(char *buf, size_t buflen, struct MsgBuf *msgbuf, unsigned int capmask)
|
|
{
|
|
if (!msgbuf_has_matching_tags(msgbuf, capmask))
|
|
return;
|
|
|
|
*buf = '@';
|
|
|
|
for (size_t i = 0; i < msgbuf->n_tags; i++)
|
|
{
|
|
if ((msgbuf->tags[i].capmask & capmask) == 0)
|
|
continue;
|
|
|
|
if (i != 0)
|
|
rb_strlcat(buf, ";", buflen);
|
|
|
|
rb_strlcat(buf, msgbuf->tags[i].key, buflen);
|
|
|
|
/* XXX properly handle escaping */
|
|
if (msgbuf->tags[i].value)
|
|
{
|
|
rb_strlcat(buf, "=", buflen);
|
|
rb_strlcat(buf, msgbuf->tags[i].value, buflen);
|
|
}
|
|
}
|
|
|
|
rb_strlcat(buf, " ", buflen);
|
|
}
|
|
|
|
void
|
|
msgbuf_unparse_prefix(char *buf, size_t buflen, struct MsgBuf *msgbuf, unsigned int capmask)
|
|
{
|
|
memset(buf, 0, buflen);
|
|
|
|
if (msgbuf->n_tags > 0)
|
|
msgbuf_unparse_tags(buf, buflen, msgbuf, capmask);
|
|
|
|
rb_snprintf_append(buf, buflen, ":%s ", msgbuf->origin != NULL ? msgbuf->origin : me.name);
|
|
|
|
if (msgbuf->cmd != NULL)
|
|
rb_snprintf_append(buf, buflen, "%s ", msgbuf->cmd);
|
|
|
|
if (msgbuf->target != NULL)
|
|
rb_snprintf_append(buf, buflen, "%s ", msgbuf->target);
|
|
}
|
|
|
|
/*
|
|
* unparse a pure MsgBuf into a buffer.
|
|
* if origin is NULL, me.name will be used.
|
|
* cmd may not be NULL.
|
|
* returns 0 on success, 1 on error.
|
|
*/
|
|
int
|
|
msgbuf_unparse(char *buf, size_t buflen, struct MsgBuf *msgbuf, unsigned int capmask)
|
|
{
|
|
msgbuf_unparse_prefix(buf, buflen, msgbuf, capmask);
|
|
|
|
for (size_t i = msgbuf->cmd != NULL ? 0 : 1; i < msgbuf->n_para; i++)
|
|
{
|
|
if (i == (msgbuf->n_para - 1))
|
|
{
|
|
if (strchr(msgbuf->para[i], ' ') != NULL)
|
|
rb_snprintf_append(buf, buflen, ":%s", msgbuf->para[i]);
|
|
else
|
|
rb_strlcat(buf, msgbuf->para[i], buflen);
|
|
}
|
|
else
|
|
rb_strlcat(buf, msgbuf->para[i], buflen);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* unparse a MsgBuf stem + format string into a buffer
|
|
* if origin is NULL, me.name will be used.
|
|
* cmd may not be NULL.
|
|
* returns 0 on success, 1 on error.
|
|
*/
|
|
int
|
|
msgbuf_vunparse_fmt(char *buf, size_t buflen, struct MsgBuf *head, unsigned int capmask, const char *fmt, va_list va)
|
|
{
|
|
char *ws;
|
|
size_t prefixlen;
|
|
|
|
msgbuf_unparse_prefix(buf, buflen, head, capmask);
|
|
prefixlen = strlen(buf);
|
|
|
|
ws = buf + prefixlen;
|
|
vsnprintf(ws, buflen - prefixlen, fmt, va);
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* unparse a MsgBuf stem + format string into a buffer (with va_list handling)
|
|
* if origin is NULL, me.name will be used.
|
|
* cmd may not be NULL.
|
|
* returns 0 on success, 1 on error.
|
|
*/
|
|
int
|
|
msgbuf_unparse_fmt(char *buf, size_t buflen, struct MsgBuf *head, unsigned int capmask, const char *fmt, ...)
|
|
{
|
|
va_list va;
|
|
int res;
|
|
|
|
va_start(va, fmt);
|
|
res = msgbuf_vunparse_fmt(buf, buflen, head, capmask, fmt, va);
|
|
va_end(va);
|
|
|
|
return res;
|
|
}
|
|
|
|
|
|
} // namespace ircd
|