rbutil/rbutilqt/base/bootloaderinstallsansa.cpp

   1 /***************************************************************************
   2  *             __________               __   ___.
   3  *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
   4  *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
   5  *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
   6  *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
   7  *                     \/            \/     \/    \/            \/
   8  *
   9  *   Copyright (C) 2008 by Dominik Riebeling
  10  *   $Id$
  11  *
  12  * All files in this archive are subject to the GNU General Public License.
  13  * See the file COPYING in the source tree root for full license agreement.
  14  *
  15  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  16  * KIND, either express or implied.
  17  *
  18  ****************************************************************************/
  19
  20 #include <QtCore>
  21 #include "bootloaderinstallbase.h"
  22 #include "bootloaderinstallsansa.h"
  23
  24 #include "../sansapatcher/sansapatcher.h"
  25 #include "autodetection.h"
  26
  27 BootloaderInstallSansa::BootloaderInstallSansa(QObject *parent)
  28         : BootloaderInstallBase(parent)
  29 {
  30     (void)parent;
  31     // initialize sector buffer. sansa_sectorbuf is instantiated by
  32     // sansapatcher.
  33     // The buffer itself is only present once, so make sure to not allocate
  34     // it if it was already allocated. The application needs to take care
  35     // no concurrent (i.e. multiple objects of this class running) requests
  36     // are done.
  37     if(sansa_sectorbuf == NULL)
  38         sansa_alloc_buffer(&sansa_sectorbuf, BUFFER_SIZE);
  39 }
  40
  41
  42 BootloaderInstallSansa::~BootloaderInstallSansa()
  43 {
  44     if(sansa_sectorbuf) {
  45         free(sansa_sectorbuf);
  46         sansa_sectorbuf = NULL;
  47     }
  48 }
  49
  50
  51 /** Start bootloader installation.
  52  */
  53 bool BootloaderInstallSansa::install(void)
  54 {
  55     if(sansa_sectorbuf == NULL) {
  56         emit logItem(tr("Error: can't allocate buffer memory!"), LOGERROR);
  57         return false;
  58         emit done(true);
  59     }
  60
  61     emit logItem(tr("Searching for Sansa"), LOGINFO);
  62
  63     struct sansa_t sansa;
  64
  65     int n = sansa_scan(&sansa);
  66     if(n == -1) {
  67         emit logItem(tr("Permission for disc access denied!\n"
  68                 "This is required to install the bootloader"),
  69                 LOGERROR);
  70         emit done(true);
  71         return false;
  72     }
  73     if(n == 0) {
  74         emit logItem(tr("No Sansa detected!"), LOGERROR);
  75         emit done(true);
  76         return false;
  77     }
  78     if(sansa.hasoldbootloader) {
  79         emit logItem(tr("OLD ROCKBOX INSTALLATION DETECTED, ABORTING.\n"
  80                "You must reinstall the original Sansa firmware before running\n"
  81                "sansapatcher for the first time.\n"
  82                "See http://www.rockbox.org/wiki/SansaE200Install\n"),
  83                LOGERROR);
  84         emit done(true);
  85         return false;
  86     }
  87     emit logItem(tr("Downloading bootloader file"), LOGINFO);
  88
  89     downloadBlStart(m_blurl);
  90     connect(this, SIGNAL(downloadDone()), this, SLOT(installStage2()));
  91     return true;
  92 }
  93
  94
  95 /** Finish bootloader installation.
  96  */
  97 void BootloaderInstallSansa::installStage2(void)
  98 {
  99     struct sansa_t sansa;
 100
 101     emit logItem(tr("Installing Rockbox bootloader"), LOGINFO);
 102     QCoreApplication::processEvents();
 103     if(!sansaInitialize(&sansa)) {
 104             emit done(true);
 105             return;
 106     }
 107
 108     if(sansa_reopen_rw(&sansa) < 0) {
 109         emit logItem(tr("Could not open Sansa in R/W mode"), LOGERROR);
 110         emit done(true);
 111         return;
 112     }
 113
 114     // check model -- if sansapatcher reports a c200 don't install an e200
 115     // bootloader and vice versa.
 116     // The model is available in the mi4 file at offset 0x1fc and matches
 117     // the targetname set by sansapatcher.
 118     emit logItem(tr("Checking downloaded bootloader"), LOGINFO);
 119     m_tempfile.open();
 120     QString blfile = m_tempfile.fileName();
 121     char magic[4];
 122     m_tempfile.seek(0x1fc);
 123     m_tempfile.read(magic, 4);
 124     m_tempfile.close();
 125     if(memcmp(sansa.targetname, magic, 4) != 0) {
 126         emit logItem(tr("Bootloader mismatch! Aborting."), LOGERROR);
 127         qDebug("[BootloaderInstallSansa] Targetname: %s, mi4 magic: %c%c%c%c",
 128                 sansa.targetname, magic[0], magic[1], magic[2], magic[3]);
 129         emit done(true);
 130         sansa_close(&sansa);
 131         return;
 132     }
 133
 134     if(sansa_add_bootloader(&sansa, blfile.toLatin1().data(),
 135             FILETYPE_MI4) == 0) {
 136         emit logItem(tr("Successfully installed bootloader"), LOGOK);
 137         logInstall(LogAdd);
 138         emit done(false);
 139         sansa_close(&sansa);
 140         return;
 141     }
 142     else {
 143         emit logItem(tr("Failed to install bootloader"), LOGERROR);
 144         sansa_close(&sansa);
 145         emit done(true);
 146         return;
 147     }
 148
 149 }
 150
 151
 152 /** Uninstall the bootloader.
 153  */
 154 bool BootloaderInstallSansa::uninstall(void)
 155 {
 156     struct sansa_t sansa;
 157
 158     emit logItem(tr("Uninstalling bootloader"), LOGINFO);
 159     QCoreApplication::processEvents();
 160
 161     if(!sansaInitialize(&sansa)) {
 162         emit done(true);
 163         return false;
 164     }
 165
 166     if (sansa.hasoldbootloader) {
 167         emit logItem(tr("OLD ROCKBOX INSTALLATION DETECTED, ABORTING.\n"
 168                     "You must reinstall the original Sansa firmware before running\n"
 169                     "sansapatcher for the first time.\n"
 170                     "See http://www.rockbox.org/wiki/SansaE200Install\n"),
 171                     LOGERROR);
 172         emit done(true);
 173         return false;
 174     }
 175
 176     if (sansa_reopen_rw(&sansa) < 0) {
 177         emit logItem(tr("Could not open Sansa in R/W mode"), LOGERROR);
 178         emit done(true);
 179         return false;
 180     }
 181
 182     if (sansa_delete_bootloader(&sansa)==0) {
 183         emit logItem(tr("Successfully removed bootloader"), LOGOK);
 184         logInstall(LogRemove);
 185         emit done(false);
 186         sansa_close(&sansa);
 187         return true;
 188     }
 189     else {
 190         emit logItem(tr("Removing bootloader failed."),LOGERROR);
 191         emit done(true);
 192         sansa_close(&sansa);
 193         return false;
 194     }
 195
 196     return false;
 197 }
 198
 199
 200 /** Check if bootloader is already installed
 201  */
 202 BootloaderInstallBase::BootloaderType BootloaderInstallSansa::installed(void)
 203 {
 204     struct sansa_t sansa;
 205     int num;
 206
 207     if(!sansaInitialize(&sansa)) {
 208         return BootloaderUnknown;
 209     }
 210     if((num = sansa_list_images(&sansa)) == 2) {
 211         sansa_close(&sansa);
 212         return BootloaderRockbox;
 213     }
 214     else if(num == 1) {
 215         sansa_close(&sansa);
 216         return BootloaderOther;
 217     }
 218     return BootloaderUnknown;
 219
 220 }
 221
 222 bool BootloaderInstallSansa::sansaInitialize(struct sansa_t *sansa)
 223 {
 224     if(!m_blfile.isEmpty()) {
 225 #if defined(Q_OS_WIN32)
 226         sprintf(sansa->diskname, "\\\\.\\PhysicalDrive%i",
 227                 Autodetection::resolveDevicename(m_blfile).toInt());
 228 #elif defined(Q_OS_MACX)
 229         sprintf(sansa->diskname,
 230             qPrintable(Autodetection::resolveDevicename(m_blfile).remove(QRegExp("s[0-9]+$"))));
 231 #else
 232         sprintf(sansa->diskname,
 233             qPrintable(Autodetection::resolveDevicename(m_blfile).remove(QRegExp("[0-9]+$"))));
 234 #endif
 235         qDebug() << "[BootloaderInstallSansa] sansapatcher: overriding scan, using"
 236                  << sansa->diskname;
 237     }
 238     else if(sansa_scan(sansa) != 1) {
 239         emit logItem(tr("Can't find Sansa"), LOGERROR);
 240         return false;
 241     }
 242
 243     if (sansa_open(sansa, 0) < 0) {
 244         emit logItem(tr("Could not open Sansa"), LOGERROR);
 245         return false;
 246     }
 247
 248     if (sansa_read_partinfo(sansa,0) < 0) {
 249         emit logItem(tr("Could not read partition table"), LOGERROR);
 250         sansa_close(sansa);
 251         return false;
 252     }
 253
 254     int i = is_sansa(sansa);
 255     if(i < 0) {
 256         emit logItem(tr("Disk is not a Sansa (Error %1), aborting.").arg(i), LOGERROR);
 257         sansa_close(sansa);
 258         return false;
 259     }
 260     return true;
 261 }
 262
 263
 264 /** Get capabilities of subclass installer.
 265  */
 266 BootloaderInstallBase::Capabilities BootloaderInstallSansa::capabilities(void)
 267 {
 268     return (Install | Uninstall | IsRaw | CanCheckInstalled);
 269 }
 270