summaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/common.php4
-rw-r--r--lib/daemon.php117
-rw-r--r--lib/queuehandler.php19
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) {