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