mirror of https://github.com/icedream/npsharp.git
add validateticket call
NTAuthority
parent
612fd05702
commit
ac7e1b1a78
|
|
@ -1,147 +1,147 @@
|
|||
using System;
|
||||
using System.IO;
|
||||
using System.Net;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using log4net;
|
||||
using NPSharp.RPC;
|
||||
using NPSharp.RPC.Messages.Client;
|
||||
using NPSharp.RPC.Messages.Server;
|
||||
|
||||
namespace NPSharp.NP
|
||||
{
|
||||
/// <summary>
|
||||
/// Represents a high-level network platform client.
|
||||
/// </summary>
|
||||
public class NPClient
|
||||
{
|
||||
private readonly string _host;
|
||||
private readonly ILog _log;
|
||||
private readonly ushort _port;
|
||||
private CancellationToken _cancellationToken;
|
||||
private CancellationTokenSource _cancellationTokenSource;
|
||||
private RPCClientStream _rpc;
|
||||
|
||||
/// <summary>
|
||||
/// Initializes the NP client with a specified host and port.
|
||||
/// </summary>
|
||||
/// <param name="host">The host to connect to.</param>
|
||||
/// <param name="port">The port to use. Default: 3025.</param>
|
||||
public NPClient(string host, ushort port = 3025)
|
||||
{
|
||||
_host = host;
|
||||
_port = port;
|
||||
_log = LogManager.GetLogger("NPClient");
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// The internal RPC client.
|
||||
using System;
|
||||
using System.IO;
|
||||
using System.Net;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using log4net;
|
||||
using NPSharp.RPC;
|
||||
using NPSharp.RPC.Messages.Client;
|
||||
using NPSharp.RPC.Messages.Server;
|
||||
|
||||
namespace NPSharp.NP
|
||||
{
|
||||
/// <summary>
|
||||
/// Represents a high-level network platform client.
|
||||
/// </summary>
|
||||
public class NPClient
|
||||
{
|
||||
private readonly string _host;
|
||||
private readonly ILog _log;
|
||||
private readonly ushort _port;
|
||||
private CancellationToken _cancellationToken;
|
||||
private CancellationTokenSource _cancellationTokenSource;
|
||||
private RPCClientStream _rpc;
|
||||
|
||||
/// <summary>
|
||||
/// Initializes the NP client with a specified host and port.
|
||||
/// </summary>
|
||||
public RPCClientStream RPCClient { get { return _rpc; } }
|
||||
|
||||
/// <summary>
|
||||
/// The assigned NP user ID. Will be set on successful authentication.
|
||||
/// </summary>
|
||||
public ulong LoginId { get; private set; }
|
||||
|
||||
/// <summary>
|
||||
/// The assigned session token for this client. Will be set on successful authentication.
|
||||
/// </summary>
|
||||
public string SessionToken { get; private set; }
|
||||
|
||||
// TODO: Handle connection failures via exception
|
||||
/// <summary>
|
||||
/// Connects the client to the NP server.
|
||||
/// </summary>
|
||||
/// <returns>True if the connection succeeded, otherwise false.</returns>
|
||||
public bool Connect()
|
||||
{
|
||||
_log.Debug("Connect() start");
|
||||
|
||||
_cancellationTokenSource = new CancellationTokenSource();
|
||||
_cancellationToken = _cancellationTokenSource.Token;
|
||||
|
||||
try
|
||||
{
|
||||
_rpc = RPCClientStream.Open(_host, _port);
|
||||
}
|
||||
catch (Exception err)
|
||||
{
|
||||
#if DEBUG
|
||||
_log.ErrorFormat(@"Could not initialize RPC: {0}", err);
|
||||
#else
|
||||
_log.ErrorFormat(@"Could not initialize RPC: {0}", err.Message);
|
||||
#endif
|
||||
return false;
|
||||
}
|
||||
|
||||
Task.Factory.StartNew(() =>
|
||||
{
|
||||
_log.Debug("Now receiving RPC messages");
|
||||
try
|
||||
{
|
||||
while (true)
|
||||
{
|
||||
if (_rpc.Read() == null)
|
||||
break;
|
||||
}
|
||||
}
|
||||
catch (ProtocolViolationException error)
|
||||
{
|
||||
_log.ErrorFormat("Protocol violation: {0}. Disconnect imminent.", error.Message);
|
||||
Disconnect();
|
||||
}
|
||||
|
||||
_log.Debug("Now not receiving RPC messages anymore");
|
||||
}, _cancellationToken);
|
||||
|
||||
_log.Debug("Connect() done");
|
||||
return true;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Disconnects the client from the NP server.
|
||||
/// </summary>
|
||||
public void Disconnect()
|
||||
{
|
||||
_log.Debug("Disconnect() start");
|
||||
|
||||
_cancellationTokenSource.Cancel(true);
|
||||
// TODO: Find a cleaner way to cancel _processingTask (focus: _rpc.Read)
|
||||
//_procTask.Wait(_cancellationToken);
|
||||
_rpc.Close();
|
||||
|
||||
LoginId = 0;
|
||||
|
||||
_log.Debug("Disconnect() done");
|
||||
}
|
||||
|
||||
// TODO: Try to use an exception for failed action instead
|
||||
/// <summary>
|
||||
/// Authenticates this connection via a token. This token has to be requested via an external interface like
|
||||
/// remauth.php.
|
||||
/// </summary>
|
||||
/// <param name="token">The token to use for authentication</param>
|
||||
/// <returns>True if the login succeeded, otherwise false.</returns>
|
||||
public async Task<bool> AuthenticateWithToken(string token)
|
||||
{
|
||||
var tcs = new TaskCompletionSource<bool>();
|
||||
|
||||
_rpc.AttachHandlerForNextMessage(packet =>
|
||||
{
|
||||
var result = packet as AuthenticateResultMessage;
|
||||
if (result == null)
|
||||
return;
|
||||
|
||||
if (result.Result != 0)
|
||||
tcs.SetResult(false);
|
||||
LoginId = result.NPID;
|
||||
SessionToken = result.SessionToken;
|
||||
tcs.SetResult(true);
|
||||
});
|
||||
_rpc.Send(new AuthenticateWithTokenMessage {Token = token});
|
||||
|
||||
return await tcs.Task;
|
||||
/// <param name="host">The host to connect to.</param>
|
||||
/// <param name="port">The port to use. Default: 3025.</param>
|
||||
public NPClient(string host, ushort port = 3025)
|
||||
{
|
||||
_host = host;
|
||||
_port = port;
|
||||
_log = LogManager.GetLogger("NPClient");
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// The internal RPC client.
|
||||
/// </summary>
|
||||
public RPCClientStream RPCClient { get { return _rpc; } }
|
||||
|
||||
/// <summary>
|
||||
/// The assigned NP user ID. Will be set on successful authentication.
|
||||
/// </summary>
|
||||
public ulong LoginId { get; private set; }
|
||||
|
||||
/// <summary>
|
||||
/// The assigned session token for this client. Will be set on successful authentication.
|
||||
/// </summary>
|
||||
public string SessionToken { get; private set; }
|
||||
|
||||
// TODO: Handle connection failures via exception
|
||||
/// <summary>
|
||||
/// Connects the client to the NP server.
|
||||
/// </summary>
|
||||
/// <returns>True if the connection succeeded, otherwise false.</returns>
|
||||
public bool Connect()
|
||||
{
|
||||
_log.Debug("Connect() start");
|
||||
|
||||
_cancellationTokenSource = new CancellationTokenSource();
|
||||
_cancellationToken = _cancellationTokenSource.Token;
|
||||
|
||||
try
|
||||
{
|
||||
_rpc = RPCClientStream.Open(_host, _port);
|
||||
}
|
||||
catch (Exception err)
|
||||
{
|
||||
#if DEBUG
|
||||
_log.ErrorFormat(@"Could not initialize RPC: {0}", err);
|
||||
#else
|
||||
_log.ErrorFormat(@"Could not initialize RPC: {0}", err.Message);
|
||||
#endif
|
||||
return false;
|
||||
}
|
||||
|
||||
Task.Factory.StartNew(() =>
|
||||
{
|
||||
_log.Debug("Now receiving RPC messages");
|
||||
try
|
||||
{
|
||||
while (true)
|
||||
{
|
||||
if (_rpc.Read() == null)
|
||||
break;
|
||||
}
|
||||
}
|
||||
catch (ProtocolViolationException error)
|
||||
{
|
||||
_log.ErrorFormat("Protocol violation: {0}. Disconnect imminent.", error.Message);
|
||||
Disconnect();
|
||||
}
|
||||
|
||||
_log.Debug("Now not receiving RPC messages anymore");
|
||||
}, _cancellationToken);
|
||||
|
||||
_log.Debug("Connect() done");
|
||||
return true;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Disconnects the client from the NP server.
|
||||
/// </summary>
|
||||
public void Disconnect()
|
||||
{
|
||||
_log.Debug("Disconnect() start");
|
||||
|
||||
_cancellationTokenSource.Cancel(true);
|
||||
// TODO: Find a cleaner way to cancel _processingTask (focus: _rpc.Read)
|
||||
//_procTask.Wait(_cancellationToken);
|
||||
_rpc.Close();
|
||||
|
||||
LoginId = 0;
|
||||
|
||||
_log.Debug("Disconnect() done");
|
||||
}
|
||||
|
||||
// TODO: Try to use an exception for failed action instead
|
||||
/// <summary>
|
||||
/// Authenticates this connection via a token. This token has to be requested via an external interface like
|
||||
/// remauth.php.
|
||||
/// </summary>
|
||||
/// <param name="token">The token to use for authentication</param>
|
||||
/// <returns>True if the login succeeded, otherwise false.</returns>
|
||||
public async Task<bool> AuthenticateWithToken(string token)
|
||||
{
|
||||
var tcs = new TaskCompletionSource<bool>();
|
||||
|
||||
_rpc.AttachHandlerForNextMessage(packet =>
|
||||
{
|
||||
var result = packet as AuthenticateResultMessage;
|
||||
if (result == null)
|
||||
return;
|
||||
|
||||
if (result.Result != 0)
|
||||
tcs.SetResult(false);
|
||||
LoginId = result.NPID;
|
||||
SessionToken = result.SessionToken;
|
||||
tcs.SetResult(true);
|
||||
});
|
||||
_rpc.Send(new AuthenticateWithTokenMessage {Token = token});
|
||||
|
||||
return await tcs.Task;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
|
|
@ -161,7 +161,12 @@ namespace NPSharp.NP
|
|||
return;
|
||||
|
||||
if (result.Result != 0)
|
||||
{
|
||||
tcs.SetResult(false);
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
LoginId = result.NPID;
|
||||
SessionToken = result.SessionToken;
|
||||
tcs.SetResult(true);
|
||||
|
|
@ -169,108 +174,132 @@ namespace NPSharp.NP
|
|||
_rpc.Send(new AuthenticateWithKeyMessage { LicenseKey = key });
|
||||
|
||||
return await tcs.Task;
|
||||
}
|
||||
|
||||
// TODO: Try to use an exception for failed action instead
|
||||
/// <summary>
|
||||
/// Uploads a user file.
|
||||
/// </summary>
|
||||
/// <param name="filename">The file name to save the contents to on the server</param>
|
||||
/// <param name="contents">The raw byte contents</param>
|
||||
/// <returns>True if the upload succeeded, otherwise false.</returns>
|
||||
public async Task<bool> UploadUserFile(string filename, byte[] contents)
|
||||
{
|
||||
var tcs = new TaskCompletionSource<bool>();
|
||||
|
||||
_rpc.AttachHandlerForNextMessage(packet =>
|
||||
{
|
||||
var result = (StorageWriteUserFileResultMessage) packet;
|
||||
if (result.Result != 0)
|
||||
tcs.SetResult(false);
|
||||
tcs.SetResult(true);
|
||||
});
|
||||
_rpc.Send(new StorageWriteUserFileMessage {FileData = contents, FileName = filename, NPID = LoginId});
|
||||
|
||||
return await tcs.Task;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Downloads a user file and returns its contents.
|
||||
/// </summary>
|
||||
/// <param name="filename">The file to download</param>
|
||||
/// <returns>File contents as byte array</returns>
|
||||
public async Task<byte[]> GetUserFile(string filename)
|
||||
{
|
||||
var tcs = new TaskCompletionSource<byte[]>();
|
||||
|
||||
_rpc.AttachHandlerForNextMessage(packet =>
|
||||
{
|
||||
var result = (StorageUserFileMessage) packet;
|
||||
if (result.Result != 0)
|
||||
{
|
||||
tcs.SetException(new NpFileException(result.Result));
|
||||
return;
|
||||
}
|
||||
tcs.SetResult(result.FileData);
|
||||
});
|
||||
_rpc.Send(new StorageGetUserFileMessage {FileName = filename, NPID = LoginId});
|
||||
|
||||
return await tcs.Task;
|
||||
}
|
||||
|
||||
|
||||
/// <summary>
|
||||
/// Downloads a user file onto the harddisk.
|
||||
/// </summary>
|
||||
/// <param name="filename">The file to download</param>
|
||||
/// <param name="targetpath">Path where to save the file</param>
|
||||
public async void DownloadUserFileTo(string filename, string targetpath)
|
||||
{
|
||||
var contents = await GetUserFile(filename);
|
||||
|
||||
File.WriteAllBytes(targetpath, contents);
|
||||
}
|
||||
|
||||
|
||||
/// <summary>
|
||||
/// Downloads a publisher file and returns its contents.
|
||||
/// </summary>
|
||||
/// <param name="filename">The file to download</param>
|
||||
/// <returns>File contents as byte array</returns>
|
||||
public async Task<byte[]> GetPublisherFile(string filename)
|
||||
{
|
||||
var tcs = new TaskCompletionSource<byte[]>();
|
||||
|
||||
_rpc.AttachHandlerForNextMessage(packet =>
|
||||
{
|
||||
var result = (StoragePublisherFileMessage) packet;
|
||||
if (result.Result != 0)
|
||||
{
|
||||
tcs.SetException(new NpFileException(result.Result));
|
||||
return;
|
||||
}
|
||||
tcs.SetResult(result.FileData);
|
||||
});
|
||||
_rpc.Send(new StorageGetPublisherFileMessage {FileName = filename});
|
||||
|
||||
return await tcs.Task;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Downloads a publisher file onto the harddisk.
|
||||
/// </summary>
|
||||
/// <param name="filename">The file to download</param>
|
||||
/// <param name="targetpath">Path where to save the file</param>
|
||||
public async void DownloadPublisherFileTo(string filename, string targetpath)
|
||||
{
|
||||
var contents = await GetPublisherFile(filename);
|
||||
|
||||
File.WriteAllBytes(targetpath, contents);
|
||||
}
|
||||
|
||||
public void SendRandomString(string data)
|
||||
{
|
||||
_rpc.Send(new StorageSendRandomStringMessage {RandomString = data});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public async Task<bool> ValidateTicket(uint clientIP, ulong npID, byte[] ticket)
|
||||
{
|
||||
var tcs = new TaskCompletionSource<bool>();
|
||||
|
||||
_rpc.AttachHandlerForNextMessage(packet =>
|
||||
{
|
||||
var result = packet as AuthenticateValidateTicketResultMessage;
|
||||
if (result == null)
|
||||
return;
|
||||
|
||||
if (result.Result != 0)
|
||||
{
|
||||
tcs.SetResult(false);
|
||||
}
|
||||
else
|
||||
{
|
||||
tcs.SetResult(true);
|
||||
}
|
||||
});
|
||||
_rpc.Send(new AuthenticateValidateTicketMessage { ClientIP = clientIP, Ticket = ticket, NPID = npID });
|
||||
|
||||
return await tcs.Task;
|
||||
}
|
||||
|
||||
// TODO: Try to use an exception for failed action instead
|
||||
/// <summary>
|
||||
/// Uploads a user file.
|
||||
/// </summary>
|
||||
/// <param name="filename">The file name to save the contents to on the server</param>
|
||||
/// <param name="contents">The raw byte contents</param>
|
||||
/// <returns>True if the upload succeeded, otherwise false.</returns>
|
||||
public async Task<bool> UploadUserFile(string filename, byte[] contents)
|
||||
{
|
||||
var tcs = new TaskCompletionSource<bool>();
|
||||
|
||||
_rpc.AttachHandlerForNextMessage(packet =>
|
||||
{
|
||||
var result = (StorageWriteUserFileResultMessage) packet;
|
||||
if (result.Result != 0)
|
||||
tcs.SetResult(false);
|
||||
tcs.SetResult(true);
|
||||
});
|
||||
_rpc.Send(new StorageWriteUserFileMessage {FileData = contents, FileName = filename, NPID = LoginId});
|
||||
|
||||
return await tcs.Task;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Downloads a user file and returns its contents.
|
||||
/// </summary>
|
||||
/// <param name="filename">The file to download</param>
|
||||
/// <returns>File contents as byte array</returns>
|
||||
public async Task<byte[]> GetUserFile(string filename)
|
||||
{
|
||||
var tcs = new TaskCompletionSource<byte[]>();
|
||||
|
||||
_rpc.AttachHandlerForNextMessage(packet =>
|
||||
{
|
||||
var result = (StorageUserFileMessage) packet;
|
||||
if (result.Result != 0)
|
||||
{
|
||||
tcs.SetException(new NpFileException(result.Result));
|
||||
return;
|
||||
}
|
||||
tcs.SetResult(result.FileData);
|
||||
});
|
||||
_rpc.Send(new StorageGetUserFileMessage {FileName = filename, NPID = LoginId});
|
||||
|
||||
return await tcs.Task;
|
||||
}
|
||||
|
||||
|
||||
/// <summary>
|
||||
/// Downloads a user file onto the harddisk.
|
||||
/// </summary>
|
||||
/// <param name="filename">The file to download</param>
|
||||
/// <param name="targetpath">Path where to save the file</param>
|
||||
public async void DownloadUserFileTo(string filename, string targetpath)
|
||||
{
|
||||
var contents = await GetUserFile(filename);
|
||||
|
||||
File.WriteAllBytes(targetpath, contents);
|
||||
}
|
||||
|
||||
|
||||
/// <summary>
|
||||
/// Downloads a publisher file and returns its contents.
|
||||
/// </summary>
|
||||
/// <param name="filename">The file to download</param>
|
||||
/// <returns>File contents as byte array</returns>
|
||||
public async Task<byte[]> GetPublisherFile(string filename)
|
||||
{
|
||||
var tcs = new TaskCompletionSource<byte[]>();
|
||||
|
||||
_rpc.AttachHandlerForNextMessage(packet =>
|
||||
{
|
||||
var result = (StoragePublisherFileMessage) packet;
|
||||
if (result.Result != 0)
|
||||
{
|
||||
tcs.SetException(new NpFileException(result.Result));
|
||||
return;
|
||||
}
|
||||
tcs.SetResult(result.FileData);
|
||||
});
|
||||
_rpc.Send(new StorageGetPublisherFileMessage {FileName = filename});
|
||||
|
||||
return await tcs.Task;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Downloads a publisher file onto the harddisk.
|
||||
/// </summary>
|
||||
/// <param name="filename">The file to download</param>
|
||||
/// <param name="targetpath">Path where to save the file</param>
|
||||
public async void DownloadPublisherFileTo(string filename, string targetpath)
|
||||
{
|
||||
var contents = await GetPublisherFile(filename);
|
||||
|
||||
File.WriteAllBytes(targetpath, contents);
|
||||
}
|
||||
|
||||
public void SendRandomString(string data)
|
||||
{
|
||||
_rpc.Send(new StorageSendRandomStringMessage {RandomString = data});
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue