Memory.cpp

Go to the documentation of this file.

/***********************************************************************************\
* (c) Copyright 1998-2025 CERN for the benefit of the LHCb and ATLAS collaborations *
*                                                                                   *
* This software is distributed under the terms of the Apache version 2 licence,     *
* copied verbatim in the file "LICENSE".                                            *
*                                                                                   *
* In applying this licence, CERN does not waive the privileges and immunities       *
* granted to it by virtue of its status as an Intergovernmental Organization        *
* or submit itself to any jurisdiction.                                             *
\***********************************************************************************/
#include "ProcessDescriptor.h"
#include <GaudiKernel/Memory.h>
#include <libgen.h>
#include <limits.h>
#include <sys/times.h>
#include <unistd.h>

long System::adjustMemory( MemoryUnit unit, long value ) {
  if ( value != -1 ) {
    switch ( unit ) {
    case Byte:
      break;
    case kByte:
      value = value / 1024;
      break;
    case MByte:
      value = ( value / 1024 ) / 1024;
      break;
    case GByte:
      value = ( ( value / 1024 ) / 1024 ) / 1024;
      break;
    case TByte:
      value = ( ( ( value / 1024 ) / 1024 ) / 1024 ) / 1024;
      break;
    case PByte:
      value = ( ( ( ( value / 1024 ) / 1024 ) / 1024 ) / 1024 ) / 1024;
      break;
    case EByte:
      value = ( ( ( ( ( value / 1024 ) / 1024 ) / 1024 ) / 1024 ) / 1024 ) / 1024;
      break;
    default:
      value = -1;
      break;
    }
  }
  return value;
}

long System::basePriority( InfoType fetch, long pid ) {
  PROCESS_BASIC_INFORMATION info;
  if ( fetch != NoFetch && getProcess()->query( pid, ProcessBasics, &info ) ) return info.BasePriority;
  return 0;
}

long System::procID() {
  static const long s_pid = ::getpid();
  return s_pid;
}

long System::parentID( InfoType fetch, long pid ) {
  PROCESS_BASIC_INFORMATION info;
  if ( fetch != NoFetch && getProcess()->query( pid, ProcessBasics, &info ) ) return info.InheritedFromUniqueProcessId;
  return 0;
}

long System::affinityMask( InfoType fetch, long pid ) {
  PROCESS_BASIC_INFORMATION info;
  if ( fetch != NoFetch && getProcess()->query( pid, ProcessBasics, &info ) ) return info.AffinityMask;
  return 0;
}

long System::exitStatus( InfoType fetch, long pid ) {
  PROCESS_BASIC_INFORMATION info;
  if ( fetch != NoFetch && getProcess()->query( pid, ProcessBasics, &info ) ) return info.ExitStatus;
  return -2;
}

long System::priorityBoost( InfoType fetch, long pid ) {
  long info;
  if ( fetch != NoFetch && getProcess()->query( pid, fetch, &info ) ) return info;
  return -2;
}

long System::nonPagedMemoryPeak( MemoryUnit unit, InfoType fetch, long pid ) {
  VM_COUNTERS info;
  if ( fetch != NoFetch && getProcess()->query( pid, fetch, &info ) )
    return adjustMemory( unit, info.QuotaPeakNonPagedPoolUsage );
  return -2;
}

long System::nonPagedMemory( MemoryUnit unit, InfoType fetch, long pid ) {
  VM_COUNTERS info;
  if ( fetch != NoFetch && getProcess()->query( pid, fetch, &info ) )
    return adjustMemory( unit, info.QuotaNonPagedPoolUsage );
  return -2;
}

long System::nonPagedMemoryLimit( MemoryUnit unit, InfoType fetch, long pid ) {
  POOLED_USAGE_AND_LIMITS quota;
  if ( fetch != NoFetch && getProcess()->query( pid, fetch, &quota ) )
    return adjustMemory( unit, quota.NonPagedPoolLimit );
  return 0;
}

long System::pagedMemory( MemoryUnit unit, InfoType fetch, long pid ) {
  VM_COUNTERS info;
  if ( fetch != NoFetch && getProcess()->query( pid, fetch, &info ) )
    return adjustMemory( unit, info.QuotaPagedPoolUsage );
  return -2;
}

