foss-vuln-tracker/molerat/src/main/java/it/unitn/molerat/repos/trackers/AbstractEvidenceTracker.java

57 lines
2.1 KiB
Java

package it.unitn.molerat.repos.trackers;
import it.unitn.molerat.evidence.Changes;
import it.unitn.molerat.repos.wrappers.RepoWrapper;
import java.util.*;
public abstract class AbstractEvidenceTracker {
protected final String fixedRevision;
protected final String vulnRevision;
protected final String repoPath;
protected final RepoWrapper repoWrapper;
protected Set<Changes> changes;
protected Set<String> commits = new LinkedHashSet<>();
protected Set<String> restOfCommits = new LinkedHashSet<>();
public AbstractEvidenceTracker(RepoWrapper wrapper, String fixedRev) throws Exception {
this.fixedRevision = fixedRev;
this.repoPath = wrapper.getBasePath();
this.repoWrapper = wrapper;
Iterator<String> commitsIter = repoWrapper.getRevisionNumbers(fixedRev).iterator();
this.vulnRevision = commitsIter.next();
while (commitsIter.hasNext()) {
restOfCommits.add(commitsIter.next());
}
String initDiff = repoWrapper.doDiff(vulnRevision, fixedRevision);
this.changes = repoWrapper.inferChangesFromDiff(initDiff, vulnRevision, fixedRevision);
}
public RepoWrapper getRepoWrapper() {
return this.repoWrapper;
}
public String getFixedRevision() {
return this.fixedRevision;
}
public Set<String> getProcessedCommits() {
return this.commits;
}
// apart from non-java files, filter the files that contain "test" in ther name as well
protected Set<Changes> filterNonJavaChanges(Set<Changes> changes) {
Set<Changes> filteredChanges = new HashSet<>();
for (Changes change : changes) {
if (!change.getPath().endsWith(".java") || change.getPath().toLowerCase().contains("test")) {
filteredChanges.add(change);
}
this.repoWrapper.filterCommentsAndBlanks(change.getDeletions());
this.repoWrapper.filterCommentsAndBlanks(change.getAdditions());
}
changes.removeAll(filteredChanges);
return changes;
}
public abstract void trackEvidence() throws Exception;
}