2 * Copyright (C) 2017 Andy Spencer <andy753421@gmail.com>
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 3 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
67 xmpp_channel_t system;
90 char msg_jid[JID_LEN];
91 char msg_usr[JID_LEN];
92 char msg_srv[JID_LEN];
93 char msg_res[JID_LEN];
95 xmpp_channel_t *msg_chan;
98 /* Helper functions */
99 static void srv_notice(xmpp_server_t *srv, const char *fmt, ...)
101 static char buf[1024];
105 vsnprintf(buf, sizeof(buf), fmt, ap);
108 chat_recv(&srv->system.channel, NULL, buf);
111 static void chan_notice(xmpp_channel_t *chan, const char *fmt, ...)
113 static char buf[1024];
117 vsnprintf(buf, sizeof(buf), fmt, ap);
120 chat_recv(&chan->channel, NULL, buf);
123 static void split_jid(const char *jid, char *usr, char *srv, char *res)
128 if (usr) usr[0] = '\0';
129 if (srv) srv[0] = '\0';
130 if (res) res[0] = '\0';
132 for (int i = 0; jid && jid[i]; i++) {
143 if (ptr && (pos+1) < JID_LEN) {
149 //debug("JID: '%s' usr=[%s] srv=[%s] res=[%s]",
150 // jid, usr, srv, res);
153 static xmpp_channel_t *find_dest(xmpp_server_t *srv,
154 const char *jid, int is_muc)
156 static char jid_usr[JID_LEN];
157 static char jid_srv[JID_LEN];
158 static char dest[JID_LEN];
159 xmpp_channel_t *chan;
161 split_jid(jid, jid_usr, jid_srv, NULL);
162 snprintf(dest, JID_LEN, "%s@%s", jid_usr,
163 jid_srv[0] ? jid_srv :
164 is_muc ? srv->muc : srv->host);
166 /* Server channels */
167 if (match(jid, srv->host))
170 /* Find existing channels */
171 for (channel_t *cur = channels; cur; cur = cur->next) {
172 if (cur->server != &srv->server)
174 chan = (xmpp_channel_t *)cur;
175 if (match(chan->dest, dest))
179 /* Create a new channel */
180 chan = (xmpp_channel_t *)add_channel(jid_usr, &srv->server);
181 strncpy(chan->dest, dest, JID_LEN);
185 static const char *find_attr(const char **attrs, const char *name)
187 for (int i = 0; attrs[i] && attrs[i+1]; i += 2)
188 if (match(attrs[i+0], name))
193 /* Callback functions */
194 static void xmpp_run(xmpp_server_t *srv,
195 const char *start, const char **attrs,
196 const char *end, const char *data);
198 static void on_start(void *_srv, const char *tag, const char **attrs)
200 xmpp_server_t *srv = _srv;
201 xmpp_run(srv, tag, attrs, NULL, reset(&srv->buf));
204 static void on_data(void *_srv, const char *data, int len)
206 xmpp_server_t *srv = _srv;
207 append(&srv->buf, data, len);
210 static void on_end(void *_srv, const char *tag)
212 xmpp_server_t *srv = _srv;
213 xmpp_run(srv, NULL, NULL, tag, reset(&srv->buf));
216 static void on_send(void *_srv)
218 xmpp_server_t *srv = _srv;
219 xmpp_run(srv, NULL, NULL, NULL, NULL);
222 static void on_recv(void *_srv, char *buf, int len)
224 xmpp_server_t *srv = _srv;
226 XML_Parse(srv->expat, buf, len, 0);
227 xmpp_run(srv, NULL, NULL, NULL, NULL);
230 static void on_err(void *_srv, int errno)
232 xmpp_server_t *srv = _srv;
233 xmpp_run(srv, NULL, NULL, NULL, NULL);
236 /* XMPP State machine */
237 static void xmpp_run(xmpp_server_t *srv,
238 const char *start, const char **attrs,
239 const char *end, const char *data)
241 xmpp_channel_t *chan = NULL;
245 debug("%*s \"%s\"", srv->indent*4, "", data);
247 debug("%*s<%s>", srv->indent*4, "", start);
248 for (int i = 0; attrs[i] && attrs[i+1]; i += 2) {
249 debug("%*s%s=\"%s\"%s",
251 attrs[i+0], attrs[i+1],
252 attrs[i+2] ? "" : ">");
260 /* Connection Handling */
261 if (srv->state == XMPP_CONNECT && !start && !end) {
262 srv->net.send = on_send;
263 srv->net.recv = on_recv;
264 srv->net.err = on_err;
266 net_open(&srv->net, srv->host, srv->port);
268 if (!(srv->expat = XML_ParserCreate(NULL)))
269 error("Error creating XML parser");
270 XML_SetUserData(srv->expat, srv);
271 XML_SetStartElementHandler(srv->expat, on_start);
272 XML_SetEndElementHandler(srv->expat, on_end);
273 XML_SetCharacterDataHandler(srv->expat, on_data);
275 debug("xmpp: connect -> stream");
276 srv->state = XMPP_SEND_STREAM;
278 if (srv->state == XMPP_ENCRYPT && !start && !end) {
279 net_encrypt(&srv->net);
281 if (!(XML_ParserReset(srv->expat, NULL)))
282 error("Error resetting XML parser");
283 XML_SetUserData(srv->expat, srv);
284 XML_SetStartElementHandler(srv->expat, on_start);
285 XML_SetEndElementHandler(srv->expat, on_end);
286 XML_SetCharacterDataHandler(srv->expat, on_data);
288 debug("xmpp: encrypt -> stream");
289 srv->state = XMPP_SEND_STREAM;
291 if (srv->state == XMPP_RESTART && !start && !end) {
292 if (!(XML_ParserReset(srv->expat, NULL)))
293 error("Error resetting XML parser");
294 XML_SetUserData(srv->expat, srv);
295 XML_SetStartElementHandler(srv->expat, on_start);
296 XML_SetEndElementHandler(srv->expat, on_end);
297 XML_SetCharacterDataHandler(srv->expat, on_data);
299 debug("xmpp: restart -> stream");
300 srv->state = XMPP_SEND_STREAM;
304 if (srv->state == XMPP_SEND_STREAM) {
305 if (net_print(&srv->net,
306 "<?xml version='1.0'?>"
312 " xmlns='jabber:client'"
313 " xmlns:stream='http://etherx.jabber.org/streams'>",
314 srv->jid, srv->host)) {
315 debug("xmpp: stream -> features");
316 srv->state = XMPP_RECV_FEATURES;
319 if (srv->state == XMPP_RECV_FEATURES) {
320 if (match(start, "starttls")) {
321 debug("xmpp: features -> starttls");
322 srv->state = XMPP_SEND_STARTTLS;
324 if (match(start, "mechanisms")) {
325 debug("xmpp: features -> auth");
326 srv->state = XMPP_SEND_AUTH;
328 if (match(start, "bind")) {
329 debug("xmpp: features -> bind");
330 srv->state = XMPP_SEND_BIND;
335 if (srv->state == XMPP_SEND_STARTTLS) {
336 if (net_print(&srv->net,
337 "<starttls xmlns='urn:ietf:params:xml:ns:xmpp-tls'/>")) {
338 debug("xmpp: startls -> proceed");
339 srv->state = XMPP_RECV_PROCEED;
342 if (srv->state == XMPP_RECV_PROCEED) {
343 if (match(start, "proceed")) {
344 debug("xmpp: proceed -> encrypt");
345 srv->state = XMPP_ENCRYPT;
350 if (srv->state == XMPP_SEND_AUTH) {
351 static char plain[AUTH_LEN];
352 static char coded[AUTH_LEN];
354 len = snprintf(plain, AUTH_LEN, "%s%c%s%c%s",
355 srv->user, '\0', srv->user, '\0', srv->pass);
356 len = base64(plain, len, coded, AUTH_LEN);
357 if (net_print(&srv->net,
359 " xmlns='urn:ietf:params:xml:ns:xmpp-sasl'"
360 " mechanism='PLAIN'>%.*s</auth>",
362 debug("xmpp: auth -> success");
363 srv->state = XMPP_RECV_SUCCESS;
366 if (srv->state == XMPP_RECV_SUCCESS) {
367 if (match(start, "success")) {
368 debug("xmpp: success -> restart");
369 srv->state = XMPP_RESTART;
374 if (srv->state == XMPP_SEND_BIND) {
375 const char *resource = srv->jid;
376 while (*resource && *resource != '/')
378 while (*resource && *resource == '/')
380 if (net_print(&srv->net,
381 "<iq id='bind' type='set'>"
382 "<bind xmlns='urn:ietf:params:xml:ns:xmpp-bind'>"
383 "<resource>%s</resource>"
387 debug("xmpp: bind -> jid");
388 srv->state = XMPP_RECV_JID;
391 if (srv->state == XMPP_RECV_JID) {
392 if (match(start, "jid")) {
393 debug("xmpp: jid -> presence");
394 srv->state = XMPP_SEND_PRESENCE;
397 if (srv->state == XMPP_SEND_PRESENCE) {
398 for (channel_t *cur = channels; cur; cur = cur->next) {
399 xmpp_channel_t *chan = (xmpp_channel_t*)cur;
400 if (cur->server != &srv->server || !chan->join)
403 "<presence id='join' from='%s' to='%s/%s'>"
404 "<x xmlns='http://jabber.org/protocol/muc'/>"
406 srv->jid, chan->dest, srv->nick);
408 srv->state = XMPP_READY;
412 if (srv->state == XMPP_READY) {
413 srv->state = match(start, "iq") ? XMPP_IN_IQ :
414 match(start, "message") ? XMPP_IN_MESSAGE :
415 match(start, "presence") ? XMPP_IN_PRESENCE :
417 if (srv->state != XMPP_READY) {
418 strncpy(srv->msg_jid, find_attr(attrs, "from"), JID_LEN);
419 split_jid(srv->msg_jid, srv->msg_usr,
420 srv->msg_srv, srv->msg_res);
422 if (match(srv->msg_srv, srv->muc)) {
423 srv->msg_from = srv->msg_res[0] ? srv->msg_res : NULL;
424 srv->msg_chan = find_dest(srv, srv->msg_jid, 1);
426 srv->msg_from = srv->msg_usr[0] ? srv->msg_usr : NULL;
427 srv->msg_chan = find_dest(srv, srv->msg_jid, 0);
430 debug("xmpp: %s -- jid=[%s] from=[%s] chan=[%s]",
432 srv->msg_jid, srv->msg_from,
433 srv->msg_chan->channel.name);
436 if (srv->state == XMPP_IN_IQ ||
437 srv->state == XMPP_IN_MESSAGE ||
438 srv->state == XMPP_IN_PRESENCE) {
439 if (srv->msg_chan && srv->msg_chan != &srv->system)
440 chan = srv->msg_chan;
444 if (srv->state == XMPP_IN_IQ) {
445 if (match(start, "item") && chan) {
446 static char res[JID_LEN];
447 split_jid(find_attr(attrs, "jid"),
449 chan_notice(chan, "user: %s", res);
451 if (match(start, "item") && !chan) {
452 srv_notice(srv, "item: [%s] %s",
453 find_attr(attrs, "jid"),
454 find_attr(attrs, "name"));
456 if (match(start, "identity")) {
457 srv_notice(srv, "identity: %s",
458 find_attr(attrs, "name"));
460 if (match(start, "feature")) {
461 srv_notice(srv, "feature: %s",
462 find_attr(attrs, "var"));
464 if (match(start, "field")) {
465 debug("xmpp: %s -- type=[%s] label=[%s]", end,
466 find_attr(attrs, "type"),
467 find_attr(attrs, "label"));
468 if (!find_attr(attrs, "label"))
470 chan_notice(chan, "%-36s -- %s (%s)",
471 find_attr(attrs, "var"),
472 find_attr(attrs, "label"),
473 find_attr(attrs, "type"));
475 if (match(end, "title")) {
476 debug("xmpp: title -- jid=[%s]",
478 chan_notice(chan, "Title: %s", data);
480 if (match(end, "instructions")) {
481 debug("xmpp: instructions -- jid=[%s]",
483 chan_notice(chan, "%s", data);
488 if (srv->state == XMPP_IN_MESSAGE) {
489 if (match(start, "delay")) {
490 const char *ts = find_attr(attrs, "stamp");
493 strptime(ts, "%Y-%m-%dT%H:%M:%S%z", &tm);
494 srv->stamp = mktime(&tm);
497 if (match(end, "body")) {
498 strset(&srv->body, data);
500 if (match(end, "message")) {
501 debug("xmpp: body (%s) -- chan=[%s] jid=[%s] from=[%s]",
502 srv->msg_from == srv->msg_usr ? "user" : "chat" ,
503 srv->msg_chan->channel.name,
504 srv->msg_jid, srv->msg_from);
506 chat_recv(&chan->channel, srv->msg_from, srv->body);
507 message_t *msg = &messages[history-1];
508 msg->when = srv->stamp ?: msg->when;
511 strset(&srv->body, NULL);
516 if (srv->state == XMPP_IN_PRESENCE) {
517 static char alias[JID_LEN];
519 if (match(start, "item")) {
520 if ((jid = find_attr(attrs, "jid")))
521 strncpy(alias, jid, JID_LEN);
523 if (match(end, "presence") && chan) {
524 if (alias[0] && !srv->quiet)
525 chan_notice(chan, "%s (%s) entered room.",
526 srv->msg_from, alias);
527 else if (!srv->quiet)
528 chan_notice(chan, "%s entered room.",
535 if (srv->state == XMPP_IN_IQ ||
536 srv->state == XMPP_IN_MESSAGE ||
537 srv->state == XMPP_IN_PRESENCE) {
538 if (match(end, "iq") ||
539 match(end, "message") ||
540 match(end, "presence")) {
541 srv->state = XMPP_READY;
543 srv->msg_jid[0] = '\0';
544 srv->msg_usr[0] = '\0';
545 srv->msg_srv[0] = '\0';
546 srv->msg_res[0] = '\0';
547 srv->msg_from = NULL;
548 srv->msg_chan = NULL;
552 if (match(start, "stream:error"))
554 if (match(end, "stream:error"))
557 if (match(end, "text")) {
558 debug("xmpp: error: %s", data);
559 srv_notice(srv, "error: %s", data);
567 for (server_t *cur = servers; cur; cur = cur->next) {
568 if (cur->protocol != XMPP)
571 xmpp_server_t *srv = (xmpp_server_t*)cur;
572 srv->system.channel.server = &srv->server;
573 srv->system.channel.name = srv->server.name;
574 srv_notice(srv, "XMPP Server: %s", srv->server.name);
579 error("jid is required");
581 xmpp_run(srv, NULL, NULL, NULL, NULL);
583 for (channel_t *cur = channels; cur; cur = cur->next) {
584 if (cur->server->protocol != XMPP)
587 xmpp_channel_t *chan = (xmpp_channel_t*)cur;
588 xmpp_server_t *srv = (xmpp_server_t*)cur->server;
590 chan->room = strcopy(cur->name);
591 snprintf(chan->dest, JID_LEN, "%s@%s",
592 chan->room, srv->muc);
596 server_t *xmpp_server(void)
598 return new0(xmpp_server_t);
601 channel_t *xmpp_channel(void)
603 return new0(xmpp_channel_t);
606 void xmpp_config(server_t *server, channel_t *channel,
607 const char *group, const char *name,
608 const char *key, const char *value)
610 xmpp_server_t *srv = (xmpp_server_t*)server;
611 xmpp_channel_t *chan = (xmpp_channel_t*)channel;
614 if (match(key, "connect"))
615 srv->connect = get_bool(value);
616 else if (match(key, "host"))
617 srv->host = get_string(value);
618 else if (match(key, "port"))
619 srv->port = get_number(value);
620 else if (match(key, "muc"))
621 srv->muc = get_string(value);
622 else if (match(key, "nick"))
623 srv->nick = get_string(value);
624 else if (match(key, "jid"))
625 srv->jid = get_string(value);
626 else if (match(key, "user"))
627 srv->user = get_string(value);
628 else if (match(key, "pass"))
629 srv->pass = get_string(value);
630 else if (match(key, "quiet"))
631 srv->quiet = get_bool(value);
634 if (match(key, "room"))
635 chan->room = get_string(value);
636 else if (match(key, "join"))
637 chan->join = get_bool(value);
641 void xmpp_send(channel_t *channel, const char *text)
643 xmpp_channel_t *chan = (xmpp_channel_t*)channel;
644 xmpp_server_t *srv = (xmpp_server_t*)channel->server;
647 /* Handle commands */
648 if (text[0] == '/') {
649 if (prefix(text, "/items", &arg)) {
651 "<iq id='items' type='get' from='%s' to='%s'>"
652 "<query xmlns='http://jabber.org/protocol/disco#items'/>"
654 srv->jid, arg ?: srv->host);
656 else if (prefix(text, "/info", &arg)) {
658 "<iq id='info' type='get' from='%s' to='%s'>"
659 "<query xmlns='http://jabber.org/protocol/disco#info'/>"
661 srv->jid, arg ?: srv->host);
663 else if (prefix(text, "/names", &arg)) {
665 chan = find_dest(srv, arg, 1);
666 if (chan == &srv->system) {
667 chan_notice(chan, "Cannot get names from server");
671 "<iq id='list' type='get' from='%s' to='%s'>"
672 "<query xmlns='http://jabber.org/protocol/disco#items'/>"
674 srv->jid, chan->dest);
676 else if (prefix(text, "/join", &arg)) {
678 chan_notice(chan, "usage: /join <channel>");
682 "<presence id='join' from='%s' to='%s/%s'>"
683 "<x xmlns='http://jabber.org/protocol/muc'/>"
685 srv->jid, chan->dest, srv->nick);
686 chan = find_dest(srv, arg, 1);
687 chan_notice(chan, "Room: %s", arg);
689 else if (prefix(text, "/config", &arg)) {
691 chan_notice(chan, "Unimplemented: /config <arg>");
695 "<iq id='config' type='get' from='%s' to='%s'>"
696 "<query xmlns='http://jabber.org/protocol/muc#owner'/>"
698 srv->jid, chan->dest);
700 else if (prefix(text, "/query", &arg)) {
702 chan_notice(chan, "usage: /query <user>");
705 chan = find_dest(srv, arg, 0);
706 chan_notice(chan, "User: %s", arg);
709 chan_notice(chan, "Unknown command %s", text);
712 debug("message: [%s]", text);
713 if (chan == &srv->system) {
714 chan_notice(chan, "Cannot send to server");
716 else if (!chan->dest) {
717 chan_notice(chan, "No destination for message");
719 else if (chan->room) {
721 "<message id='chat' from='%s' to='%s' type='groupchat'>"
724 srv->jid, chan->dest, text);
727 "<message id='chat' from='%s' to='%s'>"
730 srv->jid, chan->dest, text);
731 chat_recv(channel, srv->nick, text);