xmpp_component.cpp 17.2 KB
Newer Older
1
#include <utils/make_unique.hpp>
louiz’'s avatar
louiz’ committed
2
#include <logger/logger.hpp>
3

4
#include <xmpp/xmpp_component.hpp>
5
#include <xmpp/jid.hpp>
6

7 8
#include <utils/sha1.hpp>

louiz’'s avatar
louiz’ committed
9
#include <stdexcept>
10 11
#include <iostream>

12
#include <stdio.h>
13

14 15 16 17
#define STREAM_NS        "http://etherx.jabber.org/streams"
#define COMPONENT_NS     "jabber:component:accept"
#define MUC_NS           "http://jabber.org/protocol/muc"
#define MUC_USER_NS      MUC_NS"#user"
18
#define MUC_ADMIN_NS     MUC_NS"#admin"
19 20 21
#define DISCO_NS         "http://jabber.org/protocol/disco"
#define DISCO_ITEMS_NS   DISCO_NS"#items"
#define DISCO_INFO_NS    DISCO_NS"#info"
22
#define XHTMLIM_NS       "http://jabber.org/protocol/xhtml-im"
23
#define STANZA_NS        "urn:ietf:params:xml:ns:xmpp-stanzas"
louiz’'s avatar
louiz’ committed
24
#define STREAMS_NS       "urn:ietf:params:xml:ns:xmpp-streams"
25

26
XmppComponent::XmppComponent(const std::string& hostname, const std::string& secret):
27 28
  ever_auth(false),
  last_auth(false),
29 30
  served_hostname(hostname),
  secret(secret),
louiz’'s avatar
louiz’ committed
31 32
  authenticated(false),
  doc_open(false)
33 34 35 36 37 38 39
{
  this->parser.add_stream_open_callback(std::bind(&XmppComponent::on_remote_stream_open, this,
                                                  std::placeholders::_1));
  this->parser.add_stanza_callback(std::bind(&XmppComponent::on_stanza, this,
                                                  std::placeholders::_1));
  this->parser.add_stream_close_callback(std::bind(&XmppComponent::on_remote_stream_close, this,
                                                  std::placeholders::_1));
40
  this->stanza_handlers.emplace(COMPONENT_NS":handshake",
41
                                std::bind(&XmppComponent::handle_handshake, this,std::placeholders::_1));
42
  this->stanza_handlers.emplace(COMPONENT_NS":presence",
43
                                std::bind(&XmppComponent::handle_presence, this,std::placeholders::_1));
44
  this->stanza_handlers.emplace(COMPONENT_NS":message",
louiz’'s avatar
louiz’ committed
45
                                std::bind(&XmppComponent::handle_message, this,std::placeholders::_1));
46 47
  this->stanza_handlers.emplace(COMPONENT_NS":iq",
                                std::bind(&XmppComponent::handle_iq, this,std::placeholders::_1));
louiz’'s avatar
louiz’ committed
48 49
  this->stanza_handlers.emplace(STREAM_NS":error",
                                std::bind(&XmppComponent::handle_error, this,std::placeholders::_1));
50 51 52 53 54 55
}

XmppComponent::~XmppComponent()
{
}

56
void XmppComponent::start()
57
{
58
  this->connect("127.0.0.1", "5347");
59 60
}

louiz’'s avatar
louiz’ committed
61 62 63 64 65
bool XmppComponent::is_document_open() const
{
  return this->doc_open;
}

66 67
void XmppComponent::send_stanza(const Stanza& stanza)
{
louiz’'s avatar
louiz’ committed
68 69 70
  std::string str = stanza.to_string();
  log_debug("XMPP SENDING: " << str);
  this->send_data(std::move(str));
71 72
}

73 74 75 76 77
void XmppComponent::on_connection_failed(const std::string& reason)
{
  log_error("Failed to connect to the XMPP server: " << reason);
}

78 79
void XmppComponent::on_connected()
{
louiz’'s avatar
louiz’ committed
80
  log_info("connected to XMPP server");
81
  XmlNode node("stream:stream", nullptr);
82 83
  node["xmlns"] = COMPONENT_NS;
  node["xmlns:stream"] = STREAM_NS;
84
  node["to"] = this->served_hostname;
85
  this->send_stanza(node);
louiz’'s avatar
louiz’ committed
86
  this->doc_open = true;
87 88 89
  // We may have some pending data to send: this happens when we try to send
  // some data before we are actually connected.  We send that data right now, if any
  this->send_pending_data();
90 91 92 93
}

