bridge.cpp 16 KB
Newer Older
1
#include <bridge/bridge.hpp>
2
#include <bridge/colors.hpp>
3
#include <xmpp/xmpp_component.hpp>
louiz’'s avatar
louiz’ committed
4
#include <irc/irc_message.hpp>
5
#include <network/poller.hpp>
6
#include <utils/encoding.hpp>
7
#include <utils/tolower.hpp>
louiz’'s avatar
louiz’ committed
8
#include <logger/logger.hpp>
louiz’'s avatar
louiz’ committed
9
#include <utils/split.hpp>
louiz’'s avatar
louiz’ committed
10
#include <stdexcept>
louiz’'s avatar
louiz’ committed
11
#include <iostream>
12
#include <tuple>
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
Bridge::Bridge(const std::string& user_jid, XmppComponent* xmpp, std::shared_ptr<Poller> poller):
19 20 21 22 23 24 25 26 27 28
  user_jid(user_jid),
  xmpp(xmpp),
  poller(poller)
{
}

Bridge::~Bridge()
{
}

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

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

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

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

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

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

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

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

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

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

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

219 220
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)
221
{
louiz’'s avatar
louiz’ committed
222
  IrcClient* irc = this->get_irc_client(iid.get_server());
223
  if (irc)
224 225 226 227 228 229 230 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
    {
      irc->send_kick_command(iid.get_local(), target, reason);
      this->add_waiting_iq([this, target, iq_id, to_jid, iid](const std::string& irc_hostname, const IrcMessage& message){
          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())
                return false;
              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;
        });
    }
262 263
}

264 265
void Bridge::set_channel_topic(const Iid& iid, const std::string& subject)
{
louiz’'s avatar
louiz’ committed
266
  IrcClient* irc = this->get_irc_client(iid.get_server());
267
  if (irc)
louiz’'s avatar
louiz’ committed
268
    irc->send_topic_command(iid.get_local(), subject);
269 270
}

louiz’'s avatar
louiz’ committed
271 272 273 274 275 276 277
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");
}

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 313
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
  this->add_waiting_iq([this, target, iq_id, to_jid, irc_hostname, from_jid](const std::string& hostname, const IrcMessage& message){
      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;
    });
}


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

334 335
void Bridge::send_join_failed(const Iid& iid, const std::string& nick, const std::string& type, const std::string& condition, const std::string& text)
{
louiz’'s avatar
louiz’ committed
336
  this->xmpp->send_presence_error(std::to_string(iid), nick, this->user_jid, type, condition, text);
337 338
}

louiz’'s avatar
louiz’ committed
339
void Bridge::send_muc_leave(Iid&& iid, std::string&& nick, const std::string& message, const bool self)
louiz’'s avatar
louiz’ committed
340
{
louiz’'s avatar
louiz’ committed
341 342
  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());
343 344
  if (irc && irc->number_of_joined_channels() == 0)
    irc->send_quit_command("");
louiz’'s avatar
louiz’ committed
345 346
}

347 348 349 350 351
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
352
{
353 354 355 356
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

louiz’'s avatar
louiz’ committed
357
  this->xmpp->send_nick_change(std::to_string(iid),
358
                               old_nick, new_nick, affiliation, role, this->user_jid, self);
louiz’'s avatar
louiz’ committed
359 360
}

361 362
void Bridge::send_xmpp_message(const std::string& from, const std::string& author, const std::string& msg)
{
louiz’'s avatar
louiz’ committed
363 364
  std::string body;
  if (!author.empty())
365 366
    {
      IrcUser user(author);
louiz’'s avatar
louiz’ committed
367
      body = "\u000303"s + user.nick + (user.host.empty()?
louiz’'s avatar
louiz’ committed
368 369
                                        "\u0003: ":
                                        (" (\u000310" + user.host + "\u000303)\u0003: ")) + msg;
370
    }
louiz’'s avatar
louiz’ committed
371 372
  else
    body = msg;
373
  this->xmpp->send_message(from, this->make_xmpp_body(body), this->user_jid, "chat");
374 375
}

376 377
void Bridge::send_user_join(const std::string& hostname,
                            const std::string& chan_name,
378
                            const IrcUser* user,
379
                            const char user_mode,
380
                            const bool self)
381
{
382 383 384 385
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

386 387
  this->xmpp->send_user_join(chan_name + "%" + hostname, user->nick, user->host,
                             affiliation, role, this->user_jid, self);
388 389 390 391
}

void Bridge::send_topic(const std::string& hostname, const std::string& chan_name, const std::string topic)
{
392
  this->xmpp->send_topic(chan_name + "%" + hostname, this->make_xmpp_body(topic), this->user_jid);
393
}
louiz’'s avatar
louiz’ committed
394 395 396

std::string Bridge::get_own_nick(const Iid& iid)
{
louiz’'s avatar
louiz’ committed
397
  IrcClient* irc = this->get_irc_client(iid.get_server());
louiz’'s avatar
louiz’ committed
398 399 400 401
  if (irc)
    return irc->get_own_nick();
  return "";
}
402

403
size_t Bridge::active_clients() const
louiz’'s avatar
louiz’ committed
404 405 406 407
{
  return this->irc_clients.size();
}

408 409
void Bridge::kick_muc_user(Iid&& iid, const std::string& target, const std::string& reason, const std::string& author)
{
louiz’'s avatar
louiz’ committed
410
  this->xmpp->kick_user(std::to_string(iid), target, reason, author, this->user_jid);
411
}
412 413 414

void Bridge::send_nickname_conflict_error(const Iid& iid, const std::string& nickname)
{
louiz’'s avatar
louiz’ committed
415
  this->xmpp->send_nickname_conflict_error(std::to_string(iid), nickname, this->user_jid);
416
}
417 418 419 420 421

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

  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(mode);
louiz’'s avatar
louiz’ committed
424
  this->xmpp->send_affiliation_role_change(std::to_string(iid), target, affiliation, role, this->user_jid);
425
}
426 427 428

void Bridge::send_iq_version_request(const std::string& nick, const std::string& hostname)
{
louiz’'s avatar
louiz’ committed
429
  this->xmpp->send_iq_version_request(nick + "!" + hostname, this->user_jid);
430
}
431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446

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);
}
447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463

void Bridge::add_waiting_iq(iq_responder_callback_t&& callback)
{
  this->waiting_iq.emplace_back(std::move(callback));
}

void Bridge::trigger_response_iq(const std::string& irc_hostname, const IrcMessage& message)
{
  auto it = this->waiting_iq.begin();
  while (it != this->waiting_iq.end())
    {
      if ((*it)(irc_hostname, message) == true)
        it = this->waiting_iq.erase(it);
      else
        ++it;
    }
}