A set of minimal dependency bootstrap binaries
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
stage0/User_Interface.py

263 lines
7.4 KiB

## Copyright (C) 2016 Jeremiah Orians
## This file is part of stage0.
##
## stage0 is free software: you can redistribute it and/or modify
## it under the terms of the GNU General Public License as published by
## the Free Software Foundation, either version 3 of the License, or
## (at your option) any later version.
##
## stage0 is distributed in the hope that it will be useful,
## but WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
## GNU General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with stage0. If not, see <http://www.gnu.org/licenses/>.
import subprocess
import ctypes
import re
import sys, getopt
vm = ctypes.CDLL('./libvm.so')
vm.initialize_lilith.argtype = (ctypes.c_uint, ctypes.c_uint)
vm.get_memory.argtype = ctypes.c_uint
vm.get_memory.restype = ctypes.c_char_p
vm.step_lilith.restype = ctypes.c_uint
vm.set_register.argtype = (ctypes.c_uint, ctypes.c_uint)
vm.set_memory.argtype = (ctypes.c_uint, ctypes.c_ubyte)
vm.get_register.argtype = ctypes.c_uint
vm.get_register.restype = ctypes.c_uint
POSIX_MODE = False
def Reset_lilith():
global Memory_Size
if 0 <= Memory_Size < 1024:
unit = 'Bytes'
chunks = Memory_Size
elif 1024 <= Memory_Size < (1024 * 1024):
unit = 'KB'
chunks = Memory_Size / 1024
elif (1024 * 1024) <= Memory_Size < (1024 * 1024 * 1024):
unit = 'MB'
chunks = Memory_Size / (1024 * 1024)
else:
unit = 'GB'
chunks = Memory_Size / (1024 * 1024 * 1024)
print("Current Memory Size is: " + str(chunks) + unit)
vm.initialize_lilith(Memory_Size, POSIX_MODE)
global Current_IP
Current_IP = 0
global Watchpoints
Watchpoints = {0}
global ROM_Name
size = vm.load_lilith(ctypes.create_string_buffer(ROM_Name.encode('ascii')))
print ("Size of loaded ROM image: " + str(size) + " bytes\n")
if(POSIX_MODE):
Set_Register(15, size)
def Step_lilith():
global Current_IP
Current_IP = vm.step_lilith()
global Count
Count = Count + 1
return
def Set_Memory(address, value):
vm.set_memory(address, value)
return
def Set_Register(register, value):
vm.set_register(register, value)
return
def returnPage():
return get_header() + (vm.get_memory(Current_Page)).decode('utf-8') + get_spacer1() + get_registers(0) + get_registers(9) + get_spacer2() + get_Window_shortcut() + get_disassembled() + get_footer()
hexlookup = { 0 : '0', 1 : '1', 2 : '2', 3 : '3', 4 : '4', 5 : '5', 6 : '6', 7 : '7', 8 : '8', 9 : '9', 10 : 'A', 11 : 'B', 12 : 'C', 13 : 'D', 14 : 'E', 15 : 'F' }
def formatByte(a):
first = a >> 4
second = a % 16
return str(hexlookup[first]+hexlookup[second])
def formatAddress(a):
first = a >> 24
second = (a % 16777216) >> 16
third = (a % 65536) >> 8
fourth = a % 256
myreturn = formatByte(first) + formatByte(second) + formatByte(third) + formatByte(fourth)
return myreturn[:-1] + "x"
def formatRegister(a):
first = a >> 24
second = (a % 16777216) >> 16
third = (a % 65536) >> 8
fourth = a % 256
return formatByte(first) + formatByte(second) + formatByte(third) + formatByte(fourth)
def get_header():
return """
<!DOCTYPE html>
<html lang="en" xmlns="http://www.w3.org/1999/xhtml">
<head>
<meta charset="utf-8" />
<title>Knight CPU Debugger</title>
<link rel="stylesheet" type="text/css" href="static/style.css" />
<script type="text/javascript" src="static/jquery-1.8.3.js"> </script>
<script type="text/javascript" src="static/script.js"></script>
</head>
<body>
<div>
<a href="RUN"><button type="button">RUN</button></a>
<a href="STOP"><button type="button">STOP</button></a>
<a href="PAGEDOWN"><button type="button">PAGE DOWN</button></a>
<a href="PAGEUP"><button type="button">PAGE UP</button></a>
<a href="STEP"><button type="button">STEP</button></a>
<a href="RESET"><button type="button">RESET</button></a>
<a href="SPEEDBREAKPOINT"><button type="button">Run Until Breakpoint</button></a>
</div>
<div>
<div style="height:230px; width:60%; float:left; overflow-y: scroll;">
<table class="Memory">
<thead>
<tr>
<th>Index</th>
<th>0</th>
<th>1</th>
<th>2</th>
<th>3</th>
<th>4</th>
<th>5</th>
<th>6</th>
<th>7</th>
<th>8</th>
<th>8</th>
<th>A</th>
<th>B</th>
<th>C</th>
<th>D</th>
<th>E</th>
<th>F</th>
</tr>
</thead>
<tbody>"""
def get_spacer1():
return """ </tbody>
</table>
</div>
<div>
"""
def get_registers(index):
temp = """ <table class="Registers">
<thead>
<tr>
<th>Register</th>
<th>Value</th>
<th>Name</th>
</tr>
</thead>
<tbody>"""
if (0 == index):
for i in range(0,9):
temp = temp + """<tr><td>R""" + str(index + i) + "</td><td>" + formatRegister(vm.get_register(index + i)) + "</td>" + "<td>REG" + str(index + i) +"</tr>\n"
else:
for i in range(0,7):
temp = temp + """<tr><td>R""" + str(index + i) + "</td><td>" + formatRegister(vm.get_register(index + i)) + "</td>" + "<td>REG" + str(index + i) +"</tr>\n"
temp = temp + """<tr><td>R16</td><td>""" + formatRegister(vm.get_register(16)) + "</td><td> PC </td></tr>\n"
temp = temp + """<tr><td>R17</td><td>""" + formatRegister(vm.get_register(17)) + "</td><td>Counter</td></tr>\n"
return temp + """ </tbody>
</table> """
def get_spacer2():
return """
</div>
</div>
"""
def get_Window_shortcut():
return """
<form action="WINDOW" style="position:absolute; left:10px; top:260px;">
<input type="submit" value="Window Fast Move"> <input type="text" name="Window">
</form>
"""
def get_disassembled():
Current_IP = vm.get_register(16)
f = open('z_disassembled', "r")
temp = """ <div style="position:absolute; left:10px; top:280px; overflow-y: scroll; height:200px; width:60%;">
<table class="Debug"><tbody>"""
for line in f:
pieces = re.split(r'\t+', line)
i = int(pieces[0], 16)
if (i < Current_IP):
temp = temp + ""
elif (i == Current_IP):
temp = temp + """<tr class="current"><td>""" + pieces[0] + "</td><td>" + pieces[1] + "</td><td>" + pieces[2] + "</td></tr>\n"
elif i in Watchpoints:
temp = temp + """<tr class="breakpoint"><td>""" + pieces[0] + "</td><td>" + pieces[1] + "</td><td>" + pieces[2] + "</td></tr>\n"
else:
temp = temp + "<tr><td>" + pieces[0] + "</td><td>" + pieces[1] + "</td><td>" + pieces[2] + "</td></tr>\n"
f.close()
return temp + "</tbody></table>"
def get_footer():
return """
</div>
</body>
</html>
"""
def main(argv):
global Debug_Point
help_string = 'Knight.py --ROM=$NAME [--DEBUG=$NUMBER] [--WINDOW=$NUMBER] [--MEMORY=$SIZE]\n'
try:
opts, args = getopt.getopt(argv,"R:D:W:M:P:",["ROM=","DEBUG=","WINDOW=", "MEMORY=", "POSIX-MODE"])
except getopt.GetoptError:
print (help_string)
sys.exit(2)
for opt, arg in opts:
if opt == '-h':
print (help_string)
sys.exit()
elif opt in ("-R", "--ROM"):
global ROM_Name
ROM_Name = arg
elif opt in ("-D", "--DEBUG"):
global Debug_Point
Debug_Point = int(arg)
elif opt in ("-W", "--WINDOW"):
global Current_Page
Current_Page = int(arg, 16)
elif opt in ("-M", "--MEMORY"):
global Memory_Size
if arg.endswith('K'):
Memory_Size = (int(arg[:-1]) * 1024)
elif arg.endswith('M'):
Memory_Size = (int(arg[:-1]) * 1024 * 1024)
elif arg.endswith('G'):
Memory_Size = (int(arg[:-1]) * 1024 * 1024 * 1024)
elif opt in ("-P", "--POSIX-MODE"):
global POSIX_MODE
POSIX_MODE = True
subprocess.call("./bin/dis " + ROM_Name + " >| z_disassembled", shell=True)
Reset_lilith()
Current_IP = 0
Memory_Size = 16 * 1024
Current_Page = 0
Watchpoints = {0}
Count=0
Debug_Point = 0
ROM_Name = "rom"