eea1ce32c0
Also, move swap to use this format for specifying mounts. That way the user can turn off swap if they want with: | mounts: | - [ swap ] Other change wrapped in here is to have DataSourceEc2 read plain text (evalable) text for its metadata rather than pickl. This is really for debuging anyway, so any speed difference is not important.
126 lines
3.9 KiB
Python
126 lines
3.9 KiB
Python
import DataSource
|
|
|
|
import ec2init
|
|
import socket
|
|
import urllib2
|
|
import time
|
|
import boto_utils
|
|
|
|
class DataSourceEc2(DataSource.DataSource):
|
|
api_ver = '2009-04-04'
|
|
cachedir = ec2init.cachedir + '/ec2'
|
|
|
|
location_locale_map = {
|
|
'us' : 'en_US.UTF-8',
|
|
'eu' : 'en_GB.UTF-8',
|
|
'default' : 'en_US.UTF-8',
|
|
}
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
def get_data(self):
|
|
try:
|
|
udf = open(self.cachedir + "/user-data.raw")
|
|
self.userdata_raw = udf.read()
|
|
udf.close()
|
|
|
|
mdf = open(self.cachedir + "/meta-data.raw")
|
|
data = mdf.read()
|
|
self.metadata = eval(data)
|
|
mdf.close()
|
|
|
|
return True
|
|
except:
|
|
pass
|
|
|
|
try:
|
|
if not self.wait_for_metadata_service():
|
|
return False
|
|
self.userdata_raw = boto_utils.get_instance_userdata(self.api_ver)
|
|
self.metadata = boto_utils.get_instance_metadata(self.api_ver)
|
|
return True
|
|
except Exception as e:
|
|
print e
|
|
return False
|
|
|
|
def get_instance_id(self):
|
|
return(self.metadata['instance-id'])
|
|
|
|
def get_availability_zone(self):
|
|
return(self.metadata['placement']['availability-zone'])
|
|
|
|
def get_local_mirror(self):
|
|
return(self.get_mirror_from_availability_zone())
|
|
|
|
def get_locale(self):
|
|
az = self.metadata['placement']['availability-zone']
|
|
if self.location_locale_map.has_key(az[0:2]):
|
|
return(self.location_locale_map[az[0:2]])
|
|
else:
|
|
return(self.location_locale_map["default"])
|
|
|
|
def get_hostname(self):
|
|
hostname = self.metadata['local-hostname']
|
|
hostname = hostname.split('.')[0]
|
|
return hostname
|
|
|
|
def get_mirror_from_availability_zone(self, availability_zone = None):
|
|
# availability is like 'us-west-1b' or 'eu-west-1a'
|
|
if availability_zone == None:
|
|
availability_zone = self.get_availability_zone()
|
|
|
|
try:
|
|
host="%s.ec2.archive.ubuntu.com" % availability_zone[:-1]
|
|
socket.getaddrinfo(host, None, 0, socket.SOCK_STREAM)
|
|
return 'http://%s/ubuntu/' % host
|
|
except:
|
|
return 'http://archive.ubuntu.com/ubuntu/'
|
|
|
|
def wait_for_metadata_service(self, sleeps = 10):
|
|
sleeptime = 1
|
|
for x in range(sleeps):
|
|
s = socket.socket()
|
|
try:
|
|
address = '169.254.169.254'
|
|
port = 80
|
|
s.connect((address,port))
|
|
s.close()
|
|
return True
|
|
except socket.error, e:
|
|
print "sleeping %s" % sleeptime
|
|
time.sleep(sleeptime)
|
|
#timeout = timeout * 2
|
|
return False
|
|
|
|
def get_public_ssh_keys(self):
|
|
keys = []
|
|
if not self.metadata.has_key('public-keys'): return([])
|
|
for keyname, klist in self.metadata['public-keys'].items():
|
|
# lp:506332 uec metadata service responds with
|
|
# data that makes boto populate a string for 'klist' rather
|
|
# than a list.
|
|
if isinstance(klist,str):
|
|
klist = [ klist ]
|
|
for pkey in klist:
|
|
# there is an empty string at the end of the keylist, trim it
|
|
if pkey:
|
|
keys.append(pkey)
|
|
|
|
return(keys)
|
|
|
|
def device_name_to_device(self, name):
|
|
# consult metadata service, that has
|
|
# ephemeral0: sdb
|
|
# and return 'sdb' for input 'ephemeral0'
|
|
if not self.metadata.has_key('block-device-mapping'):
|
|
return(None)
|
|
|
|
for entname, device in self.metadata['block-device-mapping'].items():
|
|
if entname == name:
|
|
return(device)
|
|
# LP: #513842 mapping in Euca has 'ephemeral' not 'ephemeral0'
|
|
if entname == "ephemeral" and name == "ephemeral0":
|
|
return(device)
|
|
return None
|