SMBLibrary/SMBLibrary/SMB2/Structures/EncryptionCapabilities.cs

/* Copyright (C) 2024 Tal Aloni <tal.aloni.il@gmail.com>. All rights reserved.
 * 
 * You can redistribute this program and/or modify it under the terms of
 * the GNU Lesser Public License as published by the Free Software Foundation,
 * either version 3 of the License, or (at your option) any later version.
 */
using System.Collections.Generic;
using Utilities;

namespace SMBLibrary.SMB2
{
    /// <summary>
    /// [MS-SMB2] 2.2.3.1.2 SMB2_ENCRYPTION_CAPABILITIES
    /// </summary>
    public class EncryptionCapabilities
    {
        // ushort CipherCount;
        public List<CipherAlgorithm> Ciphers = new List<CipherAlgorithm>();

        public EncryptionCapabilities()
        {
        }

        public EncryptionCapabilities(byte[] buffer, int offset)
        {
            ushort cipherCount = LittleEndianConverter.ToUInt16(buffer, offset);
            for (int index = 0; index < cipherCount; index++)
            {
                Ciphers.Add((CipherAlgorithm)LittleEndianConverter.ToUInt16(buffer, offset + index * 2));
            }
        }

        public byte[] GetBytes()
        {
            int length = 2 * Ciphers.Count * 2;
            byte[] buffer = new byte[length];
            LittleEndianWriter.WriteUInt16(buffer, 0, (ushort)Ciphers.Count);
            for (int index = 0; index < Ciphers.Count; index++)
            {
                LittleEndianWriter.WriteUInt16(buffer, 2 + index * 2, (ushort)Ciphers[index]);
            }

            return buffer;
        }
    }
}