job.c File Reference

#include <k32.h>
#include <winspool.h>
#include <debug.h>

Include dependency graph for job.c:

Go to the source code of this file.

Macros
#define	NDEBUG

Functions
HANDLE WINAPI	CreateJobObjectA (IN LPSECURITY_ATTRIBUTES lpJobAttributes, IN LPCSTR lpName)
HANDLE WINAPI	CreateJobObjectW (IN LPSECURITY_ATTRIBUTES lpJobAttributes, IN LPCWSTR lpName)
HANDLE WINAPI	OpenJobObjectW (IN DWORD dwDesiredAccess, IN BOOL bInheritHandle, IN LPCWSTR lpName)
HANDLE WINAPI	OpenJobObjectA (IN DWORD dwDesiredAccess, IN BOOL bInheritHandle, IN LPCSTR lpName)
BOOL WINAPI	IsProcessInJob (IN HANDLE ProcessHandle, IN HANDLE JobHandle, OUT PBOOL Result)
BOOL WINAPI	AssignProcessToJobObject (IN HANDLE hJob, IN HANDLE hProcess)
BOOL WINAPI	QueryInformationJobObject (IN HANDLE hJob, IN JOBOBJECTINFOCLASS JobObjectInformationClass, IN LPVOID lpJobObjectInformation, IN DWORD cbJobObjectInformationLength, OUT LPDWORD lpReturnLength)
BOOL WINAPI	SetInformationJobObject (IN HANDLE hJob, IN JOBOBJECTINFOCLASS JobObjectInformationClass, IN LPVOID lpJobObjectInformation, IN DWORD cbJobObjectInformationLength)
BOOL WINAPI	TerminateJobObject (IN HANDLE hJob, IN UINT uExitCode)
BOOL WINAPI	CreateJobSet (IN ULONG NumJob, IN PJOB_SET_ARRAY UserJobSet, IN ULONG Flags)

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 16 of file job.c.

Function Documentation

AssignProcessToJobObject()

BOOL WINAPI AssignProcessToJobObject	(	IN HANDLE	hJob,
		IN HANDLE	hProcess
	)

Definition at line 99 of file job.c.

{
    NTSTATUS Status;
 
    Status = NtAssignProcessToJobObject(hJob, hProcess);
    if (!NT_SUCCESS(Status))
    {
        BaseSetLastNTError(Status);
        return FALSE;
    }
 
    return TRUE;
}

CreateJobObjectA()

HANDLE WINAPI CreateJobObjectA	(	IN LPSECURITY_ATTRIBUTES	lpJobAttributes,
		IN LPCSTR	lpName
	)

Definition at line 26 of file job.c.

{
    /* Call the W(ide) function */
    ConvertWin32AnsiObjectApiToUnicodeApi(JobObject, lpName, lpJobAttributes);
}

CreateJobObjectW()

HANDLE WINAPI CreateJobObjectW	(	IN LPSECURITY_ATTRIBUTES	lpJobAttributes,
		IN LPCWSTR	lpName
	)

Definition at line 38 of file job.c.

{
    /* Create the NT object */
    CreateNtObjectFromWin32Api(JobObject, JobObject, JOB_OBJECT_ALL_ACCESS, lpJobAttributes, lpName);
}

CreateJobSet()

BOOL WINAPI CreateJobSet	(	IN ULONG	NumJob,
		IN PJOB_SET_ARRAY	UserJobSet,
		IN ULONG	Flags
	)

Definition at line 338 of file job.c.

{
    NTSTATUS Status;
 
    Status = NtCreateJobSet(NumJob, UserJobSet, Flags);
    if (!NT_SUCCESS(Status))
    {
        BaseSetLastNTError(Status);
        return FALSE;
    }
 
    return TRUE;
}

IsProcessInJob()

BOOL WINAPI IsProcessInJob	(	IN HANDLE	ProcessHandle,
		IN HANDLE	JobHandle,
		OUT PBOOL	Result
	)

Definition at line 77 of file job.c.

{
    NTSTATUS Status;
 
    Status = NtIsProcessInJob(ProcessHandle, JobHandle);
    if (NT_SUCCESS(Status))
    {
        *Result = (Status == STATUS_PROCESS_IN_JOB);
        return TRUE;
    }
 
    BaseSetLastNTError(Status);
    return FALSE;
}

