use weakrefrence

Author: afiqiqmal

build.gradle

@@ -5,7 +5,7 @@ buildscript {
         jcenter()
     }
     dependencies {
-        classpath 'com.android.tools.build:gradle:2.3.1'
+        classpath 'com.android.tools.build:gradle:2.3.3'
         classpath 'com.github.dcendents:android-maven-gradle-plugin:1.5'
 
         // NOTE: Do not place your application dependencies here; they belong

library/src/main/java/com/zeroone/conceal/ConcealCrypto.java

@@ -22,6 +22,7 @@
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.OutputStream;
+import java.lang.ref.WeakReference;
 import java.security.MessageDigest;
 import java.security.NoSuchAlgorithmException;
 
@@ -38,6 +39,7 @@
 public class ConcealCrypto {
 
     private Crypto crypto;
+    private Context mContext;
     private KeyChain keyChain;
     private Entity mEntityPassword = Entity.create(BuildConfig.APPLICATION_ID);
     private boolean enableCrypto=true;
@@ -46,6 +48,7 @@ public class ConcealCrypto {
 
     private ConcealCrypto(CryptoBuilder builder){
         crypto = builder.crypto;
+        mContext = builder.context.get();
         mEntityPassword = builder.mEntityPassword;
         enableCrypto = builder.mEnabledCrypto;
         enableHashKey = builder.mEnableHashKey;
@@ -320,7 +323,7 @@ public String deObscureWithIteration(String cipher,int iteration){
     }
 
     public static class CryptoBuilder{
-        Context context;
+        private WeakReference<Context> context;
         private KeyChain makeKeyChain;
         private Crypto crypto;
         private CryptoConfig mKeyChain = CryptoConfig.KEY_256;
@@ -331,7 +334,7 @@ public static class CryptoBuilder{
         private String mFolderName;
 
         public CryptoBuilder(Context context) {
-            this.context = context;
+            this.context = new WeakReference<Context>(context.getApplicationContext());
         }
 
         public CryptoBuilder setKeyChain(CryptoConfig config){
@@ -365,8 +368,12 @@ public CryptoBuilder setStoredFolder(String folderName){
 
         public ConcealCrypto create(){
 
+            if (this.context == null){
+                throw new RuntimeException("Context cannot be null");
+            }
+
             mEntityPassword = Entity.create(CipherUtils.obscureEncodeSixFourString(mEntityPasswordRaw.getBytes()));
-            makeKeyChain = new SharedPrefsBackedKeyChain(context,(mKeyChain==null)?CryptoConfig.KEY_256:mKeyChain);
+            makeKeyChain = new SharedPrefsBackedKeyChain(this.context.get(),(mKeyChain==null)?CryptoConfig.KEY_256:mKeyChain);
 
             if (mKeyChain == null) {
                 crypto = AndroidConceal.get().createDefaultCrypto(makeKeyChain);

library/src/main/java/com/zeroone/conceal/ConcealPrefRepository.java

@@ -18,6 +18,7 @@
 
 import java.io.File;
 import java.io.IOException;
+import java.lang.ref.WeakReference;
 import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.LinkedHashMap;
@@ -49,7 +50,7 @@ public class ConcealPrefRepository {
 
     @SuppressLint("CommitPrefEdits")
     private ConcealPrefRepository(@NonNull PreferencesBuilder builder){
-        mContext = builder.mContext;
+        mContext = builder.mContext.get();
         mKeyChain = builder.mKeyChain;
         mEnabledCrypto = builder.mEnabledCrypto;
         mEnableCryptKey = builder.mEnableCryptKey;
@@ -851,7 +852,7 @@ public void apply() {
      ****************************************************************************************/
     public static class PreferencesBuilder{
 
-        private Context mContext;
+        private WeakReference<Context> mContext;
         private CryptoConfig mKeyChain = CryptoConfig.KEY_256;
         private String mPrefname = null;
         private String mFolderName = null;
@@ -862,7 +863,7 @@ public static class PreferencesBuilder{
         private OnDataChangeListener onDataChangeListener;
 
         public PreferencesBuilder(Context context) {
-            mContext = context;
+            mContext = new WeakReference<>(context.getApplicationContext());
         }
 
         public PreferencesBuilder useDefaultPrefStorage(){
@@ -932,6 +933,10 @@ public PreferencesBuilder setPrefListener(OnDataChangeListener listener){
          */
         public ConcealPrefRepository create(){
 
+            if (this.mContext == null){
+                throw new RuntimeException("Context cannot be null");
+            }
+
             if(mFolderName !=null){
                 File file = new File(mFolderName);
                 try {
@@ -947,10 +952,10 @@ public ConcealPrefRepository create(){
             }
 
             if (mPrefname!=null){
-                sharedPreferences = mContext.getSharedPreferences(CipherUtils.obscureEncodeSixFourString(mPrefname.getBytes()), MODE_PRIVATE);
+                sharedPreferences = this.mContext.get().getSharedPreferences(CipherUtils.obscureEncodeSixFourString(mPrefname.getBytes()), MODE_PRIVATE);
             }
             else {
-                sharedPreferences = PreferenceManager.getDefaultSharedPreferences(mContext);
+                sharedPreferences = PreferenceManager.getDefaultSharedPreferences(this.mContext.get());
             }
 
             return new ConcealPrefRepository(this);