support getting public ssh keys from ec2 metadata service

1 files changed, 10 insertions, 1 deletions

diff --git a/ec2init/DataSourceEc2.py b/ec2init/DataSourceEc2.py
index cc12c97c..5d3bab88 100644
--- a/ec2init/DataSourceEc2.py
+++ b/ec2init/DataSourceEc2.py
@@ -120,5 +120,14 @@ class DataSourceEc2(DataSource.DataSource):
         elif availability_zone.startswith('eu-'):
             return 'eu'
         raise Exception('Could not determine location')
- 
 
+    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():
+            for pkey in klist:
+                # there is an empty string at the end of the keylist, trim it
+                if pkey:
+                    keys.append(pkey)
+
+        return(keys)