crokus 0.0.2

data/lib/crokus/ast_printer.rb

@@ -0,0 +1,86 @@
+require_relative 'ast'
+require_relative 'indent'
+require_relative 'code'
+require 'colorize'
+
+module Crokus
+
+  class AstPrinter
+
+    include Indent
+
+    attr_accessor :code,:nodes_decl,:nodes_cnx
+
+    def print ast #entry method
+      @verbose=false
+      @nodes_decl=Code.new
+      @nodes_cnx=Code.new
+      @printed_cnx={} #Cosmetic ! to keep track of already printed cnx source->sink
+      @code=Code.new
+      code << "digraph G {"
+      code.indent=2
+      code << "ordering=out;"
+      code << "ranksep=.4;"
+      code << "bgcolor=\"lightgrey\";"
+      code.newline
+      code << "node [shape=box, fixedsize=false, fontsize=12, fontname=\"Helvetica-bold\", fontcolor=\"blue\""
+      code << "       width=.25, height=.25, color=\"black\", fillcolor=\"white\", style=\"filled, solid, bold\"];"
+      code << "edge [arrowsize=.5, color=\"black\", style=\"bold\"]"
+      process(ast)
+      code << @nodes_decl
+      code << @nodes_cnx
+      code.indent=0
+      code << "}"
+      clean(code)
+      return code
+    end
+
+    def process node,level=0
+      #puts "processing #{node}"
+      kname=node.class.name.split("::")[1]
+      id=node.object_id
+      (nodes_decl << "#{id} [label=\"#{kname}\"]")
+
+      node.instance_variables.each{|vname|
+        ivar=node.instance_variable_get(vname)
+        vname=vname.to_s[1..-1]
+        if ivar
+          case ivar
+          when Array
+            ivar.each_with_index{|e,idx|
+              sink=process(e,level+2)
+              @printed_cnx[id]||=[]
+              nodes_cnx << "#{id} -> #{sink} [label=\"#{vname}[#{idx}]\"]" if not @printed_cnx[id].include? sink
+              @printed_cnx[id] << sink
+            }
+          when Token
+            val=ivar.val
+            sink="#{ivar.object_id}"
+            nodes_decl << "#{sink} [label=\"#{val}\",color=\"red\"]"
+            @printed_cnx[id]||=[]
+            nodes_cnx << "#{id} -> #{sink} [label=\"#{vname}\"]" if not @printed_cnx[id].include? sink
+            @printed_cnx[id] << sink
+          else
+            sink=process(ivar,level+2)
+            @printed_cnx[id]||=[]
+            nodes_cnx << "#{id} -> #{sink} [label=\"#{vname}\"]" if not @printed_cnx[id].include? sink
+            @printed_cnx[id] << sink
+          end
+        end
+      }
+      return id
+    end
+
+    # suppress syndrom : name=""not correct""
+    def clean code
+      #puts "=> cleaning dot code"
+      code.lines.each_with_index do |line,idx|
+        if line=~/\"\"(.*)\"\"/
+          code.lines[idx].gsub!(/\"\"(.*)\"\"/,"\"#{$1}\"")
+        end
+      end
+    end
+
+
+  end #class
+end #module

data/lib/crokus/cfg.rb

@@ -0,0 +1,81 @@
+require_relative 'ast'
+require_relative 'cfg_printer'
+
+
+module Crokus
+
+  class ITE < Ast
+    attr_accessor :cond,:trueBranch,:falseBranch
+    def initialize cond,bb_t,bb_f
+      @cond=cond
+      @trueBranch=bb_t
+      @falseBranch=bb_f
+    end
+  end
+
+  class CFG
+    attr_accessor :name,:bbs,:starter
+    def initialize name
+      @name=name
+      @bbs=[]
+      @bbs << @starter=BasicBlock.new
+    end
+
+    def size
+      @bbs.size
+    end
+
+    def <<(bb)
+      @bbs << bb
+    end
+
+    def print
+      CFGPrinter.new.print(self)
+    end
+  end
+
+  class BasicBlock
+    @@id=-1
+    attr_accessor :id,:stmts,:succs
+
+    alias :label :id
+
+    def initialize
+      @@id+=1
+      @id="L"+@@id.to_s
+      @stmts=[]
+      @succs=[]
+    end
+
+    def <<(e)
+      @stmts << e
+    end
+
+    def to bb
+      @succs << bb
+    end
+
+    def trueBranch
+      unless @succs.size==2
+        raise "request for trueBranch failed because #{@succs.size} branch(es) found. Strange."
+      end
+      return @succs.first
+    end
+
+    def falseBranch
+      unless @succs.size==2
+        raise "request for falseBranch failed because #{@succs.size} branch(es) found. Strange."
+      end
+      return @succs.last
+    end
+
+    def code4dot
+      @ppr||=PrettyPrinter.new
+      @stmts.compact.collect{|stmt| stmt.accept(@ppr)}.join("\n")
+    end
+
+    def size
+      @stmts.size
+    end
+  end
+end

