/*
* shutdown_mon.c - Part of AFD, an automatic file distribution program.
* Copyright (c) 1998 - 2021 Holger Kiehl <Holger.Kiehl@dwd.de>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/
#include "afddefs.h"
DESCR__S_M3
/*
** NAME
** shutdown_mon - does a shutdown of the AFD_MON
**
** SYNOPSIS
** void shutdown_mon(int silent_shutdown, char *user)
**
** DESCRIPTION
**
** RETURN VALUES
** None.
**
** AUTHOR
** H.Kiehl
**
** HISTORY
** 30.08.1998 H.Kiehl Created
**
*/
DESCR__E_M3
#include <stdio.h> /* fprintf() */
#include <string.h>
#include <stdlib.h> /* exit() */
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/time.h> /* struct timeval */
#ifdef HAVE_FCNTL_H
#include <fcntl.h> /* O_RDWR, O_CREAT, O_WRONLY, etc */
#endif
#include <unistd.h> /* select(), unlink() */
#include <errno.h>
#include "mondefs.h"
#include "version.h"
/* External global variables. */
extern char *p_work_dir;
/*########################### shutdown_mon() ############################*/
void
#ifdef WITH_SYSTEMD
shutdown_mon(int silent_shutdown, char *user, int stop_all)
#else
shutdown_mon(int silent_shutdown, char *user)
#endif
{
int mon_cmd_fd,
mon_resp_fd,
#ifdef WITHOUT_FIFO_RW_SUPPORT
mon_cmd_writefd,
mon_resp_writefd,
#endif
status;
fd_set rset;
char buffer[DEFAULT_BUFFER_SIZE],
mon_cmd_fifo[MAX_PATH_LENGTH],
mon_resp_fifo[MAX_PATH_LENGTH];
struct timeval timeout;
/* Initialise variables. */
(void)strcpy(mon_cmd_fifo, p_work_dir);
(void)strcat(mon_cmd_fifo, FIFO_DIR);
(void)strcpy(mon_resp_fifo, mon_cmd_fifo);
(void)strcat(mon_resp_fifo, MON_RESP_FIFO);
(void)strcat(mon_cmd_fifo, MON_CMD_FIFO);
#ifdef WITHOUT_FIFO_RW_SUPPORT
if (open_fifo_rw(mon_cmd_fifo, &mon_cmd_fd, &mon_cmd_writefd) == -1)
#else
if ((mon_cmd_fd = open(mon_cmd_fifo, O_RDWR)) == -1)
#endif
{
(void)fprintf(stderr, "ERROR : Could not open fifo %s : %s (%s %d)\n",
mon_cmd_fifo, strerror(errno), __FILE__, __LINE__);
exit(INCORRECT);
}
#ifdef WITHOUT_FIFO_RW_SUPPORT
if (open_fifo_rw(mon_resp_fifo, &mon_resp_fd, &mon_resp_writefd) == -1)
#else
if ((mon_resp_fd = open(mon_resp_fifo, O_RDWR)) == -1)
#endif
{
(void)fprintf(stderr, "ERROR : Could not open fifo %s : %s (%s %d)\n",
mon_resp_fifo, strerror(errno), __FILE__, __LINE__);
exit(INCORRECT);
}
/* Tell user what we are doing. */
system_log(CONFIG_SIGN, NULL, 0,
"Starting %s shutdown (%s) ...", AFD_MON, user);
/* Send SHUTDOWN command. */
#ifdef WITHOUT_FIFO_RW_SUPPORT
# ifdef WITH_SYSTEMD
if (send_cmd((stop_all == YES) ? SHUTDOWN_ALL : SHUTDOWN, mon_cmd_writefd) < 0)
# else
if (send_cmd(SHUTDOWN, mon_cmd_writefd) < 0)
# endif
#else
# ifdef WITH_SYSTEMD
if (send_cmd((stop_all == YES) ? SHUTDOWN_ALL : SHUTDOWN, mon_cmd_fd) < 0)
# else
if (send_cmd(SHUTDOWN, mon_cmd_fd) < 0)
# endif
#endif
{
(void)fprintf(stderr,
"ERROR : Failed to send stop command to %s : %s (%s %d)\n",
AFD_MON, strerror(errno), __FILE__, __LINE__);
exit(INCORRECT);
}
/*
* Now lets wait for a reply from 'afd_mon', but not more then 40s.
*/
/* Initialise descriptor set and timeout */
FD_ZERO(&rset);
FD_SET(mon_resp_fd, &rset);
timeout.tv_usec = 0L;
timeout.tv_sec = 40L;
/* Wait for message x seconds and then continue. */
status = select(mon_resp_fd + 1, &rset, NULL, NULL, &timeout);
/* Did we get a timeout? */
if (status == 0)
{
int val;
(void)fprintf(stderr, "\nAFD_MON is NOT responding!\n");
/*
* Since the AFD_MON does not answer and we have already send
* the shutdown command, it is necessary to remove this command
* from the FIFO.
*/
if ((val = fcntl(mon_cmd_fd, F_GETFL, 0)) == -1)
{
(void)fprintf(stderr,
"ERROR : Failed to get file status flag : %s (%s %d)\n",
strerror(errno), __FILE__, __LINE__);
exit(INCORRECT);
}
val |= O_NONBLOCK;
if (fcntl(mon_cmd_fd, F_SETFL, val) == -1)
{
(void)fprintf(stderr,
"ERROR : Failed to set file status flag : %s (%s %d)\n",
strerror(errno), __FILE__, __LINE__);
exit(INCORRECT);
}
if (read(mon_cmd_fd, buffer, DEFAULT_BUFFER_SIZE) == -1)
{
(void)fprintf(stderr,
"WARNING : read() error : %s (%s %d)\n",
strerror(errno), __FILE__, __LINE__);
}
/*
* Telling the user we failed to do a shutdown is not of
* much use. It would now be better if we kill all jobs
* of AFD_MON.
*/
if (check_mon(1L) == 0)
{
(void)fprintf(stderr, "Removed all AFD_MON processes and resources.\n");
/* Remove MON_ACTIVE file */
(void)strcpy(mon_cmd_fifo, p_work_dir);
(void)strcat(mon_cmd_fifo, FIFO_DIR);
(void)strcat(mon_cmd_fifo, MON_ACTIVE_FILE);
if (unlink(mon_cmd_fifo) == -1)
{
(void)fprintf(stderr, "Failed to unlink() %s : %s (%s %d)\n",
mon_cmd_fifo, strerror(errno),
__FILE__, __LINE__);
}
}
}
else if (FD_ISSET(mon_resp_fd, &rset))
{
if (read(mon_resp_fd, buffer, 1) > 0)
{
if (buffer[0] == ACKN)
{
/* Tell user we are done */
system_log(CONFIG_SIGN, NULL, 0, "Done!");
}
else
{
(void)fprintf(stderr,
"Hmm. Something is wrong here! (%s %d)\n",
__FILE__, __LINE__);
}
}
}
else if (status < 0)
{
(void)fprintf(stderr, "select() error : %s (%s %d)\n",
strerror(errno), __FILE__, __LINE__);
exit(INCORRECT);
}
else
{
(void)fprintf(stderr, "Unknown condition. (%s %d)\n",
__FILE__, __LINE__);
exit(INCORRECT);
}
return;
}