bridge.cpp 18.4 KB
Newer Older
1
#include <bridge/bridge.hpp>
2
#include <bridge/colors.hpp>
3
#include <xmpp/xmpp_component.hpp>
4
#include <xmpp/xmpp_stanza.hpp>
louiz’'s avatar
louiz’ committed
5
#include <irc/irc_message.hpp>
6
#include <network/poller.hpp>
7
#include <utils/encoding.hpp>
8
#include <utils/tolower.hpp>
louiz’'s avatar
louiz’ committed
9
#include <logger/logger.hpp>
10
#include <utils/revstr.hpp>
louiz’'s avatar
louiz’ committed
11
#include <utils/split.hpp>
12
#include <xmpp/jid.hpp>
louiz’'s avatar
louiz’ committed
13
#include <stdexcept>
louiz’'s avatar
louiz’ committed
14
#include <iostream>
15
#include <tuple>
louiz’'s avatar
louiz’ committed
16

louiz’'s avatar
louiz’ committed
17 18
using namespace std::string_literals;

19 20
static const char* action_prefix = "\01ACTION ";

21
Bridge::Bridge(const std::string& user_jid, XmppComponent* xmpp, std::shared_ptr<Poller> poller):
22 23 24 25 26 27 28 29 30 31
  user_jid(user_jid),
  xmpp(xmpp),
  poller(poller)
{
}

Bridge::~Bridge()
{
}

32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47
/**
 * Return the role and affiliation, corresponding to the given irc mode */
static std::tuple<std::string, std::string> get_role_affiliation_from_irc_mode(const char mode)
{
  if (mode == 'a' || mode == 'q')
    return std::make_tuple("moderator", "owner");
  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");
}

48
void Bridge::shutdown(const std::string& exit_message)
louiz’'s avatar
louiz’ committed
49 50 51
{
  for (auto it = this->irc_clients.begin(); it != this->irc_clients.end(); ++it)
  {
louiz’'s avatar
louiz’ committed
52 53
    it->second->send_quit_command(exit_message);
    it->second->leave_dummy_channel(exit_message);
louiz’'s avatar
louiz’ committed
54 55 56
  }
}

louiz’'s avatar
louiz’ committed
57 58 59 60 61 62
void Bridge::clean()
{
  auto it = this->irc_clients.begin();
  while (it != this->irc_clients.end())
  {
    IrcClient* client = it->second.get();
63
    if (!client->is_connected() && !client->is_connecting())
louiz’'s avatar
louiz’ committed
64 65 66 67 68 69
      it = this->irc_clients.erase(it);
    else
      ++it;
  }
}

70 71 72 73 74
const std::string& Bridge::get_jid() const
{
  return this->user_jid;
}

75
Xmpp::body Bridge::make_xmpp_body(const std::string& str)
76 77 78 79 80 81
{
  std::string res;
  if (utils::is_valid_utf8(str.c_str()))
    res = str;
  else
    res = utils::convert_to_utf8(str, "ISO-8859-1");
82
  return irc_format_to_xhtmlim(res);
83 84
}

85 86 87 88 89 90 91 92
IrcClient* Bridge::get_irc_client(const std::string& hostname, const std::string& username)
{
  try
    {
      return this->irc_clients.at(hostname).get();
    }
  catch (const std::out_of_range& exception)
    {
93
      this->irc_clients.emplace(hostname, std::make_shared<IrcClient>(this->poller, hostname, username, this));
94 95 96 97 98
      std::shared_ptr<IrcClient> irc = this->irc_clients.at(hostname);
      return irc.get();
    }
}

louiz’'s avatar
louiz’ committed
99 100 101 102 103 104 105 106 107 108 109 110
IrcClient* Bridge::get_irc_client(const std::string& hostname)
{
  try
    {
      return this->irc_clients.at(hostname).get();
    }
  catch (const std::out_of_range& exception)
    {
      return nullptr;
    }
}

louiz’'s avatar
louiz’ committed
111
bool Bridge::join_irc_channel(const Iid& iid, const std::string& username)
112
{
louiz’'s avatar
louiz’ committed
113 114
  IrcClient* irc = this->get_irc_client(iid.get_server(), username);
  if (iid.get_local().empty())
louiz’'s avatar
louiz’ committed
115 116 117 118 119 120 121 122 123
    { // 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
124
          const IrcMessage end_join_message(std::string(iid.get_server()), "366",
louiz’'s avatar
louiz’ committed
125 126 127 128 129
                                            {irc->get_nick(),
                                                "", "End of NAMES list"});
          irc->on_channel_completely_joined(end_join_message);
        }
      else
130
        {
louiz’'s avatar
louiz’ committed
131
          irc->get_dummy_channel().joining = true;
132 133
          irc->start();
        }
louiz’'s avatar
louiz’ committed
134 135
      return true;
    }
