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 */
16
17 #include <stdio.h>
18 #include <stdio_ext.h>
19 #include <stdlib.h>
20
21 #include <benchmark/benchmark.h>
22
23 #define KB 1024
24 #define MB 1024*KB
25
26 #define AT_COMMON_SIZES \
27 Arg(1)->Arg(2)->Arg(3)->Arg(4)->Arg(8)->Arg(16)->Arg(32)->Arg(64)->Arg(512)-> \
28 Arg(1*KB)->Arg(4*KB)->Arg(8*KB)->Arg(16*KB)->Arg(64*KB)
29
30 template <typename Fn>
ReadWriteTest(benchmark::State & state,Fn f,bool buffered)31 void ReadWriteTest(benchmark::State& state, Fn f, bool buffered) {
32 size_t chunk_size = state.range_x();
33
34 FILE* fp = fopen("/dev/zero", "rw");
35 __fsetlocking(fp, FSETLOCKING_BYCALLER);
36 char* buf = new char[chunk_size];
37
38 if (!buffered) {
39 setvbuf(fp, 0, _IONBF, 0);
40 }
41
42 while (state.KeepRunning()) {
43 f(buf, chunk_size, 1, fp);
44 }
45
46 state.SetBytesProcessed(int64_t(state.iterations()) * int64_t(chunk_size));
47 delete[] buf;
48 fclose(fp);
49 }
50
BM_stdio_fread(benchmark::State & state)51 void BM_stdio_fread(benchmark::State& state) {
52 ReadWriteTest(state, fread, true);
53 }
54 BENCHMARK(BM_stdio_fread)->AT_COMMON_SIZES;
55
BM_stdio_fwrite(benchmark::State & state)56 void BM_stdio_fwrite(benchmark::State& state) {
57 ReadWriteTest(state, fwrite, true);
58 }
59 BENCHMARK(BM_stdio_fwrite)->AT_COMMON_SIZES;
60
BM_stdio_fread_unbuffered(benchmark::State & state)61 void BM_stdio_fread_unbuffered(benchmark::State& state) {
62 ReadWriteTest(state, fread, false);
63 }
64 BENCHMARK(BM_stdio_fread_unbuffered)->AT_COMMON_SIZES;
65
BM_stdio_fwrite_unbuffered(benchmark::State & state)66 void BM_stdio_fwrite_unbuffered(benchmark::State& state) {
67 ReadWriteTest(state, fwrite, false);
68 }
69 BENCHMARK(BM_stdio_fwrite_unbuffered)->AT_COMMON_SIZES;
70
FopenFgetsFclose(benchmark::State & state,bool no_locking)71 static void FopenFgetsFclose(benchmark::State& state, bool no_locking) {
72 char buf[1024];
73 while (state.KeepRunning()) {
74 FILE* fp = fopen("/proc/version", "re");
75 if (no_locking) __fsetlocking(fp, FSETLOCKING_BYCALLER);
76 if (fgets(buf, sizeof(buf), fp) == nullptr) abort();
77 fclose(fp);
78 }
79 }
80
BM_stdio_fopen_fgets_fclose_locking(benchmark::State & state)81 static void BM_stdio_fopen_fgets_fclose_locking(benchmark::State& state) {
82 FopenFgetsFclose(state, false);
83 }
84 BENCHMARK(BM_stdio_fopen_fgets_fclose_locking);
85
BM_stdio_fopen_fgets_fclose_no_locking(benchmark::State & state)86 void BM_stdio_fopen_fgets_fclose_no_locking(benchmark::State& state) {
87 FopenFgetsFclose(state, true);
88 }
89 BENCHMARK(BM_stdio_fopen_fgets_fclose_no_locking);
90