2016-07-15 04:16:25 +00:00
|
|
|
#! /usr/bin/env python3
|
|
|
|
# encoding: utf-8
|
|
|
|
import os
|
|
|
|
import logging
|
|
|
|
import subprocess
|
|
|
|
import json
|
|
|
|
|
|
|
|
|
|
|
|
class Sins():
|
|
|
|
logger = logging.getLogger()
|
|
|
|
|
|
|
|
def __init__(self, parent, seed=None, run_dir=None):
|
|
|
|
self.logger.info('execute\n{}'.format(json.dumps(
|
|
|
|
(parent, seed, run_dir))
|
|
|
|
))
|
|
|
|
|
|
|
|
if os.path.isdir(run_dir):
|
|
|
|
self.run_dir = os.path.abspath(run_dir)
|
|
|
|
|
2016-07-15 05:20:57 +00:00
|
|
|
self.seed = os.path.abspath(seed)
|
2016-07-15 04:16:25 +00:00
|
|
|
|
|
|
|
parent = os.path.abspath(parent)
|
|
|
|
if not os.path.isfile(parent):
|
|
|
|
raise ValueError('Invalid executable image path.')
|
|
|
|
|
2016-07-15 05:20:57 +00:00
|
|
|
self.parent = parent
|
2016-07-15 04:16:25 +00:00
|
|
|
|
|
|
|
parsed = self.parent, self.seed, self.run_dir
|
|
|
|
self.logger.info('parsed\n{}'.format(json.dumps(parsed)))
|
|
|
|
|
|
|
|
while True:
|
2016-07-15 05:20:57 +00:00
|
|
|
self.execute(self.parent, self.seed, self.run_dir)
|
|
|
|
paths = sorted(os.listdir(self.run_dir))
|
|
|
|
for path in paths:
|
|
|
|
scrap_path = os.path.join(self.run_dir, path)
|
|
|
|
if os.path.isfile(scrap_path):
|
|
|
|
self.execute(
|
|
|
|
self.parent,
|
|
|
|
scrap_path,
|
|
|
|
self.run_dir
|
|
|
|
)
|
2016-07-15 04:16:25 +00:00
|
|
|
|
|
|
|
def scrap_recent(self, run_dir):
|
|
|
|
scraps = sorted(os.listdir(run_dir))
|
|
|
|
|
|
|
|
if scraps:
|
|
|
|
return os.path.join(run_dir, scraps[-1])
|
|
|
|
|
|
|
|
return None
|
|
|
|
|
2016-07-15 05:20:57 +00:00
|
|
|
def execute(self, parent, scrap, cwd):
|
|
|
|
self.logger.info('execute\n{}'.format(
|
|
|
|
json.dumps((parent, scrap, cwd))
|
|
|
|
))
|
|
|
|
subprocess.run([parent, scrap], cwd=cwd)
|
2016-07-15 04:16:25 +00:00
|
|
|
|
|
|
|
|
|
|
|
def hex_dumps(scrap_dir):
|
|
|
|
scrap_dir = os.path.abspath(scrap_dir)
|
2016-07-15 05:20:57 +00:00
|
|
|
dump_dir = os.path.join(scrap_dir, 'hexdumps')
|
|
|
|
|
|
|
|
try:
|
|
|
|
os.mkdir(dump_dir)
|
|
|
|
except:
|
|
|
|
pass
|
|
|
|
|
2016-07-15 04:16:25 +00:00
|
|
|
scraps = os.listdir(scrap_dir)
|
2016-07-15 05:20:57 +00:00
|
|
|
|
2016-07-15 04:16:25 +00:00
|
|
|
for scrap in scraps:
|
2016-07-15 05:20:57 +00:00
|
|
|
scrap_path = os.path.join(scrap_dir, scrap)
|
|
|
|
hex_name = '{}.hex'.format(scrap)
|
|
|
|
hex_path = os.path.join(dump_dir, hex_name)
|
|
|
|
|
|
|
|
if os.path.isfile(scrap_path):
|
|
|
|
with open(scrap_path, 'rb') as file_in:
|
|
|
|
with open(hex_path, 'w') as file_out:
|
2016-07-15 04:16:25 +00:00
|
|
|
binary = bytes(file_in.read())
|
|
|
|
for each in binary:
|
|
|
|
file_out.writelines('\'\\x{:02X}\',\n'.format(each))
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
import argparse
|
|
|
|
|
2016-07-15 05:20:57 +00:00
|
|
|
logger = logging.getLogger()
|
|
|
|
logger.setLevel(logging.DEBUG)
|
|
|
|
|
|
|
|
file_handler = logging.FileHandler(filename='sins.log')
|
|
|
|
formatter = logging.Formatter(
|
|
|
|
'#%(asctime)s %(name)s %(levelname)s\n##%(message)s\n'
|
|
|
|
)
|
|
|
|
file_handler.setFormatter(formatter)
|
|
|
|
logger.addHandler(file_handler)
|
|
|
|
|
2016-07-15 04:16:25 +00:00
|
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument('-seed', default='build/scrap.asm.2.o')
|
|
|
|
parser.add_argument('-parent', default='build/sins')
|
|
|
|
parser.add_argument('-dir', default='sandbox')
|
|
|
|
parser.add_argument('-dumps', action='store_true')
|
2016-07-15 05:20:57 +00:00
|
|
|
parser.add_argument('-build', action='store_true')
|
2016-07-15 04:16:25 +00:00
|
|
|
args = parser.parse_args()
|
|
|
|
|
2016-07-15 05:20:57 +00:00
|
|
|
if args.build:
|
|
|
|
waf = ['python3', 'waf.py', 'configure', 'build']
|
|
|
|
waf_proc = subprocess.run(waf, stdout=subprocess.PIPE)
|
|
|
|
logger.info(waf_proc.stdout)
|
|
|
|
|
2016-07-15 04:16:25 +00:00
|
|
|
if args.dumps:
|
|
|
|
hex_dumps(args.dir)
|
|
|
|
else:
|
|
|
|
sins = Sins(parent=args.parent, seed=args.seed, run_dir=args.dir)
|