void XmppComponent::on_connection_close()
{
louiz’'s avatar
louiz’ committed
94
  log_info("XMPP server closed connection");
95 96
}

97
void XmppComponent::parse_in_buffer(const size_t size)
98
{
99 100 101 102 103 104 105 106 107 108 109 110
  if (!this->in_buf.empty())
    { // This may happen if the parser could not allocate enough space for
      // us. We try to feed it the data that was read into our in_buf
      // instead. If this fails again we are in trouble.
      this->parser.feed(this->in_buf.data(), this->in_buf.size(), false);
      this->in_buf.clear();
    }
  else
    { // Just tell the parser to parse the data that was placed into the
      // buffer it provided to us with GetBuffer
      this->parser.parse(size, false);
    }
111 112
}

louiz’'s avatar
louiz’ committed
113 114 115 116 117 118 119 120
void XmppComponent::shutdown()
{
  for (auto it = this->bridges.begin(); it != this->bridges.end(); ++it)
  {
    it->second->shutdown();
  }
}

louiz’'s avatar
louiz’ committed
121 122
void XmppComponent::clean()
{
louiz’'s avatar
louiz’ committed
123 124
  auto it = this->bridges.begin();
  while (it != this->bridges.end())
louiz’'s avatar
louiz’ committed
125 126
  {
    it->second->clean();
127
    if (it->second->active_clients() == 0)
louiz’'s avatar
louiz’ committed
128 129 130
      it = this->bridges.erase(it);
    else
      ++it;
louiz’'s avatar
louiz’ committed
131 132 133
  }
}

134 135
void XmppComponent::on_remote_stream_open(const XmlNode& node)
{
louiz’'s avatar
louiz’ committed
136
  log_debug("XMPP DOCUMENT OPEN: " << node.to_string());
137 138 139 140 141 142
  try
    {
      this->stream_id = node["id"];
    }
  catch (const AttributeNotFound& e)
    {
louiz’'s avatar
louiz’ committed
143
      log_error("Error: no attribute 'id' found");
144 145 146 147 148
      this->send_stream_error("bad-format", "missing 'id' attribute");
      this->close_document();
      return ;
    }

149
  this->last_auth = false;
150
  // Try to authenticate
151 152 153 154 155 156 157 158 159 160 161
  char digest[HASH_LENGTH * 2 + 1];
  sha1nfo sha1;
  sha1_init(&sha1);
  sha1_write(&sha1, this->stream_id.data(), this->stream_id.size());
  sha1_write(&sha1, this->secret.data(),  this->secret.size());
  const uint8_t* result = sha1_result(&sha1);
  for (int i=0; i < HASH_LENGTH; i++)
    sprintf(digest + (i*2), "%02x", result[i]);
  digest[HASH_LENGTH * 2] = '\0';

  Stanza handshake("handshake");
162 163 164 165 166 167 168
  handshake.set_inner(digest);
  handshake.close();
  this->send_stanza(handshake);
}

void XmppComponent::on_remote_stream_close(const XmlNode& node)
{
louiz’'s avatar
louiz’ committed
169
  log_debug("XMPP DOCUMENT CLOSE " << node.to_string());
170
  this->doc_open = false;
171 172
}

173 174 175 176 177
void XmppComponent::reset()
{
  this->parser.reset();
}

178 179
void XmppComponent::on_stanza(const Stanza& stanza)
{
louiz’'s avatar
louiz’ committed
180
  log_debug("XMPP RECEIVING: " << stanza.to_string());
181
  std::function<void(const Stanza&)> handler;
182 183
  try
    {
184
      handler = this->stanza_handlers.at(stanza.get_name());
185 186 187
    }
  catch (const std::out_of_range& exception)
    {
louiz’'s avatar
louiz’ committed
188
      log_warning("No handler for stanza of type " << stanza.get_name());
189 190
      return;
    }
191
  handler(stanza);
192 193 194 195 196 197
}

void XmppComponent::send_stream_error(const std::string& name, const std::string& explanation)
{
  XmlNode node("stream:error", nullptr);
  XmlNode error(name, nullptr);
198
  error["xmlns"] = STREAM_NS;
199 200 201 202 203 204 205 206 207 208
  if (!explanation.empty())
    error.set_inner(explanation);
  error.close();
  node.add_child(std::move(error));
  node.close();
  this->send_stanza(node);
}

