Gobble memory

memory-eater.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define MB 1048576
#define CONSUME 10 * MB

// Usage: ./memory-eater {total MB} {sleep seconds}
// eg ./memory-eater 700 30 to consume 700MB for 30 secs

int main(int argc, char *argv[]) {
    void *b = NULL;
    int bytes = 0;
    int max_mem = 0;
    int secs = 5;

    if( argc == 2 ) {
        max_mem = atoi(argv[1]) * MB;
    } else if(argc == 3) {
        max_mem = atoi(argv[1]) * MB;
        secs = atoi(argv[2]);
    } else {
        printf("Usage: ./memory-eater {total MB} {sleep seconds} \n");
        exit(0);
    }

    printf("Gobbling up %d bytes from RAM for %d seconds\n", max_mem, secs);
    while(1) {
        if (bytes >= max_mem) {
            printf("Reached max memory: %d bytes\n", bytes);
            break;
        }
        /* allocate xMB of memory */
        b = (void *) malloc(CONSUME);
        if (!b) {
            break;
        }

        /* consume our allocated memory */
        memset(b, 1, CONSUME);
        bytes = bytes + CONSUME;
    }
    sleep(secs);
    exit(0);
}