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);
}
