C++统计软件使用时间代码示例_C 语言

复制代码 代码如下:

// FileName: UseSoftTime.h

#pragma once

#include <vector>

struct UseTime
{
    // 开始时间
    SYSTEMTIME startTime;

    // 结束时间
    SYSTEMTIME endTime;

    // 时间差
    SYSTEMTIME subTime;
};

struct UseSoftInfo
{
    // 软件名
    CString SoftName;

    // 软件启动时间;如果在打开我们的软件之前，待监测软件已存在，默认启动时间为我们软件打开的时间
    std::vector<UseTime> useTime;

    // 累计使用时间
    SYSTEMTIME allTime;

    // 使用次数
    unsigned int nCount;

    // 状态
    bool bStatus;
};

class SoftTime
{
public:

    // 软件使用时间
    std::vector<UseSoftInfo> m_SoftUseTime;

private:

    // 定时器处理函数
    static void CALLBACK TimerProc(HWND hWnd,UINT nMsg,UINT nTimerid,DWORD dwTime);

    // 提权
    BOOL EnableDebugPrivilege (BOOL fEnable);

    // 计算时间差
    SYSTEMTIME SubTime (SYSTEMTIME t1, SYSTEMTIME t2);

    // 计算累计时间
    SYSTEMTIME AddTime (SYSTEMTIME t1, SYSTEMTIME t2);

    // 判断是否为系统进程
    bool IsSystemProcess (LPCTSTR lpProName);

    // 初始化时间为0
    void InitTime (SYSTEMTIME& st);

public:
    SoftTime();
    ~SoftTime();

public:

    // 初始化进程信息
    void InitProcessName (void);
};

复制代码 代码如下:

// FileName: UseSoftTime.cpp

#include "stdafx.h"                    // 如果编译出错请删除此行
#include "UseSoftTime.h"

#include "TlHelp32.h"
#include "StrSafe.h"
#include "Psapi.h"
// 防止错误 error LNK2019
#pragma comment(lib, "psapi.lib")

// 全局变量
SoftTime * g_pSoftTime = NULL;

SoftTime::SoftTime()
{
    g_pSoftTime = this;   
}

SoftTime::~SoftTime()
{
    if (g_pSoftTime != NULL)
    {
        g_pSoftTime = NULL;
    }
}

