由于公司应用大量装机的情况,以前使用ghost软件来处理,但是自从uefi模式出现后ghost已经不能很好的应用了,虽然现在有Acronis软件替代,但是生成的备份文件没有很好的工具去修改,所以遇到需要添加镜像驱动和补丁的时候带来非常大的麻烦。系统自带的dism工具确实好用,但是每次都需要输入命令行输入,效率不能提高。因此根据现在的实际情况,清明三天假连学带做了这款窗口版的备份还原工具,功能目前很很简单,只有备份分区(文件夹)和还原分区功能,下一步会添加驱动添加功能。废话不说,上代码。
//DismTools.h qt widget 头文件 #pragma once #include <QWidget> #include <QThread> #include "ui_DismTools.h" #include <QString> #include <QLabel> #include "BackUp.h" #include "Restore.h" #include "BcdBoot.h" class DismTools : public QWidget { Q_OBJECT public: DismTools(QWidget *parent = Q_NULLPTR); ~DismTools() { thread.quit(); thread.wait(); } private: Ui::DismToolsClass ui; private slots: void on_BackUp_clicked(); void on_BackUp_finished(const QString&); void on_Restore_clicked(); void on_Restore_finished(const QString&); private: QThread thread; BackUp *bk; Restore *restore; BcdBoot *boot; signals: void backSig(const QString &,const QString &); void restoreSig(const QString &, const QString &); void bcdbootSig(const QString &); //PWSTR pszTmpDir = L"C:\\tmp"; };
//dismtools.cpp 代码 #pragma execution_character_set("utf-8") #include "DismTools.h" #include "BackUp.h" #include <QFileInfo> #include <QFileDialog> DismTools::DismTools(QWidget *parent) : QWidget(parent) { ui.setupUi(this); connect(ui.backUp, &QPushButton::clicked, this, &DismTools::on_BackUp_clicked); connect(ui.restore, &QPushButton::clicked, this, &DismTools::on_Restore_clicked); } void DismTools::on_BackUp_clicked() { QString dir = QFileDialog::getExistingDirectory(this, tr("选择备份分区"), "/home", QFileDialog::ShowDirsOnly | QFileDialog::DontResolveSymlinks); QFileInfo fi(dir); if (!fi.exists()) { return; } QString fileName = QFileDialog::getSaveFileName(this, QString("保存镜像文件"), "/home/", QString("镜像文件 (*.wim)")); QFileInfo fi1(fileName); bk = new BackUp(); bk->moveToThread(&thread); connect(&thread, &QThread::finished, bk, &QObject::deleteLater); connect(this, &DismTools::backSig, bk, &BackUp::doBackUp); connect(bk, &BackUp::resultReady, this, &DismTools::on_BackUp_finished); thread.start(); ui.label->setText("镜像备份中..."); ui.backUp->setDisabled(true); emit backSig(fileName, dir); } void DismTools::on_BackUp_finished(const QString &finished) { ui.label->setText(finished); QMessageBox::warning(this, "", finished); ui.backUp->setDisabled(false); } void DismTools::on_Restore_clicked() { QString dir = QFileDialog::getExistingDirectory(this, tr("选择恢复分区"), "/home", QFileDialog::ShowDirsOnly | QFileDialog::DontResolveSymlinks); QFileInfo fi(dir); if (!fi.exists()) { return; } QString fileName = QFileDialog::getOpenFileName(this, QString("选择镜像文件"), "/home/", QString("镜像文件 (*.wim)")); QFileInfo fi1(fileName); restore = new Restore(); restore->moveToThread(&thread); connect(&thread, &QThread::finished, restore, &QObject::deleteLater); connect(this, &DismTools::restoreSig, restore, &Restore::doRestore); connect(restore, &Restore::resultReady, this, &DismTools::on_Restore_finished); thread.start(); ui.label->setText("镜像恢复中..."); ui.restore->setDisabled(true); emit restoreSig(fileName, dir); } void DismTools::on_Restore_finished(const QString &finished) { ui.label->setText(finished); QMessageBox::warning(this, "", finished); ui.restore->setDisabled(false); int ret = QMessageBox::question(this, "", QString("是否添加引导项?"), QMessageBox::Ok,QMessageBox::No); if (ret = 0) { QString souces = QFileDialog::getExistingDirectory(this, tr("选择Windows系统根目录"), "/home", QFileDialog::ShowDirsOnly | QFileDialog::DontResolveSymlinks); QString bootPath = QFileDialog::getExistingDirectory(this, tr("选择Windows启动分区"), "/home", QFileDialog::ShowDirsOnly | QFileDialog::DontResolveSymlinks); boot = new BcdBoot(); boot->doBcdBoot(souces, bootPath); connect(boot, &BcdBoot::resultReady, this, [=](const QString &result) { QMessageBox::information(this, "", result); }); return; } } //backup代码 #pragma once #pragma execution_character_set("utf-8") #include <qobject.h> #include <QString> #include <QMessageBox> #include "windows.h" #include "wimgapi.h" class BackUp:public QObject { Q_OBJECT public slots: void doBackUp(const QString &pszWimFile,const QString &pszCaptureDir); signals: void resultReady(const QString &result); };
//backup.cpp #include "BackUp.h" #include <QMessageBox> void BackUp::doBackUp(const QString &pszWimFile, const QString &pszCaptureDir) { bool sucess = true; const wchar_t * WimFile = reinterpret_cast<const wchar_t *>(pszWimFile.utf16()); const wchar_t * CaptureDir = reinterpret_cast<const wchar_t *>(pszCaptureDir.utf16()); DWORD dwFlags = 0, dwDisposition = WIM_CREATE_ALWAYS, dwDesiredAccess = WIM_GENERIC_WRITE, dwCreateFlags = 0, dwCaptureFlags = WIM_FLAG_VERIFY, dwCompressionType = WIM_COMPRESS_LZX, dwCreationResult = 0, dwError = 0; HANDLE hWim = WIMCreateFile(WimFile, dwDesiredAccess, dwDisposition, dwCreateFlags, dwCompressionType, &dwCreationResult); sucess = hWim; if (!sucess) { //QMessageBox::critical(this, QString("Error!"), QString("BackUp Error with code Create Fail")); //QMessageBox::critical(NULL, QString("Error!"), QString("Create Image Fail")); emit resultReady(QString("文件创建失败")); return; } //WIMSetTemporaryPath(hWim, pszTmpDir); HANDLE hImage = WIMCaptureImage(hWim, CaptureDir, dwCaptureFlags); sucess = hImage; if (!sucess) { //QMessageBox::critical(NULL, QString("Error!"), QString("BackUp Error with code BackUp Status")); emit resultReady(QString("备份失败")); return; } WIMCloseHandle(hImage); WIMCloseHandle(hWim); emit resultReady(QString("备份成功")); }
//restore.h 代码 #pragma once #pragma execution_character_set("utf-8") #include <QObject> #include <QString> #include "Windows.h" #include "wimgapi.h" class Restore : public QObject { Q_OBJECT public slots: void doRestore(const QString &pszWimFile, const QString &pszCaptureDir); signals: void resultReady(const QString &result); }; //restore.cpp #include "Restore.h" void Restore::doRestore(const QString &pszWimFile, const QString &pszCaptureDir) { bool sucess = true; const wchar_t * WimFile = reinterpret_cast<const wchar_t *>(pszWimFile.utf16()); const wchar_t * pszApplyDir = reinterpret_cast<const wchar_t *>(pszCaptureDir.utf16()); HANDLE hWim = NULL, hImage = NULL; DWORD dwCreateFlags = 0, dwCreateResult = 0, dwImageIndex = 1, dwError = 0; WIM_INFO WimInfo = { 0 }; hWim = WIMCreateFile(WimFile, // Existing .wim file to append the image to WIM_GENERIC_READ, // Access mode WIM_OPEN_EXISTING, // Open disposition dwCreateFlags, 0, // Compression type is ignored for WIM_OPEN_EXISTING. &dwCreateResult); sucess = hWim; if (!sucess) { //QMessageBox::critical(this, QString("Error!"), QString("BackUp Error with code Create Fail")); //QMessageBox::critical(NULL, QString("Error!"), QString("Create Image Fail")); emit resultReady(QString("镜像读取失败")); return; } WIMGetAttributes(hWim, &WimInfo, sizeof(WimInfo)); if (WimInfo.ImageCount < dwImageIndex) { emit resultReady(QString("镜像读取失败")); return; } WIMSetTemporaryPath(hWim, pszApplyDir); hImage = WIMLoadImage(hWim, dwImageIndex); sucess = hImage; if (!sucess) { //QMessageBox::critical(NULL, QString("Error!"), QString("BackUp Error with code BackUp Status")); emit resultReady(QString("不能获取镜像索引")); return; } WIMApplyImage(hImage, pszApplyDir, WIM_FLAG_FILEINFO); //sucess = WIMApplyImage(hImage, pszApplyDir, WIM_FLAG_NO_APPLY); if (!sucess) { //QMessageBox::critical(NULL, QString("Error!"), QString("BackUp Error with code BackUp Status")); emit resultReady(QString("镜像还原失败")); return; } WIMCloseHandle(hImage); WIMCloseHandle(hWim); emit resultReady(QString("镜像还原成功")); }
bcdboot 代码就不贴了,简单的都不不好意思。。。
附件有详细代码 测试环境 vs2017+qt msvc2017 +wadk环境,由于制作内部使用,因此界面比较丑陋,有时间优化吧
运行效果图
附件链接:代码
原文地址:http://blog.51cto.com/amgodchan/2095438
时间: 2024-10-05 05:44:43