OVFManagerExportVAPP

OVFManagerExportVAPP.java

/*
 * ****************************************************************************
 * Copyright VMware, Inc. 2010-2016.  All Rights Reserved.
 * ****************************************************************************
 *
 * This software is made available for use under the terms of the BSD
 * 3-Clause license:
 *
 * Redistribution and use in source and binary forms, with or without 
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright 
 *    notice, this list of conditions and the following disclaimer.
 * 
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in 
 *    the documentation and/or other materials provided with the 
 *    distribution.
 * 
 * 3. Neither the name of the copyright holder nor the names of its 
 *    contributors may be used to endorse or promote products derived
 *    from this software without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 
 * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
 * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
 * USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */


package com.vmware.vapp;

import com.vmware.common.annotations.Action;
import com.vmware.common.annotations.Option;
import com.vmware.common.annotations.Sample;
import com.vmware.connection.ConnectedVimServiceBase;
import com.vmware.vim25.*;

import javax.net.ssl.HttpsURLConnection;
import javax.xml.ws.BindingProvider;
import javax.xml.ws.handler.MessageContext;
import java.io.*;
import java.net.URL;
import java.util.*;

/**
 * <pre>
 * OVFManagerExportVAPP
 *
 * This sample demonstrates OVFManager.
 * Exports VMDK's and OVF Descriptor of all VM's in the vApps
 *
 * <b>Parameters:</b>
 * username         [required]: username for the authentication
 * password         [required]: password for the authentication
 * host             [required]: Name of the host system
 * vapp             [required]: Name of the vapp
 * localpath        [required]: local System Folder path
 *
 * <b>Command Line:</b>
 * run.bat com.vmware.httpfileaccess.OVFManagerExportVAPP
 * --url [URLString] --username [username] --password [password]
 * --host [Host name] --vapp [Vapp Name] --localpath [Local Path]
 * </pre>
 */
@Sample(
        name = "ovf-manager-export-vapp",
        description = "This sample demonstrates OVFManager.\n" +
                "Exports VMDK's and OVF Descriptor of all VM's in the vApps"
)
public class OVFManagerExportVAPP extends ConnectedVimServiceBase {
    private ManagedObjectReference propCollectorRef;

    private volatile long TOTAL_BYTES = 0;
    private volatile long TOTAL_BYTES_WRITTEN = 0;
    private volatile HttpNfcLeaseExtender leaseExtender;
    private volatile boolean vmdkFlag;
    private String cookieValue;

    String host;

    String vApp = null;
    String localpath = null;

    @Option(name = "vapp", description = "Name of the vapp")
    public void setvApp(String vApp) {
        this.vApp = vApp;
    }

    @Option(name = "localpath", description = "local System Folder path")
    public void setLocalpath(String localpath) {
        this.localpath = localpath;
    }

    @Option(name = "host", description = "Name of the host")
    public void setHost(String host) {
        this.host = host;
    }

    /**
     * The Class HttpNfcLeaseExtender.
     */
    private class HttpNfcLeaseExtender implements Runnable {
        private ManagedObjectReference httpNfcLease = null;

        public HttpNfcLeaseExtender(ManagedObjectReference mor,
                                    VimPortType VIM_PORT) {
            httpNfcLease = mor;
            vimPort = VIM_PORT;
        }

