1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
/*
* InspIRCd -- Internet Relay Chat Daemon
*
* Copyright (C) 2019 Sadie Powell <sadie@witchery.services>
*
* This file is part of InspIRCd. InspIRCd is free software: you can
* redistribute it and/or modify it under the terms of the GNU General Public
* License as published by the Free Software Foundation, version 2.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
* details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "inspircd.h"
#include "modules/cap.h"
#include "modules/ctctags.h"
class MsgIdTag : public ClientProtocol::MessageTagProvider
{
private:
Cap::Reference ctctagcap;
public:
MsgIdTag(Module* mod)
: ClientProtocol::MessageTagProvider(mod)
, ctctagcap(mod, "message-tags")
{
}
ModResult OnProcessTag(User* user, const std::string& tagname, std::string& tagvalue) CXX11_OVERRIDE
{
if (!irc::equals(tagname, "msgid"))
return MOD_RES_PASSTHRU;
// We should only allow this tag if it is added by a remote server.
return IS_LOCAL(user) ? MOD_RES_DENY : MOD_RES_ALLOW;
}
bool ShouldSendTag(LocalUser* user, const ClientProtocol::MessageTagData& tagdata) CXX11_OVERRIDE
{
return ctctagcap.get(user);
}
};
class MsgIdGenerator
{
uint64_t counter;
std::string strid;
const std::string::size_type baselen;
public:
MsgIdGenerator()
: counter(0)
, strid(InspIRCd::Format("%s~%lu~", ServerInstance->Config->GetSID().c_str(), ServerInstance->startup_time))
, baselen(strid.length())
{
}
const std::string& GetNext()
{
strid.erase(baselen);
strid.append(ConvToStr(counter++));
return strid;
}
};
class ModuleMsgId
: public Module
, public CTCTags::EventListener
{
private:
MsgIdTag tag;
MsgIdGenerator generator;
ModResult CopyMessageId(const ClientProtocol::TagMap& tags_in, ClientProtocol::TagMap& tags_out)
{
ClientProtocol::TagMap::const_iterator iter = tags_in.find("msgid");
if (iter != tags_in.end())
{
// If the remote server has sent a message identifier we should use that as
// identifiers need to be the same on all sides of the network.
tags_out.insert(*iter);
return MOD_RES_PASSTHRU;
}
// Otherwise, we can just create a new message identifier.
tags_out.insert(std::make_pair("msgid", ClientProtocol::MessageTagData(&tag, generator.GetNext())));
return MOD_RES_PASSTHRU;
}
public:
ModuleMsgId()
: CTCTags::EventListener(this)
, tag(this)
{
}
ModResult OnUserPreMessage(User* user, const MessageTarget& target, MessageDetails& details) CXX11_OVERRIDE
{
return CopyMessageId(details.tags_in, details.tags_out);
}
ModResult OnUserPreTagMessage(User* user, const MessageTarget& target, CTCTags::TagMessageDetails& details) CXX11_OVERRIDE
{
return CopyMessageId(details.tags_in, details.tags_out);
}
Version GetVersion() CXX11_OVERRIDE
{
return Version("Provides the msgid IRCv3 tag", VF_VENDOR);
}
};
MODULE_INIT(ModuleMsgId)
|
