1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
/* SPDX-License-Identifier: MIT
*
* Copyright (C) 2019 WireGuard LLC. All Rights Reserved.
*/
using System;
using System.IO;
using System.IO.Pipes;
using System.Runtime.InteropServices;
using System.ComponentModel;
using System.Diagnostics;
using System.Security.Principal;
using System.Threading;
namespace Tunnel
{
public class Service
{
private const string LongName = "Example WireGuard Tunnel Client";
private const string Description = "A WireGuard tunnel created by example code.";
[StructLayout(LayoutKind.Sequential)]
private struct GoString
{
public string str;
public long len;
}
[DllImport("tunnel.dll", EntryPoint = "WireGuardTunnelService", CallingConvention = CallingConvention.Cdecl)]
private static extern bool WireGuardTunnelService(GoString configFile);
public static bool Run(string configFile)
{
return WireGuardTunnelService(new GoString { str = configFile, len = configFile.Length });
}
public static NamedPipeClientStream GetPipe(string configFile)
{
var pipepath = "ProtectedPrefix\\Administrators\\WireGuard\\" + Path.GetFileNameWithoutExtension(configFile);
return new NamedPipeClientStream(pipepath);
}
public static void Add(string configFile)
{
var tunnelName = Path.GetFileNameWithoutExtension(configFile);
var shortName = String.Format("WireGuardTunnel${0}", tunnelName);
var longName = String.Format("{0}: {1}", LongName, tunnelName);
var exeName = Process.GetCurrentProcess().MainModule.FileName;
var pathAndArgs = String.Format("\"{0}\" /service \"{1}\"", exeName, configFile); //TODO: This is not the proper way to escape file args.
var accessControl = File.GetAccessControl(configFile); //TODO: TOCTOU!
accessControl.SetOwner(new NTAccount(Environment.UserDomainName, Environment.UserName));
File.SetAccessControl(configFile, accessControl);
var scm = Win32.OpenSCManager(null, null, Win32.ScmAccessRights.AllAccess);
if (scm == IntPtr.Zero)
throw new Win32Exception(Marshal.GetLastWin32Error());
try
{
var service = Win32.OpenService(scm, shortName, Win32.ServiceAccessRights.AllAccess);
if (service != IntPtr.Zero)
{
Win32.CloseServiceHandle(service);
Remove(configFile);
}
service = Win32.CreateService(scm, shortName, longName, Win32.ServiceAccessRights.AllAccess, Win32.ServiceType.Win32OwnProcess, Win32.ServiceStartType.Auto, Win32.ServiceError.Normal, pathAndArgs, null, IntPtr.Zero, "Nsi", null, null);
if (service == IntPtr.Zero)
throw new Win32Exception(Marshal.GetLastWin32Error());
try
{
var sidType = Win32.ServiceSidType.Unrestricted;
if (!Win32.ChangeServiceConfig2(service, Win32.ServiceConfigType.SidInfo, ref sidType))
throw new Win32Exception(Marshal.GetLastWin32Error());
var description = new Win32.ServiceDescription { lpDescription = Description };
if (!Win32.ChangeServiceConfig2(service, Win32.ServiceConfigType.Description, ref description))
throw new Win32Exception(Marshal.GetLastWin32Error());
if (!Win32.StartService(service, 0, null))
throw new Win32Exception(Marshal.GetLastWin32Error());
}
finally
{
Win32.CloseServiceHandle(service);
}
}
finally
{
Win32.CloseServiceHandle(scm);
}
}
public static void Remove(string configFile)
{
var tunnelName = Path.GetFileNameWithoutExtension(configFile);
var shortName = String.Format("WireGuardTunnel${0}", tunnelName);
var scm = Win32.OpenSCManager(null, null, Win32.ScmAccessRights.AllAccess);
if (scm == IntPtr.Zero)
throw new Win32Exception(Marshal.GetLastWin32Error());
try
{
var service = Win32.OpenService(scm, shortName, Win32.ServiceAccessRights.AllAccess);
if (service == IntPtr.Zero)
{
Win32.CloseServiceHandle(service);
return;
}
try
{
var serviceStatus = new Win32.ServiceStatus();
Win32.ControlService(service, Win32.ServiceControl.Stop, serviceStatus);
for (int i = 0; i < 180 && Win32.QueryServiceStatus(service, serviceStatus) && serviceStatus.dwCurrentState != Win32.ServiceState.Stopped; ++i)
Thread.Sleep(1000);
if (!Win32.DeleteService(service))
throw new Win32Exception(Marshal.GetLastWin32Error());
}
finally
{
Win32.CloseServiceHandle(service);
}
}
finally
{
Win32.CloseServiceHandle(scm);
}
}
}
}
|