void XmppComponent::close_document()
{
louiz’'s avatar
louiz’ committed
209
  log_debug("XMPP SENDING: </stream:stream>");
210
  this->send_data("</stream:stream>");
louiz’'s avatar
louiz’ committed
211
  this->doc_open = false;
212 213 214 215
}

void XmppComponent::handle_handshake(const Stanza& stanza)
{
216
  (void)stanza;
217
  this->authenticated = true;
218 219
  this->ever_auth = true;
  this->last_auth = true;
louiz’'s avatar
louiz’ committed
220
  log_info("Authenticated with the XMPP server");
221
}
222 223 224 225 226 227

void XmppComponent::handle_presence(const Stanza& stanza)
{
  Bridge* bridge = this->get_user_bridge(stanza["from"]);
  Jid to(stanza["to"]);
  Iid iid(to.local);
louiz’'s avatar
louiz’ committed
228 229 230 231 232 233 234 235 236
  std::string type;
  try {
    type = stanza["type"];
  }
  catch (const AttributeNotFound&) {}

  if (!iid.chan.empty() && !iid.chan.empty())
    { // presence toward a MUC that corresponds to an irc channel
      if (type.empty())
louiz’'s avatar
louiz’ committed
237 238 239 240
        {
          const std::string own_nick = bridge->get_own_nick(iid);
          if (!own_nick.empty() && own_nick != to.resource)
            bridge->send_irc_nick_change(iid, to.resource);
241
          bridge->join_irc_channel(iid, to.resource);
louiz’'s avatar
louiz’ committed
242
        }
louiz’'s avatar
louiz’ committed
243 244
      else if (type == "unavailable")
        {
245
          XmlNode* status = stanza.get_child(COMPONENT_NS":status");
louiz’'s avatar
louiz’ committed
246 247 248
          bridge->leave_irc_channel(std::move(iid), status ? std::move(status->get_inner()) : "");
        }
    }
249 250
}

louiz’'s avatar
louiz’ committed
251 252 253 254 255
void XmppComponent::handle_message(const Stanza& stanza)
{
  Bridge* bridge = this->get_user_bridge(stanza["from"]);
  Jid to(stanza["to"]);
  Iid iid(to.local);
256
  XmlNode* body = stanza.get_child(COMPONENT_NS":body");
louiz’'s avatar
louiz’ committed
257 258 259 260 261
  if (stanza["type"] == "groupchat")
    {
      if (to.resource.empty())
        if (body && !body->get_inner().empty())
          bridge->send_channel_message(iid, body->get_inner());
262 263 264
      XmlNode* subject = stanza.get_child(COMPONENT_NS":subject");
      if (subject)
        bridge->set_channel_topic(iid, subject->get_inner());
louiz’'s avatar
louiz’ committed
265
    }
louiz’'s avatar
louiz’ committed
266 267 268 269 270
  else
    {
      if (body && !body->get_inner().empty())
        bridge->send_private_message(iid, body->get_inner());
    }
louiz’'s avatar
louiz’ committed
271 272
}

273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312
void XmppComponent::handle_iq(const Stanza& stanza)
{
  Bridge* bridge = this->get_user_bridge(stanza["from"]);
  Jid to(stanza["to"]);
  std::string type;
  try {
    type = stanza["type"];
  }
  catch (const AttributeNotFound&)
    { return; }
  if (type == "set")
    {
      XmlNode* query;
      if ((query = stanza.get_child(MUC_ADMIN_NS":query")))
        {
          XmlNode* child;
          if ((child = query->get_child(MUC_ADMIN_NS":item")))
            {
              std::string nick;
              std::string role;
              try {
                nick = (*child)["nick"];
                role = (*child)["role"];
              }
              catch (const AttributeNotFound&)
                { return; }
              if (!nick.empty() && role == "none")
                {
                  std::string reason;
                  XmlNode* reason_el = child->get_child(MUC_ADMIN_NS":reason");
                  if (reason_el)
                    reason = reason_el->get_inner();
                  Iid iid(to.local);
                  bridge->send_irc_kick(iid, nick, reason);
                }
            }
        }
    }
}

louiz’'s avatar
louiz’ committed
313 314 315 316 317 318 319 320 321
void XmppComponent::handle_error(const Stanza& stanza)
{
  XmlNode* text = stanza.get_child(STREAMS_NS":text");
  std::string error_message("Unspecified error");
  if (text)
    error_message = text->get_inner();
  log_error("Stream error received from the XMPP server: " << error_message);
}

