From bd0551b22076b82a6d49e9f7a2b2e0c90a1b2326 Mon Sep 17 00:00:00 2001
From: Aleksander Machniak <alec@alec.pl>
Date: Fri, 05 Feb 2016 07:25:27 -0500
Subject: [PATCH] Secure also downloads of addressbook exports, managesieve script exports and Enigma keys exports

---
 plugins/squirrelmail_usercopy/squirrelmail_usercopy.php |  302 +++++++++++++++++++++++++++++++++++++------------
 1 files changed, 225 insertions(+), 77 deletions(-)

diff --git a/plugins/squirrelmail_usercopy/squirrelmail_usercopy.php b/plugins/squirrelmail_usercopy/squirrelmail_usercopy.php
index 634f837..9047efd 100644
--- a/plugins/squirrelmail_usercopy/squirrelmail_usercopy.php
+++ b/plugins/squirrelmail_usercopy/squirrelmail_usercopy.php
@@ -1,99 +1,247 @@
 <?php
 
 /**
- * Copy a new users identity and settings from a nearby Squirrelmail installation
+ * Copy a new users identities and contacts from a nearby Squirrelmail installation
  *
- * Currently only file-based data storage of Squirrelmail is supported.
- *
- * @version 1.1
- * @author Thomas Bruederli
+ * @version 1.5
+ * @author Thomas Bruederli, Johannes Hessellund, pommi, Thomas Lueder
  */
 class squirrelmail_usercopy extends rcube_plugin
 {
-	public $task = 'login|settings';
+    public $task = 'login';
 
-	private $prefs = null;
-	private $abook = array();
+    private $prefs            = null;
+    private $identities_level = 0;
+    private $abook            = array();
 
-	public function init()
-	{
-		$this->add_hook('create_user', array($this, 'create_user'));
-		$this->add_hook('create_identity', array($this, 'create_identity'));
-	}
+    public function init()
+    {
+        $this->add_hook('user_create', array($this, 'create_user'));
+        $this->add_hook('identity_create', array($this, 'create_identity'));
+    }
 
-	public function create_user($p)
-	{
-		// read prefs and add email address
-		$this->read_squirrel_prefs($p['user']);
-		if ($this->prefs['email_address'])
-			$p['user_email'] = $this->prefs['email_address'];
+    public function create_user($p)
+    {
+        $rcmail = rcmail::get_instance();
 
-		return $p;
-	}
+        // Read plugin's config
+        $this->initialize();
 
-	public function create_identity($p)
-	{
-		$rcmail = rcmail::get_instance();
+        // read prefs and add email address
+        $this->read_squirrel_prefs($p['user']);
+        if (($this->identities_level == 0 || $this->identities_level == 2)
+            && $rcmail->config->get('squirrelmail_set_alias')
+            && $this->prefs['email_address']
+        ) {
+            $p['user_email'] = $this->prefs['email_address'];
+        }
 
-		// only execute on login
-		if ($rcmail->task == 'login' && $this->prefs) {
-			if ($this->prefs['full_name'])
-				$p['record']['name'] = $this->prefs['full_name'];
-			if ($this->prefs['email_address'])
-				$p['record']['email'] = $this->prefs['email_address'];
-			if ($this->prefs['signature'])
-				$p['record']['signature'] = $this->prefs['signature'];
-                        if ($this->prefs['reply-to']) 
-                                $p['record']['reply-to'] = $this->prefs['reply-to']; 		
+        return $p;
+    }
 
-			// copy address book
-			$contacts = $rcmail->get_address_book(null, true);
-			if ($contacts && count($this->abook)) {
-				foreach ($this->abook as $rec)
-					$contacts->insert($rec, true);
-			}
-			
-			// mark identity as complete for following hooks
-			$p['complete'] = true;
-		}
+    public function create_identity($p)
+    {
+        $rcmail = rcmail::get_instance();
 
-		return $p;
-	}
+        // prefs are set in create_user()
+        if ($this->prefs) {
+            if ($this->prefs['full_name']) {
+                $p['record']['name'] = $this->prefs['full_name'];
+            }
 
-	private function read_squirrel_prefs($uname)
-	{
-		$this->load_config();
-		$rcmail = rcmail::get_instance();
+            if (($this->identities_level == 0 || $this->identities_level == 2) && $this->prefs['email_address']) {
+                $p['record']['email'] = $this->prefs['email_address'];
+            }
 
-		if ($srcdir = $rcmail->config->get('squirrelmail_data_dir')) {
-			$prefsfile = slashify($srcdir) . $uname . '.pref';
-			$abookfile = slashify($srcdir) . $uname . '.abook';
-			$sigfile = slashify($srcdir) . $uname . '.sig';
+            if ($this->prefs['___signature___']) {
+                $p['record']['signature'] = $this->prefs['___signature___'];
+            }
 
-			if (is_readable($prefsfile)) {
-				$this->prefs = array();
-				foreach (file($prefsfile) as $line) {
-					list($key, $value) = explode('=', $line);
-					$this->prefs[$key] = utf8_encode(rtrim($value));
-				}
+            if ($this->prefs['reply_to']) {
+                $p['record']['reply-to'] = $this->prefs['reply_to'];
+            }
 
-				// also read signature file if exists
-				if (is_readable($sigfile)) {
-					$this->prefs['signature'] = utf8_encode(file_get_contents($sigfile));
-				}
+            if (($this->identities_level == 0 || $this->identities_level == 1)
+                && isset($this->prefs['identities']) && $this->prefs['identities'] > 1
+            ) {
+                for ($i = 1; $i < $this->prefs['identities']; $i++) {
+                    unset($ident_data);
+                    $ident_data = array('name' => '', 'email' => ''); // required data
 
-				// parse addres book file
-				if (filesize($abookfile)) {
-					foreach(file($abookfile) as $line) {
-						list($rec['name'], $rec['firstname'], $rec['surname'], $rec['email']) = explode('|', utf8_encode(rtrim($line)));
-						if ($rec['name'] && $rec['email'])
-							$this->abook[] = $rec;
-					}
-				}
-			}
-		}
-	}
+                    if ($this->prefs['full_name'.$i]) {
+                        $ident_data['name'] = $this->prefs['full_name'.$i];
+                    }
 
+                    if ($this->identities_level == 0 && $this->prefs['email_address'.$i]) {
+                        $ident_data['email'] = $this->prefs['email_address'.$i];
+                    }
+                    else {
+                        $ident_data['email'] = $p['record']['email'];
+                    }
+
+                    if ($this->prefs['reply_to'.$i]) {
+                        $ident_data['reply-to'] = $this->prefs['reply_to'.$i];
+                    }
+
+                    if ($this->prefs['___sig'.$i.'___']) {
+                        $ident_data['signature'] = $this->prefs['___sig'.$i.'___'];
+                    }
+
+                    // insert identity
+                    $rcmail->user->insert_identity($ident_data);
+                }
+            }
+
+            // copy address book
+            $contacts  = $rcmail->get_address_book(null, true);
+            $addresses = array();
+            $groups    = array();
+
+            if ($contacts && !empty($this->abook)) {
+                foreach ($this->abook as $rec) {
+                    // #1487096: handle multi-address and/or too long items
+                    // #1487858: convert multi-address contacts into groups
+                    $emails   = preg_split('/[;,]/', $rec['email'], -1, PREG_SPLIT_NO_EMPTY);
+                    $group_id = null;
+
+                    // create group for addresses
+                    if (count($emails) > 1) {
+                        if (!($group_id = $groups[$rec['name']])) {
+                            if ($group = $contacts->create_group($rec['name'])) {
+                                $group_id = $group['id'];
+                                $groups[$rec['name']] = $group_id;
+                            }
+                        }
+                    }
+
+                    // create contacts
+                    foreach ($emails as $email) {
+                        if (!($contact_id = $addresses[$email]) && rcube_utils::check_email(rcube_utils::idn_to_ascii($email))) {
+                            $rec['email'] = rcube_utils::idn_to_utf8($email);
+                            if ($contact_id = $contacts->insert($rec, true)) {
+                                $addresses[$email] = $contact_id;
+                            }
+                        }
+
+                        if ($group_id && $contact_id) {
+                            $contacts->add_to_group($group_id, array($contact_id));
+                        }
+                    }
+                }
+            }
+
+            // mark identity as complete for following hooks
+            $p['complete'] = true;
+        }
+
+        return $p;
+    }
+
+    private function initialize()
+    {
+        $rcmail = rcmail::get_instance();
+
+        // Load plugin's config file
+        $this->load_config();
+
+        // Set identities_level for operations of this plugin
+        $ilevel = $rcmail->config->get('squirrelmail_identities_level');
+        if ($ilevel === null) {
+            $ilevel = $rcmail->config->get('identities_level', 0);
+        }
+
+        $this->identities_level = intval($ilevel);
+    }
+
+    private function read_squirrel_prefs($uname)
+    {
+        $rcmail = rcmail::get_instance();
+
+        /**** File based backend ****/
+        if ($rcmail->config->get('squirrelmail_driver') == 'file' && ($srcdir = $rcmail->config->get('squirrelmail_data_dir'))) {
+            if (($hash_level = $rcmail->config->get('squirrelmail_data_dir_hash_level')) > 0) {
+                $srcdir = slashify($srcdir).chunk_split(substr(base_convert(crc32($uname), 10, 16), 0, $hash_level), 1, '/');
+            }
+
+            $prefsfile = slashify($srcdir) . $uname . '.pref';
+            $abookfile = slashify($srcdir) . $uname . '.abook';
+            $sigfile   = slashify($srcdir) . $uname . '.sig';
+            $sigbase   = slashify($srcdir) . $uname . '.si';
+
+            if (is_readable($prefsfile)) {
+                $this->prefs = array();
+                foreach (file($prefsfile) as $line) {
+                    list($key, $value) = explode('=', $line);
+                    $this->prefs[$key] = utf8_encode(rtrim($value));
+                }
+
+                // also read signature file if exists
+                if (is_readable($sigfile)) {
+                    $this->prefs['___signature___'] = utf8_encode(file_get_contents($sigfile));
+                }
+
+                if (isset($this->prefs['identities']) && $this->prefs['identities'] > 1) {
+                    for ($i=1; $i < $this->prefs['identities']; $i++) {
+                        // read signature file if exists
+                        if (is_readable($sigbase.$i)) {
+                            $this->prefs['___sig'.$i.'___'] = utf8_encode(file_get_contents($sigbase.$i));
+                        }
+                    }
+                }
+
+                // parse addres book file
+                if (filesize($abookfile)) {
+                    foreach(file($abookfile) as $line) {
+                        list($rec['name'], $rec['firstname'], $rec['surname'], $rec['email']) = explode('|', utf8_encode(rtrim($line)));
+                        if ($rec['name'] && $rec['email']) {
+                            $this->abook[] = $rec;
+                        }
+                    }
+                }
+            }
+        }
+        // Database backend
+        else if ($rcmail->config->get('squirrelmail_driver') == 'sql') { 
+            $this->prefs = array();
+
+            // connect to squirrelmail database
+            $db = rcube_db::factory($rcmail->config->get('squirrelmail_dsn'));
+
+            $db->set_debug($rcmail->config->get('sql_debug'));
+            $db->db_connect('r'); // connect in read mode
+
+            // retrieve prefs
+            $userprefs_table = $rcmail->config->get('squirrelmail_userprefs_table');
+            $address_table   = $rcmail->config->get('squirrelmail_address_table');
+            $db_charset      = $rcmail->config->get('squirrelmail_db_charset');
+
+            if ($db_charset) {
+                $db->query('SET NAMES '.$db_charset);
+            }
+
+            $sql_result = $db->query('SELECT * FROM ' . $db->quote_identifier($userprefs_table)
+                .' WHERE `user` = ?', $uname); // ? is replaced with emailaddress
+
+            while ($sql_array = $db->fetch_assoc($sql_result) ) { // fetch one row from result
+                $this->prefs[$sql_array['prefkey']] = rcube_charset::convert(rtrim($sql_array['prefval']), $db_charset);
+            }
+
+            // retrieve address table data
+            $sql_result = $db->query('SELECT * FROM ' . $db->quote_identifier($address_table)
+                .' WHERE `owner` = ?', $uname); // ? is replaced with emailaddress
+
+            // parse addres book
+            while ($sql_array = $db->fetch_assoc($sql_result) ) { // fetch one row from result
+                $rec['name']      = rcube_charset::convert(rtrim($sql_array['nickname']), $db_charset);
+                $rec['firstname'] = rcube_charset::convert(rtrim($sql_array['firstname']), $db_charset);
+                $rec['surname']   = rcube_charset::convert(rtrim($sql_array['lastname']), $db_charset);
+                $rec['email']     = rcube_charset::convert(rtrim($sql_array['email']), $db_charset);
+                $rec['notes']     = rcube_charset::convert(rtrim($sql_array['label']), $db_charset);
+
+                if ($rec['name'] && $rec['email']) {
+                    $this->abook[] = $rec;
+                }
+            }
+        } // end if 'sql'-driver
+    }
 }
-
-?>

--
Gitblit v1.9.1