commit c053644217dd35d492701d290052130e1d76c0df
author Benjamin Kramer <benny.kra@googlemail.com> Thu Feb 24 11:03:19 2011 +0000
committer Benjamin Kramer <benny.kra@googlemail.com> Thu Feb 24 11:03:19 2011 +0000
tree d10c02b8fece631fef2d5997244903c70da1bd1c
parent a62fc89a04b2fb9d15a7c14332bdfcfb59adc129

Plug some leaks in edis.

- Don't leak parsed operands during tokenization.
- Don't leak printed insts in llvm-mc.

llvm-svn: 126381

llvm/lib/MC/MCDisassembler/EDToken.cpp

diff --git a/llvm/lib/MC/MCDisassembler/EDToken.cpp b/llvm/lib/MC/MCDisassembler/EDToken.cpp
index 400e164..de770b4 100644
--- a/llvm/lib/MC/MCDisassembler/EDToken.cpp
+++ b/llvm/lib/MC/MCDisassembler/EDToken.cpp
@@ -194,6 +194,10 @@
     tokens.push_back(token);
   }
   
+  // Free any parsed operands.
+  for (unsigned i = 0, e = parsedOperands.size(); i != e; ++i)
+    delete parsedOperands[i];
+
   return 0;
 }
 

llvm/tools/llvm-mc/Disassembler.cpp

diff --git a/llvm/tools/llvm-mc/Disassembler.cpp b/llvm/tools/llvm-mc/Disassembler.cpp
index 1f0850f..d98b57e 100644
--- a/llvm/tools/llvm-mc/Disassembler.cpp
+++ b/llvm/tools/llvm-mc/Disassembler.cpp
@@ -227,8 +227,8 @@
   }
   
   EDDisassembler::initialize();
-  EDDisassembler *disassembler =
-    EDDisassembler::getDisassembler(TS.c_str(), AS);
+  OwningPtr<EDDisassembler>
+    disassembler(EDDisassembler::getDisassembler(TS.c_str(), AS));
   
   if (disassembler == 0) {
     errs() << "error: couldn't get disassembler for " << TS << '\n';
@@ -236,8 +236,8 @@
   }
   
   while (ByteArray.size()) {
-    EDInst *inst =
-      disassembler->createInst(byteArrayReader, 0, &ByteArray);
+    OwningPtr<EDInst>
+      inst(disassembler->createInst(byteArrayReader, 0, &ByteArray));
   
     ByteArray.erase (ByteArray.begin(), ByteArray.begin() + inst->byteSize());
                                
@@ -330,7 +330,7 @@
       }
       
       uint64_t evaluatedResult;
-      void *Arg[] = { disassembler, &Out };
+      void *Arg[] = { disassembler.get(), &Out };
       if (operand->evaluate(evaluatedResult, verboseEvaluator, Arg)) {
         errs() << "error: Couldn't evaluate an operand\n";
         return -1;