mirror of
https://github.com/pgpainless/pgpainless.git
synced 2024-12-25 04:17:59 +01:00
Kotlin conversion: SymmetricKeyAlgorithmNegotiator
This commit is contained in:
parent
32aaf59f43
commit
65bd953d65
GPG key ID:
62BEE9264BF17311
2 changed files with 77 additions and 105 deletions
|
|
@ -1,105 +0,0 @@
|
|||
// SPDX-FileCopyrightText: 2021 Paul Schaub <vanitasvitae@fsfe.org>
|
||||
//
|
||||
// SPDX-License-Identifier: Apache-2.0
|
||||
|
||||
package org.pgpainless.algorithm.negotiation;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Collections;
|
||||
import java.util.HashMap;
|
||||
import java.util.LinkedHashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
|
||||
import org.pgpainless.algorithm.SymmetricKeyAlgorithm;
|
||||
import org.pgpainless.policy.Policy;
|
||||
|
||||
/**
|
||||
* Interface for symmetric key algorithm negotiation.
|
||||
*/
|
||||
public interface SymmetricKeyAlgorithmNegotiator {
|
||||
|
||||
/**
|
||||
* Negotiate a symmetric encryption algorithm.
|
||||
* If the override is non-null, it will be returned instead of performing an actual negotiation.
|
||||
* Otherwise, the list of ordered sets containing the preferences of different recipient keys will be
|
||||
* used to determine a suitable symmetric encryption algorithm.
|
||||
*
|
||||
* @param policy algorithm policy
|
||||
* @param override algorithm override (if not null, return this)
|
||||
* @param keyPreferences list of preferences per key
|
||||
* @return negotiated algorithm
|
||||
*/
|
||||
SymmetricKeyAlgorithm negotiate(
|
||||
Policy.SymmetricKeyAlgorithmPolicy policy,
|
||||
SymmetricKeyAlgorithm override,
|
||||
List<Set<SymmetricKeyAlgorithm>> keyPreferences);
|
||||
|
||||
/**
|
||||
* Return an instance that negotiates a {@link SymmetricKeyAlgorithm} by selecting the most popular acceptable
|
||||
* algorithm from the list of preferences.
|
||||
*
|
||||
* This negotiator has the best chances to select an algorithm which is understood by all recipients.
|
||||
*
|
||||
* @return negotiator that selects by popularity
|
||||
*/
|
||||
static SymmetricKeyAlgorithmNegotiator byPopularity() {
|
||||
return new SymmetricKeyAlgorithmNegotiator() {
|
||||
@Override
|
||||
public SymmetricKeyAlgorithm negotiate(
|
||||
Policy.SymmetricKeyAlgorithmPolicy policy,
|
||||
SymmetricKeyAlgorithm override,
|
||||
List<Set<SymmetricKeyAlgorithm>> preferences) {
|
||||
if (override == SymmetricKeyAlgorithm.NULL) {
|
||||
throw new IllegalArgumentException("Algorithm override cannot be NULL (plaintext).");
|
||||
}
|
||||
|
||||
if (override != null) {
|
||||
return override;
|
||||
}
|
||||
|
||||
// Count score (occurrences) of each algorithm
|
||||
Map<SymmetricKeyAlgorithm, Integer> supportWeight = new LinkedHashMap<>();
|
||||
for (Set<SymmetricKeyAlgorithm> keyPreferences : preferences) {
|
||||
for (SymmetricKeyAlgorithm preferred : keyPreferences) {
|
||||
if (supportWeight.containsKey(preferred)) {
|
||||
supportWeight.put(preferred, supportWeight.get(preferred) + 1);
|
||||
} else {
|
||||
supportWeight.put(preferred, 1);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Pivot the score map
|
||||
Map<Integer, List<SymmetricKeyAlgorithm>> byScore = new HashMap<>();
|
||||
for (SymmetricKeyAlgorithm algorithm : supportWeight.keySet()) {
|
||||
int score = supportWeight.get(algorithm);
|
||||
List<SymmetricKeyAlgorithm> withSameScore = byScore.get(score);
|
||||
if (withSameScore == null) {
|
||||
withSameScore = new ArrayList<>();
|
||||
byScore.put(score, withSameScore);
|
||||
}
|
||||
withSameScore.add(algorithm);
|
||||
}
|
||||
|
||||
List<Integer> scores = new ArrayList<>(byScore.keySet());
|
||||
|
||||
// Sort map and iterate from highest to lowest score
|
||||
Collections.sort(scores);
|
||||
for (int i = scores.size() - 1; i >= 0; i--) {
|
||||
int score = scores.get(i);
|
||||
List<SymmetricKeyAlgorithm> withSameScore = byScore.get(score);
|
||||
// Select best algorithm
|
||||
SymmetricKeyAlgorithm best = policy.selectBest(withSameScore);
|
||||
if (best != null) {
|
||||
return best;
|
||||
}
|
||||
}
|
||||
|
||||
// If no algorithm is acceptable, choose fallback
|
||||
return policy.getDefaultSymmetricKeyAlgorithm();
|
||||
}
|
||||
};
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,77 @@
|
|||
// SPDX-FileCopyrightText: 2023 Paul Schaub <vanitasvitae@fsfe.org>
|
||||
//
|
||||
// SPDX-License-Identifier: Apache-2.0
|
||||
|
||||
package org.pgpainless.algorithm.negotiation
|
||||
|
||||
import org.pgpainless.algorithm.SymmetricKeyAlgorithm
|
||||
import org.pgpainless.policy.Policy
|
||||
import java.lang.IllegalArgumentException
|
||||
|
||||
interface SymmetricKeyAlgorithmNegotiator {
|
||||
|
||||
/**
|
||||
* Negotiate a symmetric encryption algorithm.
|
||||
* If the override is non-null, it will be returned instead of performing an actual negotiation.
|
||||
* Otherwise, the list of ordered sets containing the preferences of different recipient keys will be
|
||||
* used to determine a suitable symmetric encryption algorithm.
|
||||
*
|
||||
* @param policy algorithm policy
|
||||
* @param override algorithm override (if not null, return this)
|
||||
* @param keyPreferences list of preferences per key
|
||||
* @return negotiated algorithm
|
||||
*/
|
||||
fun negotiate(policy: Policy.SymmetricKeyAlgorithmPolicy,
|
||||
override: SymmetricKeyAlgorithm?,
|
||||
keyPreferences: List<Set<SymmetricKeyAlgorithm>>): SymmetricKeyAlgorithm
|
||||
|
||||
companion object {
|
||||
@JvmStatic
|
||||
fun byPopularity(): SymmetricKeyAlgorithmNegotiator {
|
||||
return object: SymmetricKeyAlgorithmNegotiator {
|
||||
override fun negotiate(
|
||||
policy: Policy.SymmetricKeyAlgorithmPolicy,
|
||||
override: SymmetricKeyAlgorithm?,
|
||||
keyPreferences: List<Set<SymmetricKeyAlgorithm>>):
|
||||
SymmetricKeyAlgorithm {
|
||||
if (override == SymmetricKeyAlgorithm.NULL) {
|
||||
throw IllegalArgumentException("Algorithm override cannot be NULL (plaintext).")
|
||||
}
|
||||
|
||||
if (override != null) {
|
||||
return override
|
||||
}
|
||||
|
||||
// algorithm to #occurrences
|
||||
val supportWeight = buildMap {
|
||||
keyPreferences.forEach { keyPreference ->
|
||||
keyPreference.forEach { pref ->
|
||||
put(pref, getOrDefault(pref, 0) as Int + 1)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Pivot map and sort by popularity ascending
|
||||
// score to list(algo)
|
||||
val byScore = supportWeight.toList()
|
||||
.map { e -> e.second to e.first }
|
||||
.groupBy { e -> e.first }
|
||||
.map { e -> e.key to e.value.map { it.second }.toList() }
|
||||
.associate { e -> e }
|
||||
.toSortedMap()
|
||||
|
||||
// iterate in reverse over algorithms
|
||||
for (e in byScore.entries.reversed()) {
|
||||
val best = policy.selectBest(e.value)
|
||||
if (best != null) {
|
||||
return best
|
||||
}
|
||||
}
|
||||
|
||||
return policy.defaultSymmetricKeyAlgorithm
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in a new issue