• Home
• io.joern
• x2cpg_3
• 4.0.106
• source code
• XImportsPass.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

io.joern.x2cpg.passes.frontend.XImportsPass.scala Maven / Gradle / Ivy

package io.joern.x2cpg.passes.frontend

import io.joern.x2cpg.Imports.createImportNodeAndLink
import io.shiftleft.codepropertygraph.generated.Cpg
import io.shiftleft.codepropertygraph.generated.nodes.Call
import io.shiftleft.passes.ForkJoinParallelCpgPass
import io.shiftleft.semanticcpg.language.*
import io.shiftleft.semanticcpg.language.operatorextension.OpNodes.Assignment

abstract class XImportsPass(cpg: Cpg) extends ForkJoinParallelCpgPass[(Call, Assignment)](cpg) {

  protected val importCallName: String

  override def generateParts(): Array[(Call, Assignment)] = cpg
    .call(importCallName)
    .flatMap(importCallToPart)
    .toArray

  protected def importCallToPart(x: Call): Iterator[(Call, Assignment)]

  override def runOnPart(diffGraph: DiffGraphBuilder, part: (Call, Assignment)): Unit = {
    val (call, assignment) = part
    val importedEntity     = importedEntityFromCall(call)
    val importedAs         = assignment.target.code
    createImportNodeAndLink(importedEntity, importedAs, Some(call), diffGraph)
  }

  protected def importedEntityFromCall(call: Call): String

}