int ls_stdinmode(int onoff) { fd_set rmask; struct timeval timeout; struct lslibNiosHdr reqHdr, replyHdr; sigset_t newMask, oldMask; int cc; if (!nios_ok_) { lserrno = LSE_NIOS_DOWN; return(-1); } if (blockALL_SIGS_(&newMask, &oldMask) < 0) return (-1); SET_LSLIB_NIOS_HDR(reqHdr, (onoff ? LIB_NIOS_REM_ON : LIB_NIOS_REM_OFF), 0); FD_ZERO(&rmask); FD_SET(cli_nios_fd[0], &rmask); timeout.tv_sec = NIOS_TIMEOUT; timeout.tv_usec = 0; if (b_write_fix(cli_nios_fd[0], (char *) &reqHdr, sizeof(reqHdr)) != sizeof(reqHdr)) { lserrno = LSE_MSG_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } if ((cc = select(cli_nios_fd[0] + 1, &rmask, 0, 0, &timeout)) <= 0) { if (cc == 0) lserrno = LSE_TIME_OUT; else lserrno = LSE_SELECT_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } if (b_read_fix(cli_nios_fd[0], (char *) &replyHdr, sizeof(replyHdr)) == -1) { lserrno = LSE_MSG_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } if (replyHdr.opCode != REM_ONOFF) { lserrno = LSE_PROTOC_NIOS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } sigprocmask(SIG_SETMASK, &oldMask, NULL); return(0); }
int ls_donerex(void) { fd_set rmask; struct timeval timeout; struct lslibNiosHdr reqHdr, replyHdr; sigset_t newMask, oldMask; if (!nios_ok_) { lserrno = LSE_NIOS_DOWN; return(-1); } if (blockALL_SIGS_(&newMask, &oldMask) < 0) return (-1); SET_LSLIB_NIOS_HDR(reqHdr, LIB_NIOS_EXIT, 0); FD_ZERO(&rmask); FD_SET(cli_nios_fd[0], &rmask); timeout.tv_sec = NIOS_TIMEOUT; timeout.tv_usec = 0; if (b_write_fix(cli_nios_fd[0], (char *)&reqHdr, sizeof(reqHdr)) != sizeof(reqHdr)) { lserrno = LSE_MSG_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } if (select(cli_nios_fd[0] + 1, &rmask, 0, 0, &timeout) <= 0) { lserrno = LSE_SELECT_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } if (b_read_fix(cli_nios_fd[0], (char *) &replyHdr, sizeof(replyHdr)) == -1) { lserrno = LSE_MSG_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } if (replyHdr.opCode != NIOS_OK) { lserrno = LSE_PROTOC_NIOS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } nios_ok_ = FALSE; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(0); }
int ls_rtaske(char *host, char **argv, int options, char **envp) { static u_short retport; static int rpid; static int reg_ls_donerex = FALSE; struct sockaddr_in sin; int max; char c_chfd[8]; char pathbuf[MAXPATHLEN]; int d; int niosOptions = 0; char *new_argv[5]; int pid; int s; int descriptor[2]; struct resCmdBill cmdmsg; struct lslibNiosRTask taskReq; u_short taskPort = 0; sigset_t newMask; sigset_t oldMask; socklen_t len; if (!reg_ls_donerex) { atexit( (void (*) ()) ls_donerex); reg_ls_donerex = TRUE; } if (_isconnected_(host, descriptor)) s = descriptor[0]; else if ((s = ls_connect(host)) < 0) return(-1); if (blockALL_SIGS_(&newMask, &oldMask) < 0) return (-1); if (!FD_ISSET(s,&connection_ok_)){ FD_SET(s,&connection_ok_); if (ackReturnCode_(s) < 0){ closesocket(s); _lostconnection_(host); sigprocmask(SIG_SETMASK, &oldMask, NULL); return (-1); } } if (!nios_ok_) niosOptions = options & REXF_SYNCNIOS; options &= ~REXF_SYNCNIOS; cmdmsg.options = options; if (cmdmsg.options & REXF_SHMODE) cmdmsg.options |= REXF_USEPTY; if (!isatty(0) && !isatty(1)) cmdmsg.options &= ~REXF_USEPTY; else if (cmdmsg.options & REXF_USEPTY ){ if (options & REXF_TTYASYNC){ if (rstty_async_(host) < 0) { sigprocmask(SIG_SETMASK, &oldMask, NULL); return (-1); } } else { if (rstty_(host) < 0) { sigprocmask(SIG_SETMASK, &oldMask, NULL); return (-1); } } } if ( (genParams_[LSF_INTERACTIVE_STDERR].paramValue != NULL) && (strcasecmp(genParams_[LSF_INTERACTIVE_STDERR].paramValue, "y") == 0) ) { cmdmsg.options |= REXF_STDERR; } if (!nios_ok_) { initSigHandler(SIGTSTP); initSigHandler(SIGTTIN); initSigHandler(SIGTTOU); if (socketpair(AF_UNIX, SOCK_STREAM, 0, cli_nios_fd) < 0) { lserrno = LSE_SOCK_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); closesocket(s); return(-1); } if ((pid = fork()) != 0) { int mypid; close(cli_nios_fd[1]); mypid = getpid(); if (b_write_fix(cli_nios_fd[0], (char *) &mypid, sizeof (mypid)) != sizeof (mypid)) { close(cli_nios_fd[0]); sigprocmask(SIG_SETMASK, &oldMask, NULL); lserrno = LSE_MSG_SYS; return (-1); } if (b_read_fix(cli_nios_fd[0], (char *) &retport, sizeof (u_short)) != sizeof(u_short) ) { close(cli_nios_fd[0]); sigprocmask(SIG_SETMASK, &oldMask, NULL); lserrno = LSE_MSG_SYS; return (-1); } nios_ok_ = TRUE; if (waitpid(pid, 0, 0) <0) { if (errno != ECHILD) { close(cli_nios_fd[0]); nios_ok_ = FALSE; sigprocmask(SIG_SETMASK, &oldMask, NULL); lserrno = LSE_WAIT_SYS; return (-1); } } } else { if (fork()) { max = sysconf(_SC_OPEN_MAX); for (d = 3; d < max; ++d) { (void)close(d); } exit(0); } if (initenv_(NULL, NULL) <0) exit (-1); strcpy(pathbuf, genParams_[LSF_SERVERDIR].paramValue); strcat(pathbuf, "/nios"); sprintf(c_chfd, "%d", cli_nios_fd[1]); new_argv[0] = pathbuf; new_argv[1] = c_chfd; if (cmdmsg.options & REXF_USEPTY) { if (cmdmsg.options & REXF_SHMODE) new_argv[2] = "2"; else new_argv[2] = "1"; } else new_argv[2] = "0"; new_argv[3] = NULL; max = sysconf(_SC_OPEN_MAX); for (d = 3; d < max; ++d) { if (d != cli_nios_fd[1]) (void)close(d); } for (d = 1; d < NSIG; d++) Signal_(d, SIG_DFL); sigprocmask(SIG_SETMASK, &oldMask, NULL); (void)lsfExecvp(new_argv[0], new_argv); exit(-1); } } if (envp) { if (ls_rsetenv_async (host, envp) < 0) { sigprocmask(SIG_SETMASK, &oldMask, NULL); return (-1); } } if (rexcwd_[0] != '\0') strcpy(cmdmsg.cwd, rexcwd_); else if (mygetwd_(cmdmsg.cwd) == 0) { closesocket(s); _lostconnection_(host); lserrno = LSE_WDIR; sigprocmask(SIG_SETMASK, &oldMask, NULL); return (-1); } rpid++; cmdmsg.rpid = rpid; cmdmsg.retport = retport; cmdmsg.argv = argv; cmdmsg.priority = 0; if (sendCmdBill_(s, (resCmd) RES_EXEC, &cmdmsg, NULL, NULL) == -1) { closesocket(s); _lostconnection_(host); sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } if (cmdmsg.options & REXF_TASKPORT) { if ((taskPort = getTaskPort(s)) == 0) { closesocket(s); _lostconnection_(host); sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } } len = sizeof(sin); if (getpeername(s, (struct sockaddr *) &sin, &len) <0) { closesocket(s); _lostconnection_(host); lserrno = LSE_SOCK_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } SET_LSLIB_NIOS_HDR(taskReq.hdr, LIB_NIOS_RTASK, sizeof(taskReq.r)); taskReq.r.pid = rpid; taskReq.r.peer = sin.sin_addr; taskReq.r.pid = (niosOptions & REXF_SYNCNIOS)? -rpid : rpid; if (b_write_fix(cli_nios_fd[0], (char *) &taskReq, sizeof(taskReq)) != sizeof(taskReq)) { closesocket(s); _lostconnection_(host); lserrno = LSE_MSG_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return (-1); } if (tid_register(rpid, s, taskPort, host, options & REXF_TASKINFO) == -1) { closesocket(s); _lostconnection_(host); lserrno = LSE_MALLOC; sigprocmask(SIG_SETMASK, &oldMask, NULL); return (-1); } sigprocmask(SIG_SETMASK, &oldMask, NULL); return (rpid); }
static int rwait_(int tid, LS_WAIT_T *status, int options, struct rusage *ru) { int rpid; struct lslibNiosWaitReq req; struct lslibNiosHdr hdr; fd_set rmask; struct timeval timeout; sigset_t newMask, oldMask; int cc; if (tid < 0) { lserrno = LSE_BAD_ARGS; return -1; } if (!nios_ok_) { lserrno = LSE_NORCHILD; return -1; } blockALL_SIGS_(&newMask, &oldMask); Start: FD_ZERO(&rmask); FD_SET(cli_nios_fd[0], &rmask); timeout.tv_sec = NIOS_TIMEOUT; timeout.tv_usec = 0; SET_LSLIB_NIOS_HDR(req.hdr, LIB_NIOS_RWAIT, sizeof(req.r)); req.r.options = options; req.r.tid = tid; if (b_write_fix(cli_nios_fd[0], (char *) &req, sizeof(req)) != sizeof(req)) { lserrno = LSE_MSG_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return -1; } cc = select(cli_nios_fd[0] + 1, &rmask, 0, 0, &timeout); if (cc <= 0) { if (cc < 0) lserrno = LSE_SELECT_SYS; else lserrno = LSE_TIME_OUT; sigprocmask(SIG_SETMASK, &oldMask, NULL); return -1; } if (b_read_fix(cli_nios_fd[0], (char *) &hdr, sizeof(hdr)) == -1) { lserrno = LSE_MSG_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return -1; } if (WAIT_BLOCK(options) && hdr.opCode == NONB_RETRY) { restartRWait(oldMask); if(!isPamBlockWait){ goto Start; } } switch (hdr.opCode) { case CHILD_FAIL: lserrno = LSE_NORCHILD; sigprocmask(SIG_SETMASK, &oldMask, NULL); return -1; case NONB_RETRY: sigprocmask(SIG_SETMASK, &oldMask, NULL); return 0; case CHILD_OK: rpid = readWaitReply(status, ru); sigprocmask(SIG_SETMASK, &oldMask, NULL); return (rpid); default: lserrno = LSE_PROTOC_NIOS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return -1; } }
int ls_getstdin(int on, int *rpidlist, int maxlen) { fd_set rmask; struct timeval timeout; struct lslibNiosStdin req; struct lslibNiosGetStdinReply reply; sigset_t newMask, oldMask; if (!nios_ok_) { lserrno = LSE_NIOS_DOWN; return(-1); } if (blockALL_SIGS_(&newMask, &oldMask) < 0) return (-1); FD_ZERO(&rmask); FD_SET(cli_nios_fd[0], &rmask); timeout.tv_sec = NIOS_TIMEOUT; timeout.tv_usec = 0; SET_LSLIB_NIOS_HDR(req.hdr, LIB_NIOS_GETSTDIN, sizeof(req.r.set_on)); req.r.set_on = on; if (b_write_fix(cli_nios_fd[0], (char *)&req, sizeof(req.hdr) + sizeof(req.r.set_on)) != sizeof(req.hdr) + sizeof(req.r.set_on)) { lserrno = LSE_MSG_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } if (select(cli_nios_fd[0] + 1, &rmask, 0, 0, &timeout) <= 0) { lserrno = LSE_SELECT_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } if (b_read_fix(cli_nios_fd[0], (char *) &reply.hdr, sizeof(reply.hdr)) == -1) { lserrno = LSE_MSG_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } switch(reply.hdr.opCode) { case STDIN_OK: break; default: lserrno = LSE_PROTOC_NIOS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } if (reply.hdr.len) if (b_read_fix(cli_nios_fd[0], (char *)reply.rpidlist, reply.hdr.len) != reply.hdr.len) { lserrno = LSE_MSG_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } if (reply.hdr.len <= maxlen*sizeof(int)) { memcpy((char *)rpidlist, (char *)reply.rpidlist, reply.hdr.len); sigprocmask(SIG_SETMASK, &oldMask, NULL); return(reply.hdr.len/sizeof(int)); } else { lserrno = LSE_RPIDLISTLEN; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } }
int ls_setstdin(int on, int *rpidlist, int len) { fd_set rmask; struct timeval timeout; struct lslibNiosStdin req; struct lslibNiosHdr replyHdr; sigset_t newMask, oldMask; if (!nios_ok_) { lserrno = LSE_NIOS_DOWN; return(-1); } if (blockALL_SIGS_(&newMask, &oldMask) < 0) return (-1); if ((rpidlist == NULL && len != 0) || (len < 0) || (len > NOFILE)) { lserrno = LSE_SETPARAM; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } FD_ZERO(&rmask); FD_SET(cli_nios_fd[0], &rmask); timeout.tv_sec = NIOS_TIMEOUT; timeout.tv_usec = 0; SET_LSLIB_NIOS_HDR(req.hdr, LIB_NIOS_SETSTDIN, sizeof(req.r) + len * sizeof(int)); req.r.set_on = on; req.r.len = len; if (b_write_fix(cli_nios_fd[0], (char *)&req, sizeof(req.hdr) + sizeof(req.r)) != sizeof(req.hdr) + sizeof(req.r)) { lserrno = LSE_MSG_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } if (rpidlist != NULL && len != 0) { if (b_write_fix(cli_nios_fd[0], (char *)rpidlist, len*sizeof(int)) != len * sizeof(int)) { lserrno = LSE_MSG_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } } if (select(cli_nios_fd[0] + 1, &rmask, 0, 0, &timeout) <= 0) { lserrno = LSE_SELECT_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } if (b_read_fix(cli_nios_fd[0], (char *) &replyHdr, sizeof(replyHdr)) == -1) { lserrno = LSE_MSG_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } switch(replyHdr.opCode) { case STDIN_FAIL: lserrno = LSE_SETPARAM; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); case STDIN_OK: break; default: lserrno = LSE_PROTOC_NIOS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } sigprocmask(SIG_SETMASK, &oldMask, NULL); return(0); }
int ls_setstdout(int on, char *format) { fd_set rmask; struct timeval timeout; struct lslibNiosStdout req; struct lslibNiosHdr replyHdr; sigset_t newMask, oldMask; if (!nios_ok_) { lserrno = LSE_NIOS_DOWN; return(-1); } if (blockALL_SIGS_(&newMask, &oldMask) < 0) return (-1); FD_ZERO(&rmask); FD_SET(cli_nios_fd[0], &rmask); timeout.tv_sec = NIOS_TIMEOUT; timeout.tv_usec = 0; req.r.set_on = on; req.r.len = (format == NULL) ? 0 : strlen(format); if (req.r.len > 0) req.r.len++; SET_LSLIB_NIOS_HDR(req.hdr, LIB_NIOS_SETSTDOUT, sizeof(req.r) + req.r.len*sizeof(char)); if (b_write_fix(cli_nios_fd[0], (char *)&req, sizeof(req.hdr) + sizeof(req.r)) != sizeof(req.hdr) + sizeof(req.r)) { lserrno = LSE_MSG_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } if (req.r.len > 0) { if (b_write_fix(cli_nios_fd[0], (char *)format, req.r.len*sizeof(char)) != req.r.len * sizeof(char)) { lserrno = LSE_MSG_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } } if (select(cli_nios_fd[0] + 1, &rmask, 0, 0, &timeout) <= 0) { lserrno = LSE_SELECT_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } if (b_read_fix(cli_nios_fd[0], (char *) &replyHdr, sizeof(replyHdr)) == -1) { lserrno = LSE_MSG_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } switch(replyHdr.opCode) { case STDOUT_FAIL: lserrno = LSE_SETPARAM; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); case STDOUT_OK: break; default: lserrno = LSE_PROTOC_NIOS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } sigprocmask(SIG_SETMASK, &oldMask, NULL); return(0); }
int ls_niossync(int numTasks) { fd_set rmask; struct timeval timeout; struct lslibNiosHdr replyHdr, reqHdr; sigset_t newMask, oldMask; if (!nios_ok_) { lserrno = LSE_NIOS_DOWN; return(-1); } if (numTasks < 0) { lserrno = LSE_BAD_ARGS; return(-1); } if (blockALL_SIGS_(&newMask, &oldMask) < 0) return (-1); FD_ZERO(&rmask); FD_SET(cli_nios_fd[0], &rmask); timeout.tv_sec = NIOS_TIMEOUT; timeout.tv_usec = 0; SET_LSLIB_NIOS_HDR(reqHdr, LIB_NIOS_SYNC, numTasks); if (b_write_fix(cli_nios_fd[0], (char *)&reqHdr, sizeof(reqHdr)) != sizeof(reqHdr)) { lserrno = LSE_MSG_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } if (select(cli_nios_fd[0] + 1, &rmask, 0, 0, &timeout) <= 0) { lserrno = LSE_SELECT_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } if (b_read_fix(cli_nios_fd[0], (char *) &replyHdr, sizeof(replyHdr)) == -1) { lserrno = LSE_MSG_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } switch(replyHdr.opCode) { case SYNC_FAIL: lserrno = LSE_SETPARAM; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); case SYNC_OK: break; default: lserrno = LSE_PROTOC_NIOS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return(-1); } sigprocmask(SIG_SETMASK, &oldMask, NULL); return(0); }
int callRes_(int s, resCmd cmd, char *data, char *reqBuf, int reqLen, bool_t (*xdrFunc)(), int *rd, struct timeval *timeout, struct lsfAuth *auth) { int cc; int t; struct sockaddr_in from; socklen_t fromsize; int nready; sigset_t oldMask; sigset_t newMask; char hostname[MAXHOSTNAMELEN]; blockALL_SIGS_(&newMask, &oldMask); memset(hostname, 0, sizeof(hostname)); currentSN = REQUESTSN; gethostbysock_(s, hostname); if (strcmp(hostname, "LSF_HOST_NULL")) _setcurseqno_(hostname, currentSN); if ((cc = lsSendMsg_(s, cmd, 0, data, reqBuf, reqLen, xdrFunc, b_write_fix, auth)) < 0) { sigprocmask(SIG_SETMASK, &oldMask, NULL); return -1; } if (!rd) { sigprocmask(SIG_SETMASK, &oldMask, NULL); return 0; } nready = rd_select_(*rd, timeout); if (nready <= 0) { if (nready == 0) lserrno = LSE_TIME_OUT; else lserrno = LSE_SELECT_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return -1; } fromsize = sizeof(from); t = b_accept_(*rd, (struct sockaddr *)&from, &fromsize); if (t < 0) { lserrno = LSE_ACCEPT_SYS; sigprocmask(SIG_SETMASK, &oldMask, NULL); return -1; } closesocket(*rd); *rd = t; sigprocmask(SIG_SETMASK, &oldMask, NULL); return 0; } /* callRes_() */