-
Notifications
You must be signed in to change notification settings - Fork 4
/
clone_scanner.py
447 lines (398 loc) · 21 KB
/
clone_scanner.py
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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
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
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
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
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
# -*- coding: utf-8 -*-
#
# Clone Scanner, version 1.3 for WeeChat version 0.3
# Latest development version: https://github.com/FiXato/weechat_scripts
#
# A Clone Scanner that can manually scan channels and
# automatically scans joins for users on the channel
# with multiple nicknames from the same host.
#
# Upon join by a user, the user's host is compared to the infolist of
# already connected users to see if they are already online from
# another nickname. If the user is a clone, it will report it.
# With the '/clone_scanner scan' command you can manually scan a chan.
#
# See /set plugins.var.python.clone_scanner.* for all possible options
# Use the brilliant iset.pl plugin (/weeget install iset) to see what they do
# Or check the sourcecode below.
#
# Example output for an on-join scan result:
# 21:32:46 ▬▬▶ FiXato_Odie ([email protected]) has joined #lounge
# 21:32:46 FiXato_Odie is already on the channel as [email protected] and [email protected]
#
# Example output for a manual scan:
# 21:34:44 fixato.net is online from 3 nicks:
# 21:34:44 - [email protected]
# 21:34:44 - [email protected]
# 21:34:44 - [email protected]
#
## History:
### 2011-09-11: FiXato:
#
# * version 0.1: initial release.
# * Added an on-join clone scan. Any user that joins a channel will be
# matched against users already on the channel.
#
# * version 0.2: manual clone scan
# * Added a manual clone scan via /clone_scanner scan
# you can specify a target channel with:
# /clone_scanner scan #myChannelOnCurrentServer
# or:
# /clone_scanner scan Freenode.#myChanOnSpecifiedNetwork
# * Added completion
#
### 2011-09-12: FiXato:
#
# * version 0.3: Refactor galore
# * Refactored some code. Codebase should be DRYer and clearer now.
# * Manual scan report lists by host instead of nick now.
# * Case-insensitive host-matching
# * Bugfixed the infolist memleak.
# * on-join scanner works again
# * Output examples added to the comments
#
### 2011-09-19
# * version 0.4: Option galore
# * Case-insensitive buffer lookup fix.
# * Made most messages optional through settings.
# * Made on-join alert and clone report key a bit more configurable.
# * Added formatting options for on-join alerts.
# * Added format_message helper method that accepts multiple whitespace-separated weechat.color() options.
# * Added formatting options for join messages
# * Added formatting options for clone reports
# * Added format_from_config helper method that reads the given formatting key from the config
#
# * version 0.5: cs_buffer refactoring
# * dropping the manual cs_create_buffer call in favor for a cs_get_buffer() method
#
### 2012-02-10: FiXato:
#
# * version 0.6: Stop shoving that buffer in my face!
# * The clone_scanner buffer should no longer pop up by itself when you load the script.
# It should only pop up now when you actually a line needs to show up in the buffer.
#
# * version 0.7: .. but please pop it up in my current window when I ask for it
# * Added setting plugins.var.python.clone_scanner.autofocus
# This will autofocus the clone_scanner buffer in the current window if another window isn't
# already showing it, and of course only when the clone_scanner buffer is triggered
#
### 2012-02-10: FiXato:
#
# * version 0.8: .. and only when it is first created..
# * Prevents the buffer from being focused every time there is activity in it and not being shown in a window.
#
### 2012-04-01: FiXato:
#
# * version 0.9: Hurrah for bouncers...
# * Added the option plugins.var.python.clone_scanner.compare_idents
# Set it to 'on' if you don't want people with different idents to be marked as clones.
# Useful on channels with bouncers.
#
### 2012-04-02: FiXato:
#
# * version 1.0: Bugfix
# * Fixed the on-join scanner bug introduced by the 0.9 release.
# I was not properly comparing the new [email protected] key in all places yet.
# Should really have tested this better ><
#
### 2012-04-03: FiXato:
#
# * version 1.1: Stop being so sensitive!
# * Continuing to fix the on-join scanner bugs introduced by the 0.9 release.
# The [email protected] dict key wasn't being lowercased for comparison in the on-join scan.
#
# * version 1.2: So shameless!
# * Added shameless advertising for my script through /clone_scanner advertise
#
### 2013-04-09: FiXato:
# * version 1.3: Such a killer rabbit
# * Thanks to Curtis Sorensen aka killerrabbit clone_scanner.py now supports:
# * local channels (&-prefixed)
# * nameless channels (just # or &)
#
## Acknowledgements:
# * Sebastien "Flashcode" Helleu, for developing the kick-ass chat/IRC
# client WeeChat
# * ArZa, whose kickban.pl script helped me get started with using the
# infolist results.
# * LayBot, for requesting the ident comparison
# * Curtis "killerrabbit" Sorensen, for sending in two pull-requests,
# adding support for local and nameless channels.
#
## TODO:
# - Add option to enable/disable public clone reporting aka msg channels
# - Add option to enable/disable scanning on certain channels/networks
# - Add cross-channel clone scan
# - Add cross-server clone scan
#
## Copyright (c) 2011-2012 Filip H.F. "FiXato" Slagter,
# <FiXato [at] Gmail [dot] com>
# http://google.com/profiles/FiXato
#
# Permission is hereby granted, free of charge, to any person obtaining
# a copy of this software and associated documentation files (the
# "Software"), to deal in the Software without restriction, including
# without limitation the rights to use, copy, modify, merge, publish,
# distribute, sublicense, and/or sell copies of the Software, and to
# permit persons to whom the Software is furnished to do so, subject to
# the following conditions:
#
# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
# NON-INFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
#
SCRIPT_NAME = "clone_scanner"
SCRIPT_AUTHOR = "Filip H.F. 'FiXato' Slagter <fixato [at] gmail [dot] com>"
SCRIPT_VERSION = "1.3"
SCRIPT_LICENSE = "MIT"
SCRIPT_DESC = "A Clone Scanner that can manually scan channels and automatically scans joins for users on the channel with multiple nicknames from the same host."
SCRIPT_COMMAND = "clone_scanner"
SCRIPT_CLOSE_CB = "cs_close_cb"
import_ok = True
try:
import weechat
except ImportError:
print "This script must be run under WeeChat."
import_ok = False
import re
cs_buffer = None
cs_settings = (
("autofocus", "on", "Focus the clone_scanner buffer in the current window if it isn't already displayed by a window."),
("compare_idents", "off", "Match against [email protected] instead of just the hostname. Useful if you don't want different people from bouncers marked as clones"),
("display_join_messages", "off", "Display all joins in the clone_scanner buffer"),
("display_onjoin_alert_clone_buffer", "on", "Display an on-join clone alert in the clone_scanner buffer"),
("display_onjoin_alert_target_buffer", "on", "Display an on-join clone alert in the buffer where the clone was detected"),
("display_onjoin_alert_current_buffer", "off", "Display an on-join clone alert in the current buffer"),
("display_scan_report_clone_buffer", "on", "Display manual scan reports in the clone buffer"),
("display_scan_report_target_buffer", "off", "Display manual scan reports in the buffer of the scanned channel"),
("display_scan_report_current_buffer", "on", "Display manual scan reports in the current buffer"),
("clone_report_key", "mask", "Which 'key' to display in the clone report: 'mask' for full hostmasks, or 'nick' for nicks"),
("clone_onjoin_alert_key", "mask", "Which 'key' to display in the on-join alerts: 'mask' for full hostmasks, or 'nick' for nicks"),
("colors.onjoin_alert.message", "red", "The on-join clone alert's message colour. Formats are space separated."),
("colors.onjoin_alert.nick", "bold red", "The on-join clone alert's nick colour. Formats are space separated. Note: if you have colorize_nicks, this option might not work as expected."),
("colors.onjoin_alert.channel", "red", "The on-join clone alert's channel colour. Formats are space separated."),
("colors.onjoin_alert.matches", "bold red", "The on-join clone alert's matches (masks or nicks) colour. Formats are space separated. Note: if you have colorize_nicks, this option might not work as expected."),
("colors.join_messages.message", "chat", "The base colour for the join messages."),
("colors.join_messages.nick", "bold", "The colour for the 'nick'-part of the join messages. Note: if you have colorize_nicks, this option might not always work as expected."),
("colors.join_messages.identhost", "chat", "The colour for the 'ident@host'-part of the join messages."),
("colors.join_messages.channel", "bold", "The colour for the 'channel'-part of the join messages."),
("colors.clone_report.header.message", "chat", "The colour of the clone report header."),
("colors.clone_report.header.number_of_hosts", "bold", "The colour of the number of hosts in the clone report header."),
("colors.clone_report.header.channel", "bold", "The colour of the channel name in the clone report header."),
("colors.clone_report.subheader.message", "chat", "The colour of the clone report subheader."),
("colors.clone_report.subheader.host", "bold", "The colour of the host in the clone report subheader."),
("colors.clone_report.subheader.number_of_clones", "bold", "The colour of the number of clones in the clone report subheader."),
("colors.clone_report.clone.message", "chat", "The colour of the clone hit in the clone report message."),
("colors.clone_report.clone.match", "chat", "The colour of the match details (masks or nicks) in the clone report."),
("colors.mask.nick", "bold", "The formatting of the nick in the match mask."),
("colors.mask.identhost", "", "The formatting of the identhost in the match mask."),
)
def get_validated_key_from_config(setting):
key = weechat.config_get_plugin(setting)
if key != 'mask' and key != 'nick':
weechat.prnt("", "Key %s not found. Valid settings are 'nick' and 'mask'. Reverted the setting to 'mask'" % key)
weechat.config_set_plugin("clone_report_key", "mask")
key = "mask"
return key
def format_message(msg, formats, reset_color='chat'):
if type(formats) == str:
formats = formats.split()
formatted_message = msg
needs_color_reset = False
for format in formats:
if format in ['bold', 'reverse', 'italic', 'underline']:
end_format = '-%s' % format
else:
needs_color_reset = True
end_format = ""
formatted_message = "%s%s%s" % (weechat.color(format), formatted_message, weechat.color(end_format))
if needs_color_reset:
formatted_message += weechat.color(reset_color)
return formatted_message
def format_from_config(msg, config_option):
return format_message(msg, weechat.config_get_plugin(config_option))
def on_join_scan_cb(data, signal, signal_data):
network = signal.split(',')[0]
joined_nick = weechat.info_get("irc_nick_from_host", signal_data)
join_match_data = re.match(':[^!]+!([^@]+@(\S+)) JOIN :?([#&]\S*)', signal_data)
parsed_ident_host = join_match_data.group(1).lower()
parsed_host = join_match_data.group(2).lower()
if weechat.config_get_plugin("compare_idents") == "on":
hostkey = parsed_ident_host
else:
hostkey = parsed_host
chan_name = join_match_data.group(3)
network_chan_name = "%s.%s" % (network, chan_name)
chan_buffer = weechat.info_get("irc_buffer", "%s,%s" % (network, chan_name))
if not chan_buffer:
print "No IRC channel buffer found for %s" % network_chan_name
return weechat.WEECHAT_RC_OK
if weechat.config_get_plugin("display_join_messages") == "on":
message = "%s%s%s%s%s" % (
format_from_config(joined_nick, "colors.join_messages.nick"),
format_from_config("!", "colors.join_messages.message"),
format_from_config(parsed_ident_host, "colors.join_messages.identhost"),
format_from_config(" JOINed ", "colors.join_messages.message"),
format_from_config(network_chan_name, "colors.join_messages.channel"),
)
#Make sure message format is also applied if no formatting is given for nick
message = format_from_config(message, "colors.join_messages.message")
weechat.prnt(cs_get_buffer(), message)
clones = get_clones_for_buffer("%s,%s" % (network, chan_name), hostkey)
if clones:
key = get_validated_key_from_config("clone_onjoin_alert_key")
filtered_clones = filter(lambda clone: clone['nick'] != joined_nick, clones[hostkey])
match_strings = map(lambda m: format_from_config(m[key], "colors.onjoin_alert.matches"), filtered_clones)
join_string = format_from_config(' and ',"colors.onjoin_alert.message")
masks = join_string.join(match_strings)
message = "%s %s %s %s %s" % (
format_from_config(joined_nick, "colors.onjoin_alert.nick"),
format_from_config("is already on", "colors.onjoin_alert.message"),
format_from_config(network_chan_name, "colors.onjoin_alert.channel"),
format_from_config("as", "colors.onjoin_alert.message"),
masks
)
message = format_from_config(message, weechat.config_get_plugin("colors.onjoin_alert.message"))
if weechat.config_get_plugin("display_onjoin_alert_clone_buffer") == "on":
weechat.prnt(cs_get_buffer(),message)
if weechat.config_get_plugin("display_onjoin_alert_target_buffer") == "on":
weechat.prnt(chan_buffer, message)
if weechat.config_get_plugin("display_onjoin_alert_current_buffer") == "on":
weechat.prnt(weechat.current_buffer(),message)
return weechat.WEECHAT_RC_OK
def cs_get_buffer():
global cs_buffer
if not cs_buffer:
# Sets notify to 0 as this buffer does not need to be in hotlist.
cs_buffer = weechat.buffer_new("clone_scanner", "", \
"", SCRIPT_CLOSE_CB, "")
weechat.buffer_set(cs_buffer, "title", "Clone Scanner")
weechat.buffer_set(cs_buffer, "notify", "0")
weechat.buffer_set(cs_buffer, "nicklist", "0")
if weechat.config_get_plugin("autofocus") == "on":
if not weechat.window_search_with_buffer(cs_buffer):
weechat.command("", "/buffer " + weechat.buffer_get_string(cs_buffer,"name"))
return cs_buffer
def cs_close_cb(*kwargs):
""" A callback for buffer closing. """
global cs_buffer
#TODO: Ensure the clone_scanner buffer gets closed if its option is set and the script unloads
cs_buffer = None
return weechat.WEECHAT_RC_OK
def get_channel_from_buffer_args(buffer, args):
server_name = weechat.buffer_get_string(buffer, "localvar_server")
channel_name = args
if not channel_name:
channel_name = weechat.buffer_get_string(buffer, "localvar_channel")
match_data = re.match('\A(irc.)?([^.]+)\.([#&]\S*)\Z', channel_name)
if match_data:
channel_name = match_data.group(3)
server_name = match_data.group(2)
return server_name, channel_name
def get_clones_for_buffer(infolist_buffer_name, hostname_to_match=None):
matches = {}
infolist = weechat.infolist_get("irc_nick", "", infolist_buffer_name)
while(weechat.infolist_next(infolist)):
ident_hostname = weechat.infolist_string(infolist, "host")
host_matchdata = re.match('([^@]+)@(\S+)', ident_hostname)
if not host_matchdata:
continue
hostname = host_matchdata.group(2).lower()
ident = host_matchdata.group(1).lower()
if weechat.config_get_plugin("compare_idents") == "on":
hostkey = ident_hostname.lower()
else:
hostkey = hostname
if hostname_to_match and hostname_to_match.lower() != hostkey:
continue
nick = weechat.infolist_string(infolist, "name")
matches.setdefault(hostkey,[]).append({
'nick': nick,
'mask': "%s!%s" % (
format_from_config(nick, "colors.mask.nick"),
format_from_config(ident_hostname, "colors.mask.identhost")),
'ident': ident,
'ident_hostname': ident_hostname,
'hostname': hostname,
})
weechat.infolist_free(infolist)
#Select only the results that have more than 1 match for a host
return dict((k, v) for (k, v) in matches.iteritems() if len(v) > 1)
def report_clones(clones, scanned_buffer_name, target_buffer=None):
# Default to clone_scanner buffer
if not target_buffer:
target_buffer = cs_get_buffer()
if clones:
clone_report_header = "%s %s %s%s" % (
format_from_config(len(clones), "colors.clone_report.header.number_of_hosts"),
format_from_config("hosts with clones were found on", "colors.clone_report.header.message"),
format_from_config(scanned_buffer_name, "colors.clone_report.header.channel"),
format_from_config(":", "colors.clone_report.header.message"),
)
clone_report_header = format_from_config(clone_report_header, "colors.clone_report.header.message")
weechat.prnt(target_buffer, clone_report_header)
for (host, clones) in clones.iteritems():
host_message = "%s %s %s %s" % (
format_from_config(host, "colors.clone_report.subheader.host"),
format_from_config("is online from", "colors.clone_report.subheader.message"),
format_from_config(len(clones), "colors.clone_report.subheader.number_of_clones"),
format_from_config("nicks:", "colors.clone_report.subheader.message"),
)
host_message = format_from_config(host_message, "colors.clone_report.subheader.message")
weechat.prnt(target_buffer, host_message)
for user in clones:
key = get_validated_key_from_config("clone_report_key")
clone_message = "%s%s" % (" - ", format_from_config(user[key], "colors.clone_report.clone.match"))
clone_message = format_from_config(clone_message,"colors.clone_report.clone.message")
weechat.prnt(target_buffer, clone_message)
else:
weechat.prnt(target_buffer, "No clones found on %s" % scanned_buffer_name)
def cs_command_main(data, buffer, args):
if args[0:4] == 'scan':
server_name, channel_name = get_channel_from_buffer_args(buffer, args[5:])
clones = get_clones_for_buffer('%s,%s' % (server_name, channel_name))
if weechat.config_get_plugin("display_scan_report_target_buffer") == "on":
target_buffer = weechat.info_get("irc_buffer", "%s,%s" % (server_name, channel_name))
report_clones(clones, '%s.%s' % (server_name, channel_name), target_buffer)
if weechat.config_get_plugin("display_scan_report_clone_buffer") == "on":
report_clones(clones, '%s.%s' % (server_name, channel_name))
if weechat.config_get_plugin("display_scan_report_current_buffer") == "on":
report_clones(clones, '%s.%s' % (server_name, channel_name), weechat.current_buffer())
elif args[0:9] == 'advertise':
weechat.command("", "/input insert /me is using FiXato's CloneScanner v%s for WeeChat. Get the latest version from: https://github.com/FiXato/weechat_scripts/blob/master/clone_scanner.py" % SCRIPT_VERSION)
return weechat.WEECHAT_RC_OK
def cs_set_default_settings():
global cs_settings
# Set default settings
for option, default_value, description in cs_settings:
if not weechat.config_is_set_plugin(option):
weechat.config_set_plugin(option, default_value)
version = weechat.info_get("version_number", "") or 0
if int(version) >= 0x00030500:
weechat.config_set_desc_plugin(option, description)
if __name__ == "__main__" and import_ok:
if weechat.register(SCRIPT_NAME, SCRIPT_AUTHOR, SCRIPT_VERSION,
SCRIPT_LICENSE, SCRIPT_DESC, SCRIPT_CLOSE_CB, ""):
cs_set_default_settings()
cs_buffer = weechat.buffer_search("python", "clone_scanner")
weechat.hook_signal("*,irc_in2_join", "on_join_scan_cb", "")
weechat.hook_command(SCRIPT_COMMAND,
SCRIPT_DESC,
"[scan] [[plugin.][network.]channel] | [advertise] | [help]",
"the target_buffer can be: \n"
"- left out, so the current channel buffer will be scanned.\n"
"- a plain channel name, such as #weechat, in which case it will prefixed with the current network name\n"
"- a channel name prefixed with network name, such as Freenode.#weechat\n"
"- a channel name prefixed with plugin and network name, such as irc.freenode.#weechat\n"
"See /set plugins.var.python.clone_scanner.* for all possible configuration options",
" || scan %(buffers_names)"
" || advertise"
" || help",
"cs_command_main", "")