void CALLBACK SoftTime::TimerProc(HWND hWnd,UINT nMsg,UINT nTimerid,DWORD dwTime)
{
    SoftTime * pSoftTimeInfo = (SoftTime*)g_pSoftTime;

    // 当前系统普通进程的进程名称
    std::vector<CString> currentProcessName;

    HANDLE hProcessSnap = NULL;
    HANDLE hProcessDll = NULL;
    BOOL bRet = FALSE;

    // 初始化dwSize为0，不然Process32First执行失败
    PROCESSENTRY32 pe32 = {0};
    MODULEENTRY32 me32;
    LPVOID lpMsgBuf;
    LPVOID lpDisplayBuf;
    DWORD dwError;

    LPCTSTR pszFormat = TEXT("开始服务时遇到错误! %s");

    if(!pSoftTimeInfo->EnableDebugPrivilege(1))
    {
        MessageBox(NULL, _T("提权失败！"), _T("提示"), MB_OK|MB_ICONEXCLAMATION);
    }

    hProcessSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);

    if (hProcessSnap == INVALID_HANDLE_VALUE)
    {
        dwError = GetLastError();
        FormatMessage(
            FORMAT_MESSAGE_ALLOCATE_BUFFER|
            FORMAT_MESSAGE_FROM_SYSTEM|
            FORMAT_MESSAGE_IGNORE_INSERTS,
            NULL,
            dwError,
            MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
            LPTSTR(&lpMsgBuf),
            0,
            NULL);

        lpDisplayBuf = (LPVOID)LocalAlloc(
            LMEM_ZEROINIT,
            (lstrlen((LPCTSTR)lpMsgBuf)+lstrlen(pszFormat))*sizeof(TCHAR));

        // 格式化字符串
        StringCchPrintf(
            (LPTSTR)lpDisplayBuf,
            LocalSize(lpDisplayBuf),            // 字节数
            pszFormat,
            lpMsgBuf);

        CString strTemp;
        strTemp.Format(TEXT("错误编码为:%d"), dwError);
        ::MessageBox(NULL, (LPCTSTR)lpDisplayBuf, strTemp, MB_OK|MB_ICONEXCLAMATION);
        // 清理分配的内存
        LocalFree(lpMsgBuf);
        LocalFree(lpDisplayBuf);

        return;
    }

    pe32.dwSize = sizeof(PROCESSENTRY32);
    Module32First(hProcessSnap, &me32);

    if (Process32First(hProcessSnap, &pe32))
    {
        do
        {    
            WCHAR path[MAX_PATH]={0};

            HMODULE hModule;
            HANDLE hProcess;
            DWORD needed;
            hProcess=OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, false, pe32.th32ProcessID);
            if (hProcess)
            {
                // 枚举进程
                EnumProcessModules(hProcess, &hModule, sizeof(hModule), &needed);

                // 获取进程的全路径
                memset(path, 0, sizeof(WCHAR)*MAX_PATH);
                GetModuleFileNameEx(hProcess, hModule, path, sizeof(WCHAR)*MAX_PATH);

                if (!(pSoftTimeInfo->IsSystemProcess(path)))
                {
                    std::vector<CString>::iterator iter;
                    bool bIsExist = false;
                    for (iter=currentProcessName.begin(); iter!=currentProcessName.end(); iter++)
                    {
                        if (*iter ==pe32.szExeFile)
                        {
                            bIsExist = true;
                        }
                    }

                    if (!bIsExist)
                    {
                        currentProcessName.push_back(pe32.szExeFile);
                    }
                }
            }
        }
        while (Process32Next(hProcessSnap, &pe32));
    }

    // 查找已存在的进程
    std::vector<UseSoftInfo>::iterator iter1;
    std::vector<CString >::iterator iter2;

    for (iter2=currentProcessName.begin(); iter2!=currentProcessName.end(); iter2++)
    {
        bool bIsExist = false;
        int nIndex = 0;
        for (iter1=pSoftTimeInfo->m_SoftUseTime.begin(); iter1!=pSoftTimeInfo->m_SoftUseTime.end(); iter1++, nIndex++)
        {
            // 已存在的进程，更新软件使用时间
            if (*iter2 == iter1->SoftName)
            {
                // 进程已存在；
                bIsExist = true;

                if (iter1->bStatus)
                {
                    SYSTEMTIME st;
                    GetLocalTime(&st);

                    int n = pSoftTimeInfo->m_SoftUseTime[nIndex].nCount - 1;
                    //pSoftTimeInfo->m_SoftUseTime[nIndex].allTime = pSoftTimeInfo->SubTime(pSoftTimeInfo->m_SoftUseTime[nIndex].useTime[n].startTime, st);
                    pSoftTimeInfo->m_SoftUseTime[nIndex].useTime[n].subTime = pSoftTimeInfo->SubTime(pSoftTimeInfo->m_SoftUseTime[nIndex].useTime[n].startTime, st);

                    std::vector<UseTime>::iterator iter;
                    pSoftTimeInfo->InitTime(st);
                    for (iter=pSoftTimeInfo->m_SoftUseTime[nIndex].useTime.begin(); iter!=pSoftTimeInfo->m_SoftUseTime[nIndex].useTime.end(); iter++)
                    {
                        st = pSoftTimeInfo->AddTime(st, iter->subTime);
                    }
                    pSoftTimeInfo->m_SoftUseTime[nIndex].allTime = st;
                }
                else   // 第二次以后打开的情况
                {
                    pSoftTimeInfo->m_SoftUseTime[nIndex].nCount += 1;
                    pSoftTimeInfo->m_SoftUseTime[nIndex].bStatus = true;

                    UseTime useTime;
                    GetLocalTime(&useTime.startTime);
                    pSoftTimeInfo->InitTime(useTime.endTime);
                    pSoftTimeInfo->InitTime(useTime.subTime);
                    pSoftTimeInfo->m_SoftUseTime[nIndex].useTime.push_back(useTime);
                }
            }

        }

        // 新添加的进程
        if (!bIsExist)
        {
            UseSoftInfo infoTemp;
            UseTime useTime;

            GetLocalTime(&useTime.startTime);
            pSoftTimeInfo->InitTime(useTime.endTime);
            pSoftTimeInfo->InitTime(useTime.subTime);
            infoTemp.useTime.push_back(useTime);

            infoTemp.SoftName = pe32.szExeFile;
            pSoftTimeInfo->InitTime(infoTemp.allTime);

            infoTemp.nCount = 1;
            infoTemp.bStatus = true;

            pSoftTimeInfo->m_SoftUseTime.push_back(infoTemp);
        }
    }

    // 查找退出的进程
    int nIndex = 0;
    for (iter1=pSoftTimeInfo->m_SoftUseTime.begin(); iter1!=pSoftTimeInfo->m_SoftUseTime.end(); iter1++, nIndex++)
    {
        bool bIsExist = false;
        for (iter2=currentProcessName.begin(); iter2!=currentProcessName.end(); iter2++)
        {
            if (iter1->SoftName == *iter2)
            {
                bIsExist = true;
            }
        }

        // 退出的进程
        if (!bIsExist && pSoftTimeInfo->m_SoftUseTime[nIndex].bStatus)
        {
            SYSTEMTIME st;
            GetLocalTime(&st);
            pSoftTimeInfo->m_SoftUseTime[nIndex].useTime[pSoftTimeInfo->m_SoftUseTime[nIndex].nCount-1].endTime = st;
            pSoftTimeInfo->m_SoftUseTime[nIndex].useTime[pSoftTimeInfo->m_SoftUseTime[nIndex].nCount-1].subTime = \
                pSoftTimeInfo->SubTime(pSoftTimeInfo->m_SoftUseTime[nIndex].useTime[pSoftTimeInfo->m_SoftUseTime[nIndex].nCount-1].startTime, st);

            std::vector<UseTime>::iterator iter;
            pSoftTimeInfo->InitTime(st);
            for (iter=pSoftTimeInfo->m_SoftUseTime[nIndex].useTime.begin(); iter!=pSoftTimeInfo->m_SoftUseTime[nIndex].useTime.end(); iter++)
            {
                st = pSoftTimeInfo->AddTime(st, iter->subTime);
            }

            //pSoftTimeInfo->m_SoftUseTime[nIndex].allTime = st;
            pSoftTimeInfo->m_SoftUseTime[nIndex].bStatus = false;
        }
    }

 

    // 关闭特权
    pSoftTimeInfo->EnableDebugPrivilege(0);
    // 关闭内核对象
    CloseHandle(hProcessSnap );

}

