diff --git a/compiler-rt/lib/scudo/standalone/common.h b/compiler-rt/lib/scudo/standalone/common.h --- a/compiler-rt/lib/scudo/standalone/common.h +++ b/compiler-rt/lib/scudo/standalone/common.h @@ -132,6 +132,10 @@ const char *getEnv(const char *Name); +u64 GetRSSFromBuffer(const char *buf); + +u64 getRSS(); + u64 getMonotonicTime(); u32 getThreadID(); diff --git a/compiler-rt/lib/scudo/standalone/common.cpp b/compiler-rt/lib/scudo/standalone/common.cpp --- a/compiler-rt/lib/scudo/standalone/common.cpp +++ b/compiler-rt/lib/scudo/standalone/common.cpp @@ -35,4 +35,10 @@ die(); } +#if !SCUDO_LINUX +u64 GetRSSFromBuffer(const char *buf) { return 0; } + +u64 GetRSS() { return 0; } +#endif + } // namespace scudo diff --git a/compiler-rt/lib/scudo/standalone/internal_defs.h b/compiler-rt/lib/scudo/standalone/internal_defs.h --- a/compiler-rt/lib/scudo/standalone/internal_defs.h +++ b/compiler-rt/lib/scudo/standalone/internal_defs.h @@ -88,6 +88,9 @@ typedef int16_t s16; typedef int32_t s32; typedef int64_t s64; +typedef int fd_t; + +#define kInvalidFd ((fd_t)-1) // The following two functions have platform specific implementations. void outputRaw(const char *Buffer); diff --git a/compiler-rt/lib/scudo/standalone/linux.cpp b/compiler-rt/lib/scudo/standalone/linux.cpp --- a/compiler-rt/lib/scudo/standalone/linux.cpp +++ b/compiler-rt/lib/scudo/standalone/linux.cpp @@ -19,6 +19,7 @@ #include #include #include +#include #include #include #include @@ -180,6 +181,38 @@ extern "C" WEAK int async_safe_write_log(int pri, const char *tag, const char *msg); +fd_t OpenFile(const char *FileName, int flags) { return open(FileName, flags); } + +u64 GetRSSFromBuffer(const char *buf) { + // The format of the file is: + // 1084 89 69 11 0 79 0 + // We need the second number which is RSS in pages. + const char *pos = buf; + // Skip the first number. + while (*pos >= '0' && *pos <= '9') + pos++; + // Skip whitespaces. + while (!(*pos >= '0' && *pos <= '9') && *pos != 0) + pos++; + // Read the number. + u64 rss = 0; + while (*pos >= '0' && *pos <= '9') + rss = rss * 10 + static_cast(*(pos)++) - '0'; + return rss * getPageSizeCached(); +} + +u64 GetRSS() { + fd_t fd = open("/proc/self/statm", O_RDONLY); + char buf[64]; + s64 len = read(fd, buf, sizeof(buf) - 1); + close(fd); + if (len <= 0) + return 0; + buf[len] = 0; + + return GetRSSFromBuffer(buf); +} + void outputRaw(const char *Buffer) { if (&async_safe_write_log) { constexpr s32 AndroidLogInfo = 4; diff --git a/compiler-rt/lib/scudo/standalone/tests/common_test.cpp b/compiler-rt/lib/scudo/standalone/tests/common_test.cpp --- a/compiler-rt/lib/scudo/standalone/tests/common_test.cpp +++ b/compiler-rt/lib/scudo/standalone/tests/common_test.cpp @@ -69,4 +69,13 @@ unmap(P, Size, 0, &Data); } +#if SCUDO_LINUX +TEST(ScudoCommonTest, GetRssFromBuffer) { + std::string buf = "1084 89 69 11 0 79 0"; + auto result = scudo::GetRSSFromBuffer(buf.data()); + + EXPECT_EQ(result, 89 * scudo::getPageSizeCached()); +} +#endif // SCUDO_LINUX + } // namespace scudo