1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License, Version 1.0 only
6  * (the "License").  You may not use this file except in compliance
7  * with the License.
8  *
9  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10  * or http://www.opensolaris.org/os/licensing.
11  * See the License for the specific language governing permissions
12  * and limitations under the License.
13  *
14  * When distributing Covered Code, include this CDDL HEADER in each
15  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16  * If applicable, add the following below this CDDL HEADER, with the
17  * fields enclosed by brackets "[]" replaced with your own identifying
18  * information: Portions Copyright [yyyy] [name of copyright owner]
19  *
20  * CDDL HEADER END
21  */
22 /*
23  * Copyright 2004 Sun Microsystems, Inc.  All rights reserved.
24  * Use is subject to license terms.
25  */
26 
27 #pragma ident	"%Z%%M%	%I%	%E% SMI"
28 
29 /*
30  * echo inetd service - both stream and dgram based.
31  */
32 
33 #include <sys/types.h>
34 #include <sys/socket.h>
35 #include <unistd.h>
36 #include <stdio.h>
37 #include <netinet/in.h>
38 #include <inetsvc.h>
39 
40 
41 static void
42 echo_stream(int s, char *argv[])
43 {
44 	char	buffer[BUFSIZ];
45 	int	i;
46 
47 	setproctitle("echo", s, argv);
48 	while (((i = read(s, buffer, sizeof (buffer))) > 0) &&
49 	    (safe_write(s, buffer, i) == 0)) {
50 		;
51 	}
52 }
53 
54 static void
55 echo_dg(int s, const struct sockaddr *sap, int sa_size, const void *buf,
56     size_t bufsiz)
57 {
58 	(void) safe_sendto(s, buf, bufsiz, 0, sap, sa_size);
59 }
60 
61 
62 int
63 main(int argc, char *argv[])
64 {
65 	char		buf[BUFSIZ];
66 
67 	opterr = 0;	/* disable getopt error msgs */
68 	switch (getopt(argc, argv, "ds")) {
69 	case 'd':
70 		dg_template(echo_dg, STDIN_FILENO, buf, sizeof (buf));
71 		break;
72 	case 's':
73 		echo_stream(STDIN_FILENO, argv);
74 		break;
75 	default:
76 		return (1);
77 	}
78 
79 	return (0);
80 }
81