码迷,mamicode.com
首页 > Windows程序 > 详细

Windows核心编程笔记(3)--作业

时间:2015-01-26 17:11:47      阅读:282      评论:0      收藏:0      [点我收藏+]

标签:

/*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;
}
最后,运行该实例时需要从命令行来,因为资源管理器启动会默认加入系统作业。(上文有说明)
技术分享




Windows核心编程笔记(3)--作业

标签:

原文地址:http://blog.csdn.net/mfcing/article/details/43153415

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!