rwall.c revision 7c478bd95313f5f23a4c958a745db2134aa03244
/*
* CDDL HEADER START
*
* The contents of this file are subject to the terms of the
* Common Development and Distribution License, Version 1.0 only
* (the "License"). You may not use this file except in compliance
* with the License.
*
* You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
* or http://www.opensolaris.org/os/licensing.
* See the License for the specific language governing permissions
* and limitations under the License.
*
* When distributing Covered Code, include this CDDL HEADER in each
* file and include the License file at usr/src/OPENSOLARIS.LICENSE.
* If applicable, add the following below this CDDL HEADER, with the
* fields enclosed by brackets "[]" replaced with your own identifying
* information: Portions Copyright [yyyy] [name of copyright owner]
*
* CDDL HEADER END
*
* Copyright 2005 Sun Microsystems, Inc. All rights reserved.
* Use is subject to license terms.
*/
/* Copyright (c) 1983, 1984, 1985, 1986, 1987, 1988, 1989 AT&T */
/* All Rights Reserved */
/*
* University Copyright- Copyright (c) 1982, 1986, 1988
* The Regents of the University of California
* All Rights Reserved
*
* University Acknowledgment- Portions of this document are derived from
* software developed by the University of California, Berkeley, and its
* contributors.
*/
#pragma ident "%Z%%M% %I% %E% SMI"
/*
* rwall.c
* The client rwall program
*/
#include <stdio.h>
#include <sys/types.h>
#include <stdlib.h>
#include <unistd.h>
#include <thread.h>
#include <string.h>
#include <rpc/rpc.h>
#include <signal.h>
#include <pwd.h>
#include <rpcsvc/rwall.h>
#include <netconfig.h>
#include <netdb.h>
#include <sys/time.h>
#include <sys/resource.h>
static void init_who(void);
static void doall(void);
static void doit(char *);
static void *do_one(void *);
static void usage(void);
#define PATIENCE 10
#define MAX_THREADS 1024
static mutex_t tty = DEFAULTMUTEX;
static char who[9] = "???";
static char *path;
static mutex_t thr_mtx = DEFAULTMUTEX;
static int thread_count = 8; /* fudge factor for system threads/fds */
static int qflag = 0; /* quiet: we don't care about errors */
main(argc, argv)
int argc;
char **argv;
{
int msize;
char buf[BUFSIZ+1];
register i;
char hostname[256];
int hflag;
struct rlimit rl;
if (argc < 2)
usage();
if (getrlimit(RLIMIT_NOFILE, &rl) == 0) {
rl.rlim_cur = (rl.rlim_max < MAX_THREADS ?
rl.rlim_max : MAX_THREADS);
(void) setrlimit(RLIMIT_NOFILE, &rl);
}
(void) gethostname(hostname, sizeof (hostname));
init_who();
msize = snprintf(buf, sizeof (buf), "From %s@%s: ", who, hostname);
while ((i = getchar()) != EOF) {
if (msize >= (sizeof (buf) - 1)) {
(void) fprintf(stderr, "Message too long\n");
exit(1);
}
buf[msize++] = i;
}
buf[msize] = '\0';
path = buf;
hflag = 1;
while (argc > 1) {
if (argv[1][0] == '-') {
switch (argv[1][1]) {
case 'h':
hflag = 1;
break;
case 'n':
hflag = 0;
break;
case 'q':
qflag = 1;
break;
default:
usage();
break;
}
argc--;
argv++;
continue;
}
if (hflag) {
doit(argv[1]);
} else {
char *machine, *user, *domain;
(void) setnetgrent(argv[1]);
while (getnetgrent(&machine, &user, &domain)) {
if (machine)
doit(machine);
else
doall();
}
(void) endnetgrent();
}
argc--;
argv++;
}
thr_exit(NULL);
return (0);
}
static void
init_who(void)
{
char *wp;
struct passwd *pwd;
wp = getlogin();
if (wp != NULL)
(void) strncpy(who, wp, sizeof (who));
else {
pwd = getpwuid(getuid());
if (pwd)
(void) strncpy(who, pwd->pw_name, sizeof (who));
}
}
/*
* Saw a wild card, so do everything
*/
static void
doall(void)
{
(void) mutex_lock(&tty);
(void) fprintf(stderr, "writing to everyone not supported\n");
(void) mutex_unlock(&tty);
}
/*
* Fire off a detached thread for each host in the list, if the thread
* create fails simply run synchronously.
*/
static void
doit(char *hostname)
{
thread_t tid;
char *thread_hostname;
(void) mutex_lock(&thr_mtx);
while (thread_count >= MAX_THREADS) {
(void) mutex_unlock(&thr_mtx);
(void) sleep(PATIENCE/2);
(void) mutex_lock(&thr_mtx);
}
thread_count++;
(void) mutex_unlock(&thr_mtx);
thread_hostname = strdup(hostname);
if (thread_hostname == (char *)NULL) {
(void) mutex_lock(&tty);
(void) fprintf(stderr, "Ran out of memory\n");
(void) mutex_unlock(&tty);
exit(1);
}
if (thr_create(NULL, 0, do_one, thread_hostname,
THR_DETACHED, &tid) != 0) {
(void) do_one(thread_hostname);
}
}
static void *
do_one(void *arg)
{
char *hostname = arg;
CLIENT *clnt;
struct timeval tp;
void *vp = NULL;
#ifdef DEBUG
(void) mutex_lock(&tty);
(void) fprintf(stderr, "sending message to %s\n%s\n", hostname, path);
(void) mutex_unlock(&tty);
return (0);
#endif
tp.tv_sec = PATIENCE;
tp.tv_usec = 0;
clnt = clnt_create_timed(
hostname, WALLPROG, WALLVERS, "datagram_v", &tp);
if (clnt == NULL) {
if (!qflag) {
(void) mutex_lock(&tty);
(void) fprintf(stderr, "rwall: Can't send to %s\n",
hostname);
clnt_pcreateerror(hostname);
(void) mutex_unlock(&tty);
}
goto errout;
}
if (wallproc_wall_1(&path, vp, clnt) != RPC_SUCCESS) {
if (!qflag) {
(void) mutex_lock(&tty);
clnt_perror(clnt, hostname);
(void) mutex_unlock(&tty);
}
}
clnt_destroy(clnt);
errout:
(void) mutex_lock(&thr_mtx);
thread_count--;
(void) mutex_unlock(&thr_mtx);
free(hostname);
return (0);
}
static void
usage(void)
{
(void) fprintf(stderr,
"Usage: rwall [-q] host .... [-n netgroup ....] [-h host ...]\n");
exit(1);
}