ensure that cache dir is present if not created, and tighten permissions

on object cache stored there.

1 files changed, 7 insertions, 0 deletions

diff --git a/cloudinit/__init__.py b/cloudinit/__init__.py
index 65ca6d7f..8d6ec59e 100644
--- a/cloudinit/__init__.py
+++ b/cloudinit/__init__.py
@@ -104,8 +104,15 @@ class CloudInit:
 
     def write_to_cache(self):
         try:
+            os.makedirs(os.path.dirname(data_source_cache))
+        except OSError as e:
+            if e.errno != errno.EEXIST:
+                return False
+                
+        try:
             f=open(data_source_cache, "wb")
             data = cPickle.dump(self.datasource,f)
+            os.chmod(data_source_cache,0400)
             return True
         except:
             return False