/* Copyright 2009-2017 Intel Corporation Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. */ using System; using System.Net; using System.Reflection; using System.Collections; using System.Windows.Forms; using System.Security.Cryptography.X509Certificates; using Microsoft.Win32; namespace MeshCentralRouter { public partial class MainForm : Form { private int currentPanel = 0; private DateTime refreshTime = DateTime.Now; private MeshCentralServer meshcentral = null; private X509Certificate2 lastBadConnectCert = null; private string title; private string[] args; private bool debug = false; private bool autoLogin = false; private bool ignoreCert = false; private bool inaddrany = false; private bool forceExit = false; public static void saveToRegistry(string name, string value) { try { Registry.SetValue(@"HKEY_CURRENT_USER\SOFTWARE\OpenSource\MeshRouter", name, value); } catch (Exception) { } } public static string loadFromRegistry(string name) { try { return Registry.GetValue(@"HKEY_CURRENT_USER\SOFTWARE\OpenSource\MeshRouter", name, "").ToString(); } catch (Exception) { return ""; } } public MainForm(string[] args) { this.args = args; InitializeComponent(); mainPanel.Controls.Add(panel1); mainPanel.Controls.Add(panel2); mainPanel.Controls.Add(panel3); mainPanel.Controls.Add(panel4); pictureBox1.SendToBack(); Version version = Assembly.GetEntryAssembly().GetName().Version; versionLabel.Text = "v" + version.Major + "." + version.Minor + "." + version.Build; serverNameComboBox.Text = loadFromRegistry("ServerName"); userNameTextBox.Text = loadFromRegistry("UserName"); title = this.Text; int argflags = 0; foreach (string arg in this.args) { if (arg.ToLower() == "-debug") { debug = true; } if (arg.ToLower() == "-ignorecert") { ignoreCert = true; } if (arg.ToLower() == "-all") { inaddrany = true; } if (arg.ToLower() == "-inaddrany") { inaddrany = true; } if (arg.ToLower() == "-tray") { notifyIcon.Visible = true; this.ShowInTaskbar = false; this.MinimizeBox = false; } if (arg.Length > 6 && arg.Substring(0, 6).ToLower() == "-host:") { serverNameComboBox.Text = arg.Substring(6); argflags |= 1; } if (arg.Length > 6 && arg.Substring(0, 6).ToLower() == "-user:") { userNameTextBox.Text = arg.Substring(6); argflags |= 2; } if (arg.Length > 6 && arg.Substring(0, 6).ToLower() == "-pass:") { passwordTextBox.Text = arg.Substring(6); argflags |= 4; } } autoLogin = (argflags == 7); } private void setPanel(int newPanel) { if (currentPanel == newPanel) return; if (newPanel == 4) { updatePanel4(); } panel1.Visible = (newPanel == 1); panel2.Visible = (newPanel == 2); panel3.Visible = (newPanel == 3); panel4.Visible = (newPanel == 4); currentPanel = newPanel; } private void MainForm_Load(object sender, EventArgs e) { //Text += " - v" + Application.ProductVersion; //installPathTextBox.Text = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.ProgramFiles), "Open Source", "MeshCentral"); //serverModeComboBox.SelectedIndex = 0; //windowColor = serverNameTextBox.BackColor; setPanel(1); updatePanel1(null, null); // Start the multicast scanner //scanner = new MeshDiscovery(); //scanner.OnNotify += Scanner_OnNotify; //scanner.MulticastPing(); if (autoLogin) { nextButton1_Click(null, null); } } private void updatePanel1(object sender, EventArgs e) { bool ok = true; if (serverNameComboBox.Text.Length == 0) { ok = false; } if (userNameTextBox.Text.Length == 0) { ok = false; } if (passwordTextBox.Text.Length == 0) { ok = false; } nextButton1.Enabled = ok; } private void updatePanel2(object sender, EventArgs e) { bool ok = true; if (tokenTextBox.Text.Length == 0) { ok = false; } nextButton2.Enabled = ok; } private void updatePanel4() { //ServerState s = readServerStateEx(installPathTextBox.Text); //if (s.state == ServerStateEnum.Running) { label7.Text = "MeshCentral is running this computer."; } //else if (s.state == ServerStateEnum.Unknown) { label7.Text = "MeshCentral is installed on this computer."; } } private void exitToolStripMenuItem_Click(object sender, EventArgs e) { forceExit = true; Application.Exit(); } private void MainForm_FormClosing(object sender, FormClosingEventArgs e) { if ((notifyIcon.Visible == true) && (forceExit == false)) { e.Cancel = true; Visible = false; } } private void backButton5_Click(object sender, EventArgs e) { meshcentral.disconnect(); } private void nextButton1_Click(object sender, EventArgs e) { // Attempt to login addButton.Enabled = false; addRelayButton.Enabled = false; openWebSiteButton.Visible = false; Uri serverurl = new Uri("wss://" + serverNameComboBox.Text + "/control.ashx"); meshcentral = new MeshCentralServer(); meshcentral.debug = debug; meshcentral.ignoreCert = ignoreCert; meshcentral.onStateChanged += Meshcentral_onStateChanged; meshcentral.onNodesChanged += Meshcentral_onNodesChanged; meshcentral.onLoginTokenChanged += Meshcentral_onLoginTokenChanged; if (lastBadConnectCert != null) { meshcentral.okCertHash = lastBadConnectCert.GetCertHashString(); } meshcentral.connect(serverurl, userNameTextBox.Text, passwordTextBox.Text, null); } private void nextButton3_Click(object sender, EventArgs e) { // Attempt to login, ignore bad cert. addButton.Enabled = false; addRelayButton.Enabled = false; openWebSiteButton.Visible = false; Uri serverurl = new Uri("wss://" + serverNameComboBox.Text + "/control.ashx"); meshcentral = new MeshCentralServer(); meshcentral.debug = debug; meshcentral.ignoreCert = ignoreCert; meshcentral.onStateChanged += Meshcentral_onStateChanged; meshcentral.onNodesChanged += Meshcentral_onNodesChanged; meshcentral.onLoginTokenChanged += Meshcentral_onLoginTokenChanged; meshcentral.okCertHash = lastBadConnectCert.GetCertHashString(); meshcentral.connect(serverurl, userNameTextBox.Text, passwordTextBox.Text, null); } private void Meshcentral_onLoginTokenChanged() { if (this.InvokeRequired) { this.Invoke(new MeshCentralServer.onLoginTokenChangedHandler(Meshcentral_onLoginTokenChanged)); return; } openWebSiteButton.Visible = true; } private void Meshcentral_onNodesChanged() { if (this.InvokeRequired) { this.Invoke(new MeshCentralServer.onNodeListChangedHandler(Meshcentral_onNodesChanged)); return; } addRelayButton.Enabled = addButton.Enabled = ((meshcentral.nodes != null) && (meshcentral.nodes.Count > 0)); // Update any active mappings foreach (Control c in mapPanel.Controls) { if (c.GetType() == typeof(MapUserControl)) { MapUserControl cc = (MapUserControl)c; cc.UpdateInfo(); } } addArgMappings(); reconnectUdpMaps(); } private void Meshcentral_onStateChanged(int state) { if (this.InvokeRequired) { this.Invoke(new MeshCentralServer.onStateChangedHandler(Meshcentral_onStateChanged), state); return; } if (state == 0) { if (meshcentral.disconnectMsg == "tokenrequired") { tokenTextBox.Text = ""; setPanel(2); tokenTextBox.Focus(); } else { setPanel(1); } if ((meshcentral.disconnectMsg != null) && meshcentral.disconnectMsg.StartsWith("noauth")) { stateLabel.Text = "Invalid username or password"; stateLabel.Visible = true; stateClearTimer.Enabled = true; serverNameComboBox.Focus(); } else if (meshcentral.disconnectMsg == "cert") { lastBadConnectCert = meshcentral.disconnectCert; certDetailsTextBox.Text = "---Issuer---\r\n" + lastBadConnectCert.Issuer.Replace(", ", "\r\n") + "\r\n\r\n---Subject---\r\n" + lastBadConnectCert.Subject.Replace(", ", "\r\n"); setPanel(3); certDetailsButton.Focus(); } else if (meshcentral.disconnectMsg == null) { stateLabel.Text = "Unable to connect"; stateLabel.Visible = true; stateClearTimer.Enabled = true; serverNameComboBox.Focus(); } // Clean up the UI nextButton1.Enabled = true; serverNameComboBox.Enabled = true; userNameTextBox.Enabled = true; passwordTextBox.Enabled = true; this.Text = title; // Clean up all mappings foreach (Control c in mapPanel.Controls) { if (c.GetType() == typeof(MapUserControl)) { ((MapUserControl)c).Dispose(); ((MapUserControl)c).Dispose(); mapPanel.Controls.Remove(c); } } noMapLabel.Visible = true; // Clean up the server cookieRefreshTimer.Enabled = false; meshcentral.onStateChanged -= Meshcentral_onStateChanged; meshcentral.onNodesChanged -= Meshcentral_onNodesChanged; meshcentral = null; } else if (state == 1) { stateLabel.Visible = false; //setPanel(1); nextButton1.Enabled = false; serverNameComboBox.Enabled = false; userNameTextBox.Enabled = false; passwordTextBox.Enabled = false; cookieRefreshTimer.Enabled = false; } else if (state == 2) { meshcentral.disconnectMsg = "connected"; stateLabel.Visible = false; setPanel(4); addButton.Focus(); saveToRegistry("ServerName", serverNameComboBox.Text); saveToRegistry("UserName", userNameTextBox.Text); this.Text = title + " - " + userNameTextBox.Text; cookieRefreshTimer.Enabled = true; } } private void reconnectUdpMaps() { foreach (Control c in mapPanel.Controls) { if (c == noMapLabel) continue; MapUserControl map = (MapUserControl)c; if ((map.protocol == 2) && (map.mapper.totalConnectCounter == 0)) { // This is an unconnected UDP map, check if the target node is connected. foreach (NodeClass n in meshcentral.nodes.Values) { if ((map.node == n) && ((n.conn & 1) != 0)) { // Reconnect the UDP map map.Start(); } } } } } private ArrayList processedArgs = new ArrayList(); private void addArgMappings() { // Add mappings for (int i = 0; i < this.args.Length; i++) { if (processedArgs.Contains(i)) { continue; } // This map was already added string arg = this.args[i]; if (arg.Length > 5 && arg.Substring(0, 5).ToLower() == "-map:") { string[] x = arg.Substring(5).Split(':'); if (x.Length == 5) { // Protocol int protocol = 0; if (x[0].ToLower() == "tcp") { protocol = 1; } if (x[0].ToLower() == "udp") { protocol = 2; } if (protocol == 0) continue; // LocalPort ushort localPort = 0; if (ushort.TryParse(x[1], out localPort) == false) continue; // Node string nodename = x[2]; NodeClass node = null; foreach (NodeClass n in meshcentral.nodes.Values) { if (((n.conn & 1) != 0) && (n.name.ToLower() == nodename.ToLower())) { node = n; } } if (node == null) continue; // AppId int appId = 0; if (protocol == 1) { if (x[3].ToLower() == "http") { appId = 1; } else if (x[3].ToLower() == "https") { appId = 2; } else if (x[3].ToLower() == "rdp") { appId = 3; } else if (x[3].ToLower() == "putty") { appId = 4; } else if (x[3].ToLower() == "winscp") { appId = 5; } } // RemotePort ushort remotePort = 0; if (ushort.TryParse(x[4], out remotePort) == false) continue; // Add a new port map MapUserControl map = new MapUserControl(); map.xdebug = debug; map.inaddrany = inaddrany; map.protocol = protocol; map.localPort = (int)localPort; map.remotePort = (int)remotePort; map.appId = appId; map.node = node; map.host = serverNameComboBox.Text; map.authCookie = meshcentral.authCookie; map.certhash = meshcentral.wshash; map.parent = this; map.Dock = DockStyle.Top; map.Start(); mapPanel.Controls.Add(map); noMapLabel.Visible = false; processedArgs.Add(i); } } else if (arg.Length > 10 && arg.Substring(0, 10).ToLower() == "-relaymap:") { string[] x = arg.Substring(10).Split(':'); if (x.Length == 6) { // Protocol int protocol = 0; if (x[0].ToLower() == "tcp") { protocol = 1; } if (x[0].ToLower() == "udp") { protocol = 2; } if (protocol == 0) continue; // LocalPort ushort localPort = 0; if (ushort.TryParse(x[1], out localPort) == false) continue; // Node string nodename = x[2]; NodeClass node = null; foreach (NodeClass n in meshcentral.nodes.Values) { if (((n.conn & 1) != 0) && (n.name.ToLower() == nodename.ToLower())) { node = n; } } if (node == null) continue; // AppId int appId = 0; if (protocol == 1) { if (x[3].ToLower() == "http") { appId = 1; } else if (x[3].ToLower() == "https") { appId = 2; } else if (x[3].ToLower() == "rdp") { appId = 3; } else if (x[3].ToLower() == "putty") { appId = 4; } else if (x[3].ToLower() == "winscp") { appId = 5; } } // Remote host IPAddress remoteIp; if (IPAddress.TryParse(x[4], out remoteIp) == false) continue; // RemotePort ushort remotePort = 0; if (ushort.TryParse(x[5], out remotePort) == false) continue; // Add a new port map MapUserControl map = new MapUserControl(); map.xdebug = debug; map.inaddrany = inaddrany; map.protocol = protocol; map.localPort = (int)localPort; map.remoteIP = remoteIp.ToString(); map.remotePort = (int)remotePort; map.appId = appId; map.node = node; map.host = serverNameComboBox.Text; map.authCookie = meshcentral.authCookie; map.certhash = meshcentral.wshash; map.parent = this; map.Dock = DockStyle.Top; map.Start(); mapPanel.Controls.Add(map); noMapLabel.Visible = false; processedArgs.Add(i); } } } } private void addButton_Click(object sender, EventArgs e) { AddPortMapForm form = new AddPortMapForm(meshcentral); if (form.ShowDialog(this) == DialogResult.OK) { // Add a new port map MapUserControl map = new MapUserControl(); map.xdebug = debug; map.inaddrany = inaddrany; map.protocol = form.getProtocol(); map.localPort = form.getLocalPort(); map.remotePort = form.getRemotePort(); map.appId = form.getAppId(); map.node = form.getNode(); map.host = serverNameComboBox.Text; map.authCookie = meshcentral.authCookie; map.certhash = meshcentral.wshash; map.parent = this; map.Dock = DockStyle.Top; map.Start(); mapPanel.Controls.Add(map); noMapLabel.Visible = false; } } public void removeMap(MapUserControl map) { mapPanel.Controls.Remove(map); noMapLabel.Visible = (mapPanel.Controls.Count <= 1); } private void backButton2_Click(object sender, EventArgs e) { setPanel(1); } private void nextButton2_Click(object sender, EventArgs e) { // Attempt to login with token addButton.Enabled = false; addRelayButton.Enabled = false; openWebSiteButton.Visible = false; Uri serverurl = new Uri("wss://" + serverNameComboBox.Text + "/control.ashx"); meshcentral = new MeshCentralServer(); meshcentral.ignoreCert = ignoreCert; if (lastBadConnectCert != null) { meshcentral.okCertHash = lastBadConnectCert.GetCertHashString(); } meshcentral.onStateChanged += Meshcentral_onStateChanged; meshcentral.onNodesChanged += Meshcentral_onNodesChanged; meshcentral.onLoginTokenChanged += Meshcentral_onLoginTokenChanged; meshcentral.connect(serverurl, userNameTextBox.Text, passwordTextBox.Text, tokenTextBox.Text.Replace(" ", "")); } private void tokenTextBox_KeyPress(object sender, KeyPressEventArgs e) { if (e.KeyChar == 13) { nextButton2_Click(this, null); e.Handled = true; } } private void serverNameComboBox_KeyPress(object sender, KeyPressEventArgs e) { if (e.KeyChar == 13) { userNameTextBox.Focus(); e.Handled = true; } } private void userNameTextBox_KeyPress(object sender, KeyPressEventArgs e) { if (e.KeyChar == 13) { passwordTextBox.Focus(); e.Handled = true; } } private void passwordTextBox_KeyPress(object sender, KeyPressEventArgs e) { if (e.KeyChar == 13) { e.Handled = true; if (nextButton1.Enabled) { nextButton1_Click(this, null); } } } private void stateClearTimer_Tick(object sender, EventArgs e) { stateLabel.Visible = false; stateClearTimer.Enabled = false; } private void licenseLinkLabel_LinkClicked(object sender, LinkLabelLinkClickedEventArgs e) { System.Diagnostics.Process.Start("https://www.apache.org/licenses/LICENSE-2.0.html"); } private void backButton3_Click(object sender, EventArgs e) { setPanel(1); } private void certDetailsButton_Click(object sender, EventArgs e) { X509Certificate2UI.DisplayCertificate(lastBadConnectCert); } private void addRelayMapButton_Click(object sender, EventArgs e) { AddRelayMapForm form = new AddRelayMapForm(meshcentral); if (form.ShowDialog(this) == DialogResult.OK) { // Add a new port map MapUserControl map = new MapUserControl(); map.xdebug = debug; map.inaddrany = inaddrany; map.protocol = form.getProtocol(); map.localPort = form.getLocalPort(); map.remotePort = form.getRemotePort(); map.remoteIP = form.getRemoteIP(); map.appId = form.getAppId(); map.node = form.getNode(); map.host = serverNameComboBox.Text; map.authCookie = meshcentral.authCookie; map.certhash = meshcentral.wshash; map.parent = this; map.Dock = DockStyle.Top; map.Start(); mapPanel.Controls.Add(map); noMapLabel.Visible = false; } } private void helpPictureBox_Click(object sender, EventArgs e) { new MappingHelpForm().ShowDialog(this); } private void openWebSiteButton_Click(object sender, EventArgs e) { if (meshcentral.loginCookie != null) { Uri serverurl = new Uri("https://" + serverNameComboBox.Text + "?login=" + meshcentral.loginCookie); System.Diagnostics.Process.Start(serverurl.ToString()); } } private void cookieRefreshTimer_Tick(object sender, EventArgs e) { meshcentral.refreshCookies(); } private void notifyIcon_DoubleClick(object sender, EventArgs e) { if (this.Visible == false) { this.Visible = true; } else { this.Visible = false; this.Focus(); } } private void exitToolStripMenuItem_Click_1(object sender, EventArgs e) { forceExit = true; Application.Exit(); } private void openToolStripMenuItem_Click(object sender, EventArgs e) { this.WindowState = FormWindowState.Normal; this.Visible = true; this.Focus(); } private void settingsPictureBox_Click(object sender, EventArgs e) { SettingsForm f = new SettingsForm(); f.BindAllInterfaces = inaddrany; f.ShowSystemTray = (notifyIcon.Visible == true); if (f.ShowDialog(this) == DialogResult.OK) { inaddrany = f.BindAllInterfaces; if (f.ShowSystemTray) { notifyIcon.Visible = true; this.ShowInTaskbar = false; this.MinimizeBox = false; } else { notifyIcon.Visible = false; this.ShowInTaskbar = true; this.MinimizeBox = true; } } } /* private delegate void displayMessageHandler(string msg, int buttons, string extra, int progress); private void displayMessage(string msg, int buttons = 0, string extra = "", int progress = 0) { if (this.InvokeRequired) { this.Invoke(new displayMessageHandler(displayMessage), msg, buttons, extra, progress); return; } if (msg != null) { statusLabel.Text = msg; loadingLabel.Text = msg; } statusLabel2.Text = extra; label4.Text = extra; nextButton3.Enabled = (buttons == 1); backButton3.Enabled = (buttons == 2); mainProgressBar.Visible = (progress > 0); if (progress >= 0) { mainProgressBar.Value = progress; } if (buttons == 3) { setPanel(6); } linkLabel1.Visible = (progress == -1); advConfigButton.Visible = (progress == -1); } */ } }