1 /* 2 * Copyright 2014 Google Inc. 3 * 4 * Use of this source code is governed by a BSD-style license that can be 5 * found in the LICENSE file. 6 */ 7 8 #include "include/core/SkTypes.h" 9 #include "tools/ProcStats.h" 10 11 #if defined(__Fuchsia__) 12 #include <zircon/process.h> 13 #include <zircon/syscalls.h> 14 #include <zircon/syscalls/object.h> 15 #include <zircon/types.h> 16 getMaxResidentSetSizeMB()17 int sk_tools::getMaxResidentSetSizeMB() { 18 zx_info_task_stats_t task_stats; 19 zx_handle_t process = zx_process_self(); 20 zx_status_t status = zx_object_get_info( 21 process, ZX_INFO_TASK_STATS, &task_stats, sizeof(task_stats), NULL, NULL); 22 if (status != ZX_OK) { 23 return -1; 24 } 25 return (task_stats.mem_private_bytes + task_stats.mem_shared_bytes) / (1 << 20); 26 } 27 #elif defined(SK_BUILD_FOR_UNIX) || defined(SK_BUILD_FOR_MAC) || defined(SK_BUILD_FOR_IOS) || defined(SK_BUILD_FOR_ANDROID) 28 #include <sys/resource.h> getMaxResidentSetSizeMB()29 int sk_tools::getMaxResidentSetSizeMB() { 30 struct rusage ru; 31 getrusage(RUSAGE_SELF, &ru); 32 #if defined(SK_BUILD_FOR_MAC) || defined(SK_BUILD_FOR_IOS) 33 return static_cast<int>(ru.ru_maxrss / 1024 / 1024); // Darwin reports bytes. 34 #else 35 return static_cast<int>(ru.ru_maxrss / 1024); // Linux reports kilobytes. 36 #endif 37 } 38 #elif defined(SK_BUILD_FOR_WIN) 39 #include <windows.h> 40 #include <psapi.h> getMaxResidentSetSizeMB()41 int sk_tools::getMaxResidentSetSizeMB() { 42 PROCESS_MEMORY_COUNTERS info; 43 GetProcessMemoryInfo(GetCurrentProcess(), &info, sizeof(info)); 44 return static_cast<int>(info.PeakWorkingSetSize / 1024 / 1024); // Windows reports bytes. 45 } 46 #else getMaxResidentSetSizeMB()47 int sk_tools::getMaxResidentSetSizeMB() { return -1; } 48 #endif 49 50 #if defined(SK_BUILD_FOR_MAC) || defined(SK_BUILD_FOR_IOS) 51 #include <mach/mach.h> getCurrResidentSetSizeMB()52 int sk_tools::getCurrResidentSetSizeMB() { 53 mach_task_basic_info info; 54 mach_msg_type_number_t count = MACH_TASK_BASIC_INFO_COUNT; 55 if (KERN_SUCCESS != 56 task_info(mach_task_self(), MACH_TASK_BASIC_INFO, (task_info_t)&info, &count)) { 57 return -1; 58 } 59 return info.resident_size / 1024 / 1024; // Darwin reports bytes. 60 } 61 #elif defined(SK_BUILD_FOR_UNIX) || defined(SK_BUILD_FOR_ANDROID) // N.B. /proc is Linux-only. 62 #include <unistd.h> 63 #include <stdio.h> getCurrResidentSetSizeMB()64 int sk_tools::getCurrResidentSetSizeMB() { 65 const long pageSize = sysconf(_SC_PAGESIZE); 66 long long rssPages = 0; 67 if (FILE* statm = fopen("/proc/self/statm", "r")) { 68 // statm contains: program-size rss shared text lib data dirty, all in page counts. 69 int rc = fscanf(statm, "%*d %lld", &rssPages); 70 fclose(statm); 71 if (rc != 1) { 72 return -1; 73 } 74 } 75 return rssPages * pageSize / 1024 / 1024; 76 } 77 78 #elif defined(SK_BUILD_FOR_WIN) getCurrResidentSetSizeMB()79 int sk_tools::getCurrResidentSetSizeMB() { 80 PROCESS_MEMORY_COUNTERS info; 81 GetProcessMemoryInfo(GetCurrentProcess(), &info, sizeof(info)); 82 return static_cast<int>(info.WorkingSetSize / 1024 / 1024); // Windows reports bytes. 83 } 84 #else getCurrResidentSetSizeMB()85 int sk_tools::getCurrResidentSetSizeMB() { return -1; } 86 #endif 87