Homec4science
Diffusion SHRINE MedCo Fork (patch/1.22.2.4) Browse

shrine-medco/commons/crypto/src/test/scala/net/shrine/crypto2/DownStreamCertCollectionTest.scala
b85ef5a3279apatch/1.22.2.4

Last Change
Raw File

DownStreamCertCollectionTest.scala
View Options

package net.shrine.crypto2
import java.math.BigInteger
import java.security.cert.X509Certificate
import java.security.{KeyPairGenerator, SecureRandom}
import java.util.Date
import javax.security.auth.x500.X500Principal
import junit.framework.TestFailure
import net.shrine.crypto.{KeyStoreDescriptor, KeyStoreType, NewTestKeyStore}
import net.shrine.util.{NonEmptySeq, SingleHubModel}
import org.bouncycastle.asn1.ASN1Sequence
import org.bouncycastle.asn1.x500.X500Name
import org.bouncycastle.asn1.x509._
import org.bouncycastle.cert.X509v3CertificateBuilder
import org.bouncycastle.cert.jcajce.JcaX509CertificateConverter
import org.bouncycastle.operator.jcajce.JcaContentSignerBuilder
import org.bouncycastle.x509.X509V3CertificateGenerator
import org.junit.runner.RunWith
import org.scalatest.{FlatSpec, Matchers, ShouldMatchers}
import org.scalatest.junit.JUnitRunner
/**
* Created by ty on 11/1/16.
*/
@RunWith(classOf[JUnitRunner])
class DownStreamCertCollectionTest extends FlatSpec with Matchers {
val descriptor = NewTestKeyStore.descriptor
val heyo = "Heyo!".getBytes("UTF-8")
"A down stream cert collection" should "build and verify its own messages" in {
val hubCertCollection = BouncyKeyStoreCollection.fromFileRecoverWithClassPath(descriptor) match {
case hub:DownStreamCertCollection => hub
case _ => fail("This should generate a DownstreamCertCollection!")
}
val testEntry = KeyStoreEntry(TestCert.cert, NonEmptySeq("notTrusted", Nil), Some(TestCert.keyPair.getPrivate))
hubCertCollection.allEntries.size shouldBe 2
hubCertCollection.myEntry.privateKey.isDefined shouldBe true
hubCertCollection.caEntry.privateKey.isDefined shouldBe false
hubCertCollection.myEntry.aliases.first shouldBe "shrine-test"
hubCertCollection.caEntry.aliases.first shouldBe "shrine-test-ca"
hubCertCollection.caEntry.wasSignedBy(hubCertCollection.myEntry) shouldBe false
hubCertCollection.myEntry.wasSignedBy(hubCertCollection.caEntry) shouldBe true
val mySigned = hubCertCollection.myEntry.sign(heyo).get
val testSigned = testEntry.sign(heyo).get
testEntry.verify(mySigned, heyo) shouldBe false
testEntry.verify(testSigned, heyo) shouldBe true
testEntry.signed(testEntry.cert) shouldBe true
hubCertCollection.myEntry.verify(testSigned, heyo) shouldBe false
hubCertCollection.myEntry.verify(mySigned, heyo) shouldBe true
hubCertCollection.caEntry.verify(testSigned, heyo) shouldBe false
hubCertCollection.caEntry.verify(mySigned, heyo) shouldBe false
hubCertCollection.verifyBytes(hubCertCollection.signBytes(heyo), heyo) shouldBe true
}
}
object TestCert {
// generate a key pair
val keyPairGenerator = KeyPairGenerator.getInstance("RSA", "BC")
keyPairGenerator.initialize(4096, new SecureRandom())
val keyPair = keyPairGenerator.generateKeyPair()
val name: X500Name = new X500Name("cn=testing")
val subject = new X500Name("dc=stillTesting")
val serial = BigInteger.valueOf(System.currentTimeMillis())
val notBefore = new Date(0)
val notAfter = new Date(java.time.Instant.now().toEpochMilli + 1000 * 60 * 60 * 24 * 10)
val subjectPublicKeyInfo = SubjectPublicKeyInfo.getInstance(ASN1Sequence.getInstance(keyPair.getPublic.getEncoded))
val certBuilder = new X509v3CertificateBuilder(name, serial, notBefore, notAfter, subject, subjectPublicKeyInfo)
val certHolder = certBuilder.build(new JcaContentSignerBuilder("SHA256withRSA").setProvider("BC").build(keyPair.getPrivate))
val cert = new JcaX509CertificateConverter().setProvider("BC").getCertificate(certHolder)
}
c4science · Help