int
main (int argc, char *argv[])
{
char *server = NULL;
int msize = 65536;
uid_t uid = geteuid ();
int topt = 0;
Npcfsys *fs = NULL;
Npcfid *fid, *afid, *root;
int c, fd;
char buf[80], *host, *p;
hostlist_t hl;
hostlist_iterator_t itr;
int lopt = 0;
diod_log_init (argv[0]);
opterr = 0;
while ((c = GETOPT (argc, argv, OPTIONS, longopts)) != -1) {
switch (c) {
case 's': /* --server HOST[:PORT] or /path/to/socket */
server = optarg;
break;
case 'm': /* --msize SIZE */
msize = strtoul (optarg, NULL, 10);
break;
case 'u': /* --uid UID */
uid = strtoul (optarg, NULL, 10);
break;
case 't': /* --timeout SECS */
topt = strtoul (optarg, NULL, 10);
break;
case 'l': /* --long */
lopt = 1;
break;
default:
usage ();
}
}
if (signal (SIGPIPE, SIG_IGN) == SIG_ERR)
err_exit ("signal");
if (signal (SIGALRM, sigalarm) == SIG_ERR)
err_exit ("signal");
if (topt > 0)
alarm (topt);
if ((fd = diod_sock_connect (server, 0)) < 0)
exit (1);
if (!(fs = npc_start (fd, fd, msize, 0)))
errn_exit (np_rerror (), "error negotiating protocol with server");
if (!(afid = npc_auth (fs, "ctl", uid, diod_auth)) && np_rerror () != 0)
errn_exit (np_rerror (), "error authenticating to server");
if (!(root = npc_attach (fs, afid, "ctl", uid)))
errn_exit (np_rerror (), "error attaching to aname=ctl");
if (!(fid = npc_open_bypath (root, "connections", O_RDONLY)))
errn_exit (np_rerror (), "open connections");
if (!(hl = hostlist_create (NULL)))
err_exit ("hostlist_create");
while (npc_gets (fid, buf, sizeof(buf))) {
if ((p = strchr (buf, ' ')))
*p = '\0';
if (!lopt && (p = strchr (buf, '.')))
*p = '\0';
if (!hostlist_push_host (hl, buf))
err_exit ("hostlist_push_host");
}
hostlist_uniq (hl);
if (lopt) {
if (!(itr = hostlist_iterator_create (hl)))
err_exit ("hostlist_iterator_create");
while ((host = hostlist_next (itr)))
printf ("%s\n", host);
hostlist_iterator_destroy (itr);
} else {
char s[1024];
if (hostlist_ranged_string (hl, sizeof (s), s) < 0)
msg_exit ("hostlist output would be too long (use -l)");
printf ("%s\n", s);
}
hostlist_destroy (hl);
if (npc_clunk (fid) < 0)
errn_exit (np_rerror (), "clunk connections");
if (npc_clunk (root) < 0)
errn_exit (np_rerror (), "error clunking ctl");
if (npc_clunk (afid) < 0)
errn_exit (np_rerror (), "error clunking afid");
npc_finish (fs);
exit(0);
}
int
main (int argc, char *argv[])
{
char *aname = NULL;
char *server = NULL;
int msize = 65536;
uid_t uid = geteuid ();
int topt = 0;
int lopt = 0;
int fd, c;
diod_log_init (argv[0]);
opterr = 0;
while ((c = GETOPT (argc, argv, OPTIONS, longopts)) != -1) {
switch (c) {
case 'a': /* --aname NAME */
aname = optarg;
break;
case 's': /* --server HOST[:PORT] or /path/to/socket */
server = optarg;
break;
case 'm': /* --msize SIZE */
msize = strtoul (optarg, NULL, 10);
break;
case 'u': /* --uid UID */
uid = strtoul (optarg, NULL, 10);
break;
case 't': /* --timeout SECS */
topt = strtoul (optarg, NULL, 10);
break;
case 'l': /* --long */
lopt = 1;
break;
default:
usage ();
}
}
if (signal (SIGPIPE, SIG_IGN) == SIG_ERR)
err_exit ("signal");
if (signal (SIGALRM, sigalarm) == SIG_ERR)
err_exit ("signal");
if (topt > 0)
alarm (topt);
if ((fd = diod_sock_connect (server, 0)) < 0)
exit (1);
if (!aname)
aname = "ctl";
if (lsfiles (fd, uid, msize, aname, lopt, argv + optind, argc - optind) < 0)
exit (1);
close (fd);
diod_log_fini ();
exit (0);
}
int
main (int argc, char *argv[])
{
char *aname = "ctl";
char *server = NULL;
int msize = 65536;
uid_t uid = geteuid ();
int topt = 0;
int Sopt = 0;
int flags = 0;
int fd, c;
Npcfsys *fs = NULL;
Npcfid *afid = NULL, *root = NULL;
char *buf = NULL;
struct timeval tv;
struct timezone tz;
diod_log_init (argv[0]);
opterr = 0;
while ((c = GETOPT (argc, argv, OPTIONS, longopts)) != -1) {
switch (c) {
case 's': /* --server HOST[:PORT] or /path/to/socket */
server = optarg;
break;
case 'm': /* --msize SIZE */
msize = strtoul (optarg, NULL, 10);
break;
case 'u': /* --uid UID */
uid = strtoul (optarg, NULL, 10);
break;
case 't': /* --timeout SECS */
topt = strtoul (optarg, NULL, 10);
break;
case 'S': /* --set-time */
Sopt = 1;
break;
default:
usage ();
}
}
if (signal (SIGPIPE, SIG_IGN) == SIG_ERR)
err_exit ("signal");
if (signal (SIGALRM, sigalarm) == SIG_ERR)
err_exit ("signal");
if (topt > 0)
alarm (topt);
if ((fd = diod_sock_connect (server, flags)) < 0)
exit (1);
if (!(fs = npc_start (fd, fd, msize, 0))) {
errn (np_rerror (), "error negotiating protocol with server");
goto done;
}
if (!(afid = npc_auth (fs, aname, uid, diod_auth)) && np_rerror () != 0) {
errn (np_rerror (), "error authenticating to server");
goto done;
}
if (!(root = npc_attach (fs, afid, aname, uid))) {
errn (np_rerror (), "error attaching to aname='%s'", aname);
goto done;
}
buf = npc_aget (root, "date");
if (!buf) {
errn (np_rerror (), "error reading date");
goto done;
}
if (sscanf (buf, "%lu.%lu %d.%d", &tv.tv_sec, &tv.tv_usec,
&tz.tz_minuteswest, &tz.tz_dsttime) != 4) {
msg ("error scanning returned date: %s", buf);
goto done;
}
if (Sopt) {
if (settimeofday (&tv, &tz) < 0)
err_exit ("settimeofday");
} else {
time_t t = tv.tv_sec;
printf ("%s", ctime (&t));
}
done:
if (buf)
free (buf);
if (root && npc_clunk (root) < 0)
errn_exit (np_rerror (), "error clunking %s", aname);
if (afid && npc_clunk (afid) < 0)
errn_exit (np_rerror (), "error clunking afid");
if (fs)
npc_finish (fs);
close (fd);
diod_log_fini ();
exit (0);
}
static void *
loadgen (void *arg)
{
thd_t *t = (thd_t *)arg;
uid_t uid = geteuid ();
void *ret = NULL;
int n, loops = 0;
if ((t->fd = diod_sock_connect (t->host, t->port, 0)) < 0)
goto done;
if (!(t->fs = npc_start (t->fd, t->fd, t->msize, 0))) {
errn (np_rerror (), "error negotiating protocol with server");
goto done;
}
if (!(t->afid = npc_auth (t->fs, "ctl", uid, diod_auth))
&& np_rerror () != 0) {
errn (np_rerror (), "error authenticating to server");
goto done;
}
if (!(t->root = npc_attach (t->fs, t->afid, "ctl", uid))) {
errn (np_rerror (), "error attaching to aname=ctl");
goto done;
}
if (t->loadtype == LOAD_IO) {
if (!(t->infile = npc_open_bypath (t->root, "zero", O_RDONLY))) {
errn (np_rerror (), "open zero");
goto done;
}
if (!(t->outfile = npc_open_bypath (t->root, "null", O_WRONLY))) {
errn (np_rerror (), "open null");
goto done;
}
do {
if ((n = npc_pread (t->infile, t->buf, t->msize, 0)) <= 0) {
errn (np_rerror (), "read zero");
break;
}
t->readbytes += n;
if ((n = npc_pwrite (t->outfile, t->buf, t->msize, 0)) < 0) {
errn (np_rerror (), "write null");
break;
}
t->writebytes += n;
t->opcount++;
loops++;
} while ((loops % 100 != 0) || time (NULL) < t->stoptime);
} else if (t->loadtype == LOAD_GETATTR) {
if (!(t->infile = npc_walk (t->root, "null"))) {
errn (np_rerror (), "walk null");
goto done;
}
do {
struct stat sb;
if (npc_fstat (t->infile, &sb) < 0) {
errn (np_rerror (), "walk null");
break;
}
t->opcount++;
loops++;
} while ((loops % 100 != 0) || time (NULL) < t->stoptime);
}
done:
if (t->outfile && npc_clunk (t->outfile) < 0)
errn (np_rerror (), "error clunking null");
if (t->infile && npc_clunk (t->infile) < 0)
errn (np_rerror (), "error clunking zero");
if (t->root && npc_clunk (t->root) < 0)
errn (np_rerror (), "error clunking ctl");
if (t->afid && npc_clunk (t->afid) < 0)
errn (np_rerror (), "error clunking afid");
if (t->fs) {
npc_finish (t->fs); /* closes fd */
t->fd = -1;
}
if (t->fd != -1)
close (t->fd);
return ret;
}
