Skip to content

Commit d9fde58

Browse files
committed
Added fallback for images missing the build flags label
1 parent 0c241d4 commit d9fde58

File tree

1 file changed

+32
-12
lines changed

1 file changed

+32
-12
lines changed

src/main/java/de/rub/nds/tls/subject/docker/DockerTlsManagerFactory.java

Lines changed: 32 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
import de.rub.nds.tls.subject.TlsImplementationType;
1818
import de.rub.nds.tls.subject.constants.TlsImageLabels;
1919
import de.rub.nds.tls.subject.constants.TransportType;
20+
import de.rub.nds.tls.subject.docker.build.DockerBuilder;
2021
import de.rub.nds.tls.subject.exceptions.DefaultProfileNotFoundException;
2122
import de.rub.nds.tls.subject.exceptions.PropertyNotFoundException;
2223
import de.rub.nds.tls.subject.exceptions.TlsVersionNotFoundException;
@@ -416,24 +417,43 @@ public static Image getMatchingImage(
416417
return images.stream()
417418
.filter(
418419
image ->
419-
image.getLabels()
420-
.get(TlsImageLabels.VERSION.getLabelName())
421-
.equals(version))
420+
version.equals(
421+
image.getLabels()
422+
.get(TlsImageLabels.VERSION.getLabelName())))
422423
.filter(
423424
image ->
424-
image.getLabels()
425-
.get(TlsImageLabels.IMPLEMENTATION.getLabelName())
426-
.equals(type.name().toLowerCase()))
425+
type.name()
426+
.toLowerCase()
427+
.equals(
428+
image.getLabels()
429+
.get(
430+
TlsImageLabels.IMPLEMENTATION
431+
.getLabelName())))
427432
.filter(
428433
image ->
429-
image.getLabels()
430-
.get(TlsImageLabels.CONNECTION_ROLE.getLabelName())
431-
.equals(role.name().toLowerCase()))
434+
role.name()
435+
.toLowerCase()
436+
.equals(
437+
image.getLabels()
438+
.get(
439+
TlsImageLabels.CONNECTION_ROLE
440+
.getLabelName())))
432441
.filter(
433442
image ->
434-
image.getLabels()
435-
.get(TlsImageLabels.ADDITIONAL_BUILD_FLAGS.getLabelName())
436-
.equals(additionalBuildFlags))
443+
additionalBuildFlags.equals(
444+
image.getLabels()
445+
.get(
446+
TlsImageLabels
447+
.ADDITIONAL_BUILD_FLAGS
448+
.getLabelName()))
449+
|| (DockerBuilder.NO_ADDITIONAL_BUILDFLAGS.equals(
450+
additionalBuildFlags)
451+
&& image.getLabels()
452+
.get(
453+
TlsImageLabels
454+
.ADDITIONAL_BUILD_FLAGS
455+
.getLabelName())
456+
== null))
437457
.findFirst()
438458
.orElse(null);
439459
}

0 commit comments

Comments
 (0)