Java tutorial
// Licensed to the Apache Software Foundation (ASF) under one // or more contributor license agreements. See the NOTICE file // distributed with this work for additional information // regarding copyright ownership. The ASF licenses this file // to you under the Apache License, Version 2.0 (the // "License"); you may not use this file except in compliance // with the License. You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, // software distributed under the License is distributed on an // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY // KIND, either express or implied. See the License for the // specific language governing permissions and limitations // under the License. package com.cloud.hypervisor.kvm.resource; import java.io.File; import java.io.IOException; import java.io.StringReader; import java.util.ArrayList; import java.util.Collections; import java.util.List; import javax.xml.parsers.DocumentBuilder; import javax.xml.parsers.DocumentBuilderFactory; import javax.xml.parsers.ParserConfigurationException; import org.apache.commons.lang.StringUtils; import org.apache.log4j.Logger; import org.w3c.dom.Document; import org.w3c.dom.Element; import org.w3c.dom.Node; import org.w3c.dom.NodeList; import org.xml.sax.InputSource; import org.xml.sax.SAXException; import com.google.common.base.Strings; import com.cloud.hypervisor.kvm.resource.LibvirtVMDef.ChannelDef; import com.cloud.hypervisor.kvm.resource.LibvirtVMDef.DiskDef; import com.cloud.hypervisor.kvm.resource.LibvirtVMDef.InterfaceDef; import com.cloud.hypervisor.kvm.resource.LibvirtVMDef.InterfaceDef.NicModel; import com.cloud.hypervisor.kvm.resource.LibvirtVMDef.RngDef; import com.cloud.hypervisor.kvm.resource.LibvirtVMDef.RngDef.RngBackendModel; import com.cloud.hypervisor.kvm.resource.LibvirtVMDef.WatchDogDef; import com.cloud.hypervisor.kvm.resource.LibvirtVMDef.WatchDogDef.WatchDogModel; import com.cloud.hypervisor.kvm.resource.LibvirtVMDef.WatchDogDef.WatchDogAction; public class LibvirtDomainXMLParser { private static final Logger s_logger = Logger.getLogger(LibvirtDomainXMLParser.class); private final List<InterfaceDef> interfaces = new ArrayList<InterfaceDef>(); private final List<DiskDef> diskDefs = new ArrayList<DiskDef>(); private final List<RngDef> rngDefs = new ArrayList<RngDef>(); private final List<ChannelDef> channels = new ArrayList<ChannelDef>(); private final List<WatchDogDef> watchDogDefs = new ArrayList<WatchDogDef>(); private Integer vncPort; private String desc; public boolean parseDomainXML(String domXML) { DocumentBuilder builder; try { builder = DocumentBuilderFactory.newInstance().newDocumentBuilder(); InputSource is = new InputSource(); is.setCharacterStream(new StringReader(domXML)); Document doc = builder.parse(is); Element rootElement = doc.getDocumentElement(); desc = getTagValue("description", rootElement); Element devices = (Element) rootElement.getElementsByTagName("devices").item(0); NodeList disks = devices.getElementsByTagName("disk"); for (int i = 0; i < disks.getLength(); i++) { Element disk = (Element) disks.item(i); String type = disk.getAttribute("type"); DiskDef def = new DiskDef(); if (type.equalsIgnoreCase("network")) { String diskFmtType = getAttrValue("driver", "type", disk); String diskCacheMode = getAttrValue("driver", "cache", disk); String diskPath = getAttrValue("source", "name", disk); String protocol = getAttrValue("source", "protocol", disk); String authUserName = getAttrValue("auth", "username", disk); String poolUuid = getAttrValue("secret", "uuid", disk); String host = getAttrValue("host", "name", disk); int port = Integer.parseInt(getAttrValue("host", "port", disk)); String diskLabel = getAttrValue("target", "dev", disk); String bus = getAttrValue("target", "bus", disk); DiskDef.DiskFmtType fmt = null; if (diskFmtType != null) { fmt = DiskDef.DiskFmtType.valueOf(diskFmtType.toUpperCase()); } def.defNetworkBasedDisk(diskPath, host, port, authUserName, poolUuid, diskLabel, DiskDef.DiskBus.valueOf(bus.toUpperCase()), DiskDef.DiskProtocol.valueOf(protocol.toUpperCase()), fmt); def.setCacheMode(DiskDef.DiskCacheMode.valueOf(diskCacheMode.toUpperCase())); } else { String diskFmtType = getAttrValue("driver", "type", disk); String diskCacheMode = getAttrValue("driver", "cache", disk); String diskFile = getAttrValue("source", "file", disk); String diskDev = getAttrValue("source", "dev", disk); String diskLabel = getAttrValue("target", "dev", disk); String bus = getAttrValue("target", "bus", disk); String device = disk.getAttribute("device"); if (type.equalsIgnoreCase("file")) { if (device.equalsIgnoreCase("disk")) { DiskDef.DiskFmtType fmt = null; if (diskFmtType != null) { fmt = DiskDef.DiskFmtType.valueOf(diskFmtType.toUpperCase()); } def.defFileBasedDisk(diskFile, diskLabel, DiskDef.DiskBus.valueOf(bus.toUpperCase()), fmt); } else if (device.equalsIgnoreCase("cdrom")) { def.defISODisk(diskFile); } } else if (type.equalsIgnoreCase("block")) { def.defBlockBasedDisk(diskDev, diskLabel, DiskDef.DiskBus.valueOf(bus.toUpperCase())); } if (diskCacheMode != null) { def.setCacheMode(DiskDef.DiskCacheMode.valueOf(diskCacheMode.toUpperCase())); } } NodeList iotune = disk.getElementsByTagName("iotune"); if ((iotune != null) && (iotune.getLength() != 0)) { String bytesReadRateStr = getTagValue("read_bytes_sec", (Element) iotune.item(0)); if (bytesReadRateStr != null) { Long bytesReadRate = Long.parseLong(bytesReadRateStr); def.setBytesReadRate(bytesReadRate); } String bytesWriteRateStr = getTagValue("write_bytes_sec", (Element) iotune.item(0)); if (bytesWriteRateStr != null) { Long bytesWriteRate = Long.parseLong(bytesWriteRateStr); def.setBytesWriteRate(bytesWriteRate); } String iopsReadRateStr = getTagValue("read_iops_sec", (Element) iotune.item(0)); if (iopsReadRateStr != null) { Long iopsReadRate = Long.parseLong(iopsReadRateStr); def.setIopsReadRate(iopsReadRate); } String iopsWriteRateStr = getTagValue("write_iops_sec", (Element) iotune.item(0)); if (iopsWriteRateStr != null) { Long iopsWriteRate = Long.parseLong(iopsWriteRateStr); def.setIopsWriteRate(iopsWriteRate); } } diskDefs.add(def); } NodeList nics = devices.getElementsByTagName("interface"); for (int i = 0; i < nics.getLength(); i++) { Element nic = (Element) nics.item(i); String type = nic.getAttribute("type"); String mac = getAttrValue("mac", "address", nic); String dev = getAttrValue("target", "dev", nic); String model = getAttrValue("model", "type", nic); String slot = StringUtils.removeStart(getAttrValue("address", "slot", nic), "0x"); InterfaceDef def = new InterfaceDef(); NodeList bandwidth = nic.getElementsByTagName("bandwidth"); Integer networkRateKBps = 0; if ((bandwidth != null) && (bandwidth.getLength() != 0)) { Integer inbound = Integer .valueOf(getAttrValue("inbound", "average", (Element) bandwidth.item(0))); Integer outbound = Integer .valueOf(getAttrValue("outbound", "average", (Element) bandwidth.item(0))); if (inbound.equals(outbound)) { networkRateKBps = inbound; } } if (type.equalsIgnoreCase("network")) { String network = getAttrValue("source", "network", nic); def.defPrivateNet(network, dev, mac, NicModel.valueOf(model.toUpperCase()), networkRateKBps); } else if (type.equalsIgnoreCase("bridge")) { String bridge = getAttrValue("source", "bridge", nic); def.defBridgeNet(bridge, dev, mac, NicModel.valueOf(model.toUpperCase()), networkRateKBps); } else if (type.equalsIgnoreCase("ethernet")) { String scriptPath = getAttrValue("script", "path", nic); def.defEthernet(dev, mac, NicModel.valueOf(model.toUpperCase()), scriptPath, networkRateKBps); } if (StringUtils.isNotBlank(slot)) { def.setSlot(Integer.parseInt(slot, 16)); } interfaces.add(def); } NodeList ports = devices.getElementsByTagName("channel"); for (int i = 0; i < ports.getLength(); i++) { Element channel = (Element) ports.item(i); String type = channel.getAttribute("type"); String path = getAttrValue("source", "path", channel); String name = getAttrValue("target", "name", channel); String state = getAttrValue("target", "state", channel); ChannelDef def = null; if (!StringUtils.isNotBlank(state)) { def = new ChannelDef(name, ChannelDef.ChannelType.valueOf(type.toUpperCase()), new File(path)); } else { def = new ChannelDef(name, ChannelDef.ChannelType.valueOf(type.toUpperCase()), ChannelDef.ChannelState.valueOf(state.toUpperCase()), new File(path)); } channels.add(def); } Element graphic = (Element) devices.getElementsByTagName("graphics").item(0); if (graphic != null) { String port = graphic.getAttribute("port"); if (port != null) { try { vncPort = Integer.parseInt(port); if (vncPort != -1) { vncPort = vncPort - 5900; } else { vncPort = null; } } catch (NumberFormatException nfe) { vncPort = null; } } } NodeList rngs = devices.getElementsByTagName("rng"); for (int i = 0; i < rngs.getLength(); i++) { RngDef def = null; Element rng = (Element) rngs.item(i); String backendModel = getAttrValue("backend", "model", rng); String path = getTagValue("backend", rng); String bytes = getAttrValue("rate", "bytes", rng); String period = getAttrValue("rate", "period", rng); if (Strings.isNullOrEmpty(backendModel)) { def = new RngDef(path, Integer.parseInt(bytes), Integer.parseInt(period)); } else { def = new RngDef(path, RngBackendModel.valueOf(backendModel.toUpperCase()), Integer.parseInt(bytes), Integer.parseInt(period)); } rngDefs.add(def); } NodeList watchDogs = devices.getElementsByTagName("watchdog"); for (int i = 0; i < watchDogs.getLength(); i++) { WatchDogDef def = null; Element watchDog = (Element) watchDogs.item(i); String action = watchDog.getAttribute("action"); String model = watchDog.getAttribute("model"); if (Strings.isNullOrEmpty(model)) { continue; } if (Strings.isNullOrEmpty(action)) { def = new WatchDogDef(WatchDogModel.valueOf(model.toUpperCase())); } else { def = new WatchDogDef(WatchDogAction.valueOf(action.toUpperCase()), WatchDogModel.valueOf(model.toUpperCase())); } watchDogDefs.add(def); } return true; } catch (ParserConfigurationException e) { s_logger.debug(e.toString()); } catch (SAXException e) { s_logger.debug(e.toString()); } catch (IOException e) { s_logger.debug(e.toString()); } return false; } private static String getTagValue(String tag, Element eElement) { NodeList tagNodeList = eElement.getElementsByTagName(tag); if (tagNodeList == null || tagNodeList.getLength() == 0) { return null; } NodeList nlList = tagNodeList.item(0).getChildNodes(); Node nValue = nlList.item(0); return nValue.getNodeValue(); } private static String getAttrValue(String tag, String attr, Element eElement) { NodeList tagNode = eElement.getElementsByTagName(tag); if (tagNode.getLength() == 0) { return null; } Element node = (Element) tagNode.item(0); return node.getAttribute(attr); } public Integer getVncPort() { return vncPort; } public List<InterfaceDef> getInterfaces() { return interfaces; } public List<DiskDef> getDisks() { return diskDefs; } public List<RngDef> getRngs() { return rngDefs; } public List<ChannelDef> getChannels() { return Collections.unmodifiableList(channels); } public List<WatchDogDef> getWatchDogs() { return watchDogDefs; } public String getDescription() { return desc; } }