Subversion Repositories freemyipod

#!/usr/bin/env python

#

#

#    Copyright 2010 TheSeven

#

#

#    This file is part of TheSeven's iPod tools.

#

#    TheSeven's iBugger 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.

#

#    TheSeven's iBugger 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 TheSeven's iPod tools.  If not, see <http://www.gnu.org/licenses/>.

#

#

import sys

import struct

import usb.core

class ipoddfu:

  def __init__(self, generation = 0, type = 0):

    try:

      self.dev = usb.core.find(idVendor=0x05ac, idProduct=0x1220)

      if self.dev and generation in [0, 2] and type in [0, 1]:

        self.dev.set_configuration(1)

        self.generation = 2;

        self.type = 1;

        print("Connected to S5L8701 Bootrom DFU mode, USB version %s"  % self.dev.bcdDevice)

        return

    except usb.core.USBError: pass

    try:

      self.dev = usb.core.find(idVendor=0x05ac, idProduct=0x1240)

      if self.dev and generation in [0, 2] and type in [0, 2]:

        self.dev.set_configuration(1)

        self.generation = 2;

        self.type = 2;

        print("Connected to iPod Nano 2G NOR DFU mode, USB version %s"  % self.dev.bcdDevice)

        return

    except usb.core.USBError: pass

    try:

      self.dev = usb.core.find(idVendor=0x05ac, idProduct=0x1223)

      if self.dev and generation in [0, 3] and type in [0, 1]:

        self.dev.set_configuration(1)

        self.generation = 3;

        self.type = 1;

        print("Connected to S5L8702 Bootrom DFU mode, USB version %s"  % self.dev.bcdDevice)

        return

    except usb.core.USBError: pass

    try:

      self.dev = usb.core.find(idVendor=0x05ac, idProduct=0x1242)

      if self.dev and generation in [0, 3] and type in [0, 2]:

        self.dev.set_configuration(1)

        self.generation = 3;

        self.type = 2;

        print("Connected to iPod Nano 3G WTF mode, USB version %s"  % self.dev.bcdDevice)

        return

    except usb.core.USBError: pass

    try:

      self.dev = usb.core.find(idVendor=0x05ac, idProduct=0x1225)

      if self.dev and generation in [0, 4] and type in [0, 1]:

        self.dev.set_configuration(1)

        self.generation = 4;

        self.type = 1;

        print("Connected to S5L8720 Bootrom DFU mode, USB version %s"  % self.dev.bcdDevice)

        return

    except usb.core.USBError: pass

    try:

      self.dev = usb.core.find(idVendor=0x05ac, idProduct=0x1243)

      if self.dev and generation in [0, 4] and type in [0, 2]:

        self.dev.set_configuration(1)

        self.generation = 4;

        self.type = 2;

        print("Connected to iPod Nano 4G WTF mode, USB version %s"  % self.dev.bcdDevice)

        return

    except usb.core.USBError: pass

    raise Exception("Could not find specified DFU device (generation = %d, type = %d)" % (generation, type))

  @staticmethod

  def crc32(data):

    crc_table = []

    for i in range(256):

      t = i;

      for j in range(8):

        if t & 1:

          t = (t >> 1) ^ 0xedb88320

        else:

          t = t >> 1

      crc_table.append(t)

    crc = 0xffffffff

    for i in range(len(data)):

      crc = (crc >> 8) ^ crc_table[(crc ^ data[i]) & 0xff];

    return crc

  def getcpu(self):

    result = self.handle.controlMsg(0xa1, 0xff, 0x3f, 2, 0, 100)

    return struct.pack("%dB" % len(result), *result)

  def upload(self, data, exploit = 0):

    if exploit == 1 and self.generation == 2 and self.type == 1:

      data = f.read().ljust(0x200f0, b"\0") \

           + b"\xb8\x48\x02\x22\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0" \

           + b"\0\0\0\x22\0\0\0\x22\0\0\0\x22\0\0\0\x22" \

           + b"\0\0\0\x22\0\0\0\x22\0\0\0\x22\0\0\0\x22" \

           + b"\0\0\0\x22\0\0\0\x22\0\0\0\x22\0\0\0\x22" \

           + b"\0\0\0\x22\0\0\0\x22\0\0\0\x22\0\0\0\x22" \

           + b"\0\0\0\x22\0\0\0\x22\0\0\0\x22\0\0\0\x22" \

           + b"\0\0\0\x22\0\0\0\x22\0\0\0\x22\0\0\0\x22" \

           + b"\0\0\0\x22\0\0\0\x22\0\0\0\x22\0\0\0\x22" \

           + b"\0\0\0\x22\0\0\0\x22\0\0\0\x22\0\0\0\x22"

    data = data + struct.pack("<I", self.crc32(data))

    sys.stdout.write("Upload: .")

    sys.stdout.flush()

    for index in range((len(data) + 2047) // 2048):

      self.dev.ctrl_transfer(0x21, 1, index, 0, data[2048 * index : 2048 * (index + 1)],   100)

      result = (0, 0, 0, 0, 0, 0)

      while result[4] != 0x05:

        result = self.dev.ctrl_transfer(0xa1, 3, 0, 0, 6, 100)

      sys.stdout.write(".")

      sys.stdout.flush()

    self.dev.ctrl_transfer(0x21, 1, index, 0, "",  100)

    result = (0, 0, 0, 0, 0, 0)

    index = 0

    try:

      while result[4] != 0x02 and index < 1000:

        result = self.dev.ctrl_transfer(0xa1, 3, 0, 0, 6, 100)

        index = index + 1

    except:

      pass

    if (exploit == 0 and (index == 1000 or result[4] == 0x02)) or \

       (exploit == 1 and (index == 1000 or result[4] != 0x04)):

      print(" failed: %X / %X" % (result[4], result[0]))

      raise Exception("DFU upload failed! (%X / %X)" % (result[4], result[0]))

    else:

      print(" done")

  def uploadfile(self, file, exploit = 0):

    f = open(file, "rb")

    data = f.read()

    f.close()

    self.upload(data, exploit)