cryptopp 源码

// validat0.cpp - originally written and placed in the public domain by Wei Dai and Jeffrey Walton // Routines in this source file are only tested in Debug builds #include "pch.h" #include "secblock.h" #include "integer.h" #include "nbtheory.h" #include "zdeflate.h" #include "filters.h" #include "stdcpp.h" #include "default.h" #include "zinflate.h" #include "channels.h" #include "files.h" #include "gf2n.h" #include "gzip.h" #include "zlib.h" #include "ida.h" #include "hex.h" #include "asn.h" #include <iostream> #include <iomanip> #include <sstream> #include "validate.h" // Aggressive stack checking with VS2005 SP1 and above. #if (_MSC_FULL_VER >= 140050727) # pragma strict_gs_check (on) #endif #if CRYPTOPP_MSC_VERSION # pragma warning(disable: 4505 4355) #endif NAMESPACE_BEGIN(CryptoPP) NAMESPACE_BEGIN(Test) #if defined(CRYPTOPP_EXTENDED_VALIDATION) // Issue 64: "PolynomialMod2::operator<<=", http://github.com/weidai11/cryptopp/issues/64 bool TestPolynomialMod2() { bool pass1 = true, pass2 = true, pass3 = true; std::cout << "\nTesting PolynomialMod2 bit operations...\n\n"; static const unsigned int start = 0; static const unsigned int stop = 4 * WORD_BITS + 1; for (unsigned int i = start; i < stop; i++) { PolynomialMod2 p(1); p <<= i; Integer n(Integer::One()); n <<= i; std::ostringstream oss1; oss1 << p; std::string str1, str2; // str1 needs the commas removed used for grouping str1 = oss1.str(); str1.erase(std::remove(str1.begin(), str1.end(), ','), str1.end()); // str1 needs the trailing 'b' removed str1.erase(str1.end() - 1); // str2 is fine as-is str2 = IntToString(n, 2); pass1 &= (str1 == str2); } for (unsigned int i = start; i < stop; i++) { const word w((word)SIZE_MAX); PolynomialMod2 p(w); p <<= i; Integer n(Integer::POSITIVE, static_cast<lword>(w)); n <<= i; std::ostringstream oss1; oss1 << p; std::string str1, str2; // str1 needs the commas removed used for grouping str1 = oss1.str(); str1.erase(std::remove(str1.begin(), str1.end(), ','), str1.end()); // str1 needs the trailing 'b' removed str1.erase(str1.end() - 1); // str2 is fine as-is str2 = IntToString(n, 2); pass2 &= (str1 == str2); } RandomNumberGenerator& prng = GlobalRNG(); for (unsigned int i = start; i < stop; i++) { word w; // Cast to lword due to Visual Studio prng.GenerateBlock((byte*)&w, sizeof(w)); PolynomialMod2 p(w); p <<= i; Integer n(Integer::POSITIVE, static_cast<lword>(w)); n <<= i; std::ostringstream oss1; oss1 << p; std::string str1, str2; // str1 needs the commas removed used for grouping str1 = oss1.str(); str1.erase(std::remove(str1.begin(), str1.end(), ','), str1.end()); // str1 needs the trailing 'b' removed str1.erase(str1.end() - 1); // str2 is fine as-is str2 = IntToString(n, 2); if (str1 != str2) { std::cout << " Oops..." << "\n"; std::cout << " random: " << std::hex << n << std::dec << "\n"; std::cout << " str1: " << str1 << "\n"; std::cout << " str2: " << str2 << "\n"; } pass3 &= (str1 == str2); } std::cout << (!pass1 ? "FAILED" : "passed") << ": " << "1 shifted over range [" << std::dec << start << "," << stop << "]" << "\n"; std::cout << (!pass2 ? "FAILED" : "passed") << ": " << "0x" << std::hex << word(SIZE_MAX) << std::dec << " shifted over range [" << start << "," << stop << "]" << "\n"; std::cout << (!pass3 ? "FAILED" : "passed") << ": " << "random values shifted over range [" << std::dec << start << "," << stop << "]" << "\n"; return pass1 && pass2 && pass3; } #endif #if defined(CRYPTOPP_EXTENDED_VALIDATION) bool TestCompressors() { std::cout << "\nTesting Compressors and Decompressors...\n\n"; bool fail1 = false, fail2 = false, fail3 = false; try { // Gzip uses Adler32 checksums. We expect a failure to to happen on occasion. // If we see more than 2 failures in a run of 128, then we need to investigate. unsigned int truncatedCount=0; for (unsigned int i = 0; i<128; ++i) { std::string src, dest, rec; unsigned int len = GlobalRNG().GenerateWord32(4, 0xfff); RandomNumberSource(GlobalRNG(), len, true, new StringSink(src)); StringSource(src, true, new Gzip(new StringSink(dest))); StringSource(dest, true, new Gunzip(new StringSink(rec))); if (src != rec) throw Exception(Exception::OTHER_ERROR, "Gzip failed to decompress stream"); // Tamper try { StringSource(dest.substr(0, len - 2), true, new Gunzip(new StringSink(rec))); if (truncatedCount++ >= 2) { std::cout << "FAILED: Gzip failed to detect a truncated stream\n"; fail1 = true; } } catch (const Exception&) {} } } catch (const Exception& ex) { std::cout << "FAILED: " << ex.what() << "\n"; fail1 = true; } // ************************************************************** // Gzip Filename, Filetime and Comment try { std::string filename = "test.txt"; std::string comment = "This is a test"; word32 filetime = GlobalRNG().GenerateWord32(4, 0xffffff); AlgorithmParameters params = MakeParameters(Name::FileTime(), (int)filetime) (Name::FileName(), ConstByteArrayParameter(filename.c_str(), false)) (Name::Comment(), ConstByteArrayParameter(comment.c_str(), false)); std::string src, dest; unsigned int len = GlobalRNG().GenerateWord32(4, 0xfff); RandomNumberSource(GlobalRNG(), len, true, new StringSink(src)); Gunzip unzip(new StringSink(dest)); StringSource(src, true, new Gzip(params, new Redirector(unzip))); if (filename != unzip.GetFilename()) throw Exception(Exception::OTHER_ERROR, "Failed to retrieve filename"); if (filetime != unzip.GetFiletime()) throw Exception(Exception::OTHER_ERROR, "Failed to retrieve filetime"); if (comment != unzip.GetComment()) throw Exception(Exception::OTHER_ERROR, "Failed to retrieve comment"); std::cout << "passed: filenames, filetimes and comments\n"; } catch (const Exception& ex) { std::cout << "FAILED: " << ex.what() << "\n"; } // Unzip random data. See if we can induce a crash for (unsigned int i = 0; i<128; i++) { SecByteBlock src; unsigned int len = GlobalRNG().GenerateWord32(4, 0xfff); RandomNumberSource(GlobalRNG(), len, true, new ArraySink(src, src.size())); try { ArraySource(src.data(), src.size(), true, new Gunzip(new Redirector(TheBitBucket()))); } catch (const Exception&) {} } // Unzip random data. See if we can induce a crash for (unsigned int i = 0; i<128; i++) { SecByteBlock src; unsigned int len = GlobalRNG().GenerateWord32(4, 0xfff); src.resize(len); RandomNumberSource(GlobalRNG(), len, true, new ArraySink(src, src.size())); src[0] = (byte)0x1f; // magic header src[1] = (byte)0x8b; src[2] = 0x00; // extra flags src[3] = sr