MARKUS

using System;

using System.Collections.Generic;

using System.Diagnostics;

using System.Linq;

using System.Management;

using System.Text;

using System.Threading.Tasks;

namespace Markus.Service.Extensions

{

    public static class ProcessExtension

    {

        /// <summary>

        /// 프로세스의 명령줄을 가져온다.

        /// </summary>

        /// <param name="process"></param>

        /// <returns></returns>

        public static ProcessArguments Arguments(this Process process)

        {

            ProcessArguments result = new ProcessArguments();

            var commandLine = ProcessCommandLinesQuery(process.Id);

            try

            {

                result.WorkingDirectory = ProcessWorkingDirectory(commandLine);

                result.CommandLine = SplitArguments(RemoveCommadLinesFilePath(commandLine)).ToList();

            }

            catch (Exception ex)

            {

                System.Diagnostics.Debug.WriteLine(ex.ToString());

            }

            return result;

        }

        /// <summary>

        /// 프로세스의 실행 경로

        /// </summary>

        /// <param name="process"></param>

        /// <returns></returns>

        public static string WorkingDirectory(this Process process)

        {

            var commandLine = ProcessCommandLinesQuery(process.Id);

            return ProcessWorkingDirectory(commandLine);

        }

        private static string ProcessCommandLinesQuery(int processId)

        {

            string result = "";

            string wmiQuery = string.Format("select CommandLine from Win32_Process where ProcessId ='{0}'", processId);

            ManagementObjectSearcher searcher = new ManagementObjectSearcher(wmiQuery);

            ManagementObjectCollection retObjectCollection = searcher.Get();

            if (retObjectCollection.Count > 0)

            {

                if (retObjectCollection.Cast<ManagementObject>().First()["CommandLine"] != null)

                {

                    result = retObjectCollection.Cast<ManagementObject>().First()["CommandLine"]?.ToString();

                }

                else

                {

                    result = "";

                }

            }

            return result;

        }

        private static string ProcessWorkingDirectory(string commandLine)

        {

            var splitCommand = commandLine.Split(' ');

            string filePath = "";

            for (int i = 0; i < splitCommand.Count(); i++)

            {

                filePath += " " + splitCommand[i].Replace("\"", "");

                if (IO.FileExists(filePath))

                {

                    System.IO.FileInfo info = new System.IO.FileInfo(filePath);

                    filePath = info.DirectoryName;

                    break;

                }

                else if (i == splitCommand.Count() - 1)

                {

                    var firstCmd = splitCommand.First();

                    if (firstCmd.StartsWith("\"") && firstCmd.EndsWith("\""))

                    {

                        filePath = firstCmd.Replace("\"", "");

                    }

                }

                else

                {

                    filePath = commandLine;

                }

            }

            return filePath;

        }

        /// <summary>

        /// command line에서 처음에 있는 파일 경로가 \"로 묶여 있거나 묶여 있지 않은 경우가 있다.

        /// 파일 경로를 제거하기 위한 코드

        /// </summary>

        /// <param name="commandLine"></param>

        /// <returns></returns>

        private static string RemoveCommadLinesFilePath(string commandLine)

        {

            string result = "";

            var splitCommand = commandLine.Split(' ');

            string filePath = "";

            for (int i = 0; i < splitCommand.Count(); i++)

            {

                filePath += " " + splitCommand[i];

                if (IO.FileExists(filePath.Replace("\"", "")))

                {

                    result = string.Join(" ", splitCommand.Skip(i + 1).Take(splitCommand.Count() - i));

                    break;

                }

                else if (i == splitCommand.Count() - 1)

                {

                    var firstCmd = splitCommand.First();

                    if (firstCmd.StartsWith("\"") && firstCmd.EndsWith("\""))

                    {

                        result = string.Join(" ", splitCommand.Skip(1).Take(splitCommand.Count() - 1));

                    }

                }

                else

                {

                    result = commandLine;

                }

            }

            return result;

        }

        public static string[] SplitArguments(string commandLine)

        {

            List<string> args = new List<string>();

            List<char> currentArg = new List<char>();

            char? quoteSection = null; // Keeps track of a quoted section (and the type of quote that was used to open it)

            char[] quoteChars = new[] { '\'', '\"' };

            char previous = ' '; // Used for escaping double quotes

            for (var index = 0; index < commandLine.Length; index++)

            {

                char c = commandLine[index];

                if (quoteChars.Contains(c))

                {

                    if (previous == c) // Escape sequence detected

                    {

                        previous = ' '; // Prevent re-escaping

                        if (!quoteSection.HasValue)

                        {

                            quoteSection = c; // oops, we ended the quoted section prematurely

                            continue;         // don't add the 2nd quote (un-escape)

                        }

                        if (quoteSection.Value == c)

                            quoteSection = null; // appears to be an empty string (not an escape sequence)

                    }

                    else if (quoteSection.HasValue)

                    {

                        if (quoteSection == c)

                            quoteSection = null; // End quoted section

                    }

                    else

                        quoteSection = c; // Start quoted section

                }

                else if (char.IsWhiteSpace(c))

                {

                    if (!quoteSection.HasValue)

                    {

                        args.Add(new string(currentArg.ToArray()));

                        currentArg.Clear();

                        previous = c;

                        continue;

                    }

                }

                currentArg.Add(c);

                previous = c;

            }

            if (currentArg.Count > 0)

                args.Add(new string(currentArg.ToArray()));

            return args.ToArray();

        }

    }

}