protected override void OnOpen()
{
base.OnOpen();
if (Settings.Instance.Server.EnableWebSocketRoomInfo)
{
try {
List <string> paths = WebSocketListener.Server.WebSocketServices.Paths.ToList();
paths.Remove("/");
if (Settings.Instance.Server.EnableWebSocketRCON)
{
paths.Remove("/" + Settings.Instance.Server.RCONPassword);
}
List <WSRoomInfo> rooms = paths.Where(x => x.Contains("room")).Select(x => new WSRoomInfo(x)).ToList();
List <RCONChannelInfo> channels = paths.Where(x => x.Contains("channel")).Select(x => new RCONChannelInfo(x)).ToList();
ServerHubInfo info = new ServerHubInfo()
{
Version = Assembly.GetEntryAssembly().GetName().Version.ToString(),
TotalClients = HubListener.hubClients.Count + RoomsController.GetRoomsList().Sum(x => x.roomClients.Count) + RadioController.radioChannels.Sum(x => x.radioClients.Count),
Rooms = rooms,
RadioChannels = channels
};
Send(JsonConvert.SerializeObject(info));
}catch (Exception e)
{
Logger.Instance.Warning("Unable to send ServerHub info to WebSocket client! Exception: " + e);
}
}
}
protected override void OnOpen()
{
base.OnOpen();
Logger.Instance.Log($"WebSocket Client Connected! ");
string[] split = Context.RequestUri.AbsolutePath.Split('/');
int _id = Convert.ToInt32(split[split.Length - 1]);
if (split[split.Length - 2] == "room")
{
BaseRoom _room = RoomsController.GetRoomsList().FirstOrDefault(room => room.roomId == _id);
if (_room != null)
{
_room.OnOpenWebSocket();
}
}
else if (split[split.Length - 2] == "channel")
{
RadioChannel _channel = RadioController.radioChannels.FirstOrDefault(channel => channel.channelId == _id);
if (_channel != null)
{
_channel.OnOpenWebSocket();
}
}
}
private static void PingTimerCallback(object state)
{
try
{
List <Client> allClients = hubClients.Concat(RoomsController.GetRoomsList().SelectMany(x => x.roomClients)).Concat(RadioController.radioChannels.SelectMany(x => x.radioClients)).ToList();
for (int i = 0; i < allClients.Count; i++)
{
if (allClients.Count > i && allClients[i] != null)
{
if (allClients[i].playerConnection.Status == NetConnectionStatus.Disconnected)
{
ClientDisconnected(allClients[i]);
}
}
}
List <uint> emptyRooms = RoomsController.GetRoomsList().Where(x => x.roomClients.Count == 0 && !x.noHost).Select(y => y.roomId).ToList();
if (emptyRooms.Count > 0 && !Settings.Instance.TournamentMode.Enabled)
{
Logger.Instance.Log("Destroying empty rooms...");
foreach (uint roomId in emptyRooms)
{
RoomsController.DestroyRoom(roomId);
Logger.Instance.Log($"Destroyed room {roomId}!");
}
}
}
catch (Exception e)
{
#if DEBUG
Logger.Instance.Warning("PingTimerCallback Exception: " + e);
#endif
}
}
public WSRoomInfo(string _path)
{
Path = _path;
string[] split = _path.Split('/');
int _id = Convert.ToInt32(split[split.Length - 1]);
BaseRoom _room = RoomsController.GetRoomsList().Find(room => room.roomId == _id);
RoomName = _room.roomSettings.Name;
}
public string[] GetSuggestions(string text, int index)
{
try
{
var parsedArgs = Program.ParseLine(text);
if (text.EndsWith(" "))
{
parsedArgs.Add("");
}
if (parsedArgs.Count == 0)
{
return(null);
}
else if (parsedArgs.Count == 1)
{
return(Program.availableCommands.Where(x => x.name.StartsWith(text)).Select(y => y.name).ToArray());
}
else if (parsedArgs.Count == 2)
{
switch (parsedArgs[0])
{
case "blacklist":
return(new string[] { "add", "remove" }.Where(x => x.StartsWith(parsedArgs[1]) || string.IsNullOrEmpty(parsedArgs[1])).ToArray());
case "whitelist":
return(new string[] { "enable", "disable", "add", "remove" }.Where(x => x.StartsWith(parsedArgs[1]) || string.IsNullOrEmpty(parsedArgs[1])).ToArray());
case "tickrate":
return(new string[] { "30", "60", "90" }.Where(x => x.StartsWith(parsedArgs[1]) || string.IsNullOrEmpty(parsedArgs[1])).ToArray());
case "createroom":
return(Directory.GetFiles("RoomPresets/", "*.json").Where(x => Path.GetFileName(x).StartsWith(parsedArgs[1]) || string.IsNullOrEmpty(parsedArgs[1])).Select(x => $"\"{Path.GetFileName(x)}\"").ToArray());
case "cloneroom":
case "saveroom":
case "destroyroom":
case "message":
return(RoomsController.GetRoomsList().Select(x => x.roomId.ToString()).Where(x => (x.StartsWith(parsedArgs[1]) || string.IsNullOrEmpty(parsedArgs[1]))).ToArray());
case "radio":
return(new string[] { "help", "enable", "disable", "list" }.Where(x => (x.StartsWith(parsedArgs[1]) || string.IsNullOrEmpty(parsedArgs[1])) && parsedArgs[1] != x).ToArray());
default:
return(null);
}
}
else if (parsedArgs.Count == 3)
{
if (parsedArgs[0] == "radio")
{
return(new string[] { "set", "queue" }.Where(x => (x.StartsWith(parsedArgs[2]) || string.IsNullOrEmpty(parsedArgs[2])) && parsedArgs[2] != x).ToArray());
}
else
{
return(null);
}
}
else if (parsedArgs.Count == 4)
{
if (parsedArgs[0] == "radio")
{
switch (parsedArgs[2])
{
case "set":
return(new string[] { "name", "iconurl", "difficulty" }.Where(x => x.StartsWith(parsedArgs[3]) || string.IsNullOrEmpty(parsedArgs[3])).ToArray());
case "queue":
return(new string[] { "list", "clear", "remove", "add" }.Where(x => x.StartsWith(parsedArgs[3]) || string.IsNullOrEmpty(parsedArgs[3])).ToArray());
default:
return(null);
}
}
else
{
return(null);
}
}
else
{
return(null);
}
}
catch (Exception e)
{
#if DEBUG
Logger.Instance.Log("Exception: " + e);
#endif
return(null);
}
}
private static void HubLoop(object sender, HighResolutionTimerElapsedEventArgs e)
{
if (_ticksLength.Count > 30)
{
_ticksLength.RemoveAt(0);
}
_ticksLength.Add(DateTime.UtcNow.Subtract(_lastTick).Ticks / (float)TimeSpan.TicksPerMillisecond);
_lastTick = DateTime.UtcNow;
List <RoomInfo> roomsList = RoomsController.GetRoomInfosList();
string titleBuffer = $"ServerHub v{Assembly.GetEntryAssembly().GetName().Version}: {roomsList.Count} rooms, {hubClients.Count} clients in lobby, {roomsList.Select(x => x.players).Sum() + hubClients.Count} clients total {(Settings.Instance.Server.ShowTickrateInTitle ? $", {Tickrate.ToString("0.0")} tickrate" : "")}";
if (_currentTitle != titleBuffer)
{
_currentTitle = titleBuffer;
Console.Title = _currentTitle;
}
List <Client> allClients = hubClients.Concat(RoomsController.GetRoomsList().SelectMany(x => x.roomClients)).Concat(RadioController.radioChannels.SelectMany(x => x.radioClients)).ToList();
NetIncomingMessage msg;
while ((msg = ListenerServer.ReadMessage()) != null)
{
try
{
Program.networkBytesInNow += msg.LengthBytes;
switch (msg.MessageType)
{
case NetIncomingMessageType.ConnectionApproval:
{
byte[] versionBytes = msg.PeekBytes(4);
byte[] version = msg.ReadBytes(4);
if (version[0] == 0 && version[1] == 0)
{
uint versionUint = BitConverter.ToUInt32(version, 0);
uint serverVersionUint = ((uint)Assembly.GetEntryAssembly().GetName().Version.Major).ConcatUInts((uint)Assembly.GetEntryAssembly().GetName().Version.Minor).ConcatUInts((uint)Assembly.GetEntryAssembly().GetName().Version.Build).ConcatUInts((uint)Assembly.GetEntryAssembly().GetName().Version.Revision);
msg.SenderConnection.Deny($"Version mismatch!\nServer:{serverVersionUint}\nClient:{versionUint}");
Logger.Instance.Log($"Client version v{versionUint} tried to connect");
break;
}
byte[] serverVersion = new byte[4] {
(byte)Assembly.GetEntryAssembly().GetName().Version.Major, (byte)Assembly.GetEntryAssembly().GetName().Version.Minor, (byte)Assembly.GetEntryAssembly().GetName().Version.Build, (byte)Assembly.GetEntryAssembly().GetName().Version.Revision
};
if (version[0] != serverVersion[0] || version[1] != serverVersion[1] || version[2] != serverVersion[2])
{
msg.SenderConnection.Deny($"Version mismatch|{string.Join('.', serverVersion)}|{string.Join('.', version)}");
Logger.Instance.Log($"Client version v{string.Join('.', version)} tried to connect");
break;
}
PlayerInfo playerInfo = new PlayerInfo(msg);
if (Settings.Instance.Access.WhitelistEnabled)
{
if (!IsWhitelisted(msg.SenderConnection.RemoteEndPoint, playerInfo))
{
msg.SenderConnection.Deny("You are not whitelisted on this ServerHub!");
Logger.Instance.Warning($"Client {playerInfo.playerName}({playerInfo.playerId})@{msg.SenderConnection.RemoteEndPoint.Address} is not whitelisted!");
break;
}
}
if (IsBlacklisted(msg.SenderConnection.RemoteEndPoint, playerInfo))
{
msg.SenderConnection.Deny("You are banned on this ServerHub!");
Logger.Instance.Warning($"Client {playerInfo.playerName}({playerInfo.playerId})@{msg.SenderConnection.RemoteEndPoint.Address} is banned!");
break;
}
msg.SenderConnection.Approve();
Client client = new Client(msg.SenderConnection, playerInfo);
client.playerInfo.playerState = PlayerState.Lobby;
client.ClientDisconnected += ClientDisconnected;
hubClients.Add(client);
allClients.Add(client);
Logger.Instance.Log($"{playerInfo.playerName} connected!");
};
break;
case NetIncomingMessageType.Data:
{
Client client = allClients.FirstOrDefault(x => x.playerConnection.RemoteEndPoint.Equals(msg.SenderEndPoint));
switch ((CommandType)msg.ReadByte())
{
case CommandType.Disconnect:
{
if (client != null)
{
allClients.Remove(client);
ClientDisconnected(client);
}
}
break;
case CommandType.UpdatePlayerInfo:
{
if (client != null)
{
client.playerInfo = new PlayerInfo(msg);
if (Settings.Instance.Misc.PlayerColors.ContainsKey(client.playerInfo.playerId))
{
client.playerInfo.playerNameColor = Settings.Instance.Misc.PlayerColors[client.playerInfo.playerId];
}
}
}
break;
case CommandType.UpdateVoIPData:
{
if (!Settings.Instance.Server.AllowVoiceChat)
{
return;
}
if (client != null)
{
UnityVOIP.VoipFragment data = new UnityVOIP.VoipFragment(msg);
if (data.playerId == client.playerInfo.playerId)
{
client.playerVoIPQueue.Enqueue(data);
}
}
}
break;
case CommandType.JoinRoom:
{
if (client != null)
{
uint roomId = msg.ReadUInt32();
BaseRoom room = RoomsController.GetRoomsList().FirstOrDefault(x => x.roomId == roomId);
if (room != null)
{
if (room.roomSettings.UsePassword)
{
if (RoomsController.ClientJoined(client, roomId, msg.ReadString()))
{
if (hubClients.Contains(client))
{
hubClients.Remove(client);
}
client.joinedRoomID = roomId;
}
}
else
{
if (RoomsController.ClientJoined(client, roomId, ""))
{
if (hubClients.Contains(client))
{
hubClients.Remove(client);
}
client.joinedRoomID = roomId;
}
}
}
else
{
RoomsController.ClientJoined(client, roomId, "");
}
}
}
break;
case CommandType.LeaveRoom:
{
if (client != null)
{
RoomsController.ClientLeftRoom(client);
client.joinedRoomID = 0;
client.playerInfo.playerState = PlayerState.Lobby;
if (!hubClients.Contains(client))
{
hubClients.Add(client);
}
}
}
break;
case CommandType.GetRooms:
{
NetOutgoingMessage outMsg = ListenerServer.CreateMessage();
outMsg.Write((byte)CommandType.GetRooms);
RoomsController.AddRoomListToMessage(outMsg);
msg.SenderConnection.SendMessage(outMsg, NetDeliveryMethod.ReliableOrdered, 0);
Program.networkBytesOutNow += outMsg.LengthBytes;
}
break;
case CommandType.CreateRoom:
{
if (client != null)
{
uint roomId = RoomsController.CreateRoom(new RoomSettings(msg), client.playerInfo);
NetOutgoingMessage outMsg = ListenerServer.CreateMessage(5);
outMsg.Write((byte)CommandType.CreateRoom);
outMsg.Write(roomId);
msg.SenderConnection.SendMessage(outMsg, NetDeliveryMethod.ReliableOrdered, 0);
Program.networkBytesOutNow += outMsg.LengthBytes;
}
}
break;
case CommandType.GetRoomInfo:
{
if (client != null)
{
#if DEBUG
Logger.Instance.Log("GetRoomInfo: Client room=" + client.joinedRoomID);
#endif
if (client.joinedRoomID != 0)
{
BaseRoom joinedRoom = RoomsController.GetRoomsList().FirstOrDefault(x => x.roomId == client.joinedRoomID);
if (joinedRoom != null)
{
NetOutgoingMessage outMsg = ListenerServer.CreateMessage();
outMsg.Write((byte)CommandType.GetRoomInfo);
joinedRoom.GetRoomInfo().AddToMessage(outMsg);
msg.SenderConnection.SendMessage(outMsg, NetDeliveryMethod.ReliableOrdered, 0);
Program.networkBytesOutNow += outMsg.LengthBytes;
}
}
}
}
break;
case CommandType.SetSelectedSong:
{
if (client != null && client.joinedRoomID != 0)
{
BaseRoom joinedRoom = RoomsController.GetRoomsList().FirstOrDefault(x => x.roomId == client.joinedRoomID);
if (joinedRoom != null)
{
if (msg.LengthBytes < 16)
{
joinedRoom.SetSelectedSong(client.playerInfo, null);
}
else
{
joinedRoom.SetSelectedSong(client.playerInfo, new SongInfo(msg));
}
}
}
}
break;
case CommandType.SetLevelOptions:
{
if (client != null && client.joinedRoomID != 0)
{
BaseRoom joinedRoom = RoomsController.GetRoomsList().FirstOrDefault(x => x.roomId == client.joinedRoomID);
if (joinedRoom != null)
{
joinedRoom.SetLevelOptions(client.playerInfo, new StartLevelInfo(msg));
}
}
}
break;
case CommandType.StartLevel:
{
#if DEBUG
Logger.Instance.Log("Received command StartLevel");
#endif
if (client != null && client.joinedRoomID != 0)
{
BaseRoom joinedRoom = RoomsController.GetRoomsList().FirstOrDefault(x => x.roomId == client.joinedRoomID);
if (joinedRoom != null)
{
StartLevelInfo options = new StartLevelInfo(msg);
SongInfo song = new SongInfo(msg);
song.songDuration += 2.5f;
joinedRoom.StartLevel(client.playerInfo, options, song);
}
}
}
break;
case CommandType.DestroyRoom:
{
if (client != null && client.joinedRoomID != 0)
{
BaseRoom joinedRoom = RoomsController.GetRoomsList().FirstOrDefault(x => x.roomId == client.joinedRoomID);
if (joinedRoom != null)
{
joinedRoom.DestroyRoom(client.playerInfo);
}
}
}
break;
case CommandType.TransferHost:
{
if (client != null && client.joinedRoomID != 0)
{
BaseRoom joinedRoom = RoomsController.GetRoomsList().FirstOrDefault(x => x.roomId == client.joinedRoomID);
if (joinedRoom != null)
{
joinedRoom.TransferHost(client.playerInfo, new PlayerInfo(msg));
}
}
}
break;
case CommandType.PlayerReady:
{
if (client != null && client.joinedRoomID != 0)
{
BaseRoom joinedRoom = RoomsController.GetRoomsList().FirstOrDefault(x => x.roomId == client.joinedRoomID);
if (joinedRoom != null)
{
joinedRoom.ReadyStateChanged(client.playerInfo, msg.ReadBoolean());
}
}
}
break;
case CommandType.SendEventMessage:
{
if (client != null && client.joinedRoomID != 0 && Settings.Instance.Server.AllowEventMessages)
{
BaseRoom joinedRoom = RoomsController.GetRoomsList().FirstOrDefault(x => x.roomId == client.joinedRoomID);
if (joinedRoom != null)
{
string header = msg.ReadString();
string data = msg.ReadString();
joinedRoom.BroadcastEventMessage(header, data, new List <Client>()
{
client
});
joinedRoom.BroadcastWebSocket(CommandType.SendEventMessage, new EventMessage(header, data));
EventMessageReceived?.Invoke(client, header, data);
#if DEBUG
Logger.Instance.Log($"Received event message! Header=\"{header}\", Data=\"{data}\"");
#endif
}
}
}
break;
case CommandType.GetChannelInfo:
{
if (Settings.Instance.Radio.EnableRadio && RadioController.radioStarted)
{
int channelId = msg.ReadInt32();
NetOutgoingMessage outMsg = ListenerServer.CreateMessage();
outMsg.Write((byte)CommandType.GetChannelInfo);
if (RadioController.radioChannels.Count > channelId)
{
RadioController.radioChannels[channelId].channelInfo.AddToMessage(outMsg);
}
else
{
new ChannelInfo()
{
channelId = -1, currentSong = new SongInfo()
{
levelId = "FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF"
}
}.AddToMessage(outMsg);
}
msg.SenderConnection.SendMessage(outMsg, NetDeliveryMethod.ReliableOrdered, 0);
Program.networkBytesOutNow += outMsg.LengthBytes;
}
}
break;
case CommandType.JoinChannel:
{
int channelId = msg.ReadInt32();
NetOutgoingMessage outMsg = ListenerServer.CreateMessage();
outMsg.Write((byte)CommandType.JoinChannel);
if (RadioController.ClientJoinedChannel(client, channelId))
{
outMsg.Write((byte)0);
hubClients.Remove(client);
}
else
{
outMsg.Write((byte)1);
}
msg.SenderConnection.SendMessage(outMsg, NetDeliveryMethod.ReliableOrdered, 0);
Program.networkBytesOutNow += outMsg.LengthBytes;
}
break;
case CommandType.GetSongDuration:
{
foreach (RadioChannel channel in RadioController.radioChannels)
{
if (channel.radioClients.Contains(client) && channel.requestingSongDuration)
{
SongInfo info = new SongInfo(msg);
if (info.levelId == channel.channelInfo.currentSong.levelId)
{
channel.songDurationResponses.TryAdd(client, info.songDuration);
}
}
}
}
break;
case CommandType.LeaveChannel:
{
if (RadioController.radioStarted && client != null)
{
RadioController.ClientLeftChannel(client);
}
}; break;
}
};
break;
case NetIncomingMessageType.WarningMessage:
Logger.Instance.Warning(msg.ReadString());
break;
case NetIncomingMessageType.ErrorMessage:
Logger.Instance.Error(msg.ReadString());
break;
case NetIncomingMessageType.StatusChanged:
{
NetConnectionStatus status = (NetConnectionStatus)msg.ReadByte();
Client client = allClients.FirstOrDefault(x => x.playerConnection.RemoteEndPoint.Equals(msg.SenderEndPoint));
if (client != null)
{
if (status == NetConnectionStatus.Disconnected)
{
allClients.Remove(client);
ClientDisconnected(client);
}
}
}
break;
#if DEBUG
case NetIncomingMessageType.VerboseDebugMessage:
case NetIncomingMessageType.DebugMessage:
Logger.Instance.Log(msg.ReadString());
break;
default:
Logger.Instance.Log("Unhandled message type: " + msg.MessageType);
break;
#endif
}
}catch (Exception ex)
{
Logger.Instance.Log($"Exception on message received: {ex}");
}
ListenerServer.Recycle(msg);
}
}