ext2copyfile.cpp

/**
 * Copyright (C) 2010 by Manish Regmi   (regmi dot manish at gmail.com)
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the
 * Free Software Foundation, Inc.,
 * 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 **/

#include <QMessageBox>
#include <QDir>

#include "ext2copyfile.h"

Ext2CopyFile::Ext2CopyFile(Ext2File *parent, QString &dest)
{
    filename = dest;
    file = parent;
    Ext2CopyFile();
}

Ext2CopyFile::Ext2CopyFile()
{
    cancelOperation = false;

    progress = new Ui::ProgressDialog();
    progress->setupUi(this);
    progress->progressBar->setMaximum(100);
    progress->progressBar->setMinimum(0);
}

Ext2CopyFile::~Ext2CopyFile()
{
    delete progress;
}

void Ext2CopyFile::start_copy()
{
    if(!EXT2_S_ISREG(file->inode.i_mode) &&
       !EXT2_S_ISDIR(file->inode.i_mode))
        return ;

    proc = new Ext2CopyProcess(file, filename);
    qRegisterMetaType<QString>("QString&");
    QObject::connect(proc, SIGNAL(sig_updateui(QString&,QString&,QString&,int,int)),
            this, SLOT(slot_updateui(QString&,QString&,QString&,int,int)));
    QObject::connect(proc, SIGNAL(sig_copydone()), this, SLOT(slot_copydone()));
    QObject::connect(this, SIGNAL(signal_cancelprocess()), proc, SLOT(slot_cancelprocess()));
    QObject::connect(this, SIGNAL(accepted()), proc, SLOT(slot_cancelprocess()));
    QObject::connect(this, SIGNAL(rejected()), proc, SLOT(slot_cancelprocess()));
    this->show();
    proc->start();
}

bool Ext2CopyFile::showMessageBox()
{
    QMessageBox msgBox;
    msgBox.setText(QString("You pressed the cancel button on the progress dialog box."));
    msgBox.setInformativeText(QString("Are you sure you want to cancel copying?"));
    msgBox.setStandardButtons(QMessageBox::Yes | QMessageBox::No);
    msgBox.setDefaultButton(QMessageBox::No);
    int ret = msgBox.exec();

    switch(ret) {
    case QMessageBox::Yes:
        return true;
    case QMessageBox::No:
        return false;
    }

    return false;
}

void Ext2CopyFile::slot_updateui(QString &file, QString &from, QString &to, int copied, int total)
{
    int value;

    if(!file.isEmpty())
        progress->FileLabel->setText(file);
    if(!from.isEmpty())
        progress->Fromlabel->setText(from);
    if(!to.isEmpty())
        progress->ToLabel->setText(to);

    value = (copied * 100) / total;
    progress->progressBar->setValue(value);
}

void Ext2CopyFile::slot_copydone()
{
    //delete this;
    delete proc;
    this->hide();
}

Ext2CopyProcess::Ext2CopyProcess(Ext2File *parent, QString &dest)
    : QThread()
{
    filename = dest;
    file = parent;
    blksize = parent->partition->get_blocksize();
    buffer = new char [blksize];
    filetosave = NULL;
    cancelOperation = false;
    codec = QTextCodec::codecForName("utf-8");
}

Ext2CopyProcess::~Ext2CopyProcess()
{
    delete [] buffer;
    if(filetosave)
        delete filetosave;
}

void Ext2CopyProcess::slot_cancelprocess()
{
    cancelOperation = true;
}

void Ext2CopyProcess::run()
{
    if(EXT2_S_ISDIR(file->inode.i_mode))
    {
        copy_folder(filename, file);
        emit sig_copydone();
        return ;
    }
    else if(!EXT2_S_ISREG(file->inode.i_mode))
        return ;

    copy_file(filename, file);
    emit sig_copydone();
}

bool Ext2CopyProcess::copy_file(QString &destfile, Ext2File *srcfile)
{
    lloff_t blocks, blkindex;
    QString qsrc;
    QByteArray ba;
    int extra;
    int ret;

    filetosave = new QFile(destfile);
    if (!filetosave->open(QIODevice::ReadWrite | QIODevice::Truncate))
    {
        LOG("Error creating file %s.\n", srcfile->file_name.c_str());
        return false;
    }
    //ba = destfile.toAscii();
    //const char *c_str2 = ba.data();

    //LOG("Copying file %s as %s\n", srcfile->file_name.c_str(), c_str2);
    qsrc = codec->toUnicode(srcfile->file_name.c_str());
    blocks = srcfile->file_size / blksize;
    for(blkindex = 0; blkindex < blocks; blkindex++)
    {
        if(cancelOperation)
        {
            return false;
        }
        ret = srcfile->partition->read_data_block(&srcfile->inode, blkindex, buffer);
        if(ret < 0)
        {
            filetosave->close();
            return false;
        }
        filetosave->write(buffer, blksize);
        if((blkindex % 10) == 0)
            emit sig_updateui(qsrc, destfile, filename, (int)blkindex, (int)blocks);
    }

    extra = srcfile->file_size % blksize;
    if(extra)
    {
        ret = srcfile->partition->read_data_block(&srcfile->inode, blkindex, buffer);
        if(ret < 0)
        {
            filetosave->close();
            return false;
        }
        filetosave->write(buffer, extra);
    }
    filetosave->close();
    return true;
}

bool Ext2CopyProcess::copy_folder(QString &path, Ext2File *parent)
{
    QDir dir(path);
    QString filetosave;
    QString rootname = path;
    EXT2DIRENT *dirent;
    Ext2Partition *part = parent->partition;
    Ext2File *child;
    QByteArray ba;
    bool ret;


    if(!EXT2_S_ISDIR(parent->inode.i_mode))
        return false;

    dir.mkdir(codec->toUnicode(parent->file_name.c_str()));
    /*ba = path.toAscii();
    const char *c_str2 = ba.data();
    LOG("Creating Folder %s as %s\n", parent->file_name.c_str(), c_str2);
*/
    dirent = part->open_dir(parent);
    while((child = part->read_dir(dirent)) != NULL)
    {
        filetosave = rootname;
        filetosave.append(QString("/"));
        filetosave.append(codec->toUnicode(parent->file_name.c_str()));
        if(EXT2_S_ISDIR(child->inode.i_mode))
        {

            ret = copy_folder(filetosave, child);
            if((ret == false) && (cancelOperation == true))
            {
                part->close_dir(dirent);
                return false;
            }
            continue;
        }
        else if(!EXT2_S_ISREG(child->inode.i_mode))
        {
            continue;
            //part->close_dir(dirent);
            //return false;
        }

        filetosave.append(QString("/"));
        filetosave.append(codec->toUnicode(child->file_name.c_str()));
        ret = copy_file(filetosave, child);
        if((ret == false) && (cancelOperation == true))
        {
            part->close_dir(dirent);
            return false;
        }
    }
    return true;
}



void Ext2CopyFile::on_buttonBox_clicked(QAbstractButton* button)
{
     cancelOperation = showMessageBox();
     if(cancelOperation)
         emit signal_cancelprocess();
}