memtest.cpp revision 1a3794a84074d7f22b8ddaba840aedd758a14cdd 
1/*
2 * Copyright (C) 2013 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#include <stdio.h>
18#include <stdlib.h>
19#include <string.h>
20#include <ctype.h>
21#include <sys/time.h>
22#include <time.h>
23#include <unistd.h>
24#include <sys/resource.h>
25#include <sys/syscall.h>
26#include <sys/types.h>
27#include <sys/mman.h>
28
29#include "memtest.h"
30
31nsecs_t system_time()
32{
33    struct timespec t;
34    t.tv_sec = t.tv_nsec = 0;
35    clock_gettime(CLOCK_MONOTONIC, &t);
36    return nsecs_t(t.tv_sec)*1000000000LL + t.tv_nsec;
37}
38
39static void usage(char* p) {
40    printf("Usage: %s <test> <options>\n"
41           "<test> is one of the following:\n"
42           "  copy_bandwidth [--size BYTES_TO_COPY]\n"
43           "  write_bandwidth [--size BYTES_TO_WRITE]\n"
44           "  read_bandwidth [--size BYTES_TO_COPY]\n"
45           "  per_core_bandwidth [--size BYTES]\n"
46           "    --type copy_ldrd_strd | copy_ldmia_stmia | copy_vld_vst |\n"
47           "           copy_vldmia_vstmia | memcpy | write_strd | write_stmia |\n"
48           "           write_vst | write_vstmia | memset | read_ldrd |\n"
49           "           read_ldmia | read_vld | read_vldmia\n"
50           "  multithread_bandwidth [--size BYTES]\n"
51           "    --type copy_ldrd_strd | copy_ldmia_stmia | copy_vld_vst |\n"
52           "           copy_vldmia_vstmia | memcpy | write_strd | write_stmia |\n"
53           "           write_vst | write_vstmia | memset | read_ldrd |\n"
54           "           read_ldmia | read_vld | read_vldmia\n"
55           "    --num_threads NUM_THREADS_TO_RUN\n"
56           "  malloc [fill]\n"
57           "  madvise\n"
58           "  resampler\n"
59           "  crash\n"
60           "  stack (stack smasher)\n"
61           "  crawl\n"
62           , p);
63}
64
65int copy_bandwidth(int argc, char** argv);
66int write_bandwidth(int argc, char** argv);
67int read_bandwidth(int argc, char** argv);
68int per_core_bandwidth(int argc, char** argv);
69int multithread_bandwidth(int argc, char** argv);
70int malloc_test(int argc, char** argv);
71int madvise_test(int argc, char** argv);
72int crash_test(int argc, char** argv);
73int stack_smasher_test(int argc, char** argv);
74int crawl_test(int argc, char** argv);
75int fp_test(int argc, char** argv);
76
77typedef struct {
78    const char *cmd_name;
79    int (*func)(int argc, char** argv);
80} function_t;
81
82function_t function_table[] = {
83    { "malloc", malloc_test },
84    { "madvise", madvise_test },
85    { "crash", crash_test },
86    { "stack", stack_smasher_test },
87    { "crawl", crawl_test },
88    { "fp", fp_test },
89    { "copy_bandwidth", copy_bandwidth },
90    { "write_bandwidth", write_bandwidth },
91    { "read_bandwidth", read_bandwidth },
92    { "per_core_bandwidth", per_core_bandwidth },
93    { "multithread_bandwidth", multithread_bandwidth },
94};
95
96int main(int argc, char** argv)
97{
98    if (argc == 1) {
99        usage(argv[0]);
100        return 0;
101    }
102    int err = -1;
103    for (unsigned int i = 0; i < sizeof(function_table)/sizeof(function_t); i++) {
104        if (strcmp(argv[1], function_table[i].cmd_name) == 0) {
105            err = (*function_table[i].func)(argc-1, argv+1);
106            break;
107        }
108    }
109    if (err) {
110        usage(argv[0]);
111    }
112    return err;
113}
114
115int malloc_test(int argc, char** argv)
116{
117    bool fill = (argc>=2 && !strcmp(argv[1], "fill"));
118    size_t total = 0;
119    size_t size = 0x40000000;
120    while (size) {
121        void* addr = malloc(size);
122        if (addr == 0) {
123            printf("size = %9zd failed\n", size);
124            size >>= 1;
125        } else {
126            total += size;
127            printf("size = %9zd, addr = %p (total = %9zd (%zd MB))\n",
128                    size, addr, total, total / (1024*1024));
129            if (fill) {
130                printf("filling...\n");
131                fflush(stdout);
132                memset(addr, 0, size);
133            }
134            size = size + (size>>1);
135        }
136    }
137    printf("done. allocated %zd MB\n", total / (1024*1024));
138    return 0;
139}
140
141int madvise_test(int argc, char** argv)
142{
143    for (int i=0 ; i<2 ; i++) {
144        size_t size = i==0 ? 4096 : 48*1024*1024; // 48 MB
145        printf("Allocating %zd MB... ", size/(1024*1024)); fflush(stdout);
146        void* addr1 = mmap(0, size, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, -1, 0);
147        printf("%p (%s)\n", addr1, addr1==(void*)-1 ? "failed" : "OK"); fflush(stdout);
148
149        printf("touching %p...\n", addr1); fflush(stdout);
150        memset(addr1, 0x55, size);
151
152        printf("advising DONTNEED...\n"); fflush(stdout);
153        madvise(addr1, size, MADV_DONTNEED);
154
155        printf("reading back %p...\n", addr1); fflush(stdout);
156        if (*(long*)addr1 == 0) {
157            printf("madvise freed some pages\n");
158        } else if (*(long*)addr1 == 0x55555555) {
159            printf("pages are still there\n");
160        } else {
161            printf("getting garbage back\n");
162        }
163
164        printf("Allocating %zd MB... ", size/(1024*1024)); fflush(stdout);
165        void* addr2 = mmap(0, size, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, -1, 0);
166        printf("%p (%s)\n", addr2, addr2==(void*)-1 ? "failed" : "OK"); fflush(stdout);
167
168        printf("touching %p...\n", addr2); fflush(stdout);
169        memset(addr2, 0xAA, size);
170
171        printf("unmap %p ...\n", addr2); fflush(stdout);
172        munmap(addr2, size);
173
174        printf("touching %p...\n", addr1); fflush(stdout);
175        memset(addr1, 0x55, size);
176
177        printf("unmap %p ...\n", addr1); fflush(stdout);
178        munmap(addr1, size);
179    }
180
181    printf("Done\n"); fflush(stdout);
182    return 0;
183}
184
185int crash_test(int argc, char** argv)
186{
187    printf("about to crash...\n");
188    asm volatile(
189        "mov r0,  #0 \n"
190        "mov r1,  #1 \n"
191        "mov r2,  #2 \n"
192        "mov r3,  #3 \n"
193        "ldr r12, [r0] \n"
194    );
195
196    return 0;
197}
198
199int stack_smasher_test(int argc, char** argv)
200{
201    int dummy = 0;
202    printf("corrupting our stack...\n");
203    *(volatile long long*)&dummy = 0;
204    return 0;
205}
206
207// --------------------------------------------------------------------
208
209extern "C" void thumb_function_1(int*p);
210extern "C" void thumb_function_2(int*p);
211extern "C" void arm_function_3(int*p);
212extern "C" void arm_function_2(int*p);
213extern "C" void arm_function_1(int*p);
214
215void arm_function_3(int*p) {
216    int a = 0;
217    thumb_function_2(&a);
218}
219
220void arm_function_2(int*p) {
221    int a = 0;
222    thumb_function_1(&a);
223}
224
225void arm_function_1(int*p) {
226    int a = 0;
227    arm_function_2(&a);
228}
229
230int crawl_test(int argc, char** argv)
231{
232    int a = 0;
233    arm_function_1(&a);
234    return 0;
235}
236
237