From 221cdf2bf83b71509028e78fcc3d5efea1fb082b Mon Sep 17 00:00:00 2001
From: Arvid Norberg <arvid@libtorrent.org>
Date: Sat, 9 Feb 2008 21:14:49 +0000
Subject: [PATCH] fixed error handling in pe-crypto

---
 src/pe_crypto.cpp | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/pe_crypto.cpp b/src/pe_crypto.cpp
index a865b4c73..d9298e989 100644
--- a/src/pe_crypto.cpp
+++ b/src/pe_crypto.cpp
@@ -61,8 +61,7 @@ namespace libtorrent
 
 		TORRENT_ASSERT(sizeof(m_dh_prime) == DH_size(m_DH));
 		
-		DH_generate_key(m_DH);
-		if (m_DH->pub_key == 0)
+		if (DH_generate_key(m_DH) == 0 || m_DH->pub_key == 0)
 		{
 			DH_free(m_DH);
 			throw std::bad_alloc();