1*7c478bd9Sstevel@tonic-gate /*
2*7c478bd9Sstevel@tonic-gate  * CDDL HEADER START
3*7c478bd9Sstevel@tonic-gate  *
4*7c478bd9Sstevel@tonic-gate  * The contents of this file are subject to the terms of the
5*7c478bd9Sstevel@tonic-gate  * Common Development and Distribution License, Version 1.0 only
6*7c478bd9Sstevel@tonic-gate  * (the "License").  You may not use this file except in compliance
7*7c478bd9Sstevel@tonic-gate  * with the License.
8*7c478bd9Sstevel@tonic-gate  *
9*7c478bd9Sstevel@tonic-gate  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10*7c478bd9Sstevel@tonic-gate  * or http://www.opensolaris.org/os/licensing.
11*7c478bd9Sstevel@tonic-gate  * See the License for the specific language governing permissions
12*7c478bd9Sstevel@tonic-gate  * and limitations under the License.
13*7c478bd9Sstevel@tonic-gate  *
14*7c478bd9Sstevel@tonic-gate  * When distributing Covered Code, include this CDDL HEADER in each
15*7c478bd9Sstevel@tonic-gate  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16*7c478bd9Sstevel@tonic-gate  * If applicable, add the following below this CDDL HEADER, with the
17*7c478bd9Sstevel@tonic-gate  * fields enclosed by brackets "[]" replaced with your own identifying
18*7c478bd9Sstevel@tonic-gate  * information: Portions Copyright [yyyy] [name of copyright owner]
19*7c478bd9Sstevel@tonic-gate  *
20*7c478bd9Sstevel@tonic-gate  * CDDL HEADER END
21*7c478bd9Sstevel@tonic-gate  */
22*7c478bd9Sstevel@tonic-gate /*
23*7c478bd9Sstevel@tonic-gate  * Copyright 2004 Sun Microsystems, Inc.  All rights reserved.
24*7c478bd9Sstevel@tonic-gate  * Use is subject to license terms.
25*7c478bd9Sstevel@tonic-gate  */
26*7c478bd9Sstevel@tonic-gate 
27*7c478bd9Sstevel@tonic-gate #pragma ident	"%Z%%M%	%I%	%E% SMI"
28*7c478bd9Sstevel@tonic-gate 
29*7c478bd9Sstevel@tonic-gate /*
30*7c478bd9Sstevel@tonic-gate  * chargen inetd service - both stream and dgram based.
31*7c478bd9Sstevel@tonic-gate  */
32*7c478bd9Sstevel@tonic-gate 
33*7c478bd9Sstevel@tonic-gate #include <sys/types.h>
34*7c478bd9Sstevel@tonic-gate #include <sys/socket.h>
35*7c478bd9Sstevel@tonic-gate #include <unistd.h>
36*7c478bd9Sstevel@tonic-gate #include <stdio.h>
37*7c478bd9Sstevel@tonic-gate #include <strings.h>
38*7c478bd9Sstevel@tonic-gate #include <netinet/in.h>
39*7c478bd9Sstevel@tonic-gate #include <sys/sysmacros.h>
40*7c478bd9Sstevel@tonic-gate #include <ctype.h>
41*7c478bd9Sstevel@tonic-gate #include <inetsvc.h>
42*7c478bd9Sstevel@tonic-gate 
43*7c478bd9Sstevel@tonic-gate 
44*7c478bd9Sstevel@tonic-gate #define	LINESIZE		72
45*7c478bd9Sstevel@tonic-gate #define	RINGSIZE		128
46*7c478bd9Sstevel@tonic-gate 
47*7c478bd9Sstevel@tonic-gate 
48*7c478bd9Sstevel@tonic-gate static	char	ring[RINGSIZE];
49*7c478bd9Sstevel@tonic-gate static	char	*endring;
50*7c478bd9Sstevel@tonic-gate 
51*7c478bd9Sstevel@tonic-gate 
52*7c478bd9Sstevel@tonic-gate static void
initring(void)53*7c478bd9Sstevel@tonic-gate initring(void)
54*7c478bd9Sstevel@tonic-gate {
55*7c478bd9Sstevel@tonic-gate 	unsigned char ch;
56*7c478bd9Sstevel@tonic-gate 
57*7c478bd9Sstevel@tonic-gate 	endring = ring;
58*7c478bd9Sstevel@tonic-gate 
59*7c478bd9Sstevel@tonic-gate 	for (ch = 0; ch <= RINGSIZE; ++ch) {
60*7c478bd9Sstevel@tonic-gate 		if (isprint(ch))
61*7c478bd9Sstevel@tonic-gate 			*endring++ = ch;
62*7c478bd9Sstevel@tonic-gate 	}
63*7c478bd9Sstevel@tonic-gate }
64*7c478bd9Sstevel@tonic-gate 
65*7c478bd9Sstevel@tonic-gate static void
chargen_stream(int s,char * argv[])66*7c478bd9Sstevel@tonic-gate chargen_stream(int s, char *argv[])
67*7c478bd9Sstevel@tonic-gate {
68*7c478bd9Sstevel@tonic-gate 	char	text[LINESIZE+2];
69*7c478bd9Sstevel@tonic-gate 	int	i;
70*7c478bd9Sstevel@tonic-gate 	char	*rp;
71*7c478bd9Sstevel@tonic-gate 	char	*dp;
72*7c478bd9Sstevel@tonic-gate 	char	*rs = ring;
73*7c478bd9Sstevel@tonic-gate 
74*7c478bd9Sstevel@tonic-gate 	setproctitle("chargen", s, argv);
75*7c478bd9Sstevel@tonic-gate 
76*7c478bd9Sstevel@tonic-gate 	for (;;) {
77*7c478bd9Sstevel@tonic-gate 		if (rs >= endring)
78*7c478bd9Sstevel@tonic-gate 			rs = ring;
79*7c478bd9Sstevel@tonic-gate 		rp = rs++;
80*7c478bd9Sstevel@tonic-gate 		dp = text;
81*7c478bd9Sstevel@tonic-gate 		i = MIN(LINESIZE, endring - rp);
82*7c478bd9Sstevel@tonic-gate 		(void) memmove(dp, rp, i);
83*7c478bd9Sstevel@tonic-gate 		dp += i;
84*7c478bd9Sstevel@tonic-gate 		if ((rp += i) >= endring)
85*7c478bd9Sstevel@tonic-gate 			rp = ring;
86*7c478bd9Sstevel@tonic-gate 		if (i < LINESIZE) {
87*7c478bd9Sstevel@tonic-gate 			i = LINESIZE - i;
88*7c478bd9Sstevel@tonic-gate 			(void) memmove(dp, rp, i);
89*7c478bd9Sstevel@tonic-gate 			dp += i;
90*7c478bd9Sstevel@tonic-gate 			if ((rp += i) >= endring)
91*7c478bd9Sstevel@tonic-gate 				rp = ring;
92*7c478bd9Sstevel@tonic-gate 		}
93*7c478bd9Sstevel@tonic-gate 
94*7c478bd9Sstevel@tonic-gate 		*dp++ = '\r';
95*7c478bd9Sstevel@tonic-gate 		*dp++ = '\n';
96*7c478bd9Sstevel@tonic-gate 
97*7c478bd9Sstevel@tonic-gate 		if (safe_write(s, text, dp - text) != 0)
98*7c478bd9Sstevel@tonic-gate 			break;
99*7c478bd9Sstevel@tonic-gate 	}
100*7c478bd9Sstevel@tonic-gate }
101*7c478bd9Sstevel@tonic-gate 
102*7c478bd9Sstevel@tonic-gate /* ARGSUSED3 */
103*7c478bd9Sstevel@tonic-gate static void
chargen_dg(int s,const struct sockaddr * sap,int sa_size,const void * buf,size_t sz)104*7c478bd9Sstevel@tonic-gate chargen_dg(int s, const struct sockaddr *sap, int sa_size, const void *buf,
105*7c478bd9Sstevel@tonic-gate     size_t sz)
106*7c478bd9Sstevel@tonic-gate {
107*7c478bd9Sstevel@tonic-gate 	char		text[LINESIZE+2];
108*7c478bd9Sstevel@tonic-gate 	int		i;
109*7c478bd9Sstevel@tonic-gate 	char		*rp;
110*7c478bd9Sstevel@tonic-gate 	static char 	*rs = ring;
111*7c478bd9Sstevel@tonic-gate 
112*7c478bd9Sstevel@tonic-gate 	rp = rs;
113*7c478bd9Sstevel@tonic-gate 	if (rs++ >= endring)
114*7c478bd9Sstevel@tonic-gate 		rs = ring;
115*7c478bd9Sstevel@tonic-gate 	i = MIN(LINESIZE, endring - rp);
116*7c478bd9Sstevel@tonic-gate 	(void) memmove(text, rp, i);
117*7c478bd9Sstevel@tonic-gate 	if ((rp += i) >= endring)
118*7c478bd9Sstevel@tonic-gate 		rp = ring;
119*7c478bd9Sstevel@tonic-gate 	if (i < LINESIZE) {
120*7c478bd9Sstevel@tonic-gate 		(void) memmove(text, rp, i);
121*7c478bd9Sstevel@tonic-gate 		if ((rp += i) >= endring)
122*7c478bd9Sstevel@tonic-gate 			rp = ring;
123*7c478bd9Sstevel@tonic-gate 	}
124*7c478bd9Sstevel@tonic-gate 
125*7c478bd9Sstevel@tonic-gate 	text[LINESIZE - 2] = '\r';
126*7c478bd9Sstevel@tonic-gate 	text[LINESIZE - 1] = '\n';
127*7c478bd9Sstevel@tonic-gate 
128*7c478bd9Sstevel@tonic-gate 	(void) safe_sendto(s, text, sizeof (text), 0, sap, sa_size);
129*7c478bd9Sstevel@tonic-gate }
130*7c478bd9Sstevel@tonic-gate 
131*7c478bd9Sstevel@tonic-gate int
main(int argc,char * argv[])132*7c478bd9Sstevel@tonic-gate main(int argc, char *argv[])
133*7c478bd9Sstevel@tonic-gate {
134*7c478bd9Sstevel@tonic-gate 	opterr = 0;	/* disable getopt error msgs */
135*7c478bd9Sstevel@tonic-gate 
136*7c478bd9Sstevel@tonic-gate 	initring();
137*7c478bd9Sstevel@tonic-gate 
138*7c478bd9Sstevel@tonic-gate 	switch (getopt(argc, argv, "ds")) {
139*7c478bd9Sstevel@tonic-gate 	case 'd':
140*7c478bd9Sstevel@tonic-gate 		dg_template(chargen_dg, STDIN_FILENO, NULL, 0);
141*7c478bd9Sstevel@tonic-gate 		break;
142*7c478bd9Sstevel@tonic-gate 	case 's':
143*7c478bd9Sstevel@tonic-gate 		chargen_stream(STDIN_FILENO, argv);
144*7c478bd9Sstevel@tonic-gate 		break;
145*7c478bd9Sstevel@tonic-gate 	default:
146*7c478bd9Sstevel@tonic-gate 		return (1);
147*7c478bd9Sstevel@tonic-gate 	}
148*7c478bd9Sstevel@tonic-gate 
149*7c478bd9Sstevel@tonic-gate 	return (0);
150*7c478bd9Sstevel@tonic-gate }
151