krb5: Improve setting file perms for state file

1 files changed, 2 insertions, 2 deletions

diff --git a/krb5/bin/k b/krb5/bin/k
index 8b1008f..4bdfed5 100755
--- a/krb5/bin/k
+++ b/krb5/bin/k
@@ -53,7 +53,7 @@ def get_credentials(url, auth_id):
     create_state_file = False
 
     if os.path.exists(state_file):
-        os.chmod(state_file, stat.S_IREAD|stat.S_IWRITE);
+        os.chmod(state_file, stat.S_IREAD|stat.S_IWRITE)
         with open(state_file, 'r') as f:
             data = f.read()
         cred = Identity.unserialize(auth_id, data)
@@ -80,8 +80,8 @@ def get_credentials(url, auth_id):
 
     if create_state_file:
         with open(state_file, 'w') as f:
+            os.fchmod(f.fileno(), stat.S_IREAD|stat.S_IWRITE)
             f.write(cred.serialize())
-        os.chmod(state_file, stat.S_IREAD|stat.S_IWRITE);
 
     try:
         entry = c.get_logins(cred, url=url)