support compressed user data, use cPickle instead of pickle

1 files changed, 19 insertions, 1 deletions

diff --git a/ec2init/DataSource.py b/ec2init/DataSource.py
index da6170fd..b940c564 100644
--- a/ec2init/DataSource.py
+++ b/ec2init/DataSource.py
@@ -1,8 +1,26 @@
 
+import ec2init
+
 class DataSource:
+    userdata = None
+    metadata = None
+    userdata_raw = None
+
     def __init__(self):
        pass
 
+    def store_user_data_raw(self):
+        fp=fopen(user_data_raw,"wb")
+        fp.write(self.userdata_raw)
+        fp.close()
+    
+    def store_user_data(self):
+        fp=fopen(user_data,"wb")
+        fp.write(self.userdata)
+        fp.close()
+
     def get_user_data(self):
-        raise Exception("get_user_data Not-implemented")
+        if self.userdata == None:
+            self.userdata = ec2init.preprocess_user_data(self.userdata_raw)
 
+        return self.userdata