From accbb9d893ba34323919f5e17db17e6833d96f50 Mon Sep 17 00:00:00 2001
From: weidai <weidai@57ff6487-cd31-0410-9ec3-f628ee90f5f0>
Date: Sat, 19 Jun 2004 08:26:29 +0000
Subject: fix encoding/decoding of optional attributes

git-svn-id: svn://svn.code.sf.net/p/cryptopp/code/trunk/c5@167 57ff6487-cd31-0410-9ec3-f628ee90f5f0
---
 asn.cpp | 43 ++++++++++++++++++++++++++++++++++++-------
 1 file changed, 36 insertions(+), 7 deletions(-)

(limited to 'asn.cpp')

diff --git a/asn.cpp b/asn.cpp
index 41211ce..272cec3 100644
--- a/asn.cpp
+++ b/asn.cpp
@@ -195,6 +195,23 @@ unsigned int BERDecodeBitString(BufferedTransformation &bt, SecByteBlock &str, u
 	return bc-1;
 }
 
+void DERReencode(BufferedTransformation &source, BufferedTransformation &dest)
+{
+	byte tag;
+	source.Peek(tag);
+	BERGeneralDecoder decoder(source, tag);
+	DERGeneralEncoder encoder(dest, tag);
+	if (decoder.IsDefiniteLength())
+		decoder.TransferTo(encoder, decoder.RemainingLength());
+	else
+	{
+		while (!decoder.EndReached())
+			DERReencode(decoder, encoder);
+	}
+	decoder.MessageEnd();
+	encoder.MessageEnd();
+}
+
 void OID::EncodeValue(BufferedTransformation &bt, unsigned long v)
 {
 	for (unsigned int i=RoundUpToMultipleOf(STDMAX(7U,BitPrecision(v)), 7U)-7; i != 0; i-=7)
@@ -354,15 +371,16 @@ void EncodedObjectFilter::Put(const byte *inString, unsigned int length)
 BERGeneralDecoder::BERGeneralDecoder(BufferedTransformation &inQueue, byte asnTag)
 	: m_inQueue(inQueue), m_finished(false)
 {
-	byte b;
-	if (!m_inQueue.Get(b) || b != asnTag)
-		BERDecodeError();
-
-	m_definiteLength = BERLengthDecode(m_inQueue, m_length);
+	Init(asnTag);
 }
 
 BERGeneralDecoder::BERGeneralDecoder(BERGeneralDecoder &inQueue, byte asnTag)
 	: m_inQueue(inQueue), m_finished(false)
+{
+	Init(asnTag);
+}
+
+void BERGeneralDecoder::Init(byte asnTag)
 {
 	byte b;
 	if (!m_inQueue.Get(b) || b != asnTag)
@@ -370,7 +388,7 @@ BERGeneralDecoder::BERGeneralDecoder(BERGeneralDecoder &inQueue, byte asnTag)
 
 	m_definiteLength = BERLengthDecode(m_inQueue, m_length);
 	if (!m_definiteLength && !(asnTag & CONSTRUCTED))
-		BERDecodeError();	// cannot be primitive have indefinite length
+		BERDecodeError();	// cannot be primitive and have indefinite length
 }
 
 BERGeneralDecoder::~BERGeneralDecoder()
@@ -534,7 +552,8 @@ void PKCS8PrivateKey::BERDecode(BufferedTransformation &bt)
 			BERDecodeKey2(octetString, parametersPresent, privateKeyInfo.RemainingLength());
 		octetString.MessageEnd();
 
-		BERDecodeOptionalAttributes(privateKeyInfo);
+		if (!privateKeyInfo.EndReached())
+			BERDecodeOptionalAttributes(privateKeyInfo);
 	privateKeyInfo.MessageEnd();
 }
 
@@ -556,6 +575,16 @@ void PKCS8PrivateKey::DEREncode(BufferedTransformation &bt) const
 	privateKeyInfo.MessageEnd();
 }
 
+void PKCS8PrivateKey::BERDecodeOptionalAttributes(BufferedTransformation &bt)
+{
+	DERReencode(bt, m_optionalAttributes);
+}
+
+void PKCS8PrivateKey::DEREncodeOptionalAttributes(BufferedTransformation &bt) const
+{
+	m_optionalAttributes.CopyTo(bt);
+}
+
 NAMESPACE_END
 
 #endif
-- 
cgit v1.2.1