vm_openvz_plugin.inc.php

<?php
/**
 * sites_web_domain_plugin plugin
 *
 * @author Till Brehm, projektfarm GmbH
 */


class vm_openvz_plugin {

	var $plugin_name        = 'vm_openvz_plugin';
	var $class_name         = 'vm_openvz_plugin';
	var $id = 0;
	var $dataRecord = array();
	var $oldDataRecord = array();


	/*
            This function is called when the plugin is loaded
    */
	function onLoad() {
		global $app;

		//* Register for events
		$app->plugin->registerEvent('vm:openvz_vm:on_after_insert', 'vm_openvz_plugin', 'openvz_vm_insert');
		$app->plugin->registerEvent('vm:openvz_vm:on_after_update', 'vm_openvz_plugin', 'openvz_vm_update');
		$app->plugin->registerEvent('vm:openvz_vm:on_after_delete', 'vm_openvz_plugin', 'openvz_vm_delete');
	}

	/*
		Function that gets called after a new vm was inserted
    */
	function openvz_vm_insert($event_name, $page_form) {
		global $app, $conf;

		$this->id = $app->functions->intval($page_form->id);
		$this->dataRecord = $page_form->dataRecord;
		$this->oldDataRecord = $page_form->oldDataRecord;

		// make sure that the record belongs to the clinet group and not the admin group when admin inserts it
		// also make sure that the user can not delete domain created by a admin
		if($_SESSION["s"]["user"]["typ"] == 'admin' && isset($this->dataRecord["client_group_id"])) {
			$client_group_id = $app->functions->intval($this->dataRecord["client_group_id"]);
			$app->db->query("UPDATE openvz_vm SET sys_groupid = ? WHERE vm_id = ?", $client_group_id, $this->id);
		}
		if($app->auth->has_clients($_SESSION['s']['user']['userid']) && isset($this->dataRecord["client_group_id"])) {
			$client_group_id = $app->functions->intval($this->dataRecord["client_group_id"]);
			$app->db->query("UPDATE openvz_vm SET sys_groupid = ? WHERE vm_id = ?", $client_group_id, $this->id);
		}

		// Set the VEID
		$tmp = $app->db->queryOneRecord('SELECT MAX(veid) + 1 as newveid FROM openvz_vm');
		$veid = ($tmp['newveid'] > 100)?$tmp['newveid']:101;
		$app->db->query("UPDATE openvz_vm SET veid = ? WHERE vm_id = ?", $veid, $this->id);
		unset($tmp);

		// Apply template values to the advanced tab settings
		$this->applyTemplate();

		// Set the IP address
		$app->db->query("UPDATE openvz_ip SET vm_id = ? WHERE ip_address = ?", $this->id, $this->dataRecord['ip_address']);

		// Create the OpenVZ config file and store it in config field
		$this->makeOpenVZConfig();

		// Create the DNS record
		$this->createDNS();

	}

	/*
		Function that gets called after a vm was updated
    */
	function openvz_vm_update($event_name, $page_form) {
		global $app, $conf;

		$this->id = $app->functions->intval($page_form->id);
		$this->dataRecord = $page_form->dataRecord;
		$this->oldDataRecord = $page_form->oldDataRecord;

		// make sure that the record belongs to the clinet group and not the admin group when a admin inserts it
		// also make sure that the user can not delete domain created by a admin
		if($_SESSION["s"]["user"]["typ"] == 'admin' && isset($this->dataRecord["client_group_id"])) {
			$client_group_id = $app->functions->intval($this->dataRecord["client_group_id"]);
			$app->db->query("UPDATE openvz_vm SET sys_groupid = ? WHERE vm_id = ?", $client_group_id, $this->id);
		}
		if($app->auth->has_clients($_SESSION['s']['user']['userid']) && isset($this->dataRecord["client_group_id"])) {
			$client_group_id = $app->functions->intval($this->dataRecord["client_group_id"]);
			$app->db->query("UPDATE openvz_vm SET sys_groupid = ? WHERE vm_id = ?", $client_group_id, $this->id);
		}

		if(isset($this->dataRecord["ostemplate_id"]) && $this->oldDataRecord["ostemplate_id"] != $this->dataRecord["ostemplate_id"]) {
			$this->applyTemplate();
		}

		// Set the IP address
		if(isset($this->dataRecord['ip_address'])) {
			$app->db->query("UPDATE openvz_ip SET vm_id = 0 WHERE vm_id = ?", $this->id);
			$app->db->query("UPDATE openvz_ip SET vm_id = ? WHERE ip_address = ?", $this->id, $this->dataRecord['ip_address']);
		}

		// Create the OpenVZ config file and store it in config field
		$this->makeOpenVZConfig();

		// Create the DNS record
		if((isset($this->dataRecord['hostname']) && $this->dataRecord['hostname'] != $this->oldDataRecord['hostname'])
			or (isset($this->dataRecord['create_dns']) && $this->dataRecord['create_dns'] != $this->oldDataRecord['create_dns'])) {
			$this->createDNS();
		}

	}

