public void RegisterUnit(IUnit unit)
{
if (!this.unit2ContractExtractor.ContainsKey(unit.UnitIdentity))
{
var contractMethods = new ContractMethods(this);
var lazyContractProviderForLoadedUnit = new LazyContractExtractor(this, unit, contractMethods, false);
var codeContractsExtractor = new CodeContractsContractExtractor(this, lazyContractProviderForLoadedUnit);
this.unit2ContractExtractor.Add(unit.UnitIdentity, codeContractsExtractor);
}
}
/// <summary>
/// The main entry point for the program
/// </summary>
/// <param name="args">An assembly to be loaded by the program</param>
public static void Main(string[] args)
{
var options = new Options();
options.Parse(args);
if (options.HasErrors)
{
if (options.HelpRequested)
{
options.PrintOptions("");
}
Environment.Exit(-1);
}
HostEnvironment host = new HostEnvironment();
IModule module = host.LoadUnitFrom(options.assembly) as IModule;
if (module == null || module == Dummy.Module || module == Dummy.Assembly)
{
Console.WriteLine("'{0}' is not a PE file containing a CLR module or assembly.", options.assembly);
return;
}
if (options.outToFile)
{
StreamWriter consoleStream = new StreamWriter("buildOutput.txt");
Console.SetOut(consoleStream);
}
string xmlDocSaveName;
XDocument xDoc = null; //TODO: Use XmlReader instead of LINQ to XML
XElement membersElement = null;
if (!String.IsNullOrEmpty(options.xmlFile))
{
//Load the XML File
try
{
xDoc = XDocument.Load(options.xmlFile);
}
catch
{
Console.WriteLine(options.xmlFile + " is not a XML file.");
return;
}
var docEl = xDoc.Element("doc"); //Navigate to "doc"
if (docEl == null)
{
Console.WriteLine(options.xmlFile + " is not a valid XML reference file; it does not contain a \"doc\" element.");
return;
}
membersElement = docEl.Element("members"); //Navigate to "members"
if (membersElement == null)
{
Console.WriteLine(options.xmlFile + " is not a valid XML reference file; it does not contain a \"members\" element.");
return;
}
xmlDocSaveName = options.xmlFile;
}
else
{
//Build a new XML File
XDeclaration xDeclaration = new XDeclaration("1.0", null, null);
membersElement = new XElement("members");
xDoc = new XDocument(xDeclaration,
new XElement("doc",
membersElement));
// membersElement = xDoc.Element("doc").Element("members");
string fileName = options.assembly;
fileName = fileName.TrimEnd(".dll".ToCharArray());
xmlDocSaveName = fileName + ".xml";
}
//Establish the traverser
var contractMethods = new ContractMethods(host);
IContractProvider contractProvider = new Microsoft.Cci.ILToCodeModel.LazyContractProvider(host, module, contractMethods);
IMetadataVisitor traverser = new ContractTraverser(host, contractProvider, membersElement, options);
traverser.Visit(module);
xDoc.Save(xmlDocSaveName, SaveOptions.None);
Console.Out.Close();
}
