Horizon – Diff between revs 11 and 12

Subversion Repositories:

?pathlinks?

Rev 11 – Rev 20 – Go to most recent revision – Details – Blame – Last modification – View Log

 ï»¿using System;
 using System.Collections.Generic;
 using System.Collections.Specialized;
 using System.ComponentModel;
 using System.Data.SQLite;
+using System.Diagnostics;
 using System.Drawing;
 using System.IO;
 using System.Linq;
 using System.Net;
 using System.Reflection;
+using System.Text;
 using System.Threading;
 using System.Threading.Tasks;
 using System.Threading.Tasks.Dataflow;
 using System.Windows.Forms;
 using Horizon.Database;
 using Horizon.Snapshots;
 using Horizon.Utilities;
 using Horizon.Utilities.Serialization;
+using Mono.Zeroconf;
 using NetSparkleUpdater;
 using NetSparkleUpdater.Enums;
 using NetSparkleUpdater.SignatureVerifiers;
 using NetSparkleUpdater.UI.WinForms;
+using Newtonsoft.Json;
 using Serilog;
 using TrackedFolders;
+using WatsonTcp;
+using static Horizon.Utilities.Networking.Miscellaneous;
 using CaptureMode = Configuration.CaptureMode;
 using Path = System.IO.Path;
 namespace Horizon
         private LogViewForm _logViewForm;
+        private readonly LogMemorySink _memorySink;
+        private RegisterService _horizonDiscoveryService;
-        private readonly LogMemorySink _memorySink;
+        private WatsonTcpServer _horizonNetworkShare;
         #endregion
+        #region Event Handlers
+        private void networkSharingToolStripMenuItem_CheckStateChanged(object sender, EventArgs e)
+        {
+            var toolStripMenuItem = (ToolStripMenuItem)sender;
+            switch (toolStripMenuItem.CheckState)
+            {
+                case CheckState.Checked:
+                    var freePort = GetAvailableTcpPort();
+                    _horizonNetworkShare = new WatsonTcpServer("0.0.0.0", freePort);
+                    _horizonNetworkShare.Events.ClientConnected += Events_ClientConnected;
+                    _horizonNetworkShare.Events.ClientDisconnected += Events_ClientDisconnected;
+                    _horizonNetworkShare.Events.MessageReceived += Events_MessageReceived;
+#pragma warning disable CS4014
+                    _horizonNetworkShare.Start();
+#pragma warning restore CS4014
+                    try
+                    {
+                        _horizonDiscoveryService = new RegisterService();
+                        _horizonDiscoveryService.Name = $"Horizon ({Environment.MachineName})";
+                        _horizonDiscoveryService.RegType = "_horizon._tcp";
+                        _horizonDiscoveryService.ReplyDomain = "local.";
+                        _horizonDiscoveryService.UPort = freePort;
+                        _horizonDiscoveryService.Register();
+                    }
+                    catch (Exception exception)
+                    {
+                        Log.Error(exception, "Service discovery protocol could not be stared.");
+                    }
+                    Configuration.NetworkSharing = true;
+                    break;
+                case CheckState.Unchecked:
+                    if (_horizonNetworkShare != null)
+                    {
+                        _horizonNetworkShare.Dispose();
+                        _horizonNetworkShare = null;
+                    }
+                    if (_horizonDiscoveryService != null)
+                    {
+                        _horizonDiscoveryService.Dispose();
+                        _horizonDiscoveryService = null;
+                    }
+                    Configuration.NetworkSharing = false;
+                    break;
+            }
+            ChangedConfigurationContinuation.Schedule(TimeSpan.FromSeconds(1),
+                async () => { await SaveConfiguration(); }, _cancellationToken);
+        }
+        private async void Events_MessageReceived(object sender, MessageReceivedEventArgs e)
+        {
+            Log.Information($"Client {e.Client.IpPort} sent {e.Data.Length} bytes via network sharing.");
+            if (e.Data.Length == 0)
+            {
+                return;
+            }
+            try
+            {
+                var payload = Encoding.UTF8.GetString(e.Data);
+                var completeSnapshot = JsonConvert.DeserializeObject<TransferSnapshot>(payload);
+                await _snapshotDatabase.SetCompleteSnapshotAsync(completeSnapshot, _cancellationToken);
+                Log.Information($"Stored {completeSnapshot.Name} from {e.Client.IpPort}");
+            }
+            catch (Exception exception)
+            {
+                Log.Error(exception, $"Failed to process network share from {e.Client.IpPort}.");
+            }
+        }
+        private void Events_ClientDisconnected(object sender, WatsonTcp.DisconnectionEventArgs e)
+        {
+            Log.Information($"Client {e.Client.IpPort} disconnected from network sharing.");
+        }
+        private void Events_ClientConnected(object sender, WatsonTcp.ConnectionEventArgs e)
+        {
             Log.Information($"Client {e.Client.IpPort} connected to network sharing.");
         #region Event Handlers
         private void WindowToolStripMenuItem_Click(object sender, EventArgs e)
             atomicOperationsToolStripMenuItem.Checked = Configuration.AtomicOperations;
             enableToolStripMenuItem.Checked = Configuration.Enabled;
             showBalloonTooltipsToolStripMenuItem.Checked = Configuration.Enabled;
             windowToolStripMenuItem.Checked = Configuration.CaptureMode == CaptureMode.Window;
             screenToolStripMenuItem.Checked = Configuration.CaptureMode == CaptureMode.Screen;
+            networkSharingToolStripMenuItem.Checked = Configuration.NetworkSharing;
             // Load all tracked folders.
             var folders = await LoadFolders();