Windows核心编程笔记（3）----作业
/*1、如果进程已经与一个作业相关联，就无法将当前进程及其任何子进程从作业中移除，这个安全特性可以保证
/* 进程无法摆脱对它施加的限制。
/*2、在调试程序时，调试器是从资源管理器启动的，程序会从调试器继承带“PCA”前缀的作业。因此，调试程序
/* 时总是显示进程已经加入了作业。使用命令行来运行程序时就不会有这个问题了。
/*3、关闭一个作业对象，并不会终止作业内所有的进程。作业对象实际上只是加了一个删除标记，只有在作业中
/* 所有进程都终止后，才会自动销毁。
/*4、可以向作业引用以下几种限制：(SetInformationJobObject函数来实现)
4.1 基本限制和额外限制，用于防止作业中的进程独占系统资源；
4.2 基本的UI限制，用于防止作业内的进程更改用户界面；
4.3 安全限额，用于防止作业内的进程访问安全资源（文件、注册表子项等）
/*5、一旦进程已经加入一个作业，就不能再将其加入其他作业，该进程说创建的子进程也将处于其作业的限制内。
/* 可以给作业加上JOB_OBJECT_LIMIT_BREAKAWAY_OK限制，并在创建子进程时加上CREATE_BREAKAWAY_FROM_JOB标
/* 识，这样子进程就不会受到父进程作业的限制了。
/*6、终止作业中所有进程的方法， TerminateJobObject( __in HANDLE hJob, __in UINT uExitCode )函数终止作业。
*/
#include "stdafx.h"
#include <Windows.h>
#include <iostream>
using std::cout;
using std::endl;
BOOL AddThreadToJob();
int _tmain(int argc, _TCHAR* argv[])
{
AddThreadToJob();
return 0;
}
BOOL AddThreadToJob()
{
BOOL bResult = FALSE;
HANDLE hJob = NULL;
try
{
BOOL bInJob = FALSE;
//DWORD dwPid = GetCurrentProcessId();
//这里是获取进程的真实句柄
//HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, FALSE, dwPid);
//GetCurrentProcess()返回的是伪句柄0xffffffff，但是本身是没有错误的
BOOL bRet = IsProcessInJob(GetCurrentProcess(), NULL, &bInJob);
//CloseHandle(hProcess);
if ( bInJob )
{
JOBOBJECT_BASIC_PROCESS_ID_LIST jbril;
memset(&jbril, 0, sizeof(jbril));
DWORD dwLen = 0;
bRet = QueryInformationJobObject(NULL, JobObjectBasicProcessIdList, (LPVOID)&jbril, \
sizeof(JOBOBJECT_BASIC_PROCESS_ID_LIST), &dwLen);
throw L"当前进程已经加入到作业中！";
}
hJob = CreateJobObject(NULL, L"TestJob");
if ( NULL == hJob )
{
throw L"创建作业失败！";
}
if ( GetLastError() == ERROR_ALREADY_EXISTS )
{
throw L"同名作业已经创建！";
}
cout<<"给作业添加限制"<<endl;
//设置作业基本限制
JOBOBJECT_BASIC_LIMIT_INFORMATION jbli = {0};
jbli.PriorityClass = IDLE_PRIORITY_CLASS;
//对于进程占用时间超过分配时间的任何进程，系统将自动终止其运行
jbli.PerJobUserTimeLimit.QuadPart = 10000;
jbli.LimitFlags = JOB_OBJECT_LIMIT_PRIORITY_CLASS|JOB_OBJECT_LIMIT_JOB_TIME;
SetInformationJobObject(hJob, JobObjectBasicLimitInformation, (LPVOID)&jbli, sizeof(JOBOBJECT_BASIC_LIMIT_INFORMATION));
//设置作业UI限制
cout<<"设置作业UI限制"<<endl;
JOBOBJECT_BASIC_UI_RESTRICTIONS jbur;
jbur.UIRestrictionsClass = JOB_OBJECT_UILIMIT_EXITWINDOWS; //不能关闭计算机
jbur.UIRestrictionsClass |= JOB_OBJECT_UILIMIT_HANDLES; //不能访问系统里的其他对象句柄
SetInformationJobObject(hJob, JobObjectBasicUIRestrictions, (LPVOID)&jbur, sizeof(JOBOBJECT_BASIC_UI_RESTRICTIONS));
//首先必须创建进程，然后挂起这个进程使他不能执行任何代码，然后才能加入作业中（确保进程在进入作业前不自信任何代码）
//不然加入作业限制这个进程也就没意义了
STARTUPINFO si;
memset(&si, 0, sizeof(STARTUPINFO));
si.cb = sizeof(STARTUPINFO);
PROCESS_INFORMATION pi;
memset(&pi, 0, sizeof(PROCESS_INFORMATION));
cout<<"创建一个新的进程"<<endl;
bRet = CreateProcess(L"update.exe", NULL, NULL, NULL, FALSE, CREATE_SUSPENDED|CREATE_NEW_CONSOLE, NULL, \
NULL, &si, &pi);
if ( !bRet )
throw L"创建进程失败！";
cout<<"子进程创建完毕"<<endl;
//把这个进程加入到作业中去
AssignProcessToJobObject(hJob, pi.hProcess);
//现在可以恢复这个进程，让他开始执行代码了
ResumeThread(pi.hThread);
CloseHandle(pi.hThread);
//等待进程终止，或者作业所申请的CPU时间都用完
HANDLE h[2];
h[0] = pi.hProcess;
h[1] = hJob;
cout<<"开始等待进程执行完毕或者作业CPU使用完"<<endl;
DWORD dwRet = WaitForMultipleObjects(2, h, FALSE, INFINITE);
switch( dwRet-WAIT_OBJECT_0 )
{
case 0://进程终止
{
int a = 0;
cout<<"进程终止"<<endl;;
break;
}
case 1://作业的CPU时间用完
{
int a = 0;
cout<<"作业的CPU时间用完"<<endl;
break;
}
}
CloseHandle(pi.hProcess);
//获取进程执行时间信息
FILETIME createTime, exitTime, kernelTime, usedTime;
GetProcessTimes(pi.hProcess, &createTime, &exitTime, &kernelTime, &usedTime);
WCHAR szBuffer[256] = {0};
swprintf_s(szBuffer, 256, L"createTime=%u，exitTime=%u，kernelTime=%u，usedTime=%u", \
createTime.dwLowDateTime/1000, exitTime.dwLowDateTime/10000, \
kernelTime.dwLowDateTime/10000, usedTime.dwLowDateTime/10000);
MessageBox(NULL, szBuffer, L"结束：", MB_OK);
bResult = TRUE;
}
catch(WCHAR* pMsg)
{
MessageBox(NULL, pMsg, NULL, 0);
}
if ( hJob )
CloseHandle(hJob);
return bResult;
}
最后，运行该实例时需要从命令行来，因为资源管理器启动会默认加入系统作业。（上文有说明）
