From b4945ae0c3f4e108090d05b34b3e7df8a753a2ec Mon Sep 17 00:00:00 2001
From: Marius Cramer <m.cramer@pixcept.de>
Date: Mon, 07 Jul 2014 13:07:17 -0400
Subject: [PATCH] Merge branch 'master' into 'master'

---
 server/plugins-available/mail_plugin_dkim.inc.php |  218 +++++++++++++++++++++++++++++++++++------------------
 1 files changed, 143 insertions(+), 75 deletions(-)

diff --git a/server/plugins-available/mail_plugin_dkim.inc.php b/server/plugins-available/mail_plugin_dkim.inc.php
index 322ccd8..268c9c5 100644
--- a/server/plugins-available/mail_plugin_dkim.inc.php
+++ b/server/plugins-available/mail_plugin_dkim.inc.php
@@ -41,7 +41,6 @@
 	// private variables
 	var $action = '';
 
-
 	/**
 	 * This function is called during ispconfig installation to determine
 	 * if a symlink shall be created for this plugin.
@@ -57,7 +56,6 @@
 
 	}
 
-
 	/**
 	 * This function is called when the plugin is loaded
 	 */
@@ -71,7 +69,6 @@
 		$app->plugins->registerEvent('mail_domain_update', $this->plugin_name, 'domain_dkim_update');
 	}
 
-
 	/**
 	 * This function gets the amavisd-config file
 	 * @return string path to the amavisd-config for dkim-keys
@@ -80,7 +77,8 @@
 		$pos_config=array(
 			'/etc/amavisd.conf',
 			'/etc/amavisd.conf/50-user',
-			'/etc/amavis/conf.d/50-user'
+			'/etc/amavis/conf.d/50-user',
+			'/etc/amavisd/amavisd.conf'
 		);
 		$amavis_configfile='';
 		foreach($pos_config as $conf) {
@@ -89,9 +87,12 @@
 				break;
 			}
 		}
+		//* If we can use seperate config-files with amavis use 60-dkim
+		if (substr_compare($amavis_configfile, '50-user', -7) === 0)
+			$amavis_configfile = str_replace('50-user', '60-dkim', $amavis_configfile);
+
 		return $amavis_configfile;
 	}
-
 
 	/**
 	 * This function checks the relevant configs and disables dkim for the domain
@@ -101,42 +102,62 @@
 	 */
 	function check_system($data) {
 		global $app, $mail_config;
+
 		$app->uses('getconf');
 		$check=true;
+
 		/* check for amavis-config */
-		if ( $this->get_amavis_config() == '' || !is_writeable($this->get_amavis_config()) ) {
+		$amavis_configfile = $this->get_amavis_config();
+
+		//* When we can use 60-dkim for the dkim-keys create the file if it does not exists.
+		if (substr_compare($amavis_configfile, '60-dkim', -7) === 0 && !file_exists($amavis_configfile))
+			file_put_contents($amavis_configfile, '');
+
+		if ( $amavis_configfile == '' || !is_writeable($amavis_configfile) ) {
 			$app->log('Amavis-config not found or not writeable.', LOGLEVEL_ERROR);
 			$check=false;
 		}
 		/* dir for dkim-keys writeable? */
 		$mail_config = $app->getconf->get_server_config($conf['server_id'], 'mail');
-		if (isset($mail_config['dkim_path']) && isset($data['new']['dkim_private']) && !empty($data['new']['dkim_private'])) {
+		if (isset($mail_config['dkim_path']) && (!empty($mail_config['dkim_path'])) && isset($data['new']['dkim_private']) && !empty($data['new']['dkim_private'])) {
+
+            if (!is_dir($mail_config['dkim_path'])) {
+                $app->log('DKIM Path '.$mail_config['dkim_path'].' not found - (re)created.', LOGLEVEL_DEBUG);
+                mkdir($mail_config['dkim_path'], 0750, true);
+            }
+
 			if (!is_writeable($mail_config['dkim_path'])) {
-				$app->log('DKIM Path '.$mail_config['dkim_path'].' not found or not writeable.', LOGLEVEL_ERROR);
+				$app->log('DKIM Path '.$mail_config['dkim_path'].' not writeable.', LOGLEVEL_ERROR);
 				$check=false;
 			}
+
 		} else {
-			$app->log('Unable to write DKIM settings; Check your config!', LOGLEVEL_ERROR);
+			$app->log('Unable to write DKIM settings - no DKIM-Path defined', LOGLEVEL_ERROR);
 			$check=false;
 		}
 		return $check;
 	}
-
-
-
-
 
 	/**
 	 * This function restarts amavis
 	 */
 	function restart_amavis() {
 		global $app, $conf;
-		$initfile=$conf['init_scripts'].'/amavis';
-		$app->log('Restarting amavis.', LOGLEVEL_DEBUG);
-		exec(escapeshellarg($conf['init_scripts']).escapeshellarg('/amavis').' restart', $output);
+		$pos_init=array(
+			$conf['init_scripts'].'/amavis',
+			$conf['init_scripts'].'/amavisd'
+		);
+		$initfile='';
+		foreach($pos_init as $init) {
+			if (is_executable($init)) {
+				$initfile=$init;
+				break;
+				}
+		}
+		$app->log('Restarting amavis: '.$initfile.'.', LOGLEVEL_DEBUG);
+		exec(escapeshellarg($initfile).' restart', $output);
 		foreach($output as $logline) $app->log($logline, LOGLEVEL_DEBUG);
 	}
-
 
 	/**
 	 * This function writes the keyfiles (public and private)
@@ -158,11 +179,10 @@
 			/* save the DKIM Public-key in dkim-dir */
 			if (!file_put_contents($key_file.'.public', $public_key) === false)
 				$app->log('Saved DKIM Public to '.$key_domain.'.', LOGLEVEL_DEBUG);
-			else $app->log('Unable to save DKIM Public to '.$key_domain.'.', LOGLEVEL_WARNING);
+			else $app->log('Unable to save DKIM Public to '.$key_domain.'.', LOGLEVEL_DEBUG);
 		}
 		return $success;
 	}
