private void SendCommand_Execute(object o)
{
SendView sendView = new SendView() { Owner = mainView };
SendViewModel sendViewModel = new SendViewModel(sendView);
sendView.DataContext = sendViewModel;
if (sendView.ShowDialog() == true)
{
Thread createFSThread = new Thread(() =>
{
xmlfs = new XMLFileSystem();
server.Action = Resource.CreateFileSystem;
foreach (FSItem fsItem in sendViewModel.Items)
{
if (fsItem.Type == FSItemType.Folder)
xmlfs.AddFolder(fsItem.Path);
else
xmlfs.AddFile(fsItem.Path);
}
xmlfs.Save("fs.xml", true);
App.Current.Dispatcher.Invoke(new Action(() =>
{
server.Send(new System.Net.IPEndPoint(System.Net.IPAddress.Parse(sendViewModel.Address), int.Parse(sendViewModel.Port)), xmlfs);
}));
});
createFSThread.Start();
}
}
private void receive_DoWork(object o, DoWorkEventArgs args)
{
IsBusy = true;
var buf = new byte[BUFFER_SIZE];
try
{
var folder = (string)args.Argument;
Action = Resource.BeginReceiving;
socket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
socket.Connect(this.endPoint);
SRFMessage message = new SRFMessage();
socket.Receive(message.Message);
if (message.Command != Commands.Hello)
throw new Exception(Resource.ERROR05);
message = new SRFMessage { Command = Commands.OK };
socket.Send(message.Message);
message = new SRFMessage { Command = Commands.FileSystem };
socket.Send(message.Message);
socket.Receive(message.Message);
if (message.Command != Commands.FileSystem)
throw new Exception(Resource.ERROR06);
CurrentFullSize = message.CurrentFullSize;
CurrentSize = 0;
fs = new FileStream("fs.xml.gz", FileMode.Create, FileAccess.ReadWrite);
bs = new BufferedStream(fs);
timer.Start();
int count;
while (currentSize < currentFullSize)
{
count = socket.Receive(buf, 0, buf.Length, 0);
bs.Write(buf, 0, count);
bs.Flush();
CurrentSize += count;
TotalProgress = CurrentProgress = Math.Round(currentSize * 100d / currentFullSize, 2);
}
timer.Stop();
old = 0;
Speed = 0;
fs.Close();
bs.Close();
message = new SRFMessage { Command = Commands.OK };
socket.Send(message.Message);
xmlfs = new XMLFileSystem("fs.xml");
FilesCount = xmlfs.FilesCount;
FullSize = xmlfs.FullSize;
xmlfs.PrepareFolders(folder);
foreach (XElement item in xmlfs.Files.Elements("file"))
{
message = new SRFMessage { Command = Commands.File, ID = UInt32.Parse(item.Attribute("id").Value) };
socket.Send(message.Message);
socket.Receive(message.Message);
if (message.Command != Commands.OK)
throw new Exception(Resource.ERROR03);
count = 0;
FileName = item.Attribute("name").Value;
CurrentSize = 0;
CurrentFullSize = long.Parse(item.Attribute("size").Value);
CurrentProgress = 0;
CurrentFileCount++;
fs = new FileStream(folder + "\\" + item.Attribute("path").Value, FileMode.CreateNew, FileAccess.Write);
bs = new BufferedStream(fs);
timer.Start();
while (currentSize < currentFullSize)
{
count = socket.Receive(buf, 0, buf.Length, 0);
bs.Write(buf, 0, count);
bs.Flush();
CurrentSize += count;
Size += count;
CurrentProgress = Math.Round(currentSize * 100d / currentFullSize, 2);
TotalProgress = Math.Round(size * 100d / fullSize, 2);
}
timer.Stop();
old = 0;
Speed = 0;
bs.Close();
fs.Close();
message = new SRFMessage() { Command = Commands.OK };
socket.Send(message.Message);
}
message = new SRFMessage() { Command = Commands.End };
socket.Send(message.Message);
Action = Resource.EndReceiving;
}
catch (IOException ioe)
{
Action = Resource.Error;
OnError(ioe);
}
catch (SocketException se)
{
Action = Resource.Error;
OnError(se);
}
finally
{
if (timer.Enabled)
timer.Stop();
if (bs != null)
{
bs.Close();
bs = null;
}
if (fs != null)
{
fs.Close();
fs = null;
}
if (socket != null)
{
if (socket.Connected)
socket.Disconnect(false);
socket.Close();
socket = null;
}
Clear();
IsBusy = false;
}
}
private void send_DoWork(object o, DoWorkEventArgs args)
{
IsBusy = true;
Socket handler = null;
SRFMessage message = null;
xmlfs = null;
byte[] buf = new byte[BUFFER_SIZE];
int count = 0;
try
{
xmlfs = (XMLFileSystem)args.Argument;
FilesCount = xmlfs.FilesCount;
FullSize = xmlfs.FullSize;
Action = Resource.BeginSending;
socket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
socket.Bind(endPoint);
socket.Listen(0);
Action = Resource.WaitConnection;
handler = socket.Accept();
message = new SRFMessage() { Command = Commands.Hello };
handler.Send(message.Message);
handler.Receive(message.Message);
if (message.Command != Commands.OK)
throw new Exception(Resource.ERROR01);
while (message.Command != Commands.End)
{
handler.Receive(message.Message);
switch (message.Command)
{
case Commands.FileSystem:
string fsPath = "fs.xml.out.gz";
FileName = "File System";
CurrentSize = 0;
CurrentFullSize = message.CurrentFullSize = new FileInfo(fsPath).Length;
handler.Send(message.Message);
fs = new FileStream(fsPath, FileMode.Open, FileAccess.Read);
bs = new BufferedStream(fs);
timer.Start();
while ((count = bs.Read(buf, 0, buf.Length)) > 0)
{
handler.Send(buf, 0, count, 0);
CurrentSize += count;
TotalProgress = CurrentProgress = Math.Round(currentSize * 100d / currentFullSize, 2);
}
timer.Stop();
old = 0;
Speed = 0;
bs.Close();
fs.Close();
handler.Receive(message.Message);
if (message.Command != Commands.OK)
throw new Exception(Resource.ERROR02);
break;
case Commands.File:
XElement el = xmlfs.SearchFile(message.ID);
if (el == null)
{
message = new SRFMessage() { Command = Commands.Error };
handler.Send(message.Message);
throw new Exception(Resource.ERROR03);
}
message = new SRFMessage() { Command = Commands.OK };
handler.Send(message.Message);
count = 0;
currentSize = 0;
FileName = el.Attribute("name").Value;
CurrentFullSize = long.Parse(el.Attribute("size").Value);
CurrentProgress = 0;
CurrentFileCount++;
fs = new FileStream(el.Attribute("cut").Value + el.Attribute("path").Value, FileMode.Open, FileAccess.Read);
bs = new BufferedStream(fs);
timer.Start();
while ((count = bs.Read(buf, 0, buf.Length)) > 0)
{
handler.Send(buf, 0, count, 0);
CurrentSize += count;
Size += count;
CurrentProgress = Math.Round(currentSize * 100d / currentFullSize, 2);
TotalProgress = Math.Round(size * 100d / fullSize, 2);
}
timer.Stop();
old = 0;
Speed = 0;
bs.Close();
fs.Close();
handler.Receive(message.Message);
if (message.Command != Commands.OK)
throw new Exception(Resource.ERROR04);
break;
case Commands.End:
return;
default:
break;
}
Action = Resource.EndSending;
}
}
catch (IOException ioe)
{
Action = Resource.Error;
OnError(ioe);
}
catch (SocketException se)
{
Action = Resource.Error;
OnError(se);
}
finally
{
message = null;
xmlfs = null;
if (timer.Enabled)
timer.Stop();
if (bs != null)
{
bs.Close();
bs = null;
}
if (fs != null)
{
fs.Close();
fs = null;
}
if (handler != null)
{
if (handler.Connected)
handler.Disconnect(false);
handler.Close();
handler = null;
}
if (socket != null)
{
if (socket.Connected)
socket.Disconnect(false);
socket.Close();
socket = null;
}
Clear();
IsBusy = false;
}
}
public void Send(IPEndPoint endPoint, XMLFileSystem xmlfs)
{
this.endPoint = endPoint;
bw = new BackgroundWorker() { WorkerReportsProgress = true, WorkerSupportsCancellation = true };
bw.DoWork += send_DoWork;
bw.RunWorkerAsync(xmlfs);
}