Add support for AWS keys reading

ezesoler · May 23, 2017 · 6e5063b · 6e5063b
1 parent b11a71b
commit 6e5063b
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 0 deletions.
diff --git a/Linux/lazagne/config/manageModules.py b/Linux/lazagne/config/manageModules.py
@@ -9,6 +9,7 @@
 from lazagne.softwares.sysadmin.filezilla import Filezilla
 from lazagne.softwares.sysadmin.env_variable import Env_variable
 from lazagne.softwares.sysadmin.shadow import Shadow
+from lazagne.softwares.sysadmin.aws import Aws
 # chats
 from lazagne.softwares.chats.pidgin import Pidgin
 from lazagne.softwares.chats.jitsi import Jitsi
@@ -52,6 +53,7 @@ def get_modules():
 		Opera(),
 		Pidgin(),
 		Shadow(),
+        Aws(),
 		SQLDeveloper(),
 		Squirrel(),
 		Wifi(),

diff --git a/Linux/lazagne/softwares/sysadmin/aws.py b/Linux/lazagne/softwares/sysadmin/aws.py
@@ -0,0 +1,44 @@
+from lazagne.config.constant import *
+from lazagne.config.write_output import print_debug
+from lazagne.config.moduleInfo import ModuleInfo
+from lazagne.config import homes
+from ConfigParser import ConfigParser
+
+import os
+
+class Aws(ModuleInfo):
+    def __init__(self):
+        options = {'command': '-W', 'action': 'store_true', 'dest': 'aws', 'help': 'aws'}
+        suboptions = []
+        ModuleInfo.__init__(self, 'aws', 'sysadmin', options, suboptions)
+
+    def get_paths(self):
+        return homes.get(file=os.path.join('.aws', 'credentials'))
+
+    def get_creds(self, path):
+        try:
+            parser = ConfigParser()
+            parser.read(path)
+        except:
+            return
+
+        for section in parser.sections():
+            try:
+                key = parser.get(section, 'aws_access_key_id')
+                secret = parser.get(section, 'aws_secret_access_key')
+                yield section, key, secret
+            except:
+                continue
+
+    def run(self, software_name=None):
+        all_passwords = []
+        for path in self.get_paths():
+            for section, key, secret in self.get_creds(path):
+                all_passwords.append({
+                    'ID': key,
+                    'KEY': secret,
+                    'Service': 'AWS',
+                    'Name': section
+                })
+
+        return all_passwords