moel@1: /*
moel@1:   
moel@1:   Version: MPL 1.1/GPL 2.0/LGPL 2.1
moel@1: 
moel@1:   The contents of this file are subject to the Mozilla Public License Version
moel@1:   1.1 (the "License"); you may not use this file except in compliance with
moel@1:   the License. You may obtain a copy of the License at
moel@1:  
moel@1:   http://www.mozilla.org/MPL/
moel@1: 
moel@1:   Software distributed under the License is distributed on an "AS IS" basis,
moel@1:   WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
moel@1:   for the specific language governing rights and limitations under the License.
moel@1: 
moel@1:   The Original Code is the Open Hardware Monitor code.
moel@1: 
moel@1:   The Initial Developer of the Original Code is 
moel@1:   Michael Möller <m.moeller@gmx.ch>.
moel@1:   Portions created by the Initial Developer are Copyright (C) 2009-2010
moel@1:   the Initial Developer. All Rights Reserved.
moel@1: 
moel@1:   Contributor(s):
moel@1: 
moel@1:   Alternatively, the contents of this file may be used under the terms of
moel@1:   either the GNU General Public License Version 2 or later (the "GPL"), or
moel@1:   the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
moel@1:   in which case the provisions of the GPL or the LGPL are applicable instead
moel@1:   of those above. If you wish to allow use of your version of this file only
moel@1:   under the terms of either the GPL or the LGPL, and not to allow others to
moel@1:   use your version of this file under the terms of the MPL, indicate your
moel@1:   decision by deleting the provisions above and replace them with the notice
moel@1:   and other provisions required by the GPL or the LGPL. If you do not delete
moel@1:   the provisions above, a recipient may use your version of this file under
moel@1:   the terms of any one of the MPL, the GPL or the LGPL.
moel@1:  
moel@1: */
moel@1: 
moel@1: using System;
moel@1: using System.Collections.Generic;
moel@1: using System.Reflection;
moel@1: using System.Reflection.Emit;
moel@1: using System.Runtime.InteropServices;
moel@1: 
moel@1: namespace OpenHardwareMonitor.Hardware {
moel@1: 
moel@1:   public sealed class PInvokeDelegateFactory {
moel@1: 
moel@1:     private static AssemblyBuilder assemblyBuilder;
moel@1:     private static ModuleBuilder moduleBuilder;
moel@1: 
moel@1:     private static IDictionary<DllImportAttribute, Type> wrapperTypes =
moel@1:       new Dictionary<DllImportAttribute, Type>();
moel@1: 
moel@1:     static PInvokeDelegateFactory() {
moel@1: 
moel@1:       AssemblyName assemblyName = new AssemblyName();
moel@1:       assemblyName.Name = "PInvokeDelegateFactoryInternalAssembly";
moel@1: 
moel@1:       assemblyBuilder =
moel@1:         AppDomain.CurrentDomain.DefineDynamicAssembly(assemblyName,
moel@1:         AssemblyBuilderAccess.Run);
moel@1: 
moel@1:       moduleBuilder = assemblyBuilder.DefineDynamicModule(
moel@1:         "PInvokeDelegateFactoryInternalModule");
moel@1:     }
moel@1: 
moel@1:     private PInvokeDelegateFactory() { }
moel@1: 
moel@1:     public static void CreateDelegate<T>(DllImportAttribute dllImportAttribute,
moel@1:       out T newDelegate) where T : class 
moel@1:     {
moel@1:       Type wrapperType;
moel@1:       wrapperTypes.TryGetValue(dllImportAttribute, out wrapperType);
moel@1: 
moel@1:       if (wrapperType == null) {
moel@1:         wrapperType = CreateWrapperType(typeof(T), dllImportAttribute);
moel@1:         wrapperTypes.Add(dllImportAttribute, wrapperType);
moel@1:       }
moel@1: 
moel@1:       newDelegate = Delegate.CreateDelegate(typeof(T), wrapperType,
moel@1:         dllImportAttribute.EntryPoint) as T;
moel@1:     }
moel@1: 
moel@1: 
moel@1:     private static Type CreateWrapperType(Type delegateType,
moel@1:       DllImportAttribute dllImportAttribute) {
moel@1: 
moel@1:       TypeBuilder typeBuilder = moduleBuilder.DefineType(
moel@1:         "PInvokeDelegateFactoryInternalWrapperType" + wrapperTypes.Count);
moel@1: 
moel@1:       MethodInfo methodInfo = delegateType.GetMethod("Invoke");
moel@1: 
moel@1:       ParameterInfo[] parameterInfos = methodInfo.GetParameters();
moel@1:       int parameterCount = parameterInfos.GetLength(0);
moel@1: 
moel@1:       Type[] parameterTypes = new Type[parameterCount];
moel@1:       for (int i = 0; i < parameterCount; i++)
moel@1:         parameterTypes[i] = parameterInfos[i].ParameterType;
moel@1: 
moel@1:       MethodBuilder methodBuilder = typeBuilder.DefinePInvokeMethod(
moel@1:         dllImportAttribute.EntryPoint, dllImportAttribute.Value,
moel@1:         MethodAttributes.Public | MethodAttributes.Static |
moel@1:         MethodAttributes.PinvokeImpl, CallingConventions.Standard,
moel@1:         methodInfo.ReturnType, parameterTypes,
moel@1:         dllImportAttribute.CallingConvention,
moel@1:         dllImportAttribute.CharSet);
moel@1: 
moel@1:       foreach (ParameterInfo parameterInfo in parameterInfos)
moel@1:         methodBuilder.DefineParameter(parameterInfo.Position + 1,
moel@1:           parameterInfo.Attributes, parameterInfo.Name);
moel@1: 
moel@1:       if (dllImportAttribute.PreserveSig)
moel@1:         methodBuilder.SetImplementationFlags(MethodImplAttributes.PreserveSig);
moel@1: 
moel@1:       return typeBuilder.CreateType();
moel@1:     }
moel@1:   }
moel@1: }