bridge.cpp 33.7 KB
Newer Older
1
#include <bridge/bridge.hpp>
2
#include <bridge/list_element.hpp>
3
#include <xmpp/biboumi_component.hpp>
4
#include <network/poller.hpp>
5
#include <utils/empty_if_fixed_server.hpp>
6
#include <utils/encoding.hpp>
7
#include <utils/tolower.hpp>
louiz’'s avatar
louiz’ committed
8
#include <logger/logger.hpp>
9
#include <utils/revstr.hpp>
louiz’'s avatar
louiz’ committed
10
#include <utils/split.hpp>
11
#include <xmpp/jid.hpp>
12
#include <database/database.hpp>
louiz’'s avatar
louiz’ committed
13

louiz’'s avatar
louiz’ committed
14 15
using namespace std::string_literals;

16 17
static const char* action_prefix = "\01ACTION ";

18 19 20 21 22 23 24 25 26 27 28 29

static std::string in_encoding_for(const Bridge& bridge, const Iid& iid)
{
#ifdef USE_DATABASE
  const auto jid = bridge.get_bare_jid();
  auto options = Database::get_irc_channel_options_with_server_default(jid, iid.get_server(), iid.get_local());
  return options.encodingIn.value();
#else
  return {"ISO-8859-1"};
#endif
}

30
Bridge::Bridge(const std::string& user_jid, BiboumiComponent& xmpp, std::shared_ptr<Poller> poller):
31 32 33 34 35 36
  user_jid(user_jid),
  xmpp(xmpp),
  poller(poller)
{
}

37
/**
38 39
 * Return the role and affiliation, corresponding to the given irc mode
 */
40 41
static std::tuple<std::string, std::string> get_role_affiliation_from_irc_mode(const char mode)
{
42 43
  if (mode == 'a' || mode == 'q'){
    return std::make_tuple("moderator", "owner");}
44 45 46 47 48 49 50 51 52 53
  else if (mode == 'o')
    return std::make_tuple("moderator", "admin");
  else if (mode == 'h')
    return std::make_tuple("moderator", "member");
  else if (mode == 'v')
    return std::make_tuple("participant", "member");
  else
    return std::make_tuple("participant", "none");
}

54
void Bridge::shutdown(const std::string& exit_message)
louiz’'s avatar
louiz’ committed
55 56 57
{
  for (auto it = this->irc_clients.begin(); it != this->irc_clients.end(); ++it)
  {
louiz’'s avatar
louiz’ committed
58 59
    it->second->send_quit_command(exit_message);
    it->second->leave_dummy_channel(exit_message);
louiz’'s avatar
louiz’ committed
60 61 62
  }
}

louiz’'s avatar
louiz’ committed
63 64 65 66 67 68
void Bridge::clean()
{
  auto it = this->irc_clients.begin();
  while (it != this->irc_clients.end())
  {
    IrcClient* client = it->second.get();
louiz’'s avatar
louiz’ committed
69 70
    if (!client->is_connected() && !client->is_connecting() &&
        !client->get_resolver().is_resolving())
louiz’'s avatar
louiz’ committed
71 72 73 74 75 76
      it = this->irc_clients.erase(it);
    else
      ++it;
  }
}

77 78 79 80 81
const std::string& Bridge::get_jid() const
{
  return this->user_jid;
}

louiz’'s avatar
louiz’ committed
82 83 84 85 86 87
std::string Bridge::get_bare_jid() const
{
  Jid jid(this->user_jid);
  return jid.local + "@" + jid.domain;
}

88
Xmpp::body Bridge::make_xmpp_body(const std::string& str, const std::string& encoding)
89 90 91 92 93
{
  std::string res;
  if (utils::is_valid_utf8(str.c_str()))
    res = str;
  else
94
    res = utils::convert_to_utf8(str, encoding.data());
95
  return irc_format_to_xhtmlim(res);
96 97
}

98
IrcClient* Bridge::make_irc_client(const std::string& hostname, const std::string& nickname)
99 100 101 102 103 104 105
{
  try
    {
      return this->irc_clients.at(hostname).get();
    }
  catch (const std::out_of_range& exception)
    {
106 107
      auto username = nickname;
      auto realname = nickname;
louiz’'s avatar
louiz’ committed
108
      Jid jid(this->user_jid);
109 110 111 112 113 114 115 116
      if (Config::get("realname_from_jid", "false") == "true")
        {
          username = jid.local;
          realname = this->get_bare_jid();
        }
      this->irc_clients.emplace(hostname,
                                std::make_shared<IrcClient>(this->poller, hostname,
                                                            nickname, username,
louiz’'s avatar
louiz’ committed
117
                                                            realname, jid.domain,
118
                                                            *this));
119 120 121 122 123
      std::shared_ptr<IrcClient> irc = this->irc_clients.at(hostname);
      return irc.get();
    }
}

