///////////////////////////////////////////////////////////////////////////////
// MuldeR's Utilities for Qt
-// Copyright (C) 2004-2018 LoRd_MuldeR <MuldeR2@GMX.de>
+// Copyright (C) 2004-2019 LoRd_MuldeR <MuldeR2@GMX.de>
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
#include <MUtils/Global.h>
#include <MUtils/OSSupport.h>
#include <MUtils/Version.h>
+#include "Internal.h"
//Internal
#include "DirLocker.h"
#include <QProcess>
#include <QTextCodec>
#include <QPair>
+#include <QHash>
#include <QListIterator>
#include <QMutex>
#include <QThreadStorage>
#include <vld.h>
#endif
+//Global
+const QString MUtils::Internal::g_empty;
+
///////////////////////////////////////////////////////////////////////////////
// Random Support
///////////////////////////////////////////////////////////////////////////////
return rnd;
}
+quint32 MUtils::next_rand_u32(const quint32 max)
+{
+ static const uint32_t DIV_LUT[64] =
+ {
+ 0xFFFFFFFF, 0xFFFFFFFF, 0x80000000, 0x55555556, 0x40000000, 0x33333334, 0x2AAAAAAB, 0x24924925,
+ 0x20000000, 0x1C71C71D, 0x1999999A, 0x1745D175, 0x15555556, 0x13B13B14, 0x12492493, 0x11111112,
+ 0x10000000, 0x0F0F0F10, 0x0E38E38F, 0x0D79435F, 0x0CCCCCCD, 0x0C30C30D, 0x0BA2E8BB, 0x0B21642D,
+ 0x0AAAAAAB, 0x0A3D70A4, 0x09D89D8A, 0x097B425F, 0x0924924A, 0x08D3DCB1, 0x08888889, 0x08421085,
+ 0x08000000, 0x07C1F07D, 0x07878788, 0x07507508, 0x071C71C8, 0x06EB3E46, 0x06BCA1B0, 0x06906907,
+ 0x06666667, 0x063E7064, 0x06186187, 0x05F417D1, 0x05D1745E, 0x05B05B06, 0x0590B217, 0x0572620B,
+ 0x05555556, 0x0539782A, 0x051EB852, 0x05050506, 0x04EC4EC5, 0x04D4873F, 0x04BDA130, 0x04A7904B,
+ 0x04924925, 0x047DC120, 0x0469EE59, 0x0456C798, 0x04444445, 0x04325C54, 0x04210843, 0x04104105
+ };
+ return (max < 64) ? (next_rand_u32() / DIV_LUT[max]) : (next_rand_u32() / (UINT32_MAX / max + 1U));
+}
+
+
quint64 MUtils::next_rand_u64(void)
{
return (quint64(next_rand_u32()) << 32) | quint64(next_rand_u32());
QString MUtils::make_temp_file(const QString &basePath, const QString &extension, const bool placeholder)
{
+ return make_temp_file(QDir(basePath), extension, placeholder);
+}
+
+QString MUtils::make_temp_file(const QDir &basePath, const QString &extension, const bool placeholder)
+{
+ if (extension.isEmpty())
+ {
+ qWarning("Cannot generate temp file name with invalid parameters!");
+ return QString();
+ }
+
for(int i = 0; i < 4096; i++)
{
- const QString tempFileName = QString("%1/%2.%3").arg(basePath, next_rand_str(), extension);
+ const QString tempFileName = basePath.absoluteFilePath(QString("%1.%2").arg(next_rand_str(), extension));
if(!QFileInfo(tempFileName).exists())
{
if(placeholder)
return QString();
}
-QString MUtils::make_unique_file(const QString &basePath, const QString &baseName, const QString &extension, const bool fancy)
+QString MUtils::make_unique_file(const QString &basePath, const QString &baseName, const QString &extension, const bool fancy, const bool placeholder)
+{
+ return make_unique_file(QDir(basePath), baseName, extension, fancy);
+}
+
+QString MUtils::make_unique_file(const QDir &basePath, const QString &baseName, const QString &extension, const bool fancy, const bool placeholder)
{
+ if (baseName.isEmpty() || extension.isEmpty())
+ {
+ qWarning("Cannot generate unique file name with invalid parameters!");
+ return QString();
+ }
+
quint32 n = fancy ? 2 : 0;
- QString fileName = fancy ? QString("%1/%2.%3").arg(basePath, baseName, extension) : QString();
+ QString fileName = fancy ? basePath.absoluteFilePath(QString("%1.%2").arg(baseName, extension)) : QString();
while (fileName.isEmpty() || QFileInfo(fileName).exists())
{
if (n <= quint32(USHRT_MAX))
{
if (fancy)
{
- fileName = QString("%1/%2 (%3).%4").arg(basePath, baseName, QString::number(n++), extension);
+ fileName = basePath.absoluteFilePath(QString("%1 (%2).%3").arg(baseName, QString::number(n++), extension));
}
else
{
- fileName = QString("%1/%2.%3.%4").arg(basePath, baseName, QString::number(n++, 16).rightJustified(4, QLatin1Char('0')), extension);
+ fileName = basePath.absoluteFilePath(QString("%1.%2.%3").arg(baseName, QString::number(n++, 16).rightJustified(4, QLatin1Char('0')), extension));
}
}
else
return QString();
}
}
+
+ if (placeholder && (!fileName.isEmpty()))
+ {
+ QFile placeholder(fileName);
+ if (placeholder.open(QIODevice::WriteOnly))
+ {
+ placeholder.close();
+ }
+ }
+
return fileName;
}
static void prependToPath(QProcessEnvironment &env, const QString &value)
{
- const QLatin1String PATH = QLatin1String("PATH");
+ static const QLatin1String PATH("PATH");
const QString path = env.value(PATH, QString()).trimmed();
env.insert(PATH, path.isEmpty() ? value : QString("%1;%2").arg(value, path));
}
-void MUtils::init_process(QProcess &process, const QString &wokringDir, const bool bReplaceTempDir, const QStringList *const extraPaths)
+void MUtils::init_process(QProcess &process, const QString &wokringDir, const bool bReplaceTempDir, const QStringList *const extraPaths, const QHash<QString, QString> *const extraEnv)
{
//Environment variable names
- static const char *const s_envvar_names_temp[] =
+ static const char *const ENVVAR_NAMES_TMP[] =
{
"TEMP", "TMP", "TMPDIR", "HOME", "USERPROFILE", "HOMEPATH", NULL
};
- static const char *const s_envvar_names_remove[] =
+ static const char *const ENVVAR_NAMES_SYS[] =
+ {
+ "WINDIR", "SYSTEMROOT", NULL
+ };
+ static const char *const ENVVAR_NAMES_DEL[] =
{
- "WGETRC", "SYSTEM_WGETRC", "HTTP_PROXY", "FTP_PROXY", "NO_PROXY", "GNUPGHOME", "LC_ALL", "LC_COLLATE", "LC_CTYPE", "LC_MESSAGES", "LC_MONETARY", "LC_NUMERIC", "LC_TIME", "LANG", NULL
+ "HTTP_PROXY", "FTP_PROXY", "NO_PROXY", "HOME", "LC_ALL", "LC_COLLATE", "LC_CTYPE",
+ "LC_MESSAGES", "LC_MONETARY", "LC_NUMERIC", "LC_TIME", "LANG", NULL
};
//Initialize environment
QProcessEnvironment env = process.processEnvironment();
- if(env.isEmpty()) env = QProcessEnvironment::systemEnvironment();
+ if (env.isEmpty())
+ {
+ env = QProcessEnvironment::systemEnvironment();
+ }
- //Clean a number of enviroment variables that might affect our tools
- for(size_t i = 0; s_envvar_names_remove[i]; i++)
+ //Clean enviroment variables that might affect our tools
+ for(const char *const *ptr = ENVVAR_NAMES_DEL; *ptr; ++ptr)
{
- env.remove(QString::fromLatin1(s_envvar_names_remove[i]));
- env.remove(QString::fromLatin1(s_envvar_names_remove[i]).toLower());
+ env.remove(QString::fromLatin1(*ptr));
}
- const QString tempDir = QDir::toNativeSeparators(temp_folder());
+ //Set up system root directory
+ const QString sysRoot = QDir::toNativeSeparators(OS::known_folder(OS::FOLDER_SYSTROOT_DIR));
+ if (!sysRoot.isEmpty())
+ {
+ for (const char *const *ptr = ENVVAR_NAMES_SYS; *ptr; ++ptr)
+ {
+ env.insert(QString::fromLatin1(*ptr), sysRoot);
+ }
+ }
//Replace TEMP directory in environment
+ const QString tempDir = QDir::toNativeSeparators(temp_folder());
if(bReplaceTempDir)
{
- for(size_t i = 0; s_envvar_names_temp[i]; i++)
+ for (const char *const *ptr = ENVVAR_NAMES_TMP; *ptr; ++ptr)
{
- env.insert(s_envvar_names_temp[i], tempDir);
+ env.insert(QString::fromLatin1(*ptr), tempDir);
}
}
}
}
+ //Setup environment
+ if (extraEnv && (!extraEnv->isEmpty()))
+ {
+ for (QHash<QString, QString>::ConstIterator iter = extraEnv->constBegin(); iter != extraEnv->constEnd(); iter++)
+ {
+ env.insert(iter.key(), iter.value());
+ }
+ }
+
//Setup QPorcess object
process.setWorkingDirectory(wokringDir);
process.setProcessChannelMode(QProcess::MergedChannels);
}
///////////////////////////////////////////////////////////////////////////////
-// SELF-TEST
+// INITIALIZER
///////////////////////////////////////////////////////////////////////////////
-int MUtils::Internal::selfTest(const char *const buildKey, const bool debug)
+int MUtils::Internal::MUTILS_INITIALIZER(const char *const buildKey, const bool debug)
{
- static const bool MY_DEBUG_FLAG = MUTILS_DEBUG;
- static const char *const MY_BUILD_KEY = __DATE__ "@" __TIME__;
+ static const bool INTERNAL_DEBUG_FLAG = MUTILS_DEBUG;
+ static const char *const INTERNAL_BUILD_KEY = MUTILS_BUILD_KEY;
- if(strncmp(buildKey, MY_BUILD_KEY, 13) || (MY_DEBUG_FLAG != debug))
+ if((debug != INTERNAL_DEBUG_FLAG) || strncmp(buildKey, INTERNAL_BUILD_KEY, 11))
{
- MUtils::OS::system_message_err(L"MUtils", L"FATAL ERROR: MUtils library version mismatch detected!");
- MUtils::OS::system_message_wrn(L"MUtils", L"Perform a clean(!) re-install of the application to fix the problem!");
+ MUtils::OS::system_message_err(L"MUtils", L"FATAL: MUtils library initialization has failed!");
abort();
}
- return 0;
+
+ volatile int _result = MUTILS_INTERFACE;
+ return _result;
}
OSDN Git Service
