All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.hp.octane.integrations.services.vulnerabilities.fod.FODServiceImpl Maven / Gradle / Ivy

There is a newer version: 2.24.3.5
Show newest version
/**
 * Copyright 2017-2023 Open Text
 *
 * The only warranties for products and services of Open Text and
 * its affiliates and licensors (“Open Text”) are as may be set forth
 * in the express warranty statements accompanying such products and services.
 * Nothing herein should be construed as constituting an additional warranty.
 * Open Text shall not be liable for technical or editorial errors or
 * omissions contained herein. The information contained herein is subject
 * to change without notice.
 *
 * Except as specifically indicated otherwise, this document contains
 * confidential information and a valid license is required for possession,
 * use or copying. If this work is provided to the U.S. Government,
 * consistent with FAR 12.211 and 12.212, Commercial Computer Software,
 * Computer Software Documentation, and Technical Data for Commercial Items are
 * licensed to the U.S. Government under vendor's standard commercial license.
 *
 * 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 com.hp.octane.integrations.services.vulnerabilities.fod;

import com.hp.octane.integrations.OctaneSDK;
import com.hp.octane.integrations.dto.securityscans.OctaneIssue;
import com.hp.octane.integrations.exceptions.PermanentException;
import com.hp.octane.integrations.exceptions.TemporaryException;
import com.hp.octane.integrations.services.rest.RestService;
import com.hp.octane.integrations.services.vulnerabilities.ExistingIssuesInOctane;
import com.hp.octane.integrations.services.vulnerabilities.PackIssuesToOctaneUtils;
import com.hp.octane.integrations.services.vulnerabilities.VulnerabilitiesQueueItem;
import com.hp.octane.integrations.services.vulnerabilities.fod.dto.pojos.Scan;
import com.hp.octane.integrations.services.vulnerabilities.fod.dto.pojos.Vulnerability;
import com.hp.octane.integrations.services.vulnerabilities.fod.dto.pojos.VulnerabilityAllData;
import com.hp.octane.integrations.services.vulnerabilities.fod.dto.services.FODReleaseService;
import com.hp.octane.integrations.services.vulnerabilities.fod.dto.services.FODVulnerabilityService;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.*;
import java.util.stream.Collectors;

import static com.hp.octane.integrations.services.vulnerabilities.IssuesFileSerializer.*;
import static com.hp.octane.integrations.services.vulnerabilities.fod.FODValuesConverter.sameDay;
import static com.hp.octane.integrations.utils.CIPluginSDKUtils.doWait;

public class FODServiceImpl implements FODService {

    private static final Logger logger = LogManager.getLogger(FODServiceImpl.class);
    protected final OctaneSDK.SDKServicesConfigurer configurer;
    protected final RestService restService;

    public FODServiceImpl(OctaneSDK.SDKServicesConfigurer configurer, RestService restService) {
        if (restService == null) {
            throw new IllegalArgumentException("rest service MUST NOT be null");
        }
        if (configurer == null) {
            throw new IllegalArgumentException("configurer service MUST NOT be null");
        }
        this.configurer = configurer;
        this.restService = restService;
    }

    @Override
    public InputStream getVulnerabilitiesScanResultStream(VulnerabilitiesQueueItem queueItem) throws IOException {

        logger.debug(configurer.octaneConfiguration.getLocationForLog() + "Entered getVulnerabilitiesScanResultStream");
        String targetDir = getTargetDir(configurer.pluginServices.getAllowedOctaneStorage(),
                queueItem.getJobId(), queueItem.getBuildId());
        logger.debug(configurer.octaneConfiguration.getLocationForLog() + "getVulnerabilitiesScanResultStream target Dir:" + targetDir);
        InputStream cachedScanResult = getCachedScanResult(targetDir);
        if (cachedScanResult != null) {
            logger.warn("results " + queueItem.toString() + "are cached!");
            return cachedScanResult;
        }
        enrichItemWithFODParams(queueItem);
        PplnRunStatus pplnRunStatus = fodScanIsStillInProgress(queueItem);
        if (pplnRunStatus.continuePolling) {
            return null;
        } else if (pplnRunStatus.tryGetIssues) {
            List octaneIssues = fetchIssues(queueItem, getRelease(queueItem).toString());
            cacheIssues(targetDir, octaneIssues);
            return serializeIssues(octaneIssues);

        } else {
            throw new PermanentException(queueItem.getJobId() + "#" + queueItem.getBuildId() +
                    " , Polling is stopped");
        }
    }

    @Override
    public boolean vulnerabilitiesQueueItemCleanUp(VulnerabilitiesQueueItem vulnerabilitiesQueueItem) {
        String runRootDir = getTargetDir(configurer.pluginServices.getAllowedOctaneStorage(),
                vulnerabilitiesQueueItem.getJobId(),
                vulnerabilitiesQueueItem.getBuildId());
        if (runRootDir == null) {
            return false;
        }
        File directoryToBeDeleted = new File(runRootDir);
        return deleteDirectory(directoryToBeDeleted);
    }

    @Override
    public RestService getRestService() {
        return this.restService;
    }

    @Override
    public OctaneSDK.SDKServicesConfigurer getConfigurer() {
        return this.configurer;
    }

    private PplnRunStatus fodScanIsStillInProgress(VulnerabilitiesQueueItem queueItem) {

        logger.debug(configurer.octaneConfiguration.getLocationForLog() + "Check if scan is in progress." + queueItem.getJobId() + "#" + queueItem.getJobId());

        Long pplRunStartTime = queueItem.getStartTime();

        //It is done , but it's scan still active.
        if (getScan(queueItem) == null) {
            logger.debug(configurer.octaneConfiguration.getLocationForLog() + "need to retrieve the scan Id");
            List scans = FODReleaseService.getScansLastInFirstFetched(getRelease(queueItem), pplRunStartTime);
            setScanIdForItem(queueItem, scans);
            if (getScan(queueItem) == null) {
                incFailedTries(queueItem);
                logger.warn(configurer.octaneConfiguration.getLocationForLog() + "scan Id was not found yet");
            }
        } else {
            logger.debug(configurer.octaneConfiguration.getLocationForLog() + "scanId is already retrieved from previous polling:" + getScan(queueItem));
        }
        if (getScan(queueItem) != null) {
            Long release = getRelease(queueItem);
            Long scan = getScan(queueItem);
            if (scanIsCompleted(release, scan)) {
                return new PplnRunStatus(false, true);
            }
        }
        if (getFailedTries(queueItem) > 30) {
            logger.error(
                    "scan Id was not found, validate that the release in the pipeline configuration is the same as the release in the Jenkins job.");
            return new PplnRunStatus(false, false);
        }
        return new PplnRunStatus(true, false);
    }

    private void incFailedTries(VulnerabilitiesQueueItem queueItem) {
        String failedTriesToGetScanStr = queueItem.getAdditionalProperties().get("failedTriesToGetScan");
        Integer failedTriesToGetScan = 1;
        if (failedTriesToGetScanStr != null) {
            failedTriesToGetScan = Integer.parseInt(failedTriesToGetScanStr) + 1;
        }
        queueItem.getAdditionalProperties().put("failedTriesToGetScan", failedTriesToGetScan.toString());
    }

    private int getFailedTries(VulnerabilitiesQueueItem queueItem) {
        String failedTriesToGetScanStr = queueItem.getAdditionalProperties().get("failedTriesToGetScan");
        return failedTriesToGetScanStr == null ? 0 : Integer.parseInt(failedTriesToGetScanStr);
    }

    private void enrichItemWithFODParams(VulnerabilitiesQueueItem queueItem) {
        if (queueItem.getAdditionalProperties() == null) {
            queueItem.setAdditionalProperties(new HashMap<>());
        }
        String releaseId = queueItem.getAdditionalProperties().get("releaseId");
        if (releaseId == null) {
            releaseId = this.configurer.pluginServices.getFodRelease(queueItem.getJobId(), queueItem.getBuildId()).toString();
            queueItem.getAdditionalProperties().put("releaseId", releaseId);
        }
        //logger.warn(configurer.octaneConfiguration.getLocationForLog() + "FOD ReleaseId:" + releaseId);
    }

    Long getRelease(VulnerabilitiesQueueItem item) {
        return Long.valueOf(item.getAdditionalProperties().get("releaseId"));

    }

    Long getScan(VulnerabilitiesQueueItem item) {
        return !item.getAdditionalProperties().containsKey("scanId") ? null :
                Long.valueOf(item.getAdditionalProperties().get("scanId"));
    }

    private List fetchIssues(VulnerabilitiesQueueItem queueItem, String remoteTag) throws IOException {

        logger.warn(configurer.octaneConfiguration.getLocationForLog() + "Security scan is done.");
        List allVulnerabilities =
                FODVulnerabilityService.getAllVulnerabilities(getRelease(queueItem));
        List nonClosedIssues = filterOutBeforeBaselineIssues(queueItem.getBaselineDate(), allVulnerabilities);
        ExistingIssuesInOctane existingIssuesInOctane = new ExistingIssuesInOctane(this.restService.obtainOctaneRestClient(),
                this.configurer.octaneConfiguration);
        List existingIssuesInOc = existingIssuesInOctane.getRemoteIdsOpenVulnsFromOctane(queueItem.getJobId(),
                queueItem.getBuildId(), remoteTag);


        PackIssuesToOctaneUtils.SortedIssues sortedIssues =
                PackIssuesToOctaneUtils.packToOctaneIssues(nonClosedIssues, existingIssuesInOc, true);

        FODValuesConverter securityIssueValuesHelper = new FODValuesConverter();
        securityIssueValuesHelper.init();
        Map idToAllData = getVulnerabilityAllDataMap(getRelease(queueItem),
                sortedIssues.issuesRequiredExtendedData);

        logger.warn(configurer.octaneConfiguration.getLocationForLog() + "fetch existing issues from Octane");
        List octaneIssuesToUpdate =
                securityIssueValuesHelper.createOctaneIssuesFromVulns(sortedIssues.issuesToUpdate, remoteTag, idToAllData,
                        queueItem.getBaselineDate());

        List total = new ArrayList<>();
        total.addAll(octaneIssuesToUpdate);
        logger.warn(configurer.octaneConfiguration.getLocationForLog() + "ToUpdate " + octaneIssuesToUpdate.size() + " items : " + octaneIssuesToUpdate);
        total.addAll(sortedIssues.issuesToClose);
        logger.warn(configurer.octaneConfiguration.getLocationForLog() + "ToClose " + sortedIssues.issuesToClose.size() + " items : " + sortedIssues.issuesToClose);
        return total;
    }

    private Map getVulnerabilityAllDataMap(Long releaseId, List requiredExtendedData) {
        long SLEEP_MS = 2000;
        long EXPECTED_REQUEST_TIME_MS = 1000;//used to compute waiting time
        logger.warn(configurer.octaneConfiguration.getLocationForLog() + String.format("getVulnerabilityAllDataMap, requiredExtendedData.size=%s, expected processing duration is %s sec",
                requiredExtendedData.size(), requiredExtendedData.size() * (SLEEP_MS + EXPECTED_REQUEST_TIME_MS) / 1000));
        Map idToAllData = new HashMap<>();

        for (int i = 0; i < requiredExtendedData.size(); i++) {
            Vulnerability t = requiredExtendedData.get(i);
            if (i > 0) {
                //adding sleep in order to avoid FOD DOS filter that might block our requests
                doWait(SLEEP_MS);
            }
            if (i > 0 && i % 50 == 0) {
                logger.warn(configurer.octaneConfiguration.getLocationForLog() + String.format("getVulnerabilityAllDataMap, %s/%s is done, remaining processing duration is %s sec",
                        i, requiredExtendedData.size(), (requiredExtendedData.size() - i) * (SLEEP_MS + EXPECTED_REQUEST_TIME_MS) / 1000));
            }

            idToAllData.put(t.id, FODVulnerabilityService.getSingleVulnAlldata(releaseId, t.vulnId));
        }
        return idToAllData;
    }

    private List filterOutBeforeBaselineIssues(Date baseline,
                                                              List allVulnerabilities) {

        return allVulnerabilities.stream().filter(
                t -> {
                    Date date = FODValuesConverter.dateOfDateString(t.introducedDate);
                    return date.after(baseline) || sameDay(date, baseline);
                })
                .collect(Collectors.toList());

    }

    private boolean scanIsCompleted(Long releaseId, Long scanId) {
        try {
            Scan completeScan = FODReleaseService.getCompleteScan(releaseId, scanId);
            if (completeScan == null) {
                return false;
            }

            logger.debug(configurer.octaneConfiguration.getLocationForLog() + "scan:" + scanId + " is:" + completeScan.status);

            if (completeScan.status == null) {
                return false;
            }
            //Scan that has not started, and not in progress is completed.
            return (!Scan.IN_PROGRESS.equals(completeScan.status) &&
                    !Scan.NOT_STARTED.equals(completeScan.status) &&
                    !Scan.QUEUED.equals(completeScan.status));
        } catch (PermanentException e){
            throw e;
        } catch (TemporaryException e){
            throw e;
        } catch (Exception e) {
            return false;
        }
    }

    private void setScanIdForItem(VulnerabilitiesQueueItem queueItem, List scans) {
        Long relevantScanId = getRelevantScan(scans, queueItem);
        logger.debug(configurer.octaneConfiguration.getLocationForLog() + "scan Id is retrieved:" + relevantScanId);
        if (relevantScanId != null) {
            queueItem.getAdditionalProperties().put("scanId", relevantScanId.toString());
        }

    }

    private Long getRelevantScan(List scans, VulnerabilitiesQueueItem queueItem) {

        Scan scanByNotes = getScanByNotes(scans, queueItem);
        if (scanByNotes != null) {
            return scanByNotes.scanId;
        }
        for (Scan scan : scans) {
            if (Scan.IN_PROGRESS.equals(scan.status)) {
                return scan.scanId;
            }
        }
        return null;
    }

    private Scan getScanByNotes(List scans,
                                VulnerabilitiesQueueItem queueItem) {
        //   "notes": "[80] #80 - Assessment submitted from Jenkins FoD Plugin".
        for (Scan scan : scans) {
            if (scan.notes != null && scan.notes.contains("[" + queueItem.getBuildId() + "]")) {
                return scan;
            }
        }
        return null;
    }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy