freebsd-dev/usr.sbin/sysinstall/msg.c
Rebecca Cran 3affea8f99 To restart, sysinstall calls execl. Since it will create a new process, we
can't check to see if sysinstall is running as init just by checking if the
PID is 0. Introduce a new option that sets the RunningAsInit flag, and update
the code to check RunningAsInit intstead of getpid().

PR:	bin/38854
Submitted by:	Peter Sedeffow <peter at trumanbrewery.com>
Approved by:	rrs (mentor)
MFC after:	1 month
2010-08-17 09:39:06 +00:00

429 lines
9.7 KiB
C

/*
* The new sysinstall program.
*
* This is probably the last program in the `sysinstall' line - the next
* generation being essentially a complete rewrite.
*
* $FreeBSD$
*
* Copyright (c) 1995
* Jordan Hubbard. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer,
* verbatim and that no modifications are made prior to this
* point in the file.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY JORDAN HUBBARD ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL JORDAN HUBBARD OR HIS PETS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, LIFE OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*
*/
#include <sys/ioctl.h>
#include <sys/consio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h>
#include <stdarg.h>
#include <syslog.h>
#include "sysinstall.h"
Boolean
isDebug(void)
{
char *cp;
return (cp = variable_get(VAR_DEBUG)) && strcmp(cp, "no");
}
static Boolean
isNetworkUp(void)
{
if (!(RunningAsInit) ||
(variable_check("NETWORK_CONFIGURED=NO")) != TRUE) {
return TRUE;
}
return FALSE;
}
void
msgSyslog(const char *errstr)
{
struct sockaddr_in server;
struct hostent *hp;
char *host, *line;
int sock;
if (!isNetworkUp())
return;
if (!(host = variable_get(VAR_SYSLOG_SERVER)))
return;
if (!(hp = gethostbyname2(host, AF_INET)))
return;
if (!(sock = socket(AF_INET, SOCK_DGRAM, 0)))
return;
bzero(&server, sizeof(struct sockaddr_in));
server.sin_family = AF_INET;
server.sin_port = htons(514);
bcopy((char *)hp->h_addr, (char *)&server.sin_addr, hp->h_length);
asprintf(&line, "<%d>%s", LOG_NOTICE, errstr);
sendto(sock, line, strlen(line), 0, (struct sockaddr *)&server,
sizeof(struct sockaddr_in));
close(sock);
free(line);
}
/* Whack up an informational message on the status line, in stand-out */
void
msgYap(char *fmt, ...)
{
va_list args;
char *errstr;
int attrs;
errstr = (char *)alloca(FILENAME_MAX);
va_start(args, fmt);
vsnprintf(errstr, FILENAME_MAX, fmt, args);
va_end(args);
attrs = getattrs(stdscr);
attrset(A_REVERSE);
mvaddstr(StatusLine, 0, errstr);
attrset(attrs);
refresh();
}
/* Whack up an informational message on the status line */
void
msgInfo(char *fmt, ...)
{
va_list args;
char *errstr;
int i, attrs;
char line[81];
attrs = getattrs(stdscr);
/* NULL is a special convention meaning "erase the old stuff" */
if (!fmt) {
move(StatusLine, 0);
clrtoeol();
return;
}
errstr = (char *)alloca(FILENAME_MAX);
va_start(args, fmt);
vsnprintf(errstr, FILENAME_MAX, fmt, args);
va_end(args);
memset(line, ' ', 80);
for (i = 0; i < 80; i++) {
if (errstr[i])
line[i] = errstr[i];
else
break;
}
line[80] = '\0';
attrset(ATTR_TITLE);
mvaddstr(StatusLine, 0, line);
attrset(attrs);
move(StatusLine, 79);
refresh();
msgSyslog(errstr);
}
/* Whack up a warning on the status line */
void
msgWarn(char *fmt, ...)
{
va_list args;
char *errstr;
int attrs;
errstr = (char *)alloca(FILENAME_MAX);
strcpy(errstr, "Warning: ");
va_start(args, fmt);
vsnprintf((char *)(errstr + strlen(errstr)), FILENAME_MAX, fmt, args);
va_end(args);
attrs = getattrs(stdscr);
beep();
attrset(ATTR_TITLE);
mvaddstr(StatusLine, 0, errstr);
attrset(attrs);
refresh();
/* we don't want this hitting syslog twice */
if (isDebug()) {
if (OnVTY)
msgDebug("Warning message `%s'\n", errstr);
else
msgSyslog(errstr);
}
}
/* Whack up an error on the status line */
void
msgError(char *fmt, ...)
{
va_list args;
char *errstr;
int attrs;
errstr = (char *)alloca(FILENAME_MAX);
strcpy(errstr, "Error: ");
va_start(args, fmt);
vsnprintf((char *)(errstr + strlen(errstr)), FILENAME_MAX, fmt, args);
va_end(args);
beep();
attrs = getattrs(stdscr);
attrset(ATTR_TITLE);
mvaddstr(StatusLine, 0, errstr);
attrset(attrs);
refresh();
/* we don't want this hitting syslog twice */
if (isDebug()) {
if (OnVTY)
msgDebug("Error message `%s'\n", errstr);
else
msgSyslog(errstr);
}
}
/* Whack up a fatal error on the status line */
void
msgFatal(char *fmt, ...)
{
va_list args;
char *errstr;
int attrs;
errstr = (char *)alloca(FILENAME_MAX);
strcpy(errstr, "Fatal Error: ");
va_start(args, fmt);
vsnprintf((char *)(errstr + strlen(errstr)), FILENAME_MAX, fmt, args);
va_end(args);
beep();
attrs = getattrs(stdscr);
attrset(ATTR_TITLE);
mvaddstr(StatusLine, 0, errstr);
addstr(" - ");
addstr("PRESS ANY KEY TO ");
if (RunningAsInit)
addstr("REBOOT");
else
addstr("QUIT");
attrset(attrs);
refresh();
if (OnVTY)
msgDebug("Fatal error `%s'!\n", errstr);
else
msgSyslog(errstr);
getch();
systemShutdown(1);
}
/* Put up a message in a popup confirmation box */
void
msgConfirm(char *fmt, ...)
{
va_list args;
char *errstr;
WINDOW *w = savescr();
errstr = (char *)alloca(FILENAME_MAX);
va_start(args, fmt);
vsnprintf(errstr, FILENAME_MAX, fmt, args);
va_end(args);
use_helpline(NULL);
use_helpfile(NULL);
if (OnVTY) {
ioctl(0, VT_ACTIVATE, 1);
msgInfo(NULL);
}
dialog_notify(errstr);
restorescr(w);
}
/* Put up a message in a popup information box */
void
msgNotify(char *fmt, ...)
{
va_list args;
char *errstr;
errstr = (char *)alloca(FILENAME_MAX);
va_start(args, fmt);
vsnprintf(errstr, FILENAME_MAX, fmt, args);
va_end(args);
use_helpline(NULL);
use_helpfile(NULL);
if (isDebug())
msgDebug("Notify: %s\n", errstr);
dialog_msgbox(NULL, errstr, -1, -1, 0);
}
/* Put up a message in a popup yes/no box and return 0 for YES, 1 for NO */
int
msgYesNo(char *fmt, ...)
{
va_list args;
char *errstr;
int ret;
WINDOW *w = savescr();
errstr = (char *)alloca(FILENAME_MAX);
va_start(args, fmt);
vsnprintf(errstr, FILENAME_MAX, fmt, args);
va_end(args);
use_helpline(NULL);
use_helpfile(NULL);
if (OnVTY) {
ioctl(0, VT_ACTIVATE, 1); /* Switch back */
msgInfo(NULL);
}
if (variable_get(VAR_NONINTERACTIVE))
return 0; /* If non-interactive, return YES all the time */
ret = dialog_yesno("User Confirmation Requested", errstr, -1, -1);
restorescr(w);
return ret;
}
/* Put up a message in a popup no/yes box and return 0 for YES, 1 for NO */
int
msgNoYes(char *fmt, ...)
{
va_list args;
char *errstr;
int ret;
WINDOW *w = savescr();
errstr = (char *)alloca(FILENAME_MAX);
va_start(args, fmt);
vsnprintf(errstr, FILENAME_MAX, fmt, args);
va_end(args);
use_helpline(NULL);
use_helpfile(NULL);
if (OnVTY) {
ioctl(0, VT_ACTIVATE, 1); /* Switch back */
msgInfo(NULL);
}
if (variable_get(VAR_NONINTERACTIVE))
return 1; /* If non-interactive, return NO all the time */
ret = dialog_noyes("User Confirmation Requested", errstr, -1, -1);
restorescr(w);
return ret;
}
/* Put up a message in an input box and return the value */
char *
msgGetInput(char *buf, char *fmt, ...)
{
va_list args;
char *errstr;
static char input_buffer[256];
int rval;
WINDOW *w = savescr();
errstr = (char *)alloca(FILENAME_MAX);
va_start(args, fmt);
vsnprintf(errstr, FILENAME_MAX, fmt, args);
va_end(args);
use_helpline(NULL);
use_helpfile(NULL);
if (buf)
SAFE_STRCPY(input_buffer, buf);
else
input_buffer[0] = '\0';
if (OnVTY) {
ioctl(0, VT_ACTIVATE, 1); /* Switch back */
msgInfo(NULL);
}
rval = dialog_inputbox("Value Required", errstr, -1, -1, input_buffer);
restorescr(w);
if (!rval)
return input_buffer;
else
return NULL;
}
/* Write something to the debugging port */
void
msgDebug(char *fmt, ...)
{
va_list args;
char *dbg;
if (DebugFD == -1)
return;
dbg = (char *)alloca(FILENAME_MAX);
strcpy(dbg, "DEBUG: ");
va_start(args, fmt);
vsnprintf((char *)(dbg + strlen(dbg)), FILENAME_MAX, fmt, args);
va_end(args);
msgSyslog(dbg);
write(DebugFD, dbg, strlen(dbg));
}
/* Tell the user there's some output to go look at */
void
msgWeHaveOutput(char *fmt, ...)
{
va_list args;
char *errstr;
WINDOW *w = savescr();
errstr = (char *)alloca(FILENAME_MAX);
va_start(args, fmt);
vsnprintf(errstr, FILENAME_MAX, fmt, args);
va_end(args);
use_helpline(NULL);
use_helpfile(NULL);
msgDebug("Notify: %s\n", errstr);
dialog_clear_norefresh();
sleep(2);
dialog_msgbox(NULL, errstr, -1, -1, 0);
restorescr(w);
}
/* Simple versions of msgConfirm() and msgNotify() for calling from scripts */
int
msgSimpleConfirm(char *str)
{
msgConfirm("%s", str);
return DITEM_SUCCESS;
}
int
msgSimpleNotify(char *str)
{
msgNotify("%s", str);
return DITEM_SUCCESS;
}