	function openvz_vm_delete($event_name, $page_form) {
		global $app, $conf;

		//* Free the IP address
		$tmp = $app->db->queryOneRecord("SELECT ip_address_id FROM openvz_ip WHERE vm_id = ?", $page_form->id);
		$app->db->datalogUpdate('openvz_ip', array('vm_id' => 0), 'ip_address_id', $tmp['ip_address_id']);
		unset($tmp);

	}

	private function applyTemplate() {
		global $app, $conf;

		$tpl = $app->db->queryOneRecord("SELECT * FROM openvz_template WHERE template_id = ?", $this->dataRecord["template_id"]);

		$sql = "UPDATE openvz_vm SET ";
		$sql .= "diskspace = ?, ";
		$sql .= "ram = ?, ";
		$sql .= "ram_burst = ?, ";
		$sql .= "cpu_units = ?, ";
		$sql .= "cpu_num = ?, ";
		$sql .= "cpu_limit = ?, ";
		$sql .= "io_priority = ?, ";
		$sql .= "nameserver = ?, ";
		$sql .= "create_dns = ?, ";
		$sql .= "capability = ?, ";
		$sql .= "features = ?, ";
		$sql .= "iptables = ? ";
		$sql .= "WHERE vm_id = ?";
		$app->db->query($sql, $tpl['diskspace'], $tpl['ram'], $tpl['ram_burst'], $tpl['cpu_units'], $tpl['cpu_num'], $tpl['cpu_limit'], $tpl['io_priority'], $tpl['nameserver'], $tpl['create_dns'], $tpl['capability'], $tpl['features'], $tpl['iptables'], $this->id);

	}