louiz’'s avatar
louiz’ committed
124
IrcClient* Bridge::get_irc_client(const std::string& hostname)
125 126 127 128 129 130 131 132 133 134 135
{
  try
    {
      return this->irc_clients.at(hostname).get();
    }
  catch (const std::out_of_range& exception)
    {
      throw IRCNotConnected(hostname);
    }
}

136
IrcClient* Bridge::find_irc_client(const std::string& hostname) const
louiz’'s avatar
louiz’ committed
137 138 139 140 141 142 143 144 145 146 147
{
  try
    {
      return this->irc_clients.at(hostname).get();
    }
  catch (const std::out_of_range& exception)
    {
      return nullptr;
    }
}

148 149 150 151 152 153 154 155 156
bool Bridge::join_irc_channel(const Iid& iid, const std::string& nickname, const std::string& password,
                              const std::string& resource)
{
  const auto hostname = iid.get_server();
  IrcClient* irc = this->make_irc_client(hostname, nickname);
  this->add_resource_to_server(hostname, resource);
  auto res_in_chan = this->is_resource_in_chan(ChannelKey{iid.get_local(), hostname}, resource);
  if (!res_in_chan)
    this->add_resource_to_chan(ChannelKey{iid.get_local(), hostname}, resource);
louiz’'s avatar
louiz’ committed
157
  if (iid.get_local().empty())
louiz’'s avatar
louiz’ committed
158 159 160 161 162 163 164 165 166
    { // Join the dummy channel
      if (irc->is_welcomed())
        {
          if (irc->get_dummy_channel().joined)
            return false;
          // Immediately simulate a message coming from the IRC server saying that we
          // joined the channel
          const IrcMessage join_message(irc->get_nick(), "JOIN", {""});
          irc->on_channel_join(join_message);
louiz’'s avatar
louiz’ committed
167
          const IrcMessage end_join_message(std::string(iid.get_server()), "366",
louiz’'s avatar
louiz’ committed
168 169 170 171 172
                                            {irc->get_nick(),
                                                "", "End of NAMES list"});
          irc->on_channel_completely_joined(end_join_message);
        }
      else
173
        {
louiz’'s avatar
louiz’ committed
174
          irc->get_dummy_channel().joining = true;
175 176
          irc->start();
        }
louiz’'s avatar
louiz’ committed
177 178
      return true;
    }
louiz’'s avatar
louiz’ committed
179
  if (irc->is_channel_joined(iid.get_local()) == false)
louiz’'s avatar
louiz’ committed
180
    {
181
      irc->send_join_command(iid.get_local(), password);
louiz’'s avatar
louiz’ committed
182
      return true;
183 184
    } else if (!res_in_chan) {
      this->generate_channel_join_for_resource(iid, resource);
louiz’'s avatar
louiz’ committed
185 186
    }
  return false;
187 188
}

louiz’'s avatar
louiz’ committed
189 190
void Bridge::send_channel_message(const Iid& iid, const std::string& body)
{
191
  if (iid.get_server().empty())
louiz’'s avatar
louiz’ committed
192
    {
193
      for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
194 195 196 197 198
        this->xmpp.send_stanza_error("message", this->user_jid + "/" + resource, std::to_string(iid), "",
                                     "cancel", "remote-server-not-found",
                                     std::to_string(iid) + " is not a valid channel name. "
                                         "A correct room jid is of the form: #<chan>%<server>",
                                     false);
louiz’'s avatar
louiz’ committed
199 200
      return;
    }
louiz’'s avatar
louiz’ committed
201
  IrcClient* irc = this->get_irc_client(iid.get_server());
202

203 204 205 206 207 208 209 210 211 212 213 214
  // Because an IRC message cannot contain \n, we need to convert each line
  // of text into a separate IRC message. For conveniance, we also cut the
  // message into submessages on the XMPP side, this way the user of the
  // gateway sees what was actually sent over IRC.  For example if an user
  // sends “hello\n/me waves”, two messages will be generated: “hello” and
  // “/me waves”. Note that the “command” handling (messages starting with
  // /me, /mode, etc) is done for each message generated this way. In the
  // above example, the /me will be interpreted.
  std::vector<std::string> lines = utils::split(body, '\n', true);
  if (lines.empty())
    return ;
  for (const std::string& line: lines)
louiz’'s avatar
louiz’ committed
215
    {
louiz’'s avatar
louiz’ committed
216
      if (line.substr(0, 5) == "/mode")
217
        {
louiz’'s avatar
louiz’ committed
218
          std::vector<std::string> args = utils::split(line.substr(5), ' ', false);
louiz’'s avatar
louiz’ committed
219
          irc->send_mode_command(iid.get_local(), args);
220 221 222 223
          continue;             // We do not want to send that back to the
                                // XMPP user, that’s not a textual message.
        }
      else if (line.substr(0, 4) == "/me ")
louiz’'s avatar
louiz’ committed
224
        irc->send_channel_message(iid.get_local(), action_prefix + line.substr(4) + "\01");
225
      else
louiz’'s avatar
louiz’ committed
226
        irc->send_channel_message(iid.get_local(), line);
227
      for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
228 229
        this->xmpp.send_muc_message(std::to_string(iid), irc->get_own_nick(),
                                    this->make_xmpp_body(line), this->user_jid + "/" + resource);
louiz’'s avatar
louiz’ committed
230
    }
louiz’'s avatar
louiz’ committed
231 232
}

