bridge.cpp 16.1 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
336
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)
337
{
338
  this->xmpp->send_presence_error(std::to_string(iid), nick, this->user_jid, type, condition, error_code, text);
339
340
}

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

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

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

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

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

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

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

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

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

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

void Bridge::send_nickname_conflict_error(const Iid& iid, const std::string& nickname)
{
417
  this->xmpp->send_presence_error(std::to_string(iid), nickname, this->user_jid, "cancel", "conflict", "409", "");
418
}
419
420
421
422
423

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

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

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

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

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