protected override ILExpression VisitExpression(ILExpression expression) { if (expression.Code == ILCode.Stloc) { var variable = (ILVariable)expression.Operand; if (variable.Type.HasInterface("System.IDisposable")) { var key = Tuple.Create(variable, currentScope); if (starts.Keys.Any(k => k.Item1 == variable && k.Item2 != currentScope)) { LogTo.Warning("Method {0}: Using cannot be added because reassigning a variable in a condition is not supported.", method); } else { if (starts.ContainsKey(key)) { UsingRanges.Add(new ILRange { From = starts[key], To = expression.FirstILOffset() }); starts.Remove(key); } if (!currentTrys.Contains(expression.LastILOffset())) { starts.Add(key, expression.LastILOffset()); } } } } if (expression.Code == ILCode.Ret && currentScope > 1) { EarlyReturns.Add(expression.FirstILOffset()); } return(base.VisitExpression(expression)); }
protected override ILBlock VisitBlock(ILBlock block) { currentScope++; var result = base.VisitBlock(block); currentScope--; if (block.Body.Count == 0) { return(result); } var toOffset = block.LastILOffset(); if (toOffset < 0) { return(result); } foreach (var start in starts.Where(kvp => kvp.Key.Item2 == currentScope + 1).ToList()) { starts.Remove(start.Key); List <ILExpression> args; if (block.Body.Last().Match(ILCode.Ret, out args) && args.Count == 1 && args[0].MatchLdloc(start.Key.Item1)) { continue; // Returning the variable } UsingRanges.Add(new ILRange { From = start.Value, To = toOffset }); } return(result); }