runtime/tools/create_snapshot_file.py

#!/usr/bin/env python
#
# Copyright (c) 2011, the Dart project authors.  Please see the AUTHORS file
# for details. All rights reserved. Use of this source code is governed by a
# BSD-style license that can be found in the LICENSE file.

# Script to create snapshot files.

import getopt
import optparse
import string
import subprocess
import sys
import utils


HOST_OS = utils.GuessOS()
HOST_CPUS = utils.GuessCpus()


def BuildOptions():
  result = optparse.OptionParser()
  result.add_option("--vm_input_bin",
      action="store", type="string",
      help="input file name of the vm isolate snapshot in binary form")
  result.add_option("--input_bin",
      action="store", type="string",
      help="input file name of the isolate snapshot in binary form")
  result.add_option("--input_cc",
      action="store", type="string",
      help="input file name which contains the C buffer template")
  result.add_option("--output",
      action="store", type="string",
      help="output file name into which snapshot in C buffer form is generated")
  result.add_option("-v", "--verbose",
      help='Verbose output.',
      default=False, action="store_true")
  return result


def ProcessOptions(options):
  if not options.vm_input_bin:
    sys.stderr.write('--vm_input_bin not specified\n')
    return False
  if not options.input_bin:
    sys.stderr.write('--input_bin not specified\n')
    return False
  if not options.input_cc:
    sys.stderr.write('--input_cc not specified\n')
    return False
  if not options.output:
    sys.stderr.write('--output not specified\n')
    return False
  return True


def makeString(input_file):
  result = ' '
  fileHandle = open(input_file, 'rb')
  lineCounter = 0
  for byte in fileHandle.read():
    result += ' %d,' % ord(byte)
    lineCounter += 1
    if lineCounter == 10:
      result += '\n   '
      lineCounter = 0
  if lineCounter != 0:
    result += '\n   '
  return result


def makeFile(output_file, input_cc_file,
             vm_isolate_input_file, isolate_input_file):
  snapshot_cc_text = open(input_cc_file).read()
  snapshot_cc_text = snapshot_cc_text % (makeString(vm_isolate_input_file),
                                         makeString(isolate_input_file))
  open(output_file, 'w').write(snapshot_cc_text)
  return True


def Main():
  # Parse options.
  parser = BuildOptions()
  (options, args) = parser.parse_args()
  if not ProcessOptions(options):
    parser.print_help()
    return 1

  # If there are additional arguments, report error and exit.
  if args:
    parser.print_help()
    return 1

  if not makeFile(options.output, options.input_cc,
                  options.vm_input_bin, options.input_bin):
    print "Unable to generate snapshot in C buffer form"
    return -1

  return 0

if __name__ == '__main__':
  sys.exit(Main())