233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 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
void Bridge::forward_affiliation_role_change(const Iid& iid, const std::string& nick,
                                             const std::string& affiliation,
                                             const std::string& role)
{
  IrcClient* irc = this->get_irc_client(iid.get_server());
  IrcChannel* chan = irc->get_channel(iid.get_local());
  if (!chan || !chan->joined)
    return;
  IrcUser* user = chan->find_user(nick);
  if (!user)
    return;
  // For each affiliation or role, we have a “maximal” mode that we want to
  // set. We must remove any superior mode at the same time. For example if
  // the user already has +o mode, and we set its affiliation to member, we
  // remove the +o mode, and add +v.  For each “superior” mode (for example,
  // for +v, the superior modes are 'h', 'a', 'o' and 'q') we check if that
  // user has it, and if yes we remove that mode

  std::size_t nb = 1;               // the number of times the nick must be
                                    // repeated in the argument list
  std::string modes;                // The string of modes to
                                    // add/remove. For example "+v-aoh"
  std::vector<char> modes_to_remove; // List of modes to check for removal
  if (affiliation == "none")
    {
      modes = "";
      nb = 0;
      modes_to_remove = {'v', 'h', 'o', 'a', 'q'};
    }
  else if (affiliation == "member")
    {
      modes = "+v";
      modes_to_remove = {'h', 'o', 'a', 'q'};
    }
  else if (role == "moderator")
    {
      modes = "+h";
      modes_to_remove = {'o', 'a', 'q'};
    }
  else if (affiliation == "admin")
    {
      modes = "+o";
      modes_to_remove = {'a', 'q'};
    }
  else if (affiliation == "owner")
    {
      modes = "+a";
      modes_to_remove = {'q'};
    }
  else
    return;
  for (const char mode: modes_to_remove)
    if (user->modes.find(mode) != user->modes.end())
      {
        modes += "-"s + mode;
        nb++;
      }
  if (modes.empty())
    return;
  std::vector<std::string> args(nb, nick);
  args.insert(args.begin(), modes);
  irc->send_mode_command(iid.get_local(), args);
}

louiz’'s avatar
louiz’ committed
297
void Bridge::send_private_message(const Iid& iid, const std::string& body, const std::string& type)
louiz’'s avatar
louiz’ committed
298
{
louiz’'s avatar
louiz’ committed
299
  if (iid.get_local().empty() || iid.get_server().empty())
300
    {
301
      this->xmpp.send_stanza_error("message", this->user_jid, std::to_string(iid), "",
302 303 304 305
                                    "cancel", "remote-server-not-found",
                                    std::to_string(iid) + " is not a valid channel name. "
                                    "A correct room jid is of the form: #<chan>%<server>",
                                    false);
306 307
      return;
    }
308
  IrcClient* irc = this->get_irc_client(iid.get_server());
309 310 311 312 313 314
  std::vector<std::string> lines = utils::split(body, '\n', true);
  if (lines.empty())
    return ;
  for (const std::string& line: lines)
    {
      if (line.substr(0, 4) == "/me ")
louiz’'s avatar
louiz’ committed
315
        irc->send_private_message(iid.get_local(), action_prefix + line.substr(4) + "\01", type);
316
      else
louiz’'s avatar
louiz’ committed
317
        irc->send_private_message(iid.get_local(), line, type);
318
    }
louiz’'s avatar
louiz’ committed
319 320
}

louiz’'s avatar
louiz’ committed
321 322 323 324 325 326
void Bridge::send_raw_message(const std::string& hostname, const std::string& body)
{
  IrcClient* irc = this->get_irc_client(hostname);
  irc->send_raw(body);
}

327
void Bridge::leave_irc_channel(Iid&& iid, std::string&& status_message, const std::string& resource)
louiz’'s avatar
louiz’ committed
328
{
louiz’'s avatar
louiz’ committed
329
  IrcClient* irc = this->get_irc_client(iid.get_server());
330 331 332 333 334 335
  const auto key = iid.to_tuple();
  if (!this->is_resource_in_chan(key, resource))
    return ;

  const auto resources = this->number_of_resources_in_chan(key);
  if (resources == 1)
336 337 338 339 340 341
    {
      irc->send_part_command(iid.get_local(), status_message);
      // Since there are no resources left in that channel, we don't
      // want to receive private messages using this room's JID
      this->remove_all_preferred_from_jid_of_room(iid.get_local());
    }
342 343 344 345 346 347 348 349 350 351
  else
    {
      IrcChannel* chan = irc->get_channel(iid.get_local());
      if (chan)
        {
          auto nick = chan->get_self()->nick;
          this->remove_resource_from_chan(key, resource);
          this->send_muc_leave(std::move(iid), std::move(nick),
                               "Biboumi note: "s + std::to_string(resources - 1) + " resources are still in this channel.",
          true, resource);
352 353
          if (this->number_of_channels_the_resource_is_in(iid.get_server(), resource) == 0)
            this->remove_resource_from_server(iid.get_server(), resource);
354 355
        }
    }
356
}
louiz’'s avatar
louiz’ committed
357

