Drop remaining uses of PassRefPtr in WebCore
[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, size);
45         free(buffer);
46         buffer = nullptr;
47         size = 0;
48     }
49 }
50 #endif
51
52 std::optional<size_t> memoryFootprint()
53 {
54 #if OS(LINUX)
55     FILE* file = fopen("/proc/self/smaps", "r");
56     if (!file)
57         return std::nullopt;
58
59     unsigned long totalPrivateDirtyInKB = 0;
60     bool isAnonymous = false;
61     forEachLine(file, [&] (char* buffer, size_t) {
62         {
63             unsigned long start;
64             unsigned long end;
65             unsigned long offset;
66             unsigned long inode;
67             char dev[32];
68             char perms[5];
69             char path[7];
70             int scannedCount = sscanf(buffer, "%lx-%lx %4s %lx %31s %lu %6s", &start, &end, perms, &offset, dev, &inode, path);
71             if (scannedCount == 6) {
72                 isAnonymous = true;
73                 return;
74             }
75             if (scannedCount == 7) {
76                 StringView pathString(path);
77                 isAnonymous = pathString == ASCIILiteral("[heap]") || pathString.startsWith("[stack");
78                 return;
79             }
80         }
81
82         if (!isAnonymous)
83             return;
84
85         unsigned long privateDirtyInKB;
86         if (sscanf(buffer, "Private_Dirty: %lu", &privateDirtyInKB) == 1)
87             totalPrivateDirtyInKB += privateDirtyInKB;
88     });
89     fclose(file);
90     return totalPrivateDirtyInKB * KB;
91 #endif
92     return std::nullopt;
93 }
94
95 }