rbutil/rbutilqt/base/bootloaderinstallipod.cpp

   1 /***************************************************************************
   2  *             __________               __   ___.
   3  *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
   4  *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
   5  *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
   6  *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
   7  *                     \/            \/     \/    \/            \/
   8  *
   9  *   Copyright (C) 2008 by Dominik Riebeling
  10  *
  11  * All files in this archive are subject to the GNU General Public License.
  12  * See the file COPYING in the source tree root for full license agreement.
  13  *
  14  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  15  * KIND, either express or implied.
  16  *
  17  ****************************************************************************/
  18
  19 #include <QtCore>
  20 #include "bootloaderinstallbase.h"
  21 #include "bootloaderinstallipod.h"
  22
  23 #include "../ipodpatcher/ipodpatcher.h"
  24 #include "utils.h"
  25
  26
  27 BootloaderInstallIpod::BootloaderInstallIpod(QObject *parent)
  28         : BootloaderInstallBase(parent)
  29 {
  30     (void)parent;
  31     // initialize sector buffer. ipod_sectorbuf is defined in ipodpatcher.
  32     // The buffer itself is only present once, so make sure to not allocate
  33     // it if it was already allocated. The application needs to take care
  34     // no concurrent (i.e. multiple objects of this class running) requests
  35     // are done.
  36     if(ipod_sectorbuf == NULL)
  37         ipod_alloc_buffer(&ipod_sectorbuf, BUFFER_SIZE);
  38 }
  39
  40
  41 BootloaderInstallIpod::~BootloaderInstallIpod()
  42 {
  43     if(ipod_sectorbuf) {
  44         free(ipod_sectorbuf);
  45         ipod_sectorbuf = NULL;
  46     }
  47 }
  48
  49
  50 bool BootloaderInstallIpod::install(void)
  51 {
  52     if(ipod_sectorbuf == NULL) {
  53         emit logItem(tr("Error: can't allocate buffer memory!"), LOGERROR);
  54         emit done(true);
  55         return false;
  56     }
  57     memset(&ipod, 0, sizeof(struct ipod_t));
  58
  59     if(!ipodInitialize(&ipod)) {
  60         emit done(true);
  61         return false;
  62     }
  63
  64     if(ipod.nimages <= 0) {
  65         emit logItem(tr("Failed to read firmware directory"), LOGERROR);
  66         emit done(true);
  67         return false;
  68     }
  69     if(getmodel(&ipod,(ipod.ipod_directory[ipod.ososimage].vers>>8)) < 0) {
  70         emit logItem(tr("Unknown version number in firmware (%1)").arg(
  71                     ipod.ipod_directory[0].vers), LOGERROR);
  72         emit done(true);
  73         return false;
  74     }
  75     if(ipod.macpod) {
  76         emit logItem(tr("Warning: This is a MacPod, Rockbox only runs on WinPods. \n"
  77                     "See http://www.rockbox.org/wiki/IpodConversionToFAT32"), LOGERROR);
  78         emit done(true);
  79         return false;
  80     }
  81     emit logItem(tr("Downloading bootloader file"), LOGINFO);
  82
  83     downloadBlStart(m_blurl);
  84     connect(this, SIGNAL(downloadDone()), this, SLOT(installStage2()));
  85     return true;
  86 }
  87
  88
  89 void BootloaderInstallIpod::installStage2(void)
  90 {
  91     emit logItem(tr("Installing Rockbox bootloader"), LOGINFO);
  92     QCoreApplication::processEvents();
  93
  94     if(ipod_reopen_rw(&ipod) < 0) {
  95         emit logItem(tr("Could not open Ipod in R/W mode"), LOGERROR);
  96         emit done(true);
  97         return;
  98     }
  99     QCoreApplication::processEvents();
 100
 101     m_tempfile.open();
 102     QString blfile = m_tempfile.fileName();
 103     m_tempfile.close();
 104     if(add_bootloader(&ipod, blfile.toLatin1().data(), FILETYPE_DOT_IPOD) == 0) {
 105         emit logItem(tr("Successfull added bootloader"), LOGOK);
 106         ipod_close(&ipod);
 107 #if defined(Q_OS_MACX)
 108         m_remountDevice = ipod.diskname;
 109         connect(this, SIGNAL(remounted(bool)), this, SLOT(installStage3(bool)));
 110         waitRemount();
 111 #else
 112         installStage3(true);
 113 #endif
 114     }
 115     else {
 116         emit logItem(tr("Failed to add bootloader"), LOGERROR);
 117         ipod_close(&ipod);
 118         emit done(true);
 119         return;
 120     }
 121 }
 122
 123
 124 void BootloaderInstallIpod::installStage3(bool mounted)
 125 {
 126     if(mounted) {
 127         logInstall(LogAdd);
 128         emit logItem(tr("Bootloader Installation complete."), LOGINFO);
 129         emit done(false);
 130         return;
 131     }
 132     else {
 133         emit logItem(tr("Writing log aborted"), LOGERROR);
 134         emit done(true);
 135     }
 136     qDebug() << "[BootloaderInstallIpod] version installed:" << m_blversion.toString(Qt::ISODate);
 137 }
 138
 139
 140 bool BootloaderInstallIpod::uninstall(void)
 141 {
 142     struct ipod_t ipod;
 143     emit logItem(tr("Uninstalling bootloader"), LOGINFO);
 144     QCoreApplication::processEvents();
 145
 146     if(!ipodInitialize(&ipod)) {
 147         emit done(true);
 148         return false;
 149     }
 150
 151     if (ipod.nimages <= 0) {
 152         emit logItem(tr("Failed to read firmware directory"),LOGERROR);
 153         emit done(true);
 154         return false;
 155     }
 156     if (getmodel(&ipod,(ipod.ipod_directory[0].vers>>8)) < 0) {
 157         emit logItem(tr("Unknown version number in firmware (%1)").arg(
 158                     ipod.ipod_directory[0].vers), LOGERROR);
 159         emit done(true);
 160         return false;
 161     }
 162
 163     if (ipod_reopen_rw(&ipod) < 0) {
 164         emit logItem(tr("Could not open Ipod in R/W mode"), LOGERROR);
 165         emit done(true);
 166         return false;
 167     }
 168
 169     if (ipod.ipod_directory[0].entryOffset == 0) {
 170         emit logItem(tr("No bootloader detected."), LOGERROR);
 171         emit done(true);
 172         return false;
 173     }
 174
 175     if (delete_bootloader(&ipod)==0) {
 176         emit logItem(tr("Successfully removed bootloader"), LOGOK);
 177         logInstall(LogRemove);
 178         emit done(false);
 179         ipod_close(&ipod);
 180         return true;
 181     }
 182     else {
 183         emit logItem(tr("Removing bootloader failed."), LOGERROR);
 184         emit done(true);
 185         ipod_close(&ipod);
 186         return false;
 187     }
 188 }
 189
 190
 191 BootloaderInstallBase::BootloaderType BootloaderInstallIpod::installed(void)
 192 {
 193     struct ipod_t ipod;
 194     BootloaderInstallBase::BootloaderType result = BootloaderRockbox;
 195
 196     if(!ipodInitialize(&ipod)) {
 197         qDebug() << "[BootloaderInstallIpod] installed: BootloaderUnknown";
 198         result = BootloaderUnknown;
 199     }
 200     else {
 201         read_directory(&ipod);
 202         if(ipod.ipod_directory[0].entryOffset == 0 || ipod.macpod) {
 203             qDebug() << "[BootloaderInstallIpod] installed: BootloaderOther";
 204             result = BootloaderOther;
 205         }
 206         else {
 207             qDebug() << "[BootloaderInstallIpod] installed: BootloaderRockbox";
 208         }
 209     }
 210     ipod_close(&ipod);
 211
 212     return result;
 213 }
 214
 215
 216 BootloaderInstallBase::Capabilities BootloaderInstallIpod::capabilities(void)
 217 {
 218     return (Install | Uninstall | IsRaw);
 219 }
 220
 221
 222 /** @initialize Ipod by opening its file handle and checking if its an ipod.
 223  *  Note: the caller has to make sure the file handle gets closed!
 224  */
 225 bool BootloaderInstallIpod::ipodInitialize(struct ipod_t *ipod)
 226 {
 227     if(!m_blfile.isEmpty()) {
 228         QString devicename = Utils::resolveDevicename(m_blfile);
 229         if(devicename.isEmpty()) {
 230             emit logItem(tr("Error: could not retrieve device name"), LOGERROR);
 231             return false;
 232         }
 233 #if defined(Q_OS_WIN32)
 234         sprintf(ipod->diskname, "\\\\.\\PhysicalDrive%i", devicename.toInt());
 235 #elif defined(Q_OS_MACX)
 236         sprintf(ipod->diskname, "%s",
 237             qPrintable(devicename.remove(QRegExp("s[0-9]+$"))));
 238 #else
 239         sprintf(ipod->diskname, "%s",
 240             qPrintable(devicename.remove(QRegExp("[0-9]+$"))));
 241 #endif
 242         qDebug() << "[BootloaderInstallIpod] ipodpatcher: overriding scan, using"
 243                  << ipod->diskname;
 244     }
 245     else {
 246         emit logItem(tr("Error: no mountpoint specified!"), LOGERROR);
 247         qDebug() << "[BootloaderInstallIpod] no mountpoint specified!";
 248     }
 249     int result = ipod_open(ipod, 1);
 250     if(result == -2) {
 251         emit logItem(tr("Could not open Ipod: permission denied"), LOGERROR);
 252         return false;
 253     }
 254     else if(result < 0) {
 255         emit logItem(tr("Could not open Ipod"), LOGERROR);
 256         return false;
 257     }
 258
 259     if(read_partinfo(ipod, 1) < 0) {
 260         emit logItem(tr("Error reading partition table - possibly not an Ipod"), LOGERROR);
 261         ipod_close(ipod);
 262         return false;
 263     }
 264
 265     if(ipod->pinfo[0].start == 0) {
 266         emit logItem(tr("No firmware partition on disk"), LOGERROR);
 267         ipod_close(ipod);
 268         return false;
 269     }
 270     read_directory(ipod);
 271     return true;
 272 }
 273