ReactOS 0.4.16-dev-306-g647d351
tasklist.c File Reference
#include "tasklist.h"
Include dependency graph for tasklist.c:

Go to the source code of this file.

Macros

#define OPTION_INVALID   -1
 
#define OPTION_HELP   0
 
#define OPTION_NOHEADER   1
 
#define RES_STR_MAXLEN   64
 

Functions

VOID PrintSplitLine (UINT Length)
 
VOID PrintSpace (UINT Length)
 
VOID PrintString (LPCWSTR String, UINT MaxWidth, BOOL bAlignLeft)
 
VOID PrintResString (HINSTANCE hInstance, UINT uID, UINT MaxWidth, BOOL bAlignLeft)
 
VOID PrintNum (LONGLONG Number, UINT MaxWidth)
 
BOOL PrintMemory (SIZE_T MemorySizeByte, UINT MaxWidth, HINSTANCE hInstance)
 
VOID PrintHeader (HINSTANCE hInstance)
 
BOOL EnumProcessAndPrint (BOOL bNoHeader)
 
INT GetOptionType (LPCWSTR szOption)
 
BOOL ProcessArguments (INT argc, WCHAR **argv)
 
int wmain (int argc, WCHAR **argv)
 

Variables

static PCWSTR OptionList [] = { L"?", L"nh" }
 

Macro Definition Documentation

◆ OPTION_HELP

#define OPTION_HELP   0

Definition at line 15 of file tasklist.c.

◆ OPTION_INVALID

#define OPTION_INVALID   -1

Definition at line 14 of file tasklist.c.

◆ OPTION_NOHEADER

#define OPTION_NOHEADER   1

Definition at line 16 of file tasklist.c.

◆ RES_STR_MAXLEN

#define RES_STR_MAXLEN   64

Definition at line 19 of file tasklist.c.

Function Documentation

◆ EnumProcessAndPrint()

BOOL EnumProcessAndPrint ( BOOL  bNoHeader)

Definition at line 144 of file tasklist.c.

