set ProtectionDomain for newly generated Class

GLGGAG · Mar 20, 2013 · 105a0c7 · 105a0c7
1 parent 7ddef5a
commit 105a0c7
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/dubbo-common/src/main/java/com/alibaba/dubbo/common/bytecode/ClassGenerator.java b/dubbo-common/src/main/java/com/alibaba/dubbo/common/bytecode/ClassGenerator.java
@@ -18,6 +18,7 @@
 import java.lang.reflect.Constructor;
 import java.lang.reflect.Method;
 import java.lang.reflect.Modifier;
+import java.security.ProtectionDomain;
 import java.util.ArrayList;
 import java.util.HashSet;
 import java.util.LinkedList;
@@ -288,7 +289,10 @@ public ClassPool getClassPool() {
 	    return mPool;
 	}
 
-	public Class<?> toClass()
+	public Class<?> toClass(){
+		return toClass(getClass().getClassLoader(), getClass().getProtectionDomain());
+	}
+	public Class<?> toClass(ClassLoader loader, ProtectionDomain pd)
 	{
 		if( mCtc != null )
 			mCtc.detach();
@@ -334,7 +338,7 @@ public Class<?> toClass()
 					}
 				}
 			}
-			return mCtc.toClass();
+			return mCtc.toClass(loader, pd);
 		}
 		catch(RuntimeException e)
 		{