~ubuntu-branches/ubuntu/karmic/partitionmanager/karmic

/***************************************************************************

 *   Copyright (C) 2008 by Volker Lanz <vl@fidra.de>                       *

 *                                                                         *

 *   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.,                                       *

 *   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA            *

 ***************************************************************************/

 

#include "jobs/resizefilesystemjob.h"

 

#include "core/partition.h"

#include "core/device.h"

 

#include "fs/filesystem.h"

 

#include "util/report.h"

#include "util/capacity.h"

 

#include <klocale.h>

#include <kdebug.h>

 

#include <parted/parted.h>

 

/** Creates a new ResizeFileSystemJob

        @param d the Device the FileSystem to be resized is on

        @param p the Partition the FileSystem to be resized is on

        @param newlength the new length for the FileSystem; if -1, the FileSystem will be resized to fill the entire Partition

*/

ResizeFileSystemJob::ResizeFileSystemJob(Device& d, Partition& p, qint64 newlength) :

        Job(),

        m_Device(d),

        m_Partition(p),

        m_Maximize(newlength == -1),

        m_NewLength(isMaximizing() ? partition().length() : newlength)

{

}

 

qint32 ResizeFileSystemJob::numSteps() const

{

        return 100;

}

 

bool ResizeFileSystemJob::run(Report& parent)

{

        Q_ASSERT(partition().fileSystem().firstSector() != -1);

        Q_ASSERT(partition().fileSystem().lastSector() != -1);

        Q_ASSERT(newLength() <= partition().length());

 

        if (partition().fileSystem().firstSector() == -1 || partition().fileSystem().lastSector() == -1 || newLength() > partition().length())

        {

                kWarning() << "file system first sector: " << partition().fileSystem().firstSector() << ", last sector: " << partition().fileSystem().lastSector() << ", new length: " << newLength() << ", partition length: " << partition().length();

                return false;

        }

        

        bool rval = false;

        

        Report* report = jobStarted(parent);

        

        if (partition().fileSystem().length() == newLength())

        {

                report->line() << i18nc("@info/plain", "The file system on partition <filename>%1</filename> already has the requested length of %2 sectors.", partition().deviceNode(), newLength());

                rval = true;

        }

        else

        {

                report->line() << i18nc("@info/plain", "Resizing file system from %1 to %2 sectors.", partition().fileSystem().length(), newLength());

                

                FileSystem::SupportType support = (newLength() < partition().fileSystem().length()) ? partition().fileSystem().supportShrink() : partition().fileSystem().supportGrow();

 

                switch(support)

                {

                        case FileSystem::SupportLibParted:

                        {

                                Report* childReport = report->newChild();

                                childReport->line() << i18nc("@info/plain", "Resizing a %1 file system using LibParted internal functions.", partition().fileSystem().name());

                                rval = resizeFileSystemLibParted(*childReport);

                                break;

                        }

 

                        case FileSystem::SupportExternal:

                        {

                                const qint64 newLengthInByte = Capacity(newLength() * partition().sectorSize()).toInt(Capacity::Byte);

                                rval = partition().fileSystem().resize(*report, partition().deviceNode(), newLengthInByte);

                                break;

                        }

 

                        default:

                                report->line() << i18nc("@info/plain", "The file system on partition <filename>%1</filename> cannot be resized because there is no support for it.", partition().deviceNode());

                                break;

                }

                

                if (rval)

                        partition().fileSystem().setLastSector(partition().fileSystem().firstSector() + newLength() - 1);

        }

        

        jobFinished(*report, rval);

                

        return rval;

}

 

bool ResizeFileSystemJob::resizeFileSystemLibParted(Report& report)

{

        bool rval = false;

 

        if (!openPed(device().deviceNode()))

        {

                report.line() << i18nc("@info/plain", "Resizing file system on partition <filename>%1</filename>: Could not open device <filename>%2</filename>.", partition().deviceNode(), device().deviceNode());

                return false;

        }

 

        if (PedGeometry* originalGeometry = ped_geometry_new(pedDevice(), partition().fileSystem().firstSector(), partition().fileSystem().length()))

        {

                if (PedFileSystem* pedFileSystem = ped_file_system_open(originalGeometry))

                {

                        if (PedGeometry* resizedGeometry = ped_geometry_new(pedDevice(), partition().fileSystem().firstSector(), newLength()))

                        {

                                PedTimer* pedTimer = ped_timer_new(pedTimerHandler, this);

                                rval = ped_file_system_resize(pedFileSystem, resizedGeometry, pedTimer) && commit();

                                ped_timer_destroy(pedTimer);

 

                                if (rval)

                                        report.line() << i18nc("@info/plain", "Successfully resized file system using LibParted.");

                                else

                                        report.line() << i18nc("@info/plain", "Could not resize file system on partition <filename>%1</filename>.", partition().deviceNode());

                        }

                        else

                                report.line() << i18nc("@info/plain", "Could not get geometry for resized partition <filename>%1</filename> while trying to resize the file system.", partition().deviceNode());

 

                        ped_file_system_close(pedFileSystem);

                }

                else

                        report.line() << i18nc("@info/plain", "Could not open partition <filename>%1</filename> while trying to resize the file system.", partition().deviceNode());

        }

        else

                report.line() << i18nc("@info/plain", "Could not read geometry for partition <filename>%1</filename> while trying to resize the file system.", partition().deviceNode());

        

        closePed();

        

        return rval;

}

 

QString ResizeFileSystemJob::description() const

{

        if (isMaximizing())

                return i18nc("@info/plain", "Maximize file system on <filename>%1</filename> to fill the partition", partition().deviceNode());

                

        return i18nc("@info/plain", "Resize file system on partition <filename>%1</filename> to %2 sectors", partition().deviceNode(), newLength());

}