        @Override
        public void run() {
            try {
                System.out.println("---------------------- Thread for "
                        + "Checking the HTTP NFCLEASE vmdkFlag: " + vmdkFlag
                        + "----------------------");
                while (!vmdkFlag) {
                    System.out.println("#### TOTAL_BYTES_WRITTEN "
                            + TOTAL_BYTES_WRITTEN);
                    System.out.println("#### TOTAL_BYTES " + TOTAL_BYTES);
                    try {
                        vimPort.httpNfcLeaseProgress(httpNfcLease, 0);
                        Thread.sleep(290000000);
                    } catch (InterruptedException e) {
                        System.out
                                .println("---------------------- Thread interrupted "
                                        + "----------------------");
                        break;
                    }
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

    void printHttpNfcLeaseInfo(HttpNfcLeaseInfo info,
                               String hostString) {
        System.out
                .println("########################################################");
        System.out.println("HttpNfcLeaseInfo");
        System.out.println("Lease Timeout: " + info.getLeaseTimeout());
        System.out.println("Total Disk capacity: "
                + info.getTotalDiskCapacityInKB());
        List<HttpNfcLeaseDeviceUrl> deviceUrlArr = info.getDeviceUrl();
        if (deviceUrlArr != null) {
            int deviceUrlCount = 1;
            for (HttpNfcLeaseDeviceUrl durl : deviceUrlArr) {
                System.out.println("HttpNfcLeaseDeviceUrl : " + deviceUrlCount++);
                System.out.println("   Device URL Import Key: "
                        + durl.getImportKey());
                System.out.println("   Device URL Key: " + durl.getKey());
                System.out.println("   Device URL : " + durl.getUrl());
                System.out.println("   Updated device URL: "
                        + durl.getUrl().replace("*", hostString));
                System.out
                        .println("   SSL Thumbprint : " + durl.getSslThumbprint());
            }
        } else {
            System.out.println("No Device URLS Found");
            System.out
                    .println("########################################################");
        }
    }

    long writeVMDKFile(String absoluteFile, String urlString) throws IOException {
        URL urlCon = new URL(urlString);
        HttpsURLConnection conn = (HttpsURLConnection) urlCon.openConnection();

        conn.setDoInput(true);
        conn.setDoOutput(true);
        conn.setAllowUserInteraction(true);

        // Maintain session
        List<String> cookies = (List<String>) headers.get("Set-cookie");
        cookieValue = cookies.get(0);
        StringTokenizer tokenizer = new StringTokenizer(cookieValue, ";");
        cookieValue = tokenizer.nextToken();
        String path = "$" + tokenizer.nextToken();
        String cookie = "$Version=\"1\"; " + cookieValue + "; " + path;

        // set the cookie in the new request header
        Map<String, List<String>> map = new HashMap<String, List<String>>();
        map.put("Cookie", Collections.singletonList(cookie));
        ((BindingProvider) vimPort).getRequestContext().put(
                MessageContext.HTTP_REQUEST_HEADERS, map);

        conn.setRequestProperty("Cookie", cookie);
        conn.setRequestProperty("Content-Type", "application/octet-stream");
        conn.setRequestProperty("Expect", "100-continue");
        conn.setRequestMethod("GET");
        conn.setRequestProperty("Content-Length", "1024");

        InputStream in = conn.getInputStream();
        String _localpath = localpath + "/" + absoluteFile;
        OutputStream out = new FileOutputStream(new File(_localpath));
        byte[] buf = new byte[102400];
        int len = 0;
        long written = 0;
        while ((len = in.read(buf)) > 0) {
            out.write(buf, 0, len);
            written = written + len;
        }
        System.out.println("   Exported File " + absoluteFile + " : " + written);
        in.close();
        out.close();
        return written;
    }

    void exportVApp() throws RuntimeFaultFaultMsg, TaskInProgressFaultMsg, FileFaultFaultMsg, InvalidStateFaultMsg, InvalidPowerStateFaultMsg, InvalidPropertyFaultMsg, InvalidCollectorVersionFaultMsg, IOException, TimedoutFaultMsg, VmConfigFaultFaultMsg, ConcurrentAccessFaultMsg {
        File file = new File(localpath);
        if (!file.exists()) {
            System.out.println("Wrong or invalid path " + localpath);
            return;
        }
        Map<String, ManagedObjectReference> results = getMOREFs.inFolderByType(serviceContent
                .getRootFolder(), "HostSystem", new RetrieveOptions());
        ManagedObjectReference hostRef = results.get(host);
        if (hostRef == null) {
            System.out.println("Host Not Found");
        } else {
            Map<String, ManagedObjectReference> vAppList = getMOREFs.inFolderByType(serviceContent
                    .getRootFolder(), "VirtualApp", new RetrieveOptions());
            ManagedObjectReference vAppMoRef = vAppList.get(vApp);
            if (vAppMoRef != null) {
                OvfCreateDescriptorParams ovfCreateDescriptorParams =
                        new OvfCreateDescriptorParams();
                ManagedObjectReference httpNfcLease =
                        vimPort.exportVApp(vAppMoRef);
                System.out.println("Getting the HTTP NFCLEASE for the vApp: "
                        + vApp);

                Object[] result =
                        waitForValues.wait(httpNfcLease, new String[]{"state"},
                                new String[]{"state"},
                                new Object[][]{new Object[]{
                                        HttpNfcLeaseState.READY,
                                        HttpNfcLeaseState.ERROR}});
                if (result[0].equals(HttpNfcLeaseState.READY)) {

                    HttpNfcLeaseInfo httpNfcLeaseInfo =
                            (HttpNfcLeaseInfo) getMOREFs.entityProps(httpNfcLease,
                                    new String[]{"info"}).get("info");

                    httpNfcLeaseInfo.setLeaseTimeout(300000000);
                    printHttpNfcLeaseInfo(httpNfcLeaseInfo, host);
                    long diskCapacity =
                            (httpNfcLeaseInfo.getTotalDiskCapacityInKB()) * 1024;
                    System.out.println("************ " + diskCapacity);

                    TOTAL_BYTES = diskCapacity;
                    leaseExtender =
                            new OVFManagerExportVAPP().new HttpNfcLeaseExtender(
                                    httpNfcLease, vimPort);
                    Thread t = new Thread(leaseExtender);
                    t.start();

                    List<HttpNfcLeaseDeviceUrl> deviceUrlArr =
                            httpNfcLeaseInfo.getDeviceUrl();
                    if (deviceUrlArr != null) {
                        List<OvfFile> ovfFiles = new ArrayList<OvfFile>();
                        for (int i = 0; i < deviceUrlArr.size(); i++) {
                            System.out.println("Downloading Files:");
                            String deviceId = deviceUrlArr.get(i).getKey();
                            String deviceUrlStr = deviceUrlArr.get(i).getUrl();
                            String absoluteFile =
                                    deviceUrlStr.substring(deviceUrlStr
                                            .lastIndexOf("/") + 1);
                            System.out.println("   Absolute File Name: "
                                    + absoluteFile);
                            System.out.println("   VMDK URL: "
                                    + deviceUrlStr.replace("*", host));
                            long writtenSize =
                                    writeVMDKFile(absoluteFile,
                                            deviceUrlStr.replace("*", host));
                            OvfFile ovfFile = new OvfFile();
                            ovfFile.setPath(absoluteFile);
                            ovfFile.setDeviceId(deviceId);
                            ovfFile.setSize(writtenSize);
                            ovfFiles.add(ovfFile);
                        }
                        ovfCreateDescriptorParams.getOvfFiles().addAll(ovfFiles);
                        OvfCreateDescriptorResult ovfCreateDescriptorResult =
                                vimPort.createDescriptor(
                                        serviceContent.getOvfManager(), vAppMoRef,
                                        ovfCreateDescriptorParams);
                        System.out.println();
                        String outOVF = localpath + "/" + vApp + ".ovf";
                        File outFile = new File(outOVF);
                        FileWriter out = new FileWriter(outFile);
                        out.write(ovfCreateDescriptorResult.getOvfDescriptor());
                        out.close();
                        System.out.println("OVF Desriptor Written to file " + vApp
                                + ".ovf");
                        System.out.println("DONE");
                        if (!ovfCreateDescriptorResult.getError().isEmpty()) {
                            System.out.println("SOME ERRORS");
                        }
                        if (!ovfCreateDescriptorResult.getWarning().isEmpty()) {
                            System.out.println("SOME WARNINGS");
                        }
                    } else {
                        System.out.println("No Device URLS");
                    }
                    System.out.println("Completed Downloading the files");
                    vmdkFlag = true;
                    t.interrupt();
                    vimPort.httpNfcLeaseProgress(httpNfcLease, 100);
                    vimPort.httpNfcLeaseComplete(httpNfcLease);
                } else {
                    System.out.println("HttpNfcLeaseState not ready");
                    System.out.println("HttpNfcLeaseState: " + result);
                }
            } else {
                System.out.println("vApp Not Found");
            }
        }
    }

    @Action
    public void run() throws RuntimeFaultFaultMsg, TaskInProgressFaultMsg, VmConfigFaultFaultMsg, IOException, InvalidPropertyFaultMsg, FileFaultFaultMsg, ConcurrentAccessFaultMsg, InvalidStateFaultMsg, InvalidCollectorVersionFaultMsg, InvalidPowerStateFaultMsg, TimedoutFaultMsg {
        exportVApp();
    }

}