summaryrefslogtreecommitdiff
path: root/alloc-stress/mem-pressure.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'alloc-stress/mem-pressure.cpp')
-rw-r--r--alloc-stress/mem-pressure.cpp38
1 files changed, 16 insertions, 22 deletions
diff --git a/alloc-stress/mem-pressure.cpp b/alloc-stress/mem-pressure.cpp
index 82b2273c..b0920e69 100644
--- a/alloc-stress/mem-pressure.cpp
+++ b/alloc-stress/mem-pressure.cpp
@@ -8,8 +8,8 @@
#include <sys/wait.h>
#include <unistd.h>
-void *alloc_set(size_t size) {
- void *addr = NULL;
+void* alloc_set(size_t size) {
+ void* addr = NULL;
addr = malloc(size);
if (!addr) {
@@ -20,15 +20,14 @@ void *alloc_set(size_t size) {
return addr;
}
-void add_pressure(size_t *shared, size_t size, size_t step_size,
- size_t duration, const char *oom_score) {
+void add_pressure(size_t* shared, size_t size, size_t step_size, size_t duration,
+ const char* oom_score) {
int fd, ret;
fd = open("/proc/self/oom_score_adj", O_WRONLY);
ret = write(fd, oom_score, strlen(oom_score));
if (ret < 0) {
- printf("Writing oom_score_adj failed with err %s\n",
- strerror(errno));
+ printf("Writing oom_score_adj failed with err %s\n", strerror(errno));
}
close(fd);
@@ -43,31 +42,27 @@ void add_pressure(size_t *shared, size_t size, size_t step_size,
}
}
-void usage()
-{
+void usage() {
printf("Usage: [OPTIONS]\n\n"
" -d N: Duration in microsecond to sleep between each allocation.\n"
" -i N: Number of iterations to run the alloc process.\n"
" -o N: The oom_score to set the child process to before alloc.\n"
- " -s N: Number of bytes to allocate in an alloc process loop.\n"
- );
+ " -s N: Number of bytes to allocate in an alloc process loop.\n");
}
-int main(int argc, char *argv[])
-{
+int main(int argc, char* argv[]) {
pid_t pid;
- size_t *shared;
+ size_t* shared;
int c, i = 0;
size_t duration = 1000;
int iterations = 0;
- const char *oom_score = "899";
- size_t step_size = 2 * 1024 * 1024; // 2 MB
+ const char* oom_score = "899";
+ size_t step_size = 2 * 1024 * 1024; // 2 MB
size_t size = step_size;
while ((c = getopt(argc, argv, "hi:d:o:s:")) != -1) {
- switch (c)
- {
+ switch (c) {
case 'i':
iterations = atoi(optarg);
break;
@@ -85,11 +80,11 @@ int main(int argc, char *argv[])
abort();
default:
abort();
- }
+ }
}
- shared = (size_t *)mmap(NULL, sizeof(size_t), PROT_READ | PROT_WRITE,
- MAP_ANONYMOUS | MAP_SHARED, 0, 0);
+ shared = (size_t*)mmap(NULL, sizeof(size_t), PROT_READ | PROT_WRITE, MAP_ANONYMOUS | MAP_SHARED,
+ 0, 0);
while (iterations == 0 || i < iterations) {
*shared = 0;
@@ -101,8 +96,7 @@ int main(int argc, char *argv[])
exit(0);
} else {
wait(NULL);
- printf("Child %d allocated %zd MB\n", i,
- *shared / 1024 / 1024);
+ printf("Child %d allocated %zd MB\n", i, *shared / 1024 / 1024);
size = *shared / 2;
}
i++;