OpenJobObjectA()

HANDLE WINAPI OpenJobObjectA	(	IN DWORD	dwDesiredAccess,
		IN BOOL	bInheritHandle,
		IN LPCSTR	lpName
	)

Definition at line 64 of file job.c.

{
    /* Call the W(ide) function */
    ConvertOpenWin32AnsiObjectApiToUnicodeApi(JobObject, dwDesiredAccess, bInheritHandle, lpName);
}

OpenJobObjectW()

HANDLE WINAPI OpenJobObjectW	(	IN DWORD	dwDesiredAccess,
		IN BOOL	bInheritHandle,
		IN LPCWSTR	lpName
	)

Definition at line 50 of file job.c.

{
    /* Open the NT object */
    OpenNtObjectFromWin32Api(JobObject, dwDesiredAccess, bInheritHandle, lpName);
}

QueryInformationJobObject()

BOOL WINAPI QueryInformationJobObject	(	IN HANDLE	hJob,
		IN JOBOBJECTINFOCLASS	JobObjectInformationClass,
		IN LPVOID	lpJobObjectInformation,
		IN DWORD	cbJobObjectInformationLength,
		OUT LPDWORD	lpReturnLength
	)

Definition at line 119 of file job.c.

{
    NTSTATUS Status;
    PVOID JobInfo;
    JOBOBJECT_EXTENDED_LIMIT_INFORMATION LocalInfo;
    ULONG ExpectedSize;
 
    if (JobObjectInformationClass == JobObjectBasicLimitInformation)
    {
        ExpectedSize = sizeof(JOBOBJECT_BASIC_LIMIT_INFORMATION);
        JobInfo = &LocalInfo;
    }
    else if (JobObjectInformationClass == JobObjectExtendedLimitInformation)
    {
        ExpectedSize = sizeof(JOBOBJECT_EXTENDED_LIMIT_INFORMATION);
        JobInfo = &LocalInfo;
    }
    else
    {
        ExpectedSize = cbJobObjectInformationLength;
        JobInfo = lpJobObjectInformation;
    }
 
    if (cbJobObjectInformationLength != ExpectedSize)
    {
        BaseSetLastNTError(STATUS_INFO_LENGTH_MISMATCH);
        return FALSE;
    }
 
    Status = NtQueryInformationJobObject(hJob,
                                         JobObjectInformationClass,
                                         JobInfo,
                                         ExpectedSize,
                                         lpReturnLength);
    if (NT_SUCCESS(Status))
    {
        if (JobInfo != &LocalInfo) return TRUE;
 
        switch (LocalInfo.BasicLimitInformation.PriorityClass)
        {
            case PROCESS_PRIORITY_CLASS_IDLE:
                LocalInfo.BasicLimitInformation.PriorityClass =
                IDLE_PRIORITY_CLASS;
                break;
 
            case PROCESS_PRIORITY_CLASS_BELOW_NORMAL:
                LocalInfo.BasicLimitInformation.PriorityClass =
                BELOW_NORMAL_PRIORITY_CLASS;
                break;
 
            case PROCESS_PRIORITY_CLASS_NORMAL:
                LocalInfo.BasicLimitInformation.PriorityClass =
                NORMAL_PRIORITY_CLASS;
                break;
 
            case PROCESS_PRIORITY_CLASS_ABOVE_NORMAL:
                LocalInfo.BasicLimitInformation.PriorityClass =
                ABOVE_NORMAL_PRIORITY_CLASS;
                break;
 
            case PROCESS_PRIORITY_CLASS_HIGH:
                LocalInfo.BasicLimitInformation.PriorityClass =
                HIGH_PRIORITY_CLASS;
                break;
 
            case PROCESS_PRIORITY_CLASS_REALTIME:
                LocalInfo.BasicLimitInformation.PriorityClass =
                REALTIME_PRIORITY_CLASS;
                break;
 
            default:
                LocalInfo.BasicLimitInformation.PriorityClass =
                NORMAL_PRIORITY_CLASS;
                break;
        }
 
        RtlCopyMemory(lpJobObjectInformation, &LocalInfo, ExpectedSize);
        return TRUE;
    }
 
    BaseSetLastNTError(Status);
    return FALSE;
}

Referenced by test_QueryInformationJobObject().

SetInformationJobObject()

