this.buffer = buffer;
        this.accessFlags = access;
        this.localKey = generateLocalKey();
        this.remoteKey = generateRemoteKey();
        this.address = getBufferAddress(buffer);
        this.valid = true;
    }

    /**
     * Get the underlying buffer
     *
     * @return memory buffer
     * @throws IllegalStateException if memory region has been invalidated
     */

        return keyGenerator.getAndIncrement();
    }

    @Override
    protected int generateRemoteKey() {
        return keyGenerator.getAndIncrement();
    }

    @Override
    protected long getBufferAddress(ByteBuffer buffer) {
        // Return a fake address for TCP
        // In real RDMA implementations, this would be the native memory address
        return buffer.hashCode();
    }

        // In real implementation, this would get the remote key from DiSNI:
        // return memoryRegister.getRkey();
        return keyGenerator.getAndIncrement();
    }

    @Override
    protected long getBufferAddress(ByteBuffer buffer) {
        // In real implementation, this would get the native memory address:
        // For direct ByteBuffers, this could be obtained through unsafe operations
        // or DiSNI-specific methods

        this.buffer = buffer;
        this.accessFlags = access;
        this.localKey = generateLocalKey();
        this.remoteKey = generateRemoteKey();
        this.address = getBufferAddress(buffer);
        this.valid = true;
    }
    
    public ByteBuffer getBuffer() {
        if (!valid) throw new IllegalStateException("Memory region invalidated");
        return buffer;
    }