BOOL SoftTime::EnableDebugPrivilege (BOOL fEnable)
{
    BOOL fOk = FALSE;  
    HANDLE hToken;

    // 得到进程的访问令牌
    if (OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES,&hToken))
    {   
        TOKEN_PRIVILEGES tp;
        tp.PrivilegeCount = 1;
        // 查看系统特权值并返回一个LUID结构体
        LookupPrivilegeValue(NULL, SE_DEBUG_NAME, &tp.Privileges[0].Luid);
        tp.Privileges[0].Attributes = fEnable ? SE_PRIVILEGE_ENABLED : 0;
        // 启用/关闭 特权
        AdjustTokenPrivileges(hToken, FALSE, &tp, sizeof(tp), NULL, NULL);
        fOk = (GetLastError() == ERROR_SUCCESS);
        CloseHandle(hToken);
    }
    else
    {
        return 0;
    }
    return(fOk);
}

void SoftTime::InitProcessName (void)
{
    HANDLE hProcessSnap = NULL;
    HANDLE hProcessDll = NULL;
    BOOL bRet = FALSE;

    // 初始化dwSize为0，不然Process32First执行失败
    PROCESSENTRY32 pe32 = {0};
    MODULEENTRY32 me32;
    LPVOID lpMsgBuf;
    LPVOID lpDisplayBuf;
    DWORD dwError;

    LPCTSTR pszFormat = TEXT("开始服务时遇到错误! %s");

    if(!EnableDebugPrivilege(1))
    {
        MessageBox(NULL, _T("提权失败！"), _T("提示"), MB_OK|MB_ICONEXCLAMATION);
    }

    hProcessSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);

    if (hProcessSnap == INVALID_HANDLE_VALUE)
    {
        dwError = GetLastError();
        FormatMessage(
            FORMAT_MESSAGE_ALLOCATE_BUFFER|
            FORMAT_MESSAGE_FROM_SYSTEM|
            FORMAT_MESSAGE_IGNORE_INSERTS,
            NULL,
            dwError,
            MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
            LPTSTR(&lpMsgBuf),
            0,
            NULL);

        lpDisplayBuf = (LPVOID)LocalAlloc(
            LMEM_ZEROINIT,
            (lstrlen((LPCTSTR)lpMsgBuf)+lstrlen(pszFormat))*sizeof(TCHAR));

        // 格式化字符串
        StringCchPrintf(
            (LPTSTR)lpDisplayBuf,
            LocalSize(lpDisplayBuf),            // 字节数
            pszFormat,
            lpMsgBuf);

        CString strTemp;
        strTemp.Format(TEXT("错误编码为:%d"), dwError);
        ::MessageBox(NULL, (LPCTSTR)lpDisplayBuf, strTemp, MB_OK|MB_ICONEXCLAMATION);
        // 清理分配的内存
        LocalFree(lpMsgBuf);
        LocalFree(lpDisplayBuf);

        return;
    }

    pe32.dwSize = sizeof(PROCESSENTRY32);

    Module32First(hProcessSnap, &me32);

    if (Process32First(hProcessSnap, &pe32))
    {
        do
        {    
            WCHAR path[MAX_PATH]={0};

            HMODULE hModule;
            HANDLE hProcess;
            DWORD needed;
            hProcess=OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, false, pe32.th32ProcessID);
            if (hProcess)
            {
                // 枚举进程
                EnumProcessModules(hProcess, &hModule, sizeof(hModule), &needed);
                // 获取进程的全路径

                memset(path, 0, sizeof(WCHAR)*MAX_PATH);
                GetModuleFileNameEx(hProcess, hModule, path, sizeof(WCHAR)*MAX_PATH);

                // 普通进程
                if (!IsSystemProcess(path))
                {
                    UseSoftInfo infoTemp;
                    UseTime useTime;

                    GetLocalTime(&useTime.startTime);
                    InitTime(useTime.endTime);
                    InitTime(useTime.subTime);
                    infoTemp.useTime.push_back(useTime);

                    infoTemp.SoftName = pe32.szExeFile;

                    std::vector<UseSoftInfo>::iterator iter;
                    bool IsExist = false;
                    for (iter=m_SoftUseTime.begin(); iter!=m_SoftUseTime.end(); iter++)
                    {
                        if (iter->SoftName == infoTemp.SoftName)
                        {
                            IsExist = true;
                        }
                    }

                    if (!IsExist)
                    {
                        InitTime(infoTemp.allTime);
                        infoTemp.nCount = 1;
                        infoTemp.bStatus = true;

                        m_SoftUseTime.push_back(infoTemp);
                    }

                }

            }
        }
        while (Process32Next(hProcessSnap, &pe32));
    }

    // 关闭特权
    EnableDebugPrivilege(0);
    // 关闭内核对象
    CloseHandle(hProcessSnap );

    // 设置一个定时器
    ::SetTimer(NULL, 1, 1000, TimerProc);
}

