RISCI_ATOM
/
Stage0
mirror of https://git.savannah.gnu.org/git/stage0.git


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254
							## Copyright (C) 2016 Jeremiah Orians
## This file is part of stage0.
##
## stage0 is free software: you an 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
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

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)
	global Current_IP
	Current_IP = 0
	global Watchpoints
	Watchpoints = {0}
	global ROM_Name
	vm.load_lilith(ctypes.create_string_buffer(ROM_Name.encode('ascii')))

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:",["ROM=","DEBUG=","WINDOW=", "MEMORY="])
	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)

	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"