Subversion Repositories freemyipod

#!/usr/bin/env python
#
#
#    Copyright 2010 TheSeven, benedikt93
#
#
#    This file is part of emBIOS.
#
#    emBIOS 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 2 of the
#    License, or (at your option) any later version.
#
#    emBIOS 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 emBIOS.  If not, see <http://www.gnu.org/licenses/>.
#
#

# note: handles commands 1 to 20

import sys
import time
import libembios
import struct


def usage():
  print ""
  print "Please provide a command and (if needed) parameters as command line arguments"
  print ""
  print "Available commands:"
  print ""
  print "  getinfo <infotype>"
  print "    Get info on the running emBIOS."
  print "    <infotype> may be either off 'version', 'packetsize', 'usermemrange'."
  print ""
  print "  reset <force>"
  print "    Resets the device"
  print "    If <force> is 1, the reset will be forced, otherwise it will be gracefully,"
  print "    which may take some time."
  print ""
  print "  poweroff <force>"
  print "    Powers the device off"
  print "    If <force> is 1, the poweroff will be forced, otherwise it will be gracefully,"
  print "    which may take some time."
  print ""
  print ""
  print "  uploadfile <offset> <file> <usedma> <freezescheduler>"
  print "    Uploads a file to the iPod"
  print "      <offset>: the address to upload the file to"
  print "      <file>: the path to the file"
  print "      <usedma>: if 0, DMA will not be used when uploading the file,"
  print "          otherwise it will be used. It can be omitted, default is then 1"
  print "      <freezescheduler>: if not 0, the scheduler will be frozen during DMA access"
  print "          to prevent non-consistent data after the transfer."
  print "          It can be omitted, default is then 0"
  print ""
  print "  downloadfile <offset> <size> <file> <usedma> <freezescheduler>"
  print "    Uploads a file to the iPod"
  print "      <offset>: the address to upload the file to"
  print "      <size>: the number of bytes to be read"
  print "      <file>: the path to the file"
  print "      <usedma>: if 0, DMA will not be used when downloading the file,"
  print "          otherwise it will be used. It can be omitted, default is then 1"
  print "      <freezescheduler>: if not 0, the scheduler will be frozen during DMA access"
  print "          to prevent non-consistent data after the transfer"
  print "          It can be omitted, default is then 0"
  print ""
  print "  uploadint <offset> <data>"
  print "    Uploads a single integer to the iPod"
  print "      <offset>: the address to upload the integer to"
  print "      <data>: the integer to upload"
  print ""
  print "  downloadint <offset>"
  print "    Downloads a single integer from the iPod and prints it to the console window"
  print "      <offset>: the address to download the integer from"
  print ""
  print ""
  print "  i2crecv <bus> <slave> <addr> <size>"
  print "    Reads data from an I2C device"
  print "      <bus> the bus index"
  print "      <slave> the slave address"
  print "      <addr> the start address on the I2C device"
  print "      <size> the number of bytes to read"
  print ""
  print "  i2csend <bus> <slave> <addr> <db1> <db2> ... <dbN>"
  print "    Writes data to an I2C device"
  print "      <bus> the bus index"
  print "      <slave> the slave address"
  print "      <addr> the start address on the I2C device"
  print "      <db1> ... <dbN> the data in single bytes, seperated by whitespaces,"
  print "                      eg. 0x37 0x56 0x45 0x12"
  print ""
  print ""
  print "  readusbconsole <size> <outtype> <file>"
  print "    Reads data from the USB console."
  print "      <size>: the number of bytes to read"
  print "      <outtype>: defines how to output the result:"
  print "        'file': writes the result to file <file>"
  print "        'printstring': writes the result as string to the console window"
  print "        'printhex': writes the result in hexedit notation to the console window"
  print "      <file>: the file to write the result to, can be omitted"
  print "               if <outtype> is not 'file'"
  print ""
  print "  writeusbconsole file <file> <offset> <length>"
  print "    Writes the file <file> to the USB console."
  print "      Optional params <offset> <length>: specify the range in <file> to write"
  print "  writeusbconsole direct <i1> <i2> ... <iN>"
  print "    Writes the integers <i1> ... <iN> to the USB console."
  print "  writeusbconsole string <str>"
  print "    Writes the string <str> to the USB console."
  print ""
  print "  readdevconsole <bitmask> <size> <outtype> <file>"
  print "    Reads data from one or more of the device's consoles."
  print "      <bitmask>: the bitmask of the consoles to read from"
  print "      <size>: the number of bytes to read"
  print "      <outtype>: defines how to output the result:"
  print "        'file': writes the result to file <file>"
  print "        'printstring': writes the result as string to the console window"
  print "        'printhex': writes the result in hexedit notation to the console window"
  print "      <file>: the file to write the result to, can be omitted"
  print "               if <outtype> is not 'file'"
  print ""
  print "  writedevconsole file <bitmask> <file> <offset> <length>"
  print "    Writes the file <file> to the device consoles specified by <bitmask>"
  print "      Optional params <offset> <length>: specify the range in <file> to write"
  print "  writedevconsole direct <bitmask> <i1> <i2> ... <iN>"
  print "    Writes the integers <i1> ... <iN> to the device consoles specified"
  print "                         by <bitmask>"
  print "  writedevconsole string <bitmask> <str>"
  print "    Writes the string <str> to the device consoles specified by <bitmask>"
  print ""
  print "  flushconsolebuffers <bitmask>"
  print "    flushes one or more of the device consoles' buffers."
  print "      <bitmask>: the bitmask of the consoles to be flushed"
  print ""
  print ""
  print "  getprocessinformation / getprocinfo"
  print "    Fetches data on the currently running processes"
  print "     ATTENTION: this function will be print the information to the console window."
  print "                If several threads are running this might overflow the window,"
  print "                causing not everything to be shown."
  print ""
  print "  lockscheduler"
  print "    Locks (freezes) the scheduler"
  print ""
  print "  unlockscheduler"
  print "    Unlocks the scheduler"
  print ""
  print "  suspendthread <threadid>"
  print "    Suspends/resumes the thread with thread ID <threadid>"
  print ""
  print "  resumethread <threadid>"
  print "    Resumes the thread with thread ID <threadid>"
  print ""
  print "  killthread <threadid>"
  print "    Kills the thread with thread ID <threadid>"
  print ""
  print "  createthread <namepointer> <entrypoint> <stackpointer> <stacksize> <type> <priority> <state>"
  print "    Creates a new thread and returns its thread ID"
  print "      <namepointer> a pointer to the thread's name"
  print "      <entrypoint> a pointer to the entrypoint of the thread"
  print "      <stackpointer> a pointer to the stack of the thread"
  print "      <stacksize> the size of the thread's stack"
  print "      <type> the thread type, vaild are: 0 => user thread, 1 => system thread"
  print "      <priority> the priority of the thread, from 1 to 255"
  print "      <state> the thread's initial state, valid are: 1 => ready, 0 => suspended"
  print ""
  print "  execimage <offset>"
  print "    Executes the emBIOS executable image at <offset>."
  print ""
  print "  execfirmware <offset>"
  print "    Executes the firmware image at the specified address <offset>."
  print ""
  print ""
  print "  readrawbootflash <addr_bootflsh> <addr_mem> <size>"
  print "    Reads <size> bytes from bootflash to memory."
  print "      <addr_bootflsh>: the address in bootflash to read from"
  print "      <addr_mem>: the address in memory to copy the data to"
  print ""
  print "  writerawbootflash <addr_mem> <addr_bootflsh> <size>"
  print "    Writes <size> bytes from memory to bootflash."
  print "    Don't call this unless you really know what you're doing."
  print "      <addr_mem>: the address in memory to copy the data from"
  print "      <addr_bootflsh>: the address in bootflash to write to"
  print ""
  print ""
  print "  flushcaches"
  print "    Flushes the CPUs data and instruction caches."
  print ""
  print "All numbers are hexadecimal!"
  exit(2)


