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