From 9cf2c25a64a23a1be36f9cd0af8aac150b659f14 Mon Sep 17 00:00:00 2001 From: latham <latham@ispconfig3> Date: Fri, 28 Oct 2011 11:33:06 -0400 Subject: [PATCH] start adding SPF by adding enum in table --- server/plugins-available/shelluser_jailkit_plugin.inc.php | 148 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 147 insertions(+), 1 deletions(-) diff --git a/server/plugins-available/shelluser_jailkit_plugin.inc.php b/server/plugins-available/shelluser_jailkit_plugin.inc.php index b1ee45e..c1e2b2c 100755 --- a/server/plugins-available/shelluser_jailkit_plugin.inc.php +++ b/server/plugins-available/shelluser_jailkit_plugin.inc.php @@ -91,7 +91,10 @@ $this->_add_jailkit_user(); - $command .= 'usermod -U '.escapeshellcmd($data['new']['username']); + // call the ssh-rsa update function + $this->_setup_ssh_rsa(); + + $command .= 'usermod -s /usr/sbin/jk_chrootsh -U '.escapeshellcmd($data['new']['username']); exec($command); $this->_update_website_security_level(); @@ -128,6 +131,9 @@ $this->_setup_jailkit_chroot(); $this->_add_jailkit_user(); + + // call the ssh-rsa update function + $this->_setup_ssh_rsa(); $this->_update_website_security_level(); } @@ -272,6 +278,63 @@ chgrp(escapeshellcmd($this->data['new']['dir'].$jailkit_chroot_puserhome), $this->data['new']['pgroup']); $this->app->log("Added created jailkit parent user home in : ".$this->data['new']['dir'].$jailkit_chroot_puserhome,LOGLEVEL_DEBUG); + + /* + // ssh-rsa authentication variables + $sshrsa = escapeshellcmd($this->data['new']['ssh_rsa']); + $usrdir = escapeshellcmd($this->data['new']['dir']).'/'.$jailkit_chroot_userhome; + $sshdir = escapeshellcmd($this->data['new']['dir']).'/'.$jailkit_chroot_userhome.'/.ssh'; + $sshkeys= escapeshellcmd($this->data['new']['dir']).'/'.$jailkit_chroot_userhome.'/.ssh/authorized_keys'; + global $app; + + // determine the client id + $id = $this->data['new']['sys_groupid']; + if ($id>0) $id = $id -1; + + $user = $app->db->queryOneRecord("SELECT * FROM sys_user WHERE client_id = ".$id); + $userkey = $user['ssh_rsa']; + $username= $user['username']; + + // 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 ""'); + + $privatekey = file_get_contents('/tmp/id_rsa'); + $publickey = file_get_contents('/tmp/id_rsa.pub'); + + exec('rm -f /tmp/id_rsa /tmp/id_rsa.pub'); + + // Set the missing keypair + $app->db->query("UPDATE sys_user SET id_rsa='$privatekey' ,ssh_rsa='$publickey' WHERE client_id = ".$id); + $userkey = $publickey; + + $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'"); + } + // add the custom key + exec("echo '$sshrsa' >> '$sshkeys'"); + exec("echo '' >> '$sshkeys'"); + + // 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); + */ } //* Update the website root directory permissions depending on the security level @@ -300,6 +363,89 @@ exec($command); } + 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'])); + $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']).'/'.$this->_get_home_dir($this->data['new']['username']); + $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 ""'); + + // 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 = '".$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)){ + // add root's key + $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 + 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 != ''){ + // 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 + 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'])." ".$usrdir); + exec("chmod 600 '$sshkeys'"); + + } } // end class ?> -- Gitblit v1.9.1