br.com.softplan.security.zap.api.authentication.AuthenticationScript Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of zap-client-api Show documentation

Java client API for the OWASP Zed Attack Proxy (ZAP), built on top of the official ZAP Java API

The newest version!

package br.com.softplan.security.zap.api.authentication;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;
import java.net.URISyntaxException;

import org.apache.commons.io.IOUtils;

import br.com.softplan.security.zap.commons.authentication.AuthenticationScripts;
import br.com.softplan.security.zap.commons.boot.ZapDockerBoot;

public class AuthenticationScript {

	private String name;
	private String description;
	private String fileName;
	private String relativePath;
	private String path;
	
	private File scriptTempFile;
	
	public AuthenticationScript(String name, String description) {
		this.name = name;
		this.description = description;
		this.fileName = name + AuthenticationScripts.EXTENSION;
		this.relativePath = AuthenticationScripts.RELATIVE_PATH + fileName;
		
		this.path = AuthenticationScript.class.getResource(relativePath).getPath();
	}

	public String getName() {
		return name;
	}

	public String getDescription() {
		return description;
	}

	public String getFileName() {
		return fileName;
	}

	public String getRelativePath() {
		return relativePath;
	}

	public String getPath(boolean isZapRunningOnDocker) throws IOException, URISyntaxException {
		if (isZapRunningOnDocker) {
			return ZapDockerBoot.CAS_AUTH_SCRIPT_DEFAULT_DOCKER_PATH + fileName;
		}
		if (scriptFileIsNotAccessible()) {
			return getCasAuthScriptTempFile().getAbsolutePath();
		}
		return path;
	}
	
	private boolean scriptFileIsNotAccessible() throws UnsupportedEncodingException {
		return !new File(path).exists();
	}
	
	private File getCasAuthScriptTempFile() throws IOException {
		if (scriptTempFile == null) {
			scriptTempFile = createCasAuthScriptTempFile();
		}
		return scriptTempFile;
	}
	
	private File createCasAuthScriptTempFile() throws IOException {
		File tempFile = File.createTempFile(name, AuthenticationScripts.EXTENSION);
		tempFile.deleteOnExit();
		
		InputStream casAuthScriptInputStream = AuthenticationScript.class.getResourceAsStream(relativePath);
		try (FileOutputStream fileOutputStream = new FileOutputStream(tempFile)) {
			IOUtils.copy(casAuthScriptInputStream, fileOutputStream);
		}
		return tempFile;
	}
	
}