static void runCoordinatorCmd(char c,
int *coordCmdStatus = NULL,
int *numPeers = NULL,
int *isRunning = NULL)
{
_dmtcp_lock();
{
CoordinatorAPI coordinatorAPI;
dmtcp_disable_ckpt();
coordinatorAPI.connectAndSendUserCommand(c, coordCmdStatus, numPeers,
isRunning);
dmtcp_enable_ckpt();
}
_dmtcp_unlock();
}
int main ( int argc, char** argv )
{
bool quiet = false;
dmtcp::string interval = "";
dmtcp::string request = "h";
initializeJalib();
Util::initializeLogFile();
//process args
shift;
while(argc>0){
dmtcp::string s = argv[0];
if((s=="--help" || s=="-h") && argc==1){
fprintf(stderr, theUsage, "");
return 1;
} else if ((s=="--version") && argc==1){
JASSERT_STDERR << DMTCP_VERSION_AND_COPYRIGHT_INFO;
return 1;
}else if(argc>1 && (s == "-h" || s == "--host")){
setenv(ENV_VAR_NAME_HOST, argv[1], 1);
shift; shift;
}else if(argc>1 && (s == "-p" || s == "--port")){
setenv(ENV_VAR_NAME_PORT, argv[1], 1);
shift; shift;
}else if(s == "--quiet"){
quiet = true;
shift;
}else if(s == "h" || s == "-h" || s == "--help" || s == "?"){
fprintf(stderr, theUsage, "");
return 1;
}else{ // else it's a request
char* cmd = argv[0];
//ignore leading dashes
while(*cmd == '-') cmd++;
s = cmd;
if(*cmd == 'b' && *(cmd+1) != 'c'){
// If blocking ckpt, next letter must be 'c'; else print the usage
fprintf(stderr, theUsage, "");
return 1;
} else if (*cmd == 's' || *cmd == 'i' || *cmd == 'c' || *cmd == 'b'
|| *cmd == 'f' || *cmd == 'k' || *cmd == 'q') {
request = s;
if (*cmd == 'i') {
if (isdigit(cmd[1])) { // if -i5, for example
interval = cmd+1;
} else { // else -i 5
if (argc == 1) {
fprintf(stderr, theUsage, "");
return 1;
}
interval = argv[1];
shift;
}
}
shift;
}else{
fprintf(stderr, theUsage, "");
return 1;
}
}
}
if (! quiet)
printf( "DMTCP-" PACKAGE_VERSION " (+ MTCP), Copyright (C) 2006-2011"
" Jason Ansel, Michael Rieker,\n"
" Kapil Arya, and Gene Cooperman\n"
"This program comes with ABSOLUTELY NO WARRANTY.\n"
"This is free software, and you are welcome to redistribute it\n"
"under certain conditions; see COPYING file for details.\n"
"(Use flag \"--quiet\" to hide this message.)\n\n");
int coordErrorCode = CoordinatorAPI::NOERROR;
int numPeers;
int isRunning;
CoordinatorAPI coordinatorAPI;
char *cmd = (char *)request.c_str();
switch (*cmd) {
case 'h':
fprintf(stderr, theUsage, "");
return 1;
case 'i':
setenv(ENV_VAR_CKPT_INTR, interval.c_str(), 1);
coordinatorAPI.connectAndSendUserCommand(*cmd, &coordErrorCode);
printf("Interval changed to %s\n", interval.c_str());
break;
case 'b':
// blocking prefix
coordinatorAPI.connectAndSendUserCommand(*cmd, &coordErrorCode);
// actual command
coordinatorAPI.connectAndSendUserCommand(*(cmd+1), &coordErrorCode);
break;
case 's':
coordinatorAPI.connectAndSendUserCommand(*cmd, &coordErrorCode,
&numPeers, &isRunning);
case 'c':
case 'f':
case 'k':
case 'q':
coordinatorAPI.connectAndSendUserCommand(*cmd, &coordErrorCode);
break;
}
//check for error
if (coordErrorCode != CoordinatorAPI::NOERROR) {
switch(coordErrorCode){
case CoordinatorAPI::ERROR_COORDINATOR_NOT_FOUND:
if (getenv("DMTCP_PORT"))
fprintf(stderr, "Coordinator not found. Please check port and host.\n");
else
fprintf(stderr,
"Coordinator not found. Try specifying port with \'--port\'.\n");
break;
case CoordinatorAPI::ERROR_INVALID_COMMAND:
fprintf(stderr,
"Unknown command: %c, try 'dmtcp_command --help'\n", *cmd);
break;
case CoordinatorAPI::ERROR_NOT_RUNNING_STATE:
fprintf(stderr, "Error, computation not in running state."
" Either a checkpoint is\n"
" currently happening or there are no connected processes.\n");
break;
default:
fprintf(stderr, "Unknown error\n");
break;
}
return 2;
}
if(*cmd == 's'){
if (getenv(ENV_VAR_NAME_HOST))
printf(" Host: %s\n", getenv(ENV_VAR_NAME_HOST));
printf(" Port: %s\n", getenv(ENV_VAR_NAME_PORT));
printf("Status...\n");
printf("NUM_PEERS=%d\n", numPeers);
printf("RUNNING=%s\n", (isRunning?"yes":"no"));
}
return 0;
}
int main ( int argc, char** argv )
{
string interval = "";
string request = "h";
initializeJalib();
// No need to initialize the log file.
// Util::initializeLogFile();
//process args
shift;
while(argc>0){
string s = argv[0];
if((s=="--help" || s=="-h") && argc==1){
printf("%s", theUsage);
return 1;
} else if ((s=="--version") && argc==1){
printf("%s", DMTCP_VERSION_AND_COPYRIGHT_INFO);
return 1;
}else if(argc>1 && (s == "-h" || s == "--coord-host" || s == "--host")){
setenv(ENV_VAR_NAME_HOST, argv[1], 1);
shift; shift;
} else if (argc>1 && (s == "-p" || s == "--coord-port" || s == "--port")) {
setenv(ENV_VAR_NAME_PORT, argv[1], 1);
shift; shift;
} else if (argv[0][0] == '-' && argv[0][1] == 'p' &&
isdigit(argv[0][2])) { // else if -p0, for example
setenv(ENV_VAR_NAME_PORT, argv[0]+2, 1);
shift;
}else if(s == "h" || s == "-h" || s == "--help" || s == "?"){
fprintf(stderr, theUsage, "");
return 1;
}else{ // else it's a request
char* cmd = argv[0];
//ignore leading dashes
while(*cmd == '-') cmd++;
s = cmd;
if((*cmd == 'b' || *cmd == 'x') && *(cmd+1) != 'c'){
// If blocking ckpt, next letter must be 'c'; else print the usage
fprintf(stderr, theUsage, "");
return 1;
} else if (*cmd == 's' || *cmd == 'i' || *cmd == 'c' || *cmd == 'b' ||
*cmd == 'x' || *cmd == 'k' || *cmd == 'q') {
request = s;
if (*cmd == 'i') {
if (isdigit(cmd[1])) { // if -i5, for example
interval = cmd+1;
} else { // else -i 5
if (argc == 1) {
fprintf(stderr, theUsage, "");
return 1;
}
interval = argv[1];
shift;
}
}
shift;
}else{
fprintf(stderr, theUsage, "");
return 1;
}
}
}
int coordCmdStatus = CoordCmdStatus::NOERROR;
int numPeers;
int isRunning;
int ckptInterval;
CoordinatorAPI coordinatorAPI;
char *cmd = (char *)request.c_str();
switch (*cmd) {
case 'h':
fprintf(stderr, theUsage, "");
return 1;
case 'i':
setenv(ENV_VAR_CKPT_INTR, interval.c_str(), 1);
coordinatorAPI.connectAndSendUserCommand(*cmd, &coordCmdStatus);
printf("Interval changed to %s\n", interval.c_str());
break;
case 'b':
case 'x':
// blocking prefix
coordinatorAPI.connectAndSendUserCommand(*cmd, &coordCmdStatus);
// actual command
coordinatorAPI.connectAndSendUserCommand(*(cmd+1), &coordCmdStatus);
break;
case 's':
coordinatorAPI.connectAndSendUserCommand(*cmd, &coordCmdStatus,
&numPeers, &isRunning, &ckptInterval);
case 'c':
case 'k':
case 'q':
coordinatorAPI.connectAndSendUserCommand(*cmd, &coordCmdStatus);
break;
}
//check for error
if (coordCmdStatus != CoordCmdStatus::NOERROR) {
switch(coordCmdStatus){
case CoordCmdStatus::ERROR_COORDINATOR_NOT_FOUND:
if (getenv("DMTCP_COORD_PORT") || getenv("DMTCP_PORT"))
fprintf(stderr, "Coordinator not found. Please check port and host.\n");
else
fprintf(stderr,
"Coordinator not found. Try specifying port with \'--port\'.\n");
break;
case CoordCmdStatus::ERROR_INVALID_COMMAND:
fprintf(stderr,
"Unknown command: %c, try 'dmtcp_command --help'\n", *cmd);
break;
case CoordCmdStatus::ERROR_NOT_RUNNING_STATE:
fprintf(stderr, "Error, computation not in running state."
" Either a checkpoint is\n"
" currently happening or there are no connected processes.\n");
break;
default:
fprintf(stderr, "Unknown error\n");
break;
}
return 2;
}
#define QUOTE(arg) #arg
#define STRINGIFY(arg) QUOTE(arg)
if(*cmd == 's'){
printf("Coordinator:\n");
char *host = getenv(ENV_VAR_NAME_HOST);
if (host == NULL) host = getenv("DMTCP_HOST"); // deprecated
printf(" Host: %s\n", (host ? host : "localhost"));
char *port = getenv(ENV_VAR_NAME_PORT);
if (port == NULL) port = getenv("DMTCP_PORT"); // deprecated
printf(" Port: %s\n",
(port ? port : STRINGIFY(DEFAULT_PORT) " (default port)"));
printf("Status...\n");
printf(" NUM_PEERS=%d\n", numPeers);
printf(" RUNNING=%s\n", (isRunning?"yes":"no"));
if (ckptInterval) {
printf(" CKPT_INTERVAL=%d\n", ckptInterval);
} else {
printf(" CKPT_INTERVAL=0 (checkpoint manually)\n");
}
}
return 0;
}