Make setting rlimits a little better

Wiebe Cazemier
1 parent d06dcb0b
Showing 3 changed files with 18 additions and 8 deletions
main.cpp
mainapp.cpp
mainapp.h
@@ -2,7 +2,6 @@
 #include <signal.h>
 #include <memory>
 #include <string.h>
-#include <sys/resource.h>
 #include <openssl/ssl.h>
  
 #include "mainapp.h"
@@ -32,12 +31,6 @@ static void signal_handler(int signal)
  
 int register_signal_handers()
 {
-    // Quick'n dirty temp. TODO properly, with config file, checks, etc
-    rlim_t rlim = 1000000;
-    printf("Setting ulimit nofile to %ld.\n", rlim);
-    struct rlimit v = { rlim, rlim };
-    setrlimit(RLIMIT_NOFILE, &v);
-
     struct sigaction sa;
     memset(&sa, 0, sizeof (struct sigaction));
     sa.sa_handler = &signal_handler;
@@ -539,7 +539,20 @@ void MainApp::quit()
     running = false;
 }
  
-// Loaded on app start where you want it to crash, loaded from within try/catch on reload, to allow the program to continue.
+
+void MainApp::setlimits(rlim_t nofile)
+{
+    logger->logf(LOG_INFO, "Setting ulimit nofile to %ld. TODO: configurable in config file.", nofile);
+    struct rlimit v = { nofile, nofile };
+    if (setrlimit(RLIMIT_NOFILE, &v) < 0)
+    {
+        logger->logf(LOG_ERR, "Setting ulimit nofile failed: %s. This means the default is used.", strerror(errno));
+    }
+}
+
+/**
+ * @brief MainApp::loadConfig is loaded on app start where you want it to crash, loaded from within try/catch on reload, to allow the program to continue.
+ */
 void MainApp::loadConfig()
 {
     Logger *logger = Logger::getInstance();
@@ -564,6 +577,8 @@ void MainApp::loadConfig()
     logger->setLogPath(settings->logPath);
     logger->reOpen();
  
+    setlimits(1000000);
+
     for (std::shared_ptr<Listener> &l : this->listeners)
     {
         l->loadCertAndKeyFromConfig();
@@ -11,6 +11,7 @@
 #include <functional>
 #include <forward_list>
 #include <list>
+#include <sys/resource.h>
  
 #include "forward_declarations.h"
  
@@ -46,6 +47,7 @@ class MainApp
  
     Logger *logger = Logger::getInstance();
  
+    void setlimits(rlim_t nofile);
     void loadConfig();
     void reloadConfig();
     static void doHelp(const char *arg);