private void SetServiceTriggerStartOnCustomEvent(string serviceName) { using (ServiceController sc = new ServiceController(serviceName)) { // Marshal Guid struct GUID_DEVINTERFACE_DISK to native memory IntPtr pGuidCustomEvent = Marshal.AllocHGlobal(Marshal.SizeOf(typeof(Guid))); Marshal.StructureToPtr(CUSTOM_EVENT_GUID, pGuidCustomEvent, false); //// Allocate and set the SERVICE_TRIGGER_SPECIFIC_DATA_ITEM structure //SERVICE_TRIGGER_SPECIFIC_DATA_ITEM deviceData; //deviceData.dwDataType = ServiceTriggerDataType.SERVICE_TRIGGER_DATA_TYPE_STRING; //deviceData.cbData = (uint)(USBHardwareId.Length + 1) * 2; //IntPtr pUSBHardwareId = Marshal.StringToHGlobalUni(USBHardwareId); //deviceData.pData = pUSBHardwareId; //// Marshal the SERVICE_TRIGGER_SPECIFIC_DATA_ITEM struct to native memory //IntPtr pDeviceData = Marshal.AllocHGlobal(Marshal.SizeOf(typeof( // SERVICE_TRIGGER_SPECIFIC_DATA_ITEM))); //Marshal.StructureToPtr(deviceData, pDeviceData, false); // Allocate and set the SERVICE_TRIGGER structure SERVICE_TRIGGER serviceTrigger; serviceTrigger.dwTriggerType = ServiceTriggerType.SERVICE_TRIGGER_TYPE_CUSTOM; serviceTrigger.dwAction = ServiceTriggerAction.SERVICE_TRIGGER_ACTION_SERVICE_START; serviceTrigger.pTriggerSubtype = pGuidCustomEvent; serviceTrigger.cDataItems = 0; serviceTrigger.pDataItems = IntPtr.Zero; // Marshal the SERVICE_TRIGGER struct to native memory IntPtr pServiceTrigger = Marshal.AllocHGlobal(Marshal.SizeOf(typeof( SERVICE_TRIGGER))); Marshal.StructureToPtr(serviceTrigger, pServiceTrigger, false); // Allocate and set the SERVICE_TRIGGER_INFO structure. SERVICE_TRIGGER_INFO serviceTriggerInfo = new SERVICE_TRIGGER_INFO(); serviceTriggerInfo.cTriggers = 1; serviceTriggerInfo.pTriggers = pServiceTrigger; // Marshal the SERVICE_TRIGGER_INFO struct to native memory IntPtr pServiceTriggerInfo = Marshal.AllocHGlobal(Marshal.SizeOf(typeof( SERVICE_TRIGGER_INFO))); Marshal.StructureToPtr(serviceTriggerInfo, pServiceTriggerInfo, false); // Call ChangeServiceConfig2 with the SERVICE_CONFIG_TRIGGER_INFO level // and pass to it the address of the SERVICE_TRIGGER_INFO structure. bool bSuccess = ServiceNative.ChangeServiceConfig2( sc.ServiceHandle.DangerousGetHandle(), ServiceConfig2InfoLevel.SERVICE_CONFIG_TRIGGER_INFO, pServiceTriggerInfo); int errorCode = Marshal.GetLastWin32Error(); // Clean up the native resources //Marshal.FreeHGlobal(pGuidUSBDevice); //Marshal.FreeHGlobal(pUSBHardwareId); //Marshal.FreeHGlobal(pDeviceData); Marshal.FreeHGlobal(pServiceTrigger); Marshal.FreeHGlobal(pServiceTriggerInfo); // If the service failed to be set as trigger start, throw the error // returned by the ChangeServiceConfig2 function. if (!bSuccess) { Marshal.ThrowExceptionForHR(errorCode); } } }
static void DumpTriggers(SafeServiceHandle service) { using (var buf = new SafeStructureInOutBuffer <SERVICE_TRIGGER_INFO>(8192, false)) { int required = 0; if (!QueryServiceConfig2(service, SERVICE_CONFIG_TRIGGER_INFO, buf, 8192, out required)) { return; } SERVICE_TRIGGER_INFO trigger_info = buf.Result; if (trigger_info.cTriggers == 0) { return; } SERVICE_TRIGGER[] trigger_arr; using (SafeHGlobalBuffer triggers = new SafeHGlobalBuffer(trigger_info.pTriggers, trigger_info.cTriggers * Marshal.SizeOf(typeof(SERVICE_TRIGGER)), false)) { trigger_arr = new SERVICE_TRIGGER[trigger_info.cTriggers]; triggers.ReadArray(0, trigger_arr, 0, trigger_arr.Length); } for (int i = 0; i < trigger_arr.Length; ++i) { SERVICE_TRIGGER trigger = trigger_arr[i]; Console.WriteLine("Trigger: {0} - Type: {1} - Action: {2}", i, trigger.dwTriggerType, trigger.dwAction); switch (trigger.dwTriggerType) { case ServiceTriggerType.Custom: Console.WriteLine("Subtype: [ETW UUID] {0:B}", trigger.GetSubType()); break; case ServiceTriggerType.DeviceInterfaceArrival: Console.WriteLine("Subtype: [Interface Class GUID] {0:B}", trigger.GetSubType()); break; case ServiceTriggerType.GroupPolicy: { Guid sub_type = trigger.GetSubType(); if (sub_type == MACHINE_POLICY_PRESENT_GUID) { Console.WriteLine("Subtype: [Machine Policy Present]"); } else if (sub_type == USER_POLICY_PRESENT_GUID) { Console.WriteLine("Subtype: [User Policy Present]"); } else { Console.WriteLine("Subtype: [Unknown Group Policy] {0:B}", sub_type); } } break; case ServiceTriggerType.NetworkEndpoint: { Guid sub_type = trigger.GetSubType(); if (sub_type == RPC_INTERFACE_EVENT_GUID) { Console.WriteLine("Subtype: [RPC Interface]"); } else if (sub_type == NAMED_PIPE_EVENT_GUID) { Console.WriteLine("Subtype: [Named Pipe]"); } else { Console.WriteLine("Subtype: [Unknown Network Endpoint] {0:B}", sub_type); } } break; case ServiceTriggerType.DomainJoin: { Guid sub_type = trigger.GetSubType(); if (sub_type == DOMAIN_JOIN_GUID) { Console.WriteLine("Subtype: [Domain Join]"); } else if (sub_type == DOMAIN_LEAVE_GUID) { Console.WriteLine("Subtype: [Domain Leave]"); } else { Console.WriteLine("Subtype: [Unknown Domain Join] {0:B}", sub_type); } } break; case ServiceTriggerType.IPAddressAvailability: { Guid sub_type = trigger.GetSubType(); if (sub_type == NETWORK_MANAGER_FIRST_IP_ADDRESS_ARRIVAL_GUID) { Console.WriteLine("Subtype: [First IP Address Available]"); } else if (sub_type == NETWORK_MANAGER_LAST_IP_ADDRESS_REMOVAL_GUID) { Console.WriteLine("Subtype: [Last IP Address Available]"); } else { Console.WriteLine("Subtype: [Unknown IP Address Availability] {0:B}", sub_type); } } break; } if (trigger.pDataItems != IntPtr.Zero && trigger.cDataItems > 0) { DumpCustomData(trigger); } } } }
/// <summary> /// Set the service to trigger-start when the first IP address on the TCP/IP /// networking stack becomes available, and trigger-stop when the last IP address on /// the TCP/IP networking stack becomes unavailable. /// </summary> /// <param name="serviceName"></param> private void SetServiceTriggerStartOnIPAddressArrival(string serviceName) { using (ServiceController sc = new ServiceController(serviceName)) { // Marshal Guid struct NETWORK_MANAGER_FIRST_IP_ADDRESS_ARRIVAL_GUID and // NETWORK_MANAGER_LAST_IP_ADDRESS_REMOVAL_GUID to native memory IntPtr pGuidIpAddressArrival = Marshal.AllocHGlobal(Marshal.SizeOf(typeof(Guid))); Marshal.StructureToPtr(NETWORK_MANAGER_FIRST_IP_ADDRESS_ARRIVAL_GUID, pGuidIpAddressArrival, false); IntPtr pGuidIpAddressRemoval = Marshal.AllocHGlobal(Marshal.SizeOf(typeof(Guid))); Marshal.StructureToPtr(NETWORK_MANAGER_LAST_IP_ADDRESS_REMOVAL_GUID, pGuidIpAddressRemoval, false); // Allocate and set the SERVICE_TRIGGER structure for // NETWORK_MANAGER_FIRST_IP_ADDRESS_ARRIVAL_GUID to start the service SERVICE_TRIGGER serviceTrigger1 = new SERVICE_TRIGGER(); serviceTrigger1.dwTriggerType = ServiceTriggerType.SERVICE_TRIGGER_TYPE_IP_ADDRESS_AVAILABILITY; serviceTrigger1.dwAction = ServiceTriggerAction.SERVICE_TRIGGER_ACTION_SERVICE_START; serviceTrigger1.pTriggerSubtype = pGuidIpAddressArrival; // Allocate and set the SERVICE_TRIGGER structure for // NETWORK_MANAGER_LAST_IP_ADDRESS_REMOVAL_GUID to stop the service SERVICE_TRIGGER serviceTrigger2 = new SERVICE_TRIGGER(); serviceTrigger2.dwTriggerType = ServiceTriggerType.SERVICE_TRIGGER_TYPE_IP_ADDRESS_AVAILABILITY; serviceTrigger2.dwAction = ServiceTriggerAction.SERVICE_TRIGGER_ACTION_SERVICE_STOP; serviceTrigger2.pTriggerSubtype = pGuidIpAddressRemoval; // Marshal the two SERVICE_TRIGGER structs to native memory as an array IntPtr pServiceTriggers = Marshal.AllocHGlobal(Marshal.SizeOf(typeof( SERVICE_TRIGGER)) * 2 /*2 elements*/); Marshal.StructureToPtr(serviceTrigger1, pServiceTriggers, false); Marshal.StructureToPtr(serviceTrigger2, new IntPtr((long)pServiceTriggers + Marshal.SizeOf(typeof(SERVICE_TRIGGER))), false); // Allocate and set the SERVICE_TRIGGER_INFO structure SERVICE_TRIGGER_INFO serviceTriggerInfo = new SERVICE_TRIGGER_INFO(); serviceTriggerInfo.cTriggers = 2; serviceTriggerInfo.pTriggers = pServiceTriggers; // Marshal the SERVICE_TRIGGER_INFO struct to native memory IntPtr pServiceTriggerInfo = Marshal.AllocHGlobal(Marshal.SizeOf(typeof( SERVICE_TRIGGER_INFO))); Marshal.StructureToPtr(serviceTriggerInfo, pServiceTriggerInfo, false); // Call ChangeServiceConfig2 with the SERVICE_CONFIG_TRIGGER_INFO level // and pass to it the address of the SERVICE_TRIGGER_INFO structure. bool bSuccess = ServiceNative.ChangeServiceConfig2( sc.ServiceHandle.DangerousGetHandle(), ServiceConfig2InfoLevel.SERVICE_CONFIG_TRIGGER_INFO, pServiceTriggerInfo); int errorCode = Marshal.GetLastWin32Error(); // Clean up the native resources Marshal.FreeHGlobal(pGuidIpAddressArrival); Marshal.FreeHGlobal(pGuidIpAddressRemoval); Marshal.FreeHGlobal(pServiceTriggers); Marshal.FreeHGlobal(pServiceTriggerInfo); // If the service failed to be set as trigger start, throw the error // returned by the ChangeServiceConfig2 function. if (!bSuccess) { Marshal.ThrowExceptionForHR(errorCode); } } }