Only allow one instance (per user) to run

Wiebe Cazemier
1 parent d68708d3
Showing 5 changed files with 91 additions and 0 deletions
CMakeLists.txt
mainapp.cpp
mainapp.h
oneinstancelock.cpp
oneinstancelock.h
@@ -35,6 +35,7 @@ add_executable(FlashMQ
     unscopedlock.cpp
     scopedsocket.cpp
     bindaddr.cpp
+    oneinstancelock.cpp
     )
 target_link_libraries(FlashMQ pthread dl ssl crypto)
@@ -378,6 +378,13 @@ MainApp *MainApp::getMainApp()
 void MainApp::start()
 {
+#ifndef NDEBUG
+    if (fuzzFilePath.empty())
+    {
+        oneInstanceLock.lock();
+    }
+#endif
+
     timer.start();
     std::map<int, std::shared_ptr<Listener>> listenerMap; // For finding listeners by fd.
@@ -516,6 +523,8 @@ void MainApp::start()
         }
     }
+    oneInstanceLock.unlock();
+
     for(std::shared_ptr<ThreadData> &thread : threads)
     {
         thread->queueQuit();
@@ -23,6 +23,7 @@
 #include "configfileparser.h"
 #include "timer.h"
 #include "scopedsocket.h"
+#include "oneinstancelock.h"
 class MainApp
 {
@@ -44,6 +45,7 @@ class MainApp
     std::mutex quitMutex;
     std::string fuzzFilePath;
     bool fuzzWebsockets = false;
+    OneInstanceLock oneInstanceLock;
     Logger *logger = Logger::getInstance();
+#include "oneinstancelock.h"
+
+#include <stdexcept>
+
+#include "utils.h"
+
+OneInstanceLock::OneInstanceLock()
+{
+    std::string dir("/tmp");
+
+    char *d = getenv("HOME");
+    if (d != NULL && d[0] == '/')
+    {
+        dir = std::string(d);
+    }
+    lockFilePath = dir + "/.FlashMQ.lock";
+}
+
+OneInstanceLock::~OneInstanceLock()
+{
+    unlock();
+}
+
+void OneInstanceLock::lock()
+{
+    fd = open(lockFilePath.c_str(), O_RDWR | O_CREAT, 0600);
+    if (fd < 0)
+         throw std::runtime_error(formatString("Can't create '%s': %s", lockFilePath.c_str(), strerror(errno)));
+
+    struct flock fl;
+    fl.l_start = 0;
+    fl.l_len = 0;
+    fl.l_type = F_WRLCK;
+    fl.l_whence = SEEK_SET;
+    if (fcntl(fd, F_SETLK, &fl) < 0)
+    {
+        throw std::runtime_error("Can't acquire lock: another instance is already running?");
+    }
+}
+
+void OneInstanceLock::unlock()
+{
+    if (fd > 0)
+    {
+        close(fd);
+        fd = 0;
+        if (!lockFilePath.empty())
+        {
+            if (unlink(lockFilePath.c_str()) < 0)
+            {
+                logger->logf(LOG_ERR, "Can't delete '%': %s", lockFilePath.c_str(), strerror(errno));
+            }
+            lockFilePath.clear();
+        }
+    }
+}
+#ifndef ONEINSTANCELOCK_H
+#define ONEINSTANCELOCK_H
+
+#include <fcntl.h>
+#include <string.h>
+#include <unistd.h>
+#include <string>
+#include "logger.h"
+
+class OneInstanceLock
+{
+    int fd = -1;
+    std::string lockFilePath;
+    Logger *logger = Logger::getInstance();
+
+public:
+    OneInstanceLock();
+    ~OneInstanceLock();
+    void lock();
+    void unlock();
+};
+
+#endif // ONEINSTANCELOCK_H