322 323 324 325 326 327 328 329 330 331 332 333 334
Bridge* XmppComponent::get_user_bridge(const std::string& user_jid)
{
  try
    {
      return this->bridges.at(user_jid).get();
    }
  catch (const std::out_of_range& exception)
    {
      this->bridges.emplace(user_jid, std::make_unique<Bridge>(user_jid, this, this->poller));
      return this->bridges.at(user_jid).get();
    }
}

335 336 337 338 339
void* XmppComponent::get_receive_buffer(const size_t size) const
{
  return this->parser.get_buffer(size);
}

340
void XmppComponent::send_message(const std::string& from, Xmpp::body&& body, const std::string& to)
341 342 343 344 345
{
  XmlNode node("message");
  node["to"] = to;
  node["from"] = from + "@" + this->served_hostname;
  XmlNode body_node("body");
346
  body_node.set_inner(std::get<0>(body));
347 348 349 350 351 352
  body_node.close();
  node.add_child(std::move(body_node));
  node.close();
  this->send_stanza(node);
}

353 354 355
void XmppComponent::send_user_join(const std::string& from,
                                   const std::string& nick,
                                   const std::string& realjid,
356 357 358 359
                                   const std::string& affiliation,
                                   const std::string& role,
                                   const std::string& to,
                                   const bool self)
360 361 362 363 364 365
{
  XmlNode node("presence");
  node["to"] = to;
  node["from"] = from + "@" + this->served_hostname + "/" + nick;

  XmlNode x("x");
366
  x["xmlns"] = MUC_USER_NS;
367 368

  XmlNode item("item");
369 370 371 372
  if (!affiliation.empty())
    item["affiliation"] = affiliation;
  if (!role.empty())
    item["role"] = role;
373 374 375 376 377 378
  if (!realjid.empty())
    {
      const std::string preped_jid = jidprep(realjid);
      if (!preped_jid.empty())
        item["jid"] = preped_jid;
    }
379
  item.close();
380
  x.add_child(std::move(item));
381

382 383 384 385 386 387 388
  if (self)
    {
      XmlNode status("status");
      status["code"] = "110";
      status.close();
      x.add_child(std::move(status));
    }
389 390 391 392 393 394
  x.close();
  node.add_child(std::move(x));
  node.close();
  this->send_stanza(node);
}

395
void XmppComponent::send_topic(const std::string& from, Xmpp::body&& topic, const std::string& to)
396 397 398 399 400 401
{
  XmlNode message("message");
  message["to"] = to;
  message["from"] = from + "@" + this->served_hostname;
  message["type"] = "groupchat";
  XmlNode subject("subject");
402
  subject.set_inner(std::get<0>(topic));
403 404 405 406 407
  subject.close();
  message.add_child(std::move(subject));
  message.close();
  this->send_stanza(message);
}
louiz’'s avatar
louiz’ committed
408

409
void XmppComponent::send_muc_message(const std::string& muc_name, const std::string& nick, Xmpp::body&& xmpp_body, const std::string& jid_to)
louiz’'s avatar
louiz’ committed
410 411 412
{
  Stanza message("message");
  message["to"] = jid_to;
louiz’'s avatar
louiz’ committed
413 414 415 416
  if (!nick.empty())
    message["from"] = muc_name + "@" + this->served_hostname + "/" + nick;
  else // Message from the room itself
    message["from"] = muc_name + "@" + this->served_hostname;
louiz’'s avatar
louiz’ committed
417 418
  message["type"] = "groupchat";
  XmlNode body("body");
419
  body.set_inner(std::get<0>(xmpp_body));
louiz’'s avatar
louiz’ committed
420 421
  body.close();
  message.add_child(std::move(body));
422 423 424 425 426 427 428 429 430
  if (std::get<1>(xmpp_body))
    {
      XmlNode html("html");
      html["xmlns"] = XHTMLIM_NS;
      // Pass the ownership of the pointer to this xmlnode
      html.add_child(std::get<1>(xmpp_body).release());
      html.close();
      message.add_child(std::move(html));
    }
louiz’'s avatar
louiz’ committed
431 432 433
  message.close();
  this->send_stanza(message);
}
louiz’'s avatar
louiz’ committed
434