145{
146 // Call NtQuerySystemInformation for the process information
147 ULONG ProcessInfoBufferLength = 0;
149 PBYTE ProcessInfoBuffer = NULL;
150
151 // Get the buffer size we need
153
154 // New process/thread might appear before we call for the actual data.
155 // Try to avoid this by retrying several times.
157 {
158 // (Re)allocate buffer
159 ProcessInfoBufferLength = ResultLength;
160 ResultLength = 0;
161 if (ProcessInfoBuffer)
162 {
163 PBYTE NewProcessInfoBuffer = HeapReAlloc(GetProcessHeap(), 0,
164 ProcessInfoBuffer,
165 ProcessInfoBufferLength);
166 if (NewProcessInfoBuffer)
167 {
168 ProcessInfoBuffer = NewProcessInfoBuffer;
169 }
170 else
171 {
172 // out of memory
174 HeapFree(GetProcessHeap(), 0, ProcessInfoBuffer);
175 return FALSE;
176 }
177 }
178 else
179 {
180 ProcessInfoBuffer = HeapAlloc(GetProcessHeap(), 0, ProcessInfoBufferLength);
181 if (!ProcessInfoBuffer)
182 {
183 // out of memory
185 return FALSE;
186 }
187 }
188
189 // Query information
191 ProcessInfoBuffer,
192 ProcessInfoBufferLength,
193 &ResultLength);
195 {
196 break;
197 }
198 }
199
200 if (!NT_SUCCESS(Status))
201 {
202 // tried NT_SYSTEM_QUERY_MAX_RETRY times, or failed with some other reason
204 HeapFree(GetProcessHeap(), 0, ProcessInfoBuffer);
205 return FALSE;
206 }
207
210
211 ConPuts(StdOut, L"\n");
212
213 if (!bNoHeader)
214 {
216 }
217
219 pSPI = (PSYSTEM_PROCESS_INFORMATION)ProcessInfoBuffer;
220 while (pSPI)
221 {
222 PrintString(pSPI->UniqueProcessId ? pSPI->ImageName.Buffer : L"System Idle Process", COLUMNWIDTH_IMAGENAME, TRUE);
223 PrintSpace(1);
225 PrintSpace(1);
227 PrintSpace(1);
229
230 ConPuts(StdOut, L"\n");
231
232 if (pSPI->NextEntryOffset == 0)
233 break;
234 pSPI = (PSYSTEM_PROCESS_INFORMATION)((LPBYTE)pSPI + pSPI->NextEntryOffset);
235 }
236
237 HeapFree(GetProcessHeap(), 0, ProcessInfoBuffer);
238 return TRUE;
239}
void ConPuts(FILE *fp, LPCWSTR psz)
Definition: fc.c:16
#define StdOut
Definition: fc.c:14
#define StdErr
Definition: fc.c:15
#define IDS_OUT_OF_MEMORY
Definition: resource.h:5
#define IDS_ENUM_FAILED
Definition: resource.h:8
LONG NTSTATUS
Definition: precomp.h:26
HINSTANCE hInstance
Definition: charmap.c:19
_In_ PSCSI_REQUEST_BLOCK _Out_ NTSTATUS _Inout_ BOOLEAN * Retry
Definition: classpnp.h:312
#define NULL
Definition: types.h:112
#define TRUE
Definition: types.h:120
#define FALSE
Definition: types.h:117
#define NT_SUCCESS(StatCode)
Definition: apphelp.c:33
#define GetProcessHeap()
Definition: compat.h:736
#define HeapAlloc
Definition: compat.h:733
#define HeapReAlloc
Definition: compat.h:734
#define HeapFree(x, y, z)
Definition: compat.h:735
HMODULE WINAPI GetModuleHandleW(LPCWSTR lpModuleName)
Definition: loader.c:838
#define assert(x)
Definition: debug.h:53
@ SystemProcessInformation
Definition: ntddk_ex.h:16
Status
Definition: gdiplustypes.h:25
unsigned int UINT
Definition: ndis.h:50
struct _SYSTEM_PROCESS_INFORMATION * PSYSTEM_PROCESS_INFORMATION
#define L(x)
Definition: ntvdm.h:50
INT __cdecl ConResMsgPrintf(IN PCON_STREAM Stream, IN DWORD dwFlags, IN UINT uID,...)
Definition: outstream.c:1461
BYTE * PBYTE
Definition: pedump.c:66
NTSYSAPI NTSTATUS NTAPI NtQuerySystemInformation(IN SYSTEM_INFORMATION_CLASS SystemInfoClass, OUT PVOID SystemInfoBuffer, IN ULONG SystemInfoBufferSize, OUT PULONG BytesReturned OPTIONAL)
UNICODE_STRING ImageName
Definition: extypes.h:902
BOOL PrintMemory(SIZE_T MemorySizeByte, UINT MaxWidth, HINSTANCE hInstance)
Definition: tasklist.c:69
VOID PrintHeader(HINSTANCE hInstance)
Definition: tasklist.c:118
VOID PrintString(LPCWSTR String, UINT MaxWidth, BOOL bAlignLeft)
Definition: tasklist.c:39
VOID PrintNum(LONGLONG Number, UINT MaxWidth)
Definition: tasklist.c:61
VOID PrintSpace(UINT Length)
Definition: tasklist.c:31
#define NT_SYSTEM_QUERY_MAX_RETRY
Definition: tasklist.h:24
#define COLUMNWIDTH_MEMUSAGE
Definition: tasklist.h:29
#define COLUMNWIDTH_PID
Definition: tasklist.h:27
#define COLUMNWIDTH_IMAGENAME
Definition: tasklist.h:26
#define COLUMNWIDTH_SESSION
Definition: tasklist.h:28
int32_t INT_PTR
Definition: typedefs.h:64
unsigned char * LPBYTE
Definition: typedefs.h:53
int64_t LONGLONG
Definition: typedefs.h:68
uint32_t ULONG
Definition: typedefs.h:59
uint64_t ULONGLONG
Definition: typedefs.h:67
#define STATUS_INFO_LENGTH_MISMATCH
Definition: udferr_usr.h:133
_Must_inspect_result_ _In_ WDFDEVICE _In_ DEVICE_REGISTRY_PROPERTY _In_ ULONG _Out_ PULONG ResultLength
Definition: wdfdevice.h:3776

