55 lines
1.9 KiB
Kotlin
55 lines
1.9 KiB
Kotlin
// SPDX-FileCopyrightText: 2024 Paul Schaub <vanitasvitae@fsfe.org>
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package org.pgpainless.sop
|
|
|
|
import java.io.BufferedOutputStream
|
|
import java.io.InputStream
|
|
import java.io.OutputStream
|
|
import kotlin.jvm.Throws
|
|
import org.bouncycastle.util.io.Streams
|
|
import org.pgpainless.decryption_verification.OpenPgpInputStream
|
|
import org.pgpainless.util.ArmoredOutputStreamFactory
|
|
import sop.Ready
|
|
import sop.enums.ArmorLabel
|
|
import sop.exception.SOPGPException
|
|
import sop.operation.Armor
|
|
|
|
/** Implementation of the `armor` operation using PGPainless. */
|
|
class ArmorImpl : Armor {
|
|
|
|
@Throws(SOPGPException.BadData::class)
|
|
override fun data(data: InputStream): Ready {
|
|
return object : Ready() {
|
|
override fun writeTo(outputStream: OutputStream) {
|
|
// By buffering the output stream, we can improve performance drastically
|
|
val bufferedOutputStream = BufferedOutputStream(outputStream)
|
|
|
|
// Determine the nature of the given data
|
|
val openPgpIn = OpenPgpInputStream(data)
|
|
openPgpIn.reset()
|
|
|
|
if (openPgpIn.isAsciiArmored) {
|
|
// armoring already-armored data is an idempotent operation
|
|
Streams.pipeAll(openPgpIn, bufferedOutputStream)
|
|
bufferedOutputStream.flush()
|
|
openPgpIn.close()
|
|
return
|
|
}
|
|
|
|
val armor = ArmoredOutputStreamFactory.get(bufferedOutputStream)
|
|
Streams.pipeAll(openPgpIn, armor)
|
|
bufferedOutputStream.flush()
|
|
armor.close()
|
|
openPgpIn.close()
|
|
}
|
|
}
|
|
}
|
|
|
|
@Deprecated("Setting custom labels is not supported.")
|
|
override fun label(label: ArmorLabel): Armor {
|
|
throw SOPGPException.UnsupportedOption("Setting custom Armor labels not supported.")
|
|
}
|
|
}
|