Java tutorial
/* * Bootstrapper.java * Copyright (C) 2011,2012 Wannes De Smet * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see <http://www.gnu.org/licenses/>. */ package org.xenmaster.setup.debian; import java.io.ByteArrayInputStream; import java.io.File; import java.io.FileInputStream; import java.io.FileOutputStream; import java.io.IOException; import java.io.InputStream; import java.nio.file.FileSystems; import java.nio.file.Files; import java.nio.file.Path; import java.util.HashMap; import java.util.Map; import java.util.zip.GZIPInputStream; import net.wgr.settings.Settings; import net.wgr.utility.Network; import org.apache.commons.compress.archivers.tar.TarArchiveEntry; import org.apache.commons.compress.archivers.tar.TarArchiveInputStream; import org.apache.commons.io.IOUtils; import org.apache.commons.net.tftp.TFTPOptionReadRequestPacket; import org.apache.log4j.Logger; import org.xenmaster.connectivity.tftp.TFTPServer; import org.xenmaster.site.XenMasterSite; /** * * @created Oct 27, 2011 * @author double-u */ public class Bootstrapper { protected TFTPServer tftpd; protected HashMap<String, String> preseedValues; protected long lastChecked; public Bootstrapper() { tftpd = new TFTPServer(); preseedValues = new HashMap<>(); lastChecked = 0; } public void boot() { tftpd.addListener(new PXEListener()); tftpd.boot(); preseedValues.put("menuItemLabel", "^Install Debian"); } public void waitForServerToQuit() throws InterruptedException { tftpd.waitTillQuit(); } protected class PXEListener implements TFTPServer.ActivityListener { @Override public InputStream pathRequest(TFTPOptionReadRequestPacket packet) { try { if (downloadNetbootFiles()) return null; File path = new File( Settings.getInstance().getString("StorePath") + "/netboot/" + packet.getFilename()); if (!path.exists()) { return null; } File f = path.getAbsoluteFile(); if (f.exists()) { FileInputStream fis = new FileInputStream(f); if (f.getName().equals("txt.cfg")) { preseedValues .put("preseedUrl", "http://" + Network .getHostAddressInSubnet( packet.getAddress().getHostAddress(), "255.255.0.0") .getHostAddress() + ":" + Settings.getInstance().get("WebApplicationPort") + "/setup/xapi"); String txt = IOUtils.toString(fis); for (Map.Entry<String, String> entry : preseedValues.entrySet()) { // todo regex txt = txt.replace("#{" + entry.getKey() + "}", entry.getValue()); } return new ByteArrayInputStream(txt.getBytes("UTF-8")); } return fis; } } catch (IOException ex) { Logger.getLogger(getClass()).error("File not found", ex); } return null; } } protected boolean downloadNetbootFiles() { if (System.currentTimeMillis() - lastChecked < 50 * 60 * 1000) { return false; } File localVersionFile = new File(Settings.getInstance().getString("StorePath") + "/netboot/version"); int localVersion = -1; try (FileInputStream fis = new FileInputStream(localVersionFile)) { localVersion = Integer.parseInt(IOUtils.toString(fis)); } catch (IOException | NumberFormatException ex) { Logger.getLogger(getClass()).error("Failed to retrieve local version file", ex); } int remoteVersion = -1; try { remoteVersion = Integer.parseInt(IOUtils.toString(XenMasterSite.getFileAsStream("/netboot/version"))); } catch (IOException | NumberFormatException ex) { Logger.getLogger(getClass()).error("Failed to retrieve remote version file", ex); } lastChecked = System.currentTimeMillis(); if (localVersion < remoteVersion) { Logger.getLogger(getClass()) .info("New version " + remoteVersion + " found. Please hold while downloading netboot data"); try { TarArchiveInputStream tais = new TarArchiveInputStream( new GZIPInputStream(XenMasterSite.getFileAsStream("/netboot/netboot.tar.gz"))); TarArchiveEntry tae = null; FileOutputStream fos = null; while ((tae = tais.getNextTarEntry()) != null) { if (tais.canReadEntryData(tae)) { String target = Settings.getInstance().getString("StorePath") + "/" + tae.getName(); if (tae.isSymbolicLink()) { Path targetFile = FileSystems.getDefault().getPath(tae.getLinkName()); Path linkName = FileSystems.getDefault().getPath(target); // Link might already have been written as null file if (targetFile.toFile().exists()) { targetFile.toFile().delete(); } Files.createSymbolicLink(linkName, targetFile); Logger.getLogger(getClass()).info( "Created sym link " + linkName.toString() + " -> " + targetFile.toString()); } else if (tae.isDirectory()) { new File(target).mkdir(); Logger.getLogger(getClass()).info("Created dir " + target); } else if (tae.isFile()) { fos = new FileOutputStream(target); byte[] b = new byte[1024]; int curPos = 0; while (tais.available() > 0) { tais.read(b, curPos, 1024); fos.write(b, curPos, 1024); } fos.flush(); fos.close(); Logger.getLogger(getClass()).info("Wrote file " + target); } } } tais.close(); // Write local version IOUtils.write("" + remoteVersion, new FileOutputStream(localVersionFile)); } catch (IOException ex) { Logger.getLogger(getClass()).error("Failed to download netboot files", ex); } } else { return false; } return true; } }