rbutil/rbutilqt/base/bootloaderinstalltcc.cpp

   1 /***************************************************************************
   2  *             __________               __   ___.
   3  *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
   4  *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
   5  *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
   6  *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
   7  *                     \/            \/     \/    \/            \/
   8  *
   9  *   Copyright (C) 2009 by Tomer Shalev
  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  * This file is a modified version of the AMS installer by Dominik Wenger
  19  *
  20  ****************************************************************************/
  21
  22 #include <QtCore>
  23 #include "bootloaderinstallbase.h"
  24 #include "bootloaderinstalltcc.h"
  25 #include "../mktccboot/mktccboot.h"
  26
  27 BootloaderInstallTcc::BootloaderInstallTcc(QObject *parent)
  28         : BootloaderInstallBase(parent)
  29 {
  30 }
  31
  32 QString BootloaderInstallTcc::ofHint()
  33 {
  34     return tr("Bootloader installation requires you to provide "
  35                "a firmware file of the original firmware (bin file). "
  36                "You need to download this file yourself due to legal "
  37                "reasons."
  38                "Press Ok to continue and browse your computer for the firmware "
  39                "file.");
  40 }
  41
  42 bool BootloaderInstallTcc::install(void)
  43 {
  44     if(m_offile.isEmpty())
  45         return false;
  46
  47     // Download firmware from server
  48     emit logItem(tr("Downloading bootloader file"), LOGINFO);
  49
  50     connect(this, SIGNAL(downloadDone()), this, SLOT(installStage2()));
  51     downloadBlStart(m_blurl);
  52
  53     return true;
  54 }
  55
  56 void BootloaderInstallTcc::installStage2(void)
  57 {
  58     unsigned char *of_buf, *boot_buf = NULL, *patched_buf = NULL;
  59     int n, of_size, boot_size, patched_size;
  60     char errstr[200];
  61     bool ret = false;
  62
  63     m_tempfile.open();
  64     QString bootfile = m_tempfile.fileName();
  65     m_tempfile.close();
  66
  67     /* Construct path for write out.
  68      * Combine path of m_blfile with filename from OF */
  69     QString outfilename = QFileInfo(m_blfile).absolutePath() + "/" +
  70         QFileInfo(m_offile).fileName();
  71
  72     /* Write out file */
  73     QFile out(outfilename);
  74
  75     /* Load original firmware file */
  76     of_buf = file_read(m_offile.toLocal8Bit().data(), &of_size);
  77     if (of_buf == NULL)
  78     {
  79         emit logItem(errstr, LOGERROR);
  80         emit logItem(tr("Could not load %1").arg(m_offile), LOGERROR);
  81         goto exit;
  82     }
  83
  84     /* Load bootloader file */
  85     boot_buf = file_read(bootfile.toLocal8Bit().data(), &boot_size);
  86     if (boot_buf == NULL)
  87     {
  88         emit logItem(errstr, LOGERROR);
  89         emit logItem(tr("Could not load %1").arg(bootfile), LOGERROR);
  90         goto exit;
  91     }
  92
  93     /* Patch the firmware */
  94     emit logItem(tr("Patching Firmware..."), LOGINFO);
  95
  96     patched_buf = patch_firmware_tcc(of_buf, of_size, boot_buf, boot_size,
  97             &patched_size);
  98     if (patched_buf == NULL)
  99     {
 100         emit logItem(errstr, LOGERROR);
 101         emit logItem(tr("Could patch firmware"), LOGERROR);
 102         goto exit;
 103     }
 104
 105     if(!out.open(QIODevice::WriteOnly | QIODevice::Truncate))
 106     {
 107         emit logItem(tr("Could not open %1 for writing").arg(m_blfile),
 108                 LOGERROR);
 109         goto exit;
 110     }
 111
 112     n = out.write((char*)patched_buf, patched_size);
 113     out.close();
 114     if (n != patched_size)
 115     {
 116         emit logItem(tr("Could not write firmware file"), LOGERROR);
 117         goto exit;
 118     }
 119
 120     /* End of install */
 121     emit logItem(tr("Success: modified firmware file created"), LOGINFO);
 122     logInstall(LogAdd);
 123
 124     ret = true;
 125
 126 exit:
 127     if (of_buf)
 128         free(of_buf);
 129
 130     if (boot_buf)
 131         free(boot_buf);
 132
 133     if (patched_buf)
 134         free(patched_buf);
 135
 136     emit done(ret);
 137 }
 138
 139 bool BootloaderInstallTcc::uninstall(void)
 140 {
 141     emit logItem("To uninstall, perform a normal upgrade with an unmodified original firmware", LOGINFO);
 142     logInstall(LogRemove);
 143     return false;
 144 }
 145
 146 BootloaderInstallBase::BootloaderType BootloaderInstallTcc::installed(void)
 147 {
 148     return BootloaderUnknown;
 149 }
 150
 151 BootloaderInstallBase::Capabilities BootloaderInstallTcc::capabilities(void)
 152 {
 153     return (Install | NeedsOf);
 154 }