summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--mteval/external_scorer.cc7
-rw-r--r--mteval/ns_ext.cc16
2 files changed, 23 insertions, 0 deletions
diff --git a/mteval/external_scorer.cc b/mteval/external_scorer.cc
index f1b3ed6e..c7c3de1a 100644
--- a/mteval/external_scorer.cc
+++ b/mteval/external_scorer.cc
@@ -4,6 +4,7 @@
#include <cstdlib>
#include <cstring>
#include <unistd.h>
+#include <signal.h>
#include <sstream>
#include <iostream>
#include <cassert>
@@ -15,6 +16,7 @@
using namespace std;
extern const char* meteor_jar_path;
+extern void metric_child_signal_handler(int);
map<string, boost::shared_ptr<ScoreServer> > ScoreServerManager::servers_;
@@ -46,6 +48,11 @@ ScoreServer* ScoreServerManager::Instance(const string& score_type) {
}
ScoreServer::ScoreServer(const string& cmd) {
+ static bool need_init = true;
+ if (need_init) {
+ need_init = false;
+ signal(SIGCHLD, metric_child_signal_handler);
+ }
cerr << "Invoking " << cmd << " ..." << endl;
if (pipe(p2c) < 0) { perror("pipe"); exit(1); }
if (pipe(c2p) < 0) { perror("pipe"); exit(1); }
diff --git a/mteval/ns_ext.cc b/mteval/ns_ext.cc
index 1e7e2bc1..9d2c75c6 100644
--- a/mteval/ns_ext.cc
+++ b/mteval/ns_ext.cc
@@ -4,6 +4,8 @@
#include <cstdlib>
#include <cstring>
#include <unistd.h>
+#include <signal.h>
+#include <sys/wait.h>
#include <sstream>
#include <iostream>
#include <cassert>
@@ -13,6 +15,14 @@
using namespace std;
+void metric_child_signal_handler(int signo) {
+ int status = 0;
+ cerr << "Received SIGCHLD(" << signo << ") ... aborting.\n";
+ // reap zombies
+ while (waitpid(-1, &status, WNOHANG) > 0) {}
+ abort();
+}
+
struct NScoreServer {
NScoreServer(const std::string& cmd);
~NScoreServer();
@@ -27,6 +37,12 @@ struct NScoreServer {
};
NScoreServer::NScoreServer(const string& cmd) {
+ static bool need_init = true;
+ if (need_init) {
+ need_init = false;
+ signal(SIGCHLD, metric_child_signal_handler);
+ }
+
cerr << "Invoking " << cmd << " ..." << endl;
if (pipe(p2c) < 0) { perror("pipe"); exit(1); }
if (pipe(c2p) < 0) { perror("pipe"); exit(1); }