if (!connected) {
            throw new IOException("Witness client not connected");
        }

        try {
            log.debug("Registering witness for share: {}", request.getShareName());

            // Create and populate the RPC message
            WitnessRegisterMessage message = new WitnessRegisterMessage();
            // Convert int version to WitnessVersion enum

        // Check if notification is relevant to this registration
        String resourceName = notification.getResourceName();
        String shareName = registration.getShareName();

        // Match by share name or server address, safely handling nulls
        boolean shareMatch = resourceName != null && shareName != null && resourceName.equalsIgnoreCase(shareName);

        assertNotNull(registration);
        assertEquals(WitnessRegistrationState.REGISTERED, registration.getState());
        assertEquals("\\\\server\\share", registration.getShareName());
        assertEquals(serverAddress, registration.getServerAddress());
        assertEquals(WitnessServiceType.FILE_SERVER_WITNESS, registration.getServiceType());

        return System.currentTimeMillis() - lastHeartbeat > timeoutMs;
    }
    
    // Getters and setters...
    public String getRegistrationId() { return registrationId; }
    public String getShareName() { return shareName; }
    public InetAddress getServerAddress() { return serverAddress; }
    public WitnessServiceType getServiceType() { return serviceType; }
    public WitnessRegistrationState getState() { return state; }