def parsecommand(dev, argv):
  if len(argv) < 2: usage()

  elif argv[1] == "getinfo":
    if len(argv) != 3: usage()
    dev.getinfo(argv[2])
  
  elif argv[1] == "reset":
    if len(argv) != 3: usage()
    dev.reset(int(argv[2]), 16)
  
  elif argv[1] == "poweroff":
    if len(argv) != 3: usage()
    dev.poweroff(int(argv[2]), 16)

    
  elif argv[1] == "uploadfile":
    if len(argv) < 4 or len(argv) > 6: usage()
    if len(argv) > 4:
      usedma = int(argv[4], 16)
      if len(argv) > 5:
        freezesched = int(argv[5], 16)
      else:
        freezesched = 0
    else:
      freezesched = 0
      usedma = 1
    dev.uploadfile(int(argv[2], 16), argv[3], usedma, freezesched)
  
  elif argv[1] == "downloadfile":
    if len(argv) < 5 or len(argv) > 7: usage()
    if len(argv) > 5:
      usedma = int(argv[5], 16)
      if len(argv) > 6:
        freezesched = int(argv[6], 16)
      else:
        freezesched = 0
    else:
      freezesched = 0
      usedma = 1
    dev.downloadfile(int(argv[2], 16), int(argv[3], 16), argv[4], usedma, freezesched)
  
  elif argv[1] == "uploadint":
    if len(argv) != 4: usage()
    dev.uploadint(int(argv[2], 16), int(argv[3], 16))
  
  elif argv[1] == "downloadint":
    if len(argv) != 3: usage()
    dev.downloadint(int(argv[2], 16))
  
  
  elif argv[1] == "i2cread":
    if len(argv) != 6: usage()
    dev.i2crecv(int(argv[2], 16), int(argv[3], 16), int(argv[4], 16), int(argv[5], 16))
    
  elif argv[1] == "i2csend":
    if len(argv) < 6: usage()
    data = ""
    ptr = 5
    while ptr < len(argv):
      data += struct.pack("<B", int(argv[ptr], 16))
      ptr += 1
    dev.i2csend(int(argv[2], 16), int(argv[3], 16), int(argv[4], 16), data)
  
  
  elif argv[1] == "readusbconsole":
    if len(argv) not in [4, 5]: usage()
    if len(argv) == 4: argv[4] = ""
    dev.readusbcon(int(argv[2], 16), argv[3], argv[4])
  
  elif argv[1] == "writeusbconsole":
    if len(argv) < 4: usage()
    
    if argv[2] == "file":
      f = open(argv[3], "rb")
      data = f.read()
      
      if len(argv) > 4:
        offset = int(argv[4], 16)
      else:
        offset = 0
      if len(argv) > 5:
        size = int(argv[5], 16)
      else:
        size = len(data)
      if len(argv) > 6: usage()
      
      dev.writeusbcon(data, 0, offset, size)
      
    if argv[2] == "direct":
      data = ""
      ptr = 3
      while ptr < len(argv):
        data += struct.pack("<I", int(argv[ptr], 16))
        ptr += 1
      dev.writeusbcon(data)
      
    if argv[2] == "string":
      if len(argv) > 4: usage()
      dev.writeusbcon(argv[3])
    
    else: usage()
  
  elif argv[1] == "readdevconsole":
    if len(argv) not in [5, 6]: usage()
    if len(argv) == 5: argv[5] = ""
    dev.readusbcon(int(argv[2], 16), int(argv[3], 16), argv[4], argv[5])
  
  elif argv[1] == "writedevconsole":
    if len(argv) < 5: usage()
    
    if argv[2] == "file":
      f = open(argv[4], "rb")
      data = f.read()
      
      if len(argv) > 5:
        offset = int(argv[5], 16)
      else:
        offset = 0
      if len(argv) > 6:
        size = int(argv[6], 16)
      else:
        size = len(data)
      if len(argv) > 7: usage()
      
      dev.writeusbcon(int(argv[3], 16), data, 0, offset, size)
      
    if argv[2] == "direct":
      data = ""
      ptr = 4
      while ptr < len(argv):
        data += struct.pack("<I", int(argv[ptr], 16))
        ptr += 1
      dev.writeusbcon(int(argv[3], 16), data)
      
    if argv[2] == "string":
      if len(argv) > 5: usage()
      dev.writedevcon(int(argv[3], 16), argv[4])
    
    else: usage()
  
  elif argv[1] == "flushconsolebuffers":
    if len(argv) != 3: usage()
    dev.flushconsolebuffers(int(argv[2], 16))
   
   
  elif argv[1] == "getprocessinformation" or argv[1] == "getprocinfo":
    if len(argv) != 2: usage()
    dev.getprocinfo()
    
  elif argv[1] == "lockscheduler":
    if len(argv) != 2: usage()
    dev.freezescheduler(1)
    
  elif argv[1] == "unlockscheduler":
    if len(argv) != 2: usage()
    dev.freezescheduler(0)
    
  elif argv[1] == "suspendthread":
    if len(argv) != 3: usage()
    dev.suspendthread(1, int(argv[2], 16))
    
  elif argv[1] == "resumethread":
    if len(argv) != 3: usage()
    dev.suspendthread(0, int(argv[2], 16))
    
  elif argv[1] == "killthread":
    if len(argv) != 3: usage()
    dev.killthread(int(argv[2], 16))
    
  elif argv[1] == "createthread":
    if len(argv) != 9: usage()
    dev.createthread(int(argv[2], 16), int(argv[3], 16), int(argv[4], 16), int(argv[5], 16), int(argv[6], 16), int(argv[7], 16), int(argv[8], 16))
    
  elif argv[1] == "execimage":
    if len(argv) != 3: usage()
    dev.execimage(int(argv[2], 16))
    
  elif argv[1] == "execfirmware":
    if len(argv) != 3: usage()
    dev.execfirmware(int(argv[2], 16))
    
  elif argv[1] == "readrawbootflash":
    if len(argv) != 5: usage()
    dev.readrawbootflash(int(argv[2], 16), int(argv[3], 16), int(argv[4], 16))
    
  elif argv[1] == "writerawbootflash":
    if len(argv) != 5: usage()
    dev.writerawbootflash(int(argv[2], 16), int(argv[3], 16), int(argv[4], 16))
 
 
  elif argv[1] == "flushcaches":
    if len(argv) != 2: usage()
    dev.flushcaches()
    
  else: usage()

dev = libembios.embios()
parsecommand(dev, sys.argv)