louiz’'s avatar
louiz’ committed
358 359
void Bridge::send_irc_nick_change(const Iid& iid, const std::string& new_nick)
{
louiz’'s avatar
louiz’ committed
360
  IrcClient* irc = this->get_irc_client(iid.get_server());
361
  irc->send_nick_command(new_nick);
louiz’'s avatar
louiz’ committed
362 363
}

364 365 366 367 368 369
void Bridge::send_irc_channel_list_request(const Iid& iid, const std::string& iq_id,
                                           const std::string& to_jid)
{
  IrcClient* irc = this->get_irc_client(iid.get_server());

  irc->send_list_command();
370

371
  std::vector<ListElement> list;
372

373 374 375
  irc_responder_callback_t cb = [this, iid, iq_id, to_jid, list=std::move(list)](const std::string& irc_hostname,
                                                           const IrcMessage& message) mutable -> bool
    {
376 377
      if (irc_hostname != iid.get_server())
        return false;
378 379 380 381 382 383
      if (message.command == "263" || message.command == "RPL_TRYAGAIN" ||
          message.command == "ERR_TOOMANYMATCHES" || message.command == "ERR_NOSUCHSERVER")
        {
          std::string text;
          if (message.arguments.size() >= 2)
            text = message.arguments[1];
384
          this->xmpp.send_stanza_error("iq", to_jid, std::to_string(iid), iq_id,
385
                                        "wait", "service-unavailable", text, false);
386 387 388 389 390 391 392 393 394 395 396
          return true;
        }
      else if (message.command == "322" || message.command == "RPL_LIST")
        { // Add element to list
          if (message.arguments.size() == 4)
            list.emplace_back(message.arguments[1], message.arguments[2],
                              message.arguments[3]);
          return false;
        }
      else if (message.command == "323" || message.command == "RPL_LISTEND")
        { // Send the iq response with the content of the list
397
          this->xmpp.send_iq_room_list_result(iq_id, to_jid, std::to_string(iid), list);
398 399 400 401 402 403 404
          return true;
        }
      return false;
    };
  this->add_waiting_irc(std::move(cb));
}

405 406
void Bridge::send_irc_kick(const Iid& iid, const std::string& target, const std::string& reason,
                           const std::string& iq_id, const std::string& to_jid)
407
{
louiz’'s avatar
louiz’ committed
408
  IrcClient* irc = this->get_irc_client(iid.get_server());
409 410 411 412

  irc->send_kick_command(iid.get_local(), target, reason);
  irc_responder_callback_t cb = [this, target, iq_id, to_jid, iid](const std::string& irc_hostname,
                                                                   const IrcMessage& message) -> bool
413
    {
414 415 416 417 418 419 420
      if (irc_hostname != iid.get_server())
        return false;
      if (message.command == "KICK" && message.arguments.size() >= 2)
        {
          const std::string target_later = message.arguments[1];
          const std::string chan_name_later = utils::tolower(message.arguments[0]);
          if (target_later != target || chan_name_later != iid.get_local())
421
            return false;
422
          this->xmpp.send_iq_result(iq_id, to_jid, std::to_string(iid));
423 424 425 426 427 428 429 430 431
        }
      else if (message.command == "401" && message.arguments.size() >= 2)
        {
          const std::string target_later = message.arguments[1];
          if (target_later != target)
            return false;
          std::string error_message = "No such nick";
          if (message.arguments.size() >= 3)
            error_message = message.arguments[2];
432
          this->xmpp.send_stanza_error("iq", to_jid, std::to_string(iid), iq_id, "cancel", "item-not-found",
433 434 435 436 437 438 439 440 441 442
                                        error_message, false);
        }
      else if (message.command == "482" && message.arguments.size() >= 2)
        {
          const std::string chan_name_later = utils::tolower(message.arguments[1]);
          if (chan_name_later != iid.get_local())
            return false;
          std::string error_message = "You're not channel operator";
          if (message.arguments.size() >= 3)
            error_message = message.arguments[2];
443
          this->xmpp.send_stanza_error("iq", to_jid, std::to_string(iid), iq_id, "cancel", "not-allowed",
444 445 446 447 448
                                        error_message, false);
        }
      return true;
    };
  this->add_waiting_irc(std::move(cb));
449 450
}

451 452
void Bridge::set_channel_topic(const Iid& iid, const std::string& subject)
{
louiz’'s avatar
louiz’ committed
453
  IrcClient* irc = this->get_irc_client(iid.get_server());
454
  irc->send_topic_command(iid.get_local(), subject);
455 456
}

