TudbuT/DarkReflection
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity Actions

move OffsetGetter to UnsafeCompat
All checks were successful
/ Build & Publish (push) Successful in 40s
Details

Browse source
This commit is contained in:
Daniella / Tove 2024-06-29 09:30:02 +02:00
parent af2c043bab
commit e50427340d
6 changed files with 62 additions and 57 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/main/java/de/tudbut/darkreflection/Strategy.java
View file

@ -10,7 +10,7 @@ import java.lang.reflect.AccessibleObject;
import java.lang.reflect.Field;
import java.util.Arrays;
import static de.tudbut.darkreflection.UnsafeProvider.offsetGetter;
import static de.tudbut.darkreflection.UnsafeProvider.compat;
// which strategy depends on which?
@ -117,8 +117,8 @@ public class Strategy {
}),
DarkReflection((owner, field, isStatic, type) -> reflectedFieldStrategy.instance(Arrays.stream(classStrategy.instance(owner).getDeclaredFields()).filter(x -> x.getName().equals(field)).findAny().orElse(null))),
Unsafe((owner, field, isStatic, type) -> new UnsafeField<>(
isStatic ? offsetGetter.staticFieldBase(owner, field) : null,
isStatic ? offsetGetter.staticFieldOffset(owner, field) : offsetGetter.objectFieldOffset(owner, field), type)),
isStatic ? compat.staticFieldBase(owner, field) : null,
isStatic ? compat.staticFieldOffset(owner, field) : compat.objectFieldOffset(owner, field), type)),
;
private final FieldByNameStrategyProvider backend;

12
src/main/java/de/tudbut/darkreflection/UnsafeProvider.java
View file

@ -1,8 +1,8 @@
package de.tudbut.darkreflection;
import de.tudbut.darkreflection.unsafe.BasicOffsetGetter;
import de.tudbut.darkreflection.unsafe.IOffsetGetter;
import de.tudbut.darkreflection.unsafe.InternalUnsafeOffsetGetter;
import de.tudbut.darkreflection.unsafe.BasicUnsafeCompat;
import de.tudbut.darkreflection.unsafe.IUnsafeCompat;
import de.tudbut.darkreflection.unsafe.InternalUnsafeCompat;
import sun.misc.Unsafe;
import java.lang.reflect.Field;
@ -20,15 +20,15 @@ public class UnsafeProvider {
}
}
public static IOffsetGetter offsetGetter;
public static IUnsafeCompat compat;
static {
try {
Class<?> c = Class.forName("jdk.internal.misc.Unsafe");
c.getMethod("objectFieldOffset", Class.class, String.class);
Object o = DarkMethod.invoke(c.getMethod("getUnsafe"), null);
offsetGetter = new InternalUnsafeOffsetGetter(c, o);
compat = new InternalUnsafeCompat(c, o);
} catch (Throwable e) {
offsetGetter = new BasicOffsetGetter();
compat = new BasicUnsafeCompat();
}
}
}

39
src/main/java/de/tudbut/darkreflection/unsafe/BasicOffsetGetter.java
View file

@ -1,39 +0,0 @@
package de.tudbut.darkreflection.unsafe;
import de.tudbut.darkreflection.DarkField;
import java.lang.reflect.Field;
import static de.tudbut.darkreflection.UnsafeProvider.unsafe;
public class BasicOffsetGetter implements IOffsetGetter {
@Override
public long objectFieldOffset(Field f) {
return unsafe.objectFieldOffset(f);
}
@Override
public long objectFieldOffset(Class<?> clazz, String name) {
return unsafe.objectFieldOffset(DarkField.nameToField(clazz, name));
}
@Override
public long staticFieldOffset(Field f) {
return unsafe.staticFieldOffset(f);
}
@Override
public long staticFieldOffset(Class<?> clazz, String name) {
return unsafe.staticFieldOffset(DarkField.nameToField(clazz, name));
}
@Override
public Object staticFieldBase(Field f) {
return unsafe.staticFieldBase(f);
}
@Override
public Object staticFieldBase(Class<?> clazz, String name) {
return unsafe.staticFieldBase(DarkField.nameToField(clazz, name));
}
}

