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
|
###############################################################################
##
## This script is coded for Python 2.7 through Python 3.x
##
## Authors: konablend
##
###############################################################################
import glob
import hashlib
import random
import re
import os
import signal
import sys
import time
try:
from urllib.request import urlopen
except ImportError:
from urllib2 import urlopen
sys.dont_write_bytecode = True
sys.path.insert(0, os.path.join(sys.path[0], 'lib'))
import hb_distfile
###############################################################################
def signal_handler(signal, frame):
sys.stderr.write('^C')
sys.exit(1)
signal.signal(signal.SIGINT, signal_handler)
signal.signal(signal.SIGTERM, signal_handler)
###############################################################################
## simple structure object
class Struct(object):
pass
## track resources and ensure cleanup
##
## - items are lambdas accepting no args
## - item order of insertion is important
## - cleanup will run in reverse order of insertion
## - item update does not effect order
##
class Ensure(object):
def __init__(self):
super(Ensure, self).__setattr__('_items', [])
def __delattr__(self, key):
if key in self.__dict__:
self._items.remove(self.__dict__[key])
super(Ensure, self).__delattr__(key)
def __setattr__(self, key, value):
if not key in self.__dict__:
self._items.insert(0, value)
super(Ensure, self).__setattr__(key, value)
def run(self):
for item in self._items:
try:
item()
except Exception:
pass
###############################################################################
class Tool(hb_distfile.Tool):
def __init__(self):
super(Tool, self).__init__()
self.parser.prog = self.name
self.parser.usage = '%prog [OPTIONS] URL...'
self.parser.description = 'Fetch and verify distfile data integrity.'
self.parser.add_argument('--disable', default=False, action='store_true', help='do nothing and exit with error')
self.parser.add_argument('--jobs', default=1, action='store', metavar='N', help='allow N download jobs at once')
self.parser.add_argument('--sha256', default=None, action='store', metavar='HASH', help='verify sha256 HASH against data')
self.parser.add_argument('--accept-url', default=[], action='append', metavar='SPEC', help='accept URL regex pattern')
self.parser.add_argument('--deny-url', default=[], action='append', metavar='SPEC', help='deny URL regex pattern')
self.parser.add_argument('--exhaust-url', default=None, action='store_true', help='try all active distfiles')
self.parser.add_argument('--output', default=None, action='store', metavar='FILE', help='write to FILE')
self._parse()
def _load_config2(self, parser, data):
parser.values.disable = data['disable-fetch']
parser.values.jobs = data['jobs']
parser.values.accept_url = data['accept-url']
parser.values.deny_url = data['deny-url']
def _run(self, error):
# throttle instances
if tool.options.jobs < 1:
tool.options.jobs = 1
if tool.options.jobs > 20:
tool.options.jobs = 20
dirname = os.path.dirname(tool.options.output)
time.sleep(random.uniform(0.1,2))
active = len(glob.glob(dirname + '/*.tmp'))
while active >= tool.options.jobs:
time.sleep(2)
active = len(glob.glob(dirname + '/*.tmp'))
# handle disabled
if self.options.disable:
raise error('administratively disabled')
## create URL objects and keep active
urls = []
i = 0
for arg in self.args:
url = URL(arg, i)
if url.active:
urls.append(url)
i += 1
## try each URL until first success
error.op = 'download'
if not urls:
raise error('nothing to download')
while urls:
url = urls.pop(0)
try:
url.download(error)
if not self.options.exhaust_url:
break
except Exception as x:
## propagate exception if no remaining urls
if not urls:
raise
self.errln('%s failure; %s' % (error.op,x))
def run(self):
error = hb_distfile.ToolError(self.name)
try:
self._run(error)
except Exception as x:
self.debug_exception()
self.errln('%s failure; %s' % (error.op,x), exit=1)
###############################################################################
class URL(object):
def __init__(self, url, index):
self.index = index
self.url = url
self.active = True
self.rule = 'none'
self._accept()
self._deny()
tool.verbosef('URL[%d]: %s\n' % (self.index,self.url))
tool.verbosef(' active: %s\n' % ('yes' if self.active else 'no'))
tool.verbosef(' rule: %s\n' % (self.rule))
def _accept(self):
if not tool.options.accept_url:
return
index = 0
for spec in tool.options.accept_url:
if re.search(spec, self.url):
self.rule = 'via accept rule[%d]: %s' % (index,spec)
return
index += 1
self.active = False
self.rule = 'no matching accept rule'
def _deny(self):
index = 0
for spec in tool.options.deny_url:
if re.search(spec, self.url):
self.active = False
self.rule = 'via deny rule[%d]: %s' % (index,spec)
return
index += 1
def _download(self, error, ensure):
filename = tool.options.output
hasher = hashlib.sha256()
if filename:
tool.infof('downloading %s to %s\n' % (self.url,filename))
ftmp = tool.mktmpname(filename)
hout = open(ftmp, 'wb')
ensure.unlink_ftmp = lambda: os.unlink(ftmp)
ensure.close_hout = lambda: hout.close()
else:
tool.infof('downloading %s\n' % (self.url))
hin = urlopen(self.url, None, 30)
ensure.close_hin = lambda: hin.close()
info = hin.info()
try:
content_length = int(info.getheader('Content-Length'))
except:
content_length = None
data_total = 0
data_total_percent = 0.0
while True:
data = hin.read(65536)
if not data:
break
if filename:
hout.write(data)
hasher.update(data)
data_total += len(data)
if content_length and content_length > 0:
data_total_percent = float(data_total) / content_length
if data_total_percent >= 1 and data_total < content_length:
data_total_percent = 0.999999
else:
data_total_percent = -1
tool.progressf(data_total_percent, 'downloading... %9d bytes' % data_total)
if content_length and content_length != data_total:
raise error('expected %d bytes, got %d bytes' % (content_length,data_total))
s = 'download total: %9d bytes\n' % data_total
if filename:
s += 'sha256 (%s) = %s' % (filename,hasher.hexdigest())
else:
s += 'sha256 = %s' % (hasher.hexdigest())
if tool.options.sha256:
sha256_pass = tool.options.sha256 == hasher.hexdigest()
s += ' (%s)' % ('pass' if sha256_pass else 'fail; expecting %s' % tool.options.sha256)
tool.infof('%s\n' % s)
if filename and tool.options.sha256:
if sha256_pass:
if os.access(filename, os.F_OK) and not os.access(filename, os.W_OK):
raise error("permission denied: '%s'" % filename)
else:
raise error("expected sha256 hash '%s', got '%s'" % (tool.options.sha256, hasher.hexdigest()))
os.rename(ftmp,filename)
del ensure.unlink_ftmp
def download(self, error):
ensure = Ensure()
try:
self._download(error, ensure)
finally:
ensure.run()
###############################################################################
tool = Tool()
tool.run()
|