-
 
 	/**
 	 * This function removes the keyfiles
@@ -181,22 +201,44 @@
 		} else $app->log('Unable to delete the DKIM Public-key for '.$key_domain.' (not found).', LOGLEVEL_DEBUG);
 	}
 
-
 	/**
 	 * This function adds the entry to the amavisd-config
 	 * @param string $key_domain mail-domain
 	 */
-	function add_to_amavis($key_domain) {
+	function add_to_amavis($key_domain, $selector, $old_selector) {
 		global $app, $mail_config;
-		$amavis_config = file_get_contents($this->get_amavis_config());
-		$key_value="dkim_key('".$key_domain."', 'default', '".$mail_config['dkim_path']."/".$key_domain.".private');\n";
-		if(strpos($amavis_config, $key_value) !== false) $amavis_config = str_replace($key_value, '', $amavis_config);
-		if (!file_put_contents($this->get_amavis_config(), $key_value, FILE_APPEND) === false) {
-			$app->log('Adding DKIM Private-key to amavis-config.', LOGLEVEL_DEBUG);
-			$this->restart_amavis();
-		}
-	}
 
+		if (empty($selector)) $selector = 'default';
+		$restart = false;
+		$amavis_configfile = $this->get_amavis_config();
+
+		$search_regex = "/(\n|\r)?dkim_key\(\'".$key_domain."\',\ \'(".$selector."|".$old_selector."){1}?\'.*/";
+
+		//* If we are using seperate config-files with amavis remove existing keys from 50-user to avoid duplicate keys
+		if (substr_compare($amavis_configfile, '60-dkim', -7) === 0) {
+			$temp_configfile = str_replace('60-dkim', '50-user', $amavis_configfile);
+			$temp_config = file_get_contents($temp_configfile);
+			if (preg_match($search_regex, $temp_config)) {
+				$temp_config = preg_replace($search_regex, '', $temp_config)."\n";
+				file_put_contents($temp_configfile, $temp_config);
+			}
+			unset($temp_configfile);
+			unset($temp_config);
+		}
+
+		$key_value="dkim_key('".$key_domain."', '".$selector."', '".$mail_config['dkim_path']."/".$key_domain.".private');\n";
+		$amavis_config = file_get_contents($amavis_configfile);
+		$amavis_config = preg_replace($search_regex, '', $amavis_config).$key_value;
+
+		if (file_put_contents($amavis_configfile, $amavis_config)) {
+			$app->log('Adding DKIM Private-key to amavis-config.', LOGLEVEL_DEBUG);
+			$restart = true;
+		} else {
+			$app->log('Unable to add DKIM Private-key for '.$key_domain.' to amavis-config.', LOGLEVEL_ERROR);
+		}
+
+		return $restart;
+	}
 
 	/**
 	 * This function removes the entry from the amavisd-config
@@ -204,22 +246,35 @@
 	 */
 	function remove_from_amavis($key_domain) {
 		global $app;
-		$amavis_config = file($this->get_amavis_config());
-		$i=0;$found=false;
-		foreach($amavis_config as $line) {
-			if (preg_match("/^\bdkim_key\b.*\b".$key_domain."\b/", $line)) {
-				unset($amavis_config[$i]);
-				$found=true;
-			}
-			$i++;
-		}
-		if ($found) {
-			file_put_contents($this->get_amavis_config(), $amavis_config);
-			$app->log('Deleted the DKIM settings from amavis-config for '.$key_domain.'.', LOGLEVEL_DEBUG);
-			$this->restart_amavis();
-		} else $app->log('Unable to delete the DKIM settings from amavis-config for '.$key_domain.'.', LOGLEVEL_ERROR);
-	}
 
