1 files changed, 271 insertions, 0 deletions

diff --git a/utils/rbutilqt/base/bootloaderinstallhex.cpp b/utils/rbutilqt/base/bootloaderinstallhex.cpp
new file mode 100644
index 0000000000..b3dde0bbfa
--- /dev/null
+++ b/utils/rbutilqt/base/bootloaderinstallhex.cpp
@@ -0,0 +1,271 @@
+/***************************************************************************
+ *             __________               __   ___.
+ *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
+ *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
+ *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
+ *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
+ *                     \/            \/     \/    \/            \/
+ *
+ *   Copyright (C) 2008 by Dominik Riebeling
+ *
+ * All files in this archive are subject to the GNU General Public License.
+ * See the file COPYING in the source tree root for full license agreement.
+ *
+ * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
+ * KIND, either express or implied.
+ *
+ ****************************************************************************/
+
+#include <QtCore>
+#include "bootloaderinstallbase.h"
+#include "bootloaderinstallhex.h"
+#include "utils.h"
+#include "Logger.h"
+
+#include "../../tools/iriver.h"
+#include "../../tools/mkboot.h"
+
+struct md5s {
+    const char* orig;
+    const char* patched;
+};
+
+struct md5s md5sums[] = {
+#include "irivertools/h100sums.h"
+    { nullptr, nullptr },
+#include "irivertools/h120sums.h"
+    { nullptr, nullptr },
+#include "irivertools/h300sums.h"
+    { nullptr, nullptr }
+};
+
+
+BootloaderInstallHex::BootloaderInstallHex(QObject *parent)
+        : BootloaderInstallBase(parent)
+{
+}
+
+QString BootloaderInstallHex::ofHint()
+{
+    return tr("Bootloader installation requires you to provide "
+               "a firmware file of the original firmware (hex file). "
+               "You need to download this file yourself due to legal "
+               "reasons. Please refer to the "
+               "<a href='http://www.rockbox.org/manual.shtml'>manual</a> and the "
+               "<a href='http://www.rockbox.org/wiki/IriverBoot"
+               "#Download_and_extract_a_recent_ve'>IriverBoot</a> wiki page on "
+               "how to obtain this file.<br/>"
+               "Press Ok to continue and browse your computer for the firmware "
+               "file.");
+}
+
+bool BootloaderInstallHex::install(void)
+{
+    if(m_offile.isEmpty())
+        return false;
+    m_hashindex = -1;
+
+    // md5sum hex file
+    emit logItem(tr("checking MD5 hash of input file ..."), LOGINFO);
+    QByteArray filedata;
+    // read hex file into QByteArray
+    QFile file(m_offile);
+    file.open(QIODevice::ReadOnly);
+    filedata = file.readAll();
+    file.close();
+    QString hash = QCryptographicHash::hash(filedata,
+            QCryptographicHash::Md5).toHex();
+    LOG_INFO() << "hexfile hash:" << hash;
+    if(file.error() != QFile::NoError) {
+        emit logItem(tr("Could not verify original firmware file"), LOGERROR);
+        emit done(true);
+        return false;
+    }
+    // check hash and figure model from md5sum
+    int i = sizeof(md5sums) / sizeof(struct md5s);
+    m_model = 4;
+    // 3: h300, 2: h120, 1: h100, 0:invalid
+    while(i--) {
+        if(md5sums[i].orig == nullptr)
+            m_model--;
+        if(!qstrcmp(md5sums[i].orig, hash.toLatin1()))
+            break;
+    }
+    if(i < 0) {
+        emit logItem(tr("Firmware file not recognized."), LOGERROR);
+        return false;
+    }
+    else {
+        emit logItem(tr("MD5 hash ok"), LOGOK);
+        m_hashindex = i;
+    }
+
+    // check model agains download link.
+    QString match[] = {"", "h100", "h120", "h300"};
+    if(!m_blurl.path().contains(match[m_model])) {
+        emit logItem(tr("Firmware file doesn't match selected player."),
+        LOGERROR);
+        return false;
+    }
+
+    emit logItem(tr("Descrambling file"), LOGINFO);
+    m_descrambled.open();
+    int result;
+    result = iriver_decode(m_offile.toLatin1().data(),
+        m_descrambled.fileName().toLatin1().data(), FALSE, STRIP_NONE);
+    LOG_INFO() << "iriver_decode():" << result;
+
+    if(result < 0) {
+        emit logItem(tr("Error in descramble: %1").arg(scrambleError(result)), LOGERROR);
+        return false;
+    }
+
+    // download firmware from server
+    emit logItem(tr("Downloading bootloader file"), LOGINFO);
+    connect(this, &BootloaderInstallBase::downloadDone, this, &BootloaderInstallHex::installStage2);
+
+    downloadBlStart(m_blurl);
+    return true;
+}
+
+
+void BootloaderInstallHex::installStage2(void)
+{
+    emit logItem(tr("Adding bootloader to firmware file"), LOGINFO);
+    QCoreApplication::processEvents();
+
+    // local temp file
+    QTemporaryFile tempbin;
+    tempbin.open();
+    QString tempbinName = tempbin.fileName();
+    tempbin.close();
+    // get temporary files filenames -- external tools need this.
+    m_descrambled.open();
+    QString descrambledName = m_descrambled.fileName();
+    m_descrambled.close();
+    m_tempfile.open();
+    QString tempfileName = m_tempfile.fileName();
+    m_tempfile.close();
+
+    int origin = 0;
+    switch(m_model) {
+        case 3:
+            origin = 0x3f0000;
+            break;
+        case 2:
+        case 1:
+            origin = 0x1f0000;
+            break;
+        default:
+            origin = 0;
+            break;
+    }
+
+    // iriver decode already done in stage 1
+    int result;
+    if((result = mkboot_iriver(descrambledName.toLocal8Bit().constData(),
+                    tempfileName.toLocal8Bit().constData(),
+                    tempbinName.toLocal8Bit().constData(), origin)) < 0)
+    {
+        QString error;
+        switch(result) {
+            case -1: error = tr("could not open input file"); break;
+            case -2: error = tr("reading header failed"); break;
+            case -3: error = tr("reading firmware failed"); break;
+            case -4: error = tr("can't open bootloader file"); break;
+            case -5: error = tr("reading bootloader file failed"); break;
+            case -6: error = tr("can't open output file"); break;
+            case -7: error = tr("writing output file failed"); break;
+        }
+        emit logItem(tr("Error in patching: %1").arg(error), LOGERROR);
+
+        emit done(true);
+        return;
+    }
+    QTemporaryFile targethex;
+    targethex.open();
+    QString targethexName = targethex.fileName();
+    if((result = iriver_encode(tempbinName.toLocal8Bit().constData(),
+                    targethexName.toLocal8Bit().constData(), FALSE)) < 0)
+    {
+        emit logItem(tr("Error in scramble: %1").arg(scrambleError(result)), LOGERROR);
+        targethex.close();
+
+        emit done(true);
+        return;
+    }
+
+    // finally check the md5sum of the created file
+    QByteArray filedata;
+    filedata = targethex.readAll();
+    targethex.close();
+    QString hash = QCryptographicHash::hash(filedata,
+            QCryptographicHash::Md5).toHex();
+    LOG_INFO() << "created hexfile hash:" << hash;
+
+    emit logItem(tr("Checking modified firmware file"), LOGINFO);
+    if(hash != QString(md5sums[m_hashindex].patched)) {
+        emit logItem(tr("Error: modified file checksum wrong"), LOGERROR);
+        targethex.remove();
+        emit done(true);
+        return;
+    }
+    // finally copy file to player
+    if(!Utils::resolvePathCase(m_blfile).isEmpty()) {
+        emit logItem(tr("A firmware file is already present on player"), LOGERROR);
+        emit done(true);
+        return;
+    }
+    if(targethex.copy(m_blfile)) {
+        emit logItem(tr("Success: modified firmware file created"), LOGINFO);
+    }
+    else {
+        emit logItem(tr("Copying modified firmware file failed"), LOGERROR);
+        emit done(true);
+        return;
+    }
+
+    logInstall(LogAdd);
+    emit done(false);
+
+    return;
+}
+
+
+bool BootloaderInstallHex::uninstall(void)
+{
+    emit logItem(tr("Uninstallation not possible, only installation info removed"), LOGINFO);
+    logInstall(LogRemove);
+    emit done(true);
+    return false;
+}
+
+
+BootloaderInstallBase::BootloaderType BootloaderInstallHex::installed(void)
+{
+    return BootloaderUnknown;
+}
+
+
+BootloaderInstallBase::Capabilities BootloaderInstallHex::capabilities(void)
+{
+    return (Install | NeedsOf);
+}
+
+QString BootloaderInstallHex::scrambleError(int err)
+{
+    QString error;
+    switch(err) {
+        case -1: error = tr("Can't open input file"); break;
+        case -2: error = tr("Can't open output file"); break;
+        case -3: error = tr("invalid file: header length wrong"); break;
+        case -4: error = tr("invalid file: unrecognized header"); break;
+        case -5: error = tr("invalid file: \"length\" field wrong"); break;
+        case -6: error = tr("invalid file: \"length2\" field wrong"); break;
+        case -7: error = tr("invalid file: internal checksum error"); break;
+        case -8: error = tr("invalid file: \"length3\" field wrong"); break;
+        default: error = tr("unknown"); break;
+    }
+    return error;
+}
+