From b52d9be85c654f1c307307f5d0d257a7d6c7187a Mon Sep 17 00:00:00 2001
From: weidai <weidai11@users.noreply.github.com>
Date: Mon, 17 Jul 2006 14:49:51 +0000
Subject: [PATCH] additional AES test vectors

---
 datatest.cpp | 2 ++
 regtest.cpp  | 6 ++++++
 2 files changed, 8 insertions(+)

diff --git a/datatest.cpp b/datatest.cpp
index cd431750..79609e88 100644
--- a/datatest.cpp
+++ b/datatest.cpp
@@ -409,6 +409,8 @@ void OutputNameValuePairs(const NameValuePairs &v)
 void TestDataFile(const std::string &filename, unsigned int &totalTests, unsigned int &failedTests)
 {
 	std::ifstream file(filename.c_str());
+	if (!file.good())
+		throw Exception(Exception::OTHER_ERROR, "Can not open file " + filename + " for reading");
 	TestData v;
 	s_currentTestData = &v;
 	std::string name, value, lastAlgName;
diff --git a/regtest.cpp b/regtest.cpp
index b6cd3d9b..05ff75e5 100644
--- a/regtest.cpp
+++ b/regtest.cpp
@@ -18,6 +18,7 @@
 #include "tea.h"
 #include "panama.h"
 #include "pssr.h"
+#include "aes.h"
 
 USING_NAMESPACE(CryptoPP)
 
@@ -63,6 +64,11 @@ void RegisterFactories()
 	RegisterSymmetricCipherDefaultFactories<ECB_Mode<XTEA> >();
 	RegisterSymmetricCipherDefaultFactories<PanamaCipher<LittleEndian> >();
 	RegisterSymmetricCipherDefaultFactories<PanamaCipher<BigEndian> >();
+	RegisterSymmetricCipherDefaultFactories<ECB_Mode<AES> >();
+	RegisterSymmetricCipherDefaultFactories<CBC_Mode<AES> >();
+	RegisterSymmetricCipherDefaultFactories<CFB_Mode<AES> >();
+	RegisterSymmetricCipherDefaultFactories<OFB_Mode<AES> >();
+	RegisterSymmetricCipherDefaultFactories<CTR_Mode<AES> >();
 
 	s_registered = true;
 }