From 2d4d61bb4b01ede5b96e5d1c71390315e1100007 Mon Sep 17 00:00:00 2001
From: ftimme <ft@falkotimme.com>
Date: Tue, 18 Oct 2011 16:59:31 +0000
Subject: [PATCH] - Added check for duplicate SSH-RSA keys. - Replaced SSH-RSA
 input field with textarea so that you can put in multiple SSH-RSA keys.

---
 .../web/sites/lib/lang/de_shell_user.lng      |  1 +
 .../web/sites/lib/lang/en_shell_user.lng      |  1 +
 interface/web/sites/shell_user_edit.php       |  2 +
 .../web/sites/templates/shell_user_edit.htm   |  4 +-
 .../shelluser_base_plugin.inc.php             | 64 ++++++++++++-----
 .../shelluser_jailkit_plugin.inc.php          | 68 +++++++++++++------
 6 files changed, 100 insertions(+), 40 deletions(-)

diff --git a/interface/web/sites/lib/lang/de_shell_user.lng b/interface/web/sites/lib/lang/de_shell_user.lng
index 4f0b2ec24..c2373ad83 100644
--- a/interface/web/sites/lib/lang/de_shell_user.lng
+++ b/interface/web/sites/lib/lang/de_shell_user.lng
@@ -19,4 +19,5 @@ $wb['password_strength_txt'] = 'Passwortkomplexität';
 $wb['parent_domain_id_error_empty'] = 'Keine Website ausgewählt.';
 $wb['puser_txt'] = 'Web Username';
 $wb['pgroup_txt'] = 'Web Group';
+$wb['ssh_rsa_txt'] = 'SSH-RSA Public Key (f&uuml;r key-basierte Logins)';
 ?>
diff --git a/interface/web/sites/lib/lang/en_shell_user.lng b/interface/web/sites/lib/lang/en_shell_user.lng
index f98f3f980..e05fd258e 100644
--- a/interface/web/sites/lib/lang/en_shell_user.lng
+++ b/interface/web/sites/lib/lang/en_shell_user.lng
@@ -20,4 +20,5 @@ $wb["uid_error_empty"] = 'GID empty.';
 $wb["directory_error_empty"] = 'Directory empty.';
 $wb["limit_shell_user_txt"] = 'The max number of shell users is reached.';
 $wb["parent_domain_id_error_empty"] = 'No website selected.';
+$wb["ssh_rsa_txt"] = 'SSH-RSA Public Key (for key-based logins)';
 ?>
diff --git a/interface/web/sites/shell_user_edit.php b/interface/web/sites/shell_user_edit.php
index cf4a585a9..bb93bf222 100644
--- a/interface/web/sites/shell_user_edit.php
+++ b/interface/web/sites/shell_user_edit.php
@@ -112,6 +112,8 @@ class page_action extends tform_actions {
 		if(isset($this->dataRecord['username']) && trim($this->dataRecord['username']) == '') $app->tform->errorMessage .= $app->tform->lng('username_error_empty').'<br />';
 		if(isset($this->dataRecord['username']) && empty($this->dataRecord['parent_domain_id'])) $app->tform->errorMessage .= $app->tform->lng('parent_domain_id_error_empty').'<br />';
 		
+		if(isset($this->dataRecord['ssh_rsa'])) $this->dataRecord['ssh_rsa'] = trim($this->dataRecord['ssh_rsa']);
+		
 		parent::onSubmit();
 	}
 	
diff --git a/interface/web/sites/templates/shell_user_edit.htm b/interface/web/sites/templates/shell_user_edit.htm
index 45bdcc408..3924d39e7 100644
--- a/interface/web/sites/templates/shell_user_edit.htm
+++ b/interface/web/sites/templates/shell_user_edit.htm
@@ -44,8 +44,8 @@
         <input name="quota_size" id="quota_size" value="{tmpl_var name='quota_size'}" size="7" maxlength="7" type="text" class="textInput formLengthLimit" /><p class="formInputUnity">&nbsp;MB</p>
 			</div>
 	<div class="ctrlHolder">
-      		<label for="ssh_rsa">ssh_rsa</label>
-        	<input name="ssh_rsa" id="ssh_rsa" value="" maxlength="600" type="text" class="textInput" />
+      		<label for="ssh_rsa">{tmpl_var name='ssh_rsa_txt'}</label>
+			<textarea name="ssh_rsa" id="ssh_rsa" rows="10" cols="30">{tmpl_var name='ssh_rsa'}</textarea>
 			</div>
       <div class="ctrlHolder">
 				<p class="label">{tmpl_var name='active_txt'}</p>
diff --git a/server/plugins-available/shelluser_base_plugin.inc.php b/server/plugins-available/shelluser_base_plugin.inc.php
index 58858a3bc..946aaaa4c 100755
--- a/server/plugins-available/shelluser_base_plugin.inc.php
+++ b/server/plugins-available/shelluser_base_plugin.inc.php
@@ -195,6 +195,7 @@ class shelluser_base_plugin {
 	}
 	
 	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']));
@@ -207,41 +208,68 @@ class shelluser_base_plugin {
 		unset($client_data);
 		
 		// ssh-rsa authentication variables
-		$sshrsa = escapeshellcmd($this->data['new']['ssh_rsa']);
+		$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) 
-		{
+		if ($userkey == '' && $id > 0){
 			//Generate ssh-rsa-keys
 			exec('ssh-keygen -t rsa -C '.$username.'-rsa-key-'.time().' -f /tmp/id_rsa -N ""');
+			
+			// use the public key that has been generated
+			$userkey = file_get_contents('/tmp/id_rsa.pub');
+			
 			// 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 = '".file_get_contents('/tmp/id_rsa.pub')."' WHERE client_id = ".$id);
-			// and use the public key that has been generated
-			$userkey = file_get_contents('/tmp/id_rsa.pub')
-			;
+			$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);
+			
 			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))
-		{
+
+		if (!file_exists($sshkeys)){
 			// add root's key
-			exec("mkdir '$sshdir'");
-			exec("cat /root/.ssh/authorized_keys > '$sshkeys'");
-			exec("echo '' >> '$sshkeys'");
+			$app->file->mkdirs($sshdir, '0755');
+			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);
+				}
+			}
+			$final_keys = implode("\n", array_flip(array_flip($new_final_keys_arr)));
+			
 			// add the user's key
-			exec("echo '$userkey' >> '$sshkeys'");
-			exec("echo '' >> '$sshkeys'");
+			file_put_contents($sshkeys, $final_keys);
+			$app->file->remove_blank_lines($sshkeys);
 			$this->app->log("ssh-rsa authorisation keyfile created in ".$sshkeys,LOGLEVEL_DEBUG);
 		}
-		if ($sshrsa!=''){
+		if ($sshrsa != ''){
+			// Remove duplicate keys
+			$existing_keys = file($sshkeys);
+			$new_keys = explode("\n", $sshrsa);
+			$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);
+				}
+			}
+			$final_keys = implode("\n", array_flip(array_flip($new_final_keys_arr)));
+			
 			// add the custom key 
-			exec("echo '$sshrsa' >> '$sshkeys'");
-			exec("echo '' >> '$sshkeys'");
+			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
diff --git a/server/plugins-available/shelluser_jailkit_plugin.inc.php b/server/plugins-available/shelluser_jailkit_plugin.inc.php
index 31db30f55..c1e2b2cd3 100755
--- a/server/plugins-available/shelluser_jailkit_plugin.inc.php
+++ b/server/plugins-available/shelluser_jailkit_plugin.inc.php
@@ -364,6 +364,7 @@ class shelluser_jailkit_plugin {
 	}
 
 	private function _setup_ssh_rsa() {
+		global $app;
 		$this->app->log("ssh-rsa setup shelluser_jailkit",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']));
@@ -376,41 +377,68 @@ class shelluser_jailkit_plugin {
 		unset($client_data);
 		
 		// ssh-rsa authentication variables
-		$sshrsa = escapeshellcmd($this->data['new']['ssh_rsa']);
+		$sshrsa = $this->data['new']['ssh_rsa'];
 		$usrdir = escapeshellcmd($this->data['new']['dir']).'/'.$this->_get_home_dir($this->data['new']['username']);
-			$sshdir = $usrdir.'/.ssh';
-			$sshkeys= $usrdir.'/.ssh/authorized_keys';
+		$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) 
-		{
+		if ($userkey == '' && $id > 0){
 			//Generate ssh-rsa-keys
 			exec('ssh-keygen -t rsa -C '.$username.'-rsa-key-'.time().' -f /tmp/id_rsa -N ""');
+			
+			// use the public key that has been generated
+			$userkey = file_get_contents('/tmp/id_rsa.pub');
+			
 			// 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 = '".file_get_contents('/tmp/id_rsa.pub')."' WHERE client_id = ".$id);
-			// and use the public key that has been generated
-			$userkey = file_get_contents('/tmp/id_rsa.pub')
-			;
+			$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);
+
 			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))
-		{
+		if (!file_exists($sshkeys)){
 			// add root's key
-			exec("mkdir '$sshdir'");
-			exec("cat /root/.ssh/authorized_keys > '$sshkeys'");
-			exec("echo '' >> '$sshkeys'");
-		
+			$app->file->mkdirs($sshdir, '0755');
+			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);
+				}
+			}
+			$final_keys = implode("\n", array_flip(array_flip($new_final_keys_arr)));
+			
 			// add the user's key
-			exec("echo '$userkey' >> '$sshkeys'");
-			exec("echo '' >> '$sshkeys'");
+			file_put_contents($sshkeys, $final_keys);
+			$app->file->remove_blank_lines($sshkeys);
 			$this->app->log("ssh-rsa authorisation keyfile created in ".$sshkeys,LOGLEVEL_DEBUG);
 		}
-		if ($sshrsa!=''){
+		if ($sshrsa != ''){
+			// Remove duplicate keys
+			$existing_keys = file($sshkeys);
+			$new_keys = explode("\n", $sshrsa);
+			$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);
+				}
+			}
+			$final_keys = implode("\n", array_flip(array_flip($new_final_keys_arr)));
+			
 			// add the custom key 
-			exec("echo '$sshrsa' >> '$sshkeys'");
-			exec("echo '' >> '$sshkeys'");
+			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
-- 
GitLab