From: Chris Lattner <sabre@nondot.org>
Date: Sat, 10 Jan 2004 21:36:49 +0000 (+0000)
Subject: Update obsolete comments
X-Git-Url: http://demsky.eecs.uci.edu/git/?a=commitdiff_plain;h=2cacc0a30630e58379f3c57ccb7b161cebb550ca;p=oota-llvm.git

Update obsolete comments
Fix iterator invalidation problems which was causing -mstrip to miss some
entries, and read free'd memory.  This shrinks the symbol table of 254.gap
from 333 to 284 bytes!  :)


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@10751 91177308-0d34-0410-b5e6-96231b3b80d8
---

diff --git a/lib/Transforms/Scalar/SymbolStripping.cpp b/lib/Transforms/Scalar/SymbolStripping.cpp
index 32ebdd36cbc..c43a2cb76e0 100644
--- a/lib/Transforms/Scalar/SymbolStripping.cpp
+++ b/lib/Transforms/Scalar/SymbolStripping.cpp
@@ -10,9 +10,9 @@
 // This file implements stripping symbols out of symbol tables.
 //
 // Specifically, this allows you to strip all of the symbols out of:
-//   * A function
 //   * All functions in a module
-//   * All symbols in a module (all function symbols + all module scope symbols)
+//   * All non-essential symbols in a module (all function symbols + all module
+//     scope symbols)
 //
 // Notice that:
 //   * This pass makes code much less readable, so it should only be used in
@@ -30,12 +30,15 @@ using namespace llvm;
 static bool StripSymbolTable(SymbolTable &SymTab) {
   bool RemovedSymbol = false;
 
-  for (SymbolTable::iterator I = SymTab.begin(); I != SymTab.end(); ++I) {
-    std::map<const std::string, Value *> &Plane = I->second;
+  for (SymbolTable::iterator I = SymTab.begin(); I != SymTab.end();) {
+    // Removing items from the plane can cause the plane itself to get deleted.
+    // If this happens, make sure we incremented our plane iterator already!
+    std::map<const std::string, Value *> &Plane = (I++)->second;
     
-    SymbolTable::type_iterator B = Plane.begin();
-    while (B != Plane.end()) {   // Found nonempty type plane!
+    SymbolTable::type_iterator B = Plane.begin(), Bend = Plane.end();
+    while (B != Bend) {   // Found nonempty type plane!
       Value *V = B->second;
+
       if (isa<Constant>(V) || isa<Type>(V)) {
 	SymTab.type_remove(B++);
         RemovedSymbol = true;