rbutil/rbutilqt/base/bootloaderinstallams.cpp

   1 /***************************************************************************
   2  *             __________               __   ___.
   3  *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
   4  *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
   5  *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
   6  *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
   7  *                     \/            \/     \/    \/            \/
   8  *
   9  *   Copyright (C) 2008 by Dominik Wenger
  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 "bootloaderinstallams.h"
  23
  24 #include "../mkamsboot/mkamsboot.h"
  25
  26 BootloaderInstallAms::BootloaderInstallAms(QObject *parent)
  27         : BootloaderInstallBase(parent)
  28 {
  29 }
  30
  31 QString BootloaderInstallAms::ofHint()
  32 {
  33     return tr("Bootloader installation requires you to provide "
  34               "a firmware file of the original firmware (bin file). "
  35               "You need to download this file yourself due to legal "
  36               "reasons. Please refer to the "
  37               "<a href='http://www.rockbox.org/manual.shtml'>manual</a> and the "
  38               "<a href='http://www.rockbox.org/wiki/SansaAMS'>SansaAMS</a> "
  39               "wiki page on how to obtain this file.<br/>"
  40               "Press Ok to continue and browse your computer for the firmware "
  41               "file.");
  42 }
  43
  44 bool BootloaderInstallAms::install(void)
  45 {
  46     if(m_offile.isEmpty())
  47         return false;
  48
  49     qDebug() << "[BootloaderInstallAms] installing bootloader";
  50
  51     // download firmware from server
  52     emit logItem(tr("Downloading bootloader file"), LOGINFO);
  53
  54     connect(this, SIGNAL(downloadDone()), this, SLOT(installStage2()));
  55     downloadBlStart(m_blurl);
  56
  57     return true;
  58 }
  59
  60 void BootloaderInstallAms::installStage2(void)
  61 {
  62     qDebug() << "[BootloaderInstallAms] installStage2";
  63
  64     unsigned char* buf;
  65     unsigned char* of_packed;
  66     int of_packedsize;
  67     unsigned char* rb_packed;
  68     int rb_packedsize;
  69     off_t len;
  70     struct md5sums sum;
  71     char md5sum[33]; /* 32 hex digits, plus terminating zero */
  72     int n;
  73     int firmware_size;
  74     int bootloader_size;
  75     int totalsize;
  76     char errstr[200];
  77
  78     sum.md5 = md5sum;
  79
  80     m_tempfile.open();
  81     QString bootfile = m_tempfile.fileName();
  82     m_tempfile.close();
  83
  84     /* Load original firmware file */
  85     buf = load_of_file(m_offile.toLocal8Bit().data(), &len,&sum,&firmware_size,
  86                         &of_packed,&of_packedsize,errstr,sizeof(errstr));
  87     if (buf == NULL)
  88     {
  89         qDebug() << "[BootloaderInstallAms] could not load OF: " << m_offile;
  90         emit logItem(errstr, LOGERROR);
  91         emit logItem(tr("Could not load %1").arg(m_offile), LOGERROR);
  92         emit done(true);
  93         return;
  94     }
  95
  96     /* Load bootloader file */
  97     rb_packed = load_rockbox_file(bootfile.toLocal8Bit().data(), sum.model,
  98                                   &bootloader_size,&rb_packedsize,
  99                                     errstr,sizeof(errstr));
 100     if (rb_packed == NULL)
 101     {
 102         qDebug() << "[BootloaderInstallAms] could not load bootloader: " << bootfile;
 103         emit logItem(errstr, LOGERROR);
 104         emit logItem(tr("Could not load %1").arg(bootfile), LOGERROR);
 105         free(buf);
 106         free(of_packed);
 107         emit done(true);
 108         return;
 109     }
 110
 111     /* check total size */
 112     totalsize = total_size(sum.model,rb_packedsize,of_packedsize);
 113     if (totalsize > firmware_size)
 114     {
 115         qDebug() << "[BootloaderInstallAms] No room to insert bootloader";
 116         emit logItem(tr("No room to insert bootloader, try another firmware version"),
 117                      LOGERROR);
 118         free(buf);
 119         free(of_packed);
 120         free(rb_packed);
 121         emit done(true);
 122         return;
 123     }
 124
 125     /* patch the firmware */
 126     emit logItem(tr("Patching Firmware..."), LOGINFO);
 127
 128     patch_firmware(sum.model,firmware_revision(sum.model),firmware_size,buf,
 129                    len,of_packed,of_packedsize,rb_packed,rb_packedsize);
 130
 131     /* write out file */
 132     QFile out(m_blfile);
 133
 134     if(!out.open(QIODevice::WriteOnly | QIODevice::Truncate))
 135     {
 136         qDebug() << "[BootloaderInstallAms] Could not open" <<  m_blfile << "for writing";
 137         emit logItem(tr("Could not open %1 for writing").arg(m_blfile),LOGERROR);
 138         free(buf);
 139         free(of_packed);
 140         free(rb_packed);
 141         emit done(true);
 142         return;
 143     }
 144
 145     n = out.write((char*)buf, len);
 146
 147     if (n != len)
 148     {
 149         qDebug() << "[BootloaderInstallAms] Could not write firmware file";
 150         emit logItem(tr("Could not write firmware file"),LOGERROR);
 151         free(buf);
 152         free(of_packed);
 153         free(rb_packed);
 154         emit done(true);
 155         return;
 156     }
 157
 158     out.close();
 159
 160     free(buf);
 161     free(of_packed);
 162     free(rb_packed);
 163
 164     //end of install
 165     qDebug() << "[BootloaderInstallAms] install successfull";
 166     emit logItem(tr("Success: modified firmware file created"), LOGINFO);
 167     logInstall(LogAdd);
 168     emit done(false);
 169     return;
 170 }
 171
 172 bool BootloaderInstallAms::uninstall(void)
 173 {
 174     emit logItem(tr("To uninstall, perform a normal upgrade with an unmodified "
 175                     "original firmware"), LOGINFO);
 176     logInstall(LogRemove);
 177     return false;
 178 }
 179
 180 BootloaderInstallBase::BootloaderType BootloaderInstallAms::installed(void)
 181 {
 182     return BootloaderUnknown;
 183 }
 184
 185 BootloaderInstallBase::Capabilities BootloaderInstallAms::capabilities(void)
 186 {
 187     return (Install | NeedsOf);
 188 }
 189