+		$restart = false;
+		$amavis_configfile = $this->get_amavis_config();
+		$amavis_config = file_get_contents($amavis_configfile);
+
+		$search_regex = "/(\n|\r)?dkim_key.*".$key_domain.".*(\n|\r)?/";
+
+		if (preg_match($search_regex, $amavis_config)) {
+			$amavis_config = preg_replace($search_regex, '', $amavis_config);
+			file_put_contents($amavis_configfile, $amavis_config);
+			$app->log('Deleted the DKIM settings from amavis-config for '.$key_domain.'.', LOGLEVEL_DEBUG);
+			$restart = true;
+		}
+
+		//* If we are using seperate config-files with amavis remove existing keys from 50-user, too
+		if (substr_compare($amavis_configfile, '60-dkim', -7) === 0) {
+			$temp_configfile = str_replace('60-dkim', '50-user', $amavis_configfile);
+			$temp_config = file_get_contents($temp_configfile);
+			if (preg_match($search_regex, $temp_config)) {
+				$temp_config = preg_replace($search_regex, '', $temp_config);
+				file_put_contents($temp_configfile, $temp_config);
+				$restart = true;
+			}
+			unset($temp_configfile);
+			unset($temp_config);
+		}
+
+		return $restart;
+	}
 
 	/**
 	 * This function controlls new key-files and amavisd-entries
@@ -232,16 +287,16 @@
 			if ( substr($mail_config['dkim_path'], strlen($mail_config['dkim_path'])-1) == '/' )
 				$mail_config['dkim_path'] = substr($mail_config['dkim_path'], 0, strlen($mail_config['dkim_path'])-1);
 			if ($this->write_dkim_key($mail_config['dkim_path']."/".$data['new']['domain'], $data['new']['dkim_private'], $data['new']['domain'])) {
-				$this->add_to_amavis($data['new']['domain']);
+				if ($this->add_to_amavis($data['new']['domain'], $data['new']['dkim_selector'], $data['old']['dkim_selector'] )) {
+					$this->restart_amavis();
+				} else {
+					$this->remove_dkim_key($mail_config['dkim_path']."/".$data['new']['domain'], $data['new']['domain']);
+				}
 			} else {
 				$app->log('Error saving the DKIM Private-key for '.$data['new']['domain'].' - DKIM is not enabled for the domain.', LOGLEVEL_ERROR);
 			}
 		}
-		else {
-			$app->log('DKIM for '.$data['new']['domain'].' not written to disk - domain is inactive', LOGLEVEL_DEBUG);
-		}
 	}
-
 
 	/**
 	 * This function controlls the removement of keyfiles (public and private)
@@ -254,29 +309,27 @@
 		if ( substr($mail_config['dkim_path'], strlen($mail_config['dkim_path'])-1) == '/' )
 			$mail_config['dkim_path'] = substr($mail_config['dkim_path'], 0, strlen($mail_config['dkim_path'])-1);
 		$this->remove_dkim_key($mail_config['dkim_path']."/".$_data['domain'], $_data['domain']);
-		$this->remove_from_amavis($_data['domain']);
+		if ($this->remove_from_amavis($_data['domain']))
+			$this->restart_amavis();
 	}
-
 
 	/**
 	 * Function called by onLoad
 	 * deletes dkim-keys
 	 */
 	function domain_dkim_delete($event_name, $data) {
-		if (isset($data['old']['dkim']) && $data['old']['dkim'] == 'y') $this->remove_dkim($data['old']);
+		if (isset($data['old']['dkim']) && $data['old']['dkim'] == 'y' && $data['old']['active'] == 'y')
+			$this->remove_dkim($data['old']);
 	}
