From aae945c369ac9c98ab3c2c4036ff1a3453a37467 Mon Sep 17 00:00:00 2001
From: Steven Wittens <steven@10.no-reply.drupal.org>
Date: Tue, 3 Oct 2006 00:24:19 +0000
Subject: #70995: Prevent cron re-runs

---
 includes/common.inc | 58 +++++++++++++++++++++++++++++++++++++++++------------
 1 file changed, 45 insertions(+), 13 deletions(-)

(limited to 'includes/common.inc')

diff --git a/includes/common.inc b/includes/common.inc
index f2a550ad1..ef296bf9b 100644
--- a/includes/common.inc
+++ b/includes/common.inc
@@ -1698,24 +1698,56 @@ function drupal_cron_run() {
     set_time_limit(240);
   }
 
-  // Check if the last cron run completed
-  if (variable_get('cron_busy', FALSE)) {
-    watchdog('cron', t('Last cron run did not complete.'), WATCHDOG_WARNING);
+  // Fetch the cron semaphore
+  $semaphore = variable_get('cron_semaphore', FALSE);
+
+  if ($semaphore) {
+    if (time() - $semaphore > 3600) {
+      // Either cron has been running for more than an hour or the semaphore
+      // was not reset due to a database error.
+      watchdog('cron', t('Cron has been running for more than an hour and is most likely stuck.'), WATCHDOG_ERROR);
+
+      // Release cron semaphore
+      variable_del('cron_semaphore');
+    }
+    else {
+      // Cron is still running normally.
+      watchdog('cron', t('Attempting to re-run cron while it is already running.'), WATCHDOG_WARNING);
+    }
   }
   else {
-    variable_set('cron_busy', TRUE);
-  }
+    // Register shutdown callback
+    register_shutdown_function('drupal_cron_cleanup');
+
+    // Lock cron semaphore
+    variable_set('cron_semaphore', time());
+
+    // Iterate through the modules calling their cron handlers (if any):
+    module_invoke_all('cron');
 
-  // Iterate through the modules calling their cron handlers (if any):
-  module_invoke_all('cron');
+    // Record cron time
+    variable_set('cron_last', time());
+    watchdog('cron', t('Cron run completed.'), WATCHDOG_NOTICE);
 
-  // Clean up
-  variable_set('cron_busy', FALSE);
-  variable_set('cron_last', time());
-  watchdog('cron', t('Cron run completed.'), WATCHDOG_NOTICE);
+    // Release cron semaphore
+    variable_del('cron_semaphore');
 
-  // Return TRUE so other functions can check if it did run successfully
-  return TRUE;
+    // Return TRUE so other functions can check if it did run successfully
+    return TRUE;
+  }
+}
+
+/**
+ * Shutdown function for cron cleanup.
+ */
+function drupal_cron_cleanup() {
+  // See if the semaphore is still locked.
+  if (variable_get('cron_semaphore', FALSE)) {
+    watchdog('cron', t('Cron run exceeded the time limit and was aborted.'), WATCHDOG_WARNING);
+
+    // Release cron semaphore
+    variable_del('cron_semaphore');
+  }
 }
 
 /**
-- 
cgit v1.2.3