validator.validateMessageSize(1024, false);
        validator.validateMessageSize(65535, true);
        validator.validateMessageSize(1048576, false); // 1MB for SMB2
    }

    @Test
    public void testMessageSizeTooSmall() throws Exception {
        assertThrows(SmbException.class, () -> {
            validator.validateMessageSize(20, false);
        });
    }

    @Test

     * Validate SMB message size
     *
     * @param size the message size
     * @param isSmb1 whether this is SMB1 protocol
     * @throws SmbException if size is invalid
     */
    public void validateMessageSize(int size, boolean isSmb1) throws SmbException {
        totalValidations.incrementAndGet();

        if (size < MIN_SMB_HEADER_SIZE) {
            failedValidations.incrementAndGet();