91 lines
3.1 KiB
C#
91 lines
3.1 KiB
C#
/*************************************************************************
|
|
* ModernUO *
|
|
* Copyright 2019-2020 - ModernUO Development Team *
|
|
* Email: hi@modernuo.com *
|
|
* File: ServerConnectionHandler.cs *
|
|
* *
|
|
* This program is free software: you can redistribute it and/or modify *
|
|
* it under the terms of the GNU General Public License as published by *
|
|
* the Free Software Foundation, either version 3 of the License, or *
|
|
* (at your option) any later version. *
|
|
* *
|
|
* You should have received a copy of the GNU General Public License *
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>. *
|
|
*************************************************************************/
|
|
|
|
using System;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.AspNetCore.Connections;
|
|
using Microsoft.Extensions.Logging;
|
|
|
|
namespace Server.Network
|
|
{
|
|
public class ServerConnectionHandler : ConnectionHandler
|
|
{
|
|
private readonly ILogger<ServerConnectionHandler> _logger;
|
|
private readonly IMessagePumpService _messagePumpService;
|
|
|
|
public ServerConnectionHandler(
|
|
IMessagePumpService messagePumpService,
|
|
ILogger<ServerConnectionHandler> logger
|
|
)
|
|
{
|
|
_messagePumpService = messagePumpService;
|
|
_logger = logger;
|
|
}
|
|
|
|
public override async Task OnConnectedAsync(ConnectionContext connection)
|
|
{
|
|
if (!VerifySocket(connection))
|
|
{
|
|
Release(connection);
|
|
return;
|
|
}
|
|
|
|
var ns = new NetState(connection);
|
|
TcpServer.Instances.Add(ns);
|
|
Console.WriteLine($"Client: {ns}: Connected. [{TcpServer.Instances.Count} Online]");
|
|
|
|
await ns.ProcessIncoming(_messagePumpService).ConfigureAwait(false);
|
|
}
|
|
|
|
private static bool VerifySocket(ConnectionContext connection)
|
|
{
|
|
try
|
|
{
|
|
var args = new SocketConnectEventArgs(connection);
|
|
|
|
EventSink.InvokeSocketConnect(args);
|
|
|
|
return args.AllowConnection;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
NetState.TraceException(ex);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
private static void Release(ConnectionContext connection)
|
|
{
|
|
try
|
|
{
|
|
connection.Abort(new ConnectionAbortedException("Failed socket verification."));
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
NetState.TraceException(ex);
|
|
}
|
|
|
|
try
|
|
{
|
|
// TODO: Is this needed?
|
|
connection.DisposeAsync();
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
NetState.TraceException(ex);
|
|
}
|
|
}
|
|
}
|
|
}
|