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  * chargen 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 <strings.h>
38 #include <netinet/in.h>
39 #include <sys/sysmacros.h>
40 #include <ctype.h>
41 #include <inetsvc.h>
42 
43 
44 #define	LINESIZE		72
45 #define	RINGSIZE		128
46 
47 
48 static	char	ring[RINGSIZE];
49 static	char	*endring;
50 
51 
52 static void
initring(void)53 initring(void)
54 {
55 	unsigned char ch;
56 
57 	endring = ring;
58 
59 	for (ch = 0; ch <= RINGSIZE; ++ch) {
60 		if (isprint(ch))
61 			*endring++ = ch;
62 	}
63 }
64 
65 static void
chargen_stream(int s,char * argv[])66 chargen_stream(int s, char *argv[])
67 {
68 	char	text[LINESIZE+2];
69 	int	i;
70 	char	*rp;
71 	char	*dp;
72 	char	*rs = ring;
73 
74 	setproctitle("chargen", s, argv);
75 
76 	for (;;) {
77 		if (rs >= endring)
78 			rs = ring;
79 		rp = rs++;
80 		dp = text;
81 		i = MIN(LINESIZE, endring - rp);
82 		(void) memmove(dp, rp, i);
83 		dp += i;
84 		if ((rp += i) >= endring)
85 			rp = ring;
86 		if (i < LINESIZE) {
87 			i = LINESIZE - i;
88 			(void) memmove(dp, rp, i);
89 			dp += i;
90 			if ((rp += i) >= endring)
91 				rp = ring;
92 		}
93 
94 		*dp++ = '\r';
95 		*dp++ = '\n';
96 
97 		if (safe_write(s, text, dp - text) != 0)
98 			break;
99 	}
100 }
101 
102 /* ARGSUSED3 */
103 static void
chargen_dg(int s,const struct sockaddr * sap,int sa_size,const void * buf,size_t sz)104 chargen_dg(int s, const struct sockaddr *sap, int sa_size, const void *buf,
105     size_t sz)
106 {
107 	char		text[LINESIZE+2];
108 	int		i;
109 	char		*rp;
110 	static char 	*rs = ring;
111 
112 	rp = rs;
113 	if (rs++ >= endring)
114 		rs = ring;
115 	i = MIN(LINESIZE, endring - rp);
116 	(void) memmove(text, rp, i);
117 	if ((rp += i) >= endring)
118 		rp = ring;
119 	if (i < LINESIZE) {
120 		(void) memmove(text, rp, i);
121 		if ((rp += i) >= endring)
122 			rp = ring;
123 	}
124 
125 	text[LINESIZE - 2] = '\r';
126 	text[LINESIZE - 1] = '\n';
127 
128 	(void) safe_sendto(s, text, sizeof (text), 0, sap, sa_size);
129 }
130 
131 int
main(int argc,char * argv[])132 main(int argc, char *argv[])
133 {
134 	opterr = 0;	/* disable getopt error msgs */
135 
136 	initring();
137 
138 	switch (getopt(argc, argv, "ds")) {
139 	case 'd':
140 		dg_template(chargen_dg, STDIN_FILENO, NULL, 0);
141 		break;
142 	case 's':
143 		chargen_stream(STDIN_FILENO, argv);
144 		break;
145 	default:
146 		return (1);
147 	}
148 
149 	return (0);
150 }
151