com.mle.rmi.RmiServer.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of util-rmi_2.11 Show documentation

util-rmi

The newest version!

package com.mle.rmi

import com.mle.util.Log
import java.rmi.server.UnicastRemoteObject
import java.io.Closeable
import java.rmi.NoSuchObjectException
import com.mle.security.{MultiKeyStoreManager, IKeystoreSettings}

/**
 * Create a keystore with: keytool -genkey -alias rmi -keyalg RSA -validity 9999 -keystore keystore.key
 *
 * @author Mle
 */
class RmiServer(registryPort: Int = RmiRegistry.DEFAULT_PORT,
                keySettings: IKeystoreSettings = RmiUtil.keySettings)
  extends Closeable with Log {
  private val sslContext = MultiKeyStoreManager.newSslContext(keySettings)
  val remoteObject = new RmiImpl(this)
  val stub = toStub(remoteObject)
  val registry = RmiRegistry.init(registryPort, sslContext)(classOf[RmiInterface].getSimpleName -> stub)
  registry.bind(remoteObject.getClass.getSimpleName, remoteObject)
  // Wait for registry daemon thread to start
  Thread sleep 100

  private def toStub(impl: RmiInterface): RmiInterface = UnicastRemoteObject.exportObject(
    impl,
    0,
    new PickyClientSocketFactory,
    new PickyServerSocketFactory(context = sslContext)
  ).asInstanceOf[RmiInterface]

  def close() {
    UnicastRemoteObject.unexportObject(registry, true)
    try {
      UnicastRemoteObject.unexportObject(remoteObject, true)
    } catch {
      case e: NoSuchObjectException => log warn("Attempted to unexport object that wasn't exported", e)
    }
    registry.list().foreach(registry.unbind)
    onClosed()
    log info "The RMI server has shut down"
  }

  def onClosed() {}
}

object RmiServer {
  def start(keySettings: IKeystoreSettings) = {
    RmiUtil.initSecurityPolicy()
    new RmiServer(keySettings = keySettings)
  }
}