Add code origin support to the grpc server integration

Author: evanchooly

dd-java-agent/agent-bootstrap/build.gradle

@@ -21,6 +21,7 @@ dependencies {
   api project(':internal-api')
   api project(':internal-api:internal-api-9')
   api project(':dd-java-agent:agent-logging')
+  api project(':dd-java-agent:agent-debugger:debugger-bootstrap')
   api libs.slf4j
   // ^ Generally a bad idea for libraries, but we're shadowing.
 

dd-java-agent/agent-debugger/src/test/java/com/datadog/debugger/agent/CapturingTestBase.java

@@ -351,6 +351,7 @@ public static LogProbe.Builder createProbeBuilder(
 
   protected TestSnapshotListener installProbes(
       Configuration configuration, ProbeDefinition... probes) {
+
     config = mock(Config.class);
     when(config.isDebuggerEnabled()).thenReturn(true);
     when(config.isDebuggerClassFileDumpEnabled()).thenReturn(true);

dd-java-agent/instrumentation/grpc-1.5/build.gradle

@@ -39,6 +39,8 @@ dependencies {
   testImplementation group: 'io.grpc', name: 'grpc-protobuf', version: grpcVersion
   testImplementation group: 'io.grpc', name: 'grpc-stub', version: grpcVersion
   testImplementation group: 'javax.annotation', name: 'javax.annotation-api', version: '1.3.2'
+  testImplementation project(':dd-java-agent:agent-debugger')
+  testImplementation libs.bundles.mockito
 
   latestDepTestImplementation sourceSets.test.output // include the protobuf generated classes
   latestDepTestCompileOnly group: 'io.grpc', name: 'grpc-core', version: '1.+'

dd-java-agent/instrumentation/grpc-1.5/src/main/java/datadog/trace/instrumentation/grpc/server/MethodHandlersInstrumentation.java

@@ -0,0 +1,60 @@
+package datadog.trace.instrumentation.grpc.server;
+
+import static datadog.trace.agent.tooling.bytebuddy.matcher.NameMatchers.nameEndsWith;
+import static datadog.trace.bootstrap.debugger.spanorigin.CodeOriginInfo.entry;
+import static net.bytebuddy.matcher.ElementMatchers.isConstructor;
+import static net.bytebuddy.matcher.ElementMatchers.takesArguments;
+
+import com.google.auto.service.AutoService;
+import datadog.trace.agent.tooling.Instrumenter.ForTypeHierarchy;
+import datadog.trace.agent.tooling.InstrumenterModule;
+import java.lang.reflect.Method;
+import net.bytebuddy.asm.Advice;
+import net.bytebuddy.description.type.TypeDescription;
+import net.bytebuddy.matcher.ElementMatcher;
+
+@AutoService(InstrumenterModule.class)
+public class MethodHandlersInstrumentation extends InstrumenterModule.Tracing
+    implements ForTypeHierarchy {
+  private static final ElementMatcher<TypeDescription> METHOD_HANDLERS =
+      nameEndsWith("$MethodHandlers");
+
+  public MethodHandlersInstrumentation() {
+    super("grpc-server-code-origin");
+  }
+
+  @Override
+  public String hierarchyMarkerType() {
+    return "io.grpc.MethodDescriptor";
+  }
+
+  @Override
+  public ElementMatcher<TypeDescription> hierarchyMatcher() {
+    return METHOD_HANDLERS;
+  }
+
+  @Override
+  public void methodAdvice(MethodTransformer transformer) {
+    transformer.applyAdvice(
+        isConstructor().and(takesArguments(2)),
+        MethodHandlersInstrumentation.class.getName() + "$BuildAdvice");
+  }
+
+  public static class BuildAdvice {
+
+    @Advice.OnMethodEnter(suppress = Throwable.class)
+    public static void onEnter(@Advice.Argument(0) Object serviceImpl) {
+      Class<?> serviceClass = serviceImpl.getClass();
+      Class<?> superclass = serviceClass.getSuperclass();
+      if (superclass != null) {
+        for (Method method : superclass.getDeclaredMethods()) {
+          try {
+            entry(serviceClass.getDeclaredMethod(method.getName(), method.getParameterTypes()));
+          } catch (NoSuchMethodException e) {
+            // service method not overridden on the impl.  skipping instrumentation.
+          }
+        }
+      }
+    }
+  }
+}