1/*-
2 * Copyright (c) 1993
3 * The Regents of the University of California. All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. All advertising materials mentioning features or use of this software
14 * must display the following acknowledgement:
15 * This product includes software developed by the University of
16 * California, Berkeley and its contributors.
17 * 4. Neither the name of the University nor the names of its contributors
18 * may be used to endorse or promote products derived from this software
19 * without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 * SUCH DAMAGE.
32 */
33
34#ifndef lint
35static const char copyright[] =
36"@(#) Copyright (c) 1993\n\
37 The Regents of the University of California. All rights reserved.\n";
38#endif /* not lint */
39
40#ifndef lint
41#if 0
42static char sccsid[] = "@(#)jot.c 8.1 (Berkeley) 6/6/93";
43#endif
44#endif
45#include <sys/cdefs.h>
46__FBSDID("$FreeBSD: head/usr.bin/jot/jot.c 132240 2004-07-16 05:10:46Z tjr $");
47
48/*
49 * jot - print sequential or random data
50 *
51 * Author: John Kunze, Office of Comp. Affairs, UCB
52 */
53
54#include <ctype.h>
55#include <err.h>
56#include <limits.h>
57#include <stdio.h>
58#include <stdint.h>
59#include <stdlib.h>
60#include <string.h>
61#include <time.h>
62#include <unistd.h>
63
64#define REPS_DEF 100
65#define BEGIN_DEF 1
66#define ENDER_DEF 100
67#define STEP_DEF 1
68
69#define is_default(s) (strcmp((s), "-") == 0)
70
71double begin;
72double ender;
73double s;
74long reps;
75int randomize;
76int infinity;
77int boring;
78int prec;
79int longdata;
80int intdata;
81int chardata;
82int nosign;
83int nofinalnl;
84const char *sepstring = "\n";
85char format[BUFSIZ];
86
87void getformat(void);
88int getprec(char *);
89int putdata(double, long);
90static void usage(void);
91
92int
93main(int argc, char **argv)
94{
95 double xd, yd;
96 long id;
97 double *x = &xd;
98 double *y = &yd;
99 long *i = &id;
100 unsigned int mask = 0;
101 int n = 0;
102 int ch;
103
104 while ((ch = getopt(argc, argv, "rb:w:cs:np:")) != -1)
105 switch (ch) {
106 case 'r':
107 randomize = 1;
108 break;
109 case 'c':
110 chardata = 1;
111 break;
112 case 'n':
113 nofinalnl = 1;
114 break;
115 case 'b':
116 boring = 1;
117 /* FALLTHROUGH */
118 case 'w':
119 if (strlcpy(format, optarg, sizeof(format)) >=
120 sizeof(format))
121 errx(1, "-%c word too long", ch);
122 break;
123 case 's':
124 sepstring = optarg;
125 break;
126 case 'p':
127 prec = atoi(optarg);
128 if (prec <= 0)
129 errx(1, "bad precision value");
130 break;
131 default:
132 usage();
133 }
134 argc -= optind;
135 argv += optind;
136
137 switch (argc) { /* examine args right to left, falling thru cases */
138 case 4:
139 if (!is_default(argv[3])) {
140 if (!sscanf(argv[3], "%lf", &s))
141 errx(1, "bad s value: %s", argv[3]);
142 mask |= 01;
143 }
144 case 3:
145 if (!is_default(argv[2])) {
146 if (!sscanf(argv[2], "%lf", &ender))
147 ender = argv[2][strlen(argv[2])-1];
148 mask |= 02;
149 if (!prec)
150 n = getprec(argv[2]);
151 }
152 case 2:
153 if (!is_default(argv[1])) {
154 if (!sscanf(argv[1], "%lf", &begin))
155 begin = argv[1][strlen(argv[1])-1];
156 mask |= 04;
157 if (!prec)
158 prec = getprec(argv[1]);
159 if (n > prec) /* maximum precision */
160 prec = n;
161 }
162 case 1:
163 if (!is_default(argv[0])) {
164 if (!sscanf(argv[0], "%ld", &reps))
165 errx(1, "bad reps value: %s", argv[0]);
166 mask |= 010;
167 }
168 break;
169 case 0:
170 usage();
171 default:
172 errx(1, "too many arguments. What do you mean by %s?",
173 argv[4]);
174 }
175 getformat();
176 while (mask) /* 4 bit mask has 1's where last 4 args were given */
177 switch (mask) { /* fill in the 0's by default or computation */
178 case 001:
179 reps = REPS_DEF;
180 mask = 011;
181 break;
182 case 002:
183 reps = REPS_DEF;
184 mask = 012;
185 break;
186 case 003:
187 reps = REPS_DEF;
188 mask = 013;
189 break;
190 case 004:
191 reps = REPS_DEF;
192 mask = 014;
193 break;
194 case 005:
195 reps = REPS_DEF;
196 mask = 015;
197 break;
198 case 006:
199 reps = REPS_DEF;
200 mask = 016;
201 break;
202 case 007:
203 if (randomize) {
204 reps = REPS_DEF;
205 mask = 0;
206 break;
207 }
208 if (s == 0.0) {
209 reps = 0;
210 mask = 0;
211 break;
212 }
213 reps = (ender - begin + s) / s;
214 if (reps <= 0)
215 errx(1, "impossible stepsize");
216 mask = 0;
217 break;
218 case 010:
219 begin = BEGIN_DEF;
220 mask = 014;
221 break;
222 case 011:
223 begin = BEGIN_DEF;
224 mask = 015;
225 break;
226 case 012:
227 s = (randomize ? time(NULL) : STEP_DEF);
228 mask = 013;
229 break;
230 case 013:
231 if (randomize)
232 begin = BEGIN_DEF;
233 else if (reps == 0)
234 errx(1, "must specify begin if reps == 0");
235 begin = ender - reps * s + s;
236 mask = 0;
237 break;
238 case 014:
239 s = (randomize ? -1.0 : STEP_DEF);
240 mask = 015;
241 break;
242 case 015:
243 if (randomize)
244 ender = ENDER_DEF;
245 else
246 ender = begin + reps * s - s;
247 mask = 0;
248 break;
249 case 016:
250 if (randomize)
251 s = -1.0;
252 else if (reps == 0)
253 errx(1, "infinite sequences cannot be bounded");
254 else if (reps == 1)
255 s = 0.0;
256 else
257 s = (ender - begin) / (reps - 1);
258 mask = 0;
259 break;
260 case 017: /* if reps given and implied, */
261 if (!randomize && s != 0.0) {
262 long t = (ender - begin + s) / s;
263 if (t <= 0)
264 errx(1, "impossible stepsize");
265 if (t < reps) /* take lesser */
266 reps = t;
267 }
268 mask = 0;
269 break;
270 default:
271 errx(1, "bad mask");
272 }
273 if (reps == 0)
274 infinity = 1;
275 if (randomize) {
276 *x = (ender - begin) * (ender > begin ? 1 : -1);
277 for (*i = 1; *i <= reps || infinity; (*i)++) {
278 *y = arc4random() / ((double)UINT32_MAX + 1);
279 if (putdata(*y * *x + begin, reps - *i))
280 errx(1, "range error in conversion");
281 }
282 } else
283 for (*i = 1, *x = begin; *i <= reps || infinity; (*i)++, *x += s)
284 if (putdata(*x, reps - *i))
285 errx(1, "range error in conversion");
286 if (!nofinalnl)
287 putchar('\n');
288 exit(0);
289}
290
291int
292putdata(double x, long int notlast)
293{
294
295 if (boring)
296 printf("%s", format);
297 else if (longdata && nosign) {
298 if (x <= (double)ULONG_MAX && x >= (double)0)
299 printf(format, (unsigned long)x);
300 else
301 return (1);
302 } else if (longdata) {
303 if (x <= (double)LONG_MAX && x >= (double)LONG_MIN)
304 printf(format, (long)x);
305 else
306 return (1);
307 } else if (chardata || (intdata && !nosign)) {
308 if (x <= (double)INT_MAX && x >= (double)INT_MIN)
309 printf(format, (int)x);
310 else
311 return (1);
312 } else if (intdata) {
313 if (x <= (double)UINT_MAX && x >= (double)0)
314 printf(format, (unsigned int)x);
315 else
316 return (1);
317
318 } else
319 printf(format, x);
320 if (notlast != 0)
321 fputs(sepstring, stdout);
322
323 return (0);
324}
325
326static void
327usage(void)
328{
329 fprintf(stderr, "%s\n%s\n",
330 "usage: jot [-cnr] [-b word] [-w word] [-s string] [-p precision]",
331 " [reps [begin [end [s]]]]");
332 exit(1);
333}
334
335int
336getprec(char *str)
337{
338 char *p;
339 char *q;
340
341 for (p = str; *p; p++)
342 if (*p == '.')
343 break;
344 if (!*p)
345 return (0);
346 for (q = ++p; *p; p++)
347 if (!isdigit((unsigned char)*p))
348 break;
349 return (p - q);
350}
351
352void
353getformat(void)
354{
355 char *p, *p2;
356 int dot, hash, space, sign, numbers = 0;
357 size_t sz;
358
359 if (boring) /* no need to bother */
360 return;
361 for (p = format; *p; p++) /* look for '%' */
362 if (*p == '%' && *(p+1) != '%') /* leave %% alone */
363 break;
364 sz = sizeof(format) - strlen(format) - 1;
365 if (!*p && !chardata) {
366 if (snprintf(p, sz, "%%.%df", prec) >= (int)sz)
367 errx(1, "-w word too long");
368 } else if (!*p && chardata) {
369 if (strlcpy(p, "%c", sz) >= sz)
370 errx(1, "-w word too long");
371 intdata = 1;
372 } else if (!*(p+1)) {
373 if (sz <= 0)
374 errx(1, "-w word too long");
375 strcat(format, "%"); /* cannot end in single '%' */
376 } else {
377 /*
378 * Allow conversion format specifiers of the form
379 * %[#][ ][{+,-}][0-9]*[.[0-9]*]? where ? must be one of
380 * [l]{d,i,o,u,x} or {f,e,g,E,G,d,o,x,D,O,U,X,c,u}
381 */
382 p2 = p++;
383 dot = hash = space = sign = numbers = 0;
384 while (!isalpha((unsigned char)*p)) {
385 if (isdigit((unsigned char)*p)) {
386 numbers++;
387 p++;
388 } else if ((*p == '#' && !(numbers|dot|sign|space|
389 hash++)) ||
390 (*p == ' ' && !(numbers|dot|space++)) ||
391 ((*p == '+' || *p == '-') && !(numbers|dot|sign++))
392 || (*p == '.' && !(dot++)))
393 p++;
394 else
395 goto fmt_broken;
396 }
397 if (*p == 'l') {
398 longdata = 1;
399 if (*++p == 'l') {
400 if (p[1] != '\0')
401 p++;
402 goto fmt_broken;
403 }
404 }
405 switch (*p) {
406 case 'o': case 'u': case 'x': case 'X':
407 intdata = nosign = 1;
408 break;
409 case 'd': case 'i':
410 intdata = 1;
411 break;
412 case 'D':
413 if (!longdata) {
414 intdata = 1;
415 break;
416 }
417 case 'O': case 'U':
418 if (!longdata) {
419 intdata = nosign = 1;
420 break;
421 }
422 case 'c':
423 if (!(intdata | longdata)) {
424 chardata = 1;
425 break;
426 }
427 case 'h': case 'n': case 'p': case 'q': case 's': case 'L':
428 case '$': case '*':
429 goto fmt_broken;
430 case 'f': case 'e': case 'g': case 'E': case 'G':
431 if (!longdata)
432 break;
433 /* FALLTHROUGH */
434 default:
435fmt_broken:
436 *++p = '\0';
437 errx(1, "illegal or unsupported format '%s'", p2);
438 /* NOTREACHED */
439 }
440 while (*++p)
441 if (*p == '%' && *(p+1) && *(p+1) != '%')
442 errx(1, "too many conversions");
443 else if (*p == '%' && *(p+1) == '%')
444 p++;
445 else if (*p == '%' && !*(p+1)) {
446 strcat(format, "%");
447 break;
448 }
449 }
450}