hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2025-12-18 01:33:15 +01:00
Code Issues Packages Projects Releases Wiki Activity

PS: Copy extractor and various scripts from internal repo to public repo.

Browse Source
This commit is contained in:
Mathias Vorreiter Pedersen
2024-08-13 19:56:20 +01:00
parent d4c0c4059d
commit 58fc649657
192 changed files with 15205 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

122
powershell/extractor/Semmle.Extraction.PowerShell/Entities/ScriptBlockEntity.cs Normal file
View File

@@ -0,0 +1,122 @@
using System;
using System.IO;
using System.Management.Automation.Language;
namespace Semmle.Extraction.PowerShell.Entities
{
internal class ScriptBlockEntity : CachedEntity<(ScriptBlockAst, ScriptBlockAst)>
{
private ScriptBlockEntity(PowerShellContext cx, ScriptBlockAst fragment)
: base(cx, (fragment, fragment))
{
}
public override Microsoft.CodeAnalysis.Location ReportingLocation => PowerShellContext.CreateAnalysisLocation(Fragment);
public ScriptBlockAst Fragment => Symbol.Item1;
public override void Populate(TextWriter trapFile)
{
trapFile.script_block(this, Fragment.UsingStatements.Count, Fragment.ScriptRequirements?.RequiredModules.Count ?? 0, Fragment.ScriptRequirements?.RequiredAssemblies.Count ?? 0, Fragment.ScriptRequirements?.RequiredPSEditions.Count ?? 0, Fragment.ScriptRequirements?.RequiresPSSnapIns.Count ?? 0);
trapFile.script_block_location(this, TrapSuitableLocation);
if (Fragment.ScriptRequirements is not null){
trapFile.script_block_requires_elevation(this, Fragment.ScriptRequirements.IsElevationRequired);
if (Fragment.ScriptRequirements.RequiredApplicationId is not null)
{
trapFile.script_block_required_application_id(this, Fragment.ScriptRequirements.RequiredApplicationId);
}
if (Fragment.ScriptRequirements.RequiredPSVersion is not null)
{
trapFile.script_block_required_ps_version(this, Fragment.ScriptRequirements.RequiredPSVersion.ToString());
}
for(int i = 0; i < Fragment.ScriptRequirements.RequiredModules.Count; i++)
{
var theModule = ModuleSpecificationEntity.Create(PowerShellContext, Fragment.ScriptRequirements.RequiredModules[i], ReportingLocation);
trapFile.script_block_required_module(this, i, theModule);
}
for (int i = 0; i < Fragment.ScriptRequirements.RequiredAssemblies.Count; i++)
{
trapFile.script_block_required_assembly(this, i, Fragment.ScriptRequirements.RequiredAssemblies[i]);
}
for (int i = 0; i < Fragment.ScriptRequirements.RequiredPSEditions.Count; i++)
{
trapFile.script_block_required_ps_edition(this, i, Fragment.ScriptRequirements.RequiredPSEditions[i]);
}
for (int i = 0; i < Fragment.ScriptRequirements.RequiresPSSnapIns.Count; i++)
{
trapFile.script_block_requires_ps_snapin(this, i, Fragment.ScriptRequirements.RequiresPSSnapIns[i].Name, Fragment.ScriptRequirements.RequiresPSSnapIns[i].Version.ToString());
}
}
if (Fragment.ParamBlock is not null)
{
trapFile.script_block_param_block(this,
EntityConstructor.ConstructAppropriateEntity(PowerShellContext, Fragment.ParamBlock));
}
if (Fragment.BeginBlock is not null)
{
trapFile.script_block_begin_block(this,
EntityConstructor.ConstructAppropriateEntity(PowerShellContext, Fragment.BeginBlock));
}
if (Fragment.CleanBlock is not null)
{
trapFile.script_block_clean_block(this,
EntityConstructor.ConstructAppropriateEntity(PowerShellContext, Fragment.CleanBlock));
}
if (Fragment.DynamicParamBlock is not null)
{
trapFile.script_block_dynamic_param_block(this,
EntityConstructor.ConstructAppropriateEntity(PowerShellContext, Fragment.DynamicParamBlock));
}
if (Fragment.EndBlock is not null)
{
trapFile.script_block_end_block(this,
EntityConstructor.ConstructAppropriateEntity(PowerShellContext, Fragment.EndBlock));
}
if (Fragment.ProcessBlock is not null)
{
trapFile.script_block_process_block(this,
EntityConstructor.ConstructAppropriateEntity(PowerShellContext, Fragment.ProcessBlock));
}
// TODO: Fragment.Requirements, need a non-null example
for (int index = 0; index < Fragment.UsingStatements.Count; index++)
{
trapFile.script_block_using(this, index,
EntityConstructor.ConstructAppropriateEntity(PowerShellContext, Fragment.UsingStatements[index]));
}
if (Fragment.Parent is not null)
{
trapFile.parent(this, EntityConstructor.ConstructAppropriateEntity(PowerShellContext, Fragment.Parent));
}
}
public override bool NeedsPopulation => true;
public override void WriteId(EscapingTextWriter trapFile)
{
trapFile.WriteSubId(TrapSuitableLocation);
trapFile.Write(";script_block");
}
internal static ScriptBlockEntity Create(PowerShellContext cx, ScriptBlockAst fragment)
{
var init = (fragment, fragment);
return ScriptBlockEntityFactory.Instance.CreateEntity(cx, init, init);
}
private class ScriptBlockEntityFactory : CachedEntityFactory<(ScriptBlockAst, ScriptBlockAst), ScriptBlockEntity>
{
public static ScriptBlockEntityFactory Instance { get; } = new ScriptBlockEntityFactory();
public override ScriptBlockEntity Create(PowerShellContext cx, (ScriptBlockAst, ScriptBlockAst) init) =>
new ScriptBlockEntity(cx, init.Item1);
}
public override TrapStackBehaviour TrapStackBehaviour => TrapStackBehaviour.OptionalLabel;
}
}