1 /*
2 * Copyright (C) 2014 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 */
17 #include "benchmark.h"
19 #include <stdio.h>
21 #define KB 1024
22 #define MB 1024*KB
24 #define AT_COMMON_SIZES \
25 Arg(1)->Arg(2)->Arg(3)->Arg(4)->Arg(8)->Arg(16)->Arg(32)->Arg(64)->Arg(512)-> \
26 Arg(1*KB)->Arg(4*KB)->Arg(8*KB)->Arg(16*KB)->Arg(64*KB)
28 template <typename Fn>
29 static void ReadWriteTest(int iters, int chunk_size, Fn f, bool buffered) {
30 StopBenchmarkTiming();
31 FILE* fp = fopen("/dev/zero", "rw");
32 char* buf = new char[chunk_size];
33 StartBenchmarkTiming();
35 if (!buffered) {
36 setvbuf(fp, 0, _IONBF, 0);
37 }
39 for (int i = 0; i < iters; ++i) {
40 f(buf, chunk_size, 1, fp);
41 }
43 StopBenchmarkTiming();
44 SetBenchmarkBytesProcessed(int64_t(iters) * int64_t(chunk_size));
45 delete[] buf;
46 fclose(fp);
47 }
49 static void BM_stdio_fread(int iters, int chunk_size) {
50 ReadWriteTest(iters, chunk_size, fread, true);
51 }
52 BENCHMARK(BM_stdio_fread)->AT_COMMON_SIZES;
54 static void BM_stdio_fwrite(int iters, int chunk_size) {
55 ReadWriteTest(iters, chunk_size, fwrite, true);
56 }
57 BENCHMARK(BM_stdio_fwrite)->AT_COMMON_SIZES;
59 static void BM_stdio_fread_unbuffered(int iters, int chunk_size) {
60 ReadWriteTest(iters, chunk_size, fread, false);
61 }
62 BENCHMARK(BM_stdio_fread_unbuffered)->AT_COMMON_SIZES;
64 static void BM_stdio_fwrite_unbuffered(int iters, int chunk_size) {
65 ReadWriteTest(iters, chunk_size, fwrite, false);
66 }
67 BENCHMARK(BM_stdio_fwrite_unbuffered)->AT_COMMON_SIZES;