Read key files in before dropping priviledges

2018-02-16 22:09:05 -05:00
parent 4663f92307
commit 246bdba859
2 changed files with 35 additions and 6 deletions
@@ -91,4 +91,22 @@ def write_pid(milterconfig):
    else:
        if milterconfig.get('Syslog'):
            syslog.syslog('Unable to write pidfle {0}.  File exists.'.format(milterconfig.get('PidFile')))
-        raise RuntimeError('Unable to write pidfle {0}.  File exists.'.format(milterconfig.get('PidFile')))
+        raise RuntimeError('Unable to write pidfle {0}.  File exists.'.format(milterconfig.get('PidFile')))
+
+####################
+def read_keyfile(milterconfig, keytype):
+    """Read private key from file."""
+    import syslog
+    if keytype == "RSA":
+        keyfile = milterconfig.get('KeyFile')
+    if keytype == "Ed25519":
+        keyfile = milterconfig.get('KeyFileEd25519')
+    try:
+        f = open(keyfile, 'r')
+        key = f.readlines
+    except IOError as e:
+        if milterconfig.get('Syslog'):
+            syslog.syslog('Unable to read keyfile {0}.  IOError: {1}'.format(keyfile, e))
+        raise
+    f.close()
+    return key