diff options
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/common.php | 4 | ||||
| -rw-r--r-- | lib/daemon.php | 117 | ||||
| -rw-r--r-- | lib/queuehandler.php | 19 |
3 files changed, 138 insertions, 2 deletions
diff --git a/lib/common.php b/lib/common.php index e4c8e9032..80aab806f 100644 --- a/lib/common.php +++ b/lib/common.php @@ -94,6 +94,10 @@ $config = 'public' => array()), # JIDs of users who want to receive the public stream 'tag' => array('dropoff' => 864000.0), + 'daemon' => + array('piddir' => '/var/run', + 'user' => false, + 'group' => false) ); $config['db'] = &PEAR::getStaticProperty('DB_DataObject','options'); diff --git a/lib/daemon.php b/lib/daemon.php new file mode 100644 index 000000000..09144afe7 --- /dev/null +++ b/lib/daemon.php @@ -0,0 +1,117 @@ +<?php +/* + * Laconica - a distributed open-source microblogging tool + * Copyright (C) 2008, Controlez-Vous, Inc. + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU Affero General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see <http://www.gnu.org/licenses/>. + */ + +if (!defined('LACONICA')) { exit(1); } + +class Daemon { + + function name() { + return NULL; + } + + function background() { + $pid = pcntl_fork(); + if ($pid < 0) { # error + return false; + } else if ($pid > 0) { # parent + common_log(LOG_INFO, "Successfully forked."); + exit(0); + } else { # child + return true; + } + } + + function alreadyRunning() { + + $pidfilename = $this->pidFilename(); + + if (!$pidfilename) { + return false; + } + + if (!file_exists($pidfilename)) { + return false; + } + $contents = file_get_contents($pidfilename); + if (posix_kill(trim($contents),0)) { + return true; + } else { + return false; + } + } + + function writePidFile() { + $pidfilename = $this->pidFilename(); + + if (!$pidfilename) { + return false; + } + + file_put_contents($pidfilename, posix_getpid()); + } + + function clearPidFile() { + $pidfilename = $this->pidFilename(); + unlink($pidfilename); + } + + function pidFilename() { + $piddir = common_config('daemon', 'piddir'); + if (!$piddir) { + return NULL; + } + $name = $this->name(); + if (!$name) { + return NULL; + } + return $piddir . '/' . $name; + } + + function changeUser() { + + if (common_config('daemon', 'user')) { + $user_info = posix_getpwnam(common_config('daemon', 'user')); + common_log(LOG_INFO, "Setting user to " . common_config('daemon', 'user')); + posix_setuid($user_info['uid']); + } + + if (common_config('daemon', 'group')) { + $group_info = posix_getgrnam(common_config('daemon', 'group')); + common_log(LOG_INFO, "Setting group to " . common_config('daemon', 'group')); + posix_setgid($group_info['gid']); + } + } + + function runOnce() { + if ($this->alreadyRunning()) { + common_log(LOG_INFO, $this->name() . ' already running. Exiting.'); + exit(0); + } + if ($this->background()) { + $this->writePidFile(); + $this->changeUser(); + $this->run(); + $this->clearPidFile(); + } + } + + function run() { + return true; + } +} diff --git a/lib/queuehandler.php b/lib/queuehandler.php index 3115ea38d..d673f7f94 100644 --- a/lib/queuehandler.php +++ b/lib/queuehandler.php @@ -19,7 +19,11 @@ define('CLAIM_TIMEOUT', 1200); -class QueueHandler { +if (!defined('LACONICA')) { exit(1); } + +require_once(INSTALLDIR.'/lib/daemon.php'); + +class QueueHandler extends Daemon { var $_id = 'generic'; @@ -32,6 +36,10 @@ class QueueHandler { function class_name() { return ucfirst($this->transport()) . 'Handler'; } + + function name() { + return strtolower($this->class_name().'.'.$this->get_id()); + } function get_id() { return $this->_id; @@ -55,7 +63,10 @@ class QueueHandler { return true; } - function handle_queue() { + function run() { + if (!$this->start()) { + return false; + } $this->log(LOG_INFO, 'checking for queued notices'); $transport = $this->transport(); do { @@ -87,6 +98,10 @@ class QueueHandler { $this->idle(5); } } while (true); + if (!$this->finish()) { + return false; + } + return true; } function idle($timeout=0) { |