datatest_8cpp-source.html

著名的密码库Crypto++的文档 C++语言的杰作。程序员必备。

<a name="l00218"></a>00218         {
<a name="l00219"></a>00219                 SignalTestError();
<a name="l00220"></a>00220                 assert(<span class="keyword">false</span>);  <span class="comment">// TODO: implement</span>
<a name="l00221"></a>00221         }
<a name="l00222"></a>00222         <span class="keywordflow">else</span> <span class="keywordflow">if</span> (test == <span class="stringliteral">"GenerateKey"</span>)
<a name="l00223"></a>00223         {
<a name="l00224"></a>00224                 SignalTestError();
<a name="l00225"></a>00225                 assert(<span class="keyword">false</span>);
<a name="l00226"></a>00226         }
<a name="l00227"></a>00227         <span class="keywordflow">else</span>
<a name="l00228"></a>00228         {
<a name="l00229"></a>00229                 SignalTestError();
<a name="l00230"></a>00230                 assert(<span class="keyword">false</span>);
<a name="l00231"></a>00231         }
<a name="l00232"></a>00232 }
<a name="l00233"></a>00233 
<a name="l00234"></a>00234 <span class="keywordtype">void</span> TestAsymmetricCipher(TestData &amp;v)
<a name="l00235"></a>00235 {
<a name="l00236"></a>00236         std::string name = GetRequiredDatum(v, <span class="stringliteral">"Name"</span>);
<a name="l00237"></a>00237         std::string test = GetRequiredDatum(v, <span class="stringliteral">"Test"</span>);
<a name="l00238"></a>00238 
<a name="l00239"></a>00239         std::auto_ptr&lt;PK_Encryptor&gt; encryptor(<a class="code" href="class_object_factory_registry.html" title="_">ObjectFactoryRegistry&lt;PK_Encryptor&gt;::Registry</a>().CreateObject(name.c_str()));
<a name="l00240"></a>00240         std::auto_ptr&lt;PK_Decryptor&gt; decryptor(<a class="code" href="class_object_factory_registry.html" title="_">ObjectFactoryRegistry&lt;PK_Decryptor&gt;::Registry</a>().CreateObject(name.c_str()));
<a name="l00241"></a>00241 
<a name="l00242"></a>00242         std::string keyFormat = GetRequiredDatum(v, <span class="stringliteral">"KeyFormat"</span>);
<a name="l00243"></a>00243 
<a name="l00244"></a>00244         <span class="keywordflow">if</span> (keyFormat == <span class="stringliteral">"DER"</span>)
<a name="l00245"></a>00245         {
<a name="l00246"></a>00246                 decryptor-&gt;AccessMaterial().Load(<a class="code" href="class_string_store.html" title="string-based implementation of Store interface">StringStore</a>(GetDecodedDatum(v, <span class="stringliteral">"PrivateKey"</span>)).Ref());
<a name="l00247"></a>00247                 encryptor-&gt;AccessMaterial().Load(<a class="code" href="class_string_store.html" title="string-based implementation of Store interface">StringStore</a>(GetDecodedDatum(v, <span class="stringliteral">"PublicKey"</span>)).Ref());
<a name="l00248"></a>00248         }
<a name="l00249"></a>00249         <span class="keywordflow">else</span> <span class="keywordflow">if</span> (keyFormat == <span class="stringliteral">"Component"</span>)
<a name="l00250"></a>00250         {
<a name="l00251"></a>00251                 <a class="code" href="class_test_data_name_value_pairs.html">TestDataNameValuePairs</a> pairs(v);
<a name="l00252"></a>00252                 decryptor-&gt;AccessMaterial().AssignFrom(pairs);
<a name="l00253"></a>00253                 encryptor-&gt;AccessMaterial().AssignFrom(pairs);
<a name="l00254"></a>00254         }
<a name="l00255"></a>00255 
<a name="l00256"></a>00256         <span class="keywordflow">if</span> (test == <span class="stringliteral">"DecryptMatch"</span>)
<a name="l00257"></a>00257         {
<a name="l00258"></a>00258                 std::string decrypted, expected = GetDecodedDatum(v, <span class="stringliteral">"Plaintext"</span>);
<a name="l00259"></a>00259                 <a class="code" href="class_string_source.html" title="string-based implementation of Source interface">StringSource</a> ss(GetDecodedDatum(v, <span class="stringliteral">"Ciphertext"</span>), <span class="keyword">true</span>, <span class="keyword">new</span> <a class="code" href="class_p_k___decryptor_filter.html" title="proxy for the filter created by PK_Decryptor::CreateDecryptionFilter">PK_DecryptorFilter</a>(GlobalRNG(), *decryptor, <span class="keyword">new</span> <a class="code" href="class_string_sink_template.html" title="Append input to a string object.">StringSink</a>(decrypted)));
<a name="l00260"></a>00260                 <span class="keywordflow">if</span> (decrypted != expected)
<a name="l00261"></a>00261                         SignalTestFailure();
<a name="l00262"></a>00262         }
<a name="l00263"></a>00263         <span class="keywordflow">else</span> <span class="keywordflow">if</span> (test == <span class="stringliteral">"KeyPairValidAndConsistent"</span>)
<a name="l00264"></a>00264         {
<a name="l00265"></a>00265                 TestKeyPairValidAndConsistent(encryptor-&gt;AccessMaterial(), decryptor-&gt;GetMaterial());
<a name="l00266"></a>00266         }
<a name="l00267"></a>00267         <span class="keywordflow">else</span>
<a name="l00268"></a>00268         {
<a name="l00269"></a>00269                 SignalTestError();
<a name="l00270"></a>00270                 assert(<span class="keyword">false</span>);
<a name="l00271"></a>00271         }
<a name="l00272"></a>00272 }
<a name="l00273"></a>00273 
<a name="l00274"></a>00274 <span class="keywordtype">void</span> TestSymmetricCipher(TestData &amp;v)
<a name="l00275"></a>00275 {
<a name="l00276"></a>00276         std::string name = GetRequiredDatum(v, <span class="stringliteral">"Name"</span>);
<a name="l00277"></a>00277         std::string test = GetRequiredDatum(v, <span class="stringliteral">"Test"</span>);
<a name="l00278"></a>00278 
<a name="l00279"></a>00279         std::string key = GetDecodedDatum(v, <span class="stringliteral">"Key"</span>);
<a name="l00280"></a>00280         std::string plaintext = GetDecodedDatum(v, <span class="stringliteral">"Plaintext"</span>);
<a name="l00281"></a>00281 
<a name="l00282"></a>00282         <a class="code" href="class_test_data_name_value_pairs.html">TestDataNameValuePairs</a> pairs(v);
<a name="l00283"></a>00283 
<a name="l00284"></a>00284         <span class="keywordflow">if</span> (test == <span class="stringliteral">"Encrypt"</span> || test == <span class="stringliteral">"EncryptXorDigest"</span>)
<a name="l00285"></a>00285         {
<a name="l00286"></a>00286                 std::auto_ptr&lt;SymmetricCipher&gt; encryptor(<a class="code" href="class_object_factory_registry.html" title="_">ObjectFactoryRegistry&lt;SymmetricCipher, ENCRYPTION&gt;::Registry</a>().CreateObject(name.c_str()));
<a name="l00287"></a>00287                 std::auto_ptr&lt;SymmetricCipher&gt; decryptor(<a class="code" href="class_object_factory_registry.html" title="_">ObjectFactoryRegistry&lt;SymmetricCipher, DECRYPTION&gt;::Registry</a>().CreateObject(name.c_str()));
<a name="l00288"></a>00288                 <a class="code" href="class_const_byte_array_parameter.html" title="used to pass byte array input as part of a NameValuePairs object">ConstByteArrayParameter</a> iv;
<a name="l00289"></a>00289                 <span class="keywordflow">if</span> (pairs.GetValue(<a class="code" href="namespace_name.html#9b6839240a8bdf1f5bcbebf1866f4733" title="const byte *">Name::IV</a>(), iv) &amp;&amp; iv.<a class="code" href="class_const_byte_array_parameter.html#b83dbf7e4d80c0304917a3cc1d3cf144">size</a>() != encryptor-&gt;IVSize())
<a name="l00290"></a>00290                         SignalTestFailure();
<a name="l00291"></a>00291                 encryptor-&gt;SetKey((<span class="keyword">const</span> byte *)key.data(), key.size(), pairs);
<a name="l00292"></a>00292                 decryptor-&gt;SetKey((<span class="keyword">const</span> byte *)key.data(), key.size(), pairs);
<a name="l00293"></a>00293                 <span class="keywordtype">int</span> seek = pairs.GetIntValueWithDefault(<span class="stringliteral">"Seek"</span>, 0);
<a name="l00294"></a>00294                 <span class="keywordflow">if</span> (seek)
<a name="l00295"></a>00295                 {
<a name="l00296"></a>00296                         encryptor-&gt;Seek(seek);
<a name="l00297"></a>00297                         decryptor-&gt;Seek(seek);
<a name="l00298"></a>00298                 }
<a name="l00299"></a>00299                 std::string encrypted, xorDigest, ciphertext, ciphertextXorDigest;
<a name="l00300"></a>00300                 <a class="code" href="class_string_source.html" title="string-based implementation of Source interface">StringSource</a> ss(plaintext, <span class="keyword">false</span>, <span class="keyword">new</span> <a class="code" href="class_stream_transformation_filter.html" title="Filter Wrapper for StreamTransformation.">StreamTransformationFilter</a>(*encryptor, <span class="keyword">new</span> <a class="code" href="class_string_sink_template.html" title="Append input to a string object.">StringSink</a>(encrypted), <a class="code" href="class_stream_transformation_filter.html#0a08a5d90e124cde39850d544cf32e84e7a3ed9f21b92fdbb36436d921f66ca5">StreamTransformationFilter::NO_PADDING</a>));
<a name="l00301"></a>00301                 ss.Pump(plaintext.size()/2 + 1);
<a name="l00302"></a>00302                 ss.PumpAll();
<a name="l00303"></a>00303                 <span class="comment">/*{</span>
<a name="l00304"></a>00304 <span class="comment">                        std::string z;</span>
<a name="l00305"></a>00305 <span class="comment">                        encryptor-&gt;Seek(seek);</span>
<a name="l00306"></a>00306 <span class="comment">                        StringSource ss(plaintext, false, new StreamTransformationFilter(*encryptor, new StringSink(z), StreamTransformationFilter::NO_PADDING));</span>
<a name="l00307"></a>00307 <span class="comment">                        while (ss.Pump(64)) {}</span>
<a name="l00308"></a>00308 <span class="comment">                        ss.PumpAll();</span>
<a name="l00309"></a>00309 <span class="comment">                        for (int i=0; i&lt;z.length(); i++)</span>
<a name="l00310"></a>00310 <span class="comment">                                assert(encrypted[i] == z[i]);</span>
<a name="l00311"></a>00311 <span class="comment">                }*/</span>
<a name="l00312"></a>00312                 <span class="keywordflow">if</span> (test == <span class="stringliteral">"Encrypt"</span>)
<a name="l00313"></a>00313                         ciphertext = GetDecodedDatum(v, <span class="stringliteral">"Ciphertext"</span>);
<a name="l00314"></a>00314                 <span class="keywordflow">else</span>
<a name="l00315"></a>00315                 {
<a name="l00316"></a>00316                         ciphertextXorDigest = GetDecodedDatum(v, <span class="stringliteral">"CiphertextXorDigest"</span>);
<a name="l00317"></a>00317                         xorDigest.append(encrypted, 0, 64);
<a name="l00318"></a>00318                         <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=64; i&lt;encrypted.size(); i++)
<a name="l00319"></a>00319                                 xorDigest[i%64] ^= encrypted[i];
<a name="l00320"></a>00320                 }
<a name="l00321"></a>00321                 <span class="keywordflow">if</span> (test == <span class="stringliteral">"Encrypt"</span> ? encrypted != ciphertext : xorDigest != ciphertextXorDigest)
<a name="l00322"></a>00322                 {
<a name="l00323"></a>00323                         std::cout &lt;&lt; <span class="stringliteral">"incorrectly encrypted: "</span>;
<a name="l00324"></a>00324                         <a class="code" href="class_string_source.html" title="string-based implementation of Source interface">StringSource</a> xx(encrypted, <span class="keyword">false</span>, <span class="keyword">new</span> <a class="code" href="class_hex_encoder.html" title="Converts given data to base 16.">HexEncoder</a>(<span class="keyword">new</span> <a class="code" href="class_file_sink.html" title="file-based implementation of Sink interface">FileSink</a>(std::cout)));
<a name="l00325"></a>00325                         xx.Pump(256); xx.Flush(<span class="keyword">false</span>);
<a name="l00326"></a>00326                         std::cout &lt;&lt; <span class="stringliteral">"\n"</span>;
<a name="l00327"></a>00327                         SignalTestFailure();
<a name="l00328"></a>00328                 }
<a name="l00329"></a>00329                 std::string decrypted;
<a name="l00330"></a>00330                 <a class="code" href="class_string_source.html" title="string-based implementation of Source interface">StringSource</a> dd(encrypted, <span class="keyword">false</span>, <span class="keyword">new</span> <a class="code" href="class_stream_transformation_filter.html" title="Filter Wrapper for StreamTransformation.">StreamTransformationFilter</a>(*decryptor, <span class="keyword">new</span> <a class="code" href="class_string_sink_template.html" title="Append input to a string object.">StringSink</a>(decrypted), <a class="code" href="class_stream_transformation_filter.html#0a08a5d90e124cde39850d544cf32e84e7a3ed9f21b92fdbb36436d921f66ca5">StreamTransformationFilter::NO_PADDING</a>));
<a name="l00331"></a>00331                 dd.Pump(plaintext.size()/2 + 1);
<a name="l00332"></a>00332                 dd.PumpAll();
<a name="l00333"></a>00333                 <span class="keywordflow">if</span> (decrypted != plaintext)
<a name="l00334"></a>00334                 {
<a name="l00335"></a>00335                         std::cout &lt;&lt; <span class="stringliteral">"incorrectly decrypted: "</span>;
<a name="l00336"></a>00336                         <a class="code" href="class_string_source.html" title="string-based implementation of Source interface">StringSource</a> xx(decrypted, <span class="keyword">false</span>, <span class="keyword">new</span> <a class="code" href="class_hex_encoder.html" title="Converts given data to base 16.">HexEncoder</a>(<span class="keyword">new</span> <a class="code" href="class_file_sink.html" title="file-based implementation of Sink interface">FileSink</a>(std::cout)));