	private function makeOpenVZConfig() {
		global $app, $conf;

		$vm = $app->db->queryOneRecord("SELECT * FROM openvz_vm WHERE vm_id = ?",$app->functions->intval($this->id));
		$vm_template = $app->db->queryOneRecord("SELECT * FROM openvz_template WHERE template_id = ?",$app->functions->intval($vm['template_id']));
		$burst_ram = $vm['ram_burst']*256;
		$guar_ram = $vm['ram']*256;

		$app->load('tpl');
		$tpl = new tpl();
		$tpl->newTemplate('../vm/templates/openvz.conf.tpl');

		$onboot = ($vm['start_boot'] == 'y')?'yes':'no';
		$tpl->setVar('onboot', $onboot);

		$tpl->setVar('kmemsize', $vm_template['kmemsize']);
		$tpl->setVar('lockedpages', $vm_template['lockedpages']);
		$tpl->setVar('privvmpages', $burst_ram.':'.$burst_ram);
		$tpl->setVar('shmpages', $guar_ram.':'.$guar_ram);
		$tpl->setVar('numproc', $vm_template['numproc']);
		$tpl->setVar('physpages', $vm_template['physpages']);
		$tpl->setVar('vmguarpages', $guar_ram.':'.$guar_ram);
		$tpl->setVar('oomguarpages', $guar_ram.':'.$guar_ram);
		$tpl->setVar('numtcpsock', $vm_template['numtcpsock']);
		$tpl->setVar('numflock', $vm_template['numflock']);
		$tpl->setVar('numpty', $vm_template['numpty']);
		$tpl->setVar('numsiginfo', $vm_template['numsiginfo']);
		$tpl->setVar('tcpsndbuf', $vm_template['tcpsndbuf']);
		$tpl->setVar('tcprcvbuf', $vm_template['tcprcvbuf']);
		$tpl->setVar('othersockbuf', $vm_template['othersockbuf']);
		$tpl->setVar('dgramrcvbuf', $vm_template['dgramrcvbuf']);
		$tpl->setVar('numothersock', $vm_template['numothersock']);
		$tpl->setVar('dcachesize', $vm_template['dcachesize']);
		$tpl->setVar('numfile', $vm_template['numfile']);
		$tpl->setVar('avnumproc', $vm_template['avnumproc']);
		$tpl->setVar('numiptent', $vm_template['numiptent']);
		$tpl->setVar('swappages', $vm_template['swappages']);

		$diskspace = $vm['diskspace']*1048576;
		$diskinodes = $vm['diskspace']*524288;

		$tpl->setVar('diskspace', $diskspace.":".$diskspace);
		$tpl->setVar('diskinodes', $diskinodes.":".$diskinodes);
		$tpl->setVar('io_priority', $vm['io_priority']);

		$tpl->setVar('cpu_num', $vm['cpu_num']);
		$tpl->setVar('cpu_units', $vm['cpu_units']);
		$tpl->setVar('cpu_limit', $vm['cpu_limit']);

		$hostname = str_replace('{VEID}', $vm['veid'], $vm['hostname']);

		$tpl->setVar('hostname', $hostname);
		$tpl->setVar('ip_address', $vm['ip_address']);
		$tpl->setVar('nameserver', $vm['nameserver']);
		$tpl->setVar('capability', $vm['capability']);
		$tpl->setVar('features', $vm['features']);
		$tpl->setVar('iptables', $vm['iptables']);

		$tmp = $app->db->queryOneRecord("SELECT template_file FROM openvz_ostemplate WHERE ostemplate_id = ?", $app->functions->intval($vm['ostemplate_id']));
		$tpl->setVar('ostemplate', $tmp['template_file']);
		unset($tmp);

		$openvz_config = $tpl->grab();
		$app->db->query("UPDATE openvz_vm SET config = ? WHERE vm_id = ?", $openvz_config, $app->functions->intval($this->id));

		unset($tpl);

	}

	private function createDNS() {
		global $app, $conf;

		$vm = $app->db->queryOneRecord("SELECT * FROM openvz_vm WHERE vm_id = ?", $app->functions->intval($this->id));

		if($vm['create_dns'] != 'y') return;

		$full_hostname = str_replace('{VEID}', $vm['veid'], $vm['hostname']);
		$hostname_parts = explode('.', $full_hostname);
		$hostname = $hostname_parts[0];
		unset($hostname_parts[0]);
		$zone = implode('.', $hostname_parts);
		unset($hostname_parts);

		// Find the dns zone
		$zone_rec = $app->db->queryOneRecord("SELECT * FROM dns_soa WHERE origin = ?", $zone);
		$rr_rec = $app->db->queryOneRecord("SELECT * FROM dns_rr WHERE zone = ? AND name = ?", $zone_rec['id'], $hostname);

		if($zone_rec['id'] > 0) {
			$ip_address = $vm['ip_address'];
			$sys_userid = $app->functions->intval($zone_rec['sys_userid']);
			$sys_groupid = $app->functions->intval($zone_rec['sys_groupid']);
			$server_id = $app->functions->intval($zone_rec['server_id']);
			$dns_soa_id = $app->functions->intval($zone_rec['id']);

			if($rr_rec['id'] > 0) {
				$app->uses('validate_dns');
				$app->db->datalogUpdate('dns_rr', array("data" => $ip_address), 'id', $app->functions->intval($rr_rec['id']));
				$serial = $app->validate_dns->increase_serial($zone_rec['serial']);
				$app->db->datalogUpdate('dns_soa', array("serial" => $serial), 'id', $app->functions->intval($zone_rec['id']));
			} else {
				$insert_data = array(
					"sys_userid" => $sys_userid,
					"sys_groupid" => $sys_groupid,
					"sys_perm_user" => 'riud',
					"sys_perm_group" => 'riud',
					"sys_perm_other" => '',
					"server_id" => $server_id,
					"zone" => $dns_soa_id,
					"name" => $hostname,
					"type" => 'A',
					"data" => $ip_address,
					"aux" => '0',
					"ttl" => '3600',
					"active" => 'Y'
				);
				$dns_rr_id = $app->db->datalogInsert('dns_rr', $insert_data, 'id');
			}

		}
	}

}