435
void XmppComponent::send_muc_leave(std::string&& muc_name, std::string&& nick, Xmpp::body&& message, const std::string& jid_to, const bool self)
louiz’'s avatar
louiz’ committed
436 437 438 439 440
{
  Stanza presence("presence");
  presence["to"] = jid_to;
  presence["from"] = muc_name + "@" + this->served_hostname + "/" + nick;
  presence["type"] = "unavailable";
441
  const std::string message_str = std::get<0>(message);
442
  if (!message_str.empty() || self)
louiz’'s avatar
louiz’ committed
443 444
    {
      XmlNode status("status");
445 446
      if (!message_str.empty())
        status.set_inner(message_str);
louiz’'s avatar
louiz’ committed
447 448 449 450 451 452 453 454
      if (self)
        status["code"] = "110";
      status.close();
      presence.add_child(std::move(status));
    }
  presence.close();
  this->send_stanza(presence);
}
louiz’'s avatar
louiz’ committed
455

456 457 458 459 460 461 462
void XmppComponent::send_nick_change(const std::string& muc_name,
                                     const std::string& old_nick,
                                     const std::string& new_nick,
                                     const std::string& affiliation,
                                     const std::string& role,
                                     const std::string& jid_to,
                                     const bool self)
louiz’'s avatar
louiz’ committed
463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489
{
  Stanza presence("presence");
  presence["to"] = jid_to;
  presence["from"] = muc_name + "@" + this->served_hostname + "/" + old_nick;
  presence["type"] = "unavailable";
  XmlNode x("x");
  x["xmlns"] = MUC_USER_NS;
  XmlNode item("item");
  item["nick"] = new_nick;
  item.close();
  x.add_child(std::move(item));
  XmlNode status("status");
  status["code"] = "303";
  status.close();
  x.add_child(std::move(status));
  if (self)
    {
      XmlNode status2("status");
      status2["code"] = "110";
      status2.close();
      x.add_child(std::move(status2));
    }
  x.close();
  presence.add_child(std::move(x));
  presence.close();
  this->send_stanza(presence);

490
  this->send_user_join(muc_name, new_nick, "", affiliation, role, jid_to, self);
louiz’'s avatar
louiz’ committed
491
}
492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527

void XmppComponent::kick_user(const std::string& muc_name,
                                  const std::string& target,
                                  const std::string& txt,
                                  const std::string& author,
                                  const std::string& jid_to)
{
  Stanza presence("presence");
  presence["from"] = muc_name + "@" + this->served_hostname + "/" + target;
  presence["to"] = jid_to;
  presence["type"] = "unavailable";
  XmlNode x("x");
  x["xmlns"] = MUC_USER_NS;
  XmlNode item("item");
  item["affiliation"] = "none";
  item["role"] = "none";
  XmlNode actor("actor");
  actor["nick"] = author;
  actor["jid"] = author; // backward compatibility with old clients
  actor.close();
  item.add_child(std::move(actor));
  XmlNode reason("reason");
  reason.set_inner(txt);
  reason.close();
  item.add_child(std::move(reason));
  item.close();
  x.add_child(std::move(item));
  XmlNode status("status");
  status["code"] = "307";
  status.close();
  x.add_child(std::move(status));
  x.close();
  presence.add_child(std::move(x));
  presence.close();
  this->send_stanza(presence);
}
528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552

void XmppComponent::send_nickname_conflict_error(const std::string& muc_name,
                                                 const std::string& nickname,
                                                 const std::string& jid_to)
{
  Stanza presence("presence");
  presence["from"] = muc_name + "@" + this->served_hostname + "/" + nickname;
  presence["to"] = jid_to;
  XmlNode x("x");
  x["xmlns"] = MUC_NS;
  x.close();
  presence.add_child(std::move(x));
  XmlNode error("error");
  error["by"] = muc_name + "@" + this->served_hostname;
  error["type"] = "cancel";
  error["code"] = "409";
  XmlNode conflict("conflict");
  conflict["xmlns"] = STANZA_NS;
  conflict.close();
  error.add_child(std::move(conflict));
  error.close();
  presence.add_child(std::move(error));
  presence.close();
  this->send_stanza(presence);
}
553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574

void XmppComponent::send_affiliation_role_change(const std::string& muc_name,
                                                 const std::string& target,
                                                 const std::string& affiliation,
                                                 const std::string& role,
                                                 const std::string& jid_to)
{
  Stanza presence("presence");
  presence["from"] = muc_name + "@" + this->served_hostname + "/" + target;
  presence["to"] = jid_to;
  XmlNode x("x");
  x["xmlns"] = MUC_USER_NS;
  XmlNode item("item");
  item["affiliation"] = affiliation;
  item["role"] = role;
  item.close();
  x.add_child(std::move(item));
  x.close();
  presence.add_child(std::move(x));
  presence.close();
  this->send_stanza(presence);
}