public void Parse_Handles_Locked_Files(string username, int token, byte freeUploadSlots, int uploadSpeed, long queueLength)
{
var msg = new MessageBuilder()
.WriteCode(MessageCode.Peer.SearchResponse)
.WriteString(username)
.WriteInteger(token)
.WriteInteger(2) // file count
.WriteByte(0x2) // code
.WriteString("filename") // filename
.WriteLong(3) // size
.WriteString("ext") // extension
.WriteInteger(1) // attribute count
.WriteInteger((int)FileAttributeType.BitDepth) // attribute[0].type
.WriteInteger(4) // attribute[0].value
.WriteByte(0x20) // code
.WriteString("filename2") // filename
.WriteLong(30) // size
.WriteString("ext2") // extension
.WriteInteger(1) // attribute count
.WriteInteger((int)FileAttributeType.BitRate) // attribute[0].type
.WriteInteger(40) // attribute[0].value
.WriteByte(freeUploadSlots)
.WriteInteger(uploadSpeed)
.WriteLong(queueLength)
.WriteInteger(1) // locked file count
.WriteByte(0x30) // code
.WriteString("filename3") // filename
.WriteLong(40) // size
.WriteString("ext3") // extension
.WriteInteger(1) // attribute count
.WriteInteger((int)FileAttributeType.BitRate) // attribute[0].type
.WriteInteger(50) // attribute[0].value
.Compress()
.Build();
var r = SearchResponseFactory.FromByteArray(msg);
Assert.Equal(2, r.Files.Count);
var file = r.Files.ToList();
Assert.Equal(0x2, file[0].Code);
Assert.Equal("filename", file[0].Filename);
Assert.Equal(3, file[0].Size);
Assert.Equal("ext", file[0].Extension);
Assert.Equal(1, file[0].AttributeCount);
Assert.Single(file[0].Attributes);
Assert.Equal(FileAttributeType.BitDepth, file[0].Attributes.ToList()[0].Type);
Assert.Equal(4, file[0].Attributes.ToList()[0].Value);
Assert.Equal(0x20, file[1].Code);
Assert.Equal("filename2", file[1].Filename);
Assert.Equal(30, file[1].Size);
Assert.Equal("ext2", file[1].Extension);
Assert.Equal(1, file[1].AttributeCount);
Assert.Single(file[1].Attributes);
Assert.Equal(FileAttributeType.BitRate, file[1].Attributes.ToList()[0].Type);
Assert.Equal(40, file[1].Attributes.ToList()[0].Value);
var locked = r.LockedFiles.ToList();
Assert.Equal(0x30, locked[0].Code);
Assert.Equal("filename3", locked[0].Filename);
Assert.Equal(40, locked[0].Size);
Assert.Equal("ext3", locked[0].Extension);
Assert.Equal(1, locked[0].AttributeCount);
Assert.Single(locked[0].Attributes);
Assert.Equal(FileAttributeType.BitRate, locked[0].Attributes.ToList()[0].Type);
Assert.Equal(50, locked[0].Attributes.ToList()[0].Value);
}
/// <summary>
/// Handles incoming messages.
/// </summary>
/// <param name="sender">The <see cref="IMessageConnection"/> instance from which the message originated.</param>
/// <param name="message">The message.</param>
public async void HandleMessageRead(object sender, byte[] message)
{
var connection = (IMessageConnection)sender;
var code = new MessageReader <MessageCode.Peer>(message).ReadCode();
Diagnostic.Debug($"Peer message received: {code} from {connection.Username} ({connection.IPEndPoint}) (id: {connection.Id})");
try
{
switch (code)
{
case MessageCode.Peer.SearchResponse:
var searchResponse = SearchResponseFactory.FromByteArray(message);
if (SoulseekClient.Searches.TryGetValue(searchResponse.Token, out var search))
{
search.TryAddResponse(searchResponse);
}
break;
case MessageCode.Peer.BrowseResponse:
var browseWaitKey = new WaitKey(MessageCode.Peer.BrowseResponse, connection.Username);
try
{
SoulseekClient.Waiter.Complete(browseWaitKey, BrowseResponseFactory.FromByteArray(message));
}
catch (Exception ex)
{
SoulseekClient.Waiter.Throw(browseWaitKey, new MessageReadException("The peer returned an invalid browse response", ex));
throw;
}
break;
case MessageCode.Peer.InfoRequest:
UserInfo outgoingInfo;
try
{
outgoingInfo = await SoulseekClient.Options
.UserInfoResolver(connection.Username, connection.IPEndPoint).ConfigureAwait(false);
}
catch (Exception ex)
{
outgoingInfo = await new SoulseekClientOptions()
.UserInfoResolver(connection.Username, connection.IPEndPoint).ConfigureAwait(false);
Diagnostic.Warning($"Failed to resolve user info response: {ex.Message}", ex);
}
await connection.WriteAsync(outgoingInfo.ToByteArray()).ConfigureAwait(false);
break;
case MessageCode.Peer.SearchRequest:
var searchRequest = PeerSearchRequest.FromByteArray(message);
if (SoulseekClient.Options.SearchResponseResolver == default)
{
break;
}
try
{
var peerSearchResponse = await SoulseekClient.Options.SearchResponseResolver(connection.Username, searchRequest.Token, SearchQuery.FromText(searchRequest.Query)).ConfigureAwait(false);
if (peerSearchResponse != null && peerSearchResponse.FileCount + peerSearchResponse.LockedFileCount > 0)
{
await connection.WriteAsync(peerSearchResponse.ToByteArray()).ConfigureAwait(false);
}
}
catch (Exception ex)
{
Diagnostic.Warning($"Error resolving search response for query '{searchRequest.Query}' requested by {connection.Username} with token {searchRequest.Token}: {ex.Message}", ex);
}
break;
case MessageCode.Peer.BrowseRequest:
BrowseResponse browseResponse;
try
{
browseResponse = await SoulseekClient.Options.BrowseResponseResolver(connection.Username, connection.IPEndPoint).ConfigureAwait(false);
}
catch (Exception ex)
{
browseResponse = await new SoulseekClientOptions()
.BrowseResponseResolver(connection.Username, connection.IPEndPoint).ConfigureAwait(false);
Diagnostic.Warning($"Failed to resolve browse response: {ex.Message}", ex);
}
await connection.WriteAsync(browseResponse.ToByteArray()).ConfigureAwait(false);
break;
case MessageCode.Peer.FolderContentsRequest:
var folderContentsRequest = FolderContentsRequest.FromByteArray(message);
Directory outgoingFolderContents = null;
try
{
outgoingFolderContents = await SoulseekClient.Options.DirectoryContentsResolver(
connection.Username,
connection.IPEndPoint,
folderContentsRequest.Token,
folderContentsRequest.DirectoryName).ConfigureAwait(false);
}
catch (Exception ex)
{
Diagnostic.Warning($"Failed to resolve directory contents response: {ex.Message}", ex);
}
if (outgoingFolderContents != null)
{
var folderContentsResponseMessage = new FolderContentsResponse(folderContentsRequest.Token, outgoingFolderContents);
await connection.WriteAsync(folderContentsResponseMessage).ConfigureAwait(false);
}
break;
case MessageCode.Peer.FolderContentsResponse:
var folderContentsResponse = FolderContentsResponse.FromByteArray(message);
SoulseekClient.Waiter.Complete(new WaitKey(MessageCode.Peer.FolderContentsResponse, connection.Username, folderContentsResponse.Token), folderContentsResponse.Directory);
break;
case MessageCode.Peer.InfoResponse:
var incomingInfo = UserInfoResponseFactory.FromByteArray(message);
SoulseekClient.Waiter.Complete(new WaitKey(MessageCode.Peer.InfoResponse, connection.Username), incomingInfo);
break;
case MessageCode.Peer.TransferResponse:
var transferResponse = TransferResponse.FromByteArray(message);
SoulseekClient.Waiter.Complete(new WaitKey(MessageCode.Peer.TransferResponse, connection.Username, transferResponse.Token), transferResponse);
break;
case MessageCode.Peer.QueueDownload:
var queueDownloadRequest = QueueDownloadRequest.FromByteArray(message);
var(queueRejected, queueRejectionMessage) =
await TryEnqueueDownloadAsync(connection.Username, connection.IPEndPoint, queueDownloadRequest.Filename).ConfigureAwait(false);
if (queueRejected)
{
await connection.WriteAsync(new UploadDenied(queueDownloadRequest.Filename, queueRejectionMessage)).ConfigureAwait(false);
}
else
{
await TrySendPlaceInQueueAsync(connection, queueDownloadRequest.Filename).ConfigureAwait(false);
}
break;
case MessageCode.Peer.TransferRequest:
var transferRequest = TransferRequest.FromByteArray(message);
if (transferRequest.Direction == TransferDirection.Upload)
{
if (!SoulseekClient.DownloadDictionary.IsEmpty && SoulseekClient.DownloadDictionary.Values.Any(d => d.Username == connection.Username && d.Filename == transferRequest.Filename))
{
SoulseekClient.Waiter.Complete(new WaitKey(MessageCode.Peer.TransferRequest, connection.Username, transferRequest.Filename), transferRequest);
}
else
{
// reject the transfer with an empty reason. it was probably cancelled, but we can't be sure.
Diagnostic.Debug($"Rejecting unknown upload from {connection.Username} for {transferRequest.Filename} with token {transferRequest.Token}");
await connection.WriteAsync(new TransferResponse(transferRequest.Token, "Cancelled")).ConfigureAwait(false);
}
}
else
{
var(transferRejected, transferRejectionMessage) = await TryEnqueueDownloadAsync(connection.Username, connection.IPEndPoint, transferRequest.Filename).ConfigureAwait(false);
if (transferRejected)
{
await connection.WriteAsync(new TransferResponse(transferRequest.Token, transferRejectionMessage)).ConfigureAwait(false);
await connection.WriteAsync(new UploadDenied(transferRequest.Filename, transferRejectionMessage)).ConfigureAwait(false);
}
else
{
await connection.WriteAsync(new TransferResponse(transferRequest.Token, "Queued")).ConfigureAwait(false);
await TrySendPlaceInQueueAsync(connection, transferRequest.Filename).ConfigureAwait(false);
}
}
break;
case MessageCode.Peer.UploadDenied:
var uploadDeniedResponse = UploadDenied.FromByteArray(message);
Diagnostic.Debug($"Download of {uploadDeniedResponse.Filename} from {connection.Username} was denied: {uploadDeniedResponse.Message}");
SoulseekClient.Waiter.Throw(new WaitKey(MessageCode.Peer.TransferRequest, connection.Username, uploadDeniedResponse.Filename), new TransferRejectedException(uploadDeniedResponse.Message));
DownloadDenied?.Invoke(this, new DownloadDeniedEventArgs(connection.Username, uploadDeniedResponse.Filename, uploadDeniedResponse.Message));
break;
case MessageCode.Peer.PlaceInQueueResponse:
var placeInQueueResponse = PlaceInQueueResponse.FromByteArray(message);
SoulseekClient.Waiter.Complete(new WaitKey(MessageCode.Peer.PlaceInQueueResponse, connection.Username, placeInQueueResponse.Filename), placeInQueueResponse);
break;
case MessageCode.Peer.PlaceInQueueRequest:
var placeInQueueRequest = PlaceInQueueRequest.FromByteArray(message);
await TrySendPlaceInQueueAsync(connection, placeInQueueRequest.Filename).ConfigureAwait(false);
break;
case MessageCode.Peer.UploadFailed:
var uploadFailedResponse = UploadFailed.FromByteArray(message);
var msg = $"Download of {uploadFailedResponse.Filename} reported as failed by {connection.Username}";
var download = SoulseekClient.DownloadDictionary.Values.FirstOrDefault(d => d.Username == connection.Username && d.Filename == uploadFailedResponse.Filename);
if (download != null)
{
SoulseekClient.Waiter.Throw(new WaitKey(MessageCode.Peer.TransferRequest, download.Username, download.Filename), new TransferException(msg));
}
Diagnostic.Debug(msg);
DownloadFailed?.Invoke(this, new DownloadFailedEventArgs(connection.Username, uploadFailedResponse.Filename));
break;
default:
Diagnostic.Debug($"Unhandled peer message: {code} from {connection.Username} ({connection.IPEndPoint}); {message.Length} bytes");
break;
}
}
catch (Exception ex)
{
Diagnostic.Warning($"Error handling peer message: {code} from {connection.Username} ({connection.IPEndPoint}); {ex.Message}", ex);
}
}
