}

    @Test
    @DisplayName("readUShort correctly interprets little endian")
    void testReadUShort() {
        byte[] a = new byte[] { (byte) 0xAA, (byte) 0xBB };
        int result = NtlmMessage.readUShort(a, 0);
        assertEquals(0xBBAA, result, "UShort should be little‑endian");
    }

    @Test
    @DisplayName("readUShort throws on too short array")
    void testReadUShortOutOfBounds() {

    }

    static int readUShort(final byte[] src, final int index) {
        return src[index] & 0xff | (src[index + 1] & 0xff) << 8;
    }

    static byte[] readSecurityBuffer(final byte[] src, final int index) {
        final int length = readUShort(src, index);
        final int offset = readULong(src, index + 4);

    }

    static int readUShort(final byte[] src, final int index) {
        return src[index] & 0xff | (src[index + 1] & 0xff) << 8;
    }

    static byte[] readSecurityBuffer(final byte[] src, final int index) {
        final int length = readUShort(src, index);
        final int offset = readULong(src, index + 4);

    }

    @Test
    void testReadUShort() {
        // Test readUShort with various values
        byte[] data = new byte[4];
        ByteBuffer buffer = ByteBuffer.wrap(data).order(ByteOrder.LITTLE_ENDIAN);

        // Test with a positive value
        buffer.putShort(0, (short) 0x1234);
        assertEquals(0x1234, NtlmMessage.readUShort(data, 0), "Should read positive UShort correctly.");

        // Test with zero

                // For server: Type 0x0001 (NetBIOS Computer Name)

                // Verify domain part
                assertEquals(2, Type2Message.readUShort(targetInfo, 0)); // Type 0x0002
                assertEquals(TEST_DOMAIN.getBytes(Type2Message.UNI_ENCODING).length, Type2Message.readUShort(targetInfo, 2)); // Length
                assertEquals(TEST_DOMAIN,

    }

    val questionCount = buf.readShort().toInt() and 0xffff
    val answerCount = buf.readShort().toInt() and 0xffff
    buf.readShort() // authority record count
    buf.readShort() // additional record count

    for (i in 0 until questionCount) {
      skipName(buf) // name
      buf.readShort() // type
      buf.readShort() // class
    }

    for (i in 0 until answerCount) {

     * @return the short value in little-endian format
     * @throws IOException if an I/O error occurs
     */
    public short readShort() throws IOException {
        align(2);
        return Short.reverseBytes(this.dis.readShort());
    }

    /**
     * Reads a 32-bit integer value with proper alignment and byte order.
     * @return the integer value in little-endian format

   * Reads a {@code short} as specified by {@link DataInputStream#readShort()}, except using
   * little-endian byte order.
   *
   * @return the next two bytes of the input stream, interpreted as a {@code short} in little-endian
   *     byte order.
   * @throws IOException if an I/O error occurs.
   */
  @CanIgnoreReturnValue // to skip some bytes
  @Override
  public short readShort() throws IOException {

  @Override
  byte readByte();

  @CanIgnoreReturnValue // to skip a byte
  @Override
  int readUnsignedByte();

  @CanIgnoreReturnValue // to skip some bytes
  @Override
  short readShort();

  @CanIgnoreReturnValue // to skip some bytes
  @Override
  int readUnsignedShort();

  @CanIgnoreReturnValue // to skip some bytes
  @Override
  char readChar();

            }
          }
        }

        else -> {
          throw ProtocolException("unsupported address type: $addressType")
        }
      }

    val port = fromSource.readShort() and 0xffff

    when (command) {
      COMMAND_CONNECT -> {
        val toSocket = Socket(toAddress, port)
        val localAddress = toSocket.localAddress.address
        if (localAddress.size != 4) {