i2p.firefox/java/net/i2p/router/WindowsUpdatePostProcessor.java

package net.i2p.router;

import java.io.*;
import java.util.*;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.*;


import net.i2p.crypto.*;
import static net.i2p.update.UpdateType.*;
import net.i2p.I2PAppContext;
import net.i2p.update.UpdateType;
import net.i2p.update.UpdatePostProcessor;
import net.i2p.util.Log;
import net.i2p.util.SystemVersion;

import java.lang.ProcessBuilder;
import java.lang.Process;
import java.lang.InterruptedException;


public class WindowsUpdatePostProcessor implements UpdatePostProcessor {
    private final Log _log = I2PAppContext.getGlobalContext().logManager().getLog(WindowsUpdatePostProcessor.class);
    private final RouterContext ctx;
    protected static Router i2pRouter = null;
    private final AtomicBoolean hook = new AtomicBoolean();
    private volatile String version;
    private volatile File positionedFile = null;
    WindowsUpdatePostProcessor() {
    	this.ctx = null;
    }

    WindowsUpdatePostProcessor(RouterContext ctx) {
        this.ctx = ctx;
    }

    public String getVersion() {
        return version;
    }
    
    public File getFile() {
        return positionedFile;
    }

    public void updateDownloadedandVerified(UpdateType type, int fileType, String version, File file) throws IOException {
        _log.info("Got an update to post-process");

        if (type != UpdateType.ROUTER_SIGNED_SU3 && type != UpdateType.ROUTER_DEV_SU3) {
            _log.warn("Unsupported update type " + type);
            return;
        }

        if (fileType != SU3File.TYPE_EXE) {
            _log.warn("Unsupported file type " + fileType);
            return;
        }
        
        try {
            this.positionedFile = moveUpdateInstaller(file);
        } catch(IOException ioe) {
            _log.error("Error positioning update installer", ioe);
            return;
        }
        this.version = version;
        
        if (!hook.compareAndSet(false,true)) {
            _log.info("shutdown hook was already set");
            return;
        }

        _log.info("adding shutdown hook");
        ctx.addFinalShutdownTask(new WinUpdateProcess(ctx, this::getVersion, this::getFile));

    }
    
    private File moveUpdateInstaller(File file) throws IOException {
        RouterContext i2pContext = i2pRouter.getContext();
        if (i2pContext != null) {
            File appDir = i2pContext.getConfigDir();
            File newFile = new File(workDir(), file.getName());
            file.renameTo(newFile);
            return newFile;
        }
        return null;
    }

    private File workDir() throws IOException{
        RouterContext i2pContext = i2pRouter.getContext();
        if (i2pContext != null) {
            File workDir = new File(i2pContext.getConfigDir().getAbsolutePath(), "i2p_update_win");
            if (workDir.exists()) {
                if (workDir.isFile())
                    throw new IOException(workDir + " exists but is a file, get it out of the way");
                    return null;
            } else {
                workDir.mkdirs();
            }
            return workDir;
        }
        return null;
    }

    protected File selectProgramFile() {
        if (SystemVersion.isWindows()) {
            File jrehome = new File(System.getProperty("java.home"));
            File programs = jrehome.getParentFile();
            System.out.println("Windows portable jpackage wrapper found, using: " + programs + " as working config");            
            return programs.getAbsoluteFile();
        } else {
            File jrehome = new File(System.getProperty("java.home"));
            File programs = new File(jrehome.getParentFile().getParentFile(), "i2p");
            System.out.println("Linux portable jpackage wrapper found, using: " + programs + " as working config");            
            return programs.getAbsoluteFile();
        }
    }

}