summaryrefslogtreecommitdiff
path: root/stress.c
blob: 0b7e999285f5619f8918c2c526f51956d4b4b4b3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
#include "stress.h"

void usage(void)
{
    printf("Usage: stress [-t n] [-m n]\n");
    printf("\t-t n\tRun n processes parallel\n");
    printf("\t-m n\tTest n MiB of Memory\n");
    printf("\n");
    printf("\t\tCan be combined\n");
    exit(0);
}

void testMemory(unsigned long long size)
{
    char *mem=malloc(size*sizeof(char));
    unsigned long long i;
    unsigned char run=0;

    while(1==1)
    {
        for(i=0; i<size; i++)
        {
            mem[i]=(mem[i]+i+run)%256;
        }
        run++;
    }
}

void testCPU(void)
{
    long double f=0;
    long double res=0;
    while(1==1)
    {
        for(f=0; f<LDBL_MAX; f+=LDBL_EPSILON)
        {
            res=sqrtl(f);
        }
    }
}

int main(int argc, char **argv)
{
    const char *optargs = "t:m:h";
    char optc;
    unsigned long long convtemp;

    unsigned char errflags=0;
    unsigned char optflags=0;

    unsigned char threads = 1;
    unsigned long long memory = 1048576LL;

    int i, pid, wstatus;

    while((optc=getopt(argc,argv,optargs))!=-1)
    {
        switch(optc)
        {
        case 't':
            if((convtemp = strtoll(optarg, NULL, 10))==0)
            {
                errflags|=THREADS;
                perror("Error with -t: ");
            }
            else
            {
                optflags|=THREADS;
                threads=convtemp;
            }
            break;
        case 'm':
            if((convtemp = strtoll(optarg, NULL, 10))==0)
            {
                errflags|=MEMSIZE;
                perror("Error with -m: ");
            }
            else
            {
                optflags|=MEMSIZE;
                memory=convtemp*1024*1024;
            }
            break;
        case 'h':
        case '?':
        default:
            usage();
        }
    }

    if(errflags>0)
    {
        usage();
    }
    if(optflags & THREADS)
    {
        printf("Threads: %d\n", threads);
    }
    if(optflags & MEMSIZE)
    {
        printf("Memsize: %llu\n", memory);
    }

    for(i=0; i<threads; i++)
    {
        switch(pid=fork())
        {
        case -1:
            perror("fork()");
            break;
        case 0:
            if(optflags&MEMSIZE)
              testMemory(memory/threads);
            else
              testCPU();
            break;
        }
    }
    for(i=0; i<threads; i++)
    {
        wait(&wstatus);
    }

    return 0;
}