mirror of
https://github.com/hsoft/collapseos.git
synced 2024-11-27 12:58:09 +11:00
60 lines
1.6 KiB
Python
Executable File
60 lines
1.6 KiB
Python
Executable File
#!/usr/bin/python
|
|
|
|
# Read specified number of bytes in specified blkdev ID and spit it to stdout.
|
|
# The proper blkdev has to be selected and placed already.
|
|
|
|
import argparse
|
|
import os
|
|
import sys
|
|
import time
|
|
|
|
# Some place where it's safe to write 0xff bytes.
|
|
MEMPTR = '9000'
|
|
|
|
def sendcmd(fd, cmd):
|
|
# The serial link echoes back all typed characters and expects us to read
|
|
# them. We have to send each char one at a time.
|
|
print("Executing {}".format(cmd.decode()), file=sys.stderr)
|
|
for c in cmd:
|
|
os.write(fd, bytes([c]))
|
|
os.read(fd, 1)
|
|
os.write(fd, b'\n')
|
|
os.read(fd, 2) # sends back \r\n
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('device')
|
|
parser.add_argument('bytecount')
|
|
args = parser.parse_args()
|
|
|
|
try:
|
|
bytecount = int(args.bytecount, 16)
|
|
except ValueError:
|
|
print("bytecount has to be hexadecimal without prefix.")
|
|
return 1
|
|
fd = os.open(args.device, os.O_RDWR)
|
|
sendcmd(fd, 'mptr {}'.format(MEMPTR).encode())
|
|
os.read(fd, 9)
|
|
while bytecount > 0:
|
|
toread = min(bytecount, 0x100)
|
|
sendcmd(fd, 'load {:x}'.format(toread & 0xff).encode())
|
|
os.read(fd, 5)
|
|
sendcmd(fd, 'peek {:x}'.format(toread & 0xff).encode())
|
|
peek = b''
|
|
while len(peek) < toread * 2:
|
|
peek += os.read(fd, 1)
|
|
time.sleep(0.0001)
|
|
os.read(fd, 5)
|
|
while peek:
|
|
c = peek[:2]
|
|
sys.stdout.buffer.write(bytes([int(c, 16)]))
|
|
peek = peek[2:]
|
|
bytecount -= toread
|
|
os.close(fd)
|
|
return 0
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|
|
|