private void ExecuteThread(object obj)
{
try
{
// Data
CommandState state = (CommandState)obj;
// Request
Entity entity = state.Target;
TunnelComponent tunnel = entity.Get <TunnelComponent>();
SearchRequestCommand cmdRequest = new SearchRequestCommand(entity, tunnel.Connection);
string jsonData = cmdRequest.Execute(state.Data);
if (string.IsNullOrEmpty(jsonData))
{
return;
}
// Response
entity = state.Source;
tunnel = entity.Get <TunnelComponent>();
SearchResponseCommand cmdResponse = new SearchResponseCommand(entity, tunnel.Connection);
HttpCode code = cmdResponse.Execute(state.Id, jsonData);
if (code == HttpCode.Ok)
{
SearchListComponent download = entity.Get <SearchListComponent>();
download.Add(state.Target);
}
}
catch (Exception ex)
{
Console.WriteLine(ex.Message);
}
}
public void Add(ClientComponent client, ServerComponent server, GroupComponent group)
{
Entity entity = new Entity();
entity.Add(client);
ClientOptionsComponent options = new ClientOptionsComponent();
entity.Add(options);
entity.Add(server);
entity.Add(group);
FolderListComponent folderList = new FolderListComponent();
entity.Add(folderList);
FileMapComponent fileMap = new FileMapComponent();
entity.Add(fileMap);
FolderMapComponent folderMap = new FolderMapComponent();
entity.Add(folderMap);
ClientMachineComponent machine = new ClientMachineComponent();
entity.Add <MachineComponent>(machine);
SessionComponent session = new SessionComponent();
entity.Add(session);
FileComponent file = new FileComponent();
entity.Add(file);
SearchListComponent search = new SearchListComponent(SearchController.Dispatcher);
entity.Add(search);
BrowseComponent browse = new BrowseComponent();
entity.Add(browse);
DownloadListComponent download = new DownloadListComponent(DownloadController.Dispatcher);
entity.Add(download);
UploadListComponent upload = new UploadListComponent(UploadController.Dispatcher);
entity.Add(upload);
#if DEBUG
LogComponent log = new LogComponent(LogController);
entity.Add(log);
#endif
Add(entity);
}
public void Select(Entity entity)
{
SearchListComponent search = null;
if (entity != null)
{
search = entity.Get <SearchListComponent>();
}
Select(search);
}
public override void Clear()
{
base.Clear();
if (MenuController.Entity != null)
{
Entity entity = MenuController.Entity;
SearchListComponent searchList = entity.Get <SearchListComponent>();;
searchList.Id = null;
}
Select(MenuController.Entity);
}
public void Select(SearchListComponent search)
{
List = (search != null) ? search.List : EmptyList;
View.ItemsSource = Collection;
}
public JsonAction Execute(HttpRequest httpRequest, JsonPacket jsonRequest, SessionComponent session)
{
Clear();
// Connect
NetworkChannel channel = new NetworkChannel(Connection);
// Request
JsonAction jsonAction = JsonAction.None;
JsonDownloadRequestMessage jsonRequestMessage = JsonDownloadRequestMessage.Parse(jsonRequest.Message);
string jsonId = jsonRequestMessage.Id;
if (jsonId != null)
{
// Data
Entity entity = TransferMap.Get(jsonId);
if (entity == null)
{
channel.SendNotFound();
return(jsonAction);
}
ChunkComponent transfer = entity.Get <ChunkComponent>();
if (transfer == null)
{
channel.SendNotFound();
return(jsonAction);
}
JsonTransfer jsonTransfer = transfer.PopData();
if (!transfer.Finished)
{
jsonAction = JsonAction.Request;
entity.Update();
}
else
{
TransferMap.Remove(jsonId);
}
string jsonData = null;
JsonChunk jsonChunk = null;
if (jsonTransfer != null)
{
jsonData = jsonTransfer.Data;
jsonChunk = jsonTransfer.Chunk;
}
// Response
JsonDownloadResponseMessage jsonResponseMessage = new JsonDownloadResponseMessage(jsonId)
{
Chunk = jsonChunk
};
JsonPacket jsonResponse = new JsonPacket(jsonResponseMessage, jsonData);
HttpResponse httpResponse = new HttpResponse()
{
Data = session.Encrypt(jsonResponse)
};
channel.Send(httpResponse);
#if DEBUG
jsonResponse.Data = null;
Log.Add(httpRequest, httpResponse, jsonRequest, jsonResponse);
#endif
}
else
{
// Data
Entity entity = session.Owner;
SearchListComponent search = entity.Get <SearchListComponent>();
JsonClient jsonClient = jsonRequestMessage.Client;
// Request
if (jsonClient == null && search.Empty)
{
channel.SendNotFound();
return(jsonAction);
}
// Response
jsonId = SecurityUtil.CreateKeyString();
JsonDownloadResponseMessage jsonResponseMessage = new JsonDownloadResponseMessage(jsonId);
JsonPacket jsonResponse = new JsonPacket(jsonResponseMessage);
HttpResponse httpResponse = new HttpResponse()
{
Data = session.Encrypt(jsonResponse)
};
channel.Send(httpResponse);
#if DEBUG
Log.Add(httpRequest, httpResponse, jsonRequest, jsonResponse);
#endif
// Data
entity = new Entity(jsonId);
EntityIdleComponent idle = new EntityIdleComponent();
entity.Add(idle);
JsonChunk jsonChunk = jsonRequestMessage.Chunk;
ChunkComponent transfer = new ChunkComponent(jsonChunk.Size, Options.ChunkSize, Options.MaxChunks);
entity.Add(transfer);
TransferMap.Add(entity);
// Command
string jsonData = jsonRequest.Data;
jsonChunk = transfer.PopChunk();
if (jsonClient == null)
{
foreach (Entity e in search)
{
CommandState state = new CommandState()
{
Id = jsonId, Data = jsonData, Chunk = jsonChunk, Entity = e
};
Thread thread = new Thread(new ParameterizedThreadStart(ExecuteThread))
{
Priority = ThreadPriority.BelowNormal, IsBackground = true
};
thread.Start(state);
}
}
else
{
Entity e = ClientMap.Get(jsonClient.Id);
if (e == null)
{
channel.SendNotFound();
return(jsonAction);
}
CommandState state = new CommandState()
{
Id = jsonId, Data = jsonData, Chunk = jsonChunk, Entity = e
};
Thread thread = new Thread(new ParameterizedThreadStart(ExecuteThread))
{
Priority = ThreadPriority.BelowNormal, IsBackground = true
};
thread.Start(state);
}
jsonAction = JsonAction.Request;
}
return(jsonAction);
}
private void ReadClient(XmlReader reader)
{
if (reader.IsEmptyElement)
{
return;
}
bool loop = true;
string name = null;
// Components
Entity entity = new Entity();
ClientComponent client = new ClientComponent();
ClientOptionsComponent options = new ClientOptionsComponent();
GroupComponent group = new GroupComponent();
ServerComponent server = new ServerComponent();
List <string> paths = new List <string>();
client.Enabled = bool.Parse(reader.GetAttribute(XmlTag.Enabled));
while (loop && reader.Read())
{
switch (reader.NodeType)
{
case XmlNodeType.Element:
{
name = reader.Name;
switch (name)
{
case XmlTag.Router:
{
ReadRouter(reader, ref server);
break;
}
case XmlTag.Options:
{
ReadOptions(reader, ref options);
break;
}
case XmlTag.Folders:
{
ReadFolders(reader, ref paths);
break;
}
}
break;
}
case XmlNodeType.Text:
{
switch (name)
{
case XmlTag.Id:
{
entity.Id = reader.Value;
break;
}
case XmlTag.Name:
{
client.Name = reader.Value;
break;
}
case XmlTag.Download:
{
client.Download = reader.Value;
break;
}
case XmlTag.Group:
{
group.Path = reader.Value;
break;
}
}
break;
}
case XmlNodeType.EndElement:
{
if (XmlTag.Client.Equals(reader.Name))
{
loop = false;
}
break;
}
}
}
// Components
entity.Add(client);
entity.Add(options);
entity.Add(group);
entity.Add(server);
FolderListComponent folderList = new FolderListComponent();
entity.Add(folderList);
FileMapComponent fileMap = new FileMapComponent();
entity.Add(fileMap);
FolderMapComponent folderMap = new FolderMapComponent();
entity.Add(folderMap);
ClientMachineComponent machine = new ClientMachineComponent();
entity.Add <MachineComponent>(machine);
SessionComponent session = new SessionComponent();
entity.Add(session);
FileComponent file = new FileComponent();
entity.Add(file);
SearchListComponent search = new SearchListComponent(SearchController.Dispatcher);
entity.Add(search);
BrowseComponent browse = new BrowseComponent();
entity.Add(browse);
DownloadListComponent download = new DownloadListComponent(DownloadController.Dispatcher);
entity.Add(download);
UploadListComponent upload = new UploadListComponent(UploadController.Dispatcher);
entity.Add(upload);
#if DEBUG
LogComponent log = new LogComponent(LogController);
entity.Add(log);
#endif
ClientController.Add(entity);
foreach (string path in paths)
{
FileComponent shared = new FileComponent(path)
{
Owner = entity
};
folderList.Add(shared);
}
}
public void Execute(HttpRequest httpRequest, JsonPacket jsonRequest, SessionComponent session)
{
// Connect
NetworkChannel channel = new NetworkChannel(Connection);
// Request
JsonJoinRequestMessage jsonRequestMessage = JsonJoinRequestMessage.Parse(jsonRequest.Message);
// Data
JsonEntity jsonEntity = jsonRequestMessage.Entity;
Entity entity = EntityMap.Get(jsonEntity.Id);
if (entity != null)
{
entity.Shutdown();
}
entity = session.Owner;
entity.Id = jsonEntity.Id;
EntityMap.Add(entity.Id, entity);
Server.Entities = EntityMap.Count;
JsonGroup jsonGroup = jsonRequestMessage.Group;
if (jsonGroup == null)
{
jsonGroup = JsonGroup.Default;
}
JsonClient jsonClient = jsonRequestMessage.Client;
ClientComponent client = new ClientComponent(jsonClient.Id, jsonClient.Name);
entity.Add(client);
ClientMap.Add(client.Id, entity);
Server.Clients = ClientMap.Count;
GroupComponent group = new GroupComponent(jsonGroup.Id);
entity.Add(group);
GroupList.Add(group.Id, entity);
Server.Groups = GroupList.Count;
SearchListComponent download = new SearchListComponent();
entity.Add(download);
// Response
JsonJoinResponseMessage jsonResponseMessage = new JsonJoinResponseMessage()
{
Entities = Server.Entities, Sessions = Server.Sessions, Clients = Server.Clients, Groups = Server.Groups
};
JsonPacket jsonResponse = new JsonPacket(jsonResponseMessage);
HttpResponse httpResponse = new HttpResponse()
{
Data = session.Encrypt(jsonResponse)
};
channel.Send(httpResponse);
#if DEBUG
Log.Add(httpRequest, httpResponse, jsonRequest, jsonResponse);
#endif
}
public void Execute(HttpRequest httpRequest, JsonPacket jsonRequest, SessionComponent session)
{
Clear();
// Connect
NetworkChannel channel = new NetworkChannel(Connection);
// Request
JsonSearchRequestMessage jsonRequestMessage = JsonSearchRequestMessage.Parse(jsonRequest.Message);
// Data
Entity entity = session.Owner;
GroupComponent group = entity.Get <GroupComponent>();
EntityList entities = GroupList.Get(group.Id);
if (entities == null)
{
channel.SendNotFound();
return;
}
// Response
string jsonId = SecurityUtil.CreateKeyString();
JsonSearchResponseMessage jsonResponseMessage = new JsonSearchResponseMessage(jsonId);
JsonPacket jsonResponse = new JsonPacket(jsonResponseMessage);
HttpResponse httpResponse = new HttpResponse()
{
Data = session.Encrypt(jsonResponse)
};
channel.Send(httpResponse);
// Optimization: Keep track of the search list for faster downloads
SearchListComponent download = entity.Get <SearchListComponent>();
download.Clear();
// Command
lock (entities)
{
foreach (Entity e in entities)
{
// Do not search yourself
if (entity.Id.Equals(e.Id))
{
continue;
}
TunnelComponent tunnel = e.Get <TunnelComponent>();
if (!tunnel.Connected)
{
continue;
}
CommandState state = new CommandState()
{
Id = jsonId,
Data = jsonRequest.Data,
Source = entity,
Target = e
};
Thread thread = new Thread(new ParameterizedThreadStart(ExecuteThread))
{
Priority = ThreadPriority.BelowNormal, IsBackground = true
};
thread.Start(state);
}
}
#if DEBUG
Log.Add(httpRequest, httpResponse, jsonRequest, jsonResponse);
#endif
}
