inline inject-tracer.ps1

github · Aug 26, 2020 · e9bfa56 · e9bfa56
1 parent 1f2bd10
commit e9bfa56
Showing 4 changed files with 26 additions and 12 deletions.
diff --git a/lib/init.js b/lib/init.js
diff --git a/lib/init.js.map b/lib/init.js.map
diff --git a/src/init.ts b/src/init.ts
@@ -78,10 +78,22 @@ export async function runInit(
   const tracerConfig = await getCombinedTracerConfig(config, codeql);
   if (tracerConfig !== undefined) {
     if (process.platform === 'win32') {
+      const injectTracerPath = path.join(config.tempDir, 'inject-tracer.ps1');
+      fs.writeFileSync(injectTracerPath, `
+        Param(
+            [Parameter(Position=0)]
+            [String]
+            $tracer
+        )
+        Get-Process -Name Runner.Worker
+        $process=Get-Process -Name Runner.Worker
+        $id=$process.Id
+        Invoke-Expression "&$tracer --inject=$id"`);
+
       await exec.exec(
         'powershell',
         [
-          path.resolve(__dirname, '..', 'src', 'inject-tracer.ps1'),
+          injectTracerPath,
           path.resolve(path.dirname(codeql.getPath()), 'tools', 'win64', 'tracer.exe'),
         ],
         { env: { 'ODASA_TRACER_CONFIGURATION': tracerConfig.spec } });

diff --git a/src/inject-tracer.ps1 b/src/inject-tracer.ps1