SID_STARTADVEX2.cs

using Atlasd.Battlenet.Exceptions;
using Atlasd.Daemon;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;

namespace Atlasd.Battlenet.Protocols.Game.Messages
{
    class SID_STARTADVEX2 : Message
    {
        public enum Statuses : UInt32
        {
            Failed = 0,
            Success = 1,
        };

        public SID_STARTADVEX2()
        {
            Id = (byte)MessageIds.SID_STARTADVEX2;
            Buffer = new byte[0];
        }

        public SID_STARTADVEX2(byte[] buffer)
        {
            Id = (byte)MessageIds.SID_STARTADVEX2;
            Buffer = buffer;
        }

        public override bool Invoke(MessageContext context)
        {
            switch (context.Direction)
            {
                case MessageDirection.ClientToServer:
                    {
                        Logging.WriteLine(Logging.LogLevel.Debug, Logging.LogType.Client_Game, context.Client.RemoteEndPoint, $"[{Common.DirectionToString(context.Direction)}] {MessageName(Id)} ({4 + Buffer.Length} bytes)");

                        /**
                         * (UINT32) Status
                         * (UINT32) Unknown [editor's note: probably elapsed time]
                         * (UINT16) Game Type
                         * (UINT16) Unknown [editor's note: probably sub game type]
                         * (UINT32) Unknown [editor's note: probably viewing filter]
                         * (UINT32) Port
                         * (STRING) Game name
                         * (STRING) Game password
                         * (STRING) Game stats - Flags, Creator, Statstring
                         */

                        if (Buffer.Length < 23)
                            throw new GameProtocolViolationException(context.Client, $"{MessageName(Id)} buffer must be at least 23 bytes");

                        if (context.Client.GameState.ActiveAccount == null)
                            throw new GameProtocolViolationException(context.Client, $"{MessageName(Id)} was received before logon");

                        using var m = new MemoryStream(Buffer);
                        using var r = new BinaryReader(m);

                        var gameState = r.ReadUInt32();
                        var gameElapsedTime = r.ReadUInt32();
                        var gameType = r.ReadUInt16();
                        var subGameType = r.ReadUInt16();
                        var viewingFilter = r.ReadUInt32();
                        var portNumber = r.ReadUInt32();
                        var gameName = r.ReadByteString();
                        var gamePassword = r.ReadByteString();
                        var gameStatstring = r.ReadByteString();

                        Statuses status = Statuses.Failed;
                        GameAd gameAd = null;

                        lock (Battlenet.Common.ActiveGameAdsLock)
                        {
                            foreach (GameAd _ad in Battlenet.Common.ActiveGameAds)
                            {
                                if (_ad.Name.SequenceEqual(gameName))
                                {
                                    gameAd = _ad;
                                    break;
                                }
                            }

                            if (gameAd == null)
                            {
                                gameAd = new GameAd(context.Client.GameState, gameName, gamePassword, gameStatstring, 6112, (GameAd.GameTypes)gameType, subGameType, context.Client.GameState.Version.VersionByte);
                                gameAd.AddClient(context.Client.GameState);
                                Battlenet.Common.ActiveGameAds.Add(gameAd);
                            }
                        }

                        if (gameAd.HasClient(context.Client.GameState)) context.Client.GameState.GameAd = gameAd;

                        bool gameAdOwner = gameAd != null && gameAd.Owner == context.Client.GameState;
                        if (!gameAdOwner)
                        {
                            status = Statuses.Failed;
                        }
                        else
                        {
                            status = Statuses.Success;
                            gameAd.SetActiveStateFlags((GameAd.StateFlags)gameState);
                            gameAd.SetElapsedTime(gameElapsedTime);
                            gameAd.SetGameType((GameAd.GameTypes)gameType);
                            gameAd.SetName(gameName);
                            gameAd.SetPassword(gamePassword);
                            gameAd.SetPort(6112);
                            gameAd.SetStatstring(gameStatstring);
                        }

                        return new SID_STARTADVEX2().Invoke(new MessageContext(context.Client, MessageDirection.ServerToClient, new Dictionary<string, dynamic>(){{ "status", status }}));
                    }
                case MessageDirection.ServerToClient:
                    {
                        var status = context.Arguments.ContainsKey("status") ? (Statuses)context.Arguments["status"] : Statuses.Failed;

                        /**
                         * (UINT32) Status
                         */

                        Buffer = new byte[4];

                        using var m = new MemoryStream(Buffer);
                        using var w = new BinaryWriter(m);

                        w.Write((UInt32)status);

                        Logging.WriteLine(Logging.LogLevel.Debug, Logging.LogType.Client_Game, context.Client.RemoteEndPoint, $"[{Common.DirectionToString(context.Direction)}] {MessageName(Id)} ({4 + Buffer.Length} bytes)");
                        context.Client.Send(ToByteArray(context.Client.ProtocolType));
                        return true;
                    }
            }

            return false;
        }
    }
}