louiz’'s avatar
louiz’ committed
136
  if (irc->is_channel_joined(iid.get_local()) == false)
louiz’'s avatar
louiz’ committed
137
    {
louiz’'s avatar
louiz’ committed
138
      irc->send_join_command(iid.get_local());
louiz’'s avatar
louiz’ committed
139 140 141
      return true;
    }
  return false;
142 143
}

louiz’'s avatar
louiz’ committed
144 145
void Bridge::send_channel_message(const Iid& iid, const std::string& body)
{
louiz’'s avatar
louiz’ committed
146
  if (iid.get_local().empty() || iid.get_server().empty())
louiz’'s avatar
louiz’ committed
147
    {
louiz’'s avatar
louiz’ committed
148
      log_warning("Cannot send message to channel: [" << iid.get_local() << "] on server [" << iid.get_server() << "]");
louiz’'s avatar
louiz’ committed
149 150
      return;
    }
louiz’'s avatar
louiz’ committed
151
  IrcClient* irc = this->get_irc_client(iid.get_server());
louiz’'s avatar
louiz’ committed
152 153
  if (!irc)
    {
louiz’'s avatar
louiz’ committed
154
      log_warning("Cannot send message: no client exist for server " << iid.get_server());
louiz’'s avatar
louiz’ committed
155 156
      return;
    }
157 158 159 160 161 162 163 164 165 166 167 168
  // 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
169
    {
170 171 172
      if (line.substr(0, 6) == "/mode ")
        {
          std::vector<std::string> args = utils::split(line.substr(6), ' ', false);
louiz’'s avatar
louiz’ committed
173
          irc->send_mode_command(iid.get_local(), args);
174 175 176 177
          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
178
        irc->send_channel_message(iid.get_local(), action_prefix + line.substr(4) + "\01");
179
      else
louiz’'s avatar
louiz’ committed
180 181
        irc->send_channel_message(iid.get_local(), line);
      this->xmpp->send_muc_message(std::to_string(iid), irc->get_own_nick(),
182
                                   this->make_xmpp_body(line), this->user_jid);
louiz’'s avatar
louiz’ committed
183
    }
louiz’'s avatar
louiz’ committed
184 185
}

louiz’'s avatar
louiz’ committed
186
void Bridge::send_private_message(const Iid& iid, const std::string& body, const std::string& type)
louiz’'s avatar
louiz’ committed
187
{
louiz’'s avatar
louiz’ committed
188
  if (iid.get_local().empty() || iid.get_server().empty())
louiz’'s avatar
louiz’ committed
189
    return ;
louiz’'s avatar
louiz’ committed
190
  IrcClient* irc = this->get_irc_client(iid.get_server());
191 192
  if (!irc)
    {
louiz’'s avatar
louiz’ committed
193
      log_warning("Cannot send message: no client exist for server " << iid.get_server());
194 195 196 197 198 199 200 201
      return;
    }
  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
202
        irc->send_private_message(iid.get_local(), action_prefix + line.substr(4) + "\01", type);
203
      else
louiz’'s avatar
louiz’ committed
204
        irc->send_private_message(iid.get_local(), line, type);
205
    }
louiz’'s avatar
louiz’ committed
206 207
}

louiz’'s avatar
louiz’ committed
208 209
void Bridge::leave_irc_channel(Iid&& iid, std::string&& status_message)
{
louiz’'s avatar
louiz’ committed
210
  IrcClient* irc = this->get_irc_client(iid.get_server());
louiz’'s avatar
louiz’ committed
211
  if (irc)
louiz’'s avatar
louiz’ committed
212
    irc->send_part_command(iid.get_local(), status_message);
louiz’'s avatar
louiz’ committed
213 214
}

louiz’'s avatar
louiz’ committed
215 216
void Bridge::send_irc_nick_change(const Iid& iid, const std::string& new_nick)
{
louiz’'s avatar
louiz’ committed
217
  IrcClient* irc = this->get_irc_client(iid.get_server());
louiz’'s avatar
louiz’ committed
218 219 220 221
  if (irc)
    irc->send_nick_command(new_nick);
}

