author Mike Hommey <>
Fri, 25 Oct 2013 08:23:05 +0900
changeset 152035 5dd08c88e328c4b821f5880340d15eaff8b3f92b
parent 148852 c3fd83b98898bc9ae8e2f743f6f8901cdf5d7c59
permissions -rw-r--r--
Bug 929905 - Consolidate sources in r=gps

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at

from __future__ import with_statement
import os, subprocess, sys, threading, time
from win32 import procmem

def measure_vsize_threadfunc(proc, output_file):
    Measure the virtual memory usage of |proc| at regular intervals
    until it exits, then print the maximum value and write it to
    |output_file|.  Also, print something to the console every
    half an hour to prevent the build job from getting killed when
    linking a large PGOed binary.
    maxvsize = 0
    idleTime = 0
    while proc.returncode is None:
        maxvsize, vsize = procmem.get_vmsize(proc._handle)
        idleTime += 0.5
        if idleTime > 30 * 60:
          print "Still linking, 30 minutes passed..."
          idleTime = 0
    print "TinderboxPrint: linker max vsize: %d" % maxvsize
    with open(output_file, "w") as f:
        f.write("%d\n" % maxvsize)

def measure_link_vsize(output_file, args):
    Execute |args|, and measure the maximum virtual memory usage of the process,
    printing it to stdout when finished.
    proc = subprocess.Popen(args)
    t = threading.Thread(target=measure_vsize_threadfunc,
                         args=(proc, output_file))
    # Wait for the linker to finish.
    exitcode = proc.wait()
    # ...and then wait for the background thread to finish.
    return exitcode

if __name__ == "__main__":
    if sys.platform != "win32":
        print >>sys.stderr, " is only for use on Windows!"
    if len(sys.argv) < 3:
        print >>sys.stderr, "Usage: <output filename> <commandline>"
    sys.exit(measure_link_vsize(sys.argv[1], sys.argv[2:]))