// 引数
private static WebSocketParams GetArguments(string[] args)
{
WebSocketParams wsp = new WebSocketParams();
wsp.Init();
for (int i = 0; i < args.Length; i++)
{
switch (args[i].ToLower())
{
case "/file":
case "--file":
case "/f":
case "-f":
wsp = WebSocketParams.Deseiralize(args[++i]);
break;
case "/targetserver":
case "--targetserver":
case "/server":
case "--server":
case "/s":
case "-s":
wsp.TargetServer = args[++i];
wsp.IsClient = true;
break;
case "/accepturi":
case "--accepturi":
case "/accepturl":
case "--accepturl":
case "/accept":
case "-accept":
case "--accept":
case "/a":
case "-a":
wsp.AcceptUrls.Add(args[++i]);
wsp.IsServer = true;
break;
case "/addpath":
case "--addpath":
case "/p":
case "-p":
wsp.AddPath = args[++i].Split(';');
break;
case "/execute":
case "--execute":
case "/x":
case "-x":
wsp.Execute = args[++i];
break;
}
}
return(wsp);
}
// サービス開始
protected override void OnStart(string[] args)
{
// カレントディレクトリ
Environment.CurrentDirectory = Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location);
// 設定ファイル読み込み
WebSocketParams wsp = WebSocketParams.Deseiralize(CONF_FILE);
// PATH追加
if (wsp.AddPath.Length > 0)
{
Environment.SetEnvironmentVariable("PATH",
string.Join(";", wsp.AddPath.Select(x => Path.GetFullPath(x))) + ";" +
Environment.ExpandEnvironmentVariables("%PATH%"));
}
// WebSocketサーバ開始
_wss = new WebSocketServer(wsp);
_wss.Start().ConfigureAwait(false);
}
// メイン処理
static void Main(string[] args)
{
// カレントディレクトリ
Environment.CurrentDirectory = Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location);
// 引数読み取り
WebSocketParams wsp = GetArguments(args);
// WebSocketサーバ開始
WebSocketServer _wss = null;
if (wsp.IsServer && wsp.AcceptUrls.Count > 0)
{
// PATH追加
if (wsp.AddPath.Length > 0)
{
Environment.SetEnvironmentVariable("PATH",
string.Join(";", wsp.AddPath.Select(x => Path.GetFullPath(x))) + ";" +
Environment.ExpandEnvironmentVariables("%PATH%"));
}
_wss = new WebSocketServer(wsp);
if (wsp.IsClient)
{
_wss.Start().ConfigureAwait(false);
}
else
{
_wss.Start().Wait();
return;
}
}
// WebSocketサーバへ接続
string executePath = "";
switch (wsp.Execute.ToLower())
{
case "powershell":
case "powershell.exe":
case "pwsh":
case "psh":
executePath = WebSocketParams.MSG_CONSOLE_POWERSHELL;
break;
case "cmd":
case "cmd.exe":
default:
executePath = WebSocketParams.MSG_CONSOLE_CMD;
break;
}
if (wsp.IsClient && wsp.TargetServer != "")
{
WebSocketClient wsc = new WebSocketClient(wsp);
wsc.ConnectServer().ConfigureAwait(false);
wsc.SendInit(executePath).Wait();
try
{
while (true)
{
string command = Console.ReadLine();
if (command == "exit" || command == "quit")
{
wsc.SendClose().ConfigureAwait(false);
break;
}
wsc.SendMessage(WebSocketParams.HEADER_STD_IN + command).Wait();
}
}
catch { }
}
// 終了
if (_wss != null)
{
_wss.Stop();
}
}