Merge pull request #1733 from peterschrammel/mem-limit

Set memory limit utility

Author: peterschrammel

src/util/Makefile

@@ -40,6 +40,7 @@ SRC = arith_tools.cpp \
       lispexpr.cpp \
       lispirep.cpp \
       memory_info.cpp \
+      memory_limit.cpp \
       merge_irep.cpp \
       message.cpp \
       mp_arith.cpp \

src/util/memory_limit.cpp

@@ -0,0 +1,48 @@
+/*******************************************************************\
+
+Module:
+
+Author: Peter Schrammel, peter.schrammel@diffblue.com
+
+\*******************************************************************/
+
+#include "memory_limit.h"
+
+#ifdef __linux__
+#include <sys/resource.h>
+#endif
+
+#include <ostream>
+
+/// Outputs the memory limits to the given output stream
+/// \param out: output stream
+void memory_limits(std::ostream &out)
+{
+#ifdef __linux__
+  struct rlimit mem_limit;
+  getrlimit(RLIMIT_AS, &mem_limit);
+
+  out << "  soft limit: " << mem_limit.rlim_cur << '\n';
+  out << "  hard limit: " << mem_limit.rlim_max;
+#else
+  out << "  not supported";
+#endif
+}
+
+/// Sets the soft memory limit of the current process
+/// \param soft_limit: the soft limit in bytes
+/// \return: true if setting the limit succeeded
+bool set_memory_limit(std::size_t soft_limit)
+{
+#ifdef __linux__
+  struct rlimit mem_limit;
+  getrlimit(RLIMIT_AS, &mem_limit);
+  mem_limit.rlim_cur = soft_limit;
+  setrlimit(RLIMIT_AS, &mem_limit);
+  getrlimit(RLIMIT_AS, &mem_limit);
+  return mem_limit.rlim_cur == soft_limit;
+#else
+  // not supported
+  return false;
+#endif
+}

src/util/memory_limit.h

@@ -0,0 +1,18 @@
+/*******************************************************************\
+
+Module:
+
+Author: Peter Schrammel, peter.schrammel@diffblue.com
+
+\*******************************************************************/
+
+#ifndef CPROVER_UTIL_MEMORY_LIMIT_H
+#define CPROVER_UTIL_MEMORY_LIMIT_H
+
+#include <cstddef> // size_t
+#include <iosfwd>
+
+void memory_limits(std::ostream &);
+bool set_memory_limit(std::size_t soft_limit);
+
+#endif // CPROVER_UTIL_MEMORY_LIMIT_H