121 lines
4.0 KiB
C++
121 lines
4.0 KiB
C++
/*************************************************************************
|
|
* Copyright (C) 2017 by Andrius Štikonas <andrius@stikonas.eu> *
|
|
* *
|
|
* 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 3 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, see <http://www.gnu.org/licenses/>.*
|
|
*************************************************************************/
|
|
|
|
#include "plugins/sfdisk/sfdiskdevice.h"
|
|
#include "plugins/sfdisk/sfdiskpartitiontable.h"
|
|
|
|
#include "core/partitiontable.h"
|
|
|
|
#include "util/externalcommand.h"
|
|
#include "util/report.h"
|
|
|
|
SfdiskDevice::SfdiskDevice(const QString& deviceNode) :
|
|
CoreBackendDevice(deviceNode),
|
|
m_deviceNode(deviceNode)
|
|
{
|
|
}
|
|
|
|
SfdiskDevice::~SfdiskDevice()
|
|
{
|
|
}
|
|
|
|
bool SfdiskDevice::open()
|
|
{
|
|
return true;
|
|
}
|
|
|
|
bool SfdiskDevice::openExclusive()
|
|
{
|
|
setExclusive(true);
|
|
|
|
return true;
|
|
}
|
|
|
|
bool SfdiskDevice::close()
|
|
{
|
|
if (isExclusive())
|
|
setExclusive(false);
|
|
|
|
return true;
|
|
}
|
|
|
|
CoreBackendPartitionTable* SfdiskDevice::openPartitionTable()
|
|
{
|
|
CoreBackendPartitionTable* ptable = new SfdiskPartitionTable(m_deviceNode);
|
|
|
|
if (ptable == nullptr || !ptable->open()) {
|
|
delete ptable;
|
|
ptable = nullptr;
|
|
}
|
|
|
|
return ptable;
|
|
}
|
|
|
|
bool SfdiskDevice::createPartitionTable(Report& report, const PartitionTable& ptable)
|
|
{
|
|
QByteArray tableType;
|
|
if (ptable.type() == PartitionTable::msdos || ptable.type() == PartitionTable::msdos_sectorbased)
|
|
tableType = QByteArrayLiteral("dos");
|
|
else
|
|
tableType = ptable.typeName().toLocal8Bit();
|
|
|
|
ExternalCommand createCommand(report, QStringLiteral("sfdisk"), { m_deviceNode } );
|
|
if ( createCommand.start(-1) && createCommand.write(QByteArrayLiteral("label: ") + tableType +
|
|
QByteArrayLiteral("\nwrite\n")) && createCommand.waitFor() ) {
|
|
return createCommand.output().contains(QStringLiteral("Script header accepted."));
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool SfdiskDevice::readData(QByteArray& buffer, qint64 offset, qint64 size)
|
|
{
|
|
if (!isExclusive())
|
|
return false;
|
|
|
|
ExternalCommand ddCommand(QStringLiteral("dd"), {
|
|
QStringLiteral("skip=") + QString::number(offset),
|
|
QStringLiteral("bs=") + QString::number(size),
|
|
QStringLiteral("count=1"),
|
|
QStringLiteral("iflag=skip_bytes"),
|
|
QStringLiteral("if=") + m_deviceNode }, QProcess::SeparateChannels);
|
|
if (ddCommand.run(-1) && ddCommand.exitCode() == 0) {
|
|
buffer = ddCommand.rawOutput();
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool SfdiskDevice::writeData(QByteArray& buffer, qint64 offset)
|
|
{
|
|
if (!isExclusive())
|
|
return false;
|
|
|
|
ExternalCommand ddCommand(QStringLiteral("dd"), {
|
|
QStringLiteral("of=") + m_deviceNode,
|
|
QStringLiteral("seek=") + QString::number(offset),
|
|
QStringLiteral("bs=1M"),
|
|
QStringLiteral("oflag=seek_bytes"),
|
|
QStringLiteral("conv=fsync") }, QProcess::SeparateChannels);
|
|
if ( ddCommand.start(-1) && ddCommand.write(buffer) == buffer.size() && ddCommand.waitFor() && ddCommand.exitCode() == 0 ) {
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|