Subversion Repositories freemyipod

Rev

Rev 499 | Show entire file | Ignore whitespace | Details | Blame | Last modification | View Log | RSS feed

Rev 499 Rev 500
Line 1103... Line 1103...
1103 
 
 1103 
 
1104 
 
 1104 
 
1105 
if __name__ == "__main__":
 1105 
if __name__ == "__main__":
1106 
    from misc import Logger
 1106 
    from misc import Logger
1107 
    logger = Logger()
 1107 
    logger = Logger()
1108 
    if sys.argv[1] == "test":
 - 
 
1109 
        # Some tests
 - 
 
1110 
        import sys
 - 
 
1111 
        emcore = Emcore()
 - 
 
1112 
        resp = emcore.getversioninfo()
 - 
 
1113 
        logger.log("Emcore device version information: " + libemcoredata.swtypes[resp.swtypeid] + " v" + str(resp.majorv) + "." + str(resp.minorv) +
- 
 
1114 
                         "." + str(resp.patchv) + " r" + str(resp.revision) + " running on " + libemcoredata.hwtypes[resp.hwtypeid] + "\n")
 - 
 
1115 
        resp = emcore.getusermemrange()
 - 
 
1116 
        logger.log("Usermemrange: "+hex(resp.lower)+" - "+hex(resp.upper)+"\n")
 - 
 
1117 
        memaddr = resp.lower
 - 
 
1118 
        maxlen = resp.upper - resp.lower
 - 
 
1119 
        f = open("./emcore.py", "rb")
 - 
 
1120 
        logger.log("Loading test file (emcore.py) to send over USB...\n")
 - 
 
1121 
        datastr = f.read()[:maxlen]
 - 
 
1122 
        logger.log("Sending data...\n")
 - 
 
1123 
        emcore.write(memaddr, datastr)
 - 
 
1124 
        logger.log("Encrypting data with the hardware key...\n")
 - 
 
1125 
        emcore.aesencrypt(memaddr, len(datastr), 0)
 - 
 
1126 
        logger.log("Reading data back and saving it to 'libemcore-test-encrypted.bin'...\n")
 - 
 
1127 
        f = open("./libemcore-test-encrypted.bin", "wb")
 - 
 
1128 
        f.write(emcore.read(memaddr, len(datastr)))
 - 
 
1129 
        logger.log("Decrypting the data again...\n")
 - 
 
1130 
        emcore.aesdecrypt(memaddr, len(datastr), 0)
 - 
 
1131 
        logger.log("Reading data back from device...\n")
 - 
 
1132 
        readdata = emcore.read(memaddr, len(datastr))
 - 
 
1133 
        if readdata == datastr:
 - 
 
1134 
            logger.log("Data matches!")
 - 
 
1135 
        else:
 - 
 
1136 
            logger.log("Data does NOT match. Something went wrong")
 - 
 
1137
1108
1138 
    elif sys.argv[1] == "gendoc":
 1109 
    if sys.argv[1] == "gendoc":
1139 
        # Generates Documentation
 1110 
        # Generates Documentation
1140 
        from misc import gendoc
 1111 
        from misc import gendoc
1141 
        logger.log("Generating documentation\n")
 1112 
        logger.log("Generating documentation\n")
1142 
        cmddict = {}
 1113 
        cmddict = {}
1143 
        for attr, value in Emcore.__dict__.iteritems():
 1114 
        for attr, value in Emcore.__dict__.iteritems():