From: Elliott Hughes <enh@google.com>
Date: Mon, 1 Dec 2014 21:12:18 +0000 (-0800)
Subject: Benchmark fread/fwrite both buffered and unbuffered.
X-Git-Tag: android-x86-7.1-r1~757^2~108^2~203^2
X-Git-Url: http://git.osdn.net/view?a=commitdiff_plain;h=47dc7c974e71f8d6275288f8137156ce317bf53c;p=android-x86%2Fbionic.git

Benchmark fread/fwrite both buffered and unbuffered.

Bug: 18556607
Change-Id: I53905aedcea75fe550d9c423fb701c2c9bf8e831
---

diff --git a/benchmarks/stdio_benchmark.cpp b/benchmarks/stdio_benchmark.cpp
index e899df7c1..386ea04c9 100644
--- a/benchmarks/stdio_benchmark.cpp
+++ b/benchmarks/stdio_benchmark.cpp
@@ -25,14 +25,19 @@
     Arg(1)->Arg(2)->Arg(3)->Arg(4)->Arg(8)->Arg(16)->Arg(32)->Arg(64)->Arg(512)-> \
     Arg(1*KB)->Arg(4*KB)->Arg(8*KB)->Arg(16*KB)->Arg(64*KB)
 
-static void BM_stdio_fread(int iters, int chunk_size) {
+template <typename Fn>
+static void ReadWriteTest(int iters, int chunk_size, Fn f, bool buffered) {
   StopBenchmarkTiming();
   FILE* fp = fopen("/dev/zero", "rw");
   char* buf = new char[chunk_size];
   StartBenchmarkTiming();
 
+  if (!buffered) {
+    setvbuf(fp, 0, _IONBF, 0);
+  }
+
   for (int i = 0; i < iters; ++i) {
-    fread(buf, chunk_size, 1, fp);
+    f(buf, chunk_size, 1, fp);
   }
 
   StopBenchmarkTiming();
@@ -40,22 +45,23 @@ static void BM_stdio_fread(int iters, int chunk_size) {
   delete[] buf;
   fclose(fp);
 }
-BENCHMARK(BM_stdio_fread)->AT_COMMON_SIZES;
 
+static void BM_stdio_fread(int iters, int chunk_size) {
+  ReadWriteTest(iters, chunk_size, fread, true);
+}
+BENCHMARK(BM_stdio_fread)->AT_COMMON_SIZES;
 
 static void BM_stdio_fwrite(int iters, int chunk_size) {
-  StopBenchmarkTiming();
-  FILE* fp = fopen("/dev/zero", "rw");
-  char* buf = new char[chunk_size];
-  StartBenchmarkTiming();
+  ReadWriteTest(iters, chunk_size, fwrite, true);
+}
+BENCHMARK(BM_stdio_fwrite)->AT_COMMON_SIZES;
 
-  for (int i = 0; i < iters; ++i) {
-      fwrite(buf, chunk_size, 1, fp);
-  }
+static void BM_stdio_fread_unbuffered(int iters, int chunk_size) {
+  ReadWriteTest(iters, chunk_size, fread, false);
+}
+BENCHMARK(BM_stdio_fread_unbuffered)->AT_COMMON_SIZES;
 
-  StopBenchmarkTiming();
-  SetBenchmarkBytesProcessed(int64_t(iters) * int64_t(chunk_size));
-  delete[] buf;
-  fclose(fp);
+static void BM_stdio_fwrite_unbuffered(int iters, int chunk_size) {
+  ReadWriteTest(iters, chunk_size, fwrite, false);
 }
-BENCHMARK(BM_stdio_fwrite)->AT_COMMON_SIZES;
+BENCHMARK(BM_stdio_fwrite_unbuffered)->AT_COMMON_SIZES;