louiz’'s avatar
louiz’ committed
457 458 459 460 461 462 463
void Bridge::send_xmpp_version_to_irc(const Iid& iid, const std::string& name, const std::string& version, const std::string& os)
{
  std::string result(name + " " + version + " " + os);

  this->send_private_message(iid, "\01VERSION "s + result + "\01", "NOTICE");
}

464 465
void Bridge::send_irc_ping_result(const Iid& iid, const std::string& id)
{
466
  this->send_private_message(iid, "\01PING "s + utils::revstr(id) + "\01", "NOTICE");
467 468 469 470 471 472
}

void Bridge::send_irc_user_ping_request(const std::string& irc_hostname, const std::string& nick,
                                        const std::string& iq_id, const std::string& to_jid,
                                        const std::string& from_jid)
{
473
  Iid iid(nick, irc_hostname, Iid::Type::User);
474 475
  this->send_private_message(iid, "\01PING " + iq_id + "\01");

476
  irc_responder_callback_t cb = [this, nick=utils::tolower(nick), iq_id, to_jid, irc_hostname, from_jid](const std::string& hostname, const IrcMessage& message) -> bool
477
    {
louiz’'s avatar
louiz’ committed
478
      if (irc_hostname != hostname || message.arguments.size() < 2)
479 480 481
        return false;
      IrcUser user(message.prefix);
      const std::string body = message.arguments[1];
louiz’'s avatar
louiz’ committed
482 483
      if (message.command == "NOTICE" && utils::tolower(user.nick) == nick
          && body.substr(0, 6) == "\01PING ")
484
        {
485
          const std::string id = body.substr(6, body.size() - 7);
486 487
          if (id != iq_id)
            return false;
488
          this->xmpp.send_iq_result_full_jid(iq_id, to_jid, from_jid);
489 490
          return true;
        }
louiz’'s avatar
louiz’ committed
491
      if (message.command == "401" && message.arguments[1] == nick)
492 493 494 495
        {
          std::string error_message = "No such nick";
          if (message.arguments.size() >= 3)
            error_message = message.arguments[2];
496
          this->xmpp.send_stanza_error("iq", to_jid, from_jid, iq_id, "cancel", "service-unavailable",
497 498 499 500 501 502 503 504 505
                                        error_message, true);
          return true;
        }

      return false;
    };
  this->add_waiting_irc(std::move(cb));
}

506 507 508 509 510 511 512 513
void Bridge::send_irc_participant_ping_request(const Iid& iid, const std::string& nick,
                                               const std::string& iq_id, const std::string& to_jid,
                                               const std::string& from_jid)
{
  IrcClient* irc = this->get_irc_client(iid.get_server());
  IrcChannel* chan = irc->get_channel(iid.get_local());
  if (!chan->joined)
    {
514
      this->xmpp.send_stanza_error("iq", to_jid, from_jid, iq_id, "cancel", "not-allowed",
515 516 517 518 519
                                    "", true);
      return;
    }
  if (chan->get_self()->nick != nick && !chan->find_user(nick))
    {
520
      this->xmpp.send_stanza_error("iq", to_jid, from_jid, iq_id, "cancel", "item-not-found",
521 522 523 524 525 526 527 528
                                    "Recipient not in room", true);
      return;
    }

  // The user is in the room, send it a direct PING
  this->send_irc_user_ping_request(iid.get_server(), nick, iq_id, to_jid, from_jid);
}

529 530 531 532
void Bridge::on_gateway_ping(const std::string& irc_hostname, const std::string& iq_id, const std::string& to_jid,
                             const std::string& from_jid)
{
  Jid jid(from_jid);
533
  if (irc_hostname.empty() || this->find_irc_client(irc_hostname))
534
    this->xmpp.send_iq_result(iq_id, to_jid, jid.local);
535
  else
536
    this->xmpp.send_stanza_error("iq", to_jid, from_jid, iq_id, "cancel", "service-unavailable",
537 538 539
                                  "", true);
}

540 541 542 543
void Bridge::send_irc_version_request(const std::string& irc_hostname, const std::string& target,
                                      const std::string& iq_id, const std::string& to_jid,
                                      const std::string& from_jid)
{
544
  Iid iid(target, irc_hostname, Iid::Type::User);
545 546 547
  this->send_private_message(iid, "\01VERSION\01");
  // TODO, add a timer to remove that waiting iq if the server does not
  // respond with a matching command before n seconds
548 549
  irc_responder_callback_t cb = [this, target, iq_id, to_jid, irc_hostname, from_jid](const std::string& hostname, const IrcMessage& message) -> bool
    {
550 551 552 553 554 555 556 557
      if (irc_hostname != hostname)
        return false;
      IrcUser user(message.prefix);
      if (message.command == "NOTICE" && user.nick == target &&
          message.arguments.size() >= 2 && message.arguments[1].substr(0, 9) == "\01VERSION ")
        {
          // remove the "\01VERSION " and the "\01" parts from the string
          const std::string version = message.arguments[1].substr(9, message.arguments[1].size() - 10);
558
          this->xmpp.send_version(iq_id, to_jid, from_jid, version);
559 560 561 562 563 564 565 566
          return true;
        }
      if (message.command == "401" && message.arguments.size() >= 2
          && message.arguments[1] == target)
        {
          std::string error_message = "No such nick";
          if (message.arguments.size() >= 3)
            error_message = message.arguments[2];
567
          this->xmpp.send_stanza_error("iq", to_jid, from_jid, iq_id, "cancel", "item-not-found",
568 569 570 571
                                        error_message, true);
          return true;
        }
      return false;
572 573
    };
  this->add_waiting_irc(std::move(cb));
574 575
}

