126 lines
3.8 KiB
Java
126 lines
3.8 KiB
Java
package net.i2p.router;
|
|
|
|
import java.io.*;
|
|
import java.io.File;
|
|
import java.io.FileInputStream;
|
|
import java.io.FileOutputStream;
|
|
import java.io.IOException;
|
|
import java.util.Map;
|
|
import java.util.function.*;
|
|
import java.util.zip.ZipEntry;
|
|
import java.util.zip.ZipInputStream;
|
|
import net.i2p.I2PAppContext;
|
|
import net.i2p.router.*;
|
|
import net.i2p.util.Log;
|
|
|
|
public class ZipUpdateProcess implements Runnable {
|
|
private final RouterContext ctx;
|
|
private final Supplier<String> versionSupplier;
|
|
private final Supplier<File> fileSupplier;
|
|
private final Log _log;
|
|
|
|
ZipUpdateProcess(RouterContext ctx, Supplier<String> versionSupplier,
|
|
Supplier<File> fileSupplier) {
|
|
this.ctx = ctx;
|
|
this.versionSupplier = versionSupplier;
|
|
this.fileSupplier = fileSupplier;
|
|
this._log = ctx.logManager().getLog(ZipUpdateProcess.class);
|
|
}
|
|
|
|
private File workDir() throws IOException {
|
|
if (ctx != null) {
|
|
File workDir =
|
|
new File(ctx.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 workDir;
|
|
} else {
|
|
workDir.mkdirs();
|
|
}
|
|
return workDir;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
private void unzipUpdateInstaller() throws IOException {
|
|
String version = versionSupplier.get();
|
|
File file = fileSupplier.get();
|
|
if (file == null)
|
|
return;
|
|
|
|
File workingDir = workDir();
|
|
File logFile = new File(workingDir, "log-" + version + ".txt");
|
|
|
|
// check if we can write to the log file. If we can, use the
|
|
// ProcessBuilder to run the installer.
|
|
// ProcessBuilder pb = new ProcessBuilder(
|
|
// file.getAbsolutePath(), "/S", "/D=" + workingDir.getAbsolutePath());
|
|
// Map<String, String> env = pb.environment();
|
|
// env.put("OLD_I2P_VERSION", version);
|
|
// env.remove("RESTART_I2P");
|
|
|
|
int exitCode = ctx.router().scheduledGracefulExitCode();
|
|
// if (exitCode == Router.EXIT_HARD_RESTART ||
|
|
// exitCode == Router.EXIT_GRACEFUL_RESTART)
|
|
// env.put("RESTART_I2P", "true");
|
|
|
|
/*try {
|
|
pb.directory(workingDir)
|
|
.redirectErrorStream(true)
|
|
.redirectOutput(logFile)
|
|
.start();
|
|
} catch (IOException ex) {
|
|
_log.error(
|
|
"Unable to run update-program in background. Update will fail.", ex);
|
|
}*/
|
|
}
|
|
|
|
@Override
|
|
public void run() {
|
|
try {
|
|
unzipUpdateInstaller();
|
|
} catch (IOException ioe) {
|
|
_log.error("Error running updater, update may fail." + ioe);
|
|
}
|
|
}
|
|
|
|
private static void unzip(String zipFilePath, String destDir) {
|
|
File dir = new File(destDir);
|
|
// create output directory if it doesn't exist
|
|
if (!dir.exists())
|
|
dir.mkdirs();
|
|
FileInputStream fis;
|
|
// buffer for read and write data to file
|
|
byte[] buffer = new byte[1024];
|
|
try {
|
|
fis = new FileInputStream(zipFilePath);
|
|
ZipInputStream zis = new ZipInputStream(fis);
|
|
ZipEntry ze = zis.getNextEntry();
|
|
while (ze != null) {
|
|
String fileName = ze.getName();
|
|
File newFile = new File(destDir + File.separator + fileName);
|
|
System.out.println("Unzipping to " + newFile.getAbsolutePath());
|
|
// create directories for sub directories in zip
|
|
new File(newFile.getParent()).mkdirs();
|
|
FileOutputStream fos = new FileOutputStream(newFile);
|
|
int len;
|
|
while ((len = zis.read(buffer)) > 0) {
|
|
fos.write(buffer, 0, len);
|
|
}
|
|
fos.close();
|
|
// close this ZipEntry
|
|
zis.closeEntry();
|
|
ze = zis.getNextEntry();
|
|
}
|
|
// close last ZipEntry
|
|
zis.closeEntry();
|
|
zis.close();
|
|
fis.close();
|
|
} catch (IOException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
}
|