static void _read_nodes_from_stdin (struct ipmidetect_arguments *cmd_args) { char buf[IPMIDETECT_STDIN_BUFFERLEN]; int n; assert (cmd_args); memset (buf, '\0', IPMIDETECT_STDIN_BUFFERLEN); if ((n = fd_read_n (STDIN_FILENO, buf, IPMIDETECT_STDIN_BUFFERLEN)) < 0) err_exit ("error reading from standard input: %s", strerror (errno)); if (n == IPMIDETECT_STDIN_BUFFERLEN) err_exit ("overflow in standard input buffer"); if (n > 0) { char *ptr = strtok (buf, " \t\n\0"); while (ptr) { _push_inputted_nodes (cmd_args, ptr); ptr = strtok (NULL, " \t\n\0"); } } }
static int _test_cpu_owner_lock(int cpu_id, uint32_t job_id) { char tmp[64]; uint32_t in_job_id; int fd, sz; if (!slurmd_spooldir) slurmd_spooldir = slurm_get_slurmd_spooldir(); snprintf(tmp, sizeof(tmp), "%s/cpu", slurmd_spooldir); if ((mkdir(tmp, 0700) != 0) && (errno != EEXIST)) { error("%s: mkdir failed: %m %s", __func__, tmp); return -1; } snprintf(tmp, sizeof(tmp), "%s/cpu/%d", slurmd_spooldir, cpu_id); fd = open(tmp, O_RDWR, 0600); if (fd < 0) { if (errno != ENOENT) /* Race condition */ error("%s: open: %m %s", __func__, tmp); return -1; } if (_fd_lock_retry(fd) < 0) { error("%s: fd_get_write_lock: %m %s", __func__, tmp); close(fd); return -1; } sz = sizeof(uint32_t); if (fd_read_n(fd, (void *) &in_job_id, sz) != sz) { error("%s: read: %m %s", __func__, tmp); close(fd); return -1; } if (job_id != in_job_id) { /* Result of various race conditions */ debug("%s: CPU %d now owned by job %u rather than job %u", __func__, cpu_id, in_job_id, job_id); close(fd); return -1; } debug("%s: CPU %d owned by job %u as expected", __func__, cpu_id, job_id); return fd; }
/* * Derived from the rcmd() libc call, with modified interface. * Is MT-safe if gethostbyname_r is defined. * Connection can time out. * * ahost (IN) target hostname * port (IN) port to connect to * remuser (IN) remote username * cmd (IN) remote command to execute under shell * fd2p (IN) if non NULL, return stderr file descriptor here * int (RETURN) socket for I/O on success */ int mcmd(char **ahost, int port, char *remuser, char *cmd, int *fd2p, char *munge_socket) { struct sockaddr m_socket; struct sockaddr_in *getp; struct sockaddr_in sin, from; struct sockaddr_storage ss; struct hostent *h_ent = NULL; struct in_addr m_in; unsigned int rand, randl; unsigned int randy = 0; int s, s2, rv, mcount, lport; char c; char num[6] = {0}; char *mptr; char *mbuf; char *tmbuf; char *m; char *mpvers; char num_seq[12] = {0}; socklen_t len; sigset_t blockme; sigset_t oldset; #ifdef HAVE_GETHOSTBYNAME_R_6 struct hostent h_entry; int h_ent_bsize = HBUF_LEN; char h_ent_buf[HBUF_LEN] = {0}; #endif int h_ent_err = 0; unsigned char *hptr; char haddrdot[MAXHOSTNAMELEN + MRSH_LOCALHOST_KEYLEN + 1] = {0}; munge_ctx_t ctx; sigemptyset(&blockme); sigaddset(&blockme, SIGURG); sigaddset(&blockme, SIGPIPE); SET_PTHREAD(); if (fd2p != NULL) { /* * Generate a random number to send in our package to the * server. We will see it again and compare it when the * server sets up the stderr socket and sends it to us. * We need to loop for the tiny possibility we read 0 :P */ int rand_fd; if ((rand_fd = open ("/dev/urandom", O_RDONLY | O_NONBLOCK)) < 0) { perror("mcmd: Open of /dev/urandom failed"); exit(1); } do { if ((rv = read (rand_fd, &randy, sizeof(uint32_t))) < 0) { perror("mcmd: Read of /dev/urandom failed"); close(rand_fd); exit(1); } if (rv < (int) (sizeof(uint32_t))) { perror("mcmd: Read returned too few bytes"); close(rand_fd); exit(1); } } while (randy == 0); close(rand_fd); } /* Convert to decimal string, is 0 if we don't want stderr. */ snprintf(num_seq, sizeof(num_seq),"%d",randy); /* * Start setup of the stdin/stdout socket... */ lport = 0; len = sizeof(struct sockaddr_in); if ((s = socket(AF_INET, SOCK_STREAM, 0)) < 0) { perror("mcmd: socket call stdout failed"); exit(1); } memset (&ss, '\0', sizeof(ss)); ss.ss_family = AF_INET; if (bind(s, (struct sockaddr *)&ss, len) < 0) { perror("mcmd: bind failed"); goto bad; } sin.sin_family = AF_INET; #ifdef HAVE_GETHOSTBYNAME_R_6 (void) gethostbyname_r(*ahost, &h_entry, &h_ent_buf[0], h_ent_bsize, &h_ent, &h_ent_err); #else h_ent = gethostbyname(*ahost); h_ent_err = h_errno; #endif if (h_ent == NULL) { switch (h_ent_err) { case HOST_NOT_FOUND: fprintf(stderr,"mcmd: Hostname not found.\n"); goto bad; case NO_ADDRESS: fprintf(stderr,"mcmd: Can't find IP address.\n"); goto bad; case NO_RECOVERY: fprintf(stderr,"mcmd: A non-recoverable error.\n"); goto bad; case TRY_AGAIN: fprintf(stderr,"mcmd: Error on name server.\n"); goto bad; default: fprintf(stderr,"mcmd: Unknown error.\n"); goto bad; } } memcpy(&sin.sin_addr.s_addr, *h_ent->h_addr_list, h_ent->h_length); sin.sin_port = port; if (connect(s, (struct sockaddr *)&sin, sizeof(sin)) < 0) { perror("mcmd: connect failed"); goto bad; } /* save address in buffer */ if ((strcmp(*ahost, "localhost") == 0) || (strcmp(*ahost, "127.0.0.1") == 0)) { /* Special case for localhost */ char hostname[MAXHOSTNAMELEN+1]; memset(hostname, '\0', MAXHOSTNAMELEN+1); if (gethostname(hostname, MAXHOSTNAMELEN) < 0) { perror("mcmd: gethostname call failed"); exit(1); } strncpy(haddrdot, MRSH_LOCALHOST_KEY, MRSH_LOCALHOST_KEYLEN); strncat(haddrdot, hostname, MAXHOSTNAMELEN); } else { memcpy(&m_in.s_addr, *h_ent->h_addr_list, h_ent->h_length); hptr = (unsigned char *) &m_in; sprintf(haddrdot, "%u.%u.%u.%u", hptr[0], hptr[1], hptr[2], hptr[3]); } lport = 0; s2 = -1; if (fd2p != NULL) { /* * Start the socket setup for the stderr. */ struct sockaddr_in sin2; if ((s2 = socket(AF_INET, SOCK_STREAM, 0)) < 0) { perror("mcmd: socket call for stderr failed"); goto bad; } memset (&sin2, 0, sizeof(sin2)); sin2.sin_family = AF_INET; sin2.sin_addr.s_addr = htonl(INADDR_ANY); sin2.sin_port = 0; if (bind(s2, (struct sockaddr *)&sin2, sizeof(sin2)) < 0) { perror("mcmd: bind failed"); close(s2); goto bad; } len = sizeof(struct sockaddr); /* * Retrieve our port number so we can hand it to the server * for the return (stderr) connection... */ if (getsockname(s2,&m_socket,&len) < 0) { perror("mcmd: getsockname failed"); close(s2); goto bad; } getp = (struct sockaddr_in *)&m_socket; lport = ntohs(getp->sin_port); if (listen(s2, 5) < 0) { perror("mcmd: listen() failed"); close(s2); goto bad; } } /* put port in buffer. will be 0 if user didn't want stderr */ snprintf(num,sizeof(num),"%d",lport); /* * We call munge_encode which will take what we write in and * return a pointer to an munged buffer. What we get back is * a null terminated string of encrypted characters. * * The format of the unmunged buffer is as follows (each a * string terminated with a '\0' (null): * * stderr_port_number & random_number are 0 if user did not * request stderr socket * * SIZE EXAMPLE * ========== ============= * remote_user_name variable "mhaskell" * '\0' * protocol version < 12 bytes "1.2" * '\0' * IP address of requestor [1] 7-15 bytes "134.9.11.155" * '\0' * stderr_port_number 4-8 bytes "50111" * '\0' * random_number 1-8 bytes "1f79ca0e" * '\0' * users_command variable "ls -al" * '\0' '\0' * * (The last extra null is accounted for in the following * line's last strlen() call.) * * [1] - With the exception when 127.0.0.1 or "localhost" are * input by the user. In that situation, the MRSH_LOCALHOST_KEY * and hostname are concatenated and the size may be much larger * than 7-15 bytes. */ mpvers = MRSH_PROTOCOL_VERSION; mcount = ((strlen(remuser)+1) + (strlen(mpvers)+1) + (strlen(haddrdot)+1) + (strlen(num)+1) + (strlen(num_seq)+1) + strlen(cmd)+2); tmbuf = mbuf = malloc(mcount); if (tmbuf == NULL) { perror("mcmd: Error from malloc"); close(s2); goto bad; } /* * The following memset() call takes the extra trailing null * as part of its count as well. */ memset(mbuf,0,mcount); mptr = strcpy(mbuf, remuser); mptr += strlen(remuser)+1; mptr = strcpy(mptr, mpvers); mptr += strlen(mpvers)+1; mptr = strcpy(mptr, haddrdot); mptr += strlen(haddrdot)+1; mptr = strcpy(mptr, num); mptr += strlen(num)+1; mptr = strcpy(mptr, num_seq); mptr += strlen(num_seq)+1; mptr = strcpy(mptr, cmd); if ((ctx = munge_ctx_create()) == NULL) { fprintf(stderr, "munge_ctx_create: %s\n", strerror(errno)); close(s2); free(tmbuf); goto bad; } if (munge_socket) { if ((rv = munge_ctx_set (ctx, MUNGE_OPT_SOCKET, munge_socket)) != EMUNGE_SUCCESS) { fprintf(stderr,"munge_ctx_set: %s\n", munge_ctx_strerror(ctx)); munge_ctx_destroy(ctx); close(s2); free(tmbuf); goto bad; } } if ((rv = munge_encode(&m,ctx,mbuf,mcount)) != EMUNGE_SUCCESS) { fprintf(stderr,"munge_encode: %s\n", munge_ctx_strerror(ctx)); munge_ctx_destroy(ctx); close(s2); free(tmbuf); goto bad; } munge_ctx_destroy(ctx); mcount = (strlen(m)+1); /* * Write stderr port in the clear in case we can't decode for * some reason (i.e. bad credentials). May be 0 if user * doesn't want stderr. */ if (fd2p != NULL) { rv = fd_write_n(s, num, strlen(num)+1); if (rv != (ssize_t)(strlen(num)+1)) { free(m); free(tmbuf); if (rv == -1) { if (errno == EPIPE) perror("mcmd: Lost connection (EPIPE)"); else perror("mcmd: Write of stderr port"); } else fprintf(stderr, "mcmd: write incorrect number of bytes.\n"); close(s2); goto bad; } } else { write(s, "", 1); lport = 0; } /* * Write the munge_encoded blob to the socket. */ if ((rv = fd_write_n(s, m, mcount)) != mcount) { free(m); free(tmbuf); if (rv == -1) { if (errno == EPIPE) perror("mcmd: Lost connection (EPIPE)"); else perror("mcmd: Write of munge data"); } else fprintf(stderr, "mcmd: write incorrect number of bytes.\n"); close(s2); goto bad; } free(m); free(tmbuf); if (fd2p != NULL) { /* * Wait for stderr connection from daemon. */ int maxfd, s3; fd_set reads; errno = 0; FD_ZERO(&reads); FD_SET(s, &reads); FD_SET(s2, &reads); maxfd = (s > s2) ? s : s2; if (select(maxfd + 1, &reads, 0, 0, 0) < 1 || !FD_ISSET(s2, &reads)) { if (errno != 0) perror("mcmd: Select failed (setting up stderr)"); else { char buf[100]; int rv = read(s, buf, 100); if (rv == 0) fprintf(stderr, "mcmd: Connection closed by remote host.\n"); else if (rv > 0) fprintf(stderr, "mcmd: Protocol failure in circuit setup.\n"); else /* rv < 0 */ fprintf(stderr, "mcmd: %s\n", strerror(errno)); } close(s2); goto bad; } errno = 0; len = sizeof(from); /* arg to accept */ if ((s3 = accept(s2, (struct sockaddr *)&from, &len)) < 0) { perror("mcmd: accept (stderr) failed"); close(s2); goto bad; } if (from.sin_family != AF_INET) { fprintf(stderr, "mcmd: bad family type: %d\n", from.sin_family); goto bad2; } close(s2); /* * The following fixes a race condition between the daemon * and the client. The daemon is waiting for a null to * proceed. We do this to make sure that we have our * socket is up prior to the daemon running the command. */ if (write(s,"",1) != 1) { perror("mcmd: Could not communicate to daemon to proceed"); close(s3); goto bad; } /* * Read from our stderr. The server should have placed * our random number we generated onto this socket. */ rv = fd_read_n(s3, &rand, sizeof(rand)); if (rv <= 0) { if (rv == 0) perror("mcmd: Connection closed by remote host"); else perror("mcmd: Bad read of verification number"); close(s3); goto bad; } randl = ntohl(rand); if (randl != randy) { char tmpbuf[LINEBUFSIZE] = {0}; char *tptr = &tmpbuf[0]; memcpy(tptr,(char *) &rand,sizeof(rand)); tptr += sizeof(rand); if ((fd_read_line (s3, tptr, LINEBUFSIZE - sizeof(rand))) < 0) { perror("mcmd: Read error from remote host"); close(s3); goto bad; } /* Legacy rsh may consider the first byte an error code, * so don't output this byte. */ if (tmpbuf[0] == '\01') tptr = &tmpbuf[1]; else tptr = &tmpbuf[0]; fprintf(stderr,"mcmd error returned: %s\n", tptr); close(s3); goto bad; } /* * Set the stderr file descriptor for the user... */ *fd2p = s3; } if ((rv = read(s, &c, 1)) < 0) { perror("mcmd: read: protocol failure"); goto bad2; } if (rv != 1) { fprintf(stderr, "mcmd: read: protocol failure: invalid response.\n"); goto bad2; } if (c != '\0') { /* retrieve error string from remote server */ char tmpbuf[LINEBUFSIZE]; if (fd_read_line (s, &tmpbuf[0], LINEBUFSIZE ) < 0) { perror("mcmd: Error from remote host"); goto bad2; } fprintf(stderr,"mcmd error returned: %s\n",&tmpbuf[0]); goto bad2; } RESTORE_PTHREAD(); return (s); bad2: if (lport) close(*fd2p); bad: close(s); RESTORE_PTHREAD(); exit(1); }
/* * Derived from the mcmd() libc call, with modified interface. * This version is MT-safe. Errors are displayed in pdsh-compat format. * Connection can time out. * ahost (IN) target hostname * addr (IN) 4 byte internet address * locuser (IN) local username * remuser (IN) remote username * cmd (IN) remote command to execute under shell * rank (IN) not used * fd2p (IN) if non NULL, return stderr file descriptor here * int (RETURN) -1 on error, socket for I/O on success * * Originally by Mike Haskell for mrsh, modified slightly to work with pdsh by: * - making mcmd always thread safe * - using "err" function output errors. * - passing in address as addr intead of calling gethostbyname * - using default mshell port instead of calling getservbyname * */ static int mcmd(char *ahost, char *addr, char *locuser, char *remuser, char *cmd, int rank, int *fd2p, void **argp) { struct sockaddr m_socket; struct sockaddr_in *getp; struct sockaddr_in sin, from; struct sockaddr_storage ss; struct in_addr m_in; unsigned int rand, randl; unsigned char *hptr; int s, s2, rv, mcount, lport; char c; char num[6] = {0}; char *mptr; char *mbuf; char *tmbuf; char *m; char *mpvers; char num_seq[12] = {0}; socklen_t len; sigset_t blockme; sigset_t oldset; char haddrdot[MAXHOSTNAMELEN + MRSH_LOCALHOST_KEYLEN + 1] = {0}; munge_ctx_t ctx; struct xpollfd xpfds[2]; memset (xpfds, 0, sizeof (xpfds)); memset (&sin, 0, sizeof (sin)); sigemptyset(&blockme); sigaddset(&blockme, SIGURG); sigaddset(&blockme, SIGPIPE); SET_PTHREAD(); /* Convert randy to decimal string, 0 if we dont' want stderr */ if (fd2p != NULL) snprintf(num_seq, sizeof(num_seq),"%d",randy); else snprintf(num_seq, sizeof(num_seq),"%d",0); /* * Start setup of the stdin/stdout socket... */ lport = 0; len = sizeof(struct sockaddr_in); if ((s = socket(AF_INET, SOCK_STREAM, 0)) < 0) { err("%p: %S: mcmd: socket call stdout failed: %m\n", ahost); EXIT_PTHREAD(); } memset (&ss, '\0', sizeof(ss)); ss.ss_family = AF_INET; if (bind(s, (struct sockaddr *)&ss, len) < 0) { err("%p: %S: mcmd: bind failed: %m\n", ahost); goto bad; } sin.sin_family = AF_INET; memcpy(&sin.sin_addr.s_addr, addr, IP_ADDR_LEN); sin.sin_port = htons(MRSH_PORT); if (connect(s, (struct sockaddr *)&sin, sizeof(sin)) < 0) { err("%p: %S: mcmd: connect failed: %m\n", ahost); goto bad; } lport = 0; s2 = -1; if (fd2p != NULL) { /* * Start the socket setup for the stderr. */ struct sockaddr_in sin2; if ((s2 = socket(AF_INET, SOCK_STREAM, 0)) < 0) { err("%p: %S: mcmd: socket call for stderr failed: %m\n", ahost); goto bad; } memset (&sin2, 0, sizeof(sin2)); sin2.sin_family = AF_INET; sin2.sin_addr.s_addr = htonl(INADDR_ANY); sin2.sin_port = 0; if (bind(s2,(struct sockaddr *)&sin2, sizeof(sin2)) < 0) { err("%p: %S: mcmd: bind failed: %m\n", ahost); close(s2); goto bad; } len = sizeof(struct sockaddr); /* * Retrieve our port number so we can hand it to the server * for the return (stderr) connection... */ /* getsockname is thread safe */ if (getsockname(s2,&m_socket,&len) < 0) { err("%p: %S: mcmd: getsockname failed: %m\n", ahost); close(s2); goto bad; } getp = (struct sockaddr_in *)&m_socket; lport = ntohs(getp->sin_port); if (listen(s2, 5) < 0) { err("%p: %S: mcmd: listen() failed: %m\n", ahost); close(s2); goto bad; } } /* put port in buffer. will be 0 if user didn't want stderr */ snprintf(num,sizeof(num),"%d",lport); /* * Use special keyed string if target is localhost, otherwise, * encode the IP addr string. */ if (!encode_localhost_string (ahost, haddrdot, sizeof (haddrdot))) { /* inet_ntoa is not thread safe, so we use the following, * which is more or less ripped from glibc */ memcpy(&m_in.s_addr, addr, IP_ADDR_LEN); hptr = (unsigned char *)&m_in; sprintf(haddrdot, "%u.%u.%u.%u", hptr[0], hptr[1], hptr[2], hptr[3]); } /* * We call munge_encode which will take what we write in and return a * pointer to an munged buffer. What we get back is a null terminated * string of encrypted characters. * * The format of the unmunged buffer is as follows (each a string terminated * with a '\0' (null): * * stderr_port_number & /dev/urandom_client_produce_number are 0 * if user did not request stderr socket * SIZE EXAMPLE * ========== ============= * remote_user_name variable "mhaskell" * '\0' * protocol version < 12 bytes "1.2" * '\0' * dotted_decimal_address_of_this_server 7-15 bytes "134.9.11.155" * '\0' * stderr_port_number 4-8 bytes "50111" * '\0' * /dev/urandom_client_produced_number 1-8 bytes "1f79ca0e" * '\0' * users_command variable "ls -al" * '\0' '\0' * * (The last extra null is accounted for in the following line's * last strlen() call.) * */ mpvers = MRSH_PROTOCOL_VERSION; mcount = ((strlen(remuser)+1) + (strlen(mpvers)+1) + (strlen(haddrdot)+1) + (strlen(num)+1) + (strlen(num_seq)+1) + strlen(cmd)+2); tmbuf = mbuf = malloc(mcount); if (tmbuf == NULL) { err("%p: %S: mcmd: Error from malloc\n", ahost); close(s2); goto bad; } /* * The following memset() call takes the extra trailing null as * part of its count as well. */ memset(mbuf,0,mcount); mptr = strcpy(mbuf, remuser); mptr += strlen(remuser)+1; mptr = strcpy(mptr, mpvers); mptr += strlen(mpvers)+1; mptr = strcpy(mptr, haddrdot); mptr += strlen(haddrdot)+1; mptr = strcpy(mptr, num); mptr += strlen(num)+1; mptr = strcpy(mptr, num_seq); mptr += strlen(num_seq)+1; mptr = strcpy(mptr, cmd); ctx = munge_ctx_create(); if ((rv = munge_encode(&m,ctx,mbuf,mcount)) != EMUNGE_SUCCESS) { err("%p: %S: mcmd: munge_encode: %s\n", ahost, munge_ctx_strerror(ctx)); munge_ctx_destroy(ctx); if (s2 >= 0) close(s2); free(tmbuf); goto bad; } munge_ctx_destroy(ctx); mcount = (strlen(m)+1); /* * Write stderr port in the clear in case we can't decode for * some reason (i.e. bad credentials). May be 0 if user * doesn't want stderr */ if (fd2p != NULL) { rv = fd_write_n(s, num, strlen(num)+1); if (rv != (strlen(num) + 1)) { free(m); free(tmbuf); if (errno == EPIPE) err("%p: %S: mcmd: Lost connection (EPIPE): %m", ahost); else err("%p: %S: mcmd: Write of stderr port failed: %m\n", ahost); close(s2); goto bad; } } else { write(s, "", 1); lport = 0; } /* * Write the munge_encoded blob to the socket. */ rv = fd_write_n(s, m, mcount); if (rv != mcount) { free(m); free(tmbuf); if (errno == EPIPE) err("%p: %S: mcmd: Lost connection: %m\n", ahost); else err("%p: %S: mcmd: Write to socket failed: %m\n", ahost); close(s2); goto bad; } free(m); free(tmbuf); if (fd2p != NULL) { /* * Wait for stderr connection from daemon. */ int s3; errno = 0; xpfds[0].fd = s; xpfds[1].fd = s2; xpfds[0].events = xpfds[1].events = XPOLLREAD; if ( ((rv = xpoll(xpfds, 2, -1)) < 0) || rv != 1 || (xpfds[0].revents > 0)) { if (errno != 0) err("%p: %S: mcmd: xpoll (setting up stderr): %m\n", ahost); else err("%p: %S: mcmd: xpoll: protocol failure in circuit setup\n", ahost); (void) close(s2); goto bad; } errno = 0; len = sizeof(from); /* arg to accept */ if ((s3 = accept(s2, (struct sockaddr *)&from, &len)) < 0) { err("%p: %S: mcmd: accept (stderr) failed: %m\n", ahost); close(s2); goto bad; } if (from.sin_family != AF_INET) { err("%p: %S: mcmd: bad family type: %d\n", ahost, from.sin_family); goto bad2; } close(s2); /* * The following fixes a race condition between the daemon * and the client. The daemon is waiting for a null to * proceed. We do this to make sure that we have our * socket is up prior to the daemon running the command. */ if (write(s,"",1) < 0) { err("%p: %S: mcmd: Could not communicate to daemon to proceed: %m\n", ahost); close(s3); goto bad; } /* * Read from our stderr. The server should have placed our * random number we generated onto this socket. */ rv = fd_read_n(s3, &rand, sizeof(rand)); if (rv != (ssize_t) (sizeof(rand))) { err("%p: %S: mcmd: Bad read of expected verification " "number off of stderr socket: %m\n", ahost); close(s3); goto bad; } randl = ntohl(rand); if (randl != randy) { char tmpbuf[LINEBUFSIZE] = {0}; char *tptr = &tmpbuf[0]; memcpy(tptr,(char *) &rand,sizeof(rand)); tptr += sizeof(rand); if (fd_read_line (s3, tptr, LINEBUFSIZE) < 0) err("%p: %S: mcmd: Read error from remote host: %m\n", ahost); else err("%p: %S: mcmd: Error: %s\n", ahost, &tmpbuf[0]); close(s3); goto bad; } /* * Set the stderr file descriptor for the user... */ *fd2p = s3; } if ((rv = read(s, &c, 1)) < 0) { err("%p: %S: mcmd: read: protocol failure: %m\n", ahost); goto bad2; } if (rv != 1) { err("%p: %S: mcmd: read: protocol failure: invalid response\n", ahost); goto bad2; } if (c != '\0') { /* retrieve error string from remote server */ char tmpbuf[LINEBUFSIZE]; if (fd_read_line (s, &tmpbuf[0], LINEBUFSIZE) < 0) err("%p: %S: mcmd: Error from remote host\n", ahost); else err("%p: %S: mcmd: Error: %s\n", ahost, tmpbuf); goto bad2; } RESTORE_PTHREAD(); return (s); bad2: if (lport) close(*fd2p); bad: close(s); EXIT_PTHREAD(); }
void * cerebrod_event_server(void *arg) { int server_fd; _event_server_initialize(); if ((server_fd = _event_server_setup_socket(0)) < 0) CEREBROD_EXIT(("event server fd setup failed")); for (;;) { ListIterator eitr; struct cerebrod_event_connection_data *ecd; struct pollfd *pfds; int pfdslen = 0; int i; /* Note that the list_count won't grow larger after the first * mutex block, b/c the cerebrod_event_queue_monitor thread can * never add to the event_connections. It can only shrink it. */ Pthread_mutex_lock(&event_connections_lock); if (event_connections) pfdslen = List_count(event_connections); Pthread_mutex_unlock(&event_connections_lock); /* The + 1 is b/c of the server_fd. */ pfdslen++; pfds = Malloc(sizeof(struct pollfd) * pfdslen); memset(pfds, '\0', sizeof(struct pollfd) * pfdslen); pfds[0].fd = server_fd; pfds[0].events = POLLIN; pfds[0].revents = 0; /* No 'event_connections' if there are no events */ if (event_connections) { i = 1; Pthread_mutex_lock(&event_connections_lock); eitr = List_iterator_create(event_connections); while ((ecd = list_next(eitr))) { pfds[i].fd = ecd->fd; pfds[i].events = POLLIN; pfds[i].revents = 0; i++; } List_iterator_destroy(eitr); Pthread_mutex_unlock(&event_connections_lock); } Poll(pfds, pfdslen, -1); /* Deal with the server fd first */ if (pfds[0].revents & POLLERR) CEREBROD_DBG(("server_fd POLLERR")); else if (pfds[0].revents & POLLIN) { unsigned int client_addr_len; int fd; struct sockaddr_in client_addr; client_addr_len = sizeof(struct sockaddr_in); if ((fd = accept(server_fd, (struct sockaddr *)&client_addr, &client_addr_len)) < 0) server_fd = cerebrod_reinit_socket(server_fd, 0, _event_server_setup_socket, "event_server: accept"); if (fd >= 0) _event_server_service_connection(fd); } /* Deal with the connecting fds */ for (i = 1; i < pfdslen; i++) { if (pfds[i].revents & POLLERR) { CEREBROD_DBG(("fd = %d POLLERR", pfds[i].fd)); Pthread_mutex_lock(&event_connections_lock); _delete_event_connection_fd(pfds[i].fd); Pthread_mutex_unlock(&event_connections_lock); continue; } if (pfds[i].revents & POLLIN) { char buf[CEREBRO_MAX_PACKET_LEN]; int n; /* We should not expect any actual data. If * we get some, just eat it and move on. * * The common situation is that the client * closes the connection. So we need to delete * our fd. */ n = fd_read_n(pfds[i].fd, buf, CEREBRO_MAX_PACKET_LEN); if (n < 0) CEREBROD_DBG(("fd_read_n = %s", strerror(errno))); if (n <= 0) { if (conf.debug && conf.event_server_debug) { Pthread_mutex_lock(&debug_output_mutex); fprintf(stderr, "**************************************\n"); fprintf(stderr, "* Event Server Close Fd: %d\n", pfds[i].fd); fprintf(stderr, "**************************************\n"); Pthread_mutex_unlock(&debug_output_mutex); } Pthread_mutex_lock(&event_connections_lock); _delete_event_connection_fd(pfds[i].fd); Pthread_mutex_unlock(&event_connections_lock); } } } Free(pfds); } return NULL; /* NOT REACHED */ }