bool SoftTime::IsSystemProcess (LPCTSTR lpProName)
{
    CString strTemp(lpProName);
    strTemp.MakeLower();

    if (strTemp.Find(_T("windows")) != -1)
    {
        return true;
    }
    else if (strTemp == _T(""))
    {
        return true;
    }
    else if (strTemp.Find(_T("system32")) != -1)
    {
        return true;
    }
    else
    {
        return false;
    }
}

SYSTEMTIME SoftTime::SubTime (SYSTEMTIME t1, SYSTEMTIME t2)
{
    t2.wYear -= t1.wYear;
    t2.wMonth -= t1.wMonth;

    t2.wDay -= t1.wDay;

    if (t2.wHour >= t1.wHour)
    {
        t2.wHour -= t1.wHour;
    }
    else
    {
        t2.wDay -= 1;
        t2.wHour = 24 - t1.wHour + t2.wHour;
    }

    if (t2.wMinute >= t1.wMinute)
    {
        t2.wMinute -= t1.wMinute;
    }
    else
    {
        t2.wHour -= 1;
        t2.wMinute = 60 - t1.wMinute + t2.wMinute;
    }

    if (t2.wSecond >= t1.wSecond)
    {
        t2.wSecond -= t1.wSecond;
    }
    else
    {
        t2.wMinute -= 1;
        t2.wSecond = 60 - t1.wSecond + t2.wSecond;
    }

    return t2;
}

void SoftTime::InitTime (SYSTEMTIME& st)
{
    st.wYear = 0;
    st.wMonth = 0;
    st.wDay = 0;
    st.wHour = 0;
    st.wMinute = 0;
    st.wSecond = 0;
}

SYSTEMTIME SoftTime::AddTime (SYSTEMTIME t1, SYSTEMTIME t2)
{
    t1.wSecond += t2.wSecond;
    if (t1.wSecond >= 60)
    {
        t1.wSecond -= 60;
        t1.wMinute += 1;
    }

    t1.wMinute += t2.wMinute;
    if (t1.wMinute > 60)
    {
        t1.wMinute -= 60;
        t1.wHour += 1;
    }

    t1.wHour += t2.wHour;

    return t1;
}

时间： 2024-08-20 12:45:38

C++统计软件使用时间代码示例_C 语言的相关文章

C语言求矩阵的各列元素之和的代码示例_C 语言