louiz’'s avatar
louiz’ committed
576
void Bridge::send_message(const Iid& iid, const std::string& nick, const std::string& body, const bool muc)
louiz’'s avatar
louiz’ committed
577
{
578
  const auto encoding = in_encoding_for(*this, iid);
louiz’'s avatar
louiz’ committed
579
  if (muc)
580
    {
581
      for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
582 583 584 585 586
        {
          this->xmpp.send_muc_message(std::to_string(iid), nick,
                                      this->make_xmpp_body(body, encoding), this->user_jid + "/" + resource);
        }
    }
louiz’'s avatar
louiz’ committed
587
  else
588 589
    {
      std::string target = std::to_string(iid);
590
      const auto it = this->preferred_user_from.find(iid.get_local());
591 592
      if (it != this->preferred_user_from.end())
        {
593
          const auto chan_name = Iid(Jid(it->second).local, {}).get_local();
594 595 596
          for (const auto& resource: this->resources_in_chan[ChannelKey{chan_name, iid.get_server()}])
            this->xmpp.send_message(it->second, this->make_xmpp_body(body, encoding),
                                    this->user_jid + "/" + resource, "chat", true);
597
        }
598
      else
599
        {
600 601 602
          for (const auto& resource: this->resources_in_server[iid.get_server()])
            this->xmpp.send_message(std::to_string(iid), this->make_xmpp_body(body, encoding),
                                    this->user_jid + "/" + resource, "chat", false);
603
        }
604
    }
louiz’'s avatar
louiz’ committed
605 606
}

607 608 609
void Bridge::send_presence_error(const Iid& iid, const std::string& nick,
                                 const std::string& type, const std::string& condition,
                                 const std::string& error_code, const std::string& text)
610
{
611
  this->xmpp.send_presence_error(std::to_string(iid), nick, this->user_jid, type, condition, error_code, text);
612 613
}

614
void Bridge::send_muc_leave(Iid&& iid, std::string&& nick, const std::string& message, const bool self, const std::string& resource)
louiz’'s avatar
louiz’ committed
615
{
616 617 618 619 620 621 622
  if (!resource.empty())
    this->xmpp.send_muc_leave(std::to_string(iid), std::move(nick), this->make_xmpp_body(message), this->user_jid + "/" + resource,
                              self);
  else
    for (const auto& res: this->resources_in_chan[iid.to_tuple()])
      this->xmpp.send_muc_leave(std::to_string(iid), std::move(nick), this->make_xmpp_body(message), this->user_jid + "/" + res,
                                self);
623
  IrcClient* irc = this->find_irc_client(iid.get_server());
624 625
  if (irc && irc->number_of_joined_channels() == 0)
    irc->send_quit_command("");
louiz’'s avatar
louiz’ committed
626 627
}

628 629 630 631 632
void Bridge::send_nick_change(Iid&& iid,
                              const std::string& old_nick,
                              const std::string& new_nick,
                              const char user_mode,
                              const bool self)
louiz’'s avatar
louiz’ committed
633
{
634 635 636 637
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

638
  for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
639 640
    this->xmpp.send_nick_change(std::to_string(iid),
                                old_nick, new_nick, affiliation, role, this->user_jid + "/" + resource, self);
louiz’'s avatar
louiz’ committed
641 642
}

643 644
void Bridge::send_xmpp_message(const std::string& from, const std::string& author, const std::string& msg)
{
louiz’'s avatar
louiz’ committed
645 646
  std::string body;
  if (!author.empty())
647 648
    {
      IrcUser user(author);
louiz’'s avatar
louiz’ committed
649
      body = "\u000303"s + user.nick + (user.host.empty()?
louiz’'s avatar
louiz’ committed
650 651
                                        "\u0003: ":
                                        (" (\u000310" + user.host + "\u000303)\u0003: ")) + msg;
652
    }
louiz’'s avatar
louiz’ committed
653 654
  else
    body = msg;
655

656
  const auto encoding = in_encoding_for(*this, {from, this});
657 658 659 660
  for (const auto& resource: this->resources_in_server[from])
    {
        this->xmpp.send_message(from, this->make_xmpp_body(body, encoding), this->user_jid + "/" + resource, "chat");
    }
661 662
}

663 664 665 666 667 668 669 670 671 672
void Bridge::send_user_join(const std::string& hostname, const std::string& chan_name,
                            const IrcUser* user, const char user_mode, const bool self)
{
  for (const auto& resource: this->resources_in_chan[ChannelKey{chan_name, hostname}])
    this->send_user_join(hostname, chan_name, user, user_mode, self, resource);
}