Referenced by ProcessArguments().

◆ GetOptionType()

INT GetOptionType ( LPCWSTR  szOption)

Definition at line 241 of file tasklist.c.

242{
243 if (szOption[0] != L'/' && szOption[0] != L'-')
244 {
245 return OPTION_INVALID;
246 }
247 szOption++;
248
249 for (UINT i = 0; i < _countof(OptionList); i++)
250 {
251 if (!_wcsicmp(OptionList[i], szOption))
252 {
253 return i;
254 }
255 }
256 return OPTION_INVALID;
257}
GLsizei GLenum const GLvoid GLsizei GLenum GLbyte GLbyte GLbyte GLdouble GLdouble GLdouble GLfloat GLfloat GLfloat GLint GLint GLint GLshort GLshort GLshort GLubyte GLubyte GLubyte GLuint GLuint GLuint GLushort GLushort GLushort GLbyte GLbyte GLbyte GLbyte GLdouble GLdouble GLdouble GLdouble GLfloat GLfloat GLfloat GLfloat GLint GLint GLint GLint GLshort GLshort GLshort GLshort GLubyte GLubyte GLubyte GLubyte GLuint GLuint GLuint GLuint GLushort GLushort GLushort GLushort GLboolean const GLdouble const GLfloat const GLint const GLshort const GLbyte const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLdouble const GLfloat const GLfloat const GLint const GLint const GLshort const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort GLenum GLenum GLenum GLfloat GLenum GLint GLenum GLenum GLenum GLfloat GLenum GLenum GLint GLenum GLfloat GLenum GLint GLint GLushort GLenum GLenum GLfloat GLenum GLenum GLint GLfloat const GLubyte GLenum GLenum GLenum const GLfloat GLenum GLenum const GLint GLenum GLint GLint GLsizei GLsizei GLint GLenum GLenum const GLvoid GLenum GLenum const GLfloat GLenum GLenum const GLint GLenum GLenum const GLdouble GLenum GLenum const GLfloat GLenum GLenum const GLint GLsizei GLuint GLfloat GLuint GLbitfield GLfloat GLint GLuint GLboolean GLenum GLfloat GLenum GLbitfield GLenum GLfloat GLfloat GLint GLint const GLfloat GLenum GLfloat GLfloat GLint GLint GLfloat GLfloat GLint GLint const GLfloat GLint GLfloat GLfloat GLint GLfloat GLfloat GLint GLfloat GLfloat const GLdouble const GLfloat const GLdouble const GLfloat GLint i
Definition: glfuncs.h:248
_Check_return_ _CRTIMP int __cdecl _wcsicmp(_In_z_ const wchar_t *_Str1, _In_z_ const wchar_t *_Str2)
#define _countof(array)
Definition: sndvol32.h:70
static PCWSTR OptionList[]
Definition: tasklist.c:12
#define OPTION_INVALID
Definition: tasklist.c:14

Referenced by ProcessArguments().

◆ PrintHeader()

VOID PrintHeader ( HINSTANCE  hInstance)

Definition at line 118 of file tasklist.c.

119{
120 if (!hInstance)
121 return;
122
124 PrintSpace(1);
126 PrintSpace(1);
128 PrintSpace(1);
130
131 ConPuts(StdOut, L"\n");
132
134 PrintSpace(1);
136 PrintSpace(1);
138 PrintSpace(1);
140
141 ConPuts(StdOut, L"\n");
142}
#define IDS_HEADER_IMAGENAME
Definition: resource.h:11
#define IDS_HEADER_MEMUSAGE
Definition: resource.h:14
#define IDS_HEADER_PID
Definition: resource.h:12
#define IDS_HEADER_SESSION
Definition: resource.h:13
VOID PrintResString(HINSTANCE hInstance, UINT uID, UINT MaxWidth, BOOL bAlignLeft)
Definition: tasklist.c:48
VOID PrintSplitLine(UINT Length)
Definition: tasklist.c:22

Referenced by EnumProcessAndPrint().

◆ PrintMemory()

BOOL PrintMemory ( SIZE_T  MemorySizeByte,
UINT  MaxWidth,
HINSTANCE  hInstance 
)

Definition at line 69 of file tasklist.c.

70{
71 if (!hInstance)
72 return FALSE;
73
74 SIZE_T MemorySize = MemorySizeByte >> 10;
75
76 WCHAR NumberString[27] = { 0 }; // length 26 is enough to display ULLONG_MAX in decimal with comma, one more for zero-terminated.
77 C_ASSERT(sizeof(SIZE_T) <= 8);
78
79 PWCHAR pNumberStr = NumberString;
80
81 // calculate the length
82 UINT PrintLength = 0;
83 SIZE_T Tmp = MemorySize;
84 UINT Mod = 1;
85
86 do
87 {
88 Tmp /= 10;
89 PrintLength++;
90 Mod *= 10;
91 } while (Tmp);
92
93 for (UINT i = PrintLength; i; i--)
94 {
95 Mod /= 10;
96 *pNumberStr = L'0' + (MemorySize / Mod);
97 MemorySize %= Mod;
98 pNumberStr++;
99
100 if (i != 1 && i % 3 == 1)
101 {
102 *pNumberStr = L',';
103 pNumberStr++;
104 }
105 }
106
107 WCHAR FormatStr[RES_STR_MAXLEN];
108 LoadStringW(hInstance, IDS_MEMORY_STR, FormatStr, _countof(FormatStr));
109
110 WCHAR String[RES_STR_MAXLEN + _countof(NumberString)] = { 0 };
111
112 StringCchPrintfW(String, _countof(String), FormatStr, NumberString);
113 PrintString(String, MaxWidth, FALSE);
114
115 return TRUE;
116}
#define IDS_MEMORY_STR
Definition: resource.h:16
#define C_ASSERT(e)
Definition: intsafe.h:73
STRSAFEAPI StringCchPrintfW(STRSAFE_LPWSTR pszDest, size_t cchDest, STRSAFE_LPCWSTR pszFormat,...)
Definition: strsafe.h:530
static CONST DWORD MemorySize[]
Definition: svga.c:32
#define RES_STR_MAXLEN
Definition: tasklist.c:19
ULONG_PTR SIZE_T
Definition: typedefs.h:80
uint16_t * PWCHAR
Definition: typedefs.h:56
_Must_inspect_result_ _In_ WDFDEVICE _In_ WDFSTRING String
Definition: wdfdevice.h:2433
int WINAPI LoadStringW(_In_opt_ HINSTANCE hInstance, _In_ UINT uID, _Out_writes_to_(cchBufferMax, return+1) LPWSTR lpBuffer, _In_ int cchBufferMax)
__wchar_t WCHAR
Definition: xmlstorage.h:180

Referenced by EnumProcessAndPrint().

◆ PrintNum()

VOID PrintNum ( LONGLONG  Number,
UINT  MaxWidth 
)

Definition at line 61 of file tasklist.c.

62{
63 ConPrintf(StdOut, L"%*lld", MaxWidth, Number);
64}
void ConPrintf(FILE *fp, LPCWSTR psz,...)
Definition: fc.c:20
_In_opt_ PENTER_STATE_SYSTEM_HANDLER _In_opt_ PVOID _In_ LONG _In_opt_ LONG volatile * Number
Definition: ntpoapi.h:207

Referenced by EnumProcessAndPrint().

◆ PrintResString()

VOID PrintResString ( HINSTANCE  hInstance,
UINT  uID,
UINT  MaxWidth,
BOOL  bAlignLeft 
)

Definition at line 48 of file tasklist.c.

49{
50 if (!hInstance)
51 return;
52
55 PrintString(StringBuffer, MaxWidth, bAlignLeft);
56}
WCHAR StringBuffer[156]
Definition: ldrinit.c:41