long System::pagedMemoryPeak( MemoryUnit unit, InfoType fetch, long pid ) {
  VM_COUNTERS info;
  if ( fetch != NoFetch && getProcess()->query( pid, fetch, &info ) )
    return adjustMemory( unit, info.QuotaPeakPagedPoolUsage );
  return -2;
}

long System::pagedMemoryLimit( MemoryUnit unit, InfoType fetch, long pid ) {
  POOLED_USAGE_AND_LIMITS quota;
  if ( fetch != NoFetch && getProcess()->query( pid, fetch, &quota ) )
    return adjustMemory( unit, quota.PagedPoolLimit );
  return 0;
}

long System::numPageFault( InfoType fetch, long pid ) {
  VM_COUNTERS info;
  if ( fetch != NoFetch && getProcess()->query( pid, fetch, &info ) ) return info.PageFaultCount;
  return -2;
}

long System::pagefileUsage( MemoryUnit unit, InfoType fetch, long pid ) {
  VM_COUNTERS info;
  if ( fetch != NoFetch && getProcess()->query( pid, fetch, &info ) ) return adjustMemory( unit, info.PagefileUsage );
  return -2;
}

long System::pagefileUsagePeak( MemoryUnit unit, InfoType fetch, long pid ) {
  VM_COUNTERS info;
  if ( fetch != NoFetch && getProcess()->query( pid, fetch, &info ) )
    return adjustMemory( unit, info.PeakPagefileUsage );
  return -2;
}

long System::pagefileUsageLimit( MemoryUnit unit, InfoType fetch, long pid ) {
  POOLED_USAGE_AND_LIMITS quota;
  if ( fetch != NoFetch && getProcess()->query( pid, fetch, &quota ) ) {
    if ( long( quota.PagefileLimit ) < 0 ) return -1; // LONG_MAX;
    return adjustMemory( unit, quota.PagefileLimit );
  }
  return -2;
}

long System::mappedMemory( MemoryUnit unit, InfoType fetch, long pid ) {
  VM_COUNTERS info;
  if ( fetch != NoFetch && getProcess()->query( pid, fetch, &info ) ) return adjustMemory( unit, info.WorkingSetSize );
  return -2;
}

long System::mappedMemoryPeak( MemoryUnit unit, InfoType fetch, long pid ) {
  VM_COUNTERS info;
  if ( fetch != NoFetch && getProcess()->query( pid, fetch, &info ) )
    return adjustMemory( unit, info.PeakWorkingSetSize );
  return -2;
}

long System::minMemoryLimit( MemoryUnit unit, InfoType fetch, long pid ) {
  QUOTA_LIMITS quota;
  if ( fetch != NoFetch && getProcess()->query( pid, fetch, &quota ) )
    return adjustMemory( unit, quota.MinimumWorkingSetSize );
  return 0;
}

long System::maxMemoryLimit( MemoryUnit unit, InfoType fetch, long pid ) {
  QUOTA_LIMITS quota;
  if ( fetch != NoFetch && getProcess()->query( pid, fetch, &quota ) )
    return adjustMemory( unit, quota.MaximumWorkingSetSize );
  return 0;
}

long System::virtualMemory( MemoryUnit unit, InfoType fetch, long pid ) {
  VM_COUNTERS info;
  if ( fetch != NoFetch && getProcess()->query( pid, fetch, &info ) ) return adjustMemory( unit, info.VirtualSize );
  return -2;
}

long System::virtualMemoryPeak( MemoryUnit unit, InfoType fetch, long pid ) {
  VM_COUNTERS info;
  if ( fetch != NoFetch && getProcess()->query( pid, fetch, &info ) ) return adjustMemory( unit, info.PeakVirtualSize );
  return -2;
}

long System::virtualMemoryLimit( MemoryUnit unit, InfoType fetch, long pid ) {
  QUOTA_LIMITS quota;
  if ( fetch != NoFetch && getProcess()->query( pid, fetch, &quota ) ) {
    if ( long( quota.PagefileLimit ) == -1 ) return -1; // LONG_MAX;
    return adjustMemory( unit, quota.PagefileLimit );
  }
  return 0;
}