Add option to unlimit memory

Author: Ex-Origin

README.md

@@ -32,6 +32,7 @@ apt install -y gdbserver strace
       -s       halt at entry point
       -v       show debug information
       -n       disable address space randomization
+      -u       do not limit memory
     ```
 
 4. Use `gdbpwn.py` to connect to the target IP.

src/arg.c

@@ -12,6 +12,7 @@ int arg_opt_m = 0;
 int arg_opt_s = 0;
 int arg_opt_v = 0;
 int arg_opt_n = 0;
+int arg_opt_u = 0;
 
 char **arg_execve_argv = NULL;
 char *arg_popen = NULL;
@@ -62,14 +63,15 @@ int help()
                     "  -s       halt at entry point\n"
                     "  -v       show debug information\n"
                     "  -n       disable address space randomization\n"
+                    "  -u       do not limit memory\n"
     );
     exit(EXIT_FAILURE);
 }
 
 int parsing_argv(int argc, char *argv[])
 {
     int opt;
-    while ((opt = getopt(argc, argv, "e:p:o:hmsvn")) != -1) {
+    while ((opt = getopt(argc, argv, "e:p:o:hmsvnu")) != -1) {
         switch (opt) {
         case 'e':
             arg_opt_e = 1;
@@ -97,6 +99,9 @@ int parsing_argv(int argc, char *argv[])
         case 'n':
             arg_opt_n = 1;
             break;
+        case 'u':
+            arg_opt_u = 1;
+            break;
         default: /* '?' */
             help();
             break;

src/debug-server.h

@@ -5,7 +5,7 @@
 #define COMMAND_PORT    9545
 #define GDBSERVER_PORT  9549
 
-#define VERSION         "1.3.1"
+#define VERSION         "1.3.2"
 
 #define COMMAND_GDB_REGISTER        0x01
 #define COMMAND_GDBSERVER_ATTACH    0x02
@@ -54,6 +54,7 @@ extern int arg_opt_m;
 extern int arg_opt_s;
 extern int arg_opt_v;
 extern int arg_opt_n;
+extern int arg_opt_u;
 extern char **arg_execve_argv;
 extern char *arg_popen;
 extern int arg_pid;

src/service.c

@@ -131,9 +131,12 @@ int start_service(int client_sock)
     }
     else if(service_pid == 0)
     {
-        limit.rlim_cur = 0x100000000;
-        limit.rlim_max = 0x100000000;
-        CHECK(setrlimit(RLIMIT_AS, &limit) != -1);
+        if(arg_opt_u == 0)
+        {
+            limit.rlim_cur = 0x100000000;
+            limit.rlim_max = 0x100000000;
+            CHECK(setrlimit(RLIMIT_AS, &limit) != -1);
+        }
 
         CHECK(sigprocmask(SIG_SETMASK, &old_mask, NULL) != -1);