예제 #1
0
        MethodReference IImportMapper.MapMethod(MethodReference mref)
        {
            if (mref.DeclaringType.Scope.Name == _targetModule.Name)
                return mref;

            foreach (var cInj in _classInjectees.Where(i => i is TypeTargetInjecteeBase))
            {
                var injCecilType = cInj.GetInjecteeCecilType(_injectionModule);
                if (!mref.DeclaringType.TypeMatches(injCecilType))
                    continue;

                var classInjectee = cInj as TypeInjectee;
                var classStubInjectee = cInj as TypeStubInjectee;

                if (classInjectee != null)
                {
                    foreach (var inj in classInjectee.Injectees.Where(i => i is MethodInjecteeBase).Cast<MethodInjecteeBase>())
                    {
                        if (!mref.MethodMatches(inj.InjecteeMethod))
                            continue;

                        if (inj is MethodStubInjectee)
                        {
                            if (((MethodStubInjectee)inj).Attr.Mode != MethodStubMode.Redirected)
                                continue;
                        }

                        if (inj is MethodTargetInjecteeBase)
                        {
                            var target = ((MethodTargetInjecteeBase)inj).GetTargetCecilDefintion(_targetModule);
                            return FixMethodRedirecting(_targetModule, mref, target);
                        }
                    }
                }
                else if (classStubInjectee != null)
                {
                    var stubTargetClass = classStubInjectee.TargetType;
                    var tdef = _targetModule.FindMatchingType(stubTargetClass);
                    if (tdef == null)
                        throw new InvalidOperationException("Type not found for: " + stubTargetClass.Name);

                    var target = tdef.FindMatchingMethod(mref);
                    if (target == null)
                        throw new InvalidOperationException("Target method not found for: " + stubTargetClass.Name);

                    return FixMethodRedirecting(_targetModule, mref, target);
                }
            }

            return null;
        }