Add pbd/resource.h for portable access to resource limits

diff --git a/libs/pbd/pbd/resource.h b/libs/pbd/pbd/resource.h
new file mode 100644 (file)
index 0000000..ec826e4
--- /dev/null
+++ b/libs/pbd/pbd/resource.h
@@ -0,0 +1,49 @@
+/*
+    Copyright (C) 2011 Tim Mayberry
+
+    This program is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 2 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 General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program; if not, write to the Free Software
+    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+
+*/
+
+#include <stdint.h>
+
+namespace PBD {
+
+typedef uint64_t rlimit_t;
+
+enum ResourceType {
+       OpenFiles
+};
+
+struct ResourceLimit
+{
+       rlimit_t current_limit;
+       rlimit_t max_limit;
+};
+
+/**
+ * @return true on success, false on error
+ */
+bool
+get_resource_limit (ResourceType resource, ResourceLimit& limit);
+
+/**
+ * @return true on success, false on error
+ */
+bool
+set_resource_limit (ResourceType resource, const ResourceLimit& limit);
+
+
+} // namespace PBD

diff --git a/libs/pbd/resource.cc b/libs/pbd/resource.cc
new file mode 100644 (file)
index 0000000..b2988b6
--- /dev/null
+++ b/libs/pbd/resource.cc
@@ -0,0 +1,77 @@
+/*
+    Copyright (C) 2011 Tim Mayberry
+
+    This program is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 2 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 General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program; if not, write to the Free Software
+    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+
+*/
+
+#ifdef WIN32
+#include <stdio.h>
+#else
+#include <sys/time.h>
+#include <sys/resource.h>
+#endif
+
+#include "pbd/resource.h"
+
+namespace PBD {
+
+bool
+get_resource_limit (ResourceType resource, ResourceLimit& limit)
+{
+       if (resource == OpenFiles)
+       {
+#ifdef WIN32
+               limit.current_limit = _getmaxstdio();
+               limit.max_limit = 2048;
+               return true;
+#else
+               struct rlimit rl;
+               if (getrlimit (RLIMIT_NOFILE, &rl) == 0) {
+                       limit.current_limit = rl.rlim_cur;
+                       limit.max_limit = rl.rlim_max;
+                       return true;
+               }
+#endif
+       }
+
+       return false;
+}
+
+bool
+set_resource_limit (ResourceType resource, const ResourceLimit& limit)
+{
+       if (resource == OpenFiles)
+       {
+#ifdef WIN32
+               // no soft and hard limits on windows
+               rlimit_t new_max = _setmaxstdio(limit.current_limit);
+
+               if (new_max == limit.current_limit) return true;
+#else
+               struct rlimit rl;
+               rl.rlim_cur = limit.current_limit;
+               rl.rlim_max = limit.max_limit;
+               if (setrlimit (RLIMIT_NOFILE, &rl) == 0) {
+                       return true;
+               }
+
+#endif
+       }
+
+       return false;
+}
+
+} // namespace PBD

diff --git a/libs/pbd/wscript b/libs/pbd/wscript
index 77403bddf46d5845de22ca8f4d492a8e92c5ae8c..2057c728dac414aa6aaa12e9daeb3eb79f3d998d 100644 (file)
--- a/libs/pbd/wscript
+++ b/libs/pbd/wscript
@@ -60,6 +60,7 @@ libpbd_sources = [
     'property_list.cc',
     'pthread_utils.cc',
     'receiver.cc',
+    'resource.cc',
     'search_path.cc',
     'semutils.cc',
     'shortpath.cc',