From 22cac8f1a25c7d34a94bc5e00c56c0532509071f Mon Sep 17 00:00:00 2001
From: Cuchaz
Date: Mon, 30 Mar 2015 22:00:54 -0400
Subject: add publifier

---
 src/cuchaz/enigma/bytecode/ClassPublifier.java | 41 ++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 src/cuchaz/enigma/bytecode/ClassPublifier.java

(limited to 'src/cuchaz/enigma/bytecode')

diff --git a/src/cuchaz/enigma/bytecode/ClassPublifier.java b/src/cuchaz/enigma/bytecode/ClassPublifier.java
new file mode 100644
index 0000000..d8162e4
--- /dev/null
+++ b/src/cuchaz/enigma/bytecode/ClassPublifier.java
@@ -0,0 +1,41 @@
+package cuchaz.enigma.bytecode;
+
+import javassist.CtBehavior;
+import javassist.CtClass;
+import javassist.CtField;
+import javassist.bytecode.AccessFlag;
+import javassist.bytecode.InnerClassesAttribute;
+
+
+public class ClassPublifier {
+
+	public static CtClass publify(CtClass c) {
+		
+		// publify all the fields
+		for (CtField field : c.getDeclaredFields()) {
+			field.setModifiers(publify(field.getModifiers()));
+		}
+		
+		// publify all the methods and constructors
+		for (CtBehavior behavior : c.getDeclaredBehaviors()) {
+			behavior.setModifiers(publify(behavior.getModifiers()));
+		}
+		
+		// publify all the inner classes
+		InnerClassesAttribute attr = (InnerClassesAttribute)c.getClassFile().getAttribute(InnerClassesAttribute.tag);
+		if (attr != null) {
+			for (int i=0; i<attr.tableLength(); i++) {
+				attr.setAccessFlags(i, publify(attr.accessFlags(i)));
+			}
+		}
+		
+		return c;
+	}
+
+	private static int publify(int flags) {
+		if (AccessFlag.isPrivate(flags) || AccessFlag.isProtected(flags)) {
+			flags = AccessFlag.setPublic(flags);
+		}
+		return flags;
+	}
+}
-- 
cgit v1.2.3