Click here to Skip to main content
15,895,656 members
Articles / Programming Languages / MSIL

Assembly Manipulation and C# / VB.NET Code Injection

Rate me:
Please Sign up or sign in to vote.
4.95/5 (200 votes)
26 Apr 2013MIT5 min read 927K   25K   486  
Reflexil is an assembly editor and runs as a plug-in for Reflector or JustDecompile. Reflexil is able to manipulate IL code and save the modified assemblies to disk. Reflexil also supports "on-the-fly" C#/VB.NET code injection.
/*
    Reflexil .NET assembly editor.
    Copyright (C) 2007 Sebastien LEBRETON

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#region " Imports "
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.IO;
using System.Windows.Forms;
using Mono.Cecil;
using Reflexil.Utils;
#endregion

namespace Reflexil.Forms
{
	public partial class ReferenceUpdaterForm : Form
    {
 
        #region " Events "
        private void BackgroundWorker_DoWork(object sender, DoWorkEventArgs e)
        {
            BackgroundWorker worker = sender as BackgroundWorker;
            AssemblyDefinition[] assemblies = e.Argument as AssemblyDefinition[];
            int progress = 0;

            try
            {
                ProcessStrongNames(assemblies);
                ProcessReferences(assemblies);
                foreach (AssemblyDefinition asmdef in assemblies)
                {
                    progress++;
                    worker.ReportProgress((progress * 100) / assemblies.Length, asmdef.Name);
                    AssemblyFactory.SaveAssembly(asmdef, asmdef.MainModule.Image.FileInformation.FullName);
                }
            }
            catch (Exception ex)
            {
                worker.ReportProgress(0,ex);
            }
        }

        private void BackgroundWorker_ProgressChanged(object sender, ProgressChangedEventArgs e)
        {
            if (e.UserState is Exception)
            {
                MessageBox.Show(String.Format("Reflexil is unable to save this assembly: {0}", (e.UserState as Exception).Message));
            }
            else
            {
                ProgressBar.Value = e.ProgressPercentage;
                Assembly.Text = e.UserState.ToString();
            }
        }

        private void BackgroundWorker_RunWorkerCompleted(object sender, RunWorkerCompletedEventArgs e)
        {
            DialogResult = DialogResult.OK;
        }
        #endregion

        #region " Methods "
        public DialogResult ShowDialog(AssemblyDefinition[] asmdefs)
        {
            BackgroundWorker.RunWorkerAsync(asmdefs);
            return ShowDialog();
        }

        public ReferenceUpdaterForm()
		{
			InitializeComponent();
        }

        private void ProcessReferences(AssemblyDefinition[] assemblies)
        {
            foreach (AssemblyDefinition asmdef in assemblies)
            {
                ProcessReferences(asmdef, assemblies);
            }
        }

        private void ProcessReferences(AssemblyDefinition current, AssemblyDefinition[] assemblies)
        {
            foreach (AssemblyDefinition asmdef in assemblies)
            {
                if (!asmdef.Equals(current))
                {
                    ProcessReferences(current, asmdef);
                }
            }
        }

        private void ProcessReferences(AssemblyDefinition current, AssemblyDefinition other)
        {
            foreach (ModuleDefinition moddef in current.Modules)
            {
                foreach (AssemblyNameReference anref in moddef.AssemblyReferences)
                {
                    if (CecilHelper.ReferenceMatches(anref, other.Name))
                    {
                        CecilHelper.RemoveStrongNameReference(anref);
                    }
                }
            }
        }

        private void ProcessStrongNames(AssemblyDefinition[] assemblies)
        {
            foreach (AssemblyDefinition asmdef in assemblies)
            {
                CecilHelper.RemoveStrongName(asmdef);
            }
        }
        #endregion

	}
}

By viewing downloads associated with this article you agree to the Terms of Service and the article's licence.

If a file you wish to view isn't highlighted, and is a text file (not binary), please let us know and we'll add colourisation support for it.

License

This article, along with any associated source code and files, is licensed under The MIT License


Written By
Software Developer (Senior) Microsoft
United States United States
Sebastien Lebreton is a Software Engineer at Microsoft.

He is particularly interested in optimization, reverse engineering and distributed objects technologies.

Comments and Discussions