Rietveld Code Review Tool
Help | Bug tracker | Discussion group | Source code

Side by Side Diff: sitescripts/subscriptions/bin/updateMalwareDomainsList.py

Issue 29321060: Issue 2719 - Decode domains when generating the "Malware Domains" list (Closed)
Patch Set: Created June 24, 2015, 12:02 p.m.
Left:
Right:
Use n/p to move between diff chunks; N/P to move between comments.
Jump to:
View unified diff | Download patch
« no previous file with comments | « no previous file | no next file » | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 # coding: utf-8 1 # coding: utf-8
2 2
3 # This file is part of the Adblock Plus web scripts, 3 # This file is part of the Adblock Plus web scripts,
4 # Copyright (C) 2006-2015 Eyeo GmbH 4 # Copyright (C) 2006-2015 Eyeo GmbH
5 # 5 #
6 # Adblock Plus is free software: you can redistribute it and/or modify 6 # Adblock Plus is free software: you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License version 3 as 7 # it under the terms of the GNU General Public License version 3 as
8 # published by the Free Software Foundation. 8 # published by the Free Software Foundation.
9 # 9 #
10 # Adblock Plus is distributed in the hope that it will be useful, 10 # Adblock Plus is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of 11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 # GNU General Public License for more details. 13 # GNU General Public License for more details.
14 # 14 #
15 # You should have received a copy of the GNU General Public License 15 # You should have received a copy of the GNU General Public License
16 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. 16 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>.
17 17
18 import os, subprocess, codecs, urllib, zipfile, tempfile, shutil 18 import os, subprocess, codecs, urllib, zipfile, tempfile, shutil
19 from StringIO import StringIO 19 from StringIO import StringIO
20 from sitescripts.utils import get_config, setupStderr 20 from sitescripts.utils import get_config
21 21
22 if __name__ == '__main__': 22 if __name__ == '__main__':
23 setupStderr()
24
25 repository = get_config().get('subscriptionDownloads', 'malwaredomains_reposit ory') 23 repository = get_config().get('subscriptionDownloads', 'malwaredomains_reposit ory')
26 tempdir = tempfile.mkdtemp(prefix='malwaredomains') 24 tempdir = tempfile.mkdtemp(prefix='malwaredomains')
27 try: 25 try:
28 subprocess.check_call(['hg', '-q', 'clone', '-U', repository, tempdir]) 26 subprocess.check_call(['hg', '-q', 'clone', '-U', repository, tempdir])
29 subprocess.check_call(['hg', '-q', 'up', '-R', tempdir, '-r', 'default']) 27 subprocess.check_call(['hg', '-q', 'up', '-R', tempdir, '-r', 'default'])
30 28
31 path = os.path.join(tempdir, 'malwaredomains_full.txt') 29 path = os.path.join(tempdir, 'malwaredomains_full.txt')
32 file = codecs.open(path, 'wb', encoding='utf-8') 30 file = codecs.open(path, 'wb', encoding='utf-8')
33 31
34 print >>file, '''[Adblock Plus 1.1] 32 print >>file, '''[Adblock Plus 1.1]
35 ! This is a list of malware domains generated from malwaredomains.com data. 33 ! This is a list of malware domains generated from malwaredomains.com data.
36 ! Homepage: http://malwaredomains.com/?page_id=2 34 ! Homepage: http://malwaredomains.com/?page_id=2
37 ! Last modified: %timestamp% 35 ! Last modified: %timestamp%
38 ! Expires: 1d 36 ! Expires: 1d
39 !''' 37 !'''
40 38
41 data = urllib.urlopen('http://mirror3.malwaredomains.com/files/justdomains.z ip').read() 39 data = urllib.urlopen('http://mirror3.malwaredomains.com/files/justdomains.z ip').read()
42 zip = zipfile.ZipFile(StringIO(data), 'r') 40 zip = zipfile.ZipFile(StringIO(data), 'r')
43 info = zip.infolist()[0] 41 info = zip.infolist()[0]
44 for line in str(zip.read(info.filename)).splitlines(): 42 for line in str(zip.read(info.filename)).splitlines():
45 if not line: 43 if not line:
46 continue 44 continue
47 print >>file, '||%s^' % line.strip().decode('iso-8859-1') 45 print >>file, '||%s^' % line.strip().decode('idna')
48 file.close(); 46 file.close();
49 47
50 if subprocess.check_output(['hg', 'stat', '-R', tempdir]) != '': 48 if subprocess.check_output(['hg', 'stat', '-R', tempdir]) != '':
51 subprocess.check_call(['hg', '-q', 'commit', '-R', tempdir, '-A', '-u', 'h gbot', '-m', 'Updated malwaredomains.com data']) 49 subprocess.check_call(['hg', '-q', 'commit', '-R', tempdir, '-A', '-u', 'h gbot', '-m', 'Updated malwaredomains.com data'])
52 subprocess.check_call(['hg', '-q', 'push', '-R', tempdir]) 50 subprocess.check_call(['hg', '-q', 'push', '-R', tempdir])
53 finally: 51 finally:
54 shutil.rmtree(tempdir, ignore_errors=True) 52 shutil.rmtree(tempdir, ignore_errors=True)
OLDNEW
« no previous file with comments | « no previous file | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld