aboutsummaryrefslogtreecommitdiffstats
path: root/Software/Visual_Studio/VSIX/Tango.BuildExtensions/RemoteDebugCommand.cs
blob: 8c794daed5008883dc98bd5f272c2811e84f51c3 (plain)
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
using Google.Protobuf;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Runtime.InteropServices;
using System.Runtime.ExceptionServices;

namespace Tango.PMR
{
    /// <summary>
    /// Represents a standard C++ protobuf method delegate.
    /// </summary>
    /// <param name="requestArray">The request array.</param>
    /// <param name="requestArraySize">Size of the request array.</param>
    /// <param name="resultArray">The result array.</param>
    /// <returns>The size of the result array.</returns>
    [UnmanagedFunctionPointer(CallingConvention.Cdecl)]
    public delegate int NativeMethodDelegate(IntPtr requestArraypre { line-height: 125%; }
td.linenos .normal { color: inherit; background-color: transparent; padding-left: 5px; padding-right: 5px; }
span.linenos { color: inherit; background-color: transparent; padding-left: 5px; padding-right: 5px; }
td.linenos .special { color: #000000; background-color: #ffffc0; padding-left: 5px; padding-right: 5px; }
span.linenos.special { color: #000000; background-color: #ffffc0; padding-left: 5px; padding-right: 5px; }
.highlight .hll { background-color: #ffffcc }
.highlight .c { color: #888888 } /* Comment */
.highlight .err { color: #a61717; background-color: #e3d2d2 } /* Error */
.highlight .k { color: #008800; font-weight: bold } /* Keyword */
.highlight .ch { color: #888888 } /* Comment.Hashbang */
.highlight .cm { color: #888888 } /* Comment.Multiline */
.highlight .cp { color: #cc0000; font-weight: bold } /* Comment.Preproc */
.highlight .cpf { color: #888888 } /* Comment.PreprocFile */
.highlight .c1 { color: #888888 } /* Comment.Single */
.highlight .cs { color: #cc0000; font-weight: bold; background-color: #fff0f0 } /* Comment.Special */
.highlight .gd { color: #000000; background-color: #ffdddd } /* Generic.Deleted */
.highlight .ge { font-style: italic } /* Generic.Emph */
.highlight .ges { font-weight: bold; font-style: italic } /* Generic.EmphStrong */
.highlight .gr { color: #aa0000 } /* Generic.Error */
.highlight .gh { color: #333333 } /* Generic.Heading */
.highlight .gi { color: #000000; background-color: #ddffdd } /* Generic.Inserted */
.highlight .go { color: #888888 } /* Generic.Output */
.highlight .gp { color: #555555 } /* Generic.Prompt */
.highlight .gs { font-weight: bold } /* Generic.Strong */
.highlight .gu { color: #666666 } /* Generic.Subheading */
.highlight .gt { color: #aa0000 } /* Generic.Traceback */
.highlight .kc { color: #008800; font-weight: bold } /* Keyword.Constant */
.highlight .kd { color: #008800; font-weight: bold } /* Keyword.Declaration */
.highlight .kn { color: #008800; font-weight: bold } /* Keyword.Namespace */
.highlight .kp { color: #008800 } /* Keyword.Pseudo */
.highlight .kr { color: #008800; font-weight: bold } /* Keyword.Reserved */
.highlight .kt { color: #888888; font-weight: bold } /* Keyword.Type */
.highlight .m { color: #0000DD; font-weight: bold } /* Literal.Number */
.highlight .s { color: #dd2200; background-color: #fff0f0 } /* Literal.String */
.highlight .na { color: #336699 } /* Name.Attribute */
.highlight .nb { color: #003388 } /* Name.Builtin */
.highlight .nc { color: #bb0066; font-weight: bold } /* Name.Class */
.highlight .no { color: #003366; font-weight: bold } /* Name.Constant */
.highlight .nd { color: #555555 } /* Name.Decorator */
.highlight .ne { color: #bb0066; font-weight: bold } /* Name.Exception */
.highlight .nf { color: #0066bb; font-weight: bold } /* Name.Function */
.highlight .nl { color: #336699; font-style: italic } /* Name.Label */
.highlight .nn { color: #bb0066; font-weight: bold } /* Name.Namespace */
.highlight .py { color: #336699; font-weight: bold } /* Name.Property */
.highlight .nt { color: #bb0066; font-weight: bold } /* Name.Tag */
.highlight .nv { color: #336699 } /* Name.Variable */
.highlight .ow { color: #008800 } /* Operator.Word */
.highlight .w { color: #bbbbbb } /* Text.Whitespace */
.highlight .mb { color: #0000DD; font-weight: bold } /* Literal.Number.Bin */
.highlight .mf { color: #0000DD; font-weight: bold } /* Literal.Number.Float */
.highlight .mh { color: #0000DD; font-weight: bold } /* Literal.Number.Hex */
.highlight .mi { color: #0000DD; font-weight: bold } /* Literal.Number.Integer */
.highlight .mo { color: #0000DD; font-weight: bold } /* Literal.Number.Oct */
.highlight .sa { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Affix */
.highlight .sb { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Backtick */
.highlight .sc { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Char */
.highlight .dl { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Delimiter */
.highlight .sd { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Doc */
.highlight .s2 { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Double */
.highlight .se { color: #0044dd; background-color: #fff0f0 } /* Literal.String.Escape */
.highlight .sh { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Heredoc */
.highlight .si { color: #3333bb; background-color: #fff0f0 } /* Literal.String.Interpol */
.highlight .sx { color: #22bb22; background-color: #f0fff0 } /* Literal.String.Other */
.highlight .sr { color: #008800; background-color: #fff0ff } /* Literal.String.Regex */
.highlight .s1 { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Single */
.highlight .ss { color: #aa6600; background-color: #fff0f0 } /* Literal.String.Symbol */
.highlight .bp { color: #003388 } /* Name.Builtin.Pseudo */
.highlight .fm { color: #0066bb; font-weight: bold } /* Name.Function.Magic */
.highlight .vc { color: #336699 } /* Name.Variable.Class */
.highlight .vg { color: #dd7700 } /* Name.Variable.Global */
.highlight .vi { color: #3333bb } /* Name.Variable.Instance */
.highlight .vm { color: #336699 } /* Name.Variable.Magic */
.highlight .il { color: #0000DD; font-weight: bold } /* Literal.Number.Integer.Long */
//------------------------------------------------------------------------------
// <copyright file="RemoteDebugCommand.cs" company="Company">
//     Copyright (c) Company.  All rights reserved.
// </copyright>
//------------------------------------------------------------------------------

using System;
using System.ComponentModel.Design;
using System.Globalization;
using Microsoft.VisualStudio.Shell;
using Microsoft.VisualStudio.Shell.Interop;
using System.Linq;
using System.IO;
using System.Diagnostics;
using Tango.Transport;
using Tango.Transport.Adapters;
using Tango.Transport.Transporters;
using System.IO.Compression;
using Tango.Core.IO;
using Tango.PMR.IO;
using Google.Protobuf;
using EnvDTE;
using System.Collections.Generic;

namespace Tango.BuildExtensions
{
    /// <summary>
    /// Command handler
    /// </summary>
    internal sealed class RemoteDebugCommand : VSIXBase
    {
        private DebuggerEvents _dteDebuggerEvents;
        private ITransporter transporter;
        private bool _attached;
        private string processId;
        private OleMenuCommand _menuCommand;
        private IList<Project> _projects;


        /// <summary>
        /// Command ID.
        /// </summary>
        public const int CommandId = 4129;

        /// <summary>
        /// Command menu group (command set GUID).
        /// </summary>
        public static readonly Guid CommandSet = new Guid("c03a7b01-8109-4ec5-8f90-858bed027e5d");

        /// <summary>
        /// VS Package that provides this command, not null.
        /// </summary>
        private readonly Package package;

        /// <summary>
        /// Initializes a new instance of the <see cref="RemoteDebugCommand"/> class.
        /// Adds our command handlers for menu (commands must exist in the command table file)
        /// </summary>
        /// <param name="package">Owner package, not null.</param>
        private RemoteDebugCommand(Package package) : base(package)
        {
            if (package == null)
            {
                throw new ArgumentNullException("package");
            }

            _dteDebuggerEvents = DTE.Events.DebuggerEvents;

            _dteDebuggerEvents.OnEnterDesignMode += _dteDebuggerEvents_OnEnterDesignMode;

            DTE.Events.SolutionEvents.Opened += () =>
            {
                _projects = GetSolutionProjects().ToList();
            };

            this.package = package;

            OleMenuCommandService commandService = this.ServiceProvider.GetService(typeof(IMenuCommandService)) as OleMenuCommandService;
            if (commandService != null)
            {
                var menuCommandID = new CommandID(CommandSet, CommandId);
                _menuCommand = new OleMenuCommand(this.MenuItemCallback, menuCommandID);
                _menuCommand.Visible = false;
                _menuCommand.BeforeQueryStatus += _menuCommand_BeforeQueryStatus;
                commandService.AddCommand(_menuCommand);
            }
        }

        private void _menuCommand_BeforeQueryStatus(object sender, EventArgs e)
        {
            (sender as OleMenuCommand).Visible = false;

            if (DTE.Debugger.CurrentMode == dbgDebugMode.dbgDesignMode
                && DTE.Solution != null
                && DTE.Solution.IsOpen)
            {
                String projectName = DTE.Solution.Properties.Item("StartupProject").Value.ToStringSafe();

                if (projectName != null)
                {
                    var project = _projects.SingleOrDefault(x => x.Name == projectName);
                    String path = GetProjectOutputFilePath(project);

                    if (Path.GetExtension(path.ToLower()) == ".exe")
                    {
                        (sender as OleMenuCommand).Visible = true;
                    }
                }
            }
        }

#pragma warning disable VSTHRD100 // Avoid async void methods
        private async void _dteDebuggerEvents_OnEnterDesignMode(dbgEventReason Reason)
#pragma warning restore VSTHRD100 // Avoid async void methods
        {
            if (_attached)
            {
                if (Reason == dbgEventReason.dbgEventReasonStopDebugging)
                {
                    _attached = false;

                    if (transporter != null)
                    {
                        OpenVSProgress("Tango Remote Runner", "Closing remote process...", true);

                        await transporter.SendRequest<KillProcessRequest, KillProcessResponse>(new KillProcessRequest()
                        {
                            ProcessID = processId,
                        });
                        await transporter.Disconnect();
                        transporter = null;

                        await System.Threading.Tasks.Task.Delay(2000);

                        CloseVSProgress();
                    }
                }
            }
        }

        /// <summary>
        /// Gets the instance of the command.
        /// </summary>
        public static RemoteDebugCommand Instance
        {
            get;
            private set;
        }

        /// <summary>
        /// Gets the service provider from the owner package.
        /// </summary>
        private IServiceProvider ServiceProvider
        {
            get
            {
                return this.package;
            }
        }

        /// <summary>
        /// Initializes the singleton instance of the command.
        /// </summary>
        /// <param name="package">Owner package, not null.</param>
        public static void Initialize(Package package)
        {
            Instance = new RemoteDebugCommand(package);
        }

        // %ifnot% $toolWindow$
        /// <summary>
        /// This function is the callback used to execute the command when the menu item is clicked.
        /// See the constructor to see how the menu item is associated with this function using
        /// OleMenuCommandService service and MenuCommand class.
        /// </summary>
        /// <param name="sender">Event sender.</param>
        /// <param name="e">Event args.</param>
        private void MenuItemCallback(object sender, EventArgs e)
        {
            RunRemote();
        }

        private void RunRemote()
        {
            _attached = false;

            try
            {
                String projectName = DTE.Solution.Properties.Item("StartupProject").Value.ToString();

                RemoteDebugForm dlg = new RemoteDebugForm(projectName);

                if (dlg.ShowDialog() == System.Windows.Forms.DialogResult.OK)
                {
#pragma warning disable VSTHRD105 // Avoid method overloads that assume TaskScheduler.Current
                    System.Threading.Tasks.Task.Factory.StartNew(async () =>
#pragma warning restore VSTHRD105 // Avoid method overloads that assume TaskScheduler.Current
                    {
                        try
                        {
                            OpenVSProgress("Tango Remote Runner", "Connecting to " + dlg.SelectedRemoteRunner.HostName + "...", true);

                            ITransportAdapter adapter = new TcpTransportAdapter(dlg.SelectedRemoteRunner.Address, dlg.SelectedRemoteRunner.Port);
                            transporter = new BasicTransporter(adapter);
                            await transporter.Connect();

                            SetVSProgress("Building " + projectName + "...");

                            var project = GetSolutionProjects().SingleOrDefault(x => x.Name == projectName);

                            String filePath = GetProjectOutputFilePath(project);
                            String folder = Path.GetDirectoryName(filePath);
                            String fileName = Path.GetFileName(filePath);

                            DTE.Solution.SolutionBuild.BuildProject("Debug", project.FullName, true);

                            var tempFolder = TemporaryManager.Default.CreateFolder();
                            var zipFile = tempFolder.CreateFile(".zip");

                            await zipFile.DeleteAsync();
                            SetVSProgress("Packaging project...");
                            ZipFile.CreateFromDirectory(folder, zipFile.Path);

                            System.IO.FileInfo zipFileInfo = new System.IO.FileInfo(zipFile);

                            CloseVSProgress();

                            OpenVSProgress("Tango Remote Runner", "Uploading project...", false);

                            var upResponse = await transporter.SendRequest<FileUploadRequest, FileUploadResponse>(new FileUploadRequest()
                            {
                                Path = zipFile.FileName,
                                Length = zipFileInfo.Length,
                            });

                            var uploadResponse = upResponse.Message;

                            int currentFilePosition = 0;

                            do
                            {
                                using (FileStream fs = new FileStream(zipFileInfo.FullName, FileMode.Open))
                                {
                                    fs.Position = currentFilePosition;

                                    FileChunkUploadRequest chunkRequest = new FileChunkUploadRequest();
                                    chunkRequest.UploadID = uploadResponse.UploadID;

                                    long remaining = zipFileInfo.Length - currentFilePosition;

                                    byte[] buffer = new byte[uploadResponse.MaxChunkLength < remaining ? uploadResponse.MaxChunkLength : remaining];
                                    await fs.ReadAsync(buffer, 0, buffer.Length);
                                    chunkRequest.Buffer = ByteString.CopyFrom(buffer);

                                    await transporter.SendRequest<FileChunkUploadRequest, FileChunkUploadResponse>(chunkRequest);

                                    currentFilePosition += buffer.Length;

                                    SetVSProgress("Uploading project...", "", currentFilePosition, (int)zipFileInfo.Length);
                                }

                            } while (currentFilePosition < zipFileInfo.Length);

                            Wait(1000);

                            CloseVSProgress();

                            OpenVSProgress("Tango Remote Runner", "Executing remote process...", true);

                            var response = await transporter.SendRequest<ExecuteProcessRequest, ExecuteProcessResponse>(new ExecuteProcessRequest()
                            {
                                UploadID = uploadResponse.UploadID,
                                FileName = fileName,
                            }, TimeSpan.FromSeconds(30));

                            processId = response.Message.ProcessID;

                            SetVSProgress("Attaching Debugger...");

                            var debugger = DTE.Debugger as EnvDTE80.Debugger2;
                            var transport = debugger.Transports.Item("Remote");
                            var process = debugger.GetProcesses(transport, dlg.SelectedRemoteRunner.Address).Item(fileName) as EnvDTE80.Process2;
                            process.Attach();
                            _attached = true;


                            CloseVSProgress();
                        }
                        catch (Exception ex)
                        {
                            CloseVSProgress();
                            ShowMessage(ex.Message);
                        }
                    });
                }
            }
            catch (Exception ex)
            {
                CloseVSProgress();
                ShowMessage(ex.Message);
            }
        }
    }
}