public List <string> GetGPOToInvestigate()
{
List <string> output = new List <string>();
output.AddRange(GPOToInvestigate);
GPOToInvestigate.Clear();
return(output);
}
void AddDataToInvestigate(string mapping, MappingType type)
{
// avoid dealing with files
if (string.IsNullOrEmpty(mapping))
{
Trace.WriteLine("Ignoring addition of mapping " + mapping + "type = " + type);
return;
}
else
{
switch (type)
{
case MappingType.DistinguishedName:
if (!nodesByDN.ContainsKey(mapping.ToLowerInvariant()))
{
if (!CNToInvestigate.Contains(mapping.ToLowerInvariant()))
{
CNToInvestigate.Add(mapping.ToLowerInvariant());
}
}
break;
case MappingType.Sid:
if (mapping.StartsWith("S-1-5-32-", StringComparison.InvariantCultureIgnoreCase) || mapping.StartsWith(databaseInformation["DomainSid"]))
{
if (!nodesBySID.ContainsKey(mapping.ToLowerInvariant()))
{
if (!SIDToInvestigate.Contains(mapping.ToUpperInvariant()))
{
SIDToInvestigate.Add(mapping.ToUpperInvariant());
}
}
}
break;
case MappingType.FileName:
if (mapping.StartsWith("\\\\"))
{
if (!nodesByFilename.ContainsKey(mapping.ToLowerInvariant()))
{
if (!FilesToInvestigate.Contains(mapping.ToLowerInvariant()))
{
FilesToInvestigate.Add(mapping.ToLowerInvariant());
}
}
}
break;
case MappingType.GPODirectory:
if (mapping.StartsWith("\\\\"))
{
if (!nodesGPOByFilename.ContainsKey(mapping.ToLowerInvariant()))
{
if (!GPOToInvestigate.Contains(mapping.ToLowerInvariant()))
{
GPOToInvestigate.Add(mapping.ToLowerInvariant());
}
}
}
break;
}
}
}
int CreateNode(Node node)
{
//locking is important because this function can be called by threads (analysing GPO, ...)
lock (nodes)
{
Trace.WriteLine("Inserting node " + index + " name=" + node.Name + " sid=" + node.Sid + " shortname=" + node.Shortname);
// defensive programming checks
if (!string.IsNullOrEmpty(node.Dn) && nodesByDN.ContainsKey(node.Dn.ToLowerInvariant()))
{
Trace.WriteLine("DN already present");
return(nodesByDN[node.Dn.ToLowerInvariant()].Id);
}
if (!string.IsNullOrEmpty(node.Sid) && nodesBySID.ContainsKey(node.Sid.ToLowerInvariant()))
{
Trace.WriteLine("SID already present");
return(nodesBySID[node.Sid.ToLowerInvariant()].Id);
}
if (!string.IsNullOrEmpty(node.FileName) && node.Type != "gpodirectory" && (nodesByFilename.ContainsKey(node.FileName.ToLowerInvariant())))
{
Trace.WriteLine("FileName already present");
return(nodesByFilename[node.FileName.ToLowerInvariant()].Id);
}
if (!string.IsNullOrEmpty(node.FileName) && node.Type == "gpodirectory" && (nodesGPOByFilename.ContainsKey(node.FileName.ToLowerInvariant())))
{
Trace.WriteLine("FileName already present");
return(nodesGPOByFilename[node.FileName.ToLowerInvariant()].Id);
}
// inserting the node
node.Id = index;
nodes.Add(index, node);
if (!string.IsNullOrEmpty(node.FileName))
{
if (node.FileName.StartsWith("\\\\"))
{
if (node.Type != "gpodirectory")
{
nodesByFilename.Add(node.FileName.ToLowerInvariant(), node);
if (FilesToInvestigate.Contains(node.FileName.ToLowerInvariant()))
{
FilesToInvestigate.Remove(node.FileName.ToLowerInvariant());
}
}
else
{
nodesGPOByFilename.Add(node.FileName.ToLowerInvariant(), node);
if (GPOToInvestigate.Contains(node.FileName.ToLowerInvariant()))
{
GPOToInvestigate.Remove(node.FileName.ToLowerInvariant());
}
}
}
}
if (!string.IsNullOrEmpty(node.Dn))
{
nodesByDN.Add(node.Dn.ToLowerInvariant(), node);
if (CNToInvestigate.Contains(node.Dn.ToLowerInvariant()))
{
CNToInvestigate.Remove(node.Dn.ToLowerInvariant());
}
}
if (!string.IsNullOrEmpty(node.Sid))
{
nodesBySID.Add(node.Sid.ToLowerInvariant(), node);
if (SIDToInvestigate.Contains(node.Sid.ToUpperInvariant()))
{
SIDToInvestigate.Remove(node.Sid.ToUpperInvariant());
}
// handle primary group id
if (node.Type == "group")
{
if (node.Sid.StartsWith("S-1-5-21-"))
{
var part = node.Sid.Split('-');
int PGId = int.Parse(part[part.Length - 1]);
if (!KnownPGId.Contains(PGId) && !PGIdToInvestigate.Contains(PGId))
{
PGIdToInvestigate.Add(PGId);
}
}
}
}
return(index++);
}
}