49
src/main/java/de/tudbut/darkreflection/unsafe/BasicUnsafeCompat.java Normal file
View file

@ -0,0 +1,49 @@
package de.tudbut.darkreflection.unsafe;
import de.tudbut.darkreflection.DarkAccessibleObject;
import de.tudbut.darkreflection.DarkField;
import de.tudbut.darkreflection.DarkMethod;
import sun.misc.Unsafe;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.security.ProtectionDomain;
import static de.tudbut.darkreflection.UnsafeProvider.unsafe;
public class BasicUnsafeCompat implements IUnsafeCompat {
@Override
public long objectFieldOffset(Class<?> clazz, String name) {
return unsafe.objectFieldOffset(DarkField.nameToField(clazz, name));
}
@Override
public long staticFieldOffset(Class<?> clazz, String name) {
return unsafe.staticFieldOffset(DarkField.nameToField(clazz, name));
}
@Override
public Object staticFieldBase(Class<?> clazz, String name) {
return unsafe.staticFieldBase(DarkField.nameToField(clazz, name));
}
@Override
public Class<?> defineClass(String name, byte[] bytes) {
try {
Method m = Unsafe.class.getDeclaredMethod("defineClass", String.class, byte[].class, int.class, int.class, ClassLoader.class, ProtectionDomain.class);
DarkAccessibleObject.makeAccessible(m);
return (Class<?>) m.invoke(null, name, bytes, 0, bytes.length, null, null);
} catch (InvocationTargetException e) {
throw new RuntimeException(e.getCause());
} catch (NoSuchMethodException | IllegalAccessException ignore) {
}
Method m = DarkMethod.instance(ClassLoader.class, "defineClass0", String.class, byte[].class, int.class, int.class, ProtectionDomain.class);
try {
return DarkMethod.invoke(m, ClassLoader.getSystemClassLoader(), name, bytes, 0, bytes.length, null);
} catch (InvocationTargetException e) {
throw new RuntimeException(e.getCause());
}
}
}

9
src/main/java/de/tudbut/darkreflection/unsafe/IOffsetGetter.java → src/main/java/de/tudbut/darkreflection/unsafe/IUnsafeCompat.java
View file

@ -1,13 +1,8 @@
package de.tudbut.darkreflection.unsafe;
import java.lang.reflect.Field;
public interface IOffsetGetter {
long objectFieldOffset(Field f);
public interface IUnsafeCompat {
long objectFieldOffset(Class<?> clazz, String name);
long staticFieldOffset(Field f);
long staticFieldOffset(Class<?> clazz, String name);
Object staticFieldBase(Field f);
Object staticFieldBase(Class<?> clazz, String name);
Class<?> defineClass(String name, byte[] bytes);
}

4
src/main/java/de/tudbut/darkreflection/unsafe/InternalUnsafeOffsetGetter.java → src/main/java/de/tudbut/darkreflection/unsafe/InternalUnsafeCompat.java
View file

@ -5,12 +5,12 @@ import de.tudbut.darkreflection.DarkMethod;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
public class InternalUnsafeOffsetGetter extends BasicOffsetGetter {
public class InternalUnsafeCompat extends BasicUnsafeCompat {
private final Method objectFieldOffset0;
private final Object iUnsafe;
public InternalUnsafeOffsetGetter(Class<?> iUnsafeClass, Object iUnsafe) throws NoSuchMethodException {
public InternalUnsafeCompat(Class<?> iUnsafeClass, Object iUnsafe) throws NoSuchMethodException {
this.objectFieldOffset0 = iUnsafeClass.getMethod("objectFieldOffset", Class.class, String.class);
this.iUnsafe = iUnsafe;
}