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;
}

赞 (0)

相关推荐

  • Shell脚本实现C语言代码行数统计

    写了一个比较粗糙的C语言代码行数统计脚本,目前还有些bug,而且效率也不高.脚本主要就是去除大部分的注释后统计行数,相当于做了一部分预处理的工作.下面是代码: #!/bin/bash filename=$1 echo "`whoami`" if [ $# -lt 1 ];then echo "usage : ./scripts filename" exit -1 fi if [ ! -f $filename ];then echo "$filename i

  • C语言中使用lex统计文本文件字符数

    我曾经在Linux上写的一个C程序,借助Lex做词法分析来同时统计N个文本文件的字符数,单词数和行数.让我觉得Lex确实挺有意思的.确实Lex的功能非常强大,用来做小巧的词法分析非常适合,也非常好用.这个程序参考了<Lex与Yacc>上的一个例子. %{ unsigned int char_count = 0, word_count = 0, line_count = 0; %} %% [^ /t/n]+ {word_count++; char_count+=yyleng;}; /n {cha

  • C语言实现的统计素数并求和代码分享

    题目来源于PAT平台,此题又是费了一番脑子.题目要求输出给定区间内的素数个数并对他们求和.具体思路是利用循环判断素数,将结果传递给控制变量,由控制变量再来判断是否执行自增以及求和.当然这里必须要注意1既不是素数也不是合数. 下面是代码: 复制代码 代码如下: #include <stdio.h> int main () {  int a=0,b=0;  int n=0,sum=0;  int x=0,i=0;  scanf("%d %d",&a,&b);  

  • C#统计C、C++及C#程序代码行数的方法

    C#统计C、C++及C#程序代码行数的方法

    本文实例讲述了C#统计C.C++及C#程序代码行数的方法.分享给大家供大家参考.具体如下: 本文中的两个函数 1)用于统计扩展名为 .h .c .cpp .cs 文件的代码行数 public static int LinesOfCode(string filename) 2)用于递归统计一个文件夹内所有扩展名为 .h .c .cpp .cs 文件的代码行数 public static int LinesOfFolder(string foldername) 一.什么样的情况算一行代码 需要注意如

  • C语言实现的统计php代码行数功能源码（支持文件夹、多目录）

    C语言实现的统计php代码行数功能源码（支持文件夹、多目录）

    放假在家没事,睡过懒觉,看过电影,就想起来写个小程序. 统计php代码的行数,对于phper还是挺实用的.支持单个文件和目录.下面是代码和演示的例子! /**  * @date     2012-12-1  * @author bright  * @todo     统计php代码行数  */ #include <stdio.h> #include <stdlib.h> #include <string.h> #include <unistd.h> #inc

  • 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++实现第K顺序统计量的求解方法

    C++实现第K顺序统计量的求解方法

    一个n个元素组成的集合中,第K个顺序统计量(Order Statistic)指的是该集合中第K小的元素,我们这里要讨论的是如何在线性时间(linear time)里找出一个数组的第K个顺序统计量.该问题的算法对于C++程序员来说有一定的借鉴价值.具体如下: 一.问题描述: 问题:给定一个含有n个元素的无序数组,找出第k小的元素. k = 1 :最小值 k = n :最大值 k = ⌊(n+1)/2⌋ or ⌈(n+1)/2⌉ :中位数 找最大值或最小值很简单,只需要遍历一次数组并记录下最大值或最

  • C语言编程中统计输入的行数以及单词个数的方法

    统计输入的行数 标准库保证输入文本流以行序列的形式出现,每一行均以换行符结束.因此,统计行数等价于统计换行符的个数. #include <stdio.h> /* count lines in input */ main() { int c, nl; nl = 0; while ((c = getchar()) != EOF) if (c == '\n') ++nl; printf("%d\n", nl); } 在该程序中,while 循环语句的循环体是一个 if 语句,它控

  • C语言实现统计字符串单词数

    字符串单词数.c #include<stdio.h> #define BUFFERSIZE 1024 int main() { char string[BUFFERSIZE]; int i,count=0,word=0; char c; gets(string) ; for(i=0;(c=string[i])!='\0';i++) { if(c==' ') word=0; else if(word==0) { word=1; count++; } } printf("%d \n&qu

  • C++统计中英文大小写字母、数字、空格及其他字符个数的方法

    C++统计中英文大小写字母、数字、空格及其他字符个数的方法

    本文实例讲述了C++统计中英文大小写字母.数字.空格及其他字符个数的方法.分享给大家供大家参考,具体如下: /* * 作 者: 刘同宾 * 完成日期:2012 年 11 月 28 日 * 版 本 号:v1.0 * 输入描述: * 问题描述: 有一篇文章,共有三行文字,每行有80个字符.要求分别统计出其中英文大写字母.小写字母.数字.空格以及其他字符的个数. * 程序输出: * 问题分析:略 * 算法设计:略 */ #include<iostream> using namespace std;

  • c++统计文件中字符个数代码汇总

    我们先来看看下面的代码: #include<iostream> #include<fstream> #include<cstdlib> using namespace std; class CntCharacters { private: int cnt; public: CntCharacters():cnt(0){} ~CntCharacters(){} void opentxt(char* p) { ifstream fin; fin.open(p,ios_bas

随机推荐