158 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			PHP
		
	
	
	
			
		
		
	
	
			158 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			PHP
		
	
	
	
| <?php
 | |
| 
 | |
| include_once(dirname(__FILE__) . '/vm.class.php');
 | |
| # include('vm.php'); // This one fails ...
 | |
| 
 | |
| class m_lxc implements vm
 | |
| {
 | |
|   public $IP;
 | |
|   public $PORT;
 | |
|   public $TIMEOUT = 5;
 | |
|   public $error = array();
 | |
| 
 | |
|   function m_lxc() {
 | |
|     $this->IP   = variable_get('lxc_ip', '', "IP address of the Alternc's LXC server. If empty, no LXC server.");
 | |
|     $this->PORT = variable_get('lxc_port', '6504', "Port of the Alternc's LXC server");
 | |
|     $this->KEY  = variable_get('lxc_key', '', "Shared key with the Alternc's LXC server");
 | |
|   }
 | |
| 
 | |
| 
 | |
|   function hook_menu() {
 | |
|     if ( empty($this->IP)) return ; # No menu if no server
 | |
| 
 | |
|     $obj = array(
 | |
|       'title'       => _("Console access"),
 | |
|       'ico'         => 'images/ssh.png',
 | |
|       'link'        => 'vm.php',
 | |
|       'pos'         => 95,
 | |
|      ) ;
 | |
| 
 | |
|      return $obj;
 | |
|   }
 | |
| 
 | |
|   function hook_admin_del_member() {
 | |
|     global $db,$err,$cuid;
 | |
|     $err->log("lxc","alternc_del_member");
 | |
|     $db->query("DELETE FROM vm_history WHERE uid='$cuid'");
 | |
|     return true;
 | |
|   }
 | |
| 
 | |
| 
 | |
| 	private function sendMessage($action, $user = FALSE, $password=FALSE, $uid=FALSE)
 | |
| 	{
 | |
| 		$fp = fsockopen($this->IP, $this->PORT, $errno, $errstr, $this->TIMEOUT);
 | |
| 		if (!$fp) 
 | |
| 		{
 | |
| 			$this->error[] = 'Unable to connect';
 | |
| 			return FALSE;
 | |
| 		}
 | |
| 
 | |
| 		$msg = sprintf("%s|%s|%s|%d\n", $action, $user, $password, $uid);
 | |
| 		if (fwrite ($fp, $msg) < 0)
 | |
| 		{
 | |
| 			$this->error[] = 'Unable to send data';
 | |
| 			return FALSE;
 | |
| 		}
 | |
| 		$resp = '';
 | |
| 		#while (($resp .= fgets($fp, 4096)) !== FALSE);
 | |
| 		$resp = fgets($fp, 4096);
 | |
| 		fclose ($fp);
 | |
| 
 | |
| 
 | |
| 		return $resp;
 | |
| 	
 | |
| 		if (stripos($resp, 'error') > 0)
 | |
| 		{
 | |
| 			$data = unserialize($resp);
 | |
| 			$this->error[] = $data['msg'];
 | |
| 			return FALSE;
 | |
| 		}
 | |
| 		else
 | |
| 		{
 | |
| 			return $resp;
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	public function start($login = FALSE, $pass = FALSE, $uid = FALSE)
 | |
| 	{
 | |
| 		
 | |
| 		global $mem, $db, $err;
 | |
| 
 | |
| 		if ($this->getvm() !== FALSE)
 | |
| 		{
 | |
| 			$err->raise('lxc', _('VM already started'));
 | |
| 			return FALSE;
 | |
| 		}
 | |
| 
 | |
| 		$user = $login ? $login : $mem->user['login'];
 | |
| 		$pass = $pass  ? $pass  : $mem->user['pass'];
 | |
| 		$uid  = $uid   ? $uid   : $mem->user['uid'];
 | |
| 
 | |
| 		$res = $this->sendMessage('start', $user, $pass, $uid);
 | |
| 		if ($res === FALSE)
 | |
| 			return $this->error;
 | |
| 		else
 | |
| 		{
 | |
| 			$data = unserialize($res);
 | |
| 			$error = $data['error'];
 | |
| 			$hostname = $data['hostname'];
 | |
| 			$msg = $data['msg'];
 | |
| 			$date_start = 'NOW()';
 | |
| 			$uid = $mem->user['uid'];
 | |
| 
 | |
| 			if ((int)$data['error'] != 0)
 | |
| 			{
 | |
| 				$err->raise('lxc', _($data['msg']));
 | |
| 				return FALSE;
 | |
| 			}
 | |
| 
 | |
| 			$db->query("INSERT INTO vm_history (ip,date_start,uid,serialized_object) VALUES ('$hostname', $date_start, '$uid', '$res')");
 | |
| 
 | |
| 			return $res;
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 
 | |
| 	public function monit()
 | |
| 	{
 | |
| 		echo "1 / 5 used ";
 | |
| 	}
 | |
| 
 | |
| 	public function getvm()
 | |
| 	{
 | |
| 		global $db, $mem;
 | |
| 
 | |
| 		$uid = $mem->user['uid'];
 | |
| 		$res = array();
 | |
| 
 | |
| 		$res = $db->query("SELECT * FROM vm_history WHERE date_end IS NULL AND uid= '$uid' ORDER BY id DESC LIMIT 1");
 | |
| 
 | |
| 		if ($db->next_record())
 | |
| 		{
 | |
| 			$db->Record['serialized_object'] = unserialize($db->Record['serialized_object']);
 | |
| 			return $db->Record;
 | |
| 		}
 | |
| 		else
 | |
| 			return FALSE;
 | |
| 	}
 | |
| 
 | |
| 	public function stop()
 | |
| 	{
 | |
| 		global $db, $mem;
 | |
| 
 | |
| 		$vm = $this->getvm();
 | |
| 
 | |
| 		if ($vm === FALSE)
 | |
| 			return TRUE;
 | |
| 
 | |
| 		$vm_id = $vm['serialized_object']['vm'];
 | |
| 		$uid = $mem->user['uid'];
 | |
| 		$vid = $vm['id'];
 | |
| 
 | |
| 		if ($this->sendMessage('stop', $vm_id, FALSE, FALSE) === FALSE)
 | |
| 			return FALSE;
 | |
| 
 | |
| 		return $db->query("UPDATE vm_history SET date_end = NOW() WHERE uid = '$uid' AND id = '$vid' LIMIT 1");
 | |
| 	}
 | |
| }
 |