void Bridge::send_user_join(const std::string& hostname, const std::string& chan_name,
                       const IrcUser* user, const char user_mode,
                       const bool self, const std::string& resource)
673
{
674 675 676 677
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

678 679 680 681
  std::string encoded_chan_name(chan_name);
  xep0106::encode(encoded_chan_name);

  this->xmpp.send_user_join(encoded_chan_name + utils::empty_if_fixed_server("%" + hostname), user->nick, user->host,
682
                            affiliation, role, this->user_jid + "/" + resource, self);
683 684
}

685
void Bridge::send_topic(const std::string& hostname, const std::string& chan_name, const std::string& topic, const std::string& who)
686
{
687 688
  for (const auto& resource: this->resources_in_chan[ChannelKey{chan_name, hostname}])
    {
689
      this->send_topic(hostname, chan_name, topic, who, resource);
690
    }
691
}
louiz’'s avatar
louiz’ committed
692

693 694 695 696
void Bridge::send_topic(const std::string& hostname, const std::string& chan_name,
                        const std::string& topic, const std::string& who,
                        const std::string& resource)
{
697 698
  std::string encoded_chan_name(chan_name);
  xep0106::encode(encoded_chan_name);
699
  const auto encoding = in_encoding_for(*this, {encoded_chan_name, hostname, Iid::Type::Channel});
700
  this->xmpp.send_topic(encoded_chan_name + utils::empty_if_fixed_server(
701 702 703 704
      "%" + hostname), this->make_xmpp_body(topic, encoding), this->user_jid + "/" + resource, who);

}

louiz’'s avatar
louiz’ committed
705 706
std::string Bridge::get_own_nick(const Iid& iid)
{
707
  IrcClient* irc = this->find_irc_client(iid.get_server());
louiz’'s avatar
louiz’ committed
708 709 710 711
  if (irc)
    return irc->get_own_nick();
  return "";
}
712

713
size_t Bridge::active_clients() const
louiz’'s avatar
louiz’ committed
714 715 716 717
{
  return this->irc_clients.size();
}

718 719
void Bridge::kick_muc_user(Iid&& iid, const std::string& target, const std::string& reason, const std::string& author)
{
720
  for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
721
    this->xmpp.kick_user(std::to_string(iid), target, reason, author, this->user_jid + "/" + resource);
722
}
723 724 725

void Bridge::send_nickname_conflict_error(const Iid& iid, const std::string& nickname)
{
726
  for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
727
    this->xmpp.send_presence_error(std::to_string(iid), nickname, this->user_jid + "/" + resource, "cancel", "conflict", "409", "");
728
}
729 730 731 732 733

void Bridge::send_affiliation_role_change(const Iid& iid, const std::string& target, const char mode)
{
  std::string role;
  std::string affiliation;
734 735

  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(mode);
736
  for (const auto& resource: this->resources_in_chan[iid.to_tuple()])
737
    this->xmpp.send_affiliation_role_change(std::to_string(iid), target, affiliation, role, this->user_jid + "/" + resource);
738
}
739 740 741

void Bridge::send_iq_version_request(const std::string& nick, const std::string& hostname)
{
742 743
  const auto resources = this->resources_in_server[hostname];
  if (resources.begin() != resources.end())
744
    this->xmpp.send_iq_version_request(utils::tolower(nick) + "%" + utils::empty_if_fixed_server(hostname), this->user_jid + "/" + *resources.begin());
745
}
746

747 748 749 750 751 752
void Bridge::send_xmpp_ping_request(const std::string& nick, const std::string& hostname,
                                    const std::string& id)
{
  // Use revstr because the forwarded ping to target XMPP user must not be
  // the same that the request iq, but we also need to get it back easily
  // (revstr again)
753 754 755
  // Forward to the first resource (arbitrary, based on the “order” of the std::set) only
  const auto resources = this->resources_in_server[hostname];
  if (resources.begin() != resources.end())
756
    this->xmpp.send_ping_request(utils::tolower(nick) + "%" + utils::empty_if_fixed_server(hostname), this->user_jid + "/" + *resources.begin(), utils::revstr(id));
757 758
}

759 760 761 762 763 764 765 766 767 768 769 770 771 772 773
void Bridge::set_preferred_from_jid(const std::string& nick, const std::string& full_jid)
{
  auto it = this->preferred_user_from.find(nick);
  if (it == this->preferred_user_from.end())
    this->preferred_user_from.emplace(nick, full_jid);
  else
    this->preferred_user_from[nick] = full_jid;
}

void Bridge::remove_preferred_from_jid(const std::string& nick)
{
  auto it = this->preferred_user_from.find(nick);
  if (it != this->preferred_user_from.end())
    this->preferred_user_from.erase(it);
}
774

