bridge.cpp 14.4 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");
}

louiz’'s avatar
louiz’ committed
278
void Bridge::send_message(const Iid& iid, const std::string& nick, const std::string& body, const bool muc)
louiz’'s avatar
louiz’ committed
279
{
louiz’'s avatar
louiz’ committed
280
  if (muc)
louiz’'s avatar
louiz’ committed
281
    this->xmpp->send_muc_message(std::to_string(iid), nick,
282
                                 this->make_xmpp_body(body), this->user_jid);
louiz’'s avatar
louiz’ committed
283
  else
284
285
286
287
288
289
290
291
292
293
294
295
    {
      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
296
297
}

298
299
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
300
  this->xmpp->send_presence_error(std::to_string(iid), nick, this->user_jid, type, condition, text);
301
302
}

louiz’'s avatar
louiz’ committed
303
void Bridge::send_muc_leave(Iid&& iid, std::string&& nick, const std::string& message, const bool self)
louiz’'s avatar
louiz’ committed
304
{
louiz’'s avatar
louiz’ committed
305
306
  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());
307
308
  if (irc && irc->number_of_joined_channels() == 0)
    irc->send_quit_command("");
louiz’'s avatar
louiz’ committed
309
310
}

311
312
313
314
315
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
316
{
317
318
319
320
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

louiz’'s avatar
louiz’ committed
321
  this->xmpp->send_nick_change(std::to_string(iid),
322
                               old_nick, new_nick, affiliation, role, this->user_jid, self);
louiz’'s avatar
louiz’ committed
323
324
}

325
326
void Bridge::send_xmpp_message(const std::string& from, const std::string& author, const std::string& msg)
{
louiz’'s avatar
louiz’ committed
327
328
  std::string body;
  if (!author.empty())
329
330
    {
      IrcUser user(author);
louiz’'s avatar
louiz’ committed
331
      body = "\u000303"s + user.nick + (user.host.empty()?
louiz’'s avatar
louiz’ committed
332
333
                                        "\u0003: ":
                                        (" (\u000310" + user.host + "\u000303)\u0003: ")) + msg;
334
    }
louiz’'s avatar
louiz’ committed
335
336
  else
    body = msg;
337
  this->xmpp->send_message(from, this->make_xmpp_body(body), this->user_jid, "chat");
338
339
}

340
341
void Bridge::send_user_join(const std::string& hostname,
                            const std::string& chan_name,
342
                            const IrcUser* user,
343
                            const char user_mode,
344
                            const bool self)
345
{
346
347
348
349
  std::string affiliation;
  std::string role;
  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(user_mode);

350
351
  this->xmpp->send_user_join(chan_name + "%" + hostname, user->nick, user->host,
                             affiliation, role, this->user_jid, self);
352
353
354
355
}

void Bridge::send_topic(const std::string& hostname, const std::string& chan_name, const std::string topic)
{
356
  this->xmpp->send_topic(chan_name + "%" + hostname, this->make_xmpp_body(topic), this->user_jid);
357
}
louiz’'s avatar
louiz’ committed
358
359
360

std::string Bridge::get_own_nick(const Iid& iid)
{
louiz’'s avatar
louiz’ committed
361
  IrcClient* irc = this->get_irc_client(iid.get_server());
louiz’'s avatar
louiz’ committed
362
363
364
365
  if (irc)
    return irc->get_own_nick();
  return "";
}
366

367
size_t Bridge::active_clients() const
louiz’'s avatar
louiz’ committed
368
369
370
371
{
  return this->irc_clients.size();
}

372
373
void Bridge::kick_muc_user(Iid&& iid, const std::string& target, const std::string& reason, const std::string& author)
{
louiz’'s avatar
louiz’ committed
374
  this->xmpp->kick_user(std::to_string(iid), target, reason, author, this->user_jid);
375
}
376
377
378

void Bridge::send_nickname_conflict_error(const Iid& iid, const std::string& nickname)
{
louiz’'s avatar
louiz’ committed
379
  this->xmpp->send_nickname_conflict_error(std::to_string(iid), nickname, this->user_jid);
380
}
381
382
383
384
385

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

  std::tie(role, affiliation) = get_role_affiliation_from_irc_mode(mode);
louiz’'s avatar
louiz’ committed
388
  this->xmpp->send_affiliation_role_change(std::to_string(iid), target, affiliation, role, this->user_jid);
389
}
390
391
392

void Bridge::send_iq_version_request(const std::string& nick, const std::string& hostname)
{
louiz’'s avatar
louiz’ committed
393
  this->xmpp->send_iq_version_request(nick + "!" + hostname, this->user_jid);
394
}
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410

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);
}
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427

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