private TcpMessage HandleMultiRelationQuery(TcpConnectedHost host, TcpMessage message)
{
if (!message.HasType(TcpRequestType.MultiRelationQuery))
{
return(null);
}
if (!IsAuthenticated(host))
{
throw AuthException.NotLoggedIn();
}
var request = new ParamTcpMessage(message);
var response = new ObjectTcpMessage <DbRelation>(TcpRequestType.List);
if (request.HasParam("from_id"))
{
var fromId = request.GetParamAsNullableInt("from_id");
var items = _dataService.GetRelations(fromId);
response.Add(items);
return(response);
}
return(TcpMessage.Error("Missing or badly formatted query parameters"));
}
private TcpMessage HandleUniqueItemQuery(TcpConnectedHost host, TcpMessage message)
{
if (!message.HasType(TcpRequestType.UniqueItemQuery))
{
return(null);
}
if (!IsAuthenticated(host))
{
throw AuthException.NotLoggedIn();
}
var request = new ParamTcpMessage(message);
var response = new ObjectTcpMessage <DbItem>(TcpRequestType.List);
if (request.HasParam("id"))
{
var id = request.GetParamAsNullableInt("id");
if (id != null)
{
var item = _dataService.GetItem(id.Value);
response.Add(item);
return(response);
}
}
return(TcpMessage.Error("Missing or badly formatted query parameters"));
}
private TcpMessage HandleLoginRequest(TcpConnectedHost host, TcpMessage message)
{
if (!message.HasType("loginbasic"))
{
return(null);
}
var request = new ParamTcpMessage(message);
var username = request.GetParam("username");
var password = request.GetParam("password");
if (!string.IsNullOrEmpty(username) && !string.IsNullOrEmpty(password))
{
var userItem = _dataService.GetItemByRelation(UsersParentId, username);
if (userItem == null)
{
if (!AutoRegisterUsers)
{
return(TcpMessage.Error("User not found. Automatic registering of users is currently disabled."));
}
// Register
userItem = new DbItem(username);
_dataService.Insert(userItem);
_dataService.Insert(new DbRelation(UsersParentId, username, userItem.Id));
var saltItem = new DbItem(HashHelper.CreateSaltString(10));
_dataService.Insert(saltItem);
_dataService.Insert(new DbRelation(userItem.Id, "salt", saltItem.Id));
var passwordItem = new DbItem(HashHelper.GenerateSaltedHash(password, saltItem.Value));
_dataService.Insert(passwordItem);
_dataService.Insert(new DbRelation(userItem.Id, "password", passwordItem.Id));
}
else
{
// Login
var saltItem = _dataService.GetItemByRelation(userItem.Id, "salt");
var passwordItem = _dataService.GetItemByRelation(userItem.Id, "password");
if (saltItem == null || saltItem.Value == null || passwordItem == null || passwordItem.Value == null || !HashHelper.ConfirmPassword(passwordItem.Value, password, saltItem.Value))
{
return(TcpMessage.Error("Login failed. Wrong password."));
}
}
_hostUsers[host] = userItem;
var response = new ObjectTcpMessage <DbItem>(TcpRequestType.Ok);
response.Add(userItem);
return(response);
}
return(TcpMessage.Error("Missing or badly formatted login parameters"));
}
private void HandleDataClientComm(object client)
{
var tcpClient = (System.Net.Sockets.TcpClient)client;
var addressSplits = tcpClient.Client.RemoteEndPoint.ToString().Split(':');
var host = new TcpConnectedHost
{
TcpClient = tcpClient,
IPAddress = addressSplits[0]
};
if (!IsAllowed(host.IPAddress))
{
Debug.WriteLine("Client refused access to data channel. IP: " + host.IPAddress);
tcpClient.Close();
return;
}
Debug.WriteLine("Client connected to data channel. IP: " + host.IPAddress);
var stream = OnPrepareStream(host.TcpClient.GetStream());
_connectedDataClients.AddLast(host);
while (true)
{
var rawRequest = TcpProtocolHelper.Read(stream);
if (rawRequest == null)
{
break;
}
rawRequest = PrepareIncommingMessage(host, rawRequest);
Debug.WriteLine(DateTime.Now.ToLongTimeString() + " [R] " + rawRequest);
TcpMessage response = null;
var request = TcpMessage.FromRaw(rawRequest);
foreach (var handler in _dataRequestHandlers)
{
try
{
response = handler.Invoke(host, request);
if (response != null)
{
break;
}
}
catch (Exception e)
{
response = TcpMessage.Error(e.ToString());
break;
}
}
if (response == null)
{
response = TcpMessage.Error("Could not handle request");
}
var responseMessage = response.ToString();
Debug.WriteLine(DateTime.Now.ToLongTimeString() + " [S] " + responseMessage);
responseMessage = PrepareOutgoingMessage(host, responseMessage);
var success = TcpProtocolHelper.Write(stream, responseMessage);
if (!success)
{
break;
}
}
Debug.WriteLine("Client disconnected from data channel. IP: " + host.IPAddress);
tcpClient.Close();
}