88 lines
1.8 KiB
C
88 lines
1.8 KiB
C
|
/*****************************************************************/
|
||
|
|
/** Microsoft LAN Manager **/
|
||
|
|
/** Copyright(c) Microsoft Corp., 1988-1991 **/
|
||
|
|
/*****************************************************************/
|
||
|
|
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <process.h>
|
||
|
|
#include <setjmp.h>
|
||
|
|
|
||
|
|
#include <time.h>
|
||
|
|
|
||
|
|
#include <nt.h>
|
||
|
|
#include <ntrtl.h>
|
||
|
|
#include <nturtl.h>
|
||
|
|
#include <windows.h>
|
||
|
|
|
||
|
|
|
||
|
|
struct {
|
||
|
|
ULONG Flags;
|
||
|
|
PUCHAR String;
|
||
|
|
} EsFlags[] = {
|
||
|
|
ES_CONTINUOUS, "Continuous",
|
||
|
|
ES_SYSTEM_REQUIRED, "System",
|
||
|
|
ES_DISPLAY_REQUIRED, "Display",
|
||
|
|
0, NULL
|
||
|
|
} ;
|
||
|
|
|
||
|
|
|
||
|
|
PUCHAR
|
||
|
|
EsState (
|
||
|
|
IN EXECUTION_STATE State
|
||
|
|
)
|
||
|
|
{
|
||
|
|
static UCHAR text[200];
|
||
|
|
ULONG i;
|
||
|
|
PUCHAR p;
|
||
|
|
UCHAR c;
|
||
|
|
|
||
|
|
p = text;
|
||
|
|
c = '(';
|
||
|
|
for (i=0; EsFlags[i].Flags; i++) {
|
||
|
|
if (State & EsFlags[i].Flags) {
|
||
|
|
p += sprintf (p, "%c%s", c, EsFlags[i].String);
|
||
|
|
c = '|';
|
||
|
|
}
|
||
|
|
}
|
||
|
|
p += sprintf (p, ")");
|
||
|
|
return text;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
VOID __cdecl
|
||
|
|
main (argc, argv)
|
||
|
|
int argc;
|
||
|
|
char *argv[];
|
||
|
|
{
|
||
|
|
UCHAR s[200];
|
||
|
|
EXECUTION_STATE State, PriorState;
|
||
|
|
NTSTATUS Status;
|
||
|
|
|
||
|
|
for (; ;) {
|
||
|
|
printf ("con sys disp: ");
|
||
|
|
gets (s);
|
||
|
|
|
||
|
|
State = 0;
|
||
|
|
if (strstr(s, "con")) State |= ES_CONTINUOUS;
|
||
|
|
if (strstr(s, "disp")) State |= ES_DISPLAY_REQUIRED;
|
||
|
|
if (strstr(s, "sys")) State |= ES_SYSTEM_REQUIRED;
|
||
|
|
|
||
|
|
Status = NtSetThreadExecutionState (
|
||
|
|
State,
|
||
|
|
&PriorState
|
||
|
|
);
|
||
|
|
|
||
|
|
if (NT_SUCCESS(Status)) {
|
||
|
|
|
||
|
|
printf ("PriorState:%x, Set:%x\n", PriorState, State);
|
||
|
|
|
||
|
|
//printf ("PriorState%s, ", EsState(PriorState));
|
||
|
|
//printf ("Set%s\n", EsState(State));
|
||
|
|
|
||
|
|
|
||
|
|
} else {
|
||
|
|
printf ("Failed: %x\n", Status);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|