data/lib/crokus/cfg_builder.rb

@@ -0,0 +1,169 @@
+require_relative 'cfg'
+require_relative 'cfg_cleaner'
+
+require_relative 'visitor'
+require_relative 'cleaner'
+
+module Crokus
+
+  class CFGBuilder < Transformer
+
+    def initialize
+      @ind=-2
+      @verbose=false
+    end
+
+    def build ast
+      ast.accept(self)
+    end
+
+    def visitFunction func,args=nil
+      puts "   |--> visitFunction '#{func.name}'" unless $options[:mute]
+      @cfg=CFG.new(func.name)
+      @current=@cfg.starter
+      func.body.accept(self)
+      @cfg.print
+      @cfg=CFGCleaner.new.clean(@cfg)
+      @cfg.name=Ident.new(Token.create "#{@cfg.name}_clean")
+      func.cfg=@cfg
+      puts "\t|--> cfg size for '#{func.name}' : #{@cfg.size}" unless $options[:mute]
+      @cfg.print
+    end
+
+    def visitBody body,args=nil
+      body.each{|stmt| stmt.accept(self,args)}
+    end
+    #...........stmts...............
+    def visitAssign assign,args=nil
+      @current << assign
+    end
+
+    def visitPreFixAccu accu,args=nil
+      @current << accu
+    end
+
+    def visitPostFixAccu accu,args=nil
+      @current << accu
+    end
+
+    def visitSwitch switch,args=nil
+      @cfg << finalBranch=BasicBlock.new
+      @current_break_dest=finalBranch
+      for cas in switch.cases
+        cond=Binary.new(switch.expr,EQUAL,cas.expr)
+        @cfg << trueBranch=BasicBlock.new
+        @cfg << falseBranch=BasicBlock.new
+
+        @current << ITE.new(cond,trueBranch,falseBranch)
+        @current.to trueBranch
+        @current.to falseBranch
+        @current = trueBranch
+        cas.body.accept(self)
+        @current = falseBranch
+      end
+      if switch.default
+        switch.default.accept(self)
+        @current.to finalBranch
+      end
+      @current=finalBranch
+    end
+
+    def visitBreak brk,args=nil
+      @current << brk
+      @current.to @current_break_dest
+      @cfg << unreachable = BasicBlock.new
+      @current = unreachable
+    end
+
+    def visitContinue cont,args=nil
+      @current << cont
+      @current.to @current_continue_dest
+      @cfg << unreachable = BasicBlock.new
+      @current = unreachable
+    end
+
+    def visitIf if_,args=nil
+      cond=if_.cond.accept(self,:as_expr)
+      @cfg << trueBranch =BasicBlock.new
+      @cfg << falseBranch=BasicBlock.new
+      @cfg << mergeBranch=BasicBlock.new
+      @current << ITE.new(cond,trueBranch,falseBranch)
+
+      @current.to trueBranch
+      @current.to falseBranch
+      #-----------
+      @current=trueBranch
+      if_.body.accept(self) #may change @current !
+      @current.to mergeBranch
+      #
+      @current=falseBranch
+      if_.else.accept(self) if if_.else #may change @current !
+      @current.to mergeBranch
+      @current=mergeBranch
+    end
+
+    def visitElse else_,args=nil
+      else_.body.accept(self)
+    end
+
+    def visitWhile while_,args=nil
+      cond=while_.cond.accept(self,:as_expr)
+      @cfg << cond_bb     = BasicBlock.new
+      @current_cond = cond_bb # for continue stmt !
+      @cfg << trueBranch  = BasicBlock.new
+      @cfg << falseBranch = BasicBlock.new
+      @current.to cond_bb
+      cond_bb << ITE.new(cond,trueBranch,falseBranch)
+      cond_bb.to trueBranch
+      cond_bb.to falseBranch
+      @current = trueBranch
+      while_.body.accept(self) #may modify identity of @current
+      @current.to cond_bb
+      @current=falseBranch
+    end
+
+    def visitFor for_,args=nil
+      for_.init.each{|stmt| stmt.accept(self)}
+      cond=for_.cond.accept(self)
+      @cfg << cond_bb     = BasicBlock.new
+      @cfg << trueBranch  = BasicBlock.new
+      @cfg << falseBranch = BasicBlock.new
+      @cfg << postBranch  = BasicBlock.new
+      @current_continue_dest = postBranch
+      @current_break_dest    = falseBranch
+      @current.to cond_bb
+      cond_bb << ITE.new(cond,trueBranch,falseBranch)
+      cond_bb.to trueBranch
+      cond_bb.to falseBranch
+      @current= trueBranch
+      for_.body.accept(self) #may modify @current identity
+      @current.to postBranch
+      @current=postBranch
+      for_.increment.accept(self)
+      @current.to cond_bb
+      @current=falseBranch
+    end
+
+    def visitDoWhile dowhile,args=nil
+      @cfg << cond_bb     = BasicBlock.new
+      @current_continue_dest = cond_bb # for continue stmt !
+      @cfg << trueBranch  = BasicBlock.new
+      @cfg << falseBranch = BasicBlock.new
+
+      @current.to trueBranch
+      @current = trueBranch
+      dowhile.body.accept(self) # may modify @current
+
+      @current.to cond_bb
+      @current = cond_bb
+      cond=dowhile.cond.accept(self)
+
+      cond_bb << ITE.new(cond,trueBranch,falseBranch)
+
+      cond_bb.to trueBranch
+      cond_bb.to falseBranch
+      @current = falseBranch
+    end
+
+  end #class Visitor
+end #module