-
 
 	/**
 	 * Function called by onLoad
 	 * insert dkim-keys
 	 */
 	function domain_dkim_insert($event_name, $data) {
-		if (isset($data['new']['dkim']) && $data['new']['dkim']=='y' && $this->check_system($data)) {
+		if (isset($data['new']['dkim']) && $data['new']['dkim']=='y' && $this->check_system($data))
 			$this->add_dkim($data);
-		}
 	}
-
 
 	/**
 	 * Function called by onLoad
@@ -284,31 +337,46 @@
 	 */
 	function domain_dkim_update($event_name, $data) {
 		global $app;
-		/* get the config */
-		if (isset($data['new']['dkim']) && $data['new']['dkim']=='y') { /* DKIM enabled */
-			if ($this->check_system($data)) {
+		if ($this->check_system($data)) {
+			/* maildomain disabled */
+			if ($data['new']['active'] == 'n' && $data['old']['active'] == 'y' && $data['new']['dkim']=='y') {
+				$app->log('Maildomain '.$data['new']['domain'].' disabled - remove DKIM-settings', LOGLEVEL_DEBUG);
+				$this->remove_dkim($data['new']);
+			}
+			/* maildomain re-enabled */
+			if ($data['new']['active'] == 'y' && $data['old']['active'] == 'n' && $data['new']['dkim']=='y') 
+				$this->add_dkim($data);
+
+			/* maildomain active - only dkim changes */
+			if ($data['new']['active'] == 'y' && $data['old']['active'] == 'y') {
+				/* dkim disabled */
+				if ($data['new']['dkim'] != $data['old']['dkim'] && $data['new']['dkim'] == 'n') {
+					$this->remove_dkim($data['new']);
+				}
+				/* dkim enabled */
+				elseif ($data['new']['dkim'] != $data['old']['dkim'] && $data['new']['dkim'] == 'y') {
+					$this->add_dkim($data);
+				}
+				/* new private-key */
+				if ($data['new']['dkim_private'] != $data['old']['dkim_private'] && $data['new']['dkim'] == 'y') {
+					$this->add_dkim($data);
+				}
+				/* new selector */
+				if ($data['new']['dkim_selector'] != $data['old']['dkim_selector'] && $data['new']['dkim'] == 'y') {
+					$this->add_dkim($data);
+				}
 				/* new domain-name */
-				if ($data['old']['domain'] != $data['new']['domain']) {
+				if ($data['new']['domain'] != $data['old']['domain']) {
 					$this->remove_dkim($data['old']);
 					$this->add_dkim($data);
 				}
-				/* new key */
-				if (($data['old']['dkim_private'] != $data['new']['dkim_private']) || ($data['old']['dkim'] != $data['new']['dkim'])) {
-					if ($data['new']['dkim_private'] != $data['old']['dkim_private']) $this->remove_dkim($data['new']);
-					$this->add_dkim($data);
-				}
-				/* change active (on / off) */
-				if ($data['old']['active'] != $data['new']['active']) {
-					if ($data['new']['active'] == 'y') {
-						$this->add_dkim($data);
-					} else {
-						$this->remove_dkim($data['new']);
-					}
-				}
+			}
+
+			/* resync */
+			if ($data['new']['active'] == 'y' && $data['new'] == $data['old']) {
+				$this->add_dkim($data);
 			}
 		}
-		if (isset($data['new']['dkim']) && $data['old']['dkim'] != $data['new']['dkim'])
-			if ($this->check_system($data) && $data['new']['dkim'] == 'n') $this->remove_dkim($data['new']);
 	}
 
 }

--
Gitblit v1.9.1