private async Task ProcessGetSessionCmd(CommandHeader header) { var response = new ServerResponseWithData <Guid>(); try { var req = await NetworkHelper.Read <GetSessionRequest>(_networkStream, header.PayloadLength); var client = _config.Clients.Single(x => x.Id == req.ClientId); var folder = client.FolderEndpoints.Single(x => x.Id == req.EndpointId); var session = SessionStorage.Instance.GetNewSession(); session.BaseDir = folder.LocalPath; session.SyncDbDir = Path.Combine(folder.LocalPath, ".sync"); if (!Directory.Exists(session.SyncDbDir)) { var dirInfo = Directory.CreateDirectory(session.SyncDbDir); dirInfo.Attributes = dirInfo.Attributes | FileAttributes.Hidden; } session.RemovedDir = Path.Combine(session.SyncDbDir, "rem"); PathHelpers.EnsureDirExists(session.RemovedDir); session.NewDir = Path.Combine(session.SyncDbDir, "new"); PathHelpers.EnsureDirExists(session.NewDir); var helper = new SessionFileHelper(session.NewDir, session.RemovedDir, session.BaseDir, new StringBuilder()); session.FileHelper = helper; response.Data = session.Id; } catch (Exception e) { response.ErrorMsg = e.ToString(); } await CommandHelper.WriteCommandResponse(_networkStream, Commands.GetSessionCmd, response); }
public async Task <ServerResponseWithData <T> > Send(Stream stream) { return(await NetworkHelper.SendReceive(stream, this)); }
private async Task ProcessCommands() { try { var cmdHeader = await NetworkHelper.ReadCommandHeader(_networkStream); if (!ByteToCmd.TryGetValue(cmdHeader.Command, out var cmdName)) { cmdName = "Unknown command"; } var msg = $"Received {cmdName}"; if (cmdHeader.PayloadLength > 0) { msg += $" with {cmdHeader.PayloadLength} bytes of payload"; } Msg?.Invoke(msg); switch (cmdHeader.Command) { case Commands.GetSessionCmd: await ProcessGetSessionCmd(cmdHeader); break; case Commands.GetSyncListCmd: await ReturnSyncList(cmdHeader); break; case Commands.GetFileCmd: await SendFileToClient(cmdHeader); break; case Commands.SendFileCmd: await ReceiveFileFromClient(cmdHeader); break; case Commands.FinishSessionCmd: await FinishSession(cmdHeader); break; case Commands.DisconnectCmd: _connected = false; break; case Commands.GetIdCmd: await ProcessGetIdCmd(); break; case Commands.RegisterClientCmd: await ProcessRegisterClientCmd(cmdHeader); break; case Commands.GetClientEndpointsCmd: await ProcessGetClientEndpointsCmd(cmdHeader); break; default: _connected = false; break; } } catch (Exception e) { if (_connected) { Debugger.Break(); _connected = false; Console.WriteLine($"Unexpected error:\r\n{e}"); } else { // client disconnected Console.WriteLine("Unexpected error but client already disconnected, ignoring..."); } } }
private async Task ReturnSyncList(CommandHeader cmdHeader) { var remoteData = await NetworkHelper.Read <GetSyncListRequest>(_networkStream, cmdHeader.PayloadLength); var ret = new ServerResponseWithData <SyncInfo>(); var session = SessionStorage.Instance.GetSession(remoteData.SessionId); if (session == null) { ret.ErrorMsg = "Session does not exist"; } else if (session.Expired) { ret.ErrorMsg = "Session has expired"; _log.AppendLine("Session has expired"); //return ret; } else { try { Msg?.Invoke("Scanning local folder..."); var syncDb = GetSyncDb(session.BaseDir, session.SyncDbDir, out var error); if (syncDb == null) { ret.ErrorMsg = error; Msg?.Invoke($"Failed to get sync db: {error}"); } else { var syncInfo = new SyncInfo(); PathHelpers.NormalizeRelative(remoteData.Files); Msg?.Invoke("Preparing sync list..."); foreach (var localFileInfo in syncDb.Files) { var remoteFileInfo = remoteData.Files.FirstOrDefault(remoteFile => remoteFile.RelativePath == localFileInfo.RelativePath); if (remoteFileInfo == null) { if (localFileInfo.State != SyncFileState.Deleted) { syncInfo.ToDownload.Add(localFileInfo); } } else { remoteData.Files.Remove(remoteFileInfo); switch (remoteFileInfo.State) { case SyncFileState.Deleted: if (localFileInfo.State == SyncFileState.NotChanged || localFileInfo.State == SyncFileState.Deleted) { session.FileHelper.PrepareForRemove(localFileInfo.RelativePath); } else if (localFileInfo.State == SyncFileState.New) { syncInfo.ToDownload.Add(localFileInfo); } else { var fileExt = Path.GetExtension(localFileInfo.RelativePath); var newPath = Path.GetFileNameWithoutExtension(localFileInfo.RelativePath) + "_FromServer" + fileExt; localFileInfo.NewRelativePath = newPath; session.FileHelper.AddRename(localFileInfo.RelativePath, newPath); syncInfo.ToDownload.Add(localFileInfo); } break; case SyncFileState.New: if (localFileInfo.HashStr != remoteFileInfo.HashStr) { var fileExt = Path.GetExtension(remoteFileInfo.RelativePath); var newPath = Path.GetFileNameWithoutExtension(remoteFileInfo.RelativePath) + "_FromClient" + fileExt; remoteFileInfo.NewRelativePath = newPath; syncInfo.ToUpload.Add(remoteFileInfo); fileExt = Path.GetExtension(localFileInfo.RelativePath); newPath = Path.GetFileNameWithoutExtension(localFileInfo.RelativePath) + "_FromServer" + fileExt; localFileInfo.NewRelativePath = newPath; session.FileHelper.AddRename(localFileInfo.RelativePath, newPath); syncInfo.ToDownload.Add(localFileInfo); } break; case SyncFileState.Modified: if (localFileInfo.State == SyncFileState.NotChanged) { syncInfo.ToUpload.Add(localFileInfo); } else if (localFileInfo.State == SyncFileState.Modified || localFileInfo.State == SyncFileState.New) { var fileExt = Path.GetExtension(remoteFileInfo.RelativePath); var newPath = Path.GetFileNameWithoutExtension(remoteFileInfo.RelativePath) + "_FromClient" + fileExt; remoteFileInfo.NewRelativePath = newPath; syncInfo.ToUpload.Add(remoteFileInfo); fileExt = Path.GetExtension(localFileInfo.RelativePath); newPath = Path.GetFileNameWithoutExtension(localFileInfo.RelativePath) + "_FromServer" + fileExt; localFileInfo.NewRelativePath = newPath; session.FileHelper.AddRename(localFileInfo.RelativePath, newPath); syncInfo.ToDownload.Add(localFileInfo); } else if (localFileInfo.State == SyncFileState.Deleted) { var fileExt = Path.GetExtension(remoteFileInfo.RelativePath); var newPath = Path.GetFileNameWithoutExtension(remoteFileInfo.RelativePath) + "_FromClient" + fileExt; remoteFileInfo.NewRelativePath = newPath; syncInfo.ToUpload.Add(remoteFileInfo); } break; case SyncFileState.NotChanged: if (localFileInfo.State == SyncFileState.Modified) { syncInfo.ToDownload.Add(localFileInfo); } else if (localFileInfo.State == SyncFileState.Deleted) { syncInfo.ToRemove.Add(remoteFileInfo); } else if (localFileInfo.State == SyncFileState.New) { if (localFileInfo.HashStr == remoteFileInfo.HashStr) { Msg?.Invoke("Skip identical file, client already has one"); localFileInfo.State = SyncFileState.NotChanged; } else { Debugger.Break(); // not possible throw new InvalidOperationException("Invalid server state"); } } break; } } } foreach (var remoteFileInfo in remoteData.Files.Where(x => x.State != SyncFileState.Deleted)) { syncInfo.ToUpload.Add(remoteFileInfo); } ret.Data = syncInfo; session.SyncDb = syncDb; session.LastAccessTime = DateTime.Now; } } catch (Exception e) { ret.ErrorMsg = e.Message; } } await CommandHelper.WriteCommandResponse(_networkStream, Commands.GetSyncListCmd, ret); }