#ifndef MAINWINDOW_H

#define MAINWINDOW_H

#include <QMainWindow>

#include <QMap>

#include <QTimer>

namespace Ui {

class MainWindow;

}

class MainWindow : public QMainWindow

{

    Q_OBJECT

public:

    explicit MainWindow(QWidget *parent = );

    ~MainWindow();

    void getAllAppPidList(QMap<QString, qint64> &app_pid);

    void OpenServiceManager();

private slots:

    void scanTable();

    void on_pushButton_add_progress_clicked();

    void on_pushButton_add_service_clicked();

private:

    Ui::MainWindow *ui;

    QTimer* scanTimer;

};

#endif // MAINWINDOW_H

#include <windows.h>// for OpenService

#include <tlhelp32.h>// for CreateToolhelp32Snapshot

#include <Psapi.h>   // for GetModuleFileNameEx

#include "mainwindow.h"

#include "ui_mainwindow.h"

#include <QDebug>

/*

 *

 * =================================================================

 * ！！！！！！！！！！！！！本程序需要管理员身份运行！！！！！！！！！！！！！！！

 * =================================================================

 *

 * */

//使用带bom的UTF8文件格式，在Tools-Options-Text Editor-Behavior-File Encoding-UTF-8 BOM:Add If Emcoding Is UTF-8

#pragma execution_character_set("utf-8")

SC_HANDLE hSCM;

MainWindow::MainWindow(QWidget *parent) :

    QMainWindow(parent),

    ui(new Ui::MainWindow)

{

    ui->setupUi(this);

    scanTimer = new QTimer(this);

    connect( scanTimer, SIGNAL(timeout()), SLOT(scanTable()) );

    scanTimer->start(  );  // for 100fps

    OpenServiceManager();

}

MainWindow::~MainWindow()

{

    delete ui;

    CloseServiceHandle(hSCM);

}

void MainWindow::scanTable()

{

    //进程

    for(int row=; row<ui->tableWidget_progress->rowCount(); row++)

        for(int col=; col<ui->tableWidget_progress->columnCount();col++)

        {

            QTableWidgetItem* item = ui->tableWidget_progress->item(row,col);

            if(item!=NULL)

            {

                QString app=item->text();

            }

        }

    //检测服务是否在运行

    if(hSCM)

    {

        for(int row=; row<ui->tableWidget_service->rowCount(); row++)

            for(int col=; col<ui->tableWidget_service->columnCount();col++)

            {

                QTableWidgetItem* item = ui->tableWidget_service->item(row,col);

                if(item!=NULL)

                {

                    QString serviceName=item->text();

                    SC_HANDLE hService = ::OpenService( hSCM, serviceName.toStdWString().data(), SERVICE_ALL_ACCESS );

                    if(hService)

                    {

                        SERVICE_STATUS ssStatus;

                        QueryServiceStatus(hService,&ssStatus);//查看该Service的状态

                        if(ssStatus.dwCurrentState==SERVICE_STOPPED)

                        {

                            ::StartService( hService, , NULL );

                            ui->textEdit->append("StartService "+serviceName);

                        }

                        else if(ssStatus.dwCurrentState==SERVICE_RUNNING)

                        {

                            item->setBackground(Qt::green);

                        }

                        CloseServiceHandle(hService);

                    }

                    else

                    {

                        //ui->textEdit->append("OpenService Failed  "+serviceName);

                        item->setBackground(Qt::red);

                    }

                }

            }

    }

}

void MainWindow::OpenServiceManager()

{

    /*

     *

     *

     * 以下服务相关操作需要管理员权限

     *

     *

     *

     * */

    hSCM = ::OpenSCManager(NULL, // local machine

                           NULL, // ServicesActive database

                           SC_MANAGER_ALL_ACCESS); // full access

    if (hSCM) {

        //hService = ::OpenService( hSCM, QString("STEPVR_MMAP_SERVICE").toStdWString().data(), SERVICE_ALL_ACCESS );

        //if(hService==NULL)

        //    ui->textEdit->append("OpenService Failed");

    }

    else

    {

        qDebug()<<"OpenSCManager Fail"<<GetLastError();

        ui->textEdit->append("OpenSCManager Failed");

    }

}

// 根据进程号获取exe所在文件绝对路径

/*QString GetPathByProcessID(DWORD pid)

{

    HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, FALSE, pid);

    if (!hProcess)

    {

        //QMessageBox::warning(NULL,"GetPathByProcessID","无权访问该进程");

        return "";

    }

    WCHAR filePath[MAX_PATH];

    DWORD ret= GetModuleFileNameEx(hProcess, NULL, filePath, MAX_PATH) ;

    QString file = QString::fromStdWString( filePath );

    //QMessageBox::warning(NULL,"GetPathByProcessID ret=", QString::number(ret)+":"+file);

    CloseHandle(hProcess);

    return ret==0?"":file;

}

// 获取机器上正在运行的全部exe

void MainWindow::getAllAppPidList(QMap<QString, qint64> &app_pid)

{

    app_pid.clear();

    PROCESSENTRY32 pe32;

    pe32.dwSize = sizeof(pe32);

    HANDLE hProcessSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS,0);

    if(hProcessSnap == INVALID_HANDLE_VALUE)

    {

        //warningLabel->setText("CreateToolhelp32Snapshot调用失败");

        return ;

    }

    BOOL bMore = Process32First(hProcessSnap,&pe32);

    while(bMore)

    {

        //printf("进程名称：%s\n",pe32.szExeFile);

        //printf("进程ID：%u\n\n",pe32.th32ProcessID);

        QString exeName = (QString::fromUtf16(reinterpret_cast<const unsigned short *>(pe32.szExeFile)));

        QString exePath = GetPathByProcessID( pe32.th32ProcessID );

        exePath = FORMAT_PATH( exePath );

        //qDebug()<<exePath.toLower();

        if( exePath.isEmpty() )

        {

            //warningLabel->setText("获取进程 " + exeName + " 路径失败");

        }

        else

        {

            app_pid[exePath] = pe32.th32ProcessID;

        }

        bMore = Process32Next(hProcessSnap,&pe32);

    }

    CloseHandle(hProcessSnap);

}*/

void MainWindow::on_pushButton_add_progress_clicked()

{

    ui->tableWidget_progress->insertRow(ui->tableWidget_progress->rowCount());

}

void MainWindow::on_pushButton_add_service_clicked()

{

    ui->tableWidget_service->insertRow(ui->tableWidget_service->rowCount());

}