Simple Win32/text application to query the SM (incomplete/untested).

svn path=/trunk/; revision=14621

reactos/subsys/system/sm/makefile

@@ -0,0 +1,28 @@
+# $Id$
+# 
+#  ReactOS Win32 SM Query Tool
+# 
+PATH_TO_TOP = ../../..
+TOOLS_PATH = $(PATH_TO_TOP)/tools
+
+TARGET_TYPE = program
+
+TARGET_APPTYPE = console
+
+TARGET_NAME = sm
+
+TARGET_INSTALLDIR = system32
+
+TARGET_CFLAGS = -D__USE_W32API -DANONYMOUSUNIONS -Wall -Werror
+
+TARGET_OBJECTS = $(TARGET_NAME).o
+
+TARGET_SDKLIBS = smdll.a ntdll.a
+
+include $(PATH_TO_TOP)/rules.mak
+
+include $(TOOLS_PATH)/helper.mk
+
+include $(TOOLS_PATH)/depend.mk
+
+# EOF

reactos/subsys/system/sm/sm.c

@@ -0,0 +1,239 @@
+/*
+ *  ReactOS Win32 Applications
+ *  Copyright (C) 2005 ReactOS Team
+ *
+ *  This program is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation; either version 2 of the License, or
+ *  (at your option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program; if not, write to the Free Software
+ *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ */
+/* $Id$
+ *
+ * COPYRIGHT : See COPYING in the top level directory
+ * PROJECT   : ReactOS/Win32 Session Manager Control Tool
+ * FILE      : subsys/system/sm/sm.c
+ * PROGRAMMER: Emanuele Aliberti (ea@reactos.com)
+ */
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#define NTOS_MODE_USER
+#include <ntos.h>
+#include <sm/helper.h>
+
+#define SM_CMD(n) cmd_##n
+#define SM_CMD_DECL(n) int SM_CMD(n)(int argc, char * argv[])
+#define SM_CMD_CALL(n,c,v) SM_CMD(n)((c),(v))
+
+HANDLE hSmApiPort = (HANDLE) 0;
+
+typedef struct _SM_CMD_DESCRIPTOR
+{
+	const char * Name;
+	int (*EntryPoint)(int,char**);
+	const char * Synopsis;
+	const char * Description;
+	
+} SM_CMD_DESCRIPTOR, *PSM_CMD_DESCRIPTOR;
+
+SM_CMD_DECL(boot);
+SM_CMD_DECL(help);
+SM_CMD_DECL(info);
+SM_CMD_DECL(reboot);
+SM_CMD_DECL(shutdown);
+
+/* internal commands directory */
+SM_CMD_DESCRIPTOR Command [] =
+{
+	{"boot",     SM_CMD(boot),     "boot subsystem_name",   "bootstrap an optional environment subsystem;"},
+	{"help",     SM_CMD(help),     "help [command]",        "print help for command;"},
+	{"info",     SM_CMD(info),     "info [subsystem_id]",   "print information about a booted subsystem\n"
+							        "if subsystem_id is omitted, a list of booted\n"
+							        "environment subsystems is printed."},
+	{"reboot",   SM_CMD(reboot),   "reboot subsystem_id",   "reboot an optional environment subsystem;"},
+	{"shutdown", SM_CMD(shutdown), "shutdown subsystem_id", "shutdown an optional environment subsystem;"},
+};
+
+PSM_CMD_DESCRIPTOR LookupCommand (const char * CommandName)
+{
+	int i;
+	const int command_count = (sizeof Command / sizeof Command[0]);
+	
+	/* parse the command... */
+
+	for (i=0; (i < command_count); i ++)
+	{
+		if (0 == strcmp(CommandName, Command[i].Name))
+		{
+			break;
+		}
+	}
+	if (i == command_count)
+	{
+		fprintf(stderr, "Unknown command '%s'.\n", CommandName);
+		return NULL;
+	}
+	return & Command [i];
+}
+
+/* user commands */
+
+SM_CMD_DECL(boot)
+{
+	int rc = EXIT_SUCCESS;
+	ANSI_STRING ProgramA;
+	UNICODE_STRING ProgramW;
+	NTSTATUS Status = STATUS_SUCCESS;
+
+	if (3 == argc)
+	{
+		RtlInitAnsiString (& ProgramA, argv[2]);
+		RtlAnsiStringToUnicodeString (& ProgramW, & ProgramA, TRUE);
+		Status = SmExecuteProgram (hSmApiPort, & ProgramW);
+		RtlFreeUnicodeString (& ProgramW);
+		if (STATUS_SUCCESS != Status)
+		{
+			printf ("Status 0x%08lx\n", Status);
+		}
+	}
+	else
+	{
+		argv[2]="boot";
+		return SM_CMD_CALL(help,3,argv);
+	}
+	return rc;
+}
+
+SM_CMD_DECL(help)
+{
+	int i = 0;
+	PSM_CMD_DESCRIPTOR cmd = NULL;
+	int rc = EXIT_SUCCESS;
+
+	switch (argc)
+	{
+	case 2:
+		for (i=0; (i < (sizeof Command / sizeof Command[0])); i ++)
+		{
+			printf("%s\n", Command[i].Synopsis);
+		}
+		break;
+	case 3:
+		cmd = LookupCommand (argv[2]);
+		if (NULL == cmd)
+		{
+			rc = EXIT_FAILURE;
+			break;
+		}
+		printf("%s\n%s\n\n%s\n",
+			cmd->Name,
+			cmd->Synopsis,
+			cmd->Description);
+		break;
+	}
+	return rc;
+}
+
+SM_CMD_DECL(info)
+{
+	int rc = EXIT_SUCCESS;
+	NTSTATUS Status = STATUS_SUCCESS;
+	SM_BASIC_INFORMATION bi = {0,};
+	ULONG ReturnDataLength = sizeof bi;
+
+	Status = SmQueryInformation (hSmApiPort,
+				     SmBasicInformation,
+				     & bi,
+				     sizeof bi,
+				     & ReturnDataLength);
+	if (STATUS_SUCCESS == Status)
+	{
+		int i = 0;
+
+		printf ("SSID PID      Flags\n");
+		for (i = 0; i < bi.SubSystemCount; i ++)
+		{
+			printf ("%04x %08lx %04x\n",
+					bi.SubSystem[i].Id,
+			       		bi.SubSystem[i].ProcessId,
+					bi.SubSystem[i].Flags);
+		}
+	}
+	else
+	{
+		printf ("Status 0x%08lx\n", Status);
+		rc = EXIT_FAILURE;
+	}
+	return rc;
+}
+
+SM_CMD_DECL(shutdown)
+{
+	int rc = EXIT_SUCCESS;
+	
+	fprintf(stderr,"not implemented\n");
+	return rc;
+}
+
+SM_CMD_DECL(reboot)
+{
+	int rc = SM_CMD(shutdown)(argc,argv);
+	if(EXIT_SUCCESS == rc)
+	{
+		rc = SM_CMD(boot)(argc,argv);
+	}
+	return rc;
+}
+
+/* print command's synopsys */
+int print_synopsys (int argc, char *argv[])
+{
+	fprintf (stderr, "ReactOS/Win32 Session Manager Control Tool\n\n");
+	printf ("Usage:\n"
+		"\tsm\n"
+		"\tsm help [command]\n"
+		"\tsm command [arguments]\n\n"
+		"'sm help' will print the list of valid commands.\n");
+	return EXIT_SUCCESS;
+}
+
+/* parse and execute */
+int pande (int argc, char *argv[])
+{
+	PSM_CMD_DESCRIPTOR Command = NULL;
+	NTSTATUS Status = STATUS_SUCCESS;
+
+	/* Lookup the user command... */	
+	Command = LookupCommand (argv[1]);
+	if (NULL == Command)
+	{
+		return EXIT_FAILURE;
+	} 
+	/* Connect to the SM in non-registering mode. */
+	Status = SmConnectApiPort (0, 0, 0, & hSmApiPort);
+	if (STATUS_SUCCESS == Status)
+	{	
+		/* ...and execute it */
+		return Command->EntryPoint (argc, argv);
+	}
+	fprintf (stderr, "Failed to connect to the Session Manager! (Status=0x%08lx)\n", Status);
+	return EXIT_FAILURE;
+}
+
+int main (int argc, char *argv[])
+{
+	return (1==argc)
+		? print_synopsys (argc, argv)
+		: pande (argc, argv);
+}
+/* EOF */

reactos/subsys/system/sm/sm.rc

@@ -0,0 +1,4 @@
+#define REACTOS_STR_FILE_DESCRIPTION	"ReactOS/Win32 Session Manager Control Tool\0"
+#define REACTOS_STR_INTERNAL_NAME	"sm\0"
+#define REACTOS_STR_ORIGINAL_FILENAME	"sm.exe\0"
+#include <reactos/version.rc>