examples/win32_service.c - third_party/wsmancli - Git at Google

 /*******************************************************************************
  * Copyright (C) 2004-2006 Intel Corp. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are met:
  *
  *  - Redistributions of source code must retain the above copyright notice,
  *    this list of conditions and the following disclaimer.
  *
  *  - Redistributions in binary form must reproduce the above copyright notice,
  *    this list of conditions and the following disclaimer in the documentation
  *    and/or other materials provided with the distribution.
  *
  *  - Neither the name of Intel Corp. nor the names of its
  *    contributors may be used to endorse or promote products derived from this
  *    software without specific prior written permission.
  *
  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ``AS IS''
  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  * ARE DISCLAIMED. IN NO EVENT SHALL Intel Corp. OR THE CONTRIBUTORS
  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  * POSSIBILITY OF SUCH DAMAGE.
  *******************************************************************************/

 /**
  * @author Anas Nashif
  */
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <unistd.h>
 #include <errno.h>
 #include <time.h>


 #include "wsman-api.h"
 #include "u/libu.h"

 #define RESOURCE_URI "http://schemas.microsoft.com/wbem/wsman/1/wmi/root/cimv2/Win32_Service"
 #define CLASSNAME "Win32_Service"

 /*
    boolean AcceptPause;
    boolean AcceptStop;
    string Caption;
    uint32 CheckPoint;
    string CreationClassName;
    string Description;
    boolean DesktopInteract;
    string DisplayName;
    string ErrorControl;
    uint32 ExitCode;
    datetime InstallDate;
    string Name;
    string PathName;
    uint32 ProcessId;
    uint32 ServiceSpecificExitCode;
    string ServiceType;
    boolean Started;
    string StartMode;
    string StartName;
    string State;
    string Status;
    string SystemCreationClassName;
    string SystemName;
    uint32 TagId;
    uint32 WaitHint;

 */


 struct __CIM_Servie
 {
 	XML_TYPE_BOOL AcceptPause;
 	XML_TYPE_BOOL AcceptStop;
 	XML_TYPE_STR Caption;
 	XML_TYPE_UINT32 CheckPoint;
 	XML_TYPE_STR CreationClassName;
 	XML_TYPE_STR Description;
 	XML_TYPE_BOOL DesktopInteract;
 	XML_TYPE_STR  DisplayName;
 	XML_TYPE_STR ErrorControl;
 	XML_TYPE_UINT32 ExitCode;
 	XML_TYPE_STR InstallDate;
 	XML_TYPE_STR Name;
 	XML_TYPE_STR PathName;
 	XML_TYPE_UINT32 ProcessId;
 	XML_TYPE_UINT32 ServiceSpecificExitCode;
 	XML_TYPE_STR ServiceType;
 	XML_TYPE_BOOL Started;
 	XML_TYPE_STR StartMode;
 	XML_TYPE_STR StartName;
 	XML_TYPE_STR State;
 	XML_TYPE_STR Status;
 	XML_TYPE_STR SystemCreationClassName;
 	XML_TYPE_STR SystemName;
 	XML_TYPE_UINT32 TagId;
 	XML_TYPE_UINT32 WaitHint;
 };
 typedef struct __CIM_Servie CIM_Servie;

 SER_START_ITEMS(CIM_Servie)
 SER_NS_BOOL(RESOURCE_URI,"AcceptPause", 1),
 SER_NS_BOOL(RESOURCE_URI,"AcceptStop", 1),
 SER_NS_STR(RESOURCE_URI,"Caption", 1),
 SER_NS_UINT32(RESOURCE_URI,"CheckPoint", 1),
 SER_NS_STR(RESOURCE_URI,"CreationClassName", 1),
 SER_NS_STR(RESOURCE_URI,"Description", 1),
 SER_NS_BOOL(RESOURCE_URI,"DesktopInteract", 1),
 SER_NS_STR(RESOURCE_URI,"DisplayName", 1),
 SER_NS_STR(RESOURCE_URI,"ErrorControl", 1),
 SER_NS_UINT32(RESOURCE_URI,"ExitCode", 1),
 SER_NS_STR(RESOURCE_URI,"InstallDate", 1),
 SER_NS_STR(RESOURCE_URI,"Name", 1),
 SER_NS_STR(RESOURCE_URI,"PathName", 1),
 SER_NS_UINT32(RESOURCE_URI,"ProcessId", 1),
 SER_NS_UINT32(RESOURCE_URI,"ServiceSpecificExitCode", 1),
 SER_NS_STR(RESOURCE_URI,"ServiceType", 1),
 SER_NS_BOOL(RESOURCE_URI,"Started", 1),
 SER_NS_STR(RESOURCE_URI,"StartMode", 1),
 SER_NS_STR(RESOURCE_URI,"StartName", 1),
 SER_NS_STR(RESOURCE_URI,"State", 1),
 SER_NS_STR(RESOURCE_URI,"Status", 1),
 SER_NS_STR(RESOURCE_URI,"SystemCreationClassName", 1),
 SER_NS_STR(RESOURCE_URI,"SystemName", 1),
 SER_NS_UINT32(RESOURCE_URI,"TagId", 1),
 SER_NS_UINT32(RESOURCE_URI,"WaitHint", 1),
 SER_END_ITEMS(CIM_Servie);

 static char *endpoint = NULL;
 char listall = 0;
 char stop = 0;
 char start = 0;
 char desc = 0;
 char status = 0;
 char dump = 0;


 static void print_info(CIM_Servie *service)
 {
 	if (service->Name) {
 		printf("%s\n", service->Name );
 	} else {
 		printf("Error\n");
 		return;
 	}
 	if (status)
 		printf("\tState: %s\n", service->State );
 	if (desc)
 		printf("\tDescription: %s\n\n", service->Description );
 }

 static int list_services(WsManClient *cl, WsXmlDocH doc, void *data)
 {
 	if (!doc) {
 		return 0;
 	}
 	if (dump)
 		ws_xml_dump_node_tree(stdout, ws_xml_get_doc_root(doc));
 	WsXmlNodeH node = ws_xml_get_soap_body(doc);
 	node = ws_xml_get_child(node, 0,  XML_NS_ENUMERATION, WSENUM_PULL_RESP);
 	node = ws_xml_get_child(node, 0,  XML_NS_ENUMERATION, WSENUM_ITEMS);
 	if (ws_xml_get_child(node, 0, RESOURCE_URI , CLASSNAME )) {
 		CIM_Servie *service = ws_deserialize( wsmc_get_serialization_context( cl ),
 				node,
 				CIM_Servie_TypeInfo, CLASSNAME,
 				RESOURCE_URI, NULL,
 				0, 0);
 		print_info(service);
 		return 1;
 	}
 	return 0;
 }


 int main(int argc, char** argv)
 {
 	WsManClient *cl;
 	WsXmlDocH doc;
 	client_opt_t *options;
 	char retval = 0;
 	u_error_t *error = NULL;


 	u_option_entry_t opt[] = {
 		{ "endpoint",	'u',	U_OPTION_ARG_STRING,	&endpoint,
 			"Endpoint in form of a URL", "<uri>" },
 		{ "list-all", 'l',	U_OPTION_ARG_NONE,	&listall,
 			"List all services", NULL },
 		{ "desc", 'd',	U_OPTION_ARG_NONE,	&desc,
 			"Show service description", NULL },
 		{ "stop", 's',	U_OPTION_ARG_NONE,	&stop,
 			"Stop service", NULL },
 		{ "start", 'S',	U_OPTION_ARG_NONE,	&start,
 			"Start service", NULL },
 		{ "status", 'X',	U_OPTION_ARG_NONE,	&status,
 			"Get service status", NULL },
 		{ "dump", 'D',	U_OPTION_ARG_NONE,	&dump,
 			"Dump request", NULL },
 		{ NULL }
 	};

 	u_option_context_t *opt_ctx;
 	opt_ctx = u_option_context_new("");
 	u_option_context_set_ignore_unknown_options(opt_ctx, FALSE);
 	u_option_context_add_main_entries(opt_ctx, opt, "Win32 Service");
 	retval = u_option_context_parse(opt_ctx, &argc, &argv, &error);

 	u_option_context_free(opt_ctx);

 	if (error) {
 		if (error->message)
 			printf ("%s\n", error->message);
 		u_error_free(error);
 		return 1;
 	}
 	u_error_free(error);


 	u_uri_t *uri;
 	if (endpoint) {
 		u_uri_parse((const char *)endpoint, &uri);
 	}
 	if (!endpoint || !uri) {
 		fprintf(stderr, "endpoint option required\n");
 		return 1;
 	}


 	cl = wsmc_create( uri->host,
 			uri->port,
 			uri->path,
 			uri->scheme,
 			uri->user,
 			uri->pwd);
 	wsmc_transport_init(cl, NULL);
 	options = wsmc_options_init();

 	if (listall) {
 		if (dump) wsmc_set_action_option(options,FLAG_DUMP_REQUEST );
 		wsmc_action_enumerate_and_pull(cl, RESOURCE_URI, options, NULL, list_services, NULL );
 	} else if (start && argv[1]) {
 		if (dump) wsmc_set_action_option(options,FLAG_DUMP_REQUEST );
 		wsmc_add_selector(options, "Name", argv[1]);
 		doc = wsmc_action_invoke(cl, RESOURCE_URI, options,
 				"StartService", NULL);
 		ws_xml_dump_node_tree(stdout, ws_xml_get_doc_root(doc));
 		ws_xml_destroy_doc(doc);
 	} else if (stop && argv[1]) {
 		if (dump) wsmc_set_action_option(options,FLAG_DUMP_REQUEST );
 		wsmc_add_selector(options, "Name", argv[1]);
 		doc = wsmc_action_invoke(cl, RESOURCE_URI, options,
 				"StopService", NULL);
 		ws_xml_dump_node_tree(stdout, ws_xml_get_doc_root(doc));
 		ws_xml_destroy_doc(doc);
 	} else if ( argv[1] ) {
 		if (dump) wsmc_set_action_option(options,FLAG_DUMP_REQUEST );
 		wsmc_add_selector(options, "Name", argv[1]);
 		doc = wsmc_action_get(cl, RESOURCE_URI,
 				options);
 		if (doc) {
 			WsXmlNodeH node = ws_xml_get_soap_body(doc);
 			if (ws_xml_get_child(node, 0, RESOURCE_URI , CLASSNAME )) {
 				CIM_Servie *service = ws_deserialize( wsmc_get_serialization_context( cl ),
 						node,
 						CIM_Servie_TypeInfo, CLASSNAME,
 						RESOURCE_URI, NULL,
 						0, 0);
 				desc = 1;
 				status = 1;
 				print_info(service);
 			}
 		}
 		ws_xml_destroy_doc(doc);
 	}

 	if (uri) {
 		u_uri_free(uri);
 	}

 	wsmc_options_destroy(options);
 	wsmc_release(cl);
 	return 0;
 }
	/*******************************************************************************
	* Copyright (C) 2004-2006 Intel Corp. All rights reserved.
	*
	* Redistribution and use in source and binary forms, with or without
	* modification, are permitted provided that the following conditions are met:
	*
	* - Redistributions of source code must retain the above copyright notice,
	* this list of conditions and the following disclaimer.
	*
	* - Redistributions in binary form must reproduce the above copyright notice,
	* this list of conditions and the following disclaimer in the documentation
	* and/or other materials provided with the distribution.
	*
	* - Neither the name of Intel Corp. nor the names of its
	* contributors may be used to endorse or promote products derived from this
	* software without specific prior written permission.
	*
	* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ``AS IS''
	* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
	* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
	* ARE DISCLAIMED. IN NO EVENT SHALL Intel Corp. OR THE CONTRIBUTORS
	* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
	* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
	* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
	* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
	* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
	* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
	* POSSIBILITY OF SUCH DAMAGE.
	*******************************************************************************/

	/**
	* @author Anas Nashif
	*/
	#include <stdio.h>
	#include <stdlib.h>
	#include <string.h>
	#include <unistd.h>
	#include <errno.h>
	#include <time.h>


	#include "wsman-api.h"
	#include "u/libu.h"

	#define RESOURCE_URI "http://schemas.microsoft.com/wbem/wsman/1/wmi/root/cimv2/Win32_Service"
	#define CLASSNAME "Win32_Service"

	/*
	boolean AcceptPause;
	boolean AcceptStop;
	string Caption;
	uint32 CheckPoint;
	string CreationClassName;
	string Description;
	boolean DesktopInteract;
	string DisplayName;
	string ErrorControl;
	uint32 ExitCode;
	datetime InstallDate;
	string Name;
	string PathName;
	uint32 ProcessId;
	uint32 ServiceSpecificExitCode;
	string ServiceType;
	boolean Started;
	string StartMode;
	string StartName;
	string State;
	string Status;
	string SystemCreationClassName;
	string SystemName;
	uint32 TagId;
	uint32 WaitHint;

	*/


	struct __CIM_Servie
	{
	XML_TYPE_BOOL AcceptPause;
	XML_TYPE_BOOL AcceptStop;
	XML_TYPE_STR Caption;
	XML_TYPE_UINT32 CheckPoint;
	XML_TYPE_STR CreationClassName;
	XML_TYPE_STR Description;
	XML_TYPE_BOOL DesktopInteract;
	XML_TYPE_STR DisplayName;
	XML_TYPE_STR ErrorControl;
	XML_TYPE_UINT32 ExitCode;
	XML_TYPE_STR InstallDate;
	XML_TYPE_STR Name;
	XML_TYPE_STR PathName;
	XML_TYPE_UINT32 ProcessId;
	XML_TYPE_UINT32 ServiceSpecificExitCode;
	XML_TYPE_STR ServiceType;
	XML_TYPE_BOOL Started;
	XML_TYPE_STR StartMode;
	XML_TYPE_STR StartName;
	XML_TYPE_STR State;
	XML_TYPE_STR Status;
	XML_TYPE_STR SystemCreationClassName;
	XML_TYPE_STR SystemName;
	XML_TYPE_UINT32 TagId;
	XML_TYPE_UINT32 WaitHint;
	};
	typedef struct __CIM_Servie CIM_Servie;

	SER_START_ITEMS(CIM_Servie)
	SER_NS_BOOL(RESOURCE_URI,"AcceptPause", 1),
	SER_NS_BOOL(RESOURCE_URI,"AcceptStop", 1),
	SER_NS_STR(RESOURCE_URI,"Caption", 1),
	SER_NS_UINT32(RESOURCE_URI,"CheckPoint", 1),
	SER_NS_STR(RESOURCE_URI,"CreationClassName", 1),
	SER_NS_STR(RESOURCE_URI,"Description", 1),
	SER_NS_BOOL(RESOURCE_URI,"DesktopInteract", 1),
	SER_NS_STR(RESOURCE_URI,"DisplayName", 1),
	SER_NS_STR(RESOURCE_URI,"ErrorControl", 1),
	SER_NS_UINT32(RESOURCE_URI,"ExitCode", 1),
	SER_NS_STR(RESOURCE_URI,"InstallDate", 1),
	SER_NS_STR(RESOURCE_URI,"Name", 1),
	SER_NS_STR(RESOURCE_URI,"PathName", 1),
	SER_NS_UINT32(RESOURCE_URI,"ProcessId", 1),
	SER_NS_UINT32(RESOURCE_URI,"ServiceSpecificExitCode", 1),
	SER_NS_STR(RESOURCE_URI,"ServiceType", 1),
	SER_NS_BOOL(RESOURCE_URI,"Started", 1),
	SER_NS_STR(RESOURCE_URI,"StartMode", 1),
	SER_NS_STR(RESOURCE_URI,"StartName", 1),
	SER_NS_STR(RESOURCE_URI,"State", 1),
	SER_NS_STR(RESOURCE_URI,"Status", 1),
	SER_NS_STR(RESOURCE_URI,"SystemCreationClassName", 1),
	SER_NS_STR(RESOURCE_URI,"SystemName", 1),
	SER_NS_UINT32(RESOURCE_URI,"TagId", 1),
	SER_NS_UINT32(RESOURCE_URI,"WaitHint", 1),
	SER_END_ITEMS(CIM_Servie);

	static char *endpoint = NULL;
	char listall = 0;
	char stop = 0;
	char start = 0;
	char desc = 0;
	char status = 0;
	char dump = 0;




	static void print_info(CIM_Servie *service)
	{
	if (service->Name) {
	printf("%s\n", service->Name );
	} else {
	printf("Error\n");
	return;
	}
	if (status)
	printf("\tState: %s\n", service->State );
	if (desc)
	printf("\tDescription: %s\n\n", service->Description );
	}

	static int list_services(WsManClient cl, WsXmlDocH doc, void data)
	{
	if (!doc) {
	return 0;
	}
	if (dump)
	ws_xml_dump_node_tree(stdout, ws_xml_get_doc_root(doc));
	WsXmlNodeH node = ws_xml_get_soap_body(doc);
	node = ws_xml_get_child(node, 0, XML_NS_ENUMERATION, WSENUM_PULL_RESP);
	node = ws_xml_get_child(node, 0, XML_NS_ENUMERATION, WSENUM_ITEMS);
	if (ws_xml_get_child(node, 0, RESOURCE_URI , CLASSNAME )) {
	CIM_Servie *service = ws_deserialize( wsmc_get_serialization_context( cl ),
	node,
	CIM_Servie_TypeInfo, CLASSNAME,
	RESOURCE_URI, NULL,
	0, 0);
	print_info(service);
	return 1;
	}
	return 0;
	}



	int main(int argc, char** argv)
	{
	WsManClient *cl;
	WsXmlDocH doc;
	client_opt_t *options;
	char retval = 0;
	u_error_t *error = NULL;


	u_option_entry_t opt[] = {
	{ "endpoint", 'u', U_OPTION_ARG_STRING, &endpoint,
	"Endpoint in form of a URL", "<uri>" },
	{ "list-all", 'l', U_OPTION_ARG_NONE, &listall,
	"List all services", NULL },
	{ "desc", 'd', U_OPTION_ARG_NONE, &desc,
	"Show service description", NULL },
	{ "stop", 's', U_OPTION_ARG_NONE, &stop,
	"Stop service", NULL },
	{ "start", 'S', U_OPTION_ARG_NONE, &start,
	"Start service", NULL },
	{ "status", 'X', U_OPTION_ARG_NONE, &status,
	"Get service status", NULL },
	{ "dump", 'D', U_OPTION_ARG_NONE, &dump,
	"Dump request", NULL },
	{ NULL }
	};

	u_option_context_t *opt_ctx;
	opt_ctx = u_option_context_new("");
	u_option_context_set_ignore_unknown_options(opt_ctx, FALSE);
	u_option_context_add_main_entries(opt_ctx, opt, "Win32 Service");
	retval = u_option_context_parse(opt_ctx, &argc, &argv, &error);

	u_option_context_free(opt_ctx);

	if (error) {
	if (error->message)
	printf ("%s\n", error->message);
	u_error_free(error);
	return 1;
	}
	u_error_free(error);


	u_uri_t *uri;
	if (endpoint) {
	u_uri_parse((const char *)endpoint, &uri);
	}
	if (!endpoint \|\| !uri) {
	fprintf(stderr, "endpoint option required\n");
	return 1;
	}


	cl = wsmc_create( uri->host,
	uri->port,
	uri->path,
	uri->scheme,
	uri->user,
	uri->pwd);
	wsmc_transport_init(cl, NULL);
	options = wsmc_options_init();

	if (listall) {
	if (dump) wsmc_set_action_option(options,FLAG_DUMP_REQUEST );
	wsmc_action_enumerate_and_pull(cl, RESOURCE_URI, options, NULL, list_services, NULL );
	} else if (start && argv[1]) {
	if (dump) wsmc_set_action_option(options,FLAG_DUMP_REQUEST );
	wsmc_add_selector(options, "Name", argv[1]);
	doc = wsmc_action_invoke(cl, RESOURCE_URI, options,
	"StartService", NULL);
	ws_xml_dump_node_tree(stdout, ws_xml_get_doc_root(doc));
	ws_xml_destroy_doc(doc);
	} else if (stop && argv[1]) {
	if (dump) wsmc_set_action_option(options,FLAG_DUMP_REQUEST );
	wsmc_add_selector(options, "Name", argv[1]);
	doc = wsmc_action_invoke(cl, RESOURCE_URI, options,
	"StopService", NULL);
	ws_xml_dump_node_tree(stdout, ws_xml_get_doc_root(doc));
	ws_xml_destroy_doc(doc);
	} else if ( argv[1] ) {
	if (dump) wsmc_set_action_option(options,FLAG_DUMP_REQUEST );
	wsmc_add_selector(options, "Name", argv[1]);
	doc = wsmc_action_get(cl, RESOURCE_URI,
	options);
	if (doc) {
	WsXmlNodeH node = ws_xml_get_soap_body(doc);
	if (ws_xml_get_child(node, 0, RESOURCE_URI , CLASSNAME )) {
	CIM_Servie *service = ws_deserialize( wsmc_get_serialization_context( cl ),
	node,
	CIM_Servie_TypeInfo, CLASSNAME,
	RESOURCE_URI, NULL,
	0, 0);
	desc = 1;
	status = 1;
	print_info(service);
	}
	}
	ws_xml_destroy_doc(doc);
	}

	if (uri) {
	u_uri_free(uri);
	}

	wsmc_options_destroy(options);
	wsmc_release(cl);
	return 0;
	}