bridge.cpp 16.2 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>
louiz’'s avatar
louiz’ committed
10
#include <utils/split.hpp>
louiz’'s avatar
louiz’ committed
11
#include <stdexcept>
louiz’'s avatar
louiz’ committed
12
#include <iostream>
13
#include <tuple>
louiz’'s avatar
louiz’ committed
14

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

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

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

Bridge::~Bridge()
{
}

30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
/**
 * 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");
}

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

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

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

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

83 84 85 86 87 88 89 90
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)
    {
91
      this->irc_clients.emplace(hostname, std::make_shared<IrcClient>(this->poller, hostname, username, this));
92 93 94 95 96
      std::shared_ptr<IrcClient> irc = this->irc_clients.at(hostname);
      return irc.get();
    }
}

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

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

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

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

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

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

  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
231
    {
232 233 234 235 236 237 238
      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())
239
            return false;
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
          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));
267 268
}

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

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

283 284 285 286 287 288 289 290 291
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
292 293
  irc_responder_callback_t cb = [this, target, iq_id, to_jid, irc_hostname, from_jid](const std::string& hostname, const IrcMessage& message) -> bool
    {
294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315
      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;
316 317
    };
  this->add_waiting_irc(std::move(cb));
318 319
}

louiz’'s avatar
louiz’ committed
320
void Bridge::send_message(const Iid& iid, const std::string& nick, const std::string& body, const bool muc)
louiz’'s avatar
louiz’ committed
321
{
louiz’'s avatar
louiz’ committed
322
  if (muc)
louiz’'s avatar
louiz’ committed
323
    this->xmpp->send_muc_message(std::to_string(iid), nick,
324
                                 this->make_xmpp_body(body), this->user_jid);
louiz’'s avatar
louiz’ committed
325
  else
326 327 328 329 330 331 332 333 334 335 336 337
    {
      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
338 339
}

340 341 342
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)
343
{
344
  this->xmpp->send_presence_error(std::to_string(iid), nick, this->user_jid, type, condition, error_code, text);
345 346
}

louiz’'s avatar
louiz’ committed
347
void Bridge::send_muc_leave(Iid&& iid, std::string&& nick, const std::string& message, const bool self)
louiz’'s avatar
louiz’ committed
348
{
louiz’'s avatar
louiz’ committed
349 350
  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());
351 352
  if (irc && irc->number_of_joined_channels() == 0)
    irc->send_quit_command("");
louiz’'s avatar
louiz’ committed
353 354
}

355 356 357 358 359
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
360
{
361 362 363 364
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

louiz’'s avatar
louiz’ committed
365
  this->xmpp->send_nick_change(std::to_string(iid),
366
                               old_nick, new_nick, affiliation, role, this->user_jid, self);
louiz’'s avatar
louiz’ committed
367 368
}

369 370
void Bridge::send_xmpp_message(const std::string& from, const std::string& author, const std::string& msg)
{
louiz’'s avatar
louiz’ committed
371 372
  std::string body;
  if (!author.empty())
373 374
    {
      IrcUser user(author);
louiz’'s avatar
louiz’ committed
375
      body = "\u000303"s + user.nick + (user.host.empty()?
louiz’'s avatar
louiz’ committed
376 377
                                        "\u0003: ":
                                        (" (\u000310" + user.host + "\u000303)\u0003: ")) + msg;
378
    }
louiz’'s avatar
louiz’ committed
379 380
  else
    body = msg;
381
  this->xmpp->send_message(from, this->make_xmpp_body(body), this->user_jid, "chat");
382 383
}

384 385
void Bridge::send_user_join(const std::string& hostname,
                            const std::string& chan_name,
386
                            const IrcUser* user,
387
                            const char user_mode,
388
                            const bool self)
389
{
390 391 392 393
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

394 395
  this->xmpp->send_user_join(chan_name + "%" + hostname, user->nick, user->host,
                             affiliation, role, this->user_jid, self);
396 397 398 399
}

void Bridge::send_topic(const std::string& hostname, const std::string& chan_name, const std::string topic)
{
400
  this->xmpp->send_topic(chan_name + "%" + hostname, this->make_xmpp_body(topic), this->user_jid);
401
}
louiz’'s avatar
louiz’ committed
402 403 404

std::string Bridge::get_own_nick(const Iid& iid)
{
louiz’'s avatar
louiz’ committed
405
  IrcClient* irc = this->get_irc_client(iid.get_server());
louiz’'s avatar
louiz’ committed
406 407 408 409
  if (irc)
    return irc->get_own_nick();
  return "";
}
410

411
size_t Bridge::active_clients() const
louiz’'s avatar
louiz’ committed
412 413 414 415
{
  return this->irc_clients.size();
}

416 417
void Bridge::kick_muc_user(Iid&& iid, const std::string& target, const std::string& reason, const std::string& author)
{
louiz’'s avatar
louiz’ committed
418
  this->xmpp->kick_user(std::to_string(iid), target, reason, author, this->user_jid);
419
}
420 421 422

void Bridge::send_nickname_conflict_error(const Iid& iid, const std::string& nickname)
{
423
  this->xmpp->send_presence_error(std::to_string(iid), nickname, this->user_jid, "cancel", "conflict", "409", "");
424
}
425 426 427 428 429

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

  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(mode);
louiz’'s avatar
louiz’ committed
432
  this->xmpp->send_affiliation_role_change(std::to_string(iid), target, affiliation, role, this->user_jid);
433
}
434 435 436

void Bridge::send_iq_version_request(const std::string& nick, const std::string& hostname)
{
louiz’'s avatar
louiz’ committed
437
  this->xmpp->send_iq_version_request(nick + "!" + hostname, this->user_jid);
438
}
439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454

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);
}
455

456
void Bridge::add_waiting_irc(irc_responder_callback_t&& callback)
457
{
458
  this->waiting_irc.emplace_back(std::move(callback));
459 460
}

461
void Bridge::trigger_on_irc_message(const std::string& irc_hostname, const IrcMessage& message)
462
{
463 464
  auto it = this->waiting_irc.begin();
  while (it != this->waiting_irc.end())
465 466
    {
      if ((*it)(irc_hostname, message) == true)
467
        it = this->waiting_irc.erase(it);
468 469 470 471
      else
        ++it;
    }
}