Skip to content

Commit 123f9b0

Browse files
committed
Added no progressbar flag
1 parent dbe8cac commit 123f9b0

File tree

3 files changed

+20
-12
lines changed

3 files changed

+20
-12
lines changed

src/main/java/de/rub/nds/tlsscanner/MultiThreadedScanJobExecutor.java

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,6 @@
2323
import java.util.concurrent.Executors;
2424
import java.util.concurrent.Future;
2525
import me.tongfei.progressbar.ProgressBar;
26-
import me.tongfei.progressbar.ProgressBarBuilder;
27-
import me.tongfei.progressbar.ProgressBarStyle;
2826
import org.apache.logging.log4j.LogManager;
2927
import org.apache.logging.log4j.Logger;
3028

@@ -47,8 +45,8 @@ public MultiThreadedScanJobExecutor(ExecutorService executor) {
4745
}
4846

4947
public SiteReport execute(ScannerConfig config, ScanJob scanJob) {
50-
51-
if(config.getGeneralDelegate().isDebug()){
48+
49+
if (config.getGeneralDelegate().isDebug() || config.isNoProgressbar()) {
5250
return scan(config, scanJob, null);
5351
} else {
5452
int numberOfProbes = 0;
@@ -67,10 +65,10 @@ public SiteReport execute(ScannerConfig config, ScanJob scanJob) {
6765
}
6866
}
6967
}
70-
68+
7169
private SiteReport scan(ScannerConfig config, ScanJob scanJob, ProgressBar pb) {
7270
List<ProbeType> probeTypes = new LinkedList<>();
73-
if(pb != null){
71+
if (pb != null) {
7472
pb.setExtraMessage("Executing Probes");
7573
}
7674
List<Future<ProbeResult>> futureResults = new LinkedList<>();
@@ -118,7 +116,7 @@ private SiteReport scan(ScannerConfig config, ScanJob scanJob, ProgressBar pb) {
118116
ProbeResult result = probe.getNotExecutedResult();
119117
if (result != null) {
120118
resultList.add(result);
121-
if(pb != null){
119+
if (pb != null) {
122120
pb.step();
123121
}
124122
}
@@ -149,8 +147,8 @@ private SiteReport scan(ScannerConfig config, ScanJob scanJob, ProgressBar pb) {
149147
LOGGER.info("Finished scan for: " + hostname);
150148
return report;
151149
}
152-
153-
private void checkProbesDone(List<Future<ProbeResult>> futureResults, ProgressBar pb){
150+
151+
private void checkProbesDone(List<Future<ProbeResult>> futureResults, ProgressBar pb) {
154152
boolean isNotReady = true;
155153
int done = 0;
156154
int tempDone = 0;
@@ -161,7 +159,7 @@ private void checkProbesDone(List<Future<ProbeResult>> futureResults, ProgressBa
161159
tempDone++;
162160
}
163161
if (done < tempDone) {
164-
if(pb != null){
162+
if (pb != null) {
165163
pb.step();
166164
}
167165
done = tempDone;
@@ -171,7 +169,7 @@ private void checkProbesDone(List<Future<ProbeResult>> futureResults, ProgressBa
171169
}
172170
}
173171
}
174-
}
172+
}
175173

176174
@Override
177175
public void shutdown() {

src/main/java/de/rub/nds/tlsscanner/SingleThreadedScanJobExecutor.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ private String getPaddedProbeName( String probeName){
4343
@Override
4444
public SiteReport execute(ScannerConfig config, ScanJob scanJob) {
4545

46-
if(config.getGeneralDelegate().isDebug()){
46+
if(config.getGeneralDelegate().isDebug() || config.isNoProgressbar()){
4747
return scan(config, scanJob, null);
4848
} else {
4949
int numberOfProbes = 0;

src/main/java/de/rub/nds/tlsscanner/config/ScannerConfig.java

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,8 @@ public class ScannerConfig extends TLSDelegateConfig {
5252
@Parameter(names = "-aggressiv", required = false, description = "The level of concurrent handshakes (only applies to some resource intensive tests)")
5353
private int aggroLevel = 1;
5454

55+
private boolean noProgressbar = false;
56+
5557
@ParametersDelegate
5658
private StarttlsDelegate starttlsDelegate;
5759

@@ -65,6 +67,14 @@ public ScannerConfig(GeneralDelegate delegate) {
6567
addDelegate(starttlsDelegate);
6668
}
6769

70+
public boolean isNoProgressbar() {
71+
return noProgressbar;
72+
}
73+
74+
public void setNoProgressbar(boolean noProgressbar) {
75+
this.noProgressbar = noProgressbar;
76+
}
77+
6878
public int getAggroLevel() {
6979
return aggroLevel;
7080
}

0 commit comments

Comments
 (0)