d/freebsd-skq
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
b945fe4320
freebsd-skq/usr.sbin/xntpd/xntpdc/ntpdc.c

1534 lines
32 KiB
C
Raw Blame History

/*
* xntpdc - control and monitor your xntpd daemon
*/
#include <stdio.h>
#include <ctype.h>
#include <signal.h>
#include <setjmp.h>
#include <sys/types.h>
#include <sys/time.h>
#include <netdb.h>
#include "ntpdc.h"
#include "ntp_select.h"
#include "ntp_io.h"
#include "ntp_stdlib.h"
/*
* Because we now potentially understand a lot of commands (and
* it requires a lot of commands to talk to xntpd) we will run
* interactive if connected to a terminal.
*/
static int interactive = 0; /* set to 1 when we should prompt */
static char * prompt = "xntpdc> "; /* prompt to ask him about */
/*
* Keyid used for authenticated requests. Obtained on the fly.
*/
static u_long info_auth_keyid;
/*
* Type of key md5 or des
*/
#define KEY_TYPE_DES 3
#define KEY_TYPE_MD5 4
static int info_auth_keytype = KEY_TYPE_DES; /* DES */
/*
* Built in command handler declarations
*/
static int openhost P((char *));
static int sendpkt P((char *, int));
static void growpktdata P((void));
static int getresponse P((int, int, int *, int *, char **));
static int sendrequest P((int, int, int, int, int, char *));
static void getcmds P((void));
static RETSIGTYPE abortcmd P((int));
static void docmd P((char *));
static void tokenize P((char *, char **, int *));
static int findcmd P((char *, struct xcmd *, struct xcmd *, struct xcmd **));
static int getarg P((char *, int, arg_v *));
static int getnetnum P((char *, u_long *, char *));
static void help P((struct parse *, FILE *));
#if defined(sgi) || defined(SYS_BSDI) || defined(__STDC__)
static int helpsort P((const void *, const void *));
#else
static int helpsort P((char **, char **));
#endif /* sgi */
static void printusage P((struct xcmd *, FILE *));
static void timeout P((struct parse *, FILE *));
static void delay P((struct parse *, FILE *));
static void host P((struct parse *, FILE *));
static void keyid P((struct parse *, FILE *));
static void keytype P((struct parse *, FILE *));
static void passwd P((struct parse *, FILE *));
static void hostnames P((struct parse *, FILE *));
static void setdebug P((struct parse *, FILE *));
static void quit P((struct parse *, FILE *));
static void version P((struct parse *, FILE *));
static void warning P((char *, char *, char *));
static void error P((char *, char *, char *));
static u_long getkeyid P((char *));
/*
* Built-in commands we understand
*/
static struct xcmd builtins[] = {
{ "?", help, { OPT|NTP_STR, NO, NO, NO },
{ "command", "", "", "" },
"tell the use and syntax of commands" },
{ "help", help, { OPT|NTP_STR, NO, NO, NO },
{ "command", "", "", "" },
"tell the use and syntax of commands" },
{ "timeout", timeout, { OPT|UINT, NO, NO, NO },
{ "msec", "", "", "" },
"set the primary receive time out" },
{ "delay", delay, { OPT|INT, NO, NO, NO },
{ "msec", "", "", "" },
"set the delay added to encryption time stamps" },
{ "host", host, { OPT|NTP_STR, NO, NO, NO },
{ "hostname", "", "", "" },
"specify the host whose NTP server we talk to" },
{ "passwd", passwd, { OPT|NTP_STR, NO, NO, NO },
{ "", "", "", "" },
"specify a password to use for authenticated requests"},
{ "hostnames", hostnames, { OPT|NTP_STR, NO, NO, NO },
{ "yes|no", "", "", "" },
"specify whether hostnames or net numbers are printed"},
{ "debug", setdebug, { OPT|NTP_STR, NO, NO, NO },
{ "no|more|less", "", "", "" },
"set/change debugging level" },
{ "quit", quit, { NO, NO, NO, NO },
{ "", "", "", "" },
"exit xntpdc" },
{ "keyid", keyid, { OPT|UINT, NO, NO, NO },
{ "key#", "", "", "" },
"set keyid to use for authenticated requests" },
{ "keytype", keytype, { NTP_STR, NO, NO, NO },
{ "key type (md5|des)", "", "", "" },
"set key type to use for authenticated requests (des|md5)" },
{ "version", version, { NO, NO, NO, NO },
{ "", "", "", "" },
"print version number" },
{ 0, 0, { NO, NO, NO, NO },
{ "", "", "", "" }, "" }
};
/*
* Default values we use.
*/
#define DEFTIMEOUT (5) /* 5 second time out */
#define DEFSTIMEOUT (2) /* 2 second time out after first */
#define DEFDELAY 0x51EB852 /* 20 milliseconds, l_fp fraction */
#define DEFHOST "localhost" /* default host name */
#define LENHOSTNAME 256 /* host name is 256 characters long */
#define MAXCMDS 100 /* maximum commands on cmd line */
#define MAXHOSTS 100 /* maximum hosts on cmd line */
#define MAXLINE 512 /* maximum line length */
#define MAXTOKENS (1+MAXARGS+2) /* maximum number of usable tokens */
/*
* Some variables used and manipulated locally
*/
static struct timeval tvout = { DEFTIMEOUT, 0 }; /* time out for reads */
static struct timeval tvsout = { DEFSTIMEOUT, 0 }; /* secondary time out */
static l_fp delay_time; /* delay time */
static char currenthost[LENHOSTNAME]; /* current host name */
static struct sockaddr_in hostaddr = { 0 }; /* host address */
static int showhostnames = 1; /* show host names by default */
static int sockfd; /* fd socket is openned on */
static int havehost = 0; /* set to 1 when host open */
struct servent *server_entry = NULL; /* server entry for ntp */
/*
* Holds data returned from queries. We allocate INITDATASIZE
* octets to begin with, increasing this as we need to.
*/
#define INITDATASIZE (sizeof(struct resp_pkt) * 16)
#define INCDATASIZE (sizeof(struct resp_pkt) * 8)
static char *pktdata;
static int pktdatasize;
/*
* For commands typed on the command line (with the -c option)
*/
static int numcmds = 0;
static char *ccmds[MAXCMDS];
#define ADDCMD(cp) if (numcmds < MAXCMDS) ccmds[numcmds++] = (cp)
/*
* When multiple hosts are specified.
*/
static int numhosts = 0;
static char *chosts[MAXHOSTS];
#define ADDHOST(cp) if (numhosts < MAXHOSTS) chosts[numhosts++] = (cp)
/*
* Error codes for internal use
*/
#define ERR_INCOMPLETE 16
#define ERR_TIMEOUT 17
/*
* Macro definitions we use
*/
#define ISSPACE(c) ((c) == ' ' || (c) == '\t')
#define ISEOL(c) ((c) == '\n' || (c) == '\r' || (c) == '\0')
#define STREQ(a, b) (*(a) == *(b) && strcmp((a), (b)) == 0)
/*
* For converting time stamps to dates
*/
#define JAN_1970 2208988800 /* 1970 - 1900 in seconds */
/*
* Jump buffer for longjumping back to the command level
*/
static jmp_buf interrupt_buf;
static int jump = 0;
/*
* Pointer to current output unit
*/
static FILE *current_output;
/*
* Command table imported from ntpdc_ops.c
*/
extern struct xcmd opcmds[];
char *progname;
int debug;
/*
* main - parse arguments and handle options
*/
void
main(argc, argv)
int argc;
char *argv[];
{
int c;
int errflg = 0;
extern int ntp_optind;
extern char *ntp_optarg;
delay_time.l_ui = 0;
delay_time.l_uf = DEFDELAY;
progname = argv[0];
while ((c = ntp_getopt(argc, argv, "c:dilnps")) != EOF)
switch (c) {
case 'c':
ADDCMD(ntp_optarg);
break;
case 'd':
++debug;
break;
case 'i':
interactive = 1;
break;
case 'l':
ADDCMD("listpeers");
break;
case 'n':
showhostnames = 0;
break;
case 'p':
ADDCMD("peers");
break;
case 's':
ADDCMD("dmpeers");
break;
default:
errflg++;
break;
}
if (errflg) {
(void) fprintf(stderr,
"usage: %s [-dilnps] [-c cmd] host ...\n",
progname);
exit(2);
}
if (ntp_optind == argc) {
ADDHOST(DEFHOST);
} else {
for (; ntp_optind < argc; ntp_optind++)
ADDHOST(argv[ntp_optind]);
}
if (numcmds == 0 && interactive == 0
&& isatty(fileno(stdin)) && isatty(fileno(stderr))) {
interactive = 1;
}
if (interactive)
(void) signal_no_reset(SIGINT, abortcmd);
/*
* Initialize the packet data buffer
*/
pktdata = (char *)malloc(INITDATASIZE);
if (pktdata == NULL) {
(void) fprintf(stderr, "%s: malloc() failed!\n", progname);
exit(1);
}
pktdatasize = INITDATASIZE;
if (numcmds == 0) {
(void) openhost(chosts[0]);
getcmds();
} else {
int ihost;
int icmd;
for (ihost = 0; ihost < numhosts; ihost++) {
if (openhost(chosts[ihost]))
for (icmd = 0; icmd < numcmds; icmd++) {
if (numhosts > 1)
printf ("--- %s ---\n",chosts[ihost]);
docmd(ccmds[icmd]);
}
}
}
exit(0);
}
/*
* openhost - open a socket to a host
*/
static int
openhost(hname)
char *hname;
{
u_long netnum;
char temphost[LENHOSTNAME];
if (server_entry == NULL) {
server_entry = getservbyname("ntp", "udp");
if (server_entry == NULL) {
(void) fprintf(stderr, "%s: ntp/udp: unknown service\n",
progname);
exit(1);
}
if (debug > 2)
printf("Got ntp/udp service entry\n");
}
if (!getnetnum(hname, &netnum, temphost))
return 0;
if (debug > 2)
printf("Opening host %s\n", temphost);
if (havehost == 1) {
if (debug > 2)
printf("Closing old host %s\n", currenthost);
(void) close(sockfd);
havehost = 0;
}
(void) strcpy(currenthost, temphost);
hostaddr.sin_family = AF_INET;
hostaddr.sin_port = server_entry->s_port;
hostaddr.sin_addr.s_addr = netnum;
sockfd = socket(AF_INET, SOCK_DGRAM, 0);
if (sockfd == -1)
error("socket", "", "");
#if defined(SYS_HPUX) && (SYS_HPUX < 8)
#ifdef SO_RCVBUF
{ int rbufsize = INITDATASIZE + 2048; /* 2K for slop */
if (setsockopt(sockfd, SOL_SOCKET, SO_RCVBUF,
&rbufsize, sizeof(int)) == -1)
error("setsockopt", "", "");
}
#endif
#endif
if (connect(sockfd, (struct sockaddr *)&hostaddr,
sizeof(hostaddr)) == -1)
error("connect", "", "");
havehost = 1;
return 1;
}
/* XXX ELIMINATE sendpkt similar in ntpq.c, ntpdc.c, ntp_io.c, ntptrace.c */
/*
* sendpkt - send a packet to the remote host
*/
static int
sendpkt(xdata, xdatalen)
char *xdata;
int xdatalen;
{
if (write(sockfd, xdata, xdatalen) == -1) {
warning("write to %s failed", currenthost, "");
return -1;
}
return 0;
}
/*
* growpktdata - grow the packet data area
*/
static void
growpktdata()
{
pktdatasize += INCDATASIZE;
pktdata = (char *)realloc(pktdata, pktdatasize);
if (pktdata == 0) {
(void) fprintf(stderr, "%s: realloc() failed!\n", progname);
exit(1);
}
}
/*
* getresponse - get a (series of) response packet(s) and return the data
*/
static int
getresponse(implcode, reqcode, ritems, rsize, rdata)
int implcode;
int reqcode;
int *ritems;
int *rsize;
char **rdata;
{
struct resp_pkt rpkt;
struct timeval tvo;
int items;
int size;
int datasize;
char *datap;
char haveseq[MAXSEQ+1];
int firstpkt;
int lastseq;
int numrecv;
int seq;
fd_set fds;
int n;
/*
* This is pretty tricky. We may get between 1 and many packets
* back in response to the request. We peel the data out of
* each packet and collect it in one long block. When the last
* packet in the sequence is received we'll know how many we
* should have had. Note we use one long time out, should reconsider.
*/
*ritems = 0;
*rsize = 0;
firstpkt = 1;
numrecv = 0;
*rdata = datap = pktdata;
lastseq = 999; /* too big to be a sequence number */
memset(haveseq, 0, sizeof(haveseq));
FD_ZERO(&fds);
again:
if (firstpkt)
tvo = tvout;
else
tvo = tvsout;
FD_SET(sockfd, &fds);
n = select(sockfd+1, &fds, (fd_set *)0, (fd_set *)0, &tvo);
if (n == -1) {
warning("select fails", "", "");
return -1;
}
if (n == 0) {
/*
* Timed out. Return what we have
*/
if (firstpkt) {
(void) fprintf(stderr,
"%s: timed out, nothing received\n", currenthost);
return ERR_TIMEOUT;
} else {
(void) fprintf(stderr,
"%s: timed out with incomplete data\n",
currenthost);
if (debug) {
printf("Received sequence numbers");
for (n = 0; n <= MAXSEQ; n++)
if (haveseq[n])
printf(" %d,", n);
if (lastseq != 999)
printf(" last frame received\n");
else
printf(" last frame not received\n");
}
return ERR_INCOMPLETE;
}
}
n = read(sockfd, (char *)&rpkt, sizeof(rpkt));
if (n == -1) {
warning("read", "", "");
return -1;
}
/*
* Check for format errors. Bug proofing.
*/
if (n < RESP_HEADER_SIZE) {
if (debug)
printf("Short (%d byte) packet received\n", n);
goto again;
}
if (INFO_VERSION(rpkt.rm_vn_mode) != NTP_VERSION) {
if (debug)
printf("Packet received with version %d\n",
INFO_VERSION(rpkt.rm_vn_mode));
goto again;
}
if (INFO_MODE(rpkt.rm_vn_mode) != MODE_PRIVATE) {
if (debug)
printf("Packet received with mode %d\n",
INFO_MODE(rpkt.rm_vn_mode));
goto again;
}
if (INFO_IS_AUTH(rpkt.auth_seq)) {
if (debug)
printf("Encrypted packet received\n");
goto again;
}
if (!ISRESPONSE(rpkt.rm_vn_mode)) {
if (debug)
printf("Received request packet, wanted response\n");
goto again;
}
if (INFO_MBZ(rpkt.mbz_itemsize) != 0) {
if (debug)
printf("Received packet with nonzero MBZ field!\n");
goto again;
}
/*
* Check implementation/request. Could be old data getting to us.
*/
if (rpkt.implementation != implcode || rpkt.request != reqcode) {
if (debug)
printf(
"Received implementation/request of %d/%d, wanted %d/%d",
rpkt.implementation, rpkt.request,
implcode, reqcode);
goto again;
}
/*
* Check the error code. If non-zero, return it.
*/
if (INFO_ERR(rpkt.err_nitems) != INFO_OKAY) {
if (debug && ISMORE(rpkt.rm_vn_mode)) {
printf("Error code %d received on not-final packet\n",
INFO_ERR(rpkt.err_nitems));
}
return (int)INFO_ERR(rpkt.err_nitems);
}
/*
* Collect items and size. Make sure they make sense.
*/
items = INFO_NITEMS(rpkt.err_nitems);
size = INFO_ITEMSIZE(rpkt.mbz_itemsize);
if ((datasize = items*size) > (n-RESP_HEADER_SIZE)) {
if (debug)
printf(
"Received items %d, size %d (total %d), data in packet is %d\n",
items, size, datasize, n-RESP_HEADER_SIZE);
goto again;
}
/*
* If this isn't our first packet, make sure the size matches
* the other ones.
*/
if (!firstpkt && size != *rsize) {
if (debug)
printf("Received itemsize %d, previous %d\n",
size, *rsize);
goto again;
}
/*
* If we've received this before, toss it
*/
seq = INFO_SEQ(rpkt.auth_seq);
if (haveseq[seq]) {
if (debug)
printf("Received duplicate sequence number %d\n", seq);
goto again;
}
haveseq[seq] = 1;
/*
* If this is the last in the sequence, record that.
*/
if (!ISMORE(rpkt.rm_vn_mode)) {
if (lastseq != 999) {
printf("Received second end sequence packet\n");
goto again;
}
lastseq = seq;
}
/*
* So far, so good. Copy this data into the output array.
*/
if ((datap + datasize) > (pktdata + pktdatasize)) {
int offset = datap - pktdata;
growpktdata();
*rdata = pktdata; /* might have been realloced ! */
datap = pktdata + offset;
}
memmove(datap, (char *)rpkt.data, datasize);
datap += datasize;
if (firstpkt) {
firstpkt = 0;
*rsize = size;
}
*ritems += items;
/*
* Finally, check the count of received packets. If we've got them
* all, return
*/
++numrecv;
if (numrecv <= lastseq)
goto again;
return INFO_OKAY;
}
/*
* sendrequest - format and send a request packet
*/
static int
sendrequest(implcode, reqcode, auth, qitems, qsize, qdata)
int implcode;
int reqcode;
int auth;
int qitems;
int qsize;
char *qdata;
{
struct req_pkt qpkt;
int datasize;
memset((char *)&qpkt, 0, sizeof qpkt);
qpkt.rm_vn_mode = RM_VN_MODE(0, 0);
qpkt.implementation = (u_char)implcode;
qpkt.request = (u_char)reqcode;
datasize = qitems * qsize;
if (datasize != 0 && qdata != NULL) {
memmove((char *)qpkt.data, qdata, datasize);
qpkt.err_nitems = ERR_NITEMS(0, qitems);
qpkt.mbz_itemsize = MBZ_ITEMSIZE(qsize);
} else {
qpkt.err_nitems = ERR_NITEMS(0, 0);
qpkt.mbz_itemsize = MBZ_ITEMSIZE(0);
}
if (!auth) {
qpkt.auth_seq = AUTH_SEQ(0, 0);
return sendpkt((char *)&qpkt, REQ_LEN_NOMAC);
} else {
l_fp ts;
char *pass;
if (info_auth_keyid == 0) {
info_auth_keyid = getkeyid("Keyid: ");
if (info_auth_keyid == 0) {
(void) fprintf(stderr,
"Keyid must be defined, request not sent\n");
return 1;
}
}
if (!auth_havekey(info_auth_keyid)) {
pass = getpass("Password: ");
if (*pass != '\0')
authusekey(info_auth_keyid, info_auth_keytype,
pass);
}
if (auth_havekey(info_auth_keyid)) {
int maclen;
qpkt.auth_seq = AUTH_SEQ(1, 0);
qpkt.keyid = htonl(info_auth_keyid);
auth1crypt(info_auth_keyid, (U_LONG *)&qpkt,
REQ_LEN_NOMAC);
gettstamp(&ts);
L_ADD(&ts, &delay_time);
HTONL_FP(&ts, &qpkt.tstamp);
maclen = auth2crypt(info_auth_keyid, (U_LONG *)&qpkt,
REQ_LEN_NOMAC);
return sendpkt((char *)&qpkt, REQ_LEN_NOMAC+maclen);
} else {
(void) fprintf(stderr,
"No password, request not sent\n");
return 1;
}
}
/*NOTREACHED*/
}
/*
* doquery - send a request and process the response
*/
int
doquery(implcode, reqcode, auth, qitems, qsize, qdata, ritems, rsize, rdata,
quiet_mask)
int implcode;
int reqcode;
int auth;
int qitems;
int qsize;
char *qdata;
int *ritems;
int *rsize;
char **rdata;
int quiet_mask;
{
int res;
char junk[512];
fd_set fds;
struct timeval tvzero;
/*
* Check to make sure host is open
*/
if (!havehost) {
(void) fprintf(stderr, "***No host open, use `host' command\n");
return -1;
}
/*
* Poll the socket and clear out any pending data
*/
do {
tvzero.tv_sec = tvzero.tv_usec = 0;
FD_ZERO(&fds);
FD_SET(sockfd, &fds);
res = select(sockfd+1, &fds, (fd_set *)0, (fd_set *)0, &tvzero);
if (res == -1) {
warning("polling select", "", "");
return -1;
} else if (res > 0)
(void) read(sockfd, junk, sizeof junk);
} while (res > 0);
/*
* send a request
*/
res = sendrequest(implcode, reqcode, auth, qitems, qsize, qdata);
if (res != 0)
return res;
/*
* Get the response. If we got a standard error, print a message
*/
res = getresponse(implcode, reqcode, ritems, rsize, rdata);
/* log error message if not told to be quiet */
if ((res > 0) && (((1 << res) & quiet_mask) == 0)) {
switch(res) {
case INFO_ERR_IMPL:
(void) fprintf(stderr,
"***Server implementation incompatable with our own\n");
break;
case INFO_ERR_REQ:
(void) fprintf(stderr,
"***Server doesn't implement this request\n");
break;
case INFO_ERR_FMT:
(void) fprintf(stderr,
"***Server reports a format error in the received packet (shouldn't happen)\n");
break;
case INFO_ERR_NODATA:
(void) fprintf(stderr,
"***Server reports data not found\n");
break;
case INFO_ERR_AUTH:
(void) fprintf(stderr, "***Permission denied\n");
break;
case ERR_TIMEOUT:
(void) fprintf(stderr, "***Request timed out\n");
break;
case ERR_INCOMPLETE:
(void) fprintf(stderr,
"***Response from server was incomplete\n");
break;
default:
(void) fprintf(stderr,
"***Server returns unknown error code %d\n", res);
break;
}
}
return res;
}
/*
* getcmds - read commands from the standard input and execute them
*/
static void
getcmds()
{
char line[MAXLINE];
for (;;) {
if (interactive) {
(void) fputs(prompt, stderr);
(void) fflush(stderr);
}
if (fgets(line, sizeof line, stdin) == NULL)
return;
docmd(line);
}
}
/*
* abortcmd - catch interrupts and abort the current command
*/
static RETSIGTYPE
abortcmd(sig)
int sig;
{
if (current_output == stdout)
(void) fflush(stdout);
putc('\n', stderr);
(void) fflush(stderr);
if (jump) longjmp(interrupt_buf, 1);
}
/*
* docmd - decode the command line and execute a command
*/
static void
docmd(cmdline)
char *cmdline;
{
char *tokens[1+MAXARGS+2];
struct parse pcmd;
int ntok;
static int i;
struct xcmd *xcmd;
/*
* Tokenize the command line. If nothing on it, return.
*/
tokenize(cmdline, tokens, &ntok);
if (ntok == 0)
return;
/*
* Find the appropriate command description.
*/
i = findcmd(tokens[0], builtins, opcmds, &xcmd);
if (i == 0) {
(void) fprintf(stderr, "***Command `%s' unknown\n",
tokens[0]);
return;
} else if (i >= 2) {
(void) fprintf(stderr, "***Command `%s' ambiguous\n",
tokens[0]);
return;
}
/*
* Save the keyword, then walk through the arguments, interpreting
* as we go.
*/
pcmd.keyword = tokens[0];
pcmd.nargs = 0;
for (i = 0; i < MAXARGS && xcmd->arg[i] != NO; i++) {
if ((i+1) >= ntok) {
if (!(xcmd->arg[i] & OPT)) {
printusage(xcmd, stderr);
return;
}
break;
}
if ((xcmd->arg[i] & OPT) && (*tokens[i+1] == '>'))
break;
if (!getarg(tokens[i+1], (int)xcmd->arg[i], &pcmd.argval[i]))
return;
pcmd.nargs++;
}
i++;
if (i < ntok && *tokens[i] == '>') {
char *fname;
if (*(tokens[i]+1) != '\0')
fname = tokens[i]+1;
else if ((i+1) < ntok)
fname = tokens[i+1];
else {
(void) fprintf(stderr, "***No file for redirect\n");
return;
}
current_output = fopen(fname, "w");
if (current_output == NULL) {
(void) fprintf(stderr, "***Error opening %s: ", fname);
perror("");
return;
}
i = 1; /* flag we need a close */
} else {
current_output = stdout;
i = 0; /* flag no close */
}
if (interactive && setjmp(interrupt_buf)) {
return;
} else {
jump = 1;
(xcmd->handler)(&pcmd, current_output);
jump = 0;
if (i) (void) fclose(current_output);
}
}
/*
* tokenize - turn a command line into tokens
*/
static void
tokenize(line, tokens, ntok)
char *line;
char **tokens;
int *ntok;
{
register char *cp;
register char *sp;
static char tspace[MAXLINE];
sp = tspace;
cp = line;
for (*ntok = 0; *ntok < MAXTOKENS; (*ntok)++) {
tokens[*ntok] = sp;
while (ISSPACE(*cp))
cp++;
if (ISEOL(*cp))
break;
do {
*sp++ = *cp++;
} while (!ISSPACE(*cp) && !ISEOL(*cp));
*sp++ = '\0';
}
}
/*
* findcmd - find a command in a command description table
*/
static int
findcmd(str, clist1, clist2, cmd)
register char *str;
struct xcmd *clist1;
struct xcmd *clist2;
struct xcmd **cmd;
{
register struct xcmd *cl;
register int clen;
int nmatch;
struct xcmd *nearmatch = NULL;
struct xcmd *clist;
clen = strlen(str);
nmatch = 0;
if (clist1 != 0)
clist = clist1;
else if (clist2 != 0)
clist = clist2;
else
return 0;
again:
for (cl = clist; cl->keyword != 0; cl++) {
/* do a first character check, for efficiency */
if (*str != *(cl->keyword))
continue;
if (strncmp(str, cl->keyword, clen) == 0) {
/*
* Could be extact match, could be approximate.
* Is exact if the length of the keyword is the
* same as the str.
*/
if (*((cl->keyword) + clen) == '\0') {
*cmd = cl;
return 1;
}
nmatch++;
nearmatch = cl;
}
}
/*
* See if there is more to do. If so, go again. Sorry about the
* goto, too much looking at BSD sources...
*/
if (clist == clist1 && clist2 != 0) {
clist = clist2;
goto again;
}
/*
* If we got extactly 1 near match, use it, else return number
* of matches.
*/
if (nmatch == 1) {
*cmd = nearmatch;
return 1;
}
return nmatch;
}
/*
* getarg - interpret an argument token
*/
static int
getarg(str, code, argp)
char *str;
int code;
arg_v *argp;
{
int isneg;
char *cp, *np;
static char *digits = "0123456789";
switch (code & ~OPT) {
case NTP_STR:
argp->string = str;
break;
case ADD:
if (!getnetnum(str, &(argp->netnum), (char *)0)) {
return 0;
}
break;
case INT:
case UINT:
isneg = 0;
np = str;
if (*np == '-') {
np++;
isneg = 1;
}
argp->uval = 0;
do {
cp = strchr(digits, *np);
if (cp == NULL) {
(void) fprintf(stderr,
"***Illegal integer value %s\n", str);
return 0;
}
argp->uval *= 10;
argp->uval += (cp - digits);
} while (*(++np) != '\0');
if (isneg) {
if ((code & ~OPT) == UINT) {
(void) fprintf(stderr,
"***Value %s should be unsigned\n", str);
return 0;
}
argp->ival = -argp->ival;
}
break;
}
return 1;
}
/*
* getnetnum - given a host name, return its net number
* and (optional) full name
*/
static int
getnetnum(host, num, fullhost)
char *host;
u_long *num;
char *fullhost;
{
struct hostent *hp;
if (decodenetnum(host, num)) {
if (fullhost != 0) {
(void) sprintf(fullhost,
"%u.%u.%u.%u", (u_int)((htonl(*num)>>24)&0xff),
(u_int)((htonl(*num)>>16)&0xff), (u_int)((htonl(*num)>>8)&0xff),
(u_int)(htonl(*num)&0xff));
}
return 1;
} else if ((hp = gethostbyname(host)) != 0) {
memmove((char *)num, hp->h_addr, sizeof(u_long));
if (fullhost != 0)
(void) strcpy(fullhost, hp->h_name);
return 1;
} else {
(void) fprintf(stderr, "***Can't find host %s\n", host);
return 0;
}
/*NOTREACHED*/
}
/*
* nntohost - convert network number to host name. This routine enforces
* the showhostnames setting.
*/
char *
nntohost(netnum)
u_long netnum;
{
if (!showhostnames)
return numtoa(netnum);
if ((ntohl(netnum) & REFCLOCK_MASK) == REFCLOCK_ADDR)
return refnumtoa(netnum);
return numtohost(netnum);
}
/*
* Finally, the built in command handlers
*/
/*
* help - tell about commands, or details of a particular command
*/
static void
help(pcmd, fp)
struct parse *pcmd;
FILE *fp;
{
int i;
int n;
struct xcmd *xcp;
char *cmd;
char *cmdsort[100];
int length[100];
int maxlength;
int numperline;
static char *spaces = " "; /* 20 spaces */
if (pcmd->nargs == 0) {
n = 0;
for (xcp = builtins; xcp->keyword != 0; xcp++) {
if (*(xcp->keyword) != '?')
cmdsort[n++] = xcp->keyword;
}
for (xcp = opcmds; xcp->keyword != 0; xcp++)
cmdsort[n++] = xcp->keyword;
#if defined(sgi) || defined(SYS_BSDI) || defined(__STDC__)
qsort((void *)cmdsort, n, sizeof(char *), helpsort);
#else
qsort((char *)cmdsort, n, sizeof(char *), helpsort);
#endif /* sgi */
maxlength = 0;
for (i = 0; i < n; i++) {
length[i] = strlen(cmdsort[i]);
if (length[i] > maxlength)
maxlength = length[i];
}
maxlength++;
numperline = 76 / maxlength;
(void) fprintf(fp, "Commands available:\n");
for (i = 0; i < n; i++) {
if ((i % numperline) == (numperline-1)
|| i == (n-1))
(void) fprintf(fp, "%s\n", cmdsort[i]);
else
(void) fprintf(fp, "%s%s", cmdsort[i],
spaces+20-maxlength+length[i]);
}
} else {
cmd = pcmd->argval[0].string;
n = findcmd(cmd, builtins, opcmds, &xcp);
if (n == 0) {
(void) fprintf(stderr,
"Command `%s' is unknown\n", cmd);
return;
} else if (n >= 2) {
(void) fprintf(stderr,
"Command `%s' is ambiguous\n", cmd);
return;
}
(void) fprintf(fp, "function: %s\n", xcp->comment);
printusage(xcp, fp);
}
}
/*
* helpsort - do hostname qsort comparisons
*/
static int
#if defined(sgi) || defined(SYS_BSDI) || defined(__STDC__)
helpsort(t1, t2)
const void *t1;
const void *t2;
{
const char **name1 = (const char **)t1;
const char **name2 = (const char **)t2;
#else
helpsort(name1, name2)
char **name1;
char **name2;
{
#endif /* sgi || bsdi */
return strcmp(*name1, *name2);
}
/*
* printusage - print usage information for a command
*/
static void
printusage(xcp, fp)
struct xcmd *xcp;
FILE *fp;
{
register int i;
(void) fprintf(fp, "usage: %s", xcp->keyword);
for (i = 0; i < MAXARGS && xcp->arg[i] != NO; i++) {
if (xcp->arg[i] & OPT)
(void) fprintf(fp, " [ %s ]", xcp->desc[i]);
else
(void) fprintf(fp, " %s", xcp->desc[i]);
}
(void) fprintf(fp, "\n");
}
/*
* timeout - set time out time
*/
static void
timeout(pcmd, fp)
struct parse *pcmd;
FILE *fp;
{
int val;
if (pcmd->nargs == 0) {
val = tvout.tv_sec * 1000 + tvout.tv_usec / 1000;
(void) fprintf(fp, "primary timeout %d ms\n", val);
} else {
tvout.tv_sec = pcmd->argval[0].uval / 1000;
tvout.tv_usec = (pcmd->argval[0].uval - (tvout.tv_sec * 1000))
* 1000;
}
}
/*
* delay - set delay for auth requests
*/
static void
delay(pcmd, fp)
struct parse *pcmd;
FILE *fp;
{
int isneg;
u_long val;
if (pcmd->nargs == 0) {
val = delay_time.l_ui * 1000 + delay_time.l_uf / 4294967;
(void) fprintf(fp, "delay %lu ms\n", val);
} else {
if (pcmd->argval[0].ival < 0) {
isneg = 1;
val = (u_long)(-pcmd->argval[0].ival);
} else {
isneg = 0;
val = (u_long)pcmd->argval[0].ival;
}
delay_time.l_ui = val / 1000;
val %= 1000;
delay_time.l_uf = val * 4294967; /* 2**32/1000 */
if (isneg)
L_NEG(&delay_time);
}
}
/*
* host - set the host we are dealing with.
*/
static void
host(pcmd, fp)
struct parse *pcmd;
FILE *fp;
{
if (pcmd->nargs == 0) {
if (havehost)
(void) fprintf(fp, "current host is %s\n", currenthost);
else
(void) fprintf(fp, "no current host\n");
} else if (openhost(pcmd->argval[0].string)) {
(void) fprintf(fp, "current host set to %s\n", currenthost);
} else {
if (havehost)
(void) fprintf(fp,
"current host remains %s\n", currenthost);
else
(void) fprintf(fp, "still no current host\n");
}
}
/*
* keyid - get a keyid to use for authenticating requests
*/
static void
keyid(pcmd, fp)
struct parse *pcmd;
FILE *fp;
{
if (pcmd->nargs == 0) {
if (info_auth_keyid == 0)
(void) fprintf(fp, "no keyid defined\n");
else
(void) fprintf(fp, "keyid is %lu\n", info_auth_keyid);
} else {
info_auth_keyid = pcmd->argval[0].uval;
}
}
/*
* keytype - get type of key to use for authenticating requests
*/
static void
keytype(pcmd, fp)
struct parse *pcmd;
FILE *fp;
{
if (pcmd->nargs == 0)
fprintf(fp, "keytype is %s",
(info_auth_keytype == KEY_TYPE_MD5) ? "md5" : "des");
else
switch (*(pcmd->argval[0].string)) {
case 'm':
case 'M':
info_auth_keytype = KEY_TYPE_MD5;
break;
case 'd':
case 'D':
info_auth_keytype = KEY_TYPE_DES;
break;
default:
fprintf(fp, "keytype must be 'md5' or 'des'\n");
}
}
/*
* passwd - get an authentication key
*/
/*ARGSUSED*/
static void
passwd(pcmd, fp)
struct parse *pcmd;
FILE *fp;
{
char *pass;
if (info_auth_keyid == 0) {
info_auth_keyid = getkeyid("Keyid: ");
if (info_auth_keyid == 0) {
(void)fprintf(fp, "Keyid must be defined\n");
return;
}
}
if (!interactive) {
authusekey(info_auth_keyid, info_auth_keytype,
pcmd->argval[0].string);
} else {
pass = getpass("Password: ");
if (*pass == '\0')
(void) fprintf(fp, "Password unchanged\n");
else
authusekey(info_auth_keyid, info_auth_keytype, pass);
}
}
/*
* hostnames - set the showhostnames flag
*/
static void
hostnames(pcmd, fp)
struct parse *pcmd;
FILE *fp;
{
if (pcmd->nargs == 0) {
if (showhostnames)
(void) fprintf(fp, "hostnames being shown\n");
else
(void) fprintf(fp, "hostnames not being shown\n");
} else {
if (STREQ(pcmd->argval[0].string, "yes"))
showhostnames = 1;
else if (STREQ(pcmd->argval[0].string, "no"))
showhostnames = 0;
else
(void)fprintf(stderr, "What?\n");
}
}
/*
* setdebug - set/change debugging level
*/
static void
setdebug(pcmd, fp)
struct parse *pcmd;
FILE *fp;
{
if (pcmd->nargs == 0) {
(void) fprintf(fp, "debug level is %d\n", debug);
return;
} else if (STREQ(pcmd->argval[0].string, "no")) {
debug = 0;
} else if (STREQ(pcmd->argval[0].string, "more")) {
debug++;
} else if (STREQ(pcmd->argval[0].string, "less")) {
debug--;
} else {
(void) fprintf(fp, "What?\n");
return;
}
(void) fprintf(fp, "debug level set to %d\n", debug);
}
/*
* quit - stop this nonsense
*/
/*ARGSUSED*/
static void
quit(pcmd, fp)
struct parse *pcmd;
FILE *fp;
{
if (havehost)
(void) close(sockfd); /* cleanliness next to godliness */
exit(0);
}
/*
* version - print the current version number
*/
/*ARGSUSED*/
static void
version(pcmd, fp)
struct parse *pcmd;
FILE *fp;
{
extern char *Version;
(void) fprintf(fp, "%s\n", Version);
}
/*
* warning - print a warning message
*/
static void
warning(fmt, st1, st2)
char *fmt;
char *st1;
char *st2;
{
(void) fprintf(stderr, "%s: ", progname);
(void) fprintf(stderr, fmt, st1, st2);
(void) fprintf(stderr, ": ");
perror("");
}
/*
* error - print a message and exit
*/
static void
error(fmt, st1, st2)
char *fmt;
char *st1;
char *st2;
{
warning(fmt, st1, st2);
exit(1);
}
/*
* getkeyid - prompt the user for a keyid to use
*/
static u_long
getkeyid(prompt)
char *prompt;
{
register char *p;
register c;
FILE *fi;
char pbuf[20];
if ((fi = fdopen(open("/dev/tty", 2), "r")) == NULL)
fi = stdin;
else
setbuf(fi, (char *)NULL);
fprintf(stderr, "%s", prompt); fflush(stderr);
for (p=pbuf; (c = getc(fi))!='\n' && c!=EOF;) {
if (p < &pbuf[18])
*p++ = c;
}
*p = '\0';
if (fi != stdin)
fclose(fi);
return (u_long)atoi(pbuf);
}
Reference in New Issue View Git Blame Copy Permalink