Check target super-class implements FieldBackedContextAccessor before delegating to it

This avoids the need to create (and catch) `NoSuchMethodError`s when we fail to inject the super-class.

For example starting the profiler before byte-buddy can trigger loading of various JDK concurrent classes that we would usually inject context into.

Author: mcculls

dd-java-agent/agent-tooling/src/main/java/datadog/trace/agent/tooling/context/FieldBackedContextInjector.java

@@ -60,9 +60,14 @@ public final class FieldBackedContextInjector implements AsmVisitorWrapper {
       Type.getMethodDescriptor(
           Type.VOID_TYPE, Type.getType(Object.class), Type.INT_TYPE, Type.getType(Object.class));
 
+  static final String IS_ASSIGNABLE_FROM_METHOD = "isAssignableFrom";
+  static final String IS_ASSIGNABLE_FROM_METHOD_DESCRIPTOR =
+      Type.getMethodDescriptor(Type.BOOLEAN_TYPE, Type.getType(Class.class));
+
   static final String OBJECT_DESCRIPTOR = Type.getDescriptor(Object.class);
 
-  static final String LINKAGE_ERROR_CLASS = getInternalName(LinkageError.class.getName());
+  public static final Type EXPECTED_SUPER_STORE_TYPE =
+      Type.getType(FieldBackedContextAccessor.class);
 
   /** Keeps track of injection requests for the class being transformed by the current thread. */
   static final ThreadLocal<BitSet> INJECTED_STORE_IDS = new ThreadLocal<>();
@@ -293,15 +298,25 @@ private void addStoreGetter(
           Label superStoreLabel = new Label();
           Label defaultStoreLabel = new Label();
 
-          mv.visitTryCatchBlock(
-              superStoreLabel, defaultStoreLabel, defaultStoreLabel, LINKAGE_ERROR_CLASS);
+          String superName = instrumentedType.getSuperClass().asErasure().getInternalName();
+
+          mv.visitLdcInsn(EXPECTED_SUPER_STORE_TYPE);
+          mv.visitLdcInsn(Type.getObjectType(superName));
+          mv.visitMethodInsn(
+              Opcodes.INVOKEVIRTUAL,
+              "java/lang/Class",
+              IS_ASSIGNABLE_FROM_METHOD,
+              IS_ASSIGNABLE_FROM_METHOD_DESCRIPTOR,
+              false);
+          mv.visitJumpInsn(Opcodes.IFEQ, defaultStoreLabel);
+
           beginNextStore(mv, superStoreLabel);
 
-          invokeSuperGet(mv, instrumentedType.getSuperClass().asErasure().getInternalName());
+          invokeSuperGet(mv, superName);
 
           mv.visitLabel(defaultStoreLabel);
           if (frames) {
-            mv.visitFrame(Opcodes.F_SAME1, 0, null, 1, new Object[] {LINKAGE_ERROR_CLASS});
+            mv.visitFrame(Opcodes.F_SAME, 0, null, 0, null);
           }
 
           invokeWeakGet(mv);
@@ -358,15 +373,25 @@ private void addStorePutter(
           Label superStoreLabel = new Label();
           Label defaultStoreLabel = new Label();
 
-          mv.visitTryCatchBlock(
-              superStoreLabel, defaultStoreLabel, defaultStoreLabel, LINKAGE_ERROR_CLASS);
+          String superName = instrumentedType.getSuperClass().asErasure().getInternalName();
+
+          mv.visitLdcInsn(EXPECTED_SUPER_STORE_TYPE);
+          mv.visitLdcInsn(Type.getObjectType(superName));
+          mv.visitMethodInsn(
+              Opcodes.INVOKEVIRTUAL,
+              "java/lang/Class",
+              IS_ASSIGNABLE_FROM_METHOD,
+              IS_ASSIGNABLE_FROM_METHOD_DESCRIPTOR,
+              false);
+          mv.visitJumpInsn(Opcodes.IFEQ, defaultStoreLabel);
+
           beginNextStore(mv, superStoreLabel);
 
-          invokeSuperPut(mv, instrumentedType.getSuperClass().asErasure().getInternalName());
+          invokeSuperPut(mv, superName);
 
           mv.visitLabel(defaultStoreLabel);
           if (frames) {
-            mv.visitFrame(Opcodes.F_SAME1, 0, null, 1, new Object[] {LINKAGE_ERROR_CLASS});
+            mv.visitFrame(Opcodes.F_SAME, 0, null, 0, null);
           }
 
           invokeWeakPut(mv);