222 223
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)
224
{
louiz’'s avatar
louiz’ committed
225
  IrcClient* irc = this->get_irc_client(iid.get_server());
226 227 228 229 230 231 232

  if (!irc)
    return;

  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
233
    {
234 235 236 237 238 239 240
      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())
241
            return false;
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
          this->xmpp->send_iq_result(iq_id, to_jid, std::to_string(iid));
        }
      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];
          this->xmpp->send_stanza_error("iq", to_jid, std::to_string(iid), iq_id, "cancel", "item-not-found",
                                        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];
          this->xmpp->send_stanza_error("iq", to_jid, std::to_string(iid), iq_id, "cancel", "not-allowed",
                                        error_message, false);
        }
      return true;
    };
  this->add_waiting_irc(std::move(cb));
269 270
}

271 272
void Bridge::set_channel_topic(const Iid& iid, const std::string& subject)
{
louiz’'s avatar
louiz’ committed
273
  IrcClient* irc = this->get_irc_client(iid.get_server());
274
  if (irc)
louiz’'s avatar
louiz’ committed
275
    irc->send_topic_command(iid.get_local(), subject);
276 277
}

louiz’'s avatar
louiz’ committed
278 279 280 281 282 283 284
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");
}

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 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328
void Bridge::send_irc_ping_result(const Iid& iid, const std::string& id)
{
  this->send_private_message(iid, "\01PING "s + utils::revstr(id), "NOTICE");
}

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)
{
  Iid iid(nick + "!" + irc_hostname);
  this->send_private_message(iid, "\01PING " + iq_id + "\01");

  irc_responder_callback_t cb = [this, nick, iq_id, to_jid, irc_hostname, from_jid](const std::string& hostname, const IrcMessage& message) -> bool
    {
      if (irc_hostname != hostname)
        return false;
      IrcUser user(message.prefix);
      const std::string body = message.arguments[1];
      if (message.command == "NOTICE" && user.nick == nick &&
          message.arguments.size() >= 2 && body.substr(0, 6) == "\01PING ")
        {
          const std::string id = body.substr(6, body.size() - 6);
          if (id != iq_id)
            return false;
          Jid jid(from_jid);
          this->xmpp->send_iq_result(iq_id, to_jid, jid.local);
          return true;
        }
      if (message.command == "401" && message.arguments.size() >= 2
          && message.arguments[1] == nick)
        {
          std::string error_message = "No such nick";
          if (message.arguments.size() >= 3)
            error_message = message.arguments[2];
          this->xmpp->send_stanza_error("iq", to_jid, from_jid, iq_id, "cancel", "service-unavailable",
                                        error_message, true);
          return true;
        }

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

329 330 331 332 333 334 335 336 337
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)
{
  Iid iid(target + "!" + irc_hostname);
  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
338 339
  irc_responder_callback_t cb = [this, target, iq_id, to_jid, irc_hostname, from_jid](const std::string& hostname, const IrcMessage& message) -> bool
    {
340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361
      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);
          this->xmpp->send_version(iq_id, to_jid, from_jid, version);
          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];
          this->xmpp->send_stanza_error("iq", to_jid, from_jid, iq_id, "cancel", "item-not-found",
                                        error_message, true);
          return true;
        }
      return false;
362 363
    };
  this->add_waiting_irc(std::move(cb));
364 365
}

louiz’'s avatar
louiz’ committed
366
void Bridge::send_message(const Iid& iid, const std::string& nick, const std::string& body, const bool muc)
louiz’'s avatar
louiz’ committed
367
{
louiz’'s avatar
louiz’ committed
368
  if (muc)
louiz’'s avatar
louiz’ committed
369
    this->xmpp->send_muc_message(std::to_string(iid), nick,
370
                                 this->make_xmpp_body(body), this->user_jid);
louiz’'s avatar
louiz’ committed
371
  else
372 373 374 375 376 377 378 379 380 381 382 383
    {
      std::string target = std::to_string(iid);
      bool fulljid = false;
      auto it = this->preferred_user_from.find(iid.get_local());
      if (it != this->preferred_user_from.end())
        {
          target = it->second;
          fulljid = true;
        }
      this->xmpp->send_message(target, this->make_xmpp_body(body),
                               this->user_jid, "chat", fulljid);
    }
louiz’'s avatar
louiz’ committed
384 385
}

386 387 388
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)
389
{
390
  this->xmpp->send_presence_error(std::to_string(iid), nick, this->user_jid, type, condition, error_code, text);
391 392
}

