123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125 |
- #include "memorymanagment.h"
- #if defined(_WIN32)
- #include <windows.h>
- #include <psapi.h>
- #elif defined(__unix__) || defined(__unix) || defined(unix) || (defined(__APPLE__) && defined(__MACH__))
- #include <unistd.h>
- #include <sys/resource.h>
- #if defined(__APPLE__) && defined(__MACH__)
- #include <mach/mach.h>
- #elif (defined(_AIX) || defined(__TOS__AIX__)) || (defined(__sun__) || defined(__sun) || defined(sun) && (defined(__SVR4) || defined(__svr4__)))
- #include <fcntl.h>
- #include <procfs.h>
- #elif defined(__linux__) || defined(__linux) || defined(linux) || defined(__gnu_linux__)
- #include <stdio.h>
- #endif
- #else
- #error "Cannot define getPeakRSS( ) or getCurrentRSS( ) for an unknown OS."
- #endif
- size_t getPeakRSS( )
- {
- #if defined(_WIN32)
-
- PROCESS_MEMORY_COUNTERS info;
- GetProcessMemoryInfo( GetCurrentProcess( ), &info, sizeof(info) );
- return (size_t)info.PeakWorkingSetSize;
- #elif (defined(_AIX) || defined(__TOS__AIX__)) || (defined(__sun__) || defined(__sun) || defined(sun) && (defined(__SVR4) || defined(__svr4__)))
-
- struct psinfo psinfo;
- int fd = -1;
- if ( (fd = open( "/proc/self/psinfo", O_RDONLY )) == -1 )
- return (size_t)0L;
- if ( read( fd, &psinfo, sizeof(psinfo) ) != sizeof(psinfo) )
- {
- close( fd );
- return (size_t)0L;
- }
- close( fd );
- return (size_t)(psinfo.pr_rssize * 1024L);
- #elif defined(__unix__) || defined(__unix) || defined(unix) || (defined(__APPLE__) && defined(__MACH__))
-
- struct rusage rusage;
- getrusage( RUSAGE_SELF, &rusage );
- #if defined(__APPLE__) && defined(__MACH__)
- return (size_t)rusage.ru_maxrss;
- #else
- return (size_t)(rusage.ru_maxrss * 1024L);
- #endif
- #else
-
- return (size_t)0L;
- #endif
- }
- size_t getCurrentRSS( )
- {
- #if defined(_WIN32)
-
- PROCESS_MEMORY_COUNTERS info;
- GetProcessMemoryInfo( GetCurrentProcess( ), &info, sizeof(info) );
- return (size_t)info.WorkingSetSize;
- #elif defined(__APPLE__) && defined(__MACH__)
-
- struct mach_task_basic_info info;
- mach_msg_type_number_t infoCount = MACH_TASK_BASIC_INFO_COUNT;
- if ( task_info( mach_task_self( ), MACH_TASK_BASIC_INFO,
- (task_info_t)&info, &infoCount ) != KERN_SUCCESS )
- return (size_t)0L;
- return (size_t)info.resident_size;
- #elif defined(__linux__) || defined(__linux) || defined(linux) || defined(__gnu_linux__)
-
- long rss = 0L;
- FILE* fp = NULL;
- if ( (fp = fopen( "/proc/self/statm", "r" )) == NULL )
- return (size_t)0L;
- if ( fscanf( fp, "%*s%ld", &rss ) != 1 )
- {
- fclose( fp );
- return (size_t)0L;
- }
- fclose( fp );
- return (size_t)rss * (size_t)sysconf( _SC_PAGESIZE);
- #else
-
- return (size_t)0L;
- #endif
- }
|