diff options
Diffstat (limited to 'usr/src/cmd/rpcsvc/rwall.c')
| -rw-r--r-- | usr/src/cmd/rpcsvc/rwall.c | 265 |
1 files changed, 265 insertions, 0 deletions
diff --git a/usr/src/cmd/rpcsvc/rwall.c b/usr/src/cmd/rpcsvc/rwall.c new file mode 100644 index 0000000000..6c4a42ef70 --- /dev/null +++ b/usr/src/cmd/rpcsvc/rwall.c @@ -0,0 +1,265 @@ +/* + * 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); +} |