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

3 files changed, 127 insertions, 0 deletions

diff --git a/libs/pbd/pbd/resource.h b/libs/pbd/pbd/resource.h
new file mode 100644
index 0000000000..ec826e4930
--- /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
index 0000000000..b2988b6314
--- /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 77403bddf4..2057c728da 100644
--- 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',