/*
* read the config file
*/
int
qcmd_config(void)
{
FILE *fp;
int rval;
if (if_namelist != NULL)
if_freenameindex(if_namelist);
if_namelist = if_nameindex();
curifname[0] = '\0';
LOG(LOG_INFO, 0, "ALTQ config file is %s", altqconfigfile);
fp = fopen(altqconfigfile, "r");
if (fp == NULL) {
LOG(LOG_ERR, errno, "can't open %s", altqconfigfile, 0);
return (QOPERR_INVAL);
}
line_no = 0;
rval = 1;
while (rval)
rval = do_command(fp);
if (!feof(fp)) {
LOG(LOG_ERR, 0, "Error in %s, line %d. config failed.",
altqconfigfile, line_no);
(void) qcmd_destroyall();
rval = QOPERR_INVAL;
} else
rval = 0;
(void)fclose(fp);
line_no = 0;
return (rval);
}
static int
ctl_parser(char *cmdbuf)
{
char w[MAX_WORD], *cp = cmdbuf;
char *ifname;
int state;
int rval;
if (!get_ifname(&cp, &ifname)) {
printf("missing interface name in %s, line %d",
altqconfigfile, line_no);
return (0);
}
if (!next_word(&cp, w)) {
state = is_q_enabled(ifname);
printf("altq %s on %s\n",
state ? "enabled" : "disabled", ifname);
return (1);
}
if (EQUAL(w, "enable")) {
rval = qcmd_enable(ifname);
printf("altq %s on %s\n",
(rval == 0) ? "enabled" : "enable failed!", ifname);
} else if (EQUAL(w, "disable")) {
rval = qcmd_disable(ifname);
printf("altq %s on %s\n",
(rval == 0) ? "disabled" : "disable failed!", ifname);
} else if (EQUAL(w, "reload")) {
printf("reinitializing altq...\n");
qcmd_destroyall();
qcmd_init();
} else
return (0);
return (1);
}
int
main(int argc, char **argv)
{
int i, c, maxfd, rval, qpsock, fd;
fd_set fds, rfds;
FILE *fp, *client[MAX_CLIENT];
m_debug = 0;
l_debug = LOG_INFO;
fp = NULL;
for (i = 0; i < MAX_CLIENT; i++)
client[i] = NULL;
while ((c = getopt(argc, argv, "f:vdl:")) != -1) {
switch (c) {
case 'f':
altqconfigfile = optarg;
break;
case 'v':
l_debug = LOG_DEBUG;
m_debug |= DEBUG_ALTQ;
daemonize = 0;
break;
case 'd':
daemonize = 0;
break;
case 'l':
l_debug = atoi(optarg);
break;
default:
usage();
}
}
signal(SIGINT, sig_handler);
signal(SIGTERM, sig_handler);
signal(SIGHUP, sig_handler);
signal(SIGPIPE, sig_handler);
if (daemonize)
openlog("altqd", LOG_PID, LOG_DAEMON);
if (qcmd_init() != 0) {
if (daemonize)
closelog();
exit(1);
}
/*
* open a unix domain socket for altqd clients
*/
if ((qpsock = socket(AF_LOCAL, SOCK_STREAM, 0)) < 0)
LOG(LOG_ERR, errno, "can't open unix domain socket");
else {
struct sockaddr_un addr;
bzero(&addr, sizeof(addr));
addr.sun_family = AF_LOCAL;
strlcpy(addr.sun_path, QUIP_PATH, sizeof(addr.sun_path));
unlink(QUIP_PATH);
if (bind(qpsock, (struct sockaddr *)&addr,
sizeof(addr)) < 0) {
LOG(LOG_ERR, errno, "can't bind to %s", QUIP_PATH);
close(qpsock);
qpsock = -1;
}
chmod(QUIP_PATH, 0666);
if (listen(qpsock, SOMAXCONN) < 0) {
LOG(LOG_ERR, errno, "can't listen to %s", QUIP_PATH);
close(qpsock);
qpsock = -1;
}
}
if (daemonize) {
daemon(0, 0);
/* save pid to the pid file (/var/tmp/altqd.pid) */
#ifdef __FreeBSD__
{
FILE *fp;
if ((fp = fopen(ALTQD_PID_FILE, "w")) != NULL) {
fprintf(fp, "%d\n", getpid());
fclose(fp);
} else
LOG(LOG_WARNING, errno, "can't open pid file");
}
#else
pidfile(NULL);
#endif
} else {
/* interactive mode */
fp = stdin;
printf("\nEnter ? or command:\n");
printf("altqd %s> ", cur_ifname());
fflush(stdout);
}
/*
* go into the command mode.
*/
FD_ZERO(&fds);
maxfd = 0;
if (fp != NULL) {
fd = fileno(fp);
if (fd == -1)
LOG(LOG_ERR, 0, "bad file descriptor", QUIP_PATH);
} else
fd = -1;
if (fd != -1) {
FD_SET(fd, &fds);
maxfd = MAX(maxfd, fd + 1);
}
if (qpsock >= 0) {
FD_SET(qpsock, &fds);
maxfd = MAX(maxfd, qpsock + 1);
}
rval = 1;
while (rval) {
if (gotsig_hup) {
qcmd_destroyall();
gotsig_hup = 0;
LOG(LOG_INFO, 0, "reinitializing altqd...");
if (qcmd_init() != 0) {
LOG(LOG_INFO, 0, "reinitialization failed");
break;
}
}
if (gotsig_term || gotsig_int) {
LOG(LOG_INFO, 0, "Exiting on signal %d",
gotsig_term ? SIGTERM : SIGINT);
break;
}
FD_COPY(&fds, &rfds);
if (select(maxfd, &rfds, NULL, NULL, NULL) < 0) {
if (errno != EINTR)
err(1, "select");
continue;
}
/*
* if there is command input, read the input line,
* parse it, and execute.
*/
if (fp && FD_ISSET(fd, &rfds)) {
rval = do_command(fp);
if (rval == 0) {
/* quit command or eof on input */
LOG(LOG_INFO, 0, "Exiting.");
} else if (fp == stdin)
printf("altqd %s> ", cur_ifname());
fflush(stdout);
} else if (qpsock >= 0 && FD_ISSET(qpsock, &rfds)) {
/*
* quip connection request from client via unix
* domain socket; get a new socket for this
* connection and add it to the select list.
*/
int newsock = accept(qpsock, NULL, NULL);
if (newsock == -1) {
LOG(LOG_ERR, errno, "accept");
continue;
}
FD_SET(newsock, &fds);
for (i = 0; i < MAX_CLIENT; i++)
if (client[i] == NULL) {
client[i] = fdopen(newsock, "r+");
break;
}
maxfd = MAX(maxfd, newsock + 1);
} else {
/*
* check input from a client via unix domain socket
*/
for (i = 0; i < MAX_CLIENT; i++) {
int fd1;
if (client[i] == NULL)
continue;
fd1 = fileno(client[i]);
if (FD_ISSET(fd1, &rfds)) {
if (quip_input(client[i]) != 0 ||
fflush(client[i]) != 0) {
/* connection closed */
fclose(client[i]);
client[i] = NULL;
FD_CLR(fd1, &fds);
}
}
}
}
}
/* cleanup and exit */
qcmd_destroyall();
if (qpsock >= 0)
(void)close(qpsock);
unlink(QUIP_PATH);
for (i = 0; i < MAX_CLIENT; i++)
if (client[i] != NULL)
(void)fclose(client[i]);
if (daemonize) {
#ifdef __FreeBSD__
/* if we have a pid file, remove it */
unlink(ALTQD_PID_FILE);
#endif
closelog();
}
exit(0);
}