775 776 777 778
void Bridge::remove_all_preferred_from_jid_of_room(const std::string& channel_name)
{
  for (auto it = this->preferred_user_from.begin(); it != this->preferred_user_from.end();)
    {
779
      Iid iid(Jid(it->second).local, {});
780 781 782 783 784 785 786
      if (iid.get_local() == channel_name)
        it = this->preferred_user_from.erase(it);
      else
        ++it;
    }
}

787
void Bridge::add_waiting_irc(irc_responder_callback_t&& callback)
788
{
789
  this->waiting_irc.emplace_back(std::move(callback));
790 791
}

792
void Bridge::trigger_on_irc_message(const std::string& irc_hostname, const IrcMessage& message)
793
{
794 795
  auto it = this->waiting_irc.begin();
  while (it != this->waiting_irc.end())
796 797
    {
      if ((*it)(irc_hostname, message) == true)
798
        it = this->waiting_irc.erase(it);
799 800 801 802
      else
        ++it;
    }
}
803 804 805 806 807

std::unordered_map<std::string, std::shared_ptr<IrcClient>>& Bridge::get_irc_clients()
{
  return this->irc_clients;
}
808

809 810 811 812 813 814 815 816
std::set<char> Bridge::get_chantypes(const std::string& hostname) const
{
  IrcClient* irc = this->find_irc_client(hostname);
  if (!irc)
    return {'#', '&'};
  return irc->get_chantypes();
}

817
void Bridge::add_resource_to_chan(const Bridge::ChannelKey& channel, const std::string& resource)
818 819 820 821 822 823 824 825
{
  auto it = this->resources_in_chan.find(channel);
  if (it == this->resources_in_chan.end())
    this->resources_in_chan[channel] = {resource};
  else
    it->second.insert(resource);
}

826
void Bridge::remove_resource_from_chan(const Bridge::ChannelKey& channel, const std::string& resource)
827 828 829 830 831 832 833 834 835 836
{
  auto it = this->resources_in_chan.find(channel);
  if (it != this->resources_in_chan.end())
    {
      it->second.erase(resource);
      if (it->second.empty())
        this->resources_in_chan.erase(it);
    }
}

837
bool Bridge::is_resource_in_chan(const Bridge::ChannelKey& channel, const std::string& resource) const
838 839 840 841 842 843 844
{
  auto it = this->resources_in_chan.find(channel);
  if (it != this->resources_in_chan.end())
    if (it->second.count(resource) == 1)
      return true;
  return false;
}
845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874

void Bridge::add_resource_to_server(const Bridge::IrcHostname& irc_hostname, const std::string& resource)
{
  auto it = this->resources_in_server.find(irc_hostname);
  if (it == this->resources_in_server.end())
    this->resources_in_server[irc_hostname] = {resource};
  else
    it->second.insert(resource);
}

void Bridge::remove_resource_from_server(const Bridge::IrcHostname& irc_hostname, const std::string& resource)
{
  auto it = this->resources_in_server.find(irc_hostname);
  if (it != this->resources_in_server.end())
    {
      it->second.erase(resource);
      if (it->second.empty())
        this->resources_in_server.erase(it);
    }
}

bool Bridge::is_resource_in_server(const Bridge::IrcHostname& irc_hostname, const std::string& resource) const
{
  auto it = this->resources_in_server.find(irc_hostname);
  if (it != this->resources_in_server.end())
    if (it->second.count(resource) == 1)
      return true;
  return false;
}

875 876 877 878 879 880 881 882
std::size_t Bridge::number_of_resources_in_chan(const Bridge::ChannelKey& channel_key) const
{
  auto it = this->resources_in_chan.find(channel_key);
  if (it == this->resources_in_chan.end())
    return 0;
  return it->second.size();
}

883 884 885 886 887 888 889 890 891 892 893
std::size_t Bridge::number_of_channels_the_resource_is_in(const std::string& irc_hostname, const std::string& resource) const
{
  std::size_t res = 0;
  for (auto pair: this->resources_in_chan)
    {
      if (std::get<0>(pair.first) == irc_hostname && pair.second.count(resource) != 0)
        res++;
    }
  return res;
}

894 895 896 897
void Bridge::generate_channel_join_for_resource(const Iid& iid, const std::string& resource)
{
  IrcClient* irc = this->get_irc_client(iid.get_server());
  IrcChannel* channel = irc->get_channel(iid.get_local());
898 899
  const auto self = channel->get_self();

900 901 902
  // Send the occupant list
  for (const auto& user: channel->get_users())
    {
903 904 905
      if (user->nick != self->nick)
        {
          log_debug(user->nick);
906
          this->send_user_join(iid.get_server(), iid.get_encoded_local(),
907 908 909
                               user.get(), user->get_most_significant_mode(irc->get_sorted_user_modes()),
                               false, resource);
        }
910
    }
911
  this->send_user_join(iid.get_server(), iid.get_encoded_local(),
912 913
                       self, self->get_most_significant_mode(irc->get_sorted_user_modes()),
                       true, resource);
914
  this->send_topic(iid.get_server(), iid.get_encoded_local(), channel->topic, channel->topic_author, resource);
915
}