|
37 | 37 | //! Note that this module has a #[cfg(windows)] above it as none of this logic
|
38 | 38 | //! is required on Unix.
|
39 | 39 |
|
40 |
| -extern crate kernel32; |
41 |
| -extern crate winapi; |
| 40 | +#![allow(bad_style, dead_code)] |
42 | 41 |
|
43 | 42 | use std::env;
|
44 | 43 | use std::io;
|
45 | 44 | use std::mem;
|
46 | 45 |
|
47 |
| -use self::winapi::*; |
48 |
| -use self::kernel32::*; |
| 46 | +type HANDLE = *mut u8; |
| 47 | +type BOOL = i32; |
| 48 | +type DWORD = u32; |
| 49 | +type LPHANDLE = *mut HANDLE; |
| 50 | +type LPVOID = *mut u8; |
| 51 | +type JOBOBJECTINFOCLASS = i32; |
| 52 | +type SIZE_T = usize; |
| 53 | +type LARGE_INTEGER = i64; |
| 54 | +type ULONG_PTR = usize; |
| 55 | +type ULONGLONG = u64; |
| 56 | + |
| 57 | +const FALSE: BOOL = 0; |
| 58 | +const DUPLICATE_SAME_ACCESS: DWORD = 0x2; |
| 59 | +const PROCESS_DUP_HANDLE: DWORD = 0x40; |
| 60 | +const JobObjectExtendedLimitInformation: JOBOBJECTINFOCLASS = 9; |
| 61 | +const JOB_OBJECT_LIMIT_KILL_ON_JOB_CLOSE: DWORD = 0x2000; |
| 62 | + |
| 63 | +extern "system" { |
| 64 | + fn CreateJobObjectW(lpJobAttributes: *mut u8, lpName: *const u8) -> HANDLE; |
| 65 | + fn CloseHandle(hObject: HANDLE) -> BOOL; |
| 66 | + fn GetCurrentProcess() -> HANDLE; |
| 67 | + fn OpenProcess(dwDesiredAccess: DWORD, |
| 68 | + bInheritHandle: BOOL, |
| 69 | + dwProcessId: DWORD) -> HANDLE; |
| 70 | + fn DuplicateHandle(hSourceProcessHandle: HANDLE, |
| 71 | + hSourceHandle: HANDLE, |
| 72 | + hTargetProcessHandle: HANDLE, |
| 73 | + lpTargetHandle: LPHANDLE, |
| 74 | + dwDesiredAccess: DWORD, |
| 75 | + bInheritHandle: BOOL, |
| 76 | + dwOptions: DWORD) -> BOOL; |
| 77 | + fn AssignProcessToJobObject(hJob: HANDLE, hProcess: HANDLE) -> BOOL; |
| 78 | + fn SetInformationJobObject(hJob: HANDLE, |
| 79 | + JobObjectInformationClass: JOBOBJECTINFOCLASS, |
| 80 | + lpJobObjectInformation: LPVOID, |
| 81 | + cbJobObjectInformationLength: DWORD) -> BOOL; |
| 82 | +} |
| 83 | + |
| 84 | +#[repr(C)] |
| 85 | +struct JOBOBJECT_EXTENDED_LIMIT_INFORMATION { |
| 86 | + BasicLimitInformation: JOBOBJECT_BASIC_LIMIT_INFORMATION, |
| 87 | + IoInfo: IO_COUNTERS, |
| 88 | + ProcessMemoryLimit: SIZE_T, |
| 89 | + JobMemoryLimit: SIZE_T, |
| 90 | + PeakProcessMemoryUsed: SIZE_T, |
| 91 | + PeakJobMemoryUsed: SIZE_T, |
| 92 | +} |
| 93 | + |
| 94 | +#[repr(C)] |
| 95 | +struct IO_COUNTERS { |
| 96 | + ReadOperationCount: ULONGLONG, |
| 97 | + WriteOperationCount: ULONGLONG, |
| 98 | + OtherOperationCount: ULONGLONG, |
| 99 | + ReadTransferCount: ULONGLONG, |
| 100 | + WriteTransferCount: ULONGLONG, |
| 101 | + OtherTransferCount: ULONGLONG, |
| 102 | +} |
| 103 | + |
| 104 | +#[repr(C)] |
| 105 | +struct JOBOBJECT_BASIC_LIMIT_INFORMATION { |
| 106 | + PerProcessUserTimeLimit: LARGE_INTEGER, |
| 107 | + PerJobUserTimeLimit: LARGE_INTEGER, |
| 108 | + LimitFlags: DWORD, |
| 109 | + MinimumWorkingsetSize: SIZE_T, |
| 110 | + MaximumWorkingsetSize: SIZE_T, |
| 111 | + ActiveProcessLimit: DWORD, |
| 112 | + Affinity: ULONG_PTR, |
| 113 | + PriorityClass: DWORD, |
| 114 | + SchedulingClass: DWORD, |
| 115 | +} |
49 | 116 |
|
50 | 117 | pub unsafe fn setup() {
|
51 | 118 | // Create a new job object for us to use
|
|
0 commit comments