Merge pull request #79 from lainchan/clean_prod_vichan_update

Merge vichan changes from upstream and changes directly applied to the site.

1 files changed, 39 insertions, 0 deletions

diff --git a/inc/lock.php b/inc/lock.php
new file mode 100644
index 00000000..4fb2f5df
--- /dev/null
+++ b/inc/lock.php
@@ -0,0 +1,39 @@
+<?php
+class Lock {
+  function __construct($key) { global $config;
+    if ($config['lock']['enabled'] == 'fs') {
+      $key = str_replace('/', '::', $key);
+      $key = str_replace("\0", '', $key);
+
+      $this->f = fopen("tmp/locks/$key", "w");
+    }
+  }
+
+  // Get a shared lock
+  function get($nonblock = false) { global $config;
+    if ($config['lock']['enabled'] == 'fs') {
+      $wouldblock = false;
+      flock($this->f, LOCK_SH | ($nonblock ? LOCK_NB : 0), $wouldblock);
+      if ($nonblock && $wouldblock) return false;
+    }
+    return $this;
+  }
+
+  // Get an exclusive lock
+  function get_ex($nonblock = false) { global $config;
+    if ($config['lock']['enabled'] == 'fs') {
+      $wouldblock = false;
+      flock($this->f, LOCK_EX | ($nonblock ? LOCK_NB : 0), $wouldblock);
+      if ($nonblock && $wouldblock) return false;
+    }
+    return $this;
+  }
+
+  // Free a lock
+  function free() { global $config;
+    if ($config['lock']['enabled'] == 'fs') {
+      flock($this->f, LOCK_UN);
+    }
+    return $this;
+  }
+}