/***************************************************************************
 *             __________               __   ___.
 *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
 *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
 *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
 *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
 *                     \/            \/     \/    \/            \/
 * Module: rbutil
 * File: bootloaders.cpp
 *
 * Copyright (C) 2007 Dominik Wenger
 *
 * 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 "bootloaders.h"
#include "irivertools.h"
#include "md5sum.h"

#include "rbutil.h"
#include "installlog.h"


// for ipodpatcher
int verbose = 0;
// reserves memory for ipodpatcher
bool initIpodpatcher()
{
      if (ipod_alloc_buffer(&sectorbuf,BUFFER_SIZE) < 0) return true;
      else return false;
}
// uses ipodpatcher for add and rem of bootloader
bool ipodpatcher(int mode,wxString bootloadername)
{
    wxString src,dest,buf;

    struct ipod_t ipod;

    int n = ipod_scan(&ipod);
    if (n == 0)
    {
         ERR_DIALOG(wxT("[ERR]  No ipods found."), wxT("Scanning for Ipods"));
         return false;
    }
    if (n > 1)
    {
         ERR_DIALOG(wxT("[ERR]  to many ipods found."), wxT("Scanning for Ipods"));
         return false;
    }

     // downloading files
    if(mode == BOOTLOADER_ADD)
    {
        src.Printf(wxT("%s/ipod/%s.ipod"),gv->bootloader_url.c_str(),bootloadername.c_str());
        dest.Printf(wxT("%s" PATH_SEP "download" PATH_SEP "%s"),
                    gv->stdpaths->GetUserDataDir().c_str(),bootloadername.c_str());
        if ( DownloadURL(src, dest) )
        {
            wxRemoveFile(dest);
            buf.Printf(wxT("Unable to download %s"), src.c_str() );
            ERR_DIALOG(buf, wxT("Install"));
            return false;
        }
    }

    if (ipod_open(&ipod, 0) < 0)
    {
       ERR_DIALOG(wxT("[ERR] could not open ipod"), wxT("open Ipod"));
       return false;
    }

    if (read_partinfo(&ipod,0) < 0)
    {
       ERR_DIALOG(wxT("[ERR] could not read partitiontable"), wxT("reading partitiontable"));
       return false;
    }

    if (ipod.pinfo[0].start==0)
    {
       ERR_DIALOG(wxT("[ERR]  No partition 0 on disk"), wxT("reading partitiontable"));
       int i;
       double sectors_per_MB = (1024.0*1024.0)/ipod.sector_size;

       buf.Printf(wxT("[INFO] Part    Start Sector    End Sector   Size (MB)   Type\n"));
       ERR_DIALOG(buf, wxT("reading partitiontable"));
       for ( i = 0; i < 4; i++ ) {
         if (ipod.pinfo[i].start != 0) {
             buf.Printf(wxT("[INFO]    %d      %10ld    %10ld  %10.1f   %s (0x%02x)\n"),
                   i,
                   ipod.pinfo[i].start,
                   ipod.pinfo[i].start+ipod.pinfo[i].size-1,
                   ipod.pinfo[i].size/sectors_per_MB,
                   get_parttype(ipod.pinfo[i].type),
                   ipod.pinfo[i].type);
             ERR_DIALOG(buf, wxT("reading partitiontable"));
        }
    }
       return false;
    }

    read_directory(&ipod);

    if (ipod.nimages <= 0)
    {
        ERR_DIALOG(wxT("[ERR]  Failed to read firmware directory"), wxT("reading directory"));
        return false;
    }
    if (getmodel(&ipod,(ipod.ipod_directory[0].vers>>8)) < 0)
    {
       buf.Printf(wxT("[ERR] Unknown version number in firmware (%08x)\n"),
                                        ipod.ipod_directory[0].vers );
       ERR_DIALOG(buf, wxT("reading directory"));
       return false;
    }

    if (ipod.macpod)
    {
      WARN_DIALOG(wxT("Warning this is a MacPod, Rockbox doesnt work on this. Convert it to WinPod"),wxT("MacPod"));
    }


    if(mode == BOOTLOADER_ADD)
    {
        if (ipod_reopen_rw(&ipod) < 0) {
          ERR_DIALOG(wxT("[ERR] Could not open Ipod in RW mode"), wxT("Bootloader add"));
          return false;
        }

        if (add_bootloader(&ipod, (char*)dest.c_str(), FILETYPE_DOT_IPOD)==0) {

        } else {
           ERR_DIALOG(wxT("[ERR] failed to add Bootloader"), wxT("Bootloader add"));
           return false;
        }
    }
    else if(mode == BOOTLOADER_REM)
    {
        if (ipod_reopen_rw(&ipod) < 0) {
          ERR_DIALOG(wxT("[ERR] Could not open Ipod in RW mode"), wxT("Bootloader add"));
          return false;
        }

        if (ipod.ipod_directory[0].entryOffset==0) {
            ERR_DIALOG(wxT("[ERR]  No bootloader detected.\n"), wxT("Bootloader del"));
            return false;
        } else {
            if (delete_bootloader(&ipod)==0) {

            } else {
               ERR_DIALOG(wxT("[ERR]  --delete-bootloader failed.\n"), wxT("Bootloader del"));
               return false;
            }
        }
    }

    ipod_close(&ipod);
    return true;
}

// gigabeatinstallation
bool gigabeatf(int mode,wxString bootloadername,wxString deviceDir)
{
   wxString path1,path2;
   wxString err;
   wxString src,dest;

   path1.Printf(wxT("%s" PATH_SEP "GBSYSTEM" PATH_SEP "FWIMG" PATH_SEP "FWIMG01.DAT"),deviceDir.c_str());

    if(mode == BOOTLOADER_ADD)
    {
        //Files downloaden
        src.Printf(wxT("%s/gigabeat/%s"), gv->bootloader_url.c_str(),bootloadername.c_str());
        dest.Printf(wxT("%s" PATH_SEP "download" PATH_SEP "%s"),
                    gv->stdpaths->GetUserDataDir().c_str(),bootloadername.c_str());
         if( DownloadURL(src, dest) )
         {
              wxRemoveFile(dest);
              err.Printf(wxT("Unable to download %s"), src.c_str() );
              ERR_DIALOG(err, wxT("Install"));
              return false;
         }


      if(!wxFileExists(path1))
      {
          err.Printf(wxT("[ERR] Coud not find %s"),path1.c_str());
          ERR_DIALOG(err, wxT("Bootloader add"));
          return false;
      }
      path2 = path1;
      path2.Append(wxT(".ORIG"));
      if(!wxFileExists(path2))
      {
         if(!wxRenameFile(path1,path2,false))
         {
           err.Printf(wxT("[ERR] Coud not rename %s to %s"),path1.c_str(),path2.c_str());
           ERR_DIALOG(err, wxT("Bootloader add"));
           return false;
         }
      }


      if(!wxCopyFile(dest,path1))
      {
         err.Printf(wxT("[ERR] Coud not copy %s to %s"),dest.c_str(),path2.c_str());
         ERR_DIALOG(err, wxT("Bootloader add"));
         return false;
      }
    }
    else if(mode == BOOTLOADER_REM)
    {
      path2 = path1;
      path2.Append(wxT(".ORIG"));
      if(!wxFileExists(path2))
      {
          err.Printf(wxT("[ERR] Coud not find %s"),path1.c_str());
          ERR_DIALOG(err, wxT("Bootloader del"));
          return false;
      }
      if(!wxRenameFile(path2,path1,true))
      {
         err.Printf(wxT("[ERR] Coud not rename %s to %s"),path1.c_str(),path2.c_str());
         ERR_DIALOG(err, wxT("Bootloader del"));
         return false;
      }
    }
    return true;
}

// iaudio bootloader install
bool iaudiox5(int mode,wxString bootloadername,wxString deviceDir)
{
    wxString path1,path2;
    wxString err;
    wxString src,dest;

    path1.Printf(wxT("%s" PATH_SEP "FIRMWARE" PATH_SEP "%s"),deviceDir.c_str(),bootloadername.c_str());

    if(mode == BOOTLOADER_ADD)
    {
        //Files downloaden
        src.Printf(wxT("%s/iaudio/%s"),gv->bootloader_url.c_str(),bootloadername.c_str());
        dest.Printf(wxT("%s" PATH_SEP "download" PATH_SEP "%s"),
                    gv->stdpaths->GetUserDataDir().c_str(),bootloadername.c_str());
         if( DownloadURL(src, dest) )
         {
              wxRemoveFile(dest);
              err.Printf(wxT("Unable to download %s"), src.c_str() );
              ERR_DIALOG(err, wxT("Install"));
              return false;
         }

         // copy file
         if(!wxCopyFile(dest,path1))
         {
            err.Printf(wxT("[ERR] Coud not copy %s to %s"),dest.c_str(),path2.c_str());
            ERR_DIALOG(err, wxT("Bootloader add"));
            return false;
         }

         return true;   // install ready
    }
    else
      return false;   //no uninstallation possible
}

// H10 install
bool h10(int mode,wxString bootloadername,wxString deviceDir)
{
  wxString err,src,dest,path1,path2;

   int pos = bootloadername.Find('/');
   if(pos == wxNOT_FOUND) pos = 0;
   wxString firmwarename = bootloadername.SubString(pos,bootloadername.Length());
   //wxString firmDir = gv->curbootloader.SubString(0,pos);

  if(mode == BOOTLOADER_ADD)
  {
     //Files downloaden
     src.Printf(wxT("%s/iriver/%s"), gv->bootloader_url.c_str(),bootloadername.c_str());
     dest.Printf(wxT("%s" PATH_SEP "download" PATH_SEP "%s"),
                  gv->stdpaths->GetUserDataDir().c_str(),firmwarename.c_str());
     if( DownloadURL(src, dest) )
     {
           wxRemoveFile(dest);
           err.Printf(wxT("Unable to download %s"), src.c_str() );
           ERR_DIALOG(err, wxT("Install"));
           return false;
     }

     path1.Printf(wxT("%sSYSTEM" PATH_SEP "%s"),deviceDir.c_str(),firmwarename.c_str());
     path2.Printf(wxT("%sSYSTEM" PATH_SEP "Original.mi4"),deviceDir.c_str());

     if(!wxFileExists(path1))  //Firmware dosent exists on player
     {
        path1.Printf(wxT("%sSYSTEM" PATH_SEP "H10EMP.mi4"),deviceDir.c_str());   //attempt other firmwarename
        if(!wxFileExists(path1))  //Firmware dosent exists on player
        {
            err.Printf(wxT("[ERR] File %s does not Exist"),path1.c_str());
            ERR_DIALOG(err, wxT("Bootloader add"));
            return false;
        }
     }
     if(!wxFileExists(path2))  //there is already a original firmware
     {
        if(!wxRenameFile(path1,path2,false))  //rename Firmware to Original
        {
           err.Printf(wxT("[ERR] Coud not rename %s to %s"),path1.c_str(),path2.c_str());
           ERR_DIALOG(err, wxT("Bootloader add"));
           return false;
        }
     }

     if(!wxCopyFile(dest,path1))  // copy file
     {
        err.Printf(wxT("[ERR] Coud not copy %s to %s"),dest.c_str(),path1.c_str());
        ERR_DIALOG(err,wxT("Bootloader add"));
        return false;
     }

     return true;  //install ready

  }
  else if(mode == BOOTLOADER_REM)
  {
     path1.Printf(wxT("%sSYSTEM" PATH_SEP "%s"),deviceDir.c_str(),firmwarename.c_str());
     path2.Printf(wxT("%sSYSTEM" PATH_SEP "Original.mi4"),gv->curdestdir.c_str());
     if(!wxFileExists(path1))  //Firmware dosent exists on player
     {
         path1.Printf(wxT("%s" PATH_SEP "SYSTEM" PATH_SEP "H10EMP.mi4"),deviceDir.c_str());   //attempt other firmwarename
         if(!wxFileExists(path1))  //Firmware dosent exists on player
         {
            err.Printf(wxT("[ERR] File %s does not Exist"),path1.c_str());
            ERR_DIALOG(err, wxT("Bootloader rem"));
            return false;
         }
     }

     if(!wxFileExists(path2))  //Original Firmware dosent exists on player
     {
         err.Printf(wxT("[ERR] File %s does not Exist"),path2.c_str());
         ERR_DIALOG(err, wxT("Bootloader rem"));
         return false;
     }

     if(!wxRenameFile(path2,path1,true))  //rename Firmware to Original
     {
         err.Printf(wxT("[ERR] Coud not rename %s to %s"),path2.c_str(),path1.c_str());
         ERR_DIALOG(err, wxT("Bootloader add"));
         return false;
     }

  }
  // shouldn't get here
  return false;
}

// FWPatcher
bool fwpatcher(int mode,wxString bootloadername,wxString deviceDir,wxString firmware)
{
    if(mode == BOOTLOADER_ADD)
    {
		char md5sum_str[32];
        wxString src,dest,err;
        int series,table_entry;

        if (!FileMD5(firmware, md5sum_str)) {
        ERR_DIALOG(wxT("Could not open firmware"), wxT("Open Firmware"));
        return false;
       }
       else {
               /* Check firmware against md5sums in h120sums and h100sums */
                series = 0;
                table_entry = intable(md5sum_str, &h120pairs[0],
                                      sizeof(h120pairs)/sizeof(struct sumpairs));
                if (table_entry >= 0) {
                    series = 120;
                }
                else {
                    table_entry = intable(md5sum_str, &h100pairs[0],
                                          sizeof(h100pairs)/sizeof(struct sumpairs));
                    if (table_entry >= 0) {
                        series = 100;
                    }
                    else {
                        table_entry =
                            intable(md5sum_str, &h300pairs[0],
                                    sizeof(h300pairs)/sizeof(struct sumpairs));
                        if (table_entry >= 0)
                            series = 300;
                    }
                }
                if (series == 0) {
                    ERR_DIALOG(wxT("Could not detect firmware type"), wxT("Detect Player out of Firmware"));
                    return false;
                }
                else
                {
                    //Download bootloader
                    src.Printf(wxT("%s/iriver/%s"), gv->bootloader_url.c_str(),bootloadername.c_str());
                    dest.Printf(wxT("%s" PATH_SEP "download" PATH_SEP "%s"),
                            gv->stdpaths->GetUserDataDir().c_str(),bootloadername.c_str());
                    if( DownloadURL(src, dest) )
                    {
                        wxRemoveFile(dest);
                        err.Printf(wxT("Unable to download %s"), src.c_str() );
                        ERR_DIALOG(err, wxT("Install"));
                        return false;
                    }

                    if(!PatchFirmware(firmware,dest,series, table_entry))  // Patch firmware
                    {
                        ERR_DIALOG(wxT("Patching Firmware failed"), wxT("Patching Firmware"));
                        return false;
                    }
                }

                // Load patched Firmware to player
                src.Printf(wxT("%s" PATH_SEP "download" PATH_SEP "new.hex"),
                        gv->stdpaths->GetUserDataDir().c_str());

                if(gv->curplat == wxT("h100"))
                    dest.Printf(wxT("%s" PATH_SEP "ihp_100.hex"),deviceDir.c_str());
                else if(gv->curplat == wxT("h120"))
                    dest.Printf(wxT("%s" PATH_SEP "ihp_120.hex"),deviceDir.c_str());
                else if(gv->curplat == wxT("h300"))
                dest.Printf(wxT("%s" PATH_SEP "H300.hex"),deviceDir.c_str());

                if(!wxRenameFile(src,dest))
                {
                     ERR_DIALOG(wxT("Copying Firmware to Device failed"), wxT("Copying Firmware"));
                     return false;
                }
                else
                {
                    return true;
                }
         }

    }
    else
    {
        return false;  //no uninstall possible
    }
}