[Linux] Port MallocBench
[WebKit-https.git] / PerformanceTests / MallocBench / MallocBench / balloon.cpp
1 /*
2  * Copyright (C) 2014 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
24  */
25
26 #include "CPUCount.h"
27 #include "Memory.h"
28 #include "balloon.h"
29 #include <array>
30 #include <chrono>
31 #include <memory>
32 #include <stddef.h>
33 #include <strings.h>
34
35 #include "mbmalloc.h"
36
37 void benchmark_balloon(CommandLine&)
38 {
39     const size_t chunkSize = 1 * 1024;
40     const size_t balloonSize = 100 * 1024 * 1024;
41     const size_t steadySize = 10 * 1024 * 1024;
42     
43     std::array<void*, balloonSize / chunkSize> balloon;
44     std::array<void*, balloonSize / chunkSize> steady;
45
46     auto start = std::chrono::steady_clock::now();
47
48     for (size_t i = 0; i < balloon.size(); ++i) {
49         balloon[i] = mbmalloc(chunkSize);
50         bzero(balloon[i], chunkSize);
51     }
52
53     for (size_t i = 0; i < balloon.size(); ++i)
54         mbfree(balloon[i], chunkSize);
55
56     auto stop = std::chrono::steady_clock::now();
57     
58     auto benchmarkTime = stop - start;
59
60     start = std::chrono::steady_clock::now();
61
62     // Converts bytes to time -- for reporting's sake -- by waiting a while until
63     // the heap shrinks back down. This isn't great for pooling with other
64     // benchmarks in a geometric mean of throughput, but it's OK for basic testing.
65     while (currentMemoryBytes().resident > 2 * steadySize
66         && std::chrono::steady_clock::now() - start < 8 * benchmarkTime) {
67         for (size_t i = 0; i < steady.size(); ++i) {
68             steady[i] = mbmalloc(chunkSize);
69             bzero(steady[i], chunkSize);
70         }
71
72         for (size_t i = 0; i < steady.size(); ++i)
73             mbfree(steady[i], chunkSize);
74     }
75 }