问题描述:统计一个矩阵的各列元素之和.矩阵各元素为整数且绝对值不超过100.要求输入:有多个测试用例,每个测试用例的第一行是空格分隔的两个正整数n和m( 1 < n, m < 80 ),接下来的n行每行有m个空格分隔的整数,组成一个n*m的矩阵.最后一个测试用例n=0 m=0不用处理.要求输出:对每个测试用例,输出一行整数(空格分隔),顺序表示从第1列至第m列的各列元素之和.输入示例: 3 5 1 1 1 2 -1 0 1 0 7 4 0 2 0 -8 -4 2 2 1 1 0 1 0 0 输

C语言实现选择排序、冒泡排序和快速排序的代码示例_C 语言

选择和冒泡 #include<stdio.h> void maopao(int a[],int len){ int i,j,temp; for(i = 0;i < len - 1 ; i ++){//从第一个到倒数第二个 for (j = 0 ; j < len - 1 - i ; j ++)//排在后的是已经排序的 { if (a[j] > a[j + 1])//大的数换到后面去 { temp = a[j]; a[j] = a[j + 1]; a [j + 1] = tem

C语言统计字符个数代码分享_C 语言

C语言实现统计字符个数 #include<stdio.h> int main() { int sz[10]={0},zm[26]={0},z[26]={0},i,space=0,e=0,t=0; char c; printf("请输入一段字符,统计其中各字符的数量\n"); while((c=getchar())!='\n') { if(c<='z'&&c>='a') zm[c-'a']++; else if(c<='Z'&&

C++、python和go语言实现的简单客户端服务器代码示例_C 语言

工作中用到了C/S模型,所做的也无非是给服务器发数据,但开发阶段会遇到程序自身的回环测试,需要用到简单的服务端以便验证数据发送的正确性. 写软件用C++,跑测试用python,这段时间也刚好看go语言,所以都要有demo.以下三组程序实现的功能相同,这里一起做下总结. 一.C++实现 Boost.Asio是一个跨平台的C++库,它用现代C++方法为网络和底层I/O程序提供了一致的异步I/O模型. 为了跨平台,我用boost库实现,具体如下. 服务端代码: 复制代码 代码如下: /*      F

C语言解字符串逆序和单向链表逆序问题的代码示例_C 语言

字符串逆序上次面试碰到一个单向链表逆序的题目,幸好对字符串逆序比较熟悉,类比做出来了.字符串逆序比较简单,直接上代码: void stringReverse(char* p1,char* p2) { if(p1==p2)return; //swap the value of p1 ,p2 *p1=(*p1)+(*p2); *p2=(*p1)-(*p2); *p1=(*p1)-(*p2); if(p1==p2-1)return; else stringReverse(++p1,--p2); } 调

C语言 循环详解及简单代码示例_C 语言

C 循环 有的时候,我们可能需要多次执行同一块代码.一般情况下,语句是按顺序执行的:函数中的第一个语句先执行,接着是第二个语句,依此类推. 编程语言提供了更为复杂执行路径的多种控制结构. 循环语句允许我们多次执行一个语句或语句组,下面是大多数编程语言中循环语句的流程图: 循环类型 C 语言提供了以下几种循环类型.点击链接查看每个类型的细节. 循环类型 描述 while 循环 当给定条件为真时,重复语句或语句组.它会在执行循环主体之前测试条件. for 循环 多次执行一个语句序列,简化管理循环变量

使用C语言打造通讯录管理系统和教学安排系统的代码示例_C 语言

通讯录管理系统实现了通讯录的录入信息.保存信息.插入.删除.排序.查找.单个显示等功能.. 完整的代码如下: #include <stdio.h> #include <malloc.h> //得到指向大小为Size的内存区域的首字节的指针// #include <string.h> #include <stdlib.h> //标准库函数// #define NULL 0 #define LEN sizeof(struct address_list) //计算

C++中实现队列类链式存储与栈类链式存储的代码示例_C 语言

队列类链式存储 代码: linkqueue.hpp  // 队列类 #pragma once #include "linklist.hpp" template <typename T> class LinkQueue { public: LinkQueue(); ~LinkQueue(); public: int clear(); int append(T &t); int retieve(T &t); int header(T &t); int l

桶排序算法的理解及C语言版代码示例_C 语言

理解:桶排序是计数排序的变种,把计数排序中相邻的m个"小桶"放到一个"大桶"中,在分完桶后,对每个桶进行排序(一般用快排),然后合并成最后的结果.基本思想:桶排序假设序列由一个随机过程产生,该过程将元素均匀而独立地分布在区间[0,1)上.我们把区间[0,1)划分成n个相同大小的子区间,称为桶.将n个记录分布到各个桶中去.如果有多于一个记录分到同一个桶中,需要进行桶内排序.最后依次把各个桶中的记录列出来记得到有序序列.效率分析:桶排序的平均时间复杂度为线性的O(N+C