BOOL WINAPI SetInformationJobObject	(	IN HANDLE	hJob,
		IN JOBOBJECTINFOCLASS	JobObjectInformationClass,
		IN LPVOID	lpJobObjectInformation,
		IN DWORD	cbJobObjectInformationLength
	)

Definition at line 212 of file job.c.

{
    NTSTATUS Status;
    PVOID JobInfo;
    JOBOBJECT_EXTENDED_LIMIT_INFORMATION LocalInfo;
    ULONG ExpectedSize;
    PVOID State = NULL;
    ULONG Privilege = SE_INC_BASE_PRIORITY_PRIVILEGE;
 
    if (JobObjectInformationClass == JobObjectBasicLimitInformation)
    {
        ExpectedSize = sizeof(JOBOBJECT_BASIC_LIMIT_INFORMATION);
        JobInfo = &LocalInfo;
    }
    else if (JobObjectInformationClass == JobObjectExtendedLimitInformation)
    {
        ExpectedSize = sizeof(JOBOBJECT_EXTENDED_LIMIT_INFORMATION);
        JobInfo = &LocalInfo;
    }
    else
    {
        ExpectedSize = cbJobObjectInformationLength;
        JobInfo = lpJobObjectInformation;
    }
 
    if (cbJobObjectInformationLength != ExpectedSize)
    {
        BaseSetLastNTError(STATUS_INFO_LENGTH_MISMATCH);
        return FALSE;
    }
 
    if (JobInfo == &LocalInfo)
    {
        RtlCopyMemory(&LocalInfo, lpJobObjectInformation, ExpectedSize);
 
        if (LocalInfo.BasicLimitInformation.LimitFlags &
            JOB_OBJECT_LIMIT_PRIORITY_CLASS)
        {
            switch (LocalInfo.BasicLimitInformation.PriorityClass)
            {
                case IDLE_PRIORITY_CLASS:
                    LocalInfo.BasicLimitInformation.PriorityClass =
                    PROCESS_PRIORITY_CLASS_IDLE;
                    break;
 
                case BELOW_NORMAL_PRIORITY_CLASS:
                    LocalInfo.BasicLimitInformation.PriorityClass =
                    PROCESS_PRIORITY_CLASS_BELOW_NORMAL;
                    break;
 
                case NORMAL_PRIORITY_CLASS:
                    LocalInfo.BasicLimitInformation.PriorityClass =
                    PROCESS_PRIORITY_CLASS_NORMAL;
                    break;
 
                case ABOVE_NORMAL_PRIORITY_CLASS:
                    LocalInfo.BasicLimitInformation.PriorityClass =
                    PROCESS_PRIORITY_CLASS_ABOVE_NORMAL;
                    break;
 
                case HIGH_PRIORITY_CLASS:
                    LocalInfo.BasicLimitInformation.PriorityClass =
                    PROCESS_PRIORITY_CLASS_HIGH;
                    break;
 
                case REALTIME_PRIORITY_CLASS:
                    LocalInfo.BasicLimitInformation.PriorityClass =
                    PROCESS_PRIORITY_CLASS_REALTIME;
                    break;
 
                default:
                    LocalInfo.BasicLimitInformation.PriorityClass =
                    PROCESS_PRIORITY_CLASS_NORMAL;
                    break;
            }
        }
 
        if (LocalInfo.BasicLimitInformation.LimitFlags &
            JOB_OBJECT_LIMIT_WORKINGSET)
        {
            Status = RtlAcquirePrivilege(&Privilege, 1, 0, &State);
        }
    }
 
    Status = NtSetInformationJobObject(hJob,
                                       JobObjectInformationClass,
                                       JobInfo,
                                       ExpectedSize);
    if (NT_SUCCESS(Status))
    {
        if (State != NULL) RtlReleasePrivilege(State);
        return TRUE;
    }
 
    BaseSetLastNTError(Status);
    return FALSE;
}

TerminateJobObject()

BOOL WINAPI TerminateJobObject	(	IN HANDLE	hJob,
		IN UINT	uExitCode
	)

Definition at line 318 of file job.c.

{
    NTSTATUS Status;
 
    Status = NtTerminateJobObject(hJob, uExitCode);
    if (!NT_SUCCESS(Status))
    {
        BaseSetLastNTError(Status);
        return FALSE;
    }
 
    return TRUE;
}