DataObjects/Core_setting.php

[Pman.Core] / DataObjects / Core_setting.php

diff --git a/DataObjects/Core_setting.php b/DataObjects/Core_setting.php
index 5dd671e..d27007b 100644 (file)
--- a/DataObjects/Core_setting.php
+++ b/DataObjects/Core_setting.php
@@ -8,13 +8,18 @@ class Pman_Core_DataObjects_Core_setting extends DB_DataObject
     
     function initKeys()
     {
+        // return false when fail
         $dir = $this->getKeyDirectory();
         
+        if(!$dir) {
+            return false; // only fail case?
+        }
+        
         if(
             file_exists("{$dir}/pub.key") ||
             file_exists("{$dir}/pri.key")
         ){
-            return;
+            return true;
         }
         
         $ssl = openssl_pkey_new(array(
@@ -29,6 +34,8 @@ class Pman_Core_DataObjects_Core_setting extends DB_DataObject
         
         file_put_contents("{$dir}/pub.key",$pub_key);
         file_put_contents("{$dir}/pri.key",$pri_key);
+        
+        return true;
     }
     
     function lookup($m,$n)
@@ -38,9 +45,11 @@ class Pman_Core_DataObjects_Core_setting extends DB_DataObject
             'module' => $m,
             'name' => $n
         ));
+        
         if($s->find(true)) {
             return $s;
         }
+        
         return false;
     }
     
@@ -52,11 +61,19 @@ class Pman_Core_DataObjects_Core_setting extends DB_DataObject
     function getKeyDirectory()
     {
         $client_dir = HTML_FlexyFramework::get()->Pman['storedir'];
+        
         $key_dir = $client_dir.'/keys';
-        if(!file_exists($key_dir)) {
-            $this->checkWritable(get_class($this),__FUNCTION__,$client_dir);
-            exec("mkdir -m775 {$key_dir}");
+        
+        if(file_exists($key_dir)) {
+            return $key_dir;
+        }
+        
+        if(!is_writable($key_dir)) {
+            return false;
         }
+        
+        exec("mkdir -m775 {$key_dir}");
+        
         return $key_dir;
     }
     
@@ -64,8 +81,6 @@ class Pman_Core_DataObjects_Core_setting extends DB_DataObject
     // should not be sending this the values..
     function initSetting($a)
     {
-        $dir = $this->getKeyDirectory();
-        
         if(empty($a)) {
             return;
         }
@@ -105,18 +120,37 @@ class Pman_Core_DataObjects_Core_setting extends DB_DataObject
         if(!$pub_key) {
             return;
         }
-        openssl_public_encrypt($v, $cipher, $pub_key);
-        return $cipher;
+        openssl_public_encrypt($v, $ciphertext, $pub_key);
+        return $ciphertext;
     }
     
-    function checkWritable($cls_name,$func_name,$dir)
+    function getDecryptVal()
     {
-        if(!is_writable($dir)) {
-            print_r("Cannot run {$cls_name} :: {$func_name}\n");
-            print_r("Directory: {$dir} is not writable by current user\n");
-            exit;
+        $dir = $this->getKeyDirectory();
+        
+        if(!$dir) {
+            return false;
         }
         
-        return true;
+        if(empty($this->val)) {
+            return false;
+        }
+        
+        $key_dir = "{$dir}/pri.key";
+        
+        if(!file_exists($key_dir)) {
+            return false;
+        }
+        
+        $pri_key = file_get_contents($key_dir);
+        
+        if(!$pri_key) {
+            return false;
+        }
+        
+        openssl_private_decrypt($this->val, $plaintext, $pri_key);
+        
+        return $plaintext;
     }
+    
 }