org.zaproxy.libs.DownloadTools.java Source code

Introduction

Here is the source code for org.zaproxy.libs.DownloadTools.java

Source

/*
 * Zed Attack Proxy (ZAP) and its related class files.
 *
 * ZAP is an HTTP/HTTPS proxy for assessing web application security.
 *
 * Copyright 2017 The ZAP Development Team
 *
 * Licensed 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 org.zaproxy.libs;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.net.URL;
import java.util.zip.GZIPInputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;
import org.apache.commons.compress.archivers.ArchiveEntry;
import org.apache.commons.compress.archivers.tar.TarArchiveInputStream;

public class DownloadTools {

    public static void downloadDriver(String urlStr, String destDir, String destFile) {
        File dest = new File(destDir + destFile);
        if (dest.exists()) {
            System.out.println("Already exists: " + dest.getAbsolutePath());
            return;
        }
        File parent = dest.getParentFile();
        if (!parent.exists() && !parent.mkdirs()) {
            System.out.println("Failed to create directory : " + dest.getParentFile().getAbsolutePath());
        }
        byte[] buffer = new byte[1024];
        if (urlStr.endsWith(".zip")) {
            try {
                URL url = new URL(urlStr);
                ZipInputStream zipIn = new ZipInputStream(url.openStream());
                ZipEntry entry;

                boolean isFound = false;
                while ((entry = zipIn.getNextEntry()) != null) {
                    if (destFile.equals(entry.getName())) {
                        isFound = true;
                        FileOutputStream out = new FileOutputStream(dest);
                        int read = 0;
                        while ((read = zipIn.read(buffer)) != -1) {
                            out.write(buffer, 0, read);
                        }
                        out.close();
                        System.out.println("Updated: " + dest.getAbsolutePath());

                    } else {
                        System.out.println("Found " + entry.getName());
                    }
                    zipIn.closeEntry();
                    entry = zipIn.getNextEntry();
                }

                zipIn.close();

                if (!isFound) {
                    System.out.println("Failed to find " + destFile);
                    System.exit(1);
                }

            } catch (IOException ex) {
                ex.printStackTrace();
                System.exit(1);
            }
        } else if (urlStr.endsWith(".tar.gz")) {
            try {
                URL url = new URL(urlStr);
                GZIPInputStream gzis = new GZIPInputStream(url.openStream());

                File tarFile = new File(dest.getAbsolutePath() + ".tar");
                FileOutputStream out = new FileOutputStream(tarFile);

                int len;
                while ((len = gzis.read(buffer)) > 0) {
                    out.write(buffer, 0, len);
                }

                gzis.close();
                out.close();

                TarArchiveInputStream tar = new TarArchiveInputStream(new FileInputStream(tarFile));
                ArchiveEntry entry;
                boolean isFound = false;
                while ((entry = tar.getNextEntry()) != null) {
                    if (destFile.equals(entry.getName())) {
                        out = new FileOutputStream(dest);
                        isFound = true;

                        int read = 0;
                        while ((read = tar.read(buffer)) != -1) {
                            out.write(buffer, 0, read);
                        }
                        out.close();
                        System.out.println("Updated: " + dest.getAbsolutePath());
                    }
                }
                tar.close();
                tarFile.delete();
                if (!isFound) {
                    System.out.println("Failed to find " + destFile);
                    System.exit(1);
                }

            } catch (IOException ex) {
                ex.printStackTrace();
                System.exit(1);
            }
        }
    }
}