Query - Fixed issues, added to server config
This commit is contained in:
parent
92c5b305dd
commit
64cfe80c98
@ -92,7 +92,7 @@ namespace TrueCraft
|
||||
CraftingRepository = craftingRepository;
|
||||
PendingBlockUpdates = new Queue<BlockUpdate>();
|
||||
EnableClientLogging = false;
|
||||
QueryProtocol = new TrueCraft.QueryProtocol(25566);
|
||||
QueryProtocol = new TrueCraft.QueryProtocol(this);
|
||||
|
||||
AccessConfiguration = Configuration.LoadConfiguration<AccessConfiguration>("access.yaml");
|
||||
|
||||
@ -115,14 +115,16 @@ namespace TrueCraft
|
||||
Log(LogCategory.Notice, "Running TrueCraft server on {0}", EndPoint);
|
||||
NetworkWorker.Start();
|
||||
EnvironmentWorker.Change(100, 1000 / 20);
|
||||
QueryProtocol.Start();
|
||||
if(Program.ServerConfiguration.Query)
|
||||
QueryProtocol.Start();
|
||||
}
|
||||
|
||||
public void Stop()
|
||||
{
|
||||
ShuttingDown = true;
|
||||
Listener.Stop();
|
||||
QueryProtocol.Stop();
|
||||
if(Program.ServerConfiguration.Query)
|
||||
QueryProtocol.Stop();
|
||||
foreach (var w in Worlds)
|
||||
w.Save();
|
||||
foreach (var c in Clients)
|
||||
|
@ -6,6 +6,7 @@ using System.Net;
|
||||
using System.Net.Sockets;
|
||||
using System.IO;
|
||||
using System.Threading;
|
||||
using TrueCraft.API.Server;
|
||||
|
||||
namespace TrueCraft
|
||||
{
|
||||
@ -15,22 +16,24 @@ namespace TrueCraft
|
||||
private int Port;
|
||||
private Timer Timer;
|
||||
private Random Rnd;
|
||||
private IMultiplayerServer Server;
|
||||
private CancellationTokenSource CToken;
|
||||
|
||||
private readonly Tuple<byte, byte> ProtocolVersion = new Tuple<byte, byte>(0xFE, 0xFD);
|
||||
private readonly byte[] ProtocolVersion = new byte[] { 0xFE, 0xFD };
|
||||
private readonly byte Type_Handshake = 0x09;
|
||||
private readonly byte Type_Stat = 0x00;
|
||||
|
||||
private Dictionary<IPEndPoint, QueryUser> UserList;
|
||||
private object UserLock = new object();
|
||||
|
||||
public QueryProtocol(int port)
|
||||
public QueryProtocol(IMultiplayerServer server)
|
||||
{
|
||||
Port = port;
|
||||
Rnd = new Random();
|
||||
Server = server;
|
||||
}
|
||||
public void Start()
|
||||
{
|
||||
Port = Program.ServerConfiguration.QueryPort;
|
||||
Udp = new UdpClient(Port);
|
||||
Timer = new Timer(ResetUserList, null, 0, 30000);
|
||||
CToken = new CancellationTokenSource();
|
||||
@ -41,16 +44,25 @@ namespace TrueCraft
|
||||
{
|
||||
if (CToken.IsCancellationRequested) return;
|
||||
|
||||
var clientEP = new IPEndPoint(IPAddress.Any, Port);
|
||||
byte[] buffer = Udp.EndReceive(ar, ref clientEP);
|
||||
|
||||
switch (buffer.Length)
|
||||
try
|
||||
{
|
||||
case 7: HandleHandshake(buffer, clientEP); break;
|
||||
case 11: HandleBasicStat(buffer, clientEP); break;
|
||||
case 15: HandleFullStat(buffer, clientEP); break;
|
||||
}
|
||||
var clientEP = new IPEndPoint(IPAddress.Any, Port);
|
||||
byte[] buffer = Udp.EndReceive(ar, ref clientEP);
|
||||
|
||||
if (CheckVersion(buffer))
|
||||
{
|
||||
if (buffer[2] == Type_Handshake)
|
||||
HandleHandshake(buffer, clientEP);
|
||||
else if (buffer[2] == Type_Stat)
|
||||
{
|
||||
if (buffer.Length == 11)
|
||||
HandleBasicStat(buffer, clientEP);
|
||||
else if (buffer.Length == 15)
|
||||
HandleFullStat(buffer, clientEP);
|
||||
}
|
||||
}
|
||||
}
|
||||
catch { }
|
||||
if (CToken.IsCancellationRequested) return;
|
||||
|
||||
Udp.BeginReceive(HandleReceive, null);
|
||||
@ -59,7 +71,6 @@ namespace TrueCraft
|
||||
private void HandleHandshake(byte[] buffer, IPEndPoint clientEP)
|
||||
{
|
||||
var stream = GetStream(buffer);
|
||||
CheckHead(Type_Handshake, stream);
|
||||
int sessionId = GetSessionId(stream);
|
||||
|
||||
var user = new QueryUser { SessionId = sessionId, ChallengeToken = Rnd.Next() };
|
||||
@ -80,7 +91,6 @@ namespace TrueCraft
|
||||
private void HandleBasicStat(byte[] buffer, IPEndPoint clientEP)
|
||||
{
|
||||
var stream = GetStream(buffer);
|
||||
CheckHead(Type_Stat, stream);
|
||||
int sessionId = GetSessionId(stream);
|
||||
int token = GetToken(stream);
|
||||
|
||||
@ -105,7 +115,6 @@ namespace TrueCraft
|
||||
private void HandleFullStat(byte[] buffer, IPEndPoint clientEP)
|
||||
{
|
||||
var stream = GetStream(buffer);
|
||||
CheckHead(Type_Stat, stream);
|
||||
int sessionId = GetSessionId(stream);
|
||||
int token = GetToken(stream);
|
||||
|
||||
@ -131,22 +140,9 @@ namespace TrueCraft
|
||||
SendResponse(response, clientEP);
|
||||
}
|
||||
|
||||
private void CheckVersion(BinaryReader stream)
|
||||
private bool CheckVersion(byte[] ver)
|
||||
{
|
||||
byte ver1 = stream.ReadByte();
|
||||
byte ver2 = stream.ReadByte();
|
||||
if (ver1 != ProtocolVersion.Item1 || ver2 != ProtocolVersion.Item2)
|
||||
throw new Exception("Incorrect Protocol Version");
|
||||
}
|
||||
private void CheckType(byte Type, BinaryReader stream)
|
||||
{
|
||||
byte type = stream.ReadByte();
|
||||
if (type != Type) throw new Exception("Incorrect Type");
|
||||
}
|
||||
private void CheckHead(byte Type, BinaryReader stream)
|
||||
{
|
||||
CheckVersion(stream);
|
||||
CheckType(Type, stream);
|
||||
return ver[0] == ProtocolVersion[0] && ver[1] == ProtocolVersion[1];
|
||||
}
|
||||
private int GetSessionId(BinaryReader stream)
|
||||
{
|
||||
@ -187,8 +183,8 @@ namespace TrueCraft
|
||||
stats.Add("game_id", "TRUECRAFT");
|
||||
stats.Add("version", "1.0");
|
||||
stats.Add("plugins", "");
|
||||
stats.Add("map", Program.Server.Worlds.First().Name);
|
||||
stats.Add("numplayers", Program.Server.Clients.Count.ToString());
|
||||
stats.Add("map", Server.Worlds.First().Name);
|
||||
stats.Add("numplayers", Server.Clients.Count.ToString());
|
||||
stats.Add("maxplayers", "64");
|
||||
stats.Add("hostport", Program.ServerConfiguration.ServerPort.ToString());
|
||||
stats.Add("hostip", Program.ServerConfiguration.ServerAddress);
|
||||
@ -198,7 +194,7 @@ namespace TrueCraft
|
||||
{
|
||||
var names = new List<string>();
|
||||
lock (Program.Server.ClientLock)
|
||||
foreach (var client in Program.Server.Clients)
|
||||
foreach (var client in Server.Clients)
|
||||
names.Add(client.Username);
|
||||
return names;
|
||||
}
|
||||
|
@ -28,6 +28,8 @@ namespace TrueCraft
|
||||
ServerAddress = "0.0.0.0";
|
||||
WorldSaveInterval = 30;
|
||||
Singleplayer = false;
|
||||
Query = true;
|
||||
QueryPort = 25566;
|
||||
}
|
||||
|
||||
[YamlMember(Alias = "motd")]
|
||||
@ -47,5 +49,11 @@ namespace TrueCraft
|
||||
|
||||
[YamlIgnore]
|
||||
public bool Singleplayer { get; set; }
|
||||
|
||||
[YamlMember(Alias = "query")]
|
||||
public bool Query { get; set; }
|
||||
|
||||
[YamlMember(Alias = "queryPort")]
|
||||
public int QueryPort { get; set; }
|
||||
}
|
||||
}
|
Loading…
x
Reference in New Issue
Block a user