mirror of https://github.com/cutefishos/calamares
[machineid] Split helper functions into own file
Adriaan de Groot
6 years ago
parent
afe7dfbcf2
commit
10e5995144
@ -0,0 +1,154 @@
|
||||
/* === This file is part of Calamares - <https://github.com/calamares> ===
|
||||
*
|
||||
* Copyright 2014, Kevin Kofler <kevin.kofler@chello.at>
|
||||
* Copyright 2016, Philip Müller <philm@manjaro.org>
|
||||
* Copyright 2017, Alf Gaida <agaida@siduction.org>
|
||||
* Copyright 2019, Adriaan de Groot <groot@kde.org>
|
||||
*
|
||||
* Calamares 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.
|
||||
*
|
||||
* Calamares 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 Calamares. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include "Workers.h"
|
||||
|
||||
#include "utils/CalamaresUtilsSystem.h"
|
||||
#include "utils/Logger.h"
|
||||
|
||||
#include <QFile>
|
||||
|
||||
namespace MachineId
|
||||
{
|
||||
|
||||
// might need to use a helper to remove the file
|
||||
void
|
||||
removeFile( const QString& rootMountPoint, const QString& fileName )
|
||||
{
|
||||
QFile::remove( rootMountPoint + fileName );
|
||||
}
|
||||
|
||||
Calamares::JobResult
|
||||
copyFile( const QString& rootMountPoint, const QString& fileName )
|
||||
{
|
||||
QFile f( fileName );
|
||||
if ( !f.exists() )
|
||||
{
|
||||
return Calamares::JobResult::error( QObject::tr( "File not found" ), fileName );
|
||||
}
|
||||
if ( !f.copy( rootMountPoint + fileName ) )
|
||||
{
|
||||
return Calamares::JobResult::error( QObject::tr( "File not found" ), rootMountPoint + fileName );
|
||||
}
|
||||
return Calamares::JobResult::ok();
|
||||
}
|
||||
|
||||
int
|
||||
getUrandomPoolSize()
|
||||
{
|
||||
QFile f( "/proc/sys/kernel/random/poolsize" );
|
||||
constexpr const int minimumPoolSize = 512;
|
||||
int poolSize = minimumPoolSize;
|
||||
|
||||
if ( f.exists() && f.open( QIODevice::ReadOnly | QIODevice::Text ) )
|
||||
{
|
||||
QByteArray v = f.read( 16 );
|
||||
if ( v.length() > 2 )
|
||||
{
|
||||
bool ok = false;
|
||||
poolSize = v.toInt( &ok );
|
||||
if ( !ok )
|
||||
{
|
||||
poolSize = minimumPoolSize;
|
||||
}
|
||||
}
|
||||
}
|
||||
return poolSize >= minimumPoolSize ? poolSize : minimumPoolSize;
|
||||
}
|
||||
|
||||
Calamares::JobResult
|
||||
createNewEntropy( int poolSize, const QString& rootMountPoint, const QString& fileName )
|
||||
{
|
||||
QFile urandom( "/dev/urandom" );
|
||||
if ( urandom.exists() && urandom.open( QIODevice::ReadOnly ) )
|
||||
{
|
||||
QByteArray data = urandom.read( poolSize );
|
||||
urandom.close();
|
||||
|
||||
QFile entropyFile( rootMountPoint + fileName );
|
||||
if ( entropyFile.exists() )
|
||||
{
|
||||
cWarning() << "Entropy file" << ( rootMountPoint + fileName ) << "already exists.";
|
||||
return Calamares::JobResult::ok(); // .. anyway
|
||||
}
|
||||
if ( !entropyFile.open( QIODevice::WriteOnly ) )
|
||||
{
|
||||
return Calamares::JobResult::error(
|
||||
QObject::tr( "File not found" ),
|
||||
QObject::tr( "Could not create new random file <pre>%1</pre>." ).arg( fileName ) );
|
||||
}
|
||||
entropyFile.write( data );
|
||||
entropyFile.close();
|
||||
if ( entropyFile.size() < data.length() )
|
||||
{
|
||||
cWarning() << "Entropy file is" << entropyFile.size() << "bytes, random data was" << data.length();
|
||||
}
|
||||
if ( data.length() < poolSize )
|
||||
{
|
||||
cWarning() << "Entropy data is" << data.length() << "bytes, rather than poolSize" << poolSize;
|
||||
}
|
||||
}
|
||||
return Calamares::JobResult::error(
|
||||
QObject::tr( "File not found" ),
|
||||
QObject::tr( "Could not read random file <pre>%1</pre>." ).arg( QStringLiteral( "/dev/urandom" ) ) );
|
||||
}
|
||||
|
||||
|
||||
Calamares::JobResult
|
||||
createEntropy( const EntropyGeneration kind, const QString& rootMountPoint, const QString& fileName )
|
||||
{
|
||||
if ( kind == EntropyGeneration::CopyFromHost )
|
||||
{
|
||||
if ( QFile::exists( fileName ) )
|
||||
{
|
||||
auto r = copyFile( rootMountPoint, fileName );
|
||||
if ( r )
|
||||
{
|
||||
return r;
|
||||
}
|
||||
else
|
||||
{
|
||||
cWarning() << "Could not copy" << fileName << "for entropy, generating new.";
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
cWarning() << "Host system entropy does not exist at" << fileName;
|
||||
}
|
||||
}
|
||||
|
||||
int poolSize = getUrandomPoolSize();
|
||||
return createNewEntropy( poolSize, rootMountPoint, fileName );
|
||||
}
|
||||
|
||||
Calamares::JobResult
|
||||
createSystemdMachineId( const QString& rootMountPoint, const QString& fileName )
|
||||
{
|
||||
return Calamares::JobResult::internalError( QObject::tr( "Internal Error" ), QObject::tr( "Not implemented" ), 0 );
|
||||
}
|
||||
|
||||
Calamares::JobResult
|
||||
createDBusMachineId( DBusGeneration kind, const QString& rootMountPoint, const QString& fileName )
|
||||
{
|
||||
return Calamares::JobResult::internalError( QObject::tr( "Internal Error" ), QObject::tr( "Not implemented" ), 0 );
|
||||
}
|
||||
|
||||
} // namespace
|
||||
@ -0,0 +1,83 @@
|
||||
/* === This file is part of Calamares - <https://github.com/calamares> ===
|
||||
*
|
||||
* Copyright 2019, Adriaan de Groot <groot@kde.org>
|
||||
*
|
||||
* Calamares 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.
|
||||
*
|
||||
* Calamares 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 Calamares. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#ifndef WORKERS_H
|
||||
#define WORKERS_H
|
||||
|
||||
#include "Job.h"
|
||||
|
||||
/// @brief Utility functions for doing the random-data stuff for MachineId
|
||||
namespace MachineId
|
||||
{
|
||||
/** @brief Utility functions
|
||||
*
|
||||
* These probably belong in libcalamares, since they're general utilities
|
||||
* for moving files around in the target system.
|
||||
*/
|
||||
|
||||
/// @brief Remove @p fileName from the target system at @p rootMountPoint
|
||||
void removeFile( const QString& rootMountPoint, const QString& fileName );
|
||||
|
||||
/// @brief Copy @p fileName from host into target system at @p rootMountPoint
|
||||
Calamares::JobResult copyFile( const QString& rootMountPoint, const QString& fileName );
|
||||
|
||||
|
||||
/** @brief Entropy functions
|
||||
*
|
||||
* The target system may want to pre-seed the entropy pool with a suitable
|
||||
* chunk of entropy data. During installation we have lots of disk access
|
||||
* so plenty of entropy -- this is used mostly be Debian.
|
||||
*/
|
||||
|
||||
/// @brief How to generate entropy (bool-like)
|
||||
enum class EntropyGeneration
|
||||
{
|
||||
New,
|
||||
CopyFromHost
|
||||
};
|
||||
|
||||
/// @brief Returns a recommended size for the entropy pool (in bytes)
|
||||
int getUrandomPoolSize();
|
||||
|
||||
/// @brief Creates a new entropy file @p fileName in the target system at @p rootMountPoint
|
||||
Calamares::JobResult createNewEntropy( int poolSize, const QString& rootMountPoint, const QString& fileName );
|
||||
|
||||
/// @brief Create an entropy file @p fileName in the target system at @p rootMountPoint
|
||||
Calamares::JobResult createEntropy( const EntropyGeneration kind, const QString& rootMountPoint, const QString& fileName );
|
||||
|
||||
|
||||
/** @brief MachineID functions
|
||||
*
|
||||
* Creating UUIDs for DBUS and SystemD.
|
||||
*/
|
||||
|
||||
/// @brief How to create the DBus machine-id (bool-like)
|
||||
enum class DBusGeneration
|
||||
{
|
||||
New,
|
||||
SymlinkFromSystemD
|
||||
};
|
||||
|
||||
Calamares::JobResult createDBusMachineId( DBusGeneration kind, const QString& rootMountPoint, const QString& fileName );
|
||||
|
||||
Calamares::JobResult createSystemdMachineId( const QString& rootMountPoint, const QString& fileName );
|
||||
|
||||
|
||||
}
|
||||
|
||||
#endif // WORKERS_H
|
||||
Loading…
Reference in New Issue