Cómo inyectar IL en un método en tiempo de ejecución

Question

Título más o menos lo dice todo. Sobre la base de this artículo, me he llegado con esto:

public static unsafe void Replace(this MethodBase destination, MethodBase source) 
{ 
    IntPtr srcHandle = source.MethodHandle.GetFunctionPointer(); 
    IntPtr dstHandle = destination.MethodHandle.GetFunctionPointer(); 

    int* dstPtr = (int*)dstHandle.ToPointer(); 
    *dstPtr = srcHandle.ToInt32(); 
} 

Esto funciona realmente ... de vez en cuando -.-

Por ejemplo, esto funciona.

public static class Program 
{ 
    public static void Main(string[] args) 
    { 
     MethodInfo methodA = typeof(Program).GetMethod("A", BindingFlags.Public | BindingFlags.Static); 
     MethodInfo methodB = typeof(Program).GetMethod("B", BindingFlags.Public | BindingFlags.Static); 

     methodA.Replace(methodB); 

     A(); 
     B(); 
    } 

    public static void A() 
    { 
     Console.WriteLine("Hai World"); 
    } 

    public static void B() 
    { 
     Console.WriteLine("Bai World"); 
    } 
} 

Sin embargo, esto no (SEHException). Todo lo que hice fue cambiar el orden en que se definieron las funciones.

public static class Program 
{ 
    public static void Main(string[] args) 
    { 
     MethodInfo methodA = typeof(Program).GetMethod("A", BindingFlags.Public | BindingFlags.Static); 
     MethodInfo methodB = typeof(Program).GetMethod("B", BindingFlags.Public | BindingFlags.Static); 

     methodA.Replace(methodB); 

     A(); 
     B(); 
    } 

    public static void B() 
    { 
     Console.WriteLine("Bai World"); 
    } 

    public static void A() 
    { 
     Console.WriteLine("Hai World"); 
    } 
} 

En cuanto al código en el artículo ... No pude conseguir que funcione en absoluto.

¿Alguna idea/alternativa?

Answer 1

Esto está haciendo un montón de suposiciones malas que te morderán en el trasero.

Lo primero es lo primero, la estructura devuelta por reflexión no está garantizada, de ninguna manera, para apuntar a ninguna estructura de tiempo de ejecución en absoluto. Como tal, intentar modificar los punteros que contiene o devuelve es simplemente erróneo.

En segundo lugar, si desea hacer algo como inyectar invariantes de llamadas al método pre/post (como ejemplo), entonces probablemente debería construir objetos proxy en tiempo de ejecución e inyectarlos en su lugar. O utilice la construcción de métodos dinámicos a través del espacio de nombres Emitir. Intentar manipular cosas a través de comportamientos indocumentados/desconocidos (como el código anterior) simplemente no funcionará.

También debe tener en cuenta que el JIT decide cuándo se ejecutará. A veces se ejecuta en contra de una clase completa, y a veces simplemente se ejecuta en contra de un único método. Su código no intenta determinar si el método ha sido JITted aún, y ciegamente asume que el puntero a la función devuelto puede ser modificado.