louiz’'s avatar
louiz’ committed
393
void Bridge::send_muc_leave(Iid&& iid, std::string&& nick, const std::string& message, const bool self)
louiz’'s avatar
louiz’ committed
394
{
louiz’'s avatar
louiz’ committed
395 396
  this->xmpp->send_muc_leave(std::to_string(iid), std::move(nick), this->make_xmpp_body(message), this->user_jid, self);
  IrcClient* irc = this->get_irc_client(iid.get_server());
397 398
  if (irc && irc->number_of_joined_channels() == 0)
    irc->send_quit_command("");
louiz’'s avatar
louiz’ committed
399 400
}

401 402 403 404 405
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
406
{
407 408 409 410
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

louiz’'s avatar
louiz’ committed
411
  this->xmpp->send_nick_change(std::to_string(iid),
412
                               old_nick, new_nick, affiliation, role, this->user_jid, self);
louiz’'s avatar
louiz’ committed
413 414
}

415 416
void Bridge::send_xmpp_message(const std::string& from, const std::string& author, const std::string& msg)
{
louiz’'s avatar
louiz’ committed
417 418
  std::string body;
  if (!author.empty())
419 420
    {
      IrcUser user(author);
louiz’'s avatar
louiz’ committed
421
      body = "\u000303"s + user.nick + (user.host.empty()?
louiz’'s avatar
louiz’ committed
422 423
                                        "\u0003: ":
                                        (" (\u000310" + user.host + "\u000303)\u0003: ")) + msg;
424
    }
louiz’'s avatar
louiz’ committed
425 426
  else
    body = msg;
427
  this->xmpp->send_message(from, this->make_xmpp_body(body), this->user_jid, "chat");
428 429
}

430 431
void Bridge::send_user_join(const std::string& hostname,
                            const std::string& chan_name,
432
                            const IrcUser* user,
433
                            const char user_mode,
434
                            const bool self)
435
{
436 437 438 439
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

440 441
  this->xmpp->send_user_join(chan_name + "%" + hostname, user->nick, user->host,
                             affiliation, role, this->user_jid, self);
442 443 444 445
}

void Bridge::send_topic(const std::string& hostname, const std::string& chan_name, const std::string topic)
{
446
  this->xmpp->send_topic(chan_name + "%" + hostname, this->make_xmpp_body(topic), this->user_jid);
447
}
louiz’'s avatar
louiz’ committed
448 449 450

std::string Bridge::get_own_nick(const Iid& iid)
{
louiz’'s avatar
louiz’ committed
451
  IrcClient* irc = this->get_irc_client(iid.get_server());
louiz’'s avatar
louiz’ committed
452 453 454 455
  if (irc)
    return irc->get_own_nick();
  return "";
}
456

457
size_t Bridge::active_clients() const
louiz’'s avatar
louiz’ committed
458 459 460 461
{
  return this->irc_clients.size();
}

462 463
void Bridge::kick_muc_user(Iid&& iid, const std::string& target, const std::string& reason, const std::string& author)
{
louiz’'s avatar
louiz’ committed
464
  this->xmpp->kick_user(std::to_string(iid), target, reason, author, this->user_jid);
465
}
466 467 468

void Bridge::send_nickname_conflict_error(const Iid& iid, const std::string& nickname)
{
469
  this->xmpp->send_presence_error(std::to_string(iid), nickname, this->user_jid, "cancel", "conflict", "409", "");
470
}
471 472 473 474 475

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

  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(mode);
louiz’'s avatar
louiz’ committed
478
  this->xmpp->send_affiliation_role_change(std::to_string(iid), target, affiliation, role, this->user_jid);
479
}
480 481 482

void Bridge::send_iq_version_request(const std::string& nick, const std::string& hostname)
{
louiz’'s avatar
louiz’ committed
483
  this->xmpp->send_iq_version_request(nick + "!" + hostname, this->user_jid);
484
}
485

486 487 488 489 490 491 492 493 494
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)
  this->xmpp->send_ping_request(nick + "!" + hostname, this->user_jid, utils::revstr(id));
}

495 496 497 498 499 500 501 502 503 504 505 506 507 508 509
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);
}
510

511
void Bridge::add_waiting_irc(irc_responder_callback_t&& callback)
512
{
513
  this->waiting_irc.emplace_back(std::move(callback));
514 515
}

516
void Bridge::trigger_on_irc_message(const std::string& irc_hostname, const IrcMessage& message)
517
{
518 519
  auto it = this->waiting_irc.begin();
  while (it != this->waiting_irc.end())
520 521
    {
      if ((*it)(irc_hostname, message) == true)
522
        it = this->waiting_irc.erase(it);
523 524 525 526
      else
        ++it;
    }
}