check for memory usage

Log.h

@@ -81,9 +81,19 @@ static uint64_t __COUNT__ = 0; \
 __COUNT__++; \
 if (__COUNT__ % 1000 == 0) { \
 spdlog::info(string(__FUNCTION__) +  " processed " + to_string(__COUNT__) + " requests"); \
+struct sysinfo memInfo; \
+sysinfo (&memInfo); \
+long long totalPhysMem = memInfo.totalram; \
+/*Multiply in next statement to avoid int overflow on right hand side...*/ \
+totalPhysMem *= memInfo.mem_unit; \
+int usedByCurrentProcess = getValue(); \
+if ( 0.9 * totalPhysMem < usedByCurrentProcess ) { \
+  exit(-103); \
+} \
 }
 
 
+
 // if uknown error, the error is 10000 + line number
 
 

VERSION

-1.66.1
+1.70.0

common.h

@@ -31,6 +31,11 @@ using namespace std;
 #include <iostream>
 #include <map>
 #include <memory>
+
+#include <sys/types.h>
+#include <sys/sysinfo.h>
+#include <string.h>
+
 #include <vector>
 
 #include <boost/throw_exception.hpp>
@@ -69,6 +74,32 @@ inline void print_stack() {
     exit(-1);
 }
 
+inline int parseLine(char* line) {
+    // This assumes that a digit will be found and the line ends in " Kb".
+    int i = strlen(line);
+    const char* p = line;
+    while (*p <'0' || *p > '9') p++;
+    line[i-3] = '\0';
+    i = atoi(p);
+    return i;
+}
+
+inline int getValue() { //Note: this value is in KB!
+    FILE* file = fopen("/proc/self/status", "r");
+    int result = -1;
+    char line[128];
+
+    while (fgets(line, 128, file) != NULL){
+        if (strncmp(line, "VmRSS:", 6) == 0){
+            result = parseLine(line);
+            break;
+        }
+    }
+    fclose(file);
+    return result;
+}
+
+
 
 #define CHECK_STATE(_EXPRESSION_) \
     if (!(_EXPRESSION_)) { \
@@ -126,23 +157,5 @@ extern uint64_t initTime;
 #define WRITE_LOCK(__X__) std::unique_lock<std::shared_timed_mutex> __LOCK__(__X__);
 
 
-#include <boost/interprocess/sync/interprocess_semaphore.hpp>
-
-// max of 200 threads can call enclave at a time
-extern boost::interprocess::interprocess_semaphore enclaveSemaphore;
-
-class semaphore_guard {
-    boost::interprocess::interprocess_semaphore &sem;
-public:
-    semaphore_guard(boost::interprocess::interprocess_semaphore &_semaphore) : sem(_semaphore) {
-        sem.wait();
-    }
-
-    ~semaphore_guard() {
-        sem.post();
-    }
-};
-
-
 
 #endif //SGXWALLET_COMMON_H