Как получить метод, фактически вызванный инструкцией callvirt IL в FxCop - PullRequest
Новая
       9

Как получить метод, фактически вызванный инструкцией callvirt IL в FxCop

7 голосов
/ 23 июня 2011

Я все еще пытаюсь заставить работать мое правило FxCop.

В рамках этого мне нужно выяснить, какие методы вызывает метод.Ранее я использовал CallGraph.CallersFor() (делаю это в обратном порядке, что в любом случае является моей конечной целью), однако, похоже, у меня та же проблема, которую я опишу ниже.

В качестве альтернативы использованию CallGraph классапопытался посетить все вызовы методов для создания словаря на основе этого кода: 

public override void VisitMethodCall(MethodCall call)
{
    Method CalledMethod = (call.Callee as MemberBinding).BoundMember as Method;
    // ....
}

Однако оказывается, что если вызываемый метод находится в производном классе, который переопределяет метод базового класса, то BoundMember - это метод базового класса, а не метод дочернего класса (который будет фактически вызываться).

Вопрос: Как я могу получить метод, который будетвызывается в случае инструкции IL callvirt в FxCop?

1 Ответ

2 голосов
/ 27 июня 2011

В моем случае оказывается, что мне это точно не нужно (что хорошо, потому что у меня нет точного ответа в этом случае).

Поскольку FxCop является статическим средством проверки, он никогда не может знать тип экземпляра объекта, на который указывает переменная, который может быть объявлен как базовый тип. Поэтому я считаю, что то, о чем я прошу, невозможно.

Мое решение здесь состоит в том, чтобы при построении дерева вызовов я добавил дополнительные ссылки для базового класса, «вызывающего» любые производные классы. Таким образом, если я вызываю метод в базовом классе, и метод производного класса может быть вызван, я могу таким образом следовать дереву вызовов.

См. Ниже мой класс, используемый классами правил FxCop: 

public class CallGraphBuilder : BinaryReadOnlyVisitor
{
    public Dictionary<TypeNode, List<TypeNode>> ChildTypes;

    public Dictionary<Method, List<Method>> CallersOfMethod;

    private Method _CurrentMethod;

    public CallGraphBuilder()
        : base()
    {
        CallersOfMethod = new Dictionary<Method, List<Method>>();
        ChildTypes = new Dictionary<TypeNode, List<TypeNode>>();
    }

    public override void VisitMethod(Method method)
    {
        _CurrentMethod = method;

        base.VisitMethod(method);
    }

    public void CreateTypesTree(AssemblyNode Assy)
    {
        foreach (var Type in Assy.Types)
        {
            if (Type.FullName != "System.Object")
            {
                TypeNode BaseType = Type.BaseType;

                if (BaseType != null && BaseType.FullName != "System.Object")
                {
                    if (!ChildTypes.ContainsKey(BaseType))
                        ChildTypes.Add(BaseType, new List<TypeNode>());

                    if (!ChildTypes[BaseType].Contains(Type))
                        ChildTypes[BaseType].Add(Type);
                }
            }
        }
    }

    public override void VisitMethodCall(MethodCall call)
    {
        Method CalledMethod = (call.Callee as MemberBinding).BoundMember as Method;

        AddCallerOfMethod(CalledMethod, _CurrentMethod);

        Queue<Method> MethodsToCheck = new Queue<Method>();

        MethodsToCheck.Enqueue(CalledMethod);

        while (MethodsToCheck.Count != 0)
        {
            Method CurrentMethod = MethodsToCheck.Dequeue();

            if (ChildTypes.ContainsKey(CurrentMethod.DeclaringType))
            {
                foreach (var DerivedType in ChildTypes[CurrentMethod.DeclaringType])
                {
                    var DerivedCalledMethod = DerivedType.Members.OfType<Method>().Where(M => MethodHidesMethod(M, CurrentMethod)).SingleOrDefault();

                    if (DerivedCalledMethod != null)
                    {
                        AddCallerOfMethod(DerivedCalledMethod, CurrentMethod);

                        MethodsToCheck.Enqueue(DerivedCalledMethod);
                    }
                }
            }
        }

        base.VisitMethodCall(call);
    }

    private void AddCallerOfMethod(Method CalledMethod, Method CallingMethod)
    {
        if (!CallersOfMethod.ContainsKey(CalledMethod))
            CallersOfMethod.Add(CalledMethod, new List<Method>());

        if (!CallersOfMethod[CalledMethod].Contains(CallingMethod))
            CallersOfMethod[CalledMethod].Add(CallingMethod);
    }

    private bool MethodHidesMethod(Method ChildMethod, Method BaseMethod)
    {
        while (ChildMethod != null)
        {
            if (ChildMethod == BaseMethod)
                return true;

            ChildMethod = ChildMethod.OverriddenMethod ?? ChildMethod.HiddenMethod;
        }

        return false;
    }
}
...