コード例 #1
0
ファイル: Bender.cs プロジェクト: JohanLindvall/bender
        public static void DoCommand(Socket peer, Stream input, Stream output, Dictionary <string, string> fileMappings, Dictionary <Regex, string> colorMappings)
        {
            using (input)
            {
                var crash = false;
                try
                {
                    var line = ReadLine(input);
                    var i3   = line.LastIndexOf((char)3);
                    if (i3 != -1)
                    {
                        line = line.Substring(i3 + 1);
                    }
                    var lineOrig = line;
                    line = line.ToLowerInvariant();
                    switch (line)
                    {
                    case "age":
                    {
                        var serverPath = ReadServerPath(ReadLine(input), fileMappings);
                        var server     = serverPath.Item1;
                        var path       = serverPath.Item2;
                        var ms         = new MemoryStream();
                        FileTailer.Tail(server, path, 1, false, ms);
                        var str = Encoding.ASCII.GetString(ms.ToArray());
                        str = str.Substring(0, Math.Min(str.Length, 100)).Replace(",", ".");
                        string os        = null;
                        var    endsWithU = false;
                        while (str.Length > 0)
                        {
                            DateTime dt;
                            if (DateTime.TryParse(str, CultureInfo.InvariantCulture, endsWithU ? DateTimeStyles.AssumeUniversal : DateTimeStyles.AssumeLocal, out dt))
                            {
                                os = ((long)(DateTime.Now - dt).TotalSeconds).ToString();
                                break;
                            }
                            endsWithU = str.EndsWith("U") && !str.EndsWith(" U");
                            str       = str.Substring(0, str.Length - 1);
                        }
                        Write(os ?? "null", output);
                        break;
                    }

                    case "apppools":
                    {
                        Shell.Do(peer, input, output, @"c:\windows\system32\inetsrv\appcmd", "list apppool /config /xml", false);
                        break;
                    }

                    case "bend":
                    {
                        Remote.Do(peer, input, output);
                        break;
                    }

                    case "buildtime":
                    {
                        Write(Environment.MachineName + " " + Date(BuildDate.RetrieveLinkerTimestamp()) + "\r\n", output);
                        break;
                    }

                    case "crash":
                    {
                        crash = true;
                        throw new InvalidOperationException();
                    }

                    case "date":
                    {
                        var serverPath = ReadServerPath(ReadLine(input), fileMappings);
                        var server     = serverPath.Item1;
                        var path       = serverPath.Item2;
                        var pattern    = ReadLine(input);
                        DateFinder.Find(new LogStream(server, path), output, pattern);
                        break;
                    }

                    case "download":
                    {
                        var path = ReadLine(input);
                        Zip.Download(path, output);
                        break;
                    }

                    case "downloadzip":
                    {
                        var path = ReadLine(input);
                        Zip.Zipit(path, output);
                        break;
                    }

                    case "dump":
                    {
                        var path    = ReadLine(input);
                        var process = ReadLine(input);
                        DumpFile.Create(path, process);
                        break;
                    }

                    case "filecount":
                    {
                        var serverPath = ReadServerPath(ReadLine(input), fileMappings);
                        var path       = Path.GetDirectoryName(serverPath.Item2);
                        var pattern    = Path.GetFileName(serverPath.Item2);
                        Write(Directory.GetFiles(path, pattern).Length.ToString(CultureInfo.InvariantCulture), output);
                        break;
                    }

                    case "fulldump":
                    {
                        var enable = int.Parse(ReadLine(input));
                        DumpConfig.Enable(enable != 0);
                        break;
                    }

                    case "fulldumpenabled":
                    {
                        var enabled = DumpConfig.IsEnabled();
                        Write(enabled ? "1" : "0", output);
                        break;
                    }

                    case "isonline":
                    {
                        Write(Environment.MachineName + " ", output);
                        FetchUri.FetchHeaders("http://localhost/IsOnline.aspx", output);
                        Write($" {Path.GetFileName(Path.GetDirectoryName(DetermineVersion.DetermineWebSiteFolder()))} \r\n", output);
                        break;
                    }

                    case "isonline2":
                    {
                        FetchUri.FetchHeaders("http://localhost/IsOnline.aspx", output);
                        break;
                    }

                    case "ls":
                    {
                        foreach (var fi in LogStream.GetLocalFileInfos(ReadLine(input)))
                        {
                            Write($"- 1 root root {fi.Size} {fi.Modification.ToUniversalTime().ToString("o", CultureInfo.InvariantCulture)} {fi.Name}\n", output);
                        }
                        break;
                    }

                    case "mem":
                    {
                        MEMORYSTATUSEX memStatus = new MEMORYSTATUSEX();
                        if (GlobalMemoryStatusEx(memStatus))
                        {
                            Write($"{Math.Round((double)memStatus.ullTotalPhys / (1024 * 1024 * 1024))} GB", output);
                        }
                        break;
                    }

                    case "netrelease":
                    {
                        var release = Convert.ToInt64(Registry.GetValue(@"HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\NET Framework Setup\NDP\v4\Full", "Release", null));
                        Write(release.ToString(CultureInfo.InvariantCulture), output);
                        break;
                    }

                    case "ntp":
                    {
                        Shell.Do(peer, input, output, "ntpq.exe", "-np", false);
                        break;
                    }

                    case "offline":
                    {
                        Online.Do(false);
                        break;
                    }

                    case "online":
                    {
                        Online.Do(true);
                        break;
                    }

                    case "patch":
                    {
                        List <string> args = new List <string>();
                        string        arg;
                        while ((arg = ReadLine(input)).Length > 0)
                        {
                            args.Add(arg);
                        }

                        bool rebootIfNeeded = args.Contains("reboot");
                        bool onlyList       = args.Contains("onlylist");

                        Patch patch = new Patch();
                        patch.InstallPatches(rebootIfNeeded, onlyList, output);
                        break;
                    }

                    case "pc":
                    {
                        var counter = ReadLine(input);
                        Write(PerformanceCounter.GetValue(counter).ToString(CultureInfo.InvariantCulture), output);
                        break;
                    }

                    case "post / http/1.0":
                    case "post / http/1.1":
                    {
                        Http.Do(input, fileMappings, colorMappings);
                        break;
                    }

                    case "powershell":
                    {
                        List <string> scripts = new List <string>();
                        string        arg;
                        while ((arg = ReadLine(input)).Length > 0)
                        {
                            scripts.Add(arg);
                        }

                        foreach (string script in scripts)
                        {
                            Shell.Do(peer, input, output, "powershell.exe", $"-file {script}", false);
                        }
                        break;
                    }

                    case "rpc":
                    {
                        var counter = ReadLine(input);
                        var specs   = (ReadLine(input) ?? "0").Split('|');

                        var index = int.Parse(specs[0]);
                        if (index == 4)
                        {
                            Write(Date(PerformanceCounterClient.GetDate(counter)), output);
                        }
                        else
                        {
                            var value = PerformanceCounterClient.GetValue(counter, index);
                            Write(value.ToString(CultureInfo.InvariantCulture), output);
                            if (specs.Length == 2)
                            {
                                Write("\n", output);
                                switch (specs[1].ToLowerInvariant())
                                {
                                case "timespan":
                                    Write(TimeSpan.FromSeconds(value) + "\n", output);
                                    break;
                                }
                            }
                        }
                        break;
                    }

                    case "shell":
                    {
                        Shell.Do(peer, input, output);
                        break;
                    }

                    case "sites":
                    {
                        Shell.Do(peer, input, output, @"c:\windows\system32\inetsrv\appcmd", "list site /config /xml", false);
                        break;
                    }

                    case "stacktrace":
                    case "stacktracenative":
                    case "verifyheap":
                    {
                        var pidOrSpec = GetPid2(ReadLine(input));

                        switch (line)
                        {
                        case "stacktrace":
                            StackTrace.DoManaged(peer, output, pidOrSpec);
                            break;

                        case "stacktracenative":
                            StackTrace.DoNative(peer, output, pidOrSpec);
                            break;

                        case "verifyheap":
                            StackTrace.VerifyHeap(peer, output, pidOrSpec);
                            break;
                        }
                        break;
                    }

                    case "systeminfo":
                    {
                        Shell.Do(peer, input, output, "systeminfo.exe", "", false);
                        break;
                    }

                    case "tail":
                    {
                        var serverPath = ReadServerPath(ReadLine(input), fileMappings);
                        var server     = serverPath.Item1;
                        var path       = serverPath.Item2;
                        var countLine  = ReadLine(input);
                        int count      = 0;
                        if (!string.IsNullOrEmpty(countLine))
                        {
                            count = int.Parse(countLine);
                        }
                        line = ReadLine(input);
                        bool tail = false;
                        if (!string.IsNullOrEmpty(line))
                        {
                            tail = int.Parse(line) != 0;
                        }
                        FileTailer.Tail(server, path, count, tail, output);
                        break;
                    }

                    case "tailc":
                    {
                        using (var fs = new FileStream(ReadLine(input), FileMode.Open, FileAccess.Read, FileShare.ReadWrite))
                        {
                            fs.Position = long.Parse(ReadLine(input));
                            fs.CopyTo(output);
                        }
                        break;
                    }

                    case "tasklist":
                    {
                        Shell.Do(peer, input, output, "tasklist.exe", "", false);
                        break;
                    }

                    case "tasklistj":
                    {
                        Tasklist.JavaScript(output);
                        break;
                    }

                    case "tasklistl":
                    {
                        Shell.Do(peer, input, output, "tasklist.exe", "/FO LIST", false);
                        break;
                    }

                    case "tickcount":
                    {
                        var tc    = Environment.TickCount;
                        var now   = DateTime.Now;
                        var prev  = -(long)int.MinValue + tc;
                        var next  = (long)int.MaxValue - tc;
                        var bytes = Encoding.ASCII.GetBytes(Environment.MachineName + " Environment.TickCount is " + Environment.TickCount + " previous " + Date(now - TimeSpan.FromMilliseconds(prev)) + " next " + Date(now + TimeSpan.FromMilliseconds(next)) + " uptime " + GetUpTime() + "\r\n");
                        output.Write(bytes, 0, bytes.Length);
                        break;
                    }

                    case "tickcount2":
                    {
                        Write(Environment.TickCount.ToString(CultureInfo.InvariantCulture), output);
                        break;
                    }

                    case "tickcounts":
                    {
                        var tc   = Environment.TickCount;
                        var secs = ((double)int.MaxValue - tc) / 1000.0;
                        Write(secs.ToString(CultureInfo.InvariantCulture) + "\n", output);
                        Write(Date(DateTime.UtcNow.AddSeconds(secs)) + "\n", output);
                        break;
                    }

                    case "time":
                    {
                        Write(Environment.MachineName + " " + Date() + "\r\n", output);
                        break;
                    }

                    case "time2":
                    {
                        Write(Date(), output);
                        break;
                    }

                    case "update":
                    {
                        var sourcePath = ReadLine(input);
                        var destPath   = Path.GetDirectoryName(Uri.UnescapeDataString(new UriBuilder(Assembly.GetExecutingAssembly().CodeBase).Path));
                        var arguments  = string.Format("/C net stop \"{0}\" && timeout /t 5 /nobreak && move /y \"{1}\\*\" \"{2}\" && net start \"{0}\"", Program.ServiceName, sourcePath, destPath);
                        Shell.Spawn(arguments);
                        break;
                    }

                    case "upload":
                    {
                        var path = ReadLine(input);
                        Zip.Upload(path, input);
                        break;
                    }

                    case "uploadzip":
                    {
                        var path = ReadLine(input);
                        Zip.Unzipit(path, input);
                        break;
                    }

                    case "uptime":
                    {
                        Write(GetUpTime().ToString(), output);
                        break;
                    }

                    case "uptime2":
                    {
                        var tc = GetTickCount64();
                        Write((tc / 1000.0).ToString(CultureInfo.InvariantCulture) + "\n", output);
                        Write(TimeSpan.FromMilliseconds(tc) + "\n", output);
                        break;
                    }

                    case "uri":
                    {
                        var path = ReadLine(input);
                        FetchUri.Fetch(path, output);
                        break;
                    }

                    case "users":
                    {
                        Shell.Do(peer, input, output, "query.exe", "user", false);
                        break;
                    }

                    case "version":
                    case "version2":
                    case "version3":
                    {
                        var pid = GetPid(ReadLine(input));

                        if (pid != -1)
                        {
                            DetermineVersion.Do(pid, output, line == "version" ? " " : (line == "version2" ? null : "\n"));
                        }
                        break;
                    }

                    case "wertrace":
                    {
                        var process = ReadLine(input);
                        StackTrace.OpenWerDump(peer, output, process);
                        break;
                    }

                    default:
                        if (line.StartsWith("get /"))
                        {
                            Http.Do(lineOrig, input, fileMappings, colorMappings);
                        }
                        else
                        {
                            throw new InvalidOperationException($"Unknown command {line}.");
                        }
                        break;
                    }
                }
                catch (Exception e)
                {
                    if (crash)
                    {
                        throw;
                    }

                    if (!(e is IOException))
                    {
                        LogError(e);
                    }
                }
            }
        }
コード例 #2
0
        public static void Do(string getLine, Stream net, Dictionary <string, string> fileMappings, Dictionary <Regex, string> colorMappings)
        {
            var ns = net as NetworkStream;

            if (ns != null)
            {
                ns.ReadTimeout = 30 * 1000;
            }

            bool methodKnown = true;

            using (net)
            {
                while (true)
                {
                    List <string> headers = null;
                    byte[]        body;
                    int           cl           = -1;
                    bool          readAnything = methodKnown;
                    try
                    {
                        var type = string.Empty;

                        while (true)
                        {
                            var line = Bender.ReadLine(net);
                            if (string.IsNullOrEmpty(line))
                            {
                                break;
                            }
                            readAnything = true;
                            if (line.StartsWith("Get ", StringComparison.OrdinalIgnoreCase))
                            {
                                getLine     = line;
                                methodKnown = true;
                            }
                            else if (line.StartsWith("Post ", StringComparison.OrdinalIgnoreCase))
                            {
                                getLine     = null;
                                methodKnown = true;
                            }
                            if (line.StartsWith("Content-Length:", StringComparison.OrdinalIgnoreCase))
                            {
                                cl = int.Parse(line.Substring(15));
                            }
                            if (line.StartsWith("Content-Type:", StringComparison.OrdinalIgnoreCase))
                            {
                                type = line.Substring(14);
                            }
                            if (headers == null && methodKnown && !string.IsNullOrEmpty(getLine))
                            {
                                headers = new List <string> {
                                    getLine
                                };
                            }
                            else if (headers != null)
                            {
                                headers.Add(line);
                            }
                        }

                        if (!readAnything)
                        {
                            return;
                        }
                    }
                    catch (IOException)
                    {
                        return;
                    }

                    string contentType;

                    try
                    {
                        string commandString = string.Empty;
                        if (!string.IsNullOrEmpty(getLine))
                        {
                            commandString = HttpUtility.UrlDecode(getLine.Substring(4, getLine.Length - 13));
                        }
                        if (cl != -1)
                        {
                            var contents = new byte[cl];
                            var offset   = 0;
                            while (cl > 0)
                            {
                                var cl2 = net.Read(contents, offset, cl);
                                if (cl2 <= 0)
                                {
                                    throw new InvalidOperationException($"Unable to read {cl} bytes of input data. Read {cl2}.");
                                }
                                cl     -= cl2;
                                offset += cl2;
                            }
                            commandString = Encoding.UTF8.GetString(contents);
                        }

                        if (!methodKnown)
                        {
                            throw new InvalidOperationException("Illegal method");
                        }

                        if (headers == null)
                        {
                            JavaScriptSerializer ser = new JavaScriptSerializer();
                            var commands             = ser.DeserializeObject(commandString) as Dictionary <string, object>;
                            var tasks = new List <Task <Tuple <string, MemoryStream> > >();
                            foreach (var command in commands)
                            {
                                var cmds       = command.Value as object[];
                                var ms         = new MemoryStream();
                                var serializer = new StreamWriter(ms);
                                foreach (var cmd in cmds)
                                {
                                    serializer.WriteLine(cmd as string);
                                }
                                serializer.Flush();
                                ms.Position = 0;
                                var key  = command.Key;
                                var task = Task.Factory.StartNew(() =>
                                {
                                    var rs = new MemoryStream();
                                    Bender.DoCommand(null, ms, rs, fileMappings, colorMappings);
                                    return(Tuple.Create(key, rs));
                                });
                                tasks.Add(task);
                            }

                            var js     = string.Empty;
                            var result = new Dictionary <string, List <string> >();
                            foreach (var task in tasks)
                            {
                                var key = task.Result.Item1;
                                var stm = task.Result.Item2;
                                stm = new MemoryStream(stm.ToArray());
                                var lines = new List <string>();
                                using (var rdr = new StreamReader(stm))
                                {
                                    while (true)
                                    {
                                        var l = rdr.ReadLine();
                                        if (l == null)
                                        {
                                            break;
                                        }
                                        lines.Add(l);
                                    }
                                }
                                if (key.Equals("tasklistj", StringComparison.InvariantCultureIgnoreCase) && lines.Count == 1)
                                {
                                    js = lines[0];
                                }

                                result.Add(key, lines);

                                if (result.Count > 1)
                                {
                                    js = string.Empty;
                                }
                            }

                            if (string.IsNullOrEmpty(js))
                            {
                                js = ser.Serialize(result);
                            }

                            contentType = "Content-Type: application/json; charset=UTF-8";
                            body        = Encoding.UTF8.GetBytes(js);
                        }
                        else
                        {
                            body        = null;
                            contentType = null;

                            var index       = commandString.IndexOf('?');
                            var paramString = string.Empty;
                            if (index != -1)
                            {
                                paramString   = commandString.Substring(index + 1);
                                commandString = commandString.Substring(0, index);
                            }

                            if (commandString.Equals("/log", StringComparison.OrdinalIgnoreCase))
                            {
                                const string defaults       = "lines=80&tail=0&scroll=1&bw=1&newlines=1&text=0";
                                var          appendLocation = string.Empty;

                                if (!ParseParams(paramString, string.Empty).Keys.Intersect(ParseParams(defaults, string.Empty).Keys).Any())
                                {
                                    appendLocation = "&" + defaults;
                                }

                                var param  = ParseParams(paramString, defaults);
                                var bw     = param["bw"] != "0";
                                var scroll = param["scroll"] != "0";
                                var file   = param.ContainsKey("file") ? param["file"] : null;
                                var lines  = param.ContainsKey("val") ? param["val"] : param["lines"];
                                var tail   = param["tail"];
                                var text   = param["text"] != "0";

                                var newLines = param["newlines"] != "0";

                                if (lines == "-f")
                                {
                                    lines = "40";
                                    tail  = "1";
                                }

                                var serverPath = Bender.ReadServerPath(file, fileMappings);
                                var server     = param.ContainsKey("server") ? param["server"] : serverPath.Item1;
                                var path       = serverPath.Item2;
                                var logOut     = new LogOutput(net, new LogOutput.Format {
                                    AppendLocation = appendLocation, ColorMappings = bw ? null : colorMappings, Scroll = scroll, NewLine = newLines, Title = path, PlainText = text
                                });

                                FileTailer.Tail(server, path, int.Parse(lines), int.Parse(tail) > 0, (bytes, i) =>
                                {
                                    logOut.Add(bytes, 0, i);
                                });

                                logOut.End();
                            }
                            else
                            {
                                var output = new MemoryStream();
                                var title  = string.Empty;

                                var param = ParseParams(paramString, string.Empty);

                                if (commandString.Equals("/stack", StringComparison.OrdinalIgnoreCase))
                                {
                                    StackTrace.DoManaged(null, output, Bender.GetPid2(param["exe"]));
                                    title = $"Stacktrace for {param["exe"]}";
                                }
                                else if (commandString.Equals("/wer", StringComparison.OrdinalIgnoreCase))
                                {
                                    StackTrace.OpenWerDump(null, output, param["exe"]);
                                    title = $"Werstack for {param["exe"]}";
                                }
                                else if (commandString.Equals("/get", StringComparison.OrdinalIgnoreCase))
                                {
                                    var serverPath = Bender.ReadServerPath(param["uri"], fileMappings);
                                    // var server = serverPath.Item1;
                                    var path = serverPath.Item2;
                                    FetchUri.Fetch(path, output);
                                    title = $"GET {path}";
                                }

                                var arr = output.ToArray();
                                if (arr.Length > 0)
                                {
                                    var str    = Encoding.UTF8.GetString(arr);
                                    var logOut = new LogOutput(net, new LogOutput.Format {
                                        Title = title, PlainText = true
                                    });
                                    logOut.Add(str);
                                    logOut.End();
                                }
                                else
                                {
                                    contentType = "Content-Type: text/plain; charset=UTF-8";
                                    var sb = new StringBuilder();
                                    foreach (var h in headers)
                                    {
                                        sb.AppendLine(h);
                                    }
                                    body = Encoding.UTF8.GetBytes(sb.ToString());
                                }
                            }
                        }
                    }
                    catch (Exception)
                    {
                        Write(net, "HTTP/1.1 500 Internal server error\nConnection: Close\n\n", null);
                        throw;
                    }

                    if (contentType != null && body != null)
                    {
                        var header = $"HTTP/1.1 200 OK\nAccess-Control-Allow-Origin: *\n{contentType}\nContent-Length: {body.Length}\n\n";
                        Write(net, header, body);
                    }
                    methodKnown = false;
                }
            }
        }