C語言——記憶體消耗測試程式
阿新 • • 發佈：2019-01-27
/*這是一個消耗記憶體的測試c程式*/
/* 用法：
* ./test-mem arg1 arg2 arg3
* arg1表示每秒分配的記憶體數量（單位M）
* arg2表示總共分配多少記憶體
* arg3表示程式存活多久（預設300秒）
*/
# include <stdio.h>
# include <stdlib.h># include <string.h>
# include <unistd.h>
# define KB (1024)
# define MB (1024 * KB)
# define GB (1024 * MB)
int main(int argc, char *argv[])
{
char *p;
int size = 0, i = 0, j = 0;
size = atoi(argv[1]);
int end_time=300;
int total_size = -1;
if (argc > 2) {
total_size = atoi(argv[2]);
if (size > total_size)
total_size = size;
}
if (argc > 3) {
end_time = atoi(argv[3]);
}
#if 0
while ((p = (char *)malloc(GB)))
memset(p, 0, GB);
#endif
while (1) {
i = size;
if(total_size != j) {
while (i--) {
p = (char *)malloc(MB);
memset(p, 0, MB);
j++;
printf("malloc: %dMB\r", j);
fflush(stdout);
}
}
sleep(1);
end_time--;
if(end_time == 0) {
break;
}
}
#if 0while ((p = (char *)malloc(KB)))
memset(p, 0, KB);
#endif
//sleep(1);
/// goto again;
return 0;
}