PGPainless/pgpainless
PGPainless
/
pgpainless
mirror of https://github.com/pgpainless/pgpainless.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Test primary-ness of user-attributes

This commit is contained in:
Paul Schaub 2024-02-27 16:53:59 +01:00
parent 214a36ab34
commit 1aeab82c2d
Signed by: vanitasvitae
GPG Key ID: 62BEE9264BF17311
1 changed files with 76 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

79
pgpainless-core/src/test/kotlin/org/pgpainless/key/generation/OpenPgpKeyGeneratorTest.kt
View File

@ -6,6 +6,7 @@ package org.pgpainless.key.generation
import java.time.Duration
import java.time.temporal.ChronoUnit
import org.bouncycastle.bcpg.attr.ImageAttribute
import org.bouncycastle.bcpg.sig.PrimaryUserID
import org.bouncycastle.openpgp.PGPUserAttributeSubpacketVectorGenerator
import org.bouncycastle.util.encoders.Hex
@ -111,6 +112,77 @@ class OpenPgpKeyGeneratorTest {
assertEquals("Primary <primary@example.com>", info.primaryUserId)
}
@Test
fun `adding two user-attributes will mark the first one as primary`() {
val policy = Policy()
val attr1 =
PGPUserAttributeSubpacketVectorGenerator()
.apply { setImageAttribute(ImageAttribute.JPEG, byteArrayOf(0x01b)) }
.generate()
val attr2 =
PGPUserAttributeSubpacketVectorGenerator()
.apply { setImageAttribute(ImageAttribute.JPEG, byteArrayOf(0x02b)) }
.generate()
val key =
OpenPgpKeyGenerator.buildV4Key(policy)
.setPrimaryKey(KeyType.EDDSA(EdDSACurve._Ed25519)) {
addUserAttribute(attr1) // primary, since it is the first
addUserAttribute(attr2) // non-primary
}
.build()
assertTrue(
key.publicKey
.getSignaturesForUserAttribute(attr1)
.asSequence()
.single()
.hashedSubPackets
.isPrimaryUserID)
assertFalse(
key.publicKey
.getSignaturesForUserAttribute(attr2)
.asSequence()
.single()
.hashedSubPackets
.isPrimaryUserID)
}
@Test
fun `adding single user-id and single user-attribute will mark both as primary`() {
val policy = Policy()
val userId = "Alice <alice@pgpainless.org>"
val userAttribute =
PGPUserAttributeSubpacketVectorGenerator()
.apply { setImageAttribute(ImageAttribute.JPEG, byteArrayOf(0x01b)) }
.generate()
val key =
OpenPgpKeyGenerator.buildV4Key(policy)
.setPrimaryKey(KeyType.EDDSA(EdDSACurve._Ed25519)) {
addUserId(userId)
addUserAttribute(userAttribute)
}
.build()
assertTrue(
key.publicKey
.getSignaturesForUserAttribute(userAttribute)
.asSequence()
.single()
.hashedSubPackets
.isPrimaryUserID)
assertTrue(
key.publicKey
.getSignaturesForID(userId)
.asSequence()
.single()
.hashedSubPackets
.isPrimaryUserID)
}
@Test
fun `adding signing key will add embedded back-signature`() {
val key =
@ -483,9 +555,10 @@ class OpenPgpKeyGeneratorTest {
fun `opinionated set primary key without any key flags is okay`() {
val policy = Policy()
val key = OpenPgpKeyGenerator.buildV4Key(policy)
.setPrimaryKey(KeyType.EDDSA(EdDSACurve._Ed25519), keyFlags = null)
.build()
val key =
OpenPgpKeyGenerator.buildV4Key(policy)
.setPrimaryKey(KeyType.EDDSA(EdDSACurve._Ed25519), keyFlags = null)
.build()
assertNull(SignatureSubpacketsUtil.getKeyFlags(key.publicKey.signatures.next()))
}