Referenced by PrintHeader().

◆ PrintSpace()

VOID PrintSpace ( UINT  Length)

Definition at line 31 of file tasklist.c.

32{
33 ConPrintf(StdOut, L"%*ls", (INT)Length, L"");
34}
_In_ ULONG _In_ ULONG _In_ ULONG Length
Definition: ntddpcm.h:102
int32_t INT
Definition: typedefs.h:58

Referenced by EnumProcessAndPrint(), and PrintHeader().

◆ PrintSplitLine()

VOID PrintSplitLine ( UINT  Length)

Definition at line 22 of file tasklist.c.

23{
24 for (; Length; Length--)
25 {
26 ConPuts(StdOut, L"=");
27 }
28}

Referenced by PrintHeader().

◆ PrintString()

VOID PrintString ( LPCWSTR  String,
UINT  MaxWidth,
BOOL  bAlignLeft 
)

Definition at line 39 of file tasklist.c.

40{
41 ConPrintf(StdOut, bAlignLeft ? L"%-*.*ls" : L"%*.*ls", MaxWidth, MaxWidth, String);
42}

Referenced by EnumProcessAndPrint(), PrintMemory(), and PrintResString().

◆ ProcessArguments()

BOOL ProcessArguments ( INT  argc,
WCHAR **  argv 
)

Definition at line 259 of file tasklist.c.

260{
261 BOOL bHasHelp = FALSE, bHasNoHeader = FALSE;
262
263 for (INT i = 1; i < argc; i++)
264 {
265 INT Option = GetOptionType(argv[i]);
266
267 switch (Option)
268 {
269 case OPTION_HELP:
270 {
271 if (bHasHelp)
272 {
273 // -? already specified
276 return FALSE;
277 }
278 bHasHelp = TRUE;
279 break;
280 }
281 case OPTION_NOHEADER:
282 {
283 if (bHasNoHeader)
284 {
285 // -nh already specified
288 return FALSE;
289 }
290 bHasNoHeader = TRUE;
291 break;
292 }
293 case OPTION_INVALID:
294 default:
295 {
298 return FALSE;
299 }
300 }
301 }
302
303 if (bHasHelp)
304 {
305 if (argc > 2) // any arguments other than -? exists
306 {
309 return FALSE;
310 }
311 else
312 {
315 return FALSE;
316 }
317 }
318 else
319 {
320 EnumProcessAndPrint(bHasNoHeader);
321 }
322 return TRUE;
323}
static int argc
Definition: ServiceArgs.c:12
#define IDS_USAGE
Definition: resource.h:3
#define IDS_DESCRIPTION
Definition: resource.h:4
#define IDS_INVALID_OPTION
Definition: resource.h:7
#define IDS_INVALID_SYNTAX
Definition: resource.h:6
#define IDS_OPTION_TOO_MUCH
Definition: resource.h:5
unsigned int BOOL
Definition: ntddk_ex.h:94
#define argv
Definition: mplay32.c:18
#define OPTION_NOHEADER
Definition: tasklist.c:16
BOOL EnumProcessAndPrint(BOOL bNoHeader)
Definition: tasklist.c:144
INT GetOptionType(LPCWSTR szOption)
Definition: tasklist.c:241
#define OPTION_HELP
Definition: tasklist.c:15

Referenced by wmain().

◆ wmain()

int wmain ( int  argc,
WCHAR **  argv 
)

Definition at line 325 of file tasklist.c.

326{
327 /* Initialize the Console Standard Streams */
329
331 {
332 return 1;
333 }
334 return 0;
335}
#define ConInitStdStreams()
Definition: fc.c:13
BOOL ProcessArguments(INT argc, WCHAR **argv)
Definition: tasklist.c:259

Variable Documentation

◆ OptionList

PCWSTR OptionList[] = { L"?", L"nh" }
static

Definition at line 12 of file tasklist.c.

Referenced by BlCopyBootOptions(), BmGetOptionList(), and GetOptionType().