From fea974d01416c9ecb57f801b1796d1332918063a Mon Sep 17 00:00:00 2001
From: tbrehm <t.brehm@ispconfig.org>
Date: Mon, 23 Jan 2012 09:53:52 -0500
Subject: [PATCH] Fixed: FS#1997 - Check encoding of the subject in ispconfig mail sending function.

---
 server/plugins-available/shelluser_base_plugin.inc.php |  141 +++++++++++++++++++++++++++++++----------------
 1 files changed, 93 insertions(+), 48 deletions(-)

diff --git a/server/plugins-available/shelluser_base_plugin.inc.php b/server/plugins-available/shelluser_base_plugin.inc.php
index 3222d2c..f411251 100755
--- a/server/plugins-available/shelluser_base_plugin.inc.php
+++ b/server/plugins-available/shelluser_base_plugin.inc.php
@@ -194,59 +194,104 @@
 		
 	}
 	
-	function _setup_ssh_rsa() {
-			//global $app, $conf;
-			//$app->uses('system');
-			// Okay, here we have a question, .. how to determine the client id
-			//$var = "Var:".intval($this->app->system->getuid($this->data['new']['puser']));
-			//exec("echo $var >> /tmp/debug");
-			/*
-			// ssh-rsa authentication variables
-			$sshrsa = escapeshellcmd($this->data['new']['ssh_rsa']);
-			$usrdir = escapeshellcmd($this->data['new']['dir']);
-			$sshdir = escapeshellcmd($this->data['new']['dir']).'/.ssh';
-			$sshkeys= escapeshellcmd($this->data['new']['dir']).'/.ssh/authorized_keys';
+	private function _setup_ssh_rsa() {
+		global $app;
+		$this->app->log("ssh-rsa setup shelluser_base",LOGLEVEL_DEBUG);
+		// Get the client ID, username, and the key
+		$domain_data = $this->app->db->queryOneRecord('SELECT sys_groupid FROM web_domain WHERE web_domain.domain_id = '.intval($this->data['new']['parent_domain_id']));
+		$sys_group_data = $this->app->db->queryOneRecord('SELECT * FROM sys_group WHERE sys_group.groupid = '.intval($domain_data['sys_groupid']));
+		$id = intval($sys_group_data['client_id']);
+		$username= $sys_group_data['name'];
+		$client_data = $this->app->db->queryOneRecord('SELECT * FROM client WHERE client.client_id = '.$id);
+		$userkey = $client_data['ssh_rsa'];
+		unset($domain_data);
+		unset($client_data);
+		
+		// ssh-rsa authentication variables
+		$sshrsa = $this->data['new']['ssh_rsa'];
+		$usrdir = escapeshellcmd($this->data['new']['dir']);
+		$sshdir = $usrdir.'/.ssh';
+		$sshkeys= $usrdir.'/.ssh/authorized_keys';
+		
+		$app->uses('file');
+		$sshrsa = $app->file->unix_nl($sshrsa);
+		$sshrsa = $app->file->remove_blank_lines($sshrsa,0);
+		
+		// If this user has no key yet, generate a pair
+		if ($userkey == '' && $id > 0){
+			//Generate ssh-rsa-keys
+			exec('ssh-keygen -t rsa -C '.$username.'-rsa-key-'.time().' -f /tmp/id_rsa -N ""');
 			
-			// determine the client id
-			$id = $this->data['new']['sys_groupid'];
-			if ($id>0) $id = $id -1;
+			// use the public key that has been generated
+			$userkey = file_get_contents('/tmp/id_rsa.pub');
 			
-			$user = $app->db->queryOneRecord("SELECT * FROM sys_user WHERE client_id  = ".$id);
-			$userkey = $user['ssh_rsa'];
-			$username= $user['username'];
+			// save keypair in client table
+			$this->app->db->query("UPDATE client SET created_at = ".time().", id_rsa = '".file_get_contents('/tmp/id_rsa')."', ssh_rsa = '".$userkey."' WHERE client_id = ".$id);
 			
-			// If this user has no key yet, generate a pair
-			if ($userkey == '') 
-			{		
-				//Generate ssh-rsa-keys
-				exec('ssh-keygen -t rsa -C '.$username.'-rsa-key-'.time().' -f /tmp/id_rsa -N ""');
-				$app->db->query("UPDATE client SET created_at = ".time().", id_rsa = '".file_get_contents('/tmp/id_rsa')."', ssh_rsa = '".file_get_contents('/tmp/id_rsa.pub')."' WHERE client_id = ".$this->id;
-				exec('rm -f /tmp/id_rsa /tmp/id_rsa.pub');
-				
-				$this->app->log("ssh-rsa keypair generated for ".$username,LOGLEVEL_DEBUG);
-			
-			};
-			
-			if (!file_exists($sshkeys))
-			{
-				// add root's key
-				exec("mkdir '$sshdir'");
-				exec("cat /root/.ssh/authorized_keys > '$sshkeys'");
-				exec("echo '' >> '$sshkeys'");
-			
-				// add the user's key
-				exec("echo '$userkey' >> '$sshkeys'");
-				exec("echo '' >> '$sshkeys'");
+			exec('rm -f /tmp/id_rsa /tmp/id_rsa.pub');
+			$this->app->log("ssh-rsa keypair generated for ".$username,LOGLEVEL_DEBUG);
+		};
+
+		if (!file_exists($sshkeys)){
+			// add root's key
+			$app->file->mkdirs($sshdir, '0700');
+			if(is_file('/root/.ssh/authorized_keys')) file_put_contents($sshkeys, file_get_contents('/root/.ssh/authorized_keys'));
+		
+			// Remove duplicate keys
+			$existing_keys = @file($sshkeys);
+			$new_keys = explode("\n", $userkey);
+			$final_keys_arr = @array_merge($existing_keys, $new_keys);
+			$new_final_keys_arr = array();
+			if(is_array($final_keys_arr) && !empty($final_keys_arr)){
+				foreach($final_keys_arr as $key => $val){
+					$new_final_keys_arr[$key] = trim($val);
+				}
 			}
-			// add the custom key 
-			exec("echo '$sshrsa' >> '$sshkeys'");
-			exec("echo '' >> '$sshkeys'");
+			$final_keys = implode("\n", array_flip(array_flip($new_final_keys_arr)));
 			
-			// set proper file permissions
-			exec("chown -R ".escapeshellcmd($this->data['new']['puser']).":".escapeshellcmd($this->data['new']['pgroup'])." ".$usrdir);
-			exec("chmod 600 '$sshkeys'");
-			*/
-			$this->app->log("ssh-rsa key added to ".$sshkeys,LOGLEVEL_DEBUG);
+			// add the user's key
+			file_put_contents($sshkeys, $final_keys);
+			$app->file->remove_blank_lines($sshkeys);
+			$this->app->log("ssh-rsa authorisation keyfile created in ".$sshkeys,LOGLEVEL_DEBUG);
+		}
+			
+		//* Get the keys
+		$existing_keys = file($sshkeys);
+		$new_keys = explode("\n", $sshrsa);
+		$old_keys = explode("\n",$this->data['old']['ssh_rsa']);
+			
+		//* Remove all old keys
+		if(is_array($old_keys)) {
+			foreach($old_keys as $key => $val) {
+				$k = array_search(trim($val),$existing_keys);
+				unset($existing_keys[$k]);
+			}
+		}
+			
+		//* merge the remaining keys and the ones fom the ispconfig database.
+		if(is_array($new_keys)) {
+			$final_keys_arr = array_merge($existing_keys, $new_keys);
+		} else {
+			$final_keys_arr = $existing_keys;
+		}
+			
+		$new_final_keys_arr = array();
+		if(is_array($final_keys_arr) && !empty($final_keys_arr)){
+			foreach($final_keys_arr as $key => $val){
+				$new_final_keys_arr[$key] = trim($val);
+			}
+		}
+		$final_keys = implode("\n", array_flip(array_flip($new_final_keys_arr)));
+			
+		// add the custom key 
+		file_put_contents($sshkeys, $final_keys);
+		$app->file->remove_blank_lines($sshkeys);
+		$this->app->log("ssh-rsa key updated in ".$sshkeys,LOGLEVEL_DEBUG);
+		
+		// set proper file permissions
+		exec("chown -R ".escapeshellcmd($this->data['new']['puser']).":".escapeshellcmd($this->data['new']['pgroup'])." ".$sshdir);
+		exec("chmod 600 '$sshkeys'");
+		
 	}
 	
 

--
Gitblit v1.9.1