1  """Memory watchdog: periodically read the memory usage of the main test process
       2  and print it out, until terminated."""
       3  # stdin should refer to the process' /proc/<PID>/statm: we don't pass the
       4  # process' PID to avoid a race condition in case of - unlikely - PID recycling.
       5  # If the process crashes, reading from the /proc entry will fail with ESRCH.
       6  
       7  
       8  import os
       9  import sys
      10  import time
      11  
      12  
      13  try:
      14      page_size = os.sysconf('SC_PAGESIZE')
      15  except (ValueError, AttributeError):
      16      try:
      17          page_size = os.sysconf('SC_PAGE_SIZE')
      18      except (ValueError, AttributeError):
      19          page_size = 4096
      20  
      21  while True:
      22      sys.stdin.seek(0)
      23      statm = sys.stdin.read()
      24      data = int(statm.split()[5])
      25      sys.stdout.write(" ... process data size: {data:.1f}G\n"
      26                       .format(data=data * page_size / (1024 ** 3)))
      27      sys.stdout.flush()
      28      time.sleep(1)