From 14e7017ab6366f088aa3b2dba794847bb22a16d4 Mon Sep 17 00:00:00 2001
From: weidai <weidai@57ff6487-cd31-0410-9ec3-f628ee90f5f0>
Date: Sat, 11 Jul 2009 22:50:18 +0000
Subject: improve Unicode filename handling

git-svn-id: svn://svn.code.sf.net/p/cryptopp/code/trunk/c5@473 57ff6487-cd31-0410-9ec3-f628ee90f5f0
---
 files.cpp | 104 +++++++++++++++++++++++++++++---------------------------------
 1 file changed, 48 insertions(+), 56 deletions(-)

(limited to 'files.cpp')

diff --git a/files.cpp b/files.cpp
index 4d4f936..6a29ed5 100644
--- a/files.cpp
+++ b/files.cpp
@@ -27,43 +27,39 @@ void FileStore::StoreInitialize(const NameValuePairs &parameters)
 	m_stream = NULL;
 	m_file.release();
 
-	const char *fileName;
-	const wchar_t *fileNameWide;
-#ifdef CRYPTOPP_UNIX_AVAILABLE
-	std::string narrowed;
+	const char *fileName = NULL;
+#if defined(CRYPTOPP_UNIX_AVAILABLE) || _MSC_VER >= 1400
+	const wchar_t *fileNameWide = NULL;
+	if (!parameters.GetValue(Name::InputFileNameWide(), fileNameWide))
 #endif
+		if (!parameters.GetValue(Name::InputFileName(), fileName))
+		{
+			parameters.GetValue(Name::InputStreamPointer(), m_stream);
+			return;
+		}
 
-	if (parameters.GetValue(Name::InputFileName(), fileName))
-	{
+	ios::openmode binary = parameters.GetValueWithDefault(Name::InputBinaryMode(), true) ? ios::binary : ios::openmode(0);
+	m_file.reset(new std::ifstream);
 #ifdef CRYPTOPP_UNIX_AVAILABLE
-narrowName:
+	std::string narrowed;
+	if (fileNameWide)
+		fileName = (narrowed = StringNarrow(fileNameWide)).c_str();
 #endif
-		ios::openmode binary = parameters.GetValueWithDefault(Name::InputBinaryMode(), true) ? ios::binary : ios::openmode(0);
-		m_file.reset(new std::ifstream);
-		m_file->open(fileName, ios::in | binary);
-		if (!*m_file)
-			throw OpenErr(fileName);
-		m_stream = m_file.get();
-	}
-#if defined(CRYPTOPP_UNIX_AVAILABLE) || _MSC_VER >= 1400
-	else if (parameters.GetValue(Name::InputFileNameWide(), fileNameWide))
+#if _MSC_VER >= 1400
+	if (fileNameWide)
 	{
-	#if _MSC_VER >= 1400
-		ios::openmode binary = parameters.GetValueWithDefault(Name::InputBinaryMode(), true) ? ios::binary : ios::openmode(0);
-		m_file.reset(new std::ifstream);
 		m_file->open(fileNameWide, ios::in | binary);
 		if (!*m_file)
-			throw OpenErr(StringNarrow(fileNameWide));
-		m_stream = m_file.get();
-	#else
-		narrowed = StringNarrow(fileNameWide);
-		fileName = narrowed.c_str();
-		goto narrowName;
-	#endif
+			throw OpenErr(StringNarrow(fileNameWide, false));
 	}
 #endif
-	else
-		parameters.GetValue(Name::InputStreamPointer(), m_stream);
+	if (fileName)
+	{
+		m_file->open(fileName, ios::in | binary);
+		if (!*m_file)
+			throw OpenErr(fileName);
+	}
+	m_stream = m_file.get();
 }
 
 lword FileStore::MaxRetrievable() const
@@ -187,43 +183,39 @@ void FileSink::IsolatedInitialize(const NameValuePairs &parameters)
 	m_stream = NULL;
 	m_file.release();
 
-	const char *fileName;
-	const wchar_t *fileNameWide;
-#ifdef CRYPTOPP_UNIX_AVAILABLE
-	std::string narrowed;
+	const char *fileName = NULL;
+#if defined(CRYPTOPP_UNIX_AVAILABLE) || _MSC_VER >= 1400
+	const wchar_t *fileNameWide = NULL;
+	if (!parameters.GetValue(Name::OutputFileNameWide(), fileNameWide))
 #endif
+		if (!parameters.GetValue(Name::OutputFileName(), fileName))
+		{
+			parameters.GetValue(Name::OutputStreamPointer(), m_stream);
+			return;
+		}
 
-	if (parameters.GetValue(Name::OutputFileName(), fileName))
-	{
+	ios::openmode binary = parameters.GetValueWithDefault(Name::OutputBinaryMode(), true) ? ios::binary : ios::openmode(0);
+	m_file.reset(new std::ofstream);
 #ifdef CRYPTOPP_UNIX_AVAILABLE
-narrowName:
+	std::string narrowed;
+	if (fileNameWide)
+		fileName = (narrowed = StringNarrow(fileNameWide)).c_str();
 #endif
-		ios::openmode binary = parameters.GetValueWithDefault(Name::OutputBinaryMode(), true) ? ios::binary : ios::openmode(0);
-		m_file.reset(new std::ofstream);
-		m_file->open(fileName, ios::out | ios::trunc | binary);
-		if (!*m_file)
-			throw OpenErr(fileName);
-		m_stream = m_file.get();
-	}
-#if defined(CRYPTOPP_UNIX_AVAILABLE) || _MSC_VER >= 1400
-	else if (parameters.GetValue(Name::OutputFileNameWide(), fileNameWide))
+#if _MSC_VER >= 1400
+	if (fileNameWide)
 	{
-	#if _MSC_VER >= 1400
-		ios::openmode binary = parameters.GetValueWithDefault(Name::OutputBinaryMode(), true) ? ios::binary : ios::openmode(0);
-		m_file.reset(new std::ofstream);
 		m_file->open(fileNameWide, ios::out | ios::trunc | binary);
 		if (!*m_file)
-			throw OpenErr(StringNarrow(fileNameWide));
-		m_stream = m_file.get();
-	#else
-		narrowed = StringNarrow(fileNameWide);
-		fileName = narrowed.c_str();
-		goto narrowName;
-	#endif
+			throw OpenErr(StringNarrow(fileNameWide, false));
 	}
 #endif
-	else
-		parameters.GetValue(Name::OutputStreamPointer(), m_stream);
+	if (fileName)
+	{
+		m_file->open(fileName, ios::out | ios::trunc | binary);
+		if (!*m_file)
+			throw OpenErr(fileName);
+	}
+	m_stream = m_file.get();
 }
 
 bool FileSink::IsolatedFlush(bool hardFlush, bool blocking)
-- 
cgit v1.2.1