memoryFootprint should return size_t not optional<size_t>
[WebKit-https.git] / Source / WTF / wtf / linux / MemoryFootprintLinux.cpp
1 /*
2  * Copyright (C) 2017 Yusuke Suzuki <utatane.tea@gmail.com>.
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 "config.h"
27 #include "MemoryFootprint.h"
28
29 #if OS(LINUX)
30 #include <stdio.h>
31 #include <wtf/StdLibExtras.h>
32 #include <wtf/text/StringView.h>
33 #endif
34
35 namespace WTF {
36
37 #if OS(LINUX)
38 template<typename Functor>
39 static void forEachLine(FILE* file, Functor functor)
40 {
41     char* buffer = nullptr;
42     size_t size = 0;
43     while (getline(&buffer, &size, file) != -1) {
44         functor(buffer);
45     }
46     free(buffer);
47 }
48 #endif
49
50 size_t memoryFootprint()
51 {
52 #if OS(LINUX)
53     FILE* file = fopen("/proc/self/smaps", "r");
54     if (!file)
55         return 0;
56
57     unsigned long totalPrivateDirtyInKB = 0;
58     bool isAnonymous = false;
59     forEachLine(file, [&] (char* buffer) {
60         {
61             unsigned long start;
62             unsigned long end;
63             unsigned long offset;
64             unsigned long inode;
65             char dev[32];
66             char perms[5];
67             char path[7];
68             int scannedCount = sscanf(buffer, "%lx-%lx %4s %lx %31s %lu %6s", &start, &end, perms, &offset, dev, &inode, path);
69             if (scannedCount == 6) {
70                 isAnonymous = true;
71                 return;
72             }
73             if (scannedCount == 7) {
74                 StringView pathString(path);
75                 isAnonymous = pathString == "[heap]" || pathString.startsWith("[stack");
76                 return;
77             }
78         }
79
80         if (!isAnonymous)
81             return;
82
83         unsigned long privateDirtyInKB;
84         if (sscanf(buffer, "Private_Dirty: %lu", &privateDirtyInKB) == 1)
85             totalPrivateDirtyInKB += privateDirtyInKB;
86     });
87     fclose(file);
88     return totalPrivateDirtyInKB * KB;
89 #endif
90     return 0;
91 }
92
93 }