mirror of https://github.com/deavminet/dnetd.git
1116 lines
24 KiB
D
1116 lines
24 KiB
D
/**
|
|
* dconnection
|
|
*
|
|
* Client/server connection handler spawned
|
|
* by socket connection dequeue loop.
|
|
*
|
|
* Handles all interactions between
|
|
* the server and the specific client/server.
|
|
*/
|
|
|
|
module dnetd.dconnection;
|
|
|
|
import core.thread : Thread;
|
|
import std.socket : Socket;
|
|
import bmessage;
|
|
import tristanable.encoding : DataMessage;
|
|
import core.sync.mutex : Mutex;
|
|
import dnetd.dserver : DServer;
|
|
import std.string : split;
|
|
import dnetd.dchannel : DChannel;
|
|
import std.conv : to;
|
|
import std.stdio : writeln;
|
|
import std.algorithm : reverse;
|
|
import gogga;
|
|
|
|
public class DConnection : Thread
|
|
{
|
|
/* The connection type */
|
|
public enum ConnectionType
|
|
{
|
|
CLIENT, SERVER, UNSPEC
|
|
}
|
|
|
|
/* Command types */
|
|
public enum Command
|
|
{
|
|
JOIN,
|
|
PART,
|
|
AUTH,
|
|
LINK,
|
|
REGISTER,
|
|
LIST,
|
|
MSG,
|
|
MEMBER_COUNT,
|
|
MEMBER_LIST,
|
|
SERVER_INFO,
|
|
MOTD,
|
|
MEMBER_INFO,
|
|
STATUS,
|
|
CHAN_PROP,
|
|
SET_PROP,
|
|
|
|
|
|
/* User property commands */
|
|
GET_USER_PROPS,
|
|
GET_USER_PROP,
|
|
SET_USER_PROP,
|
|
DELETE_USER_PROP,
|
|
IS_USER_PROP,
|
|
|
|
|
|
UNKNOWN
|
|
}
|
|
|
|
/**
|
|
* Connection information
|
|
*/
|
|
private DServer server;
|
|
private Socket socket;
|
|
private bool hasAuthed;
|
|
private ConnectionType connType;
|
|
private string username;
|
|
private string currentStatus;
|
|
|
|
/**
|
|
* User property support
|
|
*
|
|
* `properties` - the property store
|
|
* `propertiesMutex` - the mutex
|
|
*/
|
|
private string[string] properties; /* TODO: New, replace old status mechanism */
|
|
private Mutex propertiesLock;
|
|
|
|
/* Write lock for socket */
|
|
/* TODO: Forgot how bmessage works, might need, might not, if multipel calls
|
|
* then yes, if single then no as it is based off (well glibc's write)
|
|
* thread safe code
|
|
*/
|
|
private Mutex writeLock;
|
|
|
|
/**
|
|
* Mutex to provide safe access to the status message
|
|
*/
|
|
private Mutex statusMessageLock;
|
|
|
|
this(DServer server, Socket socket)
|
|
{
|
|
/* Set the function to be called on thread start */
|
|
super(&worker);
|
|
|
|
/* Set the associated server */
|
|
this.server = server;
|
|
|
|
/* Set the socket */
|
|
this.socket = socket;
|
|
|
|
/* Set the default state */
|
|
connType = ConnectionType.UNSPEC;
|
|
|
|
/* Initialize locks */
|
|
initLocks();
|
|
|
|
/* Initialize status (TODO: Remove, we use props now) */
|
|
currentStatus = "available,Hey there I'm using DNET!";
|
|
|
|
/* Start the connection handler */
|
|
start();
|
|
}
|
|
|
|
/**
|
|
* Initializes mutexes
|
|
*/
|
|
private void initLocks()
|
|
{
|
|
/* Initialize the socket write lock */
|
|
writeLock = new Mutex();
|
|
|
|
/* Initialize the status message lock */
|
|
statusMessageLock = new Mutex();
|
|
|
|
/* Initialize the properties lock */
|
|
propertiesLock = new Mutex();
|
|
}
|
|
|
|
/**
|
|
* Byte dequeue loop
|
|
*/
|
|
private void worker()
|
|
{
|
|
/* Received bytes (for bformat) */
|
|
byte[] receivedBytes;
|
|
|
|
/* Received message */
|
|
DataMessage receivedMessage;
|
|
|
|
while(true)
|
|
{
|
|
/**
|
|
* Block to receive a bformat message
|
|
*
|
|
* (Does decoding for bformat too)
|
|
*/
|
|
bool status = receiveMessage(socket, receivedBytes);
|
|
|
|
/* TODO: Check status */
|
|
if(status)
|
|
{
|
|
/* Decode the tristanable message (tagged message) */
|
|
receivedMessage = DataMessage.decode(receivedBytes);
|
|
|
|
/* Process the message */
|
|
process(receivedMessage);
|
|
}
|
|
else
|
|
{
|
|
/* TODO: Error handling */
|
|
gprintln("Error with receive: "~to!(string)(this), DebugType.ERROR);
|
|
break;
|
|
}
|
|
}
|
|
|
|
/* Clean up */
|
|
cleanUp();
|
|
}
|
|
|
|
private void cleanUp()
|
|
{
|
|
gprintln(to!(string)(this)~" Cleaning up connection...");
|
|
|
|
/* Remove this user from all channels he is in */
|
|
DChannel[] channels = server.getChannels();
|
|
|
|
/* Loop through each channel */
|
|
foreach(DChannel currentChannel; channels)
|
|
{
|
|
/* Check if you are a member of it */
|
|
if(currentChannel.isMember(this))
|
|
{
|
|
/* Leave the channel */
|
|
currentChannel.leave(this);
|
|
gprintln(to!(string)(this)~" Leaving '"~currentChannel.getName()~"'...");
|
|
}
|
|
}
|
|
|
|
/* Remove this user from the connection queue */
|
|
server.removeConnection(this);
|
|
|
|
gprintln(to!(string)(this)~" Connection cleaned up");
|
|
}
|
|
|
|
/* TODO: add mutex for writing with message and funciton for doing so */
|
|
|
|
/**
|
|
* Write to socket
|
|
*
|
|
* Encodes the byte array as a tristanable tagged
|
|
* message and then encodes that as a bformat
|
|
* message
|
|
*
|
|
* Locks the writeLock mutex, sends it over the
|
|
* socket to the client/server, and unlocks the
|
|
* mutex
|
|
*/
|
|
public bool writeSocket(long tag, byte[] data)
|
|
{
|
|
/* Send status */
|
|
bool status;
|
|
|
|
/* Create the tagged message */
|
|
DataMessage message = new DataMessage(tag, data);
|
|
|
|
gprintln("writeSocket: mutex lock");
|
|
/* Lock the write mutex */
|
|
writeLock.lock();
|
|
|
|
/* Send the message */
|
|
gprintln("writeSocket: Data: "~to!(string)(data)~" Tag: "~to!(string)(tag));
|
|
status = sendMessage(socket, message.encode());
|
|
|
|
/* Unlock the write mutex */
|
|
writeLock.unlock();
|
|
|
|
gprintln("writeSocket: mutex unlock");
|
|
|
|
return status;
|
|
}
|
|
|
|
private Command getCommand(ubyte commandByte)
|
|
{
|
|
Command command = Command.UNKNOWN;
|
|
|
|
if(commandByte == 0)
|
|
{
|
|
command = Command.AUTH;
|
|
}
|
|
else if(commandByte == 1)
|
|
{
|
|
command = Command.LINK;
|
|
}
|
|
else if(commandByte == 2)
|
|
{
|
|
command = Command.REGISTER;
|
|
}
|
|
else if(commandByte == 3)
|
|
{
|
|
command = Command.JOIN;
|
|
}
|
|
else if(commandByte == 4)
|
|
{
|
|
command = Command.PART;
|
|
}
|
|
else if(commandByte == 5)
|
|
{
|
|
command = Command.MSG;
|
|
}
|
|
else if(commandByte == 6)
|
|
{
|
|
command = Command.LIST;
|
|
}
|
|
else if(commandByte == 7)
|
|
{
|
|
command = Command.MSG;
|
|
}
|
|
else if(commandByte == 8)
|
|
{
|
|
command = Command.MEMBER_COUNT;
|
|
}
|
|
else if(commandByte == 9)
|
|
{
|
|
command = Command.MEMBER_LIST;
|
|
}
|
|
else if(commandByte == 10)
|
|
{
|
|
command = Command.SERVER_INFO;
|
|
}
|
|
else if(commandByte == 11)
|
|
{
|
|
command = Command.MOTD;
|
|
}
|
|
else if(commandByte == 12)
|
|
{
|
|
command = Command.MEMBER_INFO;
|
|
}
|
|
else if(commandByte == 13)
|
|
{
|
|
command = Command.STATUS;
|
|
}
|
|
else if(commandByte == 14)
|
|
{
|
|
command = Command.CHAN_PROP;
|
|
}
|
|
else if(commandByte == 15)
|
|
{
|
|
command = Command.GET_USER_PROPS;
|
|
}
|
|
else if(commandByte == 16)
|
|
{
|
|
command = Command.GET_USER_PROP;
|
|
}
|
|
else if(commandByte == 17)
|
|
{
|
|
command = Command.SET_USER_PROP;
|
|
}
|
|
else if(commandByte == 18)
|
|
{
|
|
command = Command.DELETE_USER_PROP;
|
|
}
|
|
else if(commandByte == 19)
|
|
{
|
|
command = Command.IS_USER_PROP;
|
|
}
|
|
|
|
|
|
|
|
|
|
return command;
|
|
}
|
|
|
|
/**
|
|
* Process the received message
|
|
*/
|
|
private void process(DataMessage message)
|
|
{
|
|
/* Get the tag */
|
|
/**
|
|
* TODO: Client side will always do 1, because we don't have
|
|
* multi-thread job processing, only need this to differentiate
|
|
* between commands and async notifications
|
|
*/
|
|
long tag = message.tag;
|
|
gprintln("tag: "~ to!(string)(tag));
|
|
|
|
/* The reply */
|
|
byte[] reply;
|
|
|
|
/* Get the command */
|
|
byte commandByte = message.data[0];
|
|
Command command = getCommand(commandByte);
|
|
gprintln(to!(string)(this)~" ~> "~to!(string)(command));
|
|
|
|
/* If `auth` command (requires: unauthed) */
|
|
if(command == Command.AUTH && !hasAuthed)
|
|
{
|
|
/* Get the length of the username */
|
|
ubyte usernameLength = message.data[1];
|
|
|
|
/* Get the username and password */
|
|
string username = cast(string)message.data[2..2+cast(uint)usernameLength];
|
|
string password = cast(string)message.data[2+cast(uint)usernameLength..message.data.length];
|
|
|
|
/* Authenticate */
|
|
bool status = authenticate(username, password);
|
|
|
|
/* TODO: What to do on bad authetication? */
|
|
|
|
/* Set the username */
|
|
this.username = username;
|
|
|
|
/* Set the type of this connection to `client` */
|
|
connType = ConnectionType.CLIENT;
|
|
hasAuthed = true;
|
|
|
|
/* Encode the reply */
|
|
reply = [status];
|
|
}
|
|
/* If `link` command (requires: unauthed) */
|
|
else if(command == Command.LINK && !hasAuthed)
|
|
{
|
|
/* TODO: Implement me later */
|
|
|
|
/* Check if this connection is a DLink'd one */
|
|
//server.getMeyer().get
|
|
|
|
|
|
/* Set the type of this connection to `server` */
|
|
connType = ConnectionType.SERVER;
|
|
hasAuthed = true;
|
|
}
|
|
/* If `register` command (requires: unauthed, client) */
|
|
else if(command == Command.REGISTER && !hasAuthed && connType == ConnectionType.CLIENT)
|
|
{
|
|
|
|
}
|
|
/* If `join` command (requires: authed, client) */
|
|
else if(command == Command.JOIN && hasAuthed && connType == ConnectionType.CLIENT)
|
|
{
|
|
/* Get the channel names */
|
|
string channelList = cast(string)message.data[1..message.data.length];
|
|
string[] channels = split(channelList, ",");
|
|
|
|
/**
|
|
* Loop through each channel, check if it
|
|
* exists, if so join it, else create it
|
|
* and then join it
|
|
*/
|
|
bool isPresentInfo = false;
|
|
foreach(string channelName; channels)
|
|
{
|
|
/**
|
|
* Finds the channel, if it exists then it returns it,
|
|
* if it does not exist then it will create it and then
|
|
* return it
|
|
*/
|
|
DChannel channel = server.getChannel(this, channelName);
|
|
|
|
/* Join the channel */
|
|
isPresentInfo = channel.join(this);
|
|
}
|
|
|
|
/* TODO: Do reply */
|
|
/* Encode the reply */
|
|
reply = [isPresentInfo];
|
|
}
|
|
/* If `part` command (requires: authed, client) */
|
|
else if(command == Command.PART && hasAuthed && connType == ConnectionType.CLIENT)
|
|
{
|
|
/* Get the channel names */
|
|
string channelList = cast(string)message.data[1..message.data.length];
|
|
string[] channels = split(channelList, ",");
|
|
|
|
/**
|
|
* Loop through each channel, check if it
|
|
* exists, if so leave it
|
|
*/
|
|
foreach(string channelName; channels)
|
|
{
|
|
/* Attempt to find the channel */
|
|
DChannel channel = server.getChannelByName(channelName);
|
|
|
|
/* Leave a channel the channel only if it exists */
|
|
if(!(channel is null))
|
|
{
|
|
channel.leave(this);
|
|
}
|
|
}
|
|
|
|
/* TODO: Do reply */
|
|
/* Encode the reply */
|
|
reply = [true];
|
|
}
|
|
/* If `list` command (requires: authed, client) */
|
|
else if(command == Command.LIST && hasAuthed && connType == ConnectionType.CLIENT)
|
|
{
|
|
/* Get all channels */
|
|
DChannel[] channels = server.getChannels();
|
|
|
|
/* Generate a list of channel names (CSV) */
|
|
string channelList;
|
|
for(ulong i = 0; i < channels.length; i++)
|
|
{
|
|
if(i == channels.length-1)
|
|
{
|
|
channelList ~= channels[i].getName();
|
|
}
|
|
else
|
|
{
|
|
channelList ~= channels[i].getName()~",";
|
|
}
|
|
}
|
|
|
|
/* TODO: Reply */
|
|
/* Encode the reply */
|
|
reply = [true];
|
|
reply ~= channelList;
|
|
}
|
|
/* If `msg` command (requires: authed, client) */
|
|
else if(command == Command.MSG && hasAuthed && connType == ConnectionType.CLIENT)
|
|
{
|
|
/* Status */
|
|
bool status = true;
|
|
|
|
/* Get the type of message */
|
|
ubyte messageType = message.data[1];
|
|
|
|
/* Get the location length */
|
|
ubyte locationLength = message.data[2];
|
|
|
|
/* Get the channel/person name */
|
|
string destination = cast(string)message.data[3..3+cast(uint)locationLength];
|
|
|
|
/* Get the message */
|
|
string msg = cast(string)message.data[3+cast(uint)locationLength..message.data.length];
|
|
|
|
/* Send status */
|
|
bool sendStatus;
|
|
|
|
/* If we are sending to a user */
|
|
if(messageType == 0)
|
|
{
|
|
/* Send the message to the user */
|
|
sendStatus = sendUserMessage(destination, msg);
|
|
}
|
|
/* If we are sending to a channel */
|
|
else if(messageType == 1)
|
|
{
|
|
/* The channel wanting to send to */
|
|
DChannel channel = server.getChannelByName(destination);
|
|
|
|
/* If the channel exists */
|
|
if(channel)
|
|
{
|
|
/* TODO Implemet me */
|
|
sendStatus = channel.sendMessage(this, msg);
|
|
}
|
|
/* If the channel does not exist */
|
|
else
|
|
{
|
|
status = false;
|
|
}
|
|
}
|
|
/* Unknown destination type */
|
|
else
|
|
{
|
|
status = false;
|
|
}
|
|
|
|
/* TODO: Handling here, should we make the user wait? */
|
|
|
|
/* Encode the reply */
|
|
/* TODO: */
|
|
reply = [status];
|
|
}
|
|
/* If `membercount` command (requires: authed, client) */
|
|
else if(command == Command.MEMBER_COUNT && hasAuthed && connType == ConnectionType.CLIENT)
|
|
{
|
|
/* Status */
|
|
bool status = true;
|
|
|
|
/* Get the channel name */
|
|
string channelName = cast(string)message.data[1..message.data.length];
|
|
|
|
/* The memebr count */
|
|
long memberCount;
|
|
|
|
/* Get the member count */
|
|
status = getMemberCount(channelName, memberCount);
|
|
|
|
/* Encode the status */
|
|
reply = [status];
|
|
|
|
/* If there was no error fetching the member count */
|
|
if(status)
|
|
{
|
|
/* Data bytes */
|
|
byte[] numberBytes;
|
|
numberBytes.length = 8;
|
|
|
|
/* Encode the length (Big Endian) from Little Endian */
|
|
numberBytes[0] = *((cast(byte*)&memberCount)+7);
|
|
numberBytes[1] = *((cast(byte*)&memberCount)+6);
|
|
numberBytes[2] = *((cast(byte*)&memberCount)+5);
|
|
numberBytes[3] = *((cast(byte*)&memberCount)+4);
|
|
numberBytes[4] = *((cast(byte*)&memberCount)+3);
|
|
numberBytes[5] = *((cast(byte*)&memberCount)+2);
|
|
numberBytes[6] = *((cast(byte*)&memberCount)+1);
|
|
numberBytes[7] = *((cast(byte*)&memberCount)+0);
|
|
|
|
/* Append the length */
|
|
reply ~= numberBytes;
|
|
}
|
|
}
|
|
/* If `memberlist` command (requires: authed, client) */
|
|
else if(command == Command.MEMBER_LIST && hasAuthed && connType == ConnectionType.CLIENT)
|
|
{
|
|
/* Status */
|
|
bool status = true;
|
|
|
|
/* Get the channel name */
|
|
string channelName = cast(string)message.data[1..message.data.length];
|
|
|
|
/* Get the channel */
|
|
DChannel channel = server.getChannelByName(channelName);
|
|
|
|
/* Encode the status */
|
|
reply ~= [channel !is null];
|
|
|
|
/* If the channel exists */
|
|
if(channel)
|
|
{
|
|
/* Get the list of members in the channel */
|
|
DConnection[] members = channel.getMembers();
|
|
|
|
/* Construct a CSV string of the members */
|
|
string memberString;
|
|
|
|
for(ulong i = 0; i < members.length; i++)
|
|
{
|
|
if(i == members.length-1)
|
|
{
|
|
memberString ~= members[i].getUsername();
|
|
}
|
|
else
|
|
{
|
|
memberString ~= members[i].getUsername()~",";
|
|
}
|
|
}
|
|
|
|
/* Encode the string into the reply */
|
|
reply ~= cast(byte[])memberString;
|
|
}
|
|
/* If the channel does not exist */
|
|
else
|
|
{
|
|
status = false;
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
/* If `serverinfo` command (requires: authed, !unspec) */
|
|
else if(command == Command.SERVER_INFO && hasAuthed && connType != ConnectionType.UNSPEC)
|
|
{
|
|
/* Status */
|
|
bool status = true;
|
|
|
|
/* Get the server info */
|
|
string serverInfo = server.getServerInfo();
|
|
|
|
/* Encode the reply */
|
|
reply ~= [status];
|
|
reply ~= serverInfo;
|
|
}
|
|
/* If `motd` command (requires: _nothing_) */
|
|
else if(command == Command.MOTD)
|
|
{
|
|
/* Status */
|
|
bool status = true;
|
|
|
|
/* Get the message of the day */
|
|
string motd = server.getConfig().getGeneral().getMotd();
|
|
|
|
/* Encode the reply */
|
|
reply ~= [status];
|
|
reply ~= motd;
|
|
}
|
|
/* If `memberinfo` command (requires: authed, client) */
|
|
else if(command == Command.MEMBER_INFO && hasAuthed && connType == ConnectionType.CLIENT)
|
|
{
|
|
/* Status */
|
|
bool status = true;
|
|
|
|
/* TODO: Implement me */
|
|
string user = cast(string)message.data[1..message.data.length];
|
|
|
|
/* TODO: fetch longontime, serveron, status */
|
|
string logontime;
|
|
string serveron;
|
|
|
|
/* Encode the reply */
|
|
reply ~= [status];
|
|
reply ~= [cast(byte)logontime.length];
|
|
reply ~= logontime;
|
|
reply ~= [cast(byte)serveron.length];
|
|
reply ~= serveron;
|
|
reply ~= server.getStatusMessage(user);
|
|
}
|
|
/* If `status` command (requires: authed, client) */
|
|
else if(command == Command.STATUS && hasAuthed && connType == ConnectionType.CLIENT)
|
|
{
|
|
/* Get the new status line */
|
|
string statusMessage = cast(string)message.data[1..message.data.length];
|
|
|
|
/* Set the new status message */
|
|
setStatusMessage(statusMessage);
|
|
|
|
/* Encode the reply */
|
|
reply ~= [true];
|
|
}
|
|
|
|
/* If `get_user_props` (requires: authed, client) */
|
|
else if(command == Command.GET_USER_PROPS && hasAuthed && connType == ConnectionType.CLIENT)
|
|
{
|
|
/* Get the username */
|
|
string username = cast(string)message.data[1..message.data.length];
|
|
|
|
/* Get the user */
|
|
DConnection connection = server.findUser(username);
|
|
|
|
/* Get all properties of the user */
|
|
string[] propertyKeys = connection.getProperties();
|
|
|
|
/* Encode the status */
|
|
reply ~= [true];
|
|
|
|
/* Encode the keys */
|
|
for(ulong i = 0; i < propertyKeys.length; i++)
|
|
{
|
|
/* The data to add to the reply */
|
|
string replyData;
|
|
|
|
if(i == propertyKeys.length-1)
|
|
{
|
|
replyData = propertyKeys[i];
|
|
}
|
|
else
|
|
{
|
|
replyData = propertyKeys[i]~",";
|
|
}
|
|
|
|
/* Encode the `replyData` */
|
|
reply ~= replyData;
|
|
}
|
|
|
|
/* Encode the number of keys (TODO: FOr now you cannot have more than 255 keys) */
|
|
// reply ~= [cast(byte)propertyKeys.length];
|
|
}
|
|
/* If `get_user_prop` (requires: authed, client) */
|
|
else if(command == Command.GET_USER_PROP && hasAuthed && connType == ConnectionType.CLIENT)
|
|
{
|
|
/* Get the <user>,<propertyName> */
|
|
string[] dataLine = split(cast(string)message.data[1..message.data.length],",");
|
|
|
|
/* Get the username */
|
|
string username = dataLine[0];
|
|
|
|
/* Get the proerty */
|
|
string propertyName = dataLine[1];
|
|
|
|
/* Get the user */
|
|
DConnection connection = server.findUser(username);
|
|
|
|
/* Determine if it is a valid property */
|
|
bool status = connection.isProperty(propertyName);
|
|
|
|
/* Encode the status */
|
|
reply ~= [status];
|
|
|
|
/* Encode the property value if one exists */
|
|
if(status)
|
|
{
|
|
/* Get the property value */
|
|
string propertyValue = server.getProperty(username, propertyName);
|
|
|
|
/* Encode the property value */
|
|
reply ~= propertyValue;
|
|
}
|
|
}
|
|
/* If `set_user_prop` (requires: authed, client) */
|
|
else if(command == Command.SET_USER_PROP && hasAuthed && connType == ConnectionType.CLIENT)
|
|
{
|
|
/* Set the <propertyName>,<propertyValue> */
|
|
string[] dataLine = split(cast(string)message.data[1..message.data.length],",");
|
|
|
|
/* Get the property */
|
|
string propertyName = dataLine[0];
|
|
|
|
/* Get the property value */
|
|
string propertyValue = dataLine[1];
|
|
|
|
/* Encode the status */
|
|
reply ~= [true];
|
|
|
|
/* Set the property value */
|
|
setProperty(propertyName, propertyValue);
|
|
}
|
|
/* If `delete_user_prop` (requires: authed, client) */
|
|
else if(command == Command.DELETE_USER_PROP && hasAuthed && connType == ConnectionType.CLIENT)
|
|
{
|
|
/* Get the property */
|
|
string property = cast(string)message.data[1..message.data.length];
|
|
|
|
/* Check if the key exists */
|
|
bool keyExists = isProperty(property);
|
|
|
|
/* If the property exists */
|
|
if(keyExists)
|
|
{
|
|
/* Delete the property */
|
|
deleteProperty(property);
|
|
}
|
|
|
|
reply ~= [keyExists];
|
|
}
|
|
/* If `is_user_prop` (requires: authed, client) */
|
|
else if(command == Command.IS_USER_PROP && hasAuthed && connType == ConnectionType.CLIENT)
|
|
{
|
|
/* Get the <user>,<propertyName> */
|
|
string[] dataLine = split(cast(string)message.data[1..message.data.length],",");
|
|
|
|
/* Get the username */
|
|
string username = dataLine[0];
|
|
|
|
/* Get the proerty */
|
|
string propertyName = dataLine[1];
|
|
|
|
/* Get the user */
|
|
DConnection connection = server.findUser(username);
|
|
|
|
/* Determine if it is a valid property */
|
|
bool status = connection.isProperty(propertyName);
|
|
|
|
/* Encode the status */
|
|
reply ~= [true];
|
|
reply ~= [status];
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/* TODO: `CHAN_PROP`, `SET_PROP` */
|
|
/* If no matching built-in command was found */
|
|
else
|
|
{
|
|
/* TODO: Check plugins */
|
|
bool isPlugin = false;
|
|
|
|
/* A matching plugin was found */
|
|
if(isPlugin)
|
|
{
|
|
/* TODO: Implement me */
|
|
}
|
|
/* The command was invalid */
|
|
else
|
|
{
|
|
/* Write error message */
|
|
reply = [2];
|
|
}
|
|
}
|
|
|
|
/* Write the response */
|
|
writeSocket(tag, reply);
|
|
}
|
|
|
|
/**
|
|
* Authenticate
|
|
*
|
|
* Login as a user with the given credentials
|
|
*/
|
|
private bool authenticate(string username, string password)
|
|
{
|
|
/* TODO: Check username and password */
|
|
/* TODO: Multi-client/session support */
|
|
|
|
/* TODO: Implement me */
|
|
this.username = username;
|
|
|
|
/* TODO (Sessions): Generate a session ID for this connection */
|
|
|
|
|
|
return true;
|
|
}
|
|
|
|
private uint generateSessionID()
|
|
{
|
|
/* TODO: Basically find a number that isn't taken by matching usernames */
|
|
return 1;
|
|
}
|
|
|
|
private uint getMySessionID()
|
|
{
|
|
/* TODO: */
|
|
return 1;
|
|
}
|
|
|
|
/**
|
|
* Get member count
|
|
*
|
|
* Gets the member count of a given channel
|
|
*/
|
|
private bool getMemberCount(string channelName, ref long count)
|
|
{
|
|
/* Status of operation */
|
|
bool status;
|
|
|
|
/* The channel */
|
|
DChannel channel = server.getChannelByName(channelName);
|
|
|
|
/* Check if the channel exists */
|
|
if(channel)
|
|
{
|
|
/* Get the channel count */
|
|
count = channel.getMemberCount();
|
|
|
|
|
|
status = true;
|
|
}
|
|
/* If the channel does not exist */
|
|
else
|
|
{
|
|
status = false;
|
|
}
|
|
|
|
return status;
|
|
}
|
|
|
|
/**
|
|
* Send user a message
|
|
*
|
|
* Sends the provided user the specified message
|
|
*/
|
|
private bool sendUserMessage(string username, string message)
|
|
{
|
|
/* Find the user to send to */
|
|
DConnection user = server.findUser(username); /*TODO: Ins erver not just use it directly */
|
|
|
|
gprintln("sendUserMessage(): "~to!(string)(user));
|
|
|
|
/* If the user was found */
|
|
if(user)
|
|
{
|
|
/* The protocol data to send */
|
|
byte[] protocolData;
|
|
|
|
/* Set the sub-type (ntype=0 / channel/directmessage) */
|
|
protocolData ~= [0];
|
|
|
|
/* Set to user message (direct message, sub-sub-type) */
|
|
protocolData ~= [1];
|
|
|
|
/* Encode the recipients's length */
|
|
protocolData ~= [cast(byte)username.length];
|
|
|
|
/* Encode the username */
|
|
protocolData ~= cast(byte[])username;
|
|
|
|
/* Encode the sender's length */
|
|
protocolData ~= [cast(byte)this.username.length];
|
|
|
|
/* Encode the sender */
|
|
protocolData ~= cast(byte[])this.username;
|
|
|
|
/* Encode the message */
|
|
protocolData ~= cast(byte[])message;
|
|
|
|
/* Send the messge */
|
|
bool sendStatus = user.writeSocket(0, protocolData);
|
|
|
|
return sendStatus;
|
|
}
|
|
/* If the user was not found */
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
public string getUsername()
|
|
{
|
|
return username;
|
|
}
|
|
|
|
/**
|
|
* Set the status message
|
|
*/
|
|
private void setStatusMessage(string statusMessage)
|
|
{
|
|
/* Lock the status message mutex */
|
|
statusMessageLock.lock();
|
|
|
|
/* Set the status message */
|
|
currentStatus = statusMessage;
|
|
|
|
/* Unlock the statue message mutex */
|
|
statusMessageLock.unlock();
|
|
}
|
|
|
|
/**
|
|
* Returns the current status message
|
|
*/
|
|
public string getStatusMessage(string username)
|
|
{
|
|
/* The status message */
|
|
string statusMessage;
|
|
|
|
gprintln("hfsjkhfjsdkhfdskj");
|
|
|
|
/* Lock the status message mutex */
|
|
statusMessageLock.lock();
|
|
|
|
/* Get the status message */
|
|
statusMessage = currentStatus;
|
|
|
|
/* Unlock the statue message mutex */
|
|
statusMessageLock.unlock();
|
|
|
|
return statusMessage;
|
|
}
|
|
|
|
/**
|
|
* Sets a property for this user
|
|
*/
|
|
public void setProperty(string propertyName, string propertyValue)
|
|
{
|
|
/* Lock the properties store */
|
|
propertiesLock.lock();
|
|
|
|
/* Set the property's value */
|
|
properties[propertyName] = propertyValue;
|
|
|
|
/* Unlock the properties store */
|
|
propertiesLock.unlock();
|
|
}
|
|
|
|
/**
|
|
* Determines whether or not the given property
|
|
* exists
|
|
*/
|
|
public bool isProperty(string propertyName)
|
|
{
|
|
/**
|
|
* Whether or not the given property is a property
|
|
* of this user
|
|
*/
|
|
bool status;
|
|
|
|
/* Lock the properties store */
|
|
propertiesLock.lock();
|
|
|
|
/* Check for the property's existence */
|
|
status = cast(bool)(propertyName in properties);
|
|
|
|
/* Unlock the properties store */
|
|
propertiesLock.unlock();
|
|
|
|
return status;
|
|
}
|
|
|
|
/**
|
|
* Returns a property
|
|
*/
|
|
public string getProperty(string propertyName)
|
|
{
|
|
/* The fetched property */
|
|
string propertyValue;
|
|
|
|
/* Lock the properties store */
|
|
propertiesLock.lock();
|
|
|
|
/* Check for the property's existence */
|
|
propertyValue = properties[propertyName];
|
|
|
|
/* Unlock the properties store */
|
|
propertiesLock.unlock();
|
|
|
|
/* TODO: Error handling */
|
|
return propertyValue;
|
|
}
|
|
|
|
/**
|
|
* Returns a list of proerties
|
|
*/
|
|
public string[] getProperties()
|
|
{
|
|
/* The list of keys */
|
|
string[] propertiesString;
|
|
|
|
/* Lock the properties store */
|
|
propertiesLock.lock();
|
|
|
|
/* Check for the property's existence */
|
|
propertiesString = properties.keys();
|
|
|
|
/* Unlock the properties store */
|
|
propertiesLock.unlock();
|
|
|
|
return propertiesString;
|
|
}
|
|
|
|
/**
|
|
* Deletes a given proerty
|
|
*/
|
|
public void deleteProperty(string propertyName)
|
|
{
|
|
/* Lock the properties store */
|
|
propertiesLock.lock();
|
|
|
|
/* Remove the property */
|
|
properties.remove(propertyName);
|
|
|
|
/* Unlock the properties store */
|
|
propertiesLock.unlock();
|
|
}
|
|
|
|
public ConnectionType getConnectionType()
|
|
{
|
|
return connType;
|
|
}
|
|
|
|
public override string toString()
|
|
{
|
|
string toStr = "["~to!(string)(connType)~" (";
|
|
toStr ~= socket.remoteAddress.toString();
|
|
|
|
|
|
toStr ~= ")]: ";
|
|
|
|
|
|
if(connType == ConnectionType.CLIENT)
|
|
{
|
|
toStr = toStr ~ getUsername();
|
|
}
|
|
else
|
|
{
|
|
/* TODO Implement me */
|
|
}
|
|
|
|
return toStr;
|
|
}
|
|
}
|