Bug 2540 - trap ctrl-C and unlink output files

1 files changed, 5 insertions, 0 deletions

diff --git a/cpp/src/slice2rb/Main.cpp b/cpp/src/slice2rb/Main.cpp
index 67bae275a63..8250a4a49ca 100644
--- a/cpp/src/slice2rb/Main.cpp
+++ b/cpp/src/slice2rb/Main.cpp
@@ -11,6 +11,7 @@
 #include <IceUtil/Options.h>
 #include <Slice/Preprocessor.h>
 #include <Slice/RubyUtil.h>
+#include <Slice/SignalHandler.h>
 
 #include <fstream>
 
@@ -137,6 +138,8 @@ main(int argc, char* argv[])
 
     for(i = args.begin(); i != args.end(); ++i)
     {
+        SignalHandler sigHandler;
+
         Preprocessor icecpp(argv[0], *i, cppArgs);
         FILE* cppHandle = icecpp.preprocess(false);
 
@@ -189,6 +192,7 @@ main(int argc, char* argv[])
                 {
                     file = output + '/' + file;
                 }
+                SignalHandler::addFile(file);
 
                 IceUtilInternal::Output out;
                 out.open(file.c_str());
@@ -206,6 +210,7 @@ main(int argc, char* argv[])
                 // Generate the Ruby mapping.
                 //
                 generate(u, all, checksum, includePaths, out);
+
             }
 
             u->destroy();