data/lib/crokus/cfg_cleaner.rb

@@ -0,0 +1,62 @@
+module Crokus
+
+  class CFGCleaner
+
+    def clean cfg
+      puts "\t|--> cleaning '#{cfg.name}'" unless $options[:mute]
+      @cfg=cfg
+      @visited=[]
+      @new_succs={}
+      clean_rec cfg.starter
+      update
+      rename
+      cfg
+    end
+
+    private
+
+    def clean_rec bb
+      @visited << bb
+      @new_succs[bb]=[]
+      bb.succs.each_with_index do |succ,idx|
+        cand=find_next_rec(succ)
+        if cand
+          #puts "  |--> #{bb.label} next #{idx} is #{cand.label}"
+          @new_succs[bb] << cand
+          unless @visited.include? cand
+            clean_rec cand
+          end
+        end
+      end
+    end
+
+    def find_next_rec bb
+      if bb.size>0
+        return bb
+      else
+        if bb.succs.any?
+          @cfg.bbs.delete(bb)
+          return find_next_rec(bb.succs.first)
+        else
+          return bb #ending
+        end
+      end
+    end
+
+    def update
+      @cfg.bbs.each{|bb| bb.succs=@new_succs[bb]}
+      @cfg.bbs.each{|bb|
+        if (ite=bb.stmts.last).is_a? ITE
+          ite.trueBranch=bb.succs.first
+          ite.falseBranch=bb.succs.last
+        end
+      }
+    end
+
+    def rename
+      @cfg.bbs.each_with_index do |bb,idx|
+        bb.id="L#{idx}"
+      end
+    end
+  end
+end

data/lib/crokus/cfg_printer.rb

@@ -0,0 +1,73 @@
+require_relative 'code'
+
+module Crokus
+
+  class CFGPrinter
+
+    attr_accessor :code
+
+    def print cfg
+      @code=Code.new
+      @code << header
+      @visited=[]
+      visitRec(cfg.starter)
+      @code << footer
+      dot_name="cfg_#{cfg.name}.dot"
+      @code.save_as dot_name
+      puts "\t|--> graphviz file saved as '#{dot_name}'" unless $options[:mute]
+    end
+
+    def header
+      ret=Code.new
+      ret << "digraph ControlFlowGraph {"
+      ret.indent=2
+      ret << 'forcelabels=true;'
+      ret << 'graph [ label="",'
+      ret << '        bgcolor="white",'
+      ret << '        fontname="Arail",'
+      ret << '        rankdir="TB"]'
+      ret.newline
+      ret << 'node  [ fontname="Arial",'
+      ret << '            shape="box",'
+      ret << '            style="filled",'
+      ret << '            fillcolor="AliceBlue"]'
+      ret.newline
+      ret << 'edge  [ fontname="Arial",'
+      ret << '        color="Blue",'
+      ret << '        dir="forward"]'
+      ret.newline
+      ret
+    end
+
+    def footer
+      "}"
+    end
+
+    def visitRec bb
+      while !@visited.include?(bb)
+        @visited << bb
+        c_code=bb.code4dot
+        c_code=clean4dot(c_code)
+        code << "#{id(bb)} [label=\"#{c_code}\",shape=rectangle, xlabel=#{bb.label}]"
+        bb.succs.each_with_index do |succ,idx|
+          if bb.succs.size>1
+            label = (idx==0 ? "true" : "false")
+            label = "[label=#{label}]"
+          end
+          code << "#{id(bb)} -> #{id(succ)} #{label}"
+          visitRec(succ)
+        end
+      end
+    end
+
+    def id bb
+      "bb_#{bb.id.to_s}"
+    end
+
+    def clean4dot str
+      str=str.gsub(/\\n\"/,'"')
+      str=str.gsub(/"/,'\"')
+    end
+
+  end
+end