From 3a306131a55c78e7a70a6d59b7b02f8366feec46 Mon Sep 17 00:00:00 2001
From: Chris Conlon <chris@wolfssl.com>
Date: Thu, 27 Apr 2017 16:39:36 -0600
Subject: [PATCH] JNI: fix for DH state check when loading private/public keys

---
 src/main/java/com/wolfssl/wolfcrypt/Dh.java | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/wolfssl/wolfcrypt/Dh.java b/src/main/java/com/wolfssl/wolfcrypt/Dh.java
index 93914b3..5e430e0 100644
--- a/src/main/java/com/wolfssl/wolfcrypt/Dh.java
+++ b/src/main/java/com/wolfssl/wolfcrypt/Dh.java
@@ -84,7 +84,7 @@ public class Dh extends NativeStruct {
 	}
 
 	public void setPrivateKey(byte[] priv) {
-		if (state == WolfCryptState.READY) {
+		if (state != WolfCryptState.UNINITIALIZED) {
 			if (privateKey != null)
 				for (int i = 0; i < privateKey.length; i++)
 					privateKey[i] = 0;
@@ -97,7 +97,7 @@ public class Dh extends NativeStruct {
 	}
 
 	public void setPublicKey(byte[] pub) {
-		if (state == WolfCryptState.READY) {
+		if (state != WolfCryptState.UNINITIALIZED) {
 			if (publicKey != null)
 				for (int i = 0; i < publicKey.length; i++)
 					publicKey[i] = 0;