3 files changed, 234 insertions, 0 deletions

diff --git a/private/sdktools/logoff/logoff.c b/private/sdktools/logoff/logoff.c
new file mode 100644
index 000000000..c103a896d
--- /dev/null
+++ b/private/sdktools/logoff/logoff.c
@@ -0,0 +1,195 @@
+#include <windows.h>
+#include <stdlib.h>
+#include <stdio.h>
+
+#define ERR(s, c) if (opterr){ (void) fprintf(stderr, "%s%s%c\n", argv[0], s, c); }
+
+int getopt(int argc, char **argv, char *opts);
+BOOL fixpriv(void);
+
+int _cdecl
+main(int argc, char ** argv)
+{
+    int c;
+    int err = 0;
+    int flags = EWX_LOGOFF;
+
+    extern int optind;
+
+    while (!err && (c = getopt(argc, argv, "qQfFrRsS")) != EOF) {
+
+        switch (c) {
+
+        case 'f':
+        case 'F':
+        case 'q':
+        case 'Q':
+            flags |= EWX_FORCE;
+            break;
+
+        case 'r':
+        case 'R':
+            flags |= EWX_REBOOT;
+            break;
+
+        case 's':
+        case 'S':
+            flags |= EWX_SHUTDOWN;
+            break;
+
+        default:
+        case '?':
+            err = 1;
+            break;
+
+        }
+
+    }
+
+    if (err || optind < argc) {
+        fprintf(stderr, "usage: %s [-f] [{-r|-s}]\n", argv[0]);
+        if (c != '?') {
+            exit(1);
+        }
+        printf("With no options, just log off.\n");
+        printf("-f or -q forces operation with no popups.\n");
+        printf("-r Reboot the system\n");
+        printf("-s Shut down the system\n");
+        exit(0);
+    }
+
+    if ( (flags & (EWX_SHUTDOWN | EWX_REBOOT)) == (EWX_SHUTDOWN | EWX_REBOOT)) {
+        fprintf(stderr, "%s: -r and -s flags are exclusive", argv[0]);
+        exit(1);
+    }
+
+    if (flags & (EWX_SHUTDOWN | EWX_REBOOT)) {
+        if (!fixpriv()) {
+            fprintf(stderr, "%s: can't get permission for operation\n", argv[0]);
+            exit(1);
+        }
+    }
+
+    ExitWindowsEx(flags, 0);
+
+    return 0;
+}
+
+
+BOOL
+fixpriv(void)
+{
+    HANDLE  hToken;
+    LUID    ShutdownValue;
+    TOKEN_PRIVILEGES tkp;
+
+    if (!OpenProcessToken(GetCurrentProcess(),
+         TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken)) {
+        return FALSE;
+    }
+
+    if (!LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME, &ShutdownValue)) {
+        return FALSE;
+    }
+
+    tkp.PrivilegeCount = 1;
+    tkp.Privileges[0].Luid = ShutdownValue;
+    tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
+
+    AdjustTokenPrivileges(hToken,
+        FALSE, &tkp, sizeof(TOKEN_PRIVILEGES), NULL, NULL);
+
+    if (GetLastError() != ERROR_SUCCESS) {
+        return FALSE;
+    }
+
+    return TRUE;
+}
+
+
+/*
+**	@(#)getopt.c	2.5 (smail) 9/15/87
+*/
+
+/*
+ * Here's something you've all been waiting for:  the AT&T public domain
+ * source for getopt(3).  It is the code which was given out at the 1985
+ * UNIFORUM conference in Dallas.  I obtained it by electronic mail
+ * directly from AT&T.  The people there assure me that it is indeed
+ * in the public domain.
+ * 
+ * There is no manual page.  That is because the one they gave out at
+ * UNIFORUM was slightly different from the current System V Release 2
+ * manual page.  The difference apparently involved a note about the
+ * famous rules 5 and 6, recommending using white space between an option
+ * and its first argument, and not grouping options that have arguments.
+ * Getopt itself is currently lenient about both of these things White
+ * space is allowed, but not mandatory, and the last option in a group can
+ * have an argument.  That particular version of the man page evidently
+ * has no official existence, and my source at AT&T did not send a copy.
+ * The current SVR2 man page reflects the actual behavor of this getopt.
+ * However, I am not about to post a copy of anything licensed by AT&T.
+ */
+
+
+#if 0
+/*LINTLIBRARY*/
+#define NULL	0
+#define EOF	(-1)
+#define ERR(s, c)	if(opterr){\
+	extern int write();\
+	char errbuf[2];\
+	errbuf[0] = c; errbuf[1] = '\n';\
+	(void) write(2, argv[0], (unsigned)strlen(argv[0]));\
+	(void) write(2, s, (unsigned)strlen(s));\
+	(void) write(2, errbuf, 2);}
+#endif
+
+int	opterr = 1;
+int	optind = 1;
+int	optopt;
+char	*optarg;
+
+int
+getopt(int argc, char **argv, char *opts)
+{
+	static int sp = 1;
+	register int c;
+	register char *cp;
+
+	if(sp == 1)
+		if(optind >= argc ||
+		   (argv[optind][0] != '-' && argv[optind][0] != '/') || argv[optind][1] == '\0')
+			return(EOF);
+		else if(strcmp(argv[optind], "--") == 0) {
+			optind++;
+			return(EOF);
+		}
+	optopt = c = argv[optind][sp];
+	if(c == ':' || (cp=strchr(opts, c)) == NULL) {
+		ERR(": illegal option -- ", c);
+		if(argv[optind][++sp] == '\0') {
+			optind++;
+			sp = 1;
+		}
+		return('?');
+	}
+	if(*++cp == ':') {
+		if(argv[optind][sp+1] != '\0')
+			optarg = &argv[optind++][sp+1];
+		else if(++optind >= argc) {
+			ERR(": option requires an argument -- ", c);
+			sp = 1;
+			return('?');
+		} else
+			optarg = argv[optind++];
+		sp = 1;
+	} else {
+		if(argv[optind][++sp] == '\0') {
+			sp = 1;
+			optind++;
+		}
+		optarg = NULL;
+	}
+	return(c);
+}
diff --git a/private/sdktools/logoff/makefile b/private/sdktools/logoff/makefile
new file mode 100644
index 000000000..6ee4f43fa
--- /dev/null
+++ b/private/sdktools/logoff/makefile
@@ -0,0 +1,6 @@
+#
+# DO NOT EDIT THIS FILE!!!  Edit .\sources. if you want to add a new source
+# file to this component.  This file merely indirects to the real make file
+# that is shared by all the components of NT OS/2
+#
+!INCLUDE $(NTMAKEENV)\makefile.def
diff --git a/private/sdktools/logoff/sources b/private/sdktools/logoff/sources
new file mode 100644
index 000000000..3928beafe
--- /dev/null
+++ b/private/sdktools/logoff/sources
@@ -0,0 +1,33 @@
+!IF 0
+
+Copyright (c) 1993  Microsoft Corporation
+
+Module Name:
+
+    sources.
+
+Abstract:
+
+    build template for logoff
+
+Author:
+
+    Kent Forschmiedt (a-kentf) 03-Mar-1993
+
+!ENDIF
+
+
+
+MAJORCOMP=utils
+MINORCOMP=logoff
+
+TARGETNAME=logoff
+TARGETPATH=obj
+TARGETTYPE=UMAPPL_NOLIB
+### USE_CRTDLL=1
+
+SOURCES=logoff.c
+
+UMTYPE=console
+UMAPPL=logoff
+UMLIBS=$(BASEDIR)\public\sdk\lib\*\user32.lib