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
|
import errno
import time
import urllib
import urllib2
from contextlib import closing
from cloudinit import log as logging
LOG = logging.getLogger(__name__)
def ok_http_code(st):
return st in xrange(200, 400)
def readurl(url, data=None, timeout=None, retries=0, sec_between=1, headers=None):
openargs = {}
if timeout is not None:
openargs['timeout'] = int(timeout)
if data is None:
req = urllib2.Request(url, headers=headers)
else:
req = urllib2.Request(url, data=urllib.urlencode(data), headers=headers)
if retries <= 0:
retries = 1
attempts = retries + 1
last_excp = None
LOG.debug("Attempting to read from %s with %s attempts to be performed", url, attempts)
for i in range(0, attempts):
try:
with closing(urllib2.urlopen(req, **openargs)) as rh:
return (rh.read(), rh.getcode())
except urllib2.HTTPError as e:
last_excp = e
LOG.exception("Failed at reading from %s.", url)
except urllib2.URLError as e:
# This can be a message string or
# another exception instance (socket.error for remote URLs, OSError for local URLs).
if (isinstance(e.reason, OSError) and
e.reason.errno == errno.ENOENT):
last_excp = e.reason
else:
last_excp = e
LOG.exception("Failed at reading from %s.", url)
if i + 1 < attempts:
LOG.debug("Please wait %s seconds while we wait to try again.", sec_between)
time.sleep(sec_between)
# Didn't work out
LOG.warn("Failed downloading from %s after %s attempts", url, attempts)
if last_excp is not None:
raise last_excp
def wait_for_url(urls, max_wait=None, timeout=None,
status_cb=None, headers_cb=None, sleep_time=1):
"""
urls: a list of urls to try
max_wait: roughly the maximum time to wait before giving up
The max time is *actually* len(urls)*timeout as each url will
be tried once and given the timeout provided.
timeout: the timeout provided to urllib2.urlopen
status_cb: call method with string message when a url is not available
headers_cb: call method with single argument of url to get headers
for request.
the idea of this routine is to wait for the EC2 metdata service to
come up. On both Eucalyptus and EC2 we have seen the case where
the instance hit the MD before the MD service was up. EC2 seems
to have permenantely fixed this, though.
In openstack, the metadata service might be painfully slow, and
unable to avoid hitting a timeout of even up to 10 seconds or more
(LP: #894279) for a simple GET.
Offset those needs with the need to not hang forever (and block boot)
on a system where cloud-init is configured to look for EC2 Metadata
service but is not going to find one. It is possible that the instance
data host (169.254.169.254) may be firewalled off Entirely for a sytem,
meaning that the connection will block forever unless a timeout is set.
"""
starttime = time.time()
def nullstatus_cb(msg):
return
if status_cb is None:
status_cb = nullstatus_cb
def timeup(max_wait, starttime):
return ((max_wait <= 0 or max_wait is None) or
(time.time() - starttime > max_wait))
loop_n = 0
while True:
sleeptime = int(loop_n / 5) + 1
for url in urls:
now = time.time()
if loop_n != 0:
if timeup(max_wait, starttime):
break
if timeout and (now + timeout > (starttime + max_wait)):
# shorten timeout to not run way over max_time
timeout = int((starttime + max_wait) - now)
reason = ""
try:
if headers_cb is not None:
headers = headers_cb(url)
else:
headers = {}
(resp, sc) = readurl(url, headers=headers, timeout=timeout)
if not resp:
reason = "empty response [%s]" % sc
elif not ok_http_code(sc):
reason = "bad status code [%s]" % sc
else:
return url
except urllib2.HTTPError as e:
reason = "http error [%s]" % e.code
except urllib2.URLError as e:
reason = "url error [%s]" % e.reason
except socket.timeout as e:
reason = "socket timeout [%s]" % e
except Exception as e:
reason = "unexpected error [%s]" % e
status_cb("'%s' failed [%s/%ss]: %s" %
(url, int(time.time() - starttime), max_wait,
reason))
if timeup(max_wait, starttime):
break
loop_n = loop_n + 1
time.sleep(sleeptime)
return False
|
