3 import Analysis.CallGraph.*;
4 import Analysis.OwnershipAnalysis.*;
12 public class MLPAnalysis {
14 // data from the compiler
16 private TypeUtil typeUtil;
17 private CallGraph callGraph;
18 private OwnershipAnalysis ownAnalysis;
21 // an implicit SESE is automatically spliced into
22 // the IR graph around the C main before this analysis--it
23 // is nothing special except that we can make assumptions
24 // about it, such as the whole program ends when it ends
25 private FlatSESEEnterNode mainSESE;
27 // SESEs that are the root of an SESE tree belong to this
28 // set--the main SESE is always a root, statically SESEs
29 // inside methods are a root because we don't know how they
30 // will fit into the runtime tree of SESEs
31 private Set<FlatSESEEnterNode> rootSESEs;
33 // simply a set of every reachable SESE in the program, not
34 // including caller placeholder SESEs
35 private Set<FlatSESEEnterNode> allSESEs;
38 // A mapping of flat nodes to the stack of SESEs for that node, where
39 // an SESE is the child of the SESE directly below it on the stack.
40 // These stacks do not reflect the heirarchy over methods calls--whenever
41 // there is an empty stack it means all variables are available.
42 private Hashtable< FlatNode, Stack<FlatSESEEnterNode> > seseStacks;
44 private Hashtable< FlatNode, Set<TempDescriptor> > livenessRootView;
45 private Hashtable< FlatNode, Set<TempDescriptor> > livenessVirtualReads;
46 private Hashtable< FlatNode, VarSrcTokTable > variableResults;
47 private Hashtable< FlatNode, Set<TempDescriptor> > notAvailableResults;
48 private Hashtable< FlatNode, CodePlan > codePlans;
50 private Hashtable< FlatEdge, FlatWriteDynamicVarNode > wdvNodesToSpliceIn;
52 private Hashtable< MethodContext, HashSet<AllocationSite>> mapMethodContextToLiveInAllocationSiteSet;
54 public static int maxSESEage = -1;
57 // use these methods in BuildCode to have access to analysis results
58 public FlatSESEEnterNode getMainSESE() {
62 public Set<FlatSESEEnterNode> getRootSESEs() {
66 public Set<FlatSESEEnterNode> getAllSESEs() {
70 public int getMaxSESEage() {
75 public CodePlan getCodePlan( FlatNode fn ) {
76 CodePlan cp = codePlans.get( fn );
81 public MLPAnalysis( State state,
84 OwnershipAnalysis ownAnalysis
87 double timeStartAnalysis = (double) System.nanoTime();
91 this.callGraph = callGraph;
92 this.ownAnalysis = ownAnalysis;
93 this.maxSESEage = state.MLP_MAXSESEAGE;
95 rootSESEs = new HashSet<FlatSESEEnterNode>();
96 allSESEs = new HashSet<FlatSESEEnterNode>();
98 seseStacks = new Hashtable< FlatNode, Stack<FlatSESEEnterNode> >();
99 livenessRootView = new Hashtable< FlatNode, Set<TempDescriptor> >();
100 livenessVirtualReads = new Hashtable< FlatNode, Set<TempDescriptor> >();
101 variableResults = new Hashtable< FlatNode, VarSrcTokTable >();
102 notAvailableResults = new Hashtable< FlatNode, Set<TempDescriptor> >();
103 codePlans = new Hashtable< FlatNode, CodePlan >();
104 wdvNodesToSpliceIn = new Hashtable< FlatEdge, FlatWriteDynamicVarNode >();
106 mapMethodContextToLiveInAllocationSiteSet = new Hashtable< MethodContext, HashSet<AllocationSite>>();
109 FlatMethod fmMain = state.getMethodFlat( typeUtil.getMain() );
111 mainSESE = (FlatSESEEnterNode) fmMain.getNext(0);
112 mainSESE.setfmEnclosing( fmMain );
113 mainSESE.setmdEnclosing( fmMain.getMethod() );
114 mainSESE.setcdEnclosing( fmMain.getMethod().getClassDesc() );
118 // run analysis on each method that is actually called
119 // reachability analysis already computed this so reuse
120 Iterator<Descriptor> methItr = ownAnalysis.descriptorsToAnalyze.iterator();
121 while( methItr.hasNext() ) {
122 Descriptor d = methItr.next();
123 FlatMethod fm = state.getMethodFlat( d );
125 // find every SESE from methods that may be called
126 // and organize them into roots and children
127 buildForestForward( fm );
131 // 2nd pass, results are saved in FlatSESEEnterNode, so
132 // intermediate results, for safety, are discarded
133 Iterator<FlatSESEEnterNode> rootItr = rootSESEs.iterator();
134 while( rootItr.hasNext() ) {
135 FlatSESEEnterNode root = rootItr.next();
136 livenessAnalysisBackward( root,
143 methItr = ownAnalysis.descriptorsToAnalyze.iterator();
144 while( methItr.hasNext() ) {
145 Descriptor d = methItr.next();
146 FlatMethod fm = state.getMethodFlat( d );
148 // starting from roots do a forward, fixed-point
149 // variable analysis for refinement and stalls
150 variableAnalysisForward( fm );
153 // 4th pass, compute liveness contribution from
154 // virtual reads discovered in variable pass
155 rootItr = rootSESEs.iterator();
156 while( rootItr.hasNext() ) {
157 FlatSESEEnterNode root = rootItr.next();
158 livenessAnalysisBackward( root,
165 SOMETHING IS WRONG WITH THIS, DON'T USE IT UNTIL IT CAN BE FIXED
168 methItr = ownAnalysis.descriptorsToAnalyze.iterator();
169 while( methItr.hasNext() ) {
170 Descriptor d = methItr.next();
171 FlatMethod fm = state.getMethodFlat( d );
173 // prune variable results in one traversal
174 // by removing reference variables that are not live
175 pruneVariableResultsWithLiveness( fm );
181 methItr = ownAnalysis.descriptorsToAnalyze.iterator();
182 while( methItr.hasNext() ) {
183 Descriptor d = methItr.next();
184 FlatMethod fm = state.getMethodFlat( d );
186 // compute what is not available at every program
187 // point, in a forward fixed-point pass
188 notAvailableForward( fm );
192 methItr = ownAnalysis.descriptorsToAnalyze.iterator();
193 JavaCallGraph javaCallGraph = new JavaCallGraph(state,tu);
194 while( methItr.hasNext() ) {
195 Descriptor d = methItr.next();
196 FlatMethod fm = state.getMethodFlat( d );
197 methodEffects(fm,javaCallGraph);
200 // disjoint analysis with a set of flagged allocation sites of live-in variable
202 OwnershipAnalysis oa2 = new OwnershipAnalysis(state, tu, callGraph,
203 state.OWNERSHIPALLOCDEPTH, false,
204 false, state.OWNERSHIPALIASFILE,
206 mapMethodContextToLiveInAllocationSiteSet);
208 methItr = oa2.descriptorsToAnalyze.iterator();
209 while (methItr.hasNext()) {
210 Descriptor d = methItr.next();
211 FlatMethod fm = state.getMethodFlat(d);
212 debugFunction(oa2, fm);
215 } catch (IOException e) {
216 System.err.println(e);
222 methItr = ownAnalysis.descriptorsToAnalyze.iterator();
223 while( methItr.hasNext() ) {
224 Descriptor d = methItr.next();
225 FlatMethod fm = state.getMethodFlat( d );
227 // compute a plan for code injections
228 codePlansForward( fm );
232 // splice new IR nodes into graph after all
233 // analysis passes are complete
234 Iterator spliceItr = wdvNodesToSpliceIn.entrySet().iterator();
235 while( spliceItr.hasNext() ) {
236 Map.Entry me = (Map.Entry) spliceItr.next();
237 FlatWriteDynamicVarNode fwdvn = (FlatWriteDynamicVarNode) me.getValue();
238 fwdvn.spliceIntoIR();
242 double timeEndAnalysis = (double) System.nanoTime();
243 double dt = (timeEndAnalysis - timeStartAnalysis)/(Math.pow( 10.0, 9.0 ) );
244 String treport = String.format( "The mlp analysis took %.3f sec.", dt );
245 System.out.println( treport );
247 if( state.MLPDEBUG ) {
249 writeReports( treport );
250 } catch( IOException e ) {}
255 private void buildForestForward( FlatMethod fm ) {
257 // start from flat method top, visit every node in
258 // method exactly once, find SESEs and remember
259 // roots and child relationships
260 Set<FlatNode> flatNodesToVisit = new HashSet<FlatNode>();
261 flatNodesToVisit.add( fm );
263 Set<FlatNode> visited = new HashSet<FlatNode>();
265 Stack<FlatSESEEnterNode> seseStackFirst = new Stack<FlatSESEEnterNode>();
266 seseStacks.put( fm, seseStackFirst );
268 while( !flatNodesToVisit.isEmpty() ) {
269 Iterator<FlatNode> fnItr = flatNodesToVisit.iterator();
270 FlatNode fn = fnItr.next();
272 Stack<FlatSESEEnterNode> seseStack = seseStacks.get( fn );
273 assert seseStack != null;
275 flatNodesToVisit.remove( fn );
278 buildForest_nodeActions( fn, seseStack, fm );
280 for( int i = 0; i < fn.numNext(); i++ ) {
281 FlatNode nn = fn.getNext( i );
283 if( !visited.contains( nn ) ) {
284 flatNodesToVisit.add( nn );
286 // clone stack and send along each analysis path
287 seseStacks.put( nn, (Stack<FlatSESEEnterNode>)seseStack.clone() );
293 private void buildForest_nodeActions( FlatNode fn,
294 Stack<FlatSESEEnterNode> seseStack,
296 switch( fn.kind() ) {
298 case FKind.FlatSESEEnterNode: {
299 FlatSESEEnterNode fsen = (FlatSESEEnterNode) fn;
301 if( !fsen.getIsCallerSESEplaceholder() ) {
302 allSESEs.add( fsen );
305 fsen.setfmEnclosing( fm );
306 fsen.setmdEnclosing( fm.getMethod() );
307 fsen.setcdEnclosing( fm.getMethod().getClassDesc() );
309 if( seseStack.empty() ) {
310 rootSESEs.add( fsen );
311 fsen.setParent( null );
313 seseStack.peek().addChild( fsen );
314 fsen.setParent( seseStack.peek() );
317 seseStack.push( fsen );
320 case FKind.FlatSESEExitNode: {
321 FlatSESEExitNode fsexn = (FlatSESEExitNode) fn;
322 assert !seseStack.empty();
323 FlatSESEEnterNode fsen = seseStack.pop();
326 case FKind.FlatReturnNode: {
327 FlatReturnNode frn = (FlatReturnNode) fn;
328 if( !seseStack.empty() &&
329 !seseStack.peek().getIsCallerSESEplaceholder()
331 throw new Error( "Error: return statement enclosed within SESE "+
332 seseStack.peek().getPrettyIdentifier() );
340 private void livenessAnalysisBackward( FlatSESEEnterNode fsen,
342 Hashtable< FlatSESEExitNode, Set<TempDescriptor> > liveout ) {
344 // start from an SESE exit, visit nodes in reverse up to
345 // SESE enter in a fixed-point scheme, where children SESEs
346 // should already be analyzed and therefore can be skipped
347 // because child SESE enter node has all necessary info
348 Set<FlatNode> flatNodesToVisit = new HashSet<FlatNode>();
351 flatNodesToVisit.add( fsen.getfmEnclosing().getFlatExit() );
353 flatNodesToVisit.add( fsen.getFlatExit() );
356 Hashtable<FlatNode, Set<TempDescriptor>> livenessResults =
357 new Hashtable< FlatNode, Set<TempDescriptor> >();
360 liveout = new Hashtable< FlatSESEExitNode, Set<TempDescriptor> >();
363 while( !flatNodesToVisit.isEmpty() ) {
364 FlatNode fn = (FlatNode) flatNodesToVisit.iterator().next();
365 flatNodesToVisit.remove( fn );
367 Set<TempDescriptor> prev = livenessResults.get( fn );
369 // merge sets from control flow joins
370 Set<TempDescriptor> u = new HashSet<TempDescriptor>();
371 for( int i = 0; i < fn.numNext(); i++ ) {
372 FlatNode nn = fn.getNext( i );
373 Set<TempDescriptor> s = livenessResults.get( nn );
379 Set<TempDescriptor> curr = liveness_nodeActions( fn, u, fsen, toplevel, liveout);
381 // if a new result, schedule backward nodes for analysis
382 if( !curr.equals( prev ) ) {
383 livenessResults.put( fn, curr );
385 // don't flow backwards past current SESE enter
386 if( !fn.equals( fsen ) ) {
387 for( int i = 0; i < fn.numPrev(); i++ ) {
388 FlatNode nn = fn.getPrev( i );
389 flatNodesToVisit.add( nn );
395 Set<TempDescriptor> s = livenessResults.get( fsen );
397 fsen.addInVarSet( s );
400 // remember liveness per node from the root view as the
401 // global liveness of variables for later passes to use
403 livenessRootView.putAll( livenessResults );
406 // post-order traversal, so do children first
407 Iterator<FlatSESEEnterNode> childItr = fsen.getChildren().iterator();
408 while( childItr.hasNext() ) {
409 FlatSESEEnterNode fsenChild = childItr.next();
410 livenessAnalysisBackward( fsenChild, false, liveout );
414 private Set<TempDescriptor> liveness_nodeActions( FlatNode fn,
415 Set<TempDescriptor> liveIn,
416 FlatSESEEnterNode currentSESE,
418 Hashtable< FlatSESEExitNode, Set<TempDescriptor> > liveout
420 switch( fn.kind() ) {
422 case FKind.FlatSESEExitNode:
424 FlatSESEExitNode fsexn = (FlatSESEExitNode) fn;
425 if( !liveout.containsKey( fsexn ) ) {
426 liveout.put( fsexn, new HashSet<TempDescriptor>() );
428 liveout.get( fsexn ).addAll( liveIn );
430 // no break, sese exits should also execute default actions
433 // handle effects of statement in reverse, writes then reads
434 TempDescriptor [] writeTemps = fn.writesTemps();
435 for( int i = 0; i < writeTemps.length; ++i ) {
436 liveIn.remove( writeTemps[i] );
439 FlatSESEExitNode fsexn = currentSESE.getFlatExit();
440 Set<TempDescriptor> livetemps = liveout.get( fsexn );
441 if( livetemps != null &&
442 livetemps.contains( writeTemps[i] ) ) {
443 // write to a live out temp...
444 // need to put in SESE liveout set
445 currentSESE.addOutVar( writeTemps[i] );
450 TempDescriptor [] readTemps = fn.readsTemps();
451 for( int i = 0; i < readTemps.length; ++i ) {
452 liveIn.add( readTemps[i] );
455 Set<TempDescriptor> virtualReadTemps = livenessVirtualReads.get( fn );
456 if( virtualReadTemps != null ) {
457 liveIn.addAll( virtualReadTemps );
468 private void variableAnalysisForward( FlatMethod fm ) {
470 Set<FlatNode> flatNodesToVisit = new HashSet<FlatNode>();
471 flatNodesToVisit.add( fm );
473 while( !flatNodesToVisit.isEmpty() ) {
474 FlatNode fn = (FlatNode) flatNodesToVisit.iterator().next();
475 flatNodesToVisit.remove( fn );
477 Stack<FlatSESEEnterNode> seseStack = seseStacks.get( fn );
478 assert seseStack != null;
480 VarSrcTokTable prev = variableResults.get( fn );
482 // merge sets from control flow joins
483 VarSrcTokTable curr = new VarSrcTokTable();
484 for( int i = 0; i < fn.numPrev(); i++ ) {
485 FlatNode nn = fn.getPrev( i );
486 VarSrcTokTable incoming = variableResults.get( nn );
487 curr.merge( incoming );
490 if( !seseStack.empty() ) {
491 variable_nodeActions( fn, curr, seseStack.peek() );
494 // if a new result, schedule forward nodes for analysis
495 if( !curr.equals( prev ) ) {
496 variableResults.put( fn, curr );
498 for( int i = 0; i < fn.numNext(); i++ ) {
499 FlatNode nn = fn.getNext( i );
500 flatNodesToVisit.add( nn );
506 private void variable_nodeActions( FlatNode fn,
507 VarSrcTokTable vstTable,
508 FlatSESEEnterNode currentSESE ) {
509 switch( fn.kind() ) {
511 case FKind.FlatSESEEnterNode: {
512 FlatSESEEnterNode fsen = (FlatSESEEnterNode) fn;
513 assert fsen.equals( currentSESE );
515 vstTable.age( currentSESE );
516 vstTable.assertConsistency();
519 case FKind.FlatSESEExitNode: {
520 FlatSESEExitNode fsexn = (FlatSESEExitNode) fn;
521 FlatSESEEnterNode fsen = fsexn.getFlatEnter();
522 assert currentSESE.getChildren().contains( fsen );
524 vstTable.remapChildTokens( fsen );
526 // liveness virtual reads are things that might be
527 // written by an SESE and should be added to the in-set
528 // anything virtually read by this SESE should be pruned
529 // of parent or sibling sources
530 Set<TempDescriptor> liveVars = livenessRootView.get( fn );
531 Set<TempDescriptor> fsenVirtReads = vstTable.calcVirtReadsAndPruneParentAndSiblingTokens( fsen, liveVars );
532 Set<TempDescriptor> fsenVirtReadsOld = livenessVirtualReads.get( fn );
533 if( fsenVirtReadsOld != null ) {
534 fsenVirtReads.addAll( fsenVirtReadsOld );
536 livenessVirtualReads.put( fn, fsenVirtReads );
539 // then all child out-set tokens are guaranteed
540 // to be filled in, so clobber those entries with
541 // the latest, clean sources
542 Iterator<TempDescriptor> outVarItr = fsen.getOutVarSet().iterator();
543 while( outVarItr.hasNext() ) {
544 TempDescriptor outVar = outVarItr.next();
545 HashSet<TempDescriptor> ts = new HashSet<TempDescriptor>();
547 VariableSourceToken vst =
548 new VariableSourceToken( ts,
553 vstTable.remove( outVar );
556 vstTable.assertConsistency();
560 case FKind.FlatOpNode: {
561 FlatOpNode fon = (FlatOpNode) fn;
563 if( fon.getOp().getOp() == Operation.ASSIGN ) {
564 TempDescriptor lhs = fon.getDest();
565 TempDescriptor rhs = fon.getLeft();
567 vstTable.remove( lhs );
569 Set<VariableSourceToken> forAddition = new HashSet<VariableSourceToken>();
571 Iterator<VariableSourceToken> itr = vstTable.get( rhs ).iterator();
572 while( itr.hasNext() ) {
573 VariableSourceToken vst = itr.next();
575 HashSet<TempDescriptor> ts = new HashSet<TempDescriptor>();
578 if( currentSESE.getChildren().contains( vst.getSESE() ) ) {
579 // if the source comes from a child, copy it over
580 forAddition.add( new VariableSourceToken( ts,
587 // otherwise, stamp it as us as the source
588 forAddition.add( new VariableSourceToken( ts,
597 vstTable.addAll( forAddition );
599 // only break if this is an ASSIGN op node,
600 // otherwise fall through to default case
601 vstTable.assertConsistency();
606 // note that FlatOpNode's that aren't ASSIGN
607 // fall through to this default case
609 TempDescriptor [] writeTemps = fn.writesTemps();
610 if( writeTemps.length > 0 ) {
613 // for now, when writeTemps > 1, make sure
614 // its a call node, programmer enforce only
615 // doing stuff like calling a print routine
616 //assert writeTemps.length == 1;
617 if( writeTemps.length > 1 ) {
618 assert fn.kind() == FKind.FlatCall ||
619 fn.kind() == FKind.FlatMethod;
623 vstTable.remove( writeTemps[0] );
625 HashSet<TempDescriptor> ts = new HashSet<TempDescriptor>();
626 ts.add( writeTemps[0] );
628 vstTable.add( new VariableSourceToken( ts,
636 vstTable.assertConsistency();
643 private void pruneVariableResultsWithLiveness( FlatMethod fm ) {
645 // start from flat method top, visit every node in
646 // method exactly once
647 Set<FlatNode> flatNodesToVisit = new HashSet<FlatNode>();
648 flatNodesToVisit.add( fm );
650 Set<FlatNode> visited = new HashSet<FlatNode>();
652 while( !flatNodesToVisit.isEmpty() ) {
653 Iterator<FlatNode> fnItr = flatNodesToVisit.iterator();
654 FlatNode fn = fnItr.next();
656 flatNodesToVisit.remove( fn );
659 Set<TempDescriptor> rootLiveSet = livenessRootView.get( fn );
660 VarSrcTokTable vstTable = variableResults.get( fn );
662 vstTable.pruneByLiveness( rootLiveSet );
664 for( int i = 0; i < fn.numNext(); i++ ) {
665 FlatNode nn = fn.getNext( i );
667 if( !visited.contains( nn ) ) {
668 flatNodesToVisit.add( nn );
675 private void notAvailableForward( FlatMethod fm ) {
677 Set<FlatNode> flatNodesToVisit = new HashSet<FlatNode>();
678 flatNodesToVisit.add( fm );
680 while( !flatNodesToVisit.isEmpty() ) {
681 FlatNode fn = (FlatNode) flatNodesToVisit.iterator().next();
682 flatNodesToVisit.remove( fn );
684 Stack<FlatSESEEnterNode> seseStack = seseStacks.get( fn );
685 assert seseStack != null;
687 Set<TempDescriptor> prev = notAvailableResults.get( fn );
689 Set<TempDescriptor> curr = new HashSet<TempDescriptor>();
690 for( int i = 0; i < fn.numPrev(); i++ ) {
691 FlatNode nn = fn.getPrev( i );
692 Set<TempDescriptor> notAvailIn = notAvailableResults.get( nn );
693 if( notAvailIn != null ) {
694 curr.addAll( notAvailIn );
698 if( !seseStack.empty() ) {
699 notAvailable_nodeActions( fn, curr, seseStack.peek() );
702 // if a new result, schedule forward nodes for analysis
703 if( !curr.equals( prev ) ) {
704 notAvailableResults.put( fn, curr );
706 for( int i = 0; i < fn.numNext(); i++ ) {
707 FlatNode nn = fn.getNext( i );
708 flatNodesToVisit.add( nn );
714 private void notAvailable_nodeActions( FlatNode fn,
715 Set<TempDescriptor> notAvailSet,
716 FlatSESEEnterNode currentSESE ) {
718 // any temps that are removed from the not available set
719 // at this node should be marked in this node's code plan
720 // as temps to be grabbed at runtime!
722 switch( fn.kind() ) {
724 case FKind.FlatSESEEnterNode: {
725 FlatSESEEnterNode fsen = (FlatSESEEnterNode) fn;
726 assert fsen.equals( currentSESE );
730 case FKind.FlatSESEExitNode: {
731 FlatSESEExitNode fsexn = (FlatSESEExitNode) fn;
732 FlatSESEEnterNode fsen = fsexn.getFlatEnter();
733 assert currentSESE.getChildren().contains( fsen );
734 notAvailSet.addAll( fsen.getOutVarSet() );
737 case FKind.FlatMethod: {
741 case FKind.FlatOpNode: {
742 FlatOpNode fon = (FlatOpNode) fn;
744 if( fon.getOp().getOp() == Operation.ASSIGN ) {
745 TempDescriptor lhs = fon.getDest();
746 TempDescriptor rhs = fon.getLeft();
748 // copy makes lhs same availability as rhs
749 if( notAvailSet.contains( rhs ) ) {
750 notAvailSet.add( lhs );
752 notAvailSet.remove( lhs );
755 // only break if this is an ASSIGN op node,
756 // otherwise fall through to default case
761 // note that FlatOpNode's that aren't ASSIGN
762 // fall through to this default case
764 TempDescriptor [] writeTemps = fn.writesTemps();
765 for( int i = 0; i < writeTemps.length; i++ ) {
766 TempDescriptor wTemp = writeTemps[i];
767 notAvailSet.remove( wTemp );
769 TempDescriptor [] readTemps = fn.readsTemps();
770 for( int i = 0; i < readTemps.length; i++ ) {
771 TempDescriptor rTemp = readTemps[i];
772 notAvailSet.remove( rTemp );
774 // if this variable has exactly one source, potentially
775 // get other things from this source as well
776 VarSrcTokTable vstTable = variableResults.get( fn );
779 vstTable.getRefVarSrcType( rTemp,
781 currentSESE.getParent() );
783 if( srcType.equals( VarSrcTokTable.SrcType_STATIC ) ) {
785 VariableSourceToken vst = vstTable.get( rTemp ).iterator().next();
787 Iterator<VariableSourceToken> availItr = vstTable.get( vst.getSESE(),
791 // look through things that are also available from same source
792 while( availItr.hasNext() ) {
793 VariableSourceToken vstAlsoAvail = availItr.next();
795 Iterator<TempDescriptor> refVarItr = vstAlsoAvail.getRefVars().iterator();
796 while( refVarItr.hasNext() ) {
797 TempDescriptor refVarAlso = refVarItr.next();
799 // if a variable is available from the same source, AND it ALSO
800 // only comes from one statically known source, mark it available
801 Integer srcTypeAlso =
802 vstTable.getRefVarSrcType( refVarAlso,
804 currentSESE.getParent() );
805 if( srcTypeAlso.equals( VarSrcTokTable.SrcType_STATIC ) ) {
806 notAvailSet.remove( refVarAlso );
817 private void debugFunction(OwnershipAnalysis oa2, FlatMethod fm) {
819 String methodName="SomeWork";
821 MethodDescriptor md=fm.getMethod();
822 HashSet<MethodContext> mcSet=oa2.getAllMethodContextSetByDescriptor(md);
823 Iterator<MethodContext> mcIter=mcSet.iterator();
825 while(mcIter.hasNext()){
826 MethodContext mc=mcIter.next();
828 OwnershipGraph og=oa2.getOwnvershipGraphByMethodContext(mc);
830 if(fm.toString().indexOf(methodName)>0){
832 og.writeGraph(fm.toString() + "SECONDGRAPH", true, true, true, true, false);
833 } catch (IOException e) {
834 System.out.println("Error writing debug capture.");
845 private void methodEffects(FlatMethod fm, CallGraph callGraph) {
847 MethodDescriptor md=fm.getMethod();
848 HashSet<MethodContext> mcSet=ownAnalysis.getAllMethodContextSetByDescriptor(md);
849 Iterator<MethodContext> mcIter=mcSet.iterator();
851 while(mcIter.hasNext()){
852 MethodContext mc=mcIter.next();
854 Set<FlatNode> visited = new HashSet<FlatNode>();
856 Set<FlatNode> flatNodesToVisit = new HashSet<FlatNode>();
857 flatNodesToVisit.add(fm);
859 while (!flatNodesToVisit.isEmpty()) {
860 FlatNode fn = (FlatNode) flatNodesToVisit.iterator().next();
861 flatNodesToVisit.remove(fn);
863 Stack<FlatSESEEnterNode> seseStack = seseStacks.get(fn);
864 assert seseStack != null;
866 if (!seseStack.empty()) {
867 effects_nodeActions(mc, fn, seseStack.peek(), callGraph);
870 flatNodesToVisit.remove(fn);
873 for (int i = 0; i < fn.numNext(); i++) {
874 FlatNode nn = fn.getNext(i);
875 if (!visited.contains(nn)) {
876 flatNodesToVisit.add(nn);
887 private void analyzeRelatedAllocationSite(MethodDescriptor callerMD,
888 MethodContext calleeMC, HashSet<Integer> paramIndexSet,
889 HashSet<HeapRegionNode> visitedHRN) {
891 HashSet<MethodContext> mcSet = ownAnalysis
892 .getAllMethodContextSetByDescriptor(callerMD);
896 Iterator<MethodContext> mcIter = mcSet.iterator();
898 FlatMethod callerFM = state.getMethodFlat(callerMD);
900 while (mcIter.hasNext()) {
901 MethodContext mc = mcIter.next();
903 Set<FlatNode> visited = new HashSet<FlatNode>();
904 Set<FlatNode> flatNodesToVisit = new HashSet<FlatNode>();
905 flatNodesToVisit.add(callerFM);
907 while (!flatNodesToVisit.isEmpty()) {
908 FlatNode fn = (FlatNode) flatNodesToVisit.iterator().next();
909 flatNodesToVisit.remove(fn);
911 analyzeRelatedAllocationSite_NodeAction(fn, mc, calleeMC,
912 paramIndexSet,visitedHRN);
914 flatNodesToVisit.remove(fn);
917 for (int i = 0; i < fn.numNext(); i++) {
918 FlatNode nn = fn.getNext(i);
919 if (!visited.contains(nn)) {
920 flatNodesToVisit.add(nn);
929 private void analyzeRelatedAllocationSite_NodeAction(FlatNode fn, MethodContext callerMC,
930 MethodContext calleeMC,
931 HashSet<Integer> paramIndexSet, HashSet<HeapRegionNode> visitedHRN) {
933 OwnershipGraph og = ownAnalysis
934 .getOwnvershipGraphByMethodContext(callerMC);
938 case FKind.FlatCall: {
940 FlatCall fc = (FlatCall) fn;
943 MethodContext calleeMCfromOG = ownAnalysis.getCalleeMethodContext(
946 // disable below condition. currently collect all possible
947 // allocation sites without regarding method context
949 // if (calleeMC.equals(calleeMCfromOG)) { // in this case, this
950 // method context calls corresponding callee.
953 if (((MethodDescriptor) calleeMC.getDescriptor()).isStatic()) {
959 for (Iterator iterator = paramIndexSet.iterator(); iterator
961 Integer integer = (Integer) iterator.next();
963 int paramIdx = integer - base;
965 // if paramIdx is less than 0, assumes that it is
966 // related with wrong method contexts.
967 TempDescriptor arg = fc.getArg(paramIdx);
968 LabelNode argLN = og.td2ln.get(arg);
970 Iterator<ReferenceEdge> iterEdge = argLN
971 .iteratorToReferencees();
972 while (iterEdge.hasNext()) {
973 ReferenceEdge referenceEdge = (ReferenceEdge) iterEdge
976 HeapRegionNode dstHRN = referenceEdge.getDst();
977 if (dstHRN.isParameter()) {
978 if (!visitedHRN.contains(dstHRN)) {
979 setupRelatedAllocSiteAnalysis(og, callerMC,
983 addLiveInAllocationSite(callerMC, dstHRN
984 .getAllocationSite());
1001 private void setupRelatedAllocSiteAnalysis(OwnershipGraph og,
1002 MethodContext mc, HeapRegionNode dstHRN,
1003 HashSet<HeapRegionNode> visitedHRN) {
1005 HashSet<Integer> paramIndexSet = new HashSet<Integer>();
1007 // collect corresponding param index
1008 Set<Integer> pIndexSet = og.idPrimary2paramIndexSet.get(dstHRN.getID());
1009 if (pIndexSet != null) {
1010 for (Iterator iterator = pIndexSet.iterator(); iterator.hasNext();) {
1011 Integer integer = (Integer) iterator.next();
1012 paramIndexSet.add(integer);
1016 Set<Integer> sIndexSet = og.idSecondary2paramIndexSet.get(dstHRN
1018 if (sIndexSet != null) {
1019 for (Iterator iterator = sIndexSet.iterator(); iterator.hasNext();) {
1020 Integer integer = (Integer) iterator.next();
1021 paramIndexSet.add(integer);
1025 if (mc.getDescriptor() instanceof MethodDescriptor) {
1026 Set callerSet = callGraph.getCallerSet((MethodDescriptor) mc
1028 for (Iterator iterator = callerSet.iterator(); iterator.hasNext();) {
1029 Object obj = (Object) iterator.next();
1030 if (obj instanceof MethodDescriptor) {
1031 MethodDescriptor callerMD = (MethodDescriptor) obj;
1033 analyzeRelatedAllocationSite(callerMD, mc, paramIndexSet,visitedHRN);
1040 private void effects_nodeActions(MethodContext mc, FlatNode fn,
1041 FlatSESEEnterNode currentSESE, CallGraph callGraph) {
1043 OwnershipGraph og = ownAnalysis.getOwnvershipGraphByMethodContext(mc);
1045 switch (fn.kind()) {
1047 case FKind.FlatSESEEnterNode: {
1049 FlatSESEEnterNode fsen = (FlatSESEEnterNode) fn;
1050 assert fsen.equals(currentSESE);
1052 // uniquely taint each live-in variable
1053 Set<TempDescriptor> set = fsen.getInVarSet();
1054 Iterator<TempDescriptor> iter = set.iterator();
1056 while (iter.hasNext()) {
1057 TempDescriptor td = iter.next();
1058 LabelNode ln = og.td2ln.get(td);
1060 int taint = (int) Math.pow(2, idx);
1061 taintLabelNode(ln, taint);
1063 // collects related allocation sites
1064 Iterator<ReferenceEdge> referenceeIter = ln
1065 .iteratorToReferencees();
1066 while (referenceeIter.hasNext()) {
1067 ReferenceEdge referenceEdge = (ReferenceEdge) referenceeIter
1069 HeapRegionNode dstHRN = referenceEdge.getDst();
1070 if (dstHRN.isParameter()) {
1072 HashSet<HeapRegionNode> visitedHRN=new HashSet<HeapRegionNode>();
1073 visitedHRN.add(dstHRN);
1074 setupRelatedAllocSiteAnalysis(og,mc,dstHRN,visitedHRN);
1077 addLiveInAllocationSite(mc, dstHRN
1078 .getAllocationSite());
1090 case FKind.FlatSESEExitNode: {
1091 FlatSESEExitNode fsexit = (FlatSESEExitNode) fn;
1093 FlatSESEEnterNode enterNode = fsexit.getFlatEnter();
1094 FlatSESEEnterNode parent = enterNode.getParent();
1095 if (parent != null) {
1097 SESEEffectsSet set = enterNode.getSeseEffectsSet();
1098 Hashtable<TempDescriptor, HashSet<SESEEffectsKey>> readTable = set
1100 Hashtable<TempDescriptor, HashSet<SESEEffectsKey>> parentReadTable = parent
1101 .getSeseEffectsSet().getReadTable();
1102 Set<TempDescriptor> keys = readTable.keySet();
1103 Iterator<TempDescriptor> keyIter = keys.iterator();
1104 while (keyIter.hasNext()) {
1105 TempDescriptor td = (TempDescriptor) keyIter.next();
1106 HashSet<SESEEffectsKey> effectsSet = readTable.get(td);
1107 HashSet<SESEEffectsKey> parentEffectsSet = parentReadTable
1109 if (parentEffectsSet == null) {
1110 parentEffectsSet = new HashSet<SESEEffectsKey>();
1113 for (Iterator iterator = effectsSet.iterator(); iterator
1115 SESEEffectsKey seseKey = (SESEEffectsKey) iterator
1117 parentEffectsSet.add(new SESEEffectsKey(seseKey
1118 .getFieldDescriptor(), seseKey
1119 .getTypeDescriptor(), seseKey.getHRNId()));
1122 parentReadTable.put(td, parentEffectsSet);
1125 Hashtable<TempDescriptor, HashSet<SESEEffectsKey>> writeTable = set
1127 Hashtable<TempDescriptor, HashSet<SESEEffectsKey>> parentWriteTable = parent
1128 .getSeseEffectsSet().getWriteTable();
1129 keys = writeTable.keySet();
1130 keyIter = keys.iterator();
1131 while (keyIter.hasNext()) {
1132 TempDescriptor td = (TempDescriptor) keyIter.next();
1133 HashSet<SESEEffectsKey> effectsSet = writeTable.get(td);
1134 HashSet<SESEEffectsKey> parentEffectsSet = parentWriteTable
1136 if (parentEffectsSet == null) {
1137 parentEffectsSet = new HashSet<SESEEffectsKey>();
1140 for (Iterator iterator = effectsSet.iterator(); iterator
1142 SESEEffectsKey seseKey = (SESEEffectsKey) iterator
1144 parentEffectsSet.add(new SESEEffectsKey(seseKey
1145 .getFieldDescriptor(), seseKey
1146 .getTypeDescriptor(), seseKey.getHRNId()));
1149 parentWriteTable.put(td, parentEffectsSet);
1152 Hashtable<TempDescriptor, HashSet<SESEEffectsKey>> strongUpdateTable = set
1153 .getStrongUpdateTable();
1154 Hashtable<TempDescriptor, HashSet<SESEEffectsKey>> parentstrongUpdateTable = parent
1155 .getSeseEffectsSet().getStrongUpdateTable();
1156 keys = strongUpdateTable.keySet();
1157 keyIter = keys.iterator();
1158 while (keyIter.hasNext()) {
1159 TempDescriptor td = (TempDescriptor) keyIter.next();
1160 HashSet<SESEEffectsKey> effectsSet = strongUpdateTable
1162 HashSet<SESEEffectsKey> parentEffectsSet = parentstrongUpdateTable
1164 if (parentEffectsSet == null) {
1165 parentEffectsSet = new HashSet<SESEEffectsKey>();
1168 for (Iterator iterator = effectsSet.iterator(); iterator
1170 SESEEffectsKey seseKey = (SESEEffectsKey) iterator
1172 parentEffectsSet.add(new SESEEffectsKey(seseKey
1173 .getFieldDescriptor(), seseKey
1174 .getTypeDescriptor(), seseKey.getHRNId()));
1177 parentstrongUpdateTable.put(td, parentEffectsSet);
1185 case FKind.FlatFieldNode: {
1187 FlatFieldNode ffn = (FlatFieldNode) fn;
1188 TempDescriptor src = ffn.getSrc();
1189 FieldDescriptor field = ffn.getField();
1191 LabelNode srcLN = og.td2ln.get(src);
1192 if (srcLN != null) {
1193 HashSet<TempDescriptor> affectedTDSet = getAccessedTaintNodeSet(srcLN);
1194 Iterator<TempDescriptor> affectedIter = affectedTDSet
1196 while (affectedIter.hasNext()) {
1197 TempDescriptor affectedTD = affectedIter.next();
1199 if (currentSESE.getInVarSet().contains(affectedTD)) {
1201 HashSet<HeapRegionNode> hrnSet = getReferenceHeapIDSet(
1203 Iterator<HeapRegionNode> hrnIter = hrnSet.iterator();
1204 while (hrnIter.hasNext()) {
1205 HeapRegionNode hrn = hrnIter.next();
1207 Iterator<ReferenceEdge> referencers = hrn
1208 .iteratorToReferencers();
1209 while (referencers.hasNext()) {
1210 ReferenceEdge referenceEdge = (ReferenceEdge) referencers
1212 if (field.getSymbol().equals(
1213 referenceEdge.getField())) {
1214 currentSESE.readEffects(affectedTD, field
1215 .getSymbol(), src.getType(),
1216 referenceEdge.getDst().getID());
1224 // handle tainted case
1226 Iterator<ReferenceEdge> edgeIter = srcLN
1227 .iteratorToReferencees();
1228 while (edgeIter.hasNext()) {
1229 ReferenceEdge edge = edgeIter.next();
1230 HeapRegionNode accessHRN = edge.getDst();
1231 // / follow the chain of reference to identify possible
1233 Iterator<ReferenceEdge> referIter = accessHRN
1234 .iteratorToReferencers();
1235 while (referIter.hasNext()) {
1236 ReferenceEdge referEdge = (ReferenceEdge) referIter
1239 // if (referEdge.getTaintIdentifier() >0 ||
1240 // referEdge.getSESETaintIdentifier()>0 ) {
1241 HashSet<TempDescriptor> referSet = new HashSet<TempDescriptor>();
1242 followReference(accessHRN, referSet,
1243 new HashSet<HeapRegionNode>(), currentSESE);
1245 Iterator<TempDescriptor> referSetIter = referSet
1247 while (referSetIter.hasNext()) {
1248 TempDescriptor tempDescriptor = (TempDescriptor) referSetIter
1250 currentSESE.readEffects(tempDescriptor, field
1251 .getSymbol(), src.getType(), accessHRN
1257 if (edge.getTaintIdentifier() > 0
1258 || edge.getSESETaintIdentifier() > 0) {
1260 affectedTDSet = getReferenceNodeSet(accessHRN);
1261 affectedIter = affectedTDSet.iterator();
1262 while (affectedIter.hasNext()) {
1263 TempDescriptor affectedTD = affectedIter.next();
1265 if (currentSESE.getInVarSet().contains(affectedTD)) {
1267 HashSet<HeapRegionNode> hrnSet = getReferenceHeapIDSet(
1269 Iterator<HeapRegionNode> hrnIter = hrnSet
1271 while (hrnIter.hasNext()) {
1272 HeapRegionNode hrn = hrnIter.next();
1273 currentSESE.readEffects(affectedTD, field
1274 .getSymbol(), src.getType(), hrn
1288 case FKind.FlatSetFieldNode: {
1290 FlatSetFieldNode fsen = (FlatSetFieldNode) fn;
1291 TempDescriptor dst = fsen.getDst();
1292 FieldDescriptor field = fsen.getField();
1294 LabelNode dstLN = og.td2ln.get(dst);
1295 if (dstLN != null) {
1296 // check possible strong updates
1297 boolean strongUpdate = false;
1299 if (!field.getType().isImmutable() || field.getType().isArray()) {
1300 Iterator<ReferenceEdge> itrXhrn = dstLN
1301 .iteratorToReferencees();
1302 while (itrXhrn.hasNext()) {
1303 ReferenceEdge edgeX = itrXhrn.next();
1304 HeapRegionNode hrnX = edgeX.getDst();
1306 // we can do a strong update here if one of two cases
1309 && field != OwnershipAnalysis
1310 .getArrayField(field.getType())
1311 && ((hrnX.getNumReferencers() == 1) || // case 1
1312 (hrnX.isSingleObject() && dstLN
1313 .getNumReferencees() == 1) // case 2
1315 strongUpdate = true;
1320 HashSet<TempDescriptor> affectedTDSet = getAccessedTaintNodeSet(dstLN);
1322 Iterator<TempDescriptor> affectedIter = affectedTDSet
1325 while (affectedIter.hasNext()) {
1326 TempDescriptor affectedTD = affectedIter.next();
1327 if (currentSESE.getInVarSet().contains(affectedTD)) {
1329 HashSet<HeapRegionNode> hrnSet = getReferenceHeapIDSet(
1331 Iterator<HeapRegionNode> hrnIter = hrnSet.iterator();
1332 while (hrnIter.hasNext()) {
1333 HeapRegionNode hrn = hrnIter.next();
1335 Iterator<ReferenceEdge> referencers = hrn
1336 .iteratorToReferencers();
1337 while (referencers.hasNext()) {
1338 ReferenceEdge referenceEdge = (ReferenceEdge) referencers
1340 if (field.getSymbol().equals(
1341 referenceEdge.getField())) {
1342 currentSESE.writeEffects(affectedTD, field
1343 .getSymbol(), dst.getType(),
1344 referenceEdge.getDst().getID(),
1353 // handle tainted case
1354 Iterator<ReferenceEdge> edgeIter = dstLN
1355 .iteratorToReferencees();
1356 while (edgeIter.hasNext()) {
1357 ReferenceEdge edge = edgeIter.next();
1359 HeapRegionNode accessHRN = edge.getDst();
1360 // / follow the chain of reference to identify possible
1362 Iterator<ReferenceEdge> referIter = accessHRN
1363 .iteratorToReferencers();
1364 while (referIter.hasNext()) {
1365 ReferenceEdge referEdge = (ReferenceEdge) referIter
1368 // if (referEdge.getTaintIdentifier() > 0 ||
1369 // referEdge.getSESETaintIdentifier() > 0 ) {
1370 HashSet<TempDescriptor> referSet = new HashSet<TempDescriptor>();
1371 followReference(accessHRN, referSet,
1372 new HashSet<HeapRegionNode>(), currentSESE);
1373 Iterator<TempDescriptor> referSetIter = referSet
1375 while (referSetIter.hasNext()) {
1376 TempDescriptor tempDescriptor = (TempDescriptor) referSetIter
1378 currentSESE.writeEffects(tempDescriptor, field
1379 .getSymbol(), dst.getType(), accessHRN
1380 .getID(), strongUpdate);
1385 if (edge.getTaintIdentifier() > 0
1386 || edge.getSESETaintIdentifier() > 0) {
1387 affectedTDSet = getReferenceNodeSet(accessHRN);
1388 affectedIter = affectedTDSet.iterator();
1389 while (affectedIter.hasNext()) {
1390 TempDescriptor affectedTD = affectedIter.next();
1391 if (currentSESE.getInVarSet().contains(affectedTD)) {
1393 HashSet<HeapRegionNode> hrnSet = getReferenceHeapIDSet(
1395 Iterator<HeapRegionNode> hrnIter = hrnSet
1397 while (hrnIter.hasNext()) {
1398 HeapRegionNode hrn = hrnIter.next();
1399 currentSESE.writeEffects(affectedTD, field
1400 .getSymbol(), dst.getType(), hrn
1401 .getID(), strongUpdate);
1416 case FKind.FlatCall: {
1417 FlatCall fc = (FlatCall) fn;
1419 MethodContext calleeMC = ownAnalysis.getCalleeMethodContext(mc, fc);
1421 MethodEffects me = ownAnalysis.getMethodEffectsAnalysis()
1422 .getMethodEffectsByMethodContext(calleeMC);
1424 OwnershipGraph calleeOG = ownAnalysis
1425 .getOwnvershipGraphByMethodContext(calleeMC);
1427 FlatMethod fm = state.getMethodFlat(fc.getMethod());
1428 ParameterDecomposition decomp = new ParameterDecomposition(
1429 ownAnalysis, fc, fm, calleeMC, calleeOG, og);
1432 if (((MethodDescriptor) calleeMC.getDescriptor()).isStatic()) {
1438 for (int i = 0; i < fc.numArgs(); i++) {
1440 TempDescriptor arg = fc.getArg(i);
1441 Set<EffectsKey> readSet = me.getEffects().getReadingSet(
1443 Set<EffectsKey> writeSet = me.getEffects().getWritingSet(
1446 Set<EffectsKey> strongUpdateSet = me.getEffects()
1447 .getStrongUpdateSet(i + base);
1449 LabelNode argLN = og.td2ln.get(arg);
1450 if (argLN != null) {
1451 HashSet<TempDescriptor> affectedTDSet = getAccessedTaintNodeSet(argLN);
1452 Iterator<TempDescriptor> affectedIter = affectedTDSet
1455 while (affectedIter.hasNext()) {
1457 TempDescriptor affectedTD = affectedIter.next();
1458 if (currentSESE.getInVarSet().contains(affectedTD)) {
1460 if (readSet != null) {
1461 Iterator<EffectsKey> readIter = readSet
1463 while (readIter.hasNext()) {
1464 EffectsKey key = readIter.next();
1465 Set<Integer> hrnSet = getCallerHRNId(
1466 new Integer(i + base), calleeOG,
1467 key.getHRNId(), decomp);
1468 Iterator<Integer> hrnIter = hrnSet
1470 while (hrnIter.hasNext()) {
1471 Integer hrnID = (Integer) hrnIter
1473 currentSESE.readEffects(affectedTD, key
1474 .getFieldDescriptor(), key
1475 .getTypeDescriptor(), hrnID);
1480 if (writeSet != null) {
1481 Iterator<EffectsKey> writeIter = writeSet
1483 while (writeIter.hasNext()) {
1484 EffectsKey key = writeIter.next();
1486 Set<Integer> hrnSet = getCallerHRNId(
1487 new Integer(i + base), calleeOG,
1488 key.getHRNId(), decomp);
1489 Iterator<Integer> hrnIter = hrnSet
1491 while (hrnIter.hasNext()) {
1492 Integer hrnID = (Integer) hrnIter
1494 currentSESE.writeEffects(affectedTD,
1495 key.getFieldDescriptor(), key
1496 .getTypeDescriptor(),
1503 if (strongUpdateSet != null) {
1504 Iterator<EffectsKey> strongUpdateIter = strongUpdateSet
1506 while (strongUpdateIter.hasNext()) {
1507 EffectsKey key = strongUpdateIter.next();
1509 Set<Integer> hrnSet = getCallerHRNId(
1510 new Integer(i + base), calleeOG,
1511 key.getHRNId(), decomp);
1512 Iterator<Integer> hrnIter = hrnSet
1514 while (hrnIter.hasNext()) {
1515 Integer hrnID = (Integer) hrnIter
1517 currentSESE.writeEffects(affectedTD,
1518 key.getFieldDescriptor(), key
1519 .getTypeDescriptor(),
1540 private void addLiveInAllocationSite(MethodContext mc, AllocationSite ac){
1541 HashSet<AllocationSite> set=mapMethodContextToLiveInAllocationSiteSet.get(mc);
1543 set=new HashSet<AllocationSite>();
1546 mapMethodContextToLiveInAllocationSiteSet.put(mc, set);
1549 private void followReference(HeapRegionNode hrn,HashSet<TempDescriptor> tdSet, HashSet<HeapRegionNode> visited, FlatSESEEnterNode currentSESE){
1551 Iterator<ReferenceEdge> referIter=hrn.iteratorToReferencers();
1552 // check whether hrn is referenced by TD
1553 while (referIter.hasNext()) {
1554 ReferenceEdge referEdge = (ReferenceEdge) referIter.next();
1555 if(referEdge.getSrc() instanceof LabelNode){
1556 LabelNode ln=(LabelNode)referEdge.getSrc();
1557 if(currentSESE.getInVarSet().contains(ln.getTempDescriptor())){
1558 tdSet.add(ln.getTempDescriptor());
1560 }else if(referEdge.getSrc() instanceof HeapRegionNode){
1561 HeapRegionNode nextHRN=(HeapRegionNode)referEdge.getSrc();
1562 if(!visited.contains(nextHRN)){
1563 visited.add(nextHRN);
1564 followReference(nextHRN,tdSet,visited,currentSESE);
1572 private Set<Integer> getCallerHRNId(Integer paramIdx,
1573 OwnershipGraph calleeOG, Integer calleeHRNId,
1574 ParameterDecomposition paramDecom) {
1576 Integer hrnPrimaryID = calleeOG.paramIndex2idPrimary.get(paramIdx);
1577 Integer hrnSecondaryID = calleeOG.paramIndex2idSecondary.get(paramIdx);
1579 if (calleeHRNId.equals(hrnPrimaryID)) {
1580 // it references to primary param heap region
1581 return paramDecom.getParamObject_hrnIDs(paramIdx);
1582 } else if (calleeHRNId.equals(hrnSecondaryID)) {
1583 // it references to secondary param heap region
1584 return paramDecom.getParamReachable_hrnIDs(paramIdx);
1587 return new HashSet<Integer>();
1590 private void taintLabelNode(LabelNode ln, int identifier) {
1592 Iterator<ReferenceEdge> edgeIter = ln.iteratorToReferencees();
1593 while (edgeIter.hasNext()) {
1594 ReferenceEdge edge = edgeIter.next();
1595 HeapRegionNode hrn = edge.getDst();
1597 Iterator<ReferenceEdge> edgeReferencerIter = hrn
1598 .iteratorToReferencers();
1599 while (edgeReferencerIter.hasNext()) {
1600 ReferenceEdge referencerEdge = edgeReferencerIter.next();
1601 OwnershipNode node = referencerEdge.getSrc();
1602 if (node instanceof LabelNode) {
1603 referencerEdge.unionSESETaintIdentifier(identifier);
1604 }else if(node instanceof HeapRegionNode){
1605 referencerEdge.unionSESETaintIdentifier(identifier);
1613 private HashSet<TempDescriptor> getReferenceNodeSet(HeapRegionNode hrn){
1615 HashSet<TempDescriptor> returnSet=new HashSet<TempDescriptor>();
1617 Iterator<ReferenceEdge> edgeIter=hrn.iteratorToReferencers();
1618 while(edgeIter.hasNext()){
1619 ReferenceEdge edge=edgeIter.next();
1620 if(edge.getSrc() instanceof LabelNode){
1621 LabelNode ln=(LabelNode)edge.getSrc();
1622 returnSet.add(ln.getTempDescriptor());
1631 private HashSet<HeapRegionNode> getReferenceHeapIDSet(OwnershipGraph og, TempDescriptor td){
1633 HashSet<HeapRegionNode> returnSet=new HashSet<HeapRegionNode>();
1635 LabelNode ln=og.td2ln.get(td);
1636 Iterator<ReferenceEdge> edgeIter=ln.iteratorToReferencees();
1637 while(edgeIter.hasNext()){
1638 ReferenceEdge edge=edgeIter.next();
1639 HeapRegionNode hrn=edge.getDst();
1646 private HashSet<TempDescriptor> getAccessedTaintNodeSet(LabelNode ln) {
1648 HashSet<TempDescriptor> returnSet = new HashSet<TempDescriptor>();
1650 Iterator<ReferenceEdge> edgeIter = ln.iteratorToReferencees();
1651 while (edgeIter.hasNext()) {
1652 ReferenceEdge edge = edgeIter.next();
1653 HeapRegionNode hrn = edge.getDst();
1655 Iterator<ReferenceEdge> edgeReferencerIter = hrn
1656 .iteratorToReferencers();
1657 while (edgeReferencerIter.hasNext()) {
1658 ReferenceEdge referencerEdge = edgeReferencerIter.next();
1660 if (referencerEdge.getSrc() instanceof LabelNode) {
1661 if (!((LabelNode) referencerEdge.getSrc()).equals(ln)) {
1663 if (referencerEdge.getSESETaintIdentifier() > 0) {
1664 TempDescriptor td = ((LabelNode) referencerEdge
1665 .getSrc()).getTempDescriptor();
1679 private void codePlansForward( FlatMethod fm ) {
1681 // start from flat method top, visit every node in
1682 // method exactly once
1683 Set<FlatNode> flatNodesToVisit = new HashSet<FlatNode>();
1684 flatNodesToVisit.add( fm );
1686 Set<FlatNode> visited = new HashSet<FlatNode>();
1688 while( !flatNodesToVisit.isEmpty() ) {
1689 Iterator<FlatNode> fnItr = flatNodesToVisit.iterator();
1690 FlatNode fn = fnItr.next();
1692 flatNodesToVisit.remove( fn );
1695 Stack<FlatSESEEnterNode> seseStack = seseStacks.get( fn );
1696 assert seseStack != null;
1698 // use incoming results as "dot statement" or just
1699 // before the current statement
1700 VarSrcTokTable dotSTtable = new VarSrcTokTable();
1701 for( int i = 0; i < fn.numPrev(); i++ ) {
1702 FlatNode nn = fn.getPrev( i );
1703 dotSTtable.merge( variableResults.get( nn ) );
1706 // find dt-st notAvailableSet also
1707 Set<TempDescriptor> dotSTnotAvailSet = new HashSet<TempDescriptor>();
1708 for( int i = 0; i < fn.numPrev(); i++ ) {
1709 FlatNode nn = fn.getPrev( i );
1710 Set<TempDescriptor> notAvailIn = notAvailableResults.get( nn );
1711 if( notAvailIn != null ) {
1712 dotSTnotAvailSet.addAll( notAvailIn );
1716 Set<TempDescriptor> dotSTlive = livenessRootView.get( fn );
1718 if( !seseStack.empty() ) {
1719 codePlans_nodeActions( fn,
1727 for( int i = 0; i < fn.numNext(); i++ ) {
1728 FlatNode nn = fn.getNext( i );
1730 if( !visited.contains( nn ) ) {
1731 flatNodesToVisit.add( nn );
1737 private void codePlans_nodeActions( FlatNode fn,
1738 Set<TempDescriptor> liveSetIn,
1739 VarSrcTokTable vstTableIn,
1740 Set<TempDescriptor> notAvailSetIn,
1741 FlatSESEEnterNode currentSESE ) {
1743 CodePlan plan = new CodePlan( currentSESE);
1745 switch( fn.kind() ) {
1747 case FKind.FlatSESEEnterNode: {
1748 FlatSESEEnterNode fsen = (FlatSESEEnterNode) fn;
1750 // track the source types of the in-var set so generated
1751 // code at this SESE issue can compute the number of
1752 // dependencies properly
1753 Iterator<TempDescriptor> inVarItr = fsen.getInVarSet().iterator();
1754 while( inVarItr.hasNext() ) {
1755 TempDescriptor inVar = inVarItr.next();
1757 vstTableIn.getRefVarSrcType( inVar,
1761 // the current SESE needs a local space to track the dynamic
1762 // variable and the child needs space in its SESE record
1763 if( srcType.equals( VarSrcTokTable.SrcType_DYNAMIC ) ) {
1764 fsen.addDynamicInVar( inVar );
1765 fsen.getParent().addDynamicVar( inVar );
1767 } else if( srcType.equals( VarSrcTokTable.SrcType_STATIC ) ) {
1768 fsen.addStaticInVar( inVar );
1769 VariableSourceToken vst = vstTableIn.get( inVar ).iterator().next();
1770 fsen.putStaticInVar2src( inVar, vst );
1771 fsen.addStaticInVarSrc( new SESEandAgePair( vst.getSESE(),
1777 assert srcType.equals( VarSrcTokTable.SrcType_READY );
1778 fsen.addReadyInVar( inVar );
1784 case FKind.FlatSESEExitNode: {
1785 FlatSESEExitNode fsexn = (FlatSESEExitNode) fn;
1788 case FKind.FlatOpNode: {
1789 FlatOpNode fon = (FlatOpNode) fn;
1791 if( fon.getOp().getOp() == Operation.ASSIGN ) {
1792 TempDescriptor lhs = fon.getDest();
1793 TempDescriptor rhs = fon.getLeft();
1795 // if this is an op node, don't stall, copy
1796 // source and delay until we need to use value
1798 // ask whether lhs and rhs sources are dynamic, static, etc.
1800 = vstTableIn.getRefVarSrcType( lhs,
1802 currentSESE.getParent() );
1805 = vstTableIn.getRefVarSrcType( rhs,
1807 currentSESE.getParent() );
1809 if( rhsSrcType.equals( VarSrcTokTable.SrcType_DYNAMIC ) ) {
1810 // if rhs is dynamic going in, lhs will definitely be dynamic
1811 // going out of this node, so track that here
1812 plan.addDynAssign( lhs, rhs );
1813 currentSESE.addDynamicVar( lhs );
1814 currentSESE.addDynamicVar( rhs );
1816 } else if( lhsSrcType.equals( VarSrcTokTable.SrcType_DYNAMIC ) ) {
1817 // otherwise, if the lhs is dynamic, but the rhs is not, we
1818 // need to update the variable's dynamic source as "current SESE"
1819 plan.addDynAssign( lhs );
1822 // only break if this is an ASSIGN op node,
1823 // otherwise fall through to default case
1828 // note that FlatOpNode's that aren't ASSIGN
1829 // fall through to this default case
1832 // a node with no live set has nothing to stall for
1833 if( liveSetIn == null ) {
1837 TempDescriptor[] readarray = fn.readsTemps();
1838 for( int i = 0; i < readarray.length; i++ ) {
1839 TempDescriptor readtmp = readarray[i];
1841 // ignore temps that are definitely available
1842 // when considering to stall on it
1843 if( !notAvailSetIn.contains( readtmp ) ) {
1847 // check the source type of this variable
1849 = vstTableIn.getRefVarSrcType( readtmp,
1851 currentSESE.getParent() );
1853 if( srcType.equals( VarSrcTokTable.SrcType_DYNAMIC ) ) {
1854 // 1) It is not clear statically where this variable will
1855 // come from statically, so dynamically we must keep track
1856 // along various control paths, and therefore when we stall,
1857 // just stall for the exact thing we need and move on
1858 plan.addDynamicStall( readtmp );
1859 currentSESE.addDynamicVar( readtmp );
1861 } else if( srcType.equals( VarSrcTokTable.SrcType_STATIC ) ) {
1862 // 2) Single token/age pair: Stall for token/age pair, and copy
1863 // all live variables with same token/age pair at the same
1864 // time. This is the same stuff that the notavaialable analysis
1865 // marks as now available.
1867 VariableSourceToken vst = vstTableIn.get( readtmp ).iterator().next();
1869 Iterator<VariableSourceToken> availItr =
1870 vstTableIn.get( vst.getSESE(), vst.getAge() ).iterator();
1872 while( availItr.hasNext() ) {
1873 VariableSourceToken vstAlsoAvail = availItr.next();
1875 // only grab additional stuff that is live
1876 Set<TempDescriptor> copySet = new HashSet<TempDescriptor>();
1878 Iterator<TempDescriptor> refVarItr = vstAlsoAvail.getRefVars().iterator();
1879 while( refVarItr.hasNext() ) {
1880 TempDescriptor refVar = refVarItr.next();
1881 if( liveSetIn.contains( refVar ) ) {
1882 copySet.add( refVar );
1886 if( !copySet.isEmpty() ) {
1887 plan.addStall2CopySet( vstAlsoAvail, copySet );
1892 // the other case for srcs is READY, so do nothing
1895 // assert that everything being stalled for is in the
1896 // "not available" set coming into this flat node and
1897 // that every VST identified is in the possible "stall set"
1898 // that represents VST's from children SESE's
1906 // identify sese-age pairs that are statically useful
1907 // and should have an associated SESE variable in code
1908 // JUST GET ALL SESE/AGE NAMES FOR NOW, PRUNE LATER,
1909 // AND ALWAYS GIVE NAMES TO PARENTS
1910 Set<VariableSourceToken> staticSet = vstTableIn.get();
1911 Iterator<VariableSourceToken> vstItr = staticSet.iterator();
1912 while( vstItr.hasNext() ) {
1913 VariableSourceToken vst = vstItr.next();
1915 // placeholder source tokens are useful results, but
1916 // the placeholder static name is never needed
1917 if( vst.getSESE().getIsCallerSESEplaceholder() ) {
1921 FlatSESEEnterNode sese = currentSESE;
1922 while( sese != null ) {
1923 sese.addNeededStaticName(
1924 new SESEandAgePair( vst.getSESE(), vst.getAge() )
1926 sese.mustTrackAtLeastAge( vst.getAge() );
1928 sese = sese.getParent();
1933 codePlans.put( fn, plan );
1936 // if any variables at this-node-*dot* have a static source (exactly one vst)
1937 // but go to a dynamic source at next-node-*dot*, create a new IR graph
1938 // node on that edge to track the sources dynamically
1939 VarSrcTokTable thisVstTable = variableResults.get( fn );
1940 for( int i = 0; i < fn.numNext(); i++ ) {
1941 FlatNode nn = fn.getNext( i );
1942 VarSrcTokTable nextVstTable = variableResults.get( nn );
1943 Set<TempDescriptor> nextLiveIn = livenessRootView.get( nn );
1945 // the table can be null if it is one of the few IR nodes
1946 // completely outside of the root SESE scope
1947 if( nextVstTable != null && nextLiveIn != null ) {
1949 Hashtable<TempDescriptor, VariableSourceToken> static2dynamicSet =
1950 thisVstTable.getStatic2DynamicSet( nextVstTable,
1953 currentSESE.getParent()
1956 if( !static2dynamicSet.isEmpty() ) {
1958 // either add these results to partial fixed-point result
1959 // or make a new one if we haven't made any here yet
1960 FlatEdge fe = new FlatEdge( fn, nn );
1961 FlatWriteDynamicVarNode fwdvn = wdvNodesToSpliceIn.get( fe );
1963 if( fwdvn == null ) {
1964 fwdvn = new FlatWriteDynamicVarNode( fn,
1969 wdvNodesToSpliceIn.put( fe, fwdvn );
1971 fwdvn.addMoreVar2Src( static2dynamicSet );
1979 public void writeReports( String timeReport ) throws java.io.IOException {
1981 BufferedWriter bw = new BufferedWriter( new FileWriter( "mlpReport_summary.txt" ) );
1982 bw.write( "MLP Analysis Results\n\n" );
1983 bw.write( timeReport+"\n\n" );
1984 printSESEHierarchy( bw );
1986 printSESEInfo( bw );
1989 Iterator<Descriptor> methItr = ownAnalysis.descriptorsToAnalyze.iterator();
1990 while( methItr.hasNext() ) {
1991 MethodDescriptor md = (MethodDescriptor) methItr.next();
1992 FlatMethod fm = state.getMethodFlat( md );
1993 bw = new BufferedWriter( new FileWriter( "mlpReport_"+
1994 md.getClassMethodName()+
1995 md.getSafeMethodDescriptor()+
1997 bw.write( "MLP Results for "+md+"\n-------------------\n");
1998 bw.write( "\n\nLive-In, Root View\n------------------\n" +fm.printMethod( livenessRootView ) );
1999 bw.write( "\n\nVariable Results-Out\n----------------\n" +fm.printMethod( variableResults ) );
2000 bw.write( "\n\nNot Available Results-Out\n---------------------\n"+fm.printMethod( notAvailableResults ) );
2001 bw.write( "\n\nCode Plans\n----------\n" +fm.printMethod( codePlans ) );
2002 bw.write("\n\nSESE Effects\n----------------------\n"+printSESEEffects());
2007 private String printSESEEffects() {
2009 StringWriter writer = new StringWriter();
2011 Iterator<FlatSESEEnterNode> keyIter = allSESEs.iterator();
2013 while (keyIter.hasNext()) {
2014 FlatSESEEnterNode seseEnter = keyIter.next();
2015 String result = seseEnter.getSeseEffectsSet().printSet();
2016 if (result.length() > 0) {
2017 writer.write("\nSESE " + seseEnter + "\n");
2018 writer.write(result);
2021 keyIter = rootSESEs.iterator();
2022 while (keyIter.hasNext()) {
2023 FlatSESEEnterNode seseEnter = keyIter.next();
2024 if (seseEnter.getIsCallerSESEplaceholder()) {
2025 if (!seseEnter.getChildren().isEmpty()) {
2026 String result = seseEnter.getSeseEffectsSet().printSet();
2027 if (result.length() > 0) {
2028 writer.write("\nSESE " + seseEnter + "\n");
2029 writer.write(result);
2035 return writer.toString();
2039 private void printSESEHierarchy( BufferedWriter bw ) throws java.io.IOException {
2040 bw.write( "SESE Hierarchy\n--------------\n" );
2041 Iterator<FlatSESEEnterNode> rootItr = rootSESEs.iterator();
2042 while( rootItr.hasNext() ) {
2043 FlatSESEEnterNode root = rootItr.next();
2044 if( root.getIsCallerSESEplaceholder() ) {
2045 if( !root.getChildren().isEmpty() ) {
2046 printSESEHierarchyTree( bw, root, 0 );
2049 printSESEHierarchyTree( bw, root, 0 );
2054 private void printSESEHierarchyTree( BufferedWriter bw,
2055 FlatSESEEnterNode fsen,
2057 ) throws java.io.IOException {
2058 for( int i = 0; i < depth; ++i ) {
2061 bw.write( "- "+fsen.getPrettyIdentifier()+"\n" );
2063 Iterator<FlatSESEEnterNode> childItr = fsen.getChildren().iterator();
2064 while( childItr.hasNext() ) {
2065 FlatSESEEnterNode fsenChild = childItr.next();
2066 printSESEHierarchyTree( bw, fsenChild, depth + 1 );
2071 private void printSESEInfo( BufferedWriter bw ) throws java.io.IOException {
2072 bw.write("\nSESE info\n-------------\n" );
2073 Iterator<FlatSESEEnterNode> rootItr = rootSESEs.iterator();
2074 while( rootItr.hasNext() ) {
2075 FlatSESEEnterNode root = rootItr.next();
2076 if( root.getIsCallerSESEplaceholder() ) {
2077 if( !root.getChildren().isEmpty() ) {
2078 printSESEInfoTree( bw, root );
2081 printSESEInfoTree( bw, root );
2086 private void printSESEInfoTree( BufferedWriter bw,
2087 FlatSESEEnterNode fsen
2088 ) throws java.io.IOException {
2090 if( !fsen.getIsCallerSESEplaceholder() ) {
2091 bw.write( "SESE "+fsen.getPrettyIdentifier()+" {\n" );
2093 bw.write( " in-set: "+fsen.getInVarSet()+"\n" );
2094 Iterator<TempDescriptor> tItr = fsen.getInVarSet().iterator();
2095 while( tItr.hasNext() ) {
2096 TempDescriptor inVar = tItr.next();
2097 if( fsen.getReadyInVarSet().contains( inVar ) ) {
2098 bw.write( " (ready) "+inVar+"\n" );
2100 if( fsen.getStaticInVarSet().contains( inVar ) ) {
2101 bw.write( " (static) "+inVar+"\n" );
2103 if( fsen.getDynamicInVarSet().contains( inVar ) ) {
2104 bw.write( " (dynamic)"+inVar+"\n" );
2108 bw.write( " out-set: "+fsen.getOutVarSet()+"\n" );
2112 Iterator<FlatSESEEnterNode> childItr = fsen.getChildren().iterator();
2113 while( childItr.hasNext() ) {
2114 FlatSESEEnterNode fsenChild = childItr.next();
2115 printSESEInfoTree( bw, fsenChild );