1 /* $OpenBSD: fmt.c,v 1.21 2004/04/01 23:14:19 tedu Exp $ */ 2 3 /* Sensible version of fmt 4 * 5 * Syntax: fmt [ options ] [ goal [ max ] ] [ filename ... ] 6 * 7 * Since the documentation for the original fmt is so poor, here 8 * is an accurate description of what this one does. It's usually 9 * the same. The *mechanism* used may differ from that suggested 10 * here. Note that we are *not* entirely compatible with fmt, 11 * because fmt gets so many things wrong. 12 * 13 * 1. Tabs are expanded, assuming 8-space tab stops. 14 * If the `-t <n>' option is given, we assume <n>-space 15 * tab stops instead. 16 * Trailing blanks are removed from all lines. 17 * x\b == nothing, for any x other than \b. 18 * Other control characters are simply stripped. This 19 * includes \r. 20 * 2. Each line is split into leading whitespace and 21 * everything else. Maximal consecutive sequences of 22 * lines with the same leading whitespace are considered 23 * to form paragraphs, except that a blank line is always 24 * a paragraph to itself. 25 * If the `-p' option is given then the first line of a 26 * paragraph is permitted to have indentation different 27 * from that of the other lines. 28 * If the `-m' option is given then a line that looks 29 * like a mail message header, if it is not immediately 30 * preceded by a non-blank non-message-header line, is 31 * taken to start a new paragraph, which also contains 32 * any subsequent lines with non-empty leading whitespace. 33 * Unless the `-n' option is given, lines beginning with 34 * a . (dot) are not formatted. 35 * 3. The "everything else" is split into words; a word 36 * includes its trailing whitespace, and a word at the 37 * end of a line is deemed to be followed by a single 38 * space, or two spaces if it ends with a sentence-end 39 * character. (See the `-d' option for how to change that.) 40 * If the `-s' option has been given, then a word's trailing 41 * whitespace is replaced by what it would have had if it 42 * had occurred at end of line. 43 * 4. Each paragraph is sent to standard output as follows. 44 * We output the leading whitespace, and then enough words 45 * to make the line length as near as possible to the goal 46 * without exceeding the maximum. (If a single word would 47 * exceed the maximum, we output that anyway.) Of course 48 * the trailing whitespace of the last word is ignored. 49 * We then emit a newline and start again if there are any 50 * words left. 51 * Note that for a blank line this translates as "We emit 52 * a newline". 53 * If the `-l <n>' option is given, then leading whitespace 54 * is modified slightly: <n> spaces are replaced by a tab. 55 * Indented paragraphs (see above under `-p') make matters 56 * more complicated than this suggests. Actually every paragraph 57 * has two `leading whitespace' values; the value for the first 58 * line, and the value for the most recent line. (While processing 59 * the first line, the two are equal. When `-p' has not been 60 * given, they are always equal.) The leading whitespace 61 * actually output is that of the first line (for the first 62 * line of *output*) or that of the most recent line (for 63 * all other lines of output). 64 * When `-m' has been given, message header paragraphs are 65 * taken as having first-leading-whitespace empty and 66 * subsequent-leading-whitespace two spaces. 67 * 68 * Multiple input files are formatted one at a time, so that a file 69 * never ends in the middle of a line. 70 * 71 * There's an alternative mode of operation, invoked by giving 72 * the `-c' option. In that case we just center every line, 73 * and most of the other options are ignored. This should 74 * really be in a separate program, but we must stay compatible 75 * with old `fmt'. 76 * 77 * QUERY: Should `-m' also try to do the right thing with quoted text? 78 * QUERY: `-b' to treat backslashed whitespace as old `fmt' does? 79 * QUERY: Option meaning `never join lines'? 80 * QUERY: Option meaning `split in mid-word to avoid overlong lines'? 81 * (Those last two might not be useful, since we have `fold'.) 82 * 83 * Differences from old `fmt': 84 * 85 * - We have many more options. Options that aren't understood 86 * generate a lengthy usage message, rather than being 87 * treated as filenames. 88 * - Even with `-m', our handling of message headers is 89 * significantly different. (And much better.) 90 * - We don't treat `\ ' as non-word-breaking. 91 * - Downward changes of indentation start new paragraphs 92 * for us, as well as upward. (I think old `fmt' behaves 93 * in the way it does in order to allow indented paragraphs, 94 * but this is a broken way of making indented paragraphs 95 * behave right.) 96 * - Given the choice of going over or under |goal_length| 97 * by the same amount, we go over; old `fmt' goes under. 98 * - We treat `?' as ending a sentence, and not `:'. Old `fmt' 99 * does the reverse. 100 * - We return approved return codes. Old `fmt' returns 101 * 1 for some errors, and *the number of unopenable files* 102 * when that was all that went wrong. 103 * - We have fewer crashes and more helpful error messages. 104 * - We don't turn spaces into tabs at starts of lines unless 105 * specifically requested. 106 * - New `fmt' is somewhat smaller and slightly faster than 107 * old `fmt'. 108 * 109 * Bugs: 110 * 111 * None known. There probably are some, though. 112 * 113 * Portability: 114 * 115 * I believe this code to be pretty portable. It does require 116 * that you have `getopt'. If you need to include "getopt.h" 117 * for this (e.g., if your system didn't come with `getopt' 118 * and you installed it yourself) then you should arrange for 119 * NEED_getopt_h to be #defined. 120 * 121 * Everything here should work OK even on nasty 16-bit 122 * machines and nice 64-bit ones. However, it's only really 123 * been tested on my FreeBSD machine. Your mileage may vary. 124 */ 125 126 /* Copyright (c) 1997 Gareth McCaughan. All rights reserved. 127 * 128 * Redistribution and use of this code, in source or binary forms, 129 * with or without modification, are permitted subject to the following 130 * conditions: 131 * 132 * - Redistribution of source code must retain the above copyright 133 * notice, this list of conditions and the following disclaimer. 134 * 135 * - If you distribute modified source code it must also include 136 * a notice saying that it has been modified, and giving a brief 137 * description of what changes have been made. 138 * 139 * Disclaimer: I am not responsible for the results of using this code. 140 * If it formats your hard disc, sends obscene messages to 141 * your boss and kills your children then that's your problem 142 * not mine. I give absolutely no warranty of any sort as to 143 * what the program will do, and absolutely refuse to be held 144 * liable for any consequences of your using it. 145 * Thank you. Have a nice day. 146 */ 147 148 /* RCS change log: 149 * Revision 1.5 1998/03/02 18:02:21 gjm11 150 * Minor changes for portability. 151 * 152 * Revision 1.4 1997/10/01 11:51:28 gjm11 153 * Repair broken indented-paragraph handling. 154 * Add mail message header stuff. 155 * Improve comments and layout. 156 * Make usable with non-BSD systems. 157 * Add revision display to usage message. 158 * 159 * Revision 1.3 1997/09/30 16:24:47 gjm11 160 * Add copyright notice, rcsid string and log message. 161 * 162 * Revision 1.2 1997/09/30 16:13:39 gjm11 163 * Add options: -d <chars>, -l <width>, -p, -s, -t <width>, -h . 164 * Parse options with `getopt'. Clean up code generally. 165 * Make comments more accurate. 166 * 167 * Revision 1.1 1997/09/30 11:29:57 gjm11 168 * Initial revision 169 */ 170 171 #ifndef lint 172 static const char copyright[] = 173 "Copyright (c) 1997 Gareth McCaughan. All rights reserved.\n"; 174 #endif /* not lint */ 175 #include <sys/cdefs.h> 176 __FBSDID("$FreeBSD$"); 177 178 #include <err.h> 179 #include <limits.h> 180 #include <locale.h> 181 #include <stdio.h> 182 #include <stdlib.h> 183 #include <string.h> 184 #include <sysexits.h> 185 #include <unistd.h> 186 #include <wchar.h> 187 #include <wctype.h> 188 189 /* Something that, we hope, will never be a genuine line length, 190 * indentation etc. 191 */ 192 #define SILLY ((size_t)-1) 193 194 /* I used to use |strtoul| for this, but (1) not all systems have it 195 * and (2) it's probably better to use |strtol| to detect negative 196 * numbers better. 197 * If |fussyp==0| then we don't complain about non-numbers 198 * (returning 0 instead), but we do complain about bad numbers. 199 */ 200 static size_t 201 get_positive(const char *s, const char *err_mess, int fussyP) 202 { 203 char *t; 204 long result = strtol(s, &t, 0); 205 206 if (*t) { 207 if (fussyP) 208 goto Lose; 209 else 210 return 0; 211 } 212 if (result <= 0) { 213 Lose: errx(EX_USAGE, "%s", err_mess); 214 } 215 return (size_t)result; 216 } 217 218 static size_t 219 get_nonnegative(const char *s, const char *err_mess, int fussyP) 220 { 221 char *t; 222 long result = strtol(s, &t, 0); 223 224 if (*t) { 225 if (fussyP) 226 goto Lose; 227 else 228 return 0; 229 } 230 if (result < 0) { 231 Lose: errx(EX_USAGE, "%s", err_mess); 232 } 233 return (size_t)result; 234 } 235 236 /* Global variables */ 237 238 static int centerP = 0; /* Try to center lines? */ 239 static size_t goal_length = 0; /* Target length for output lines */ 240 static size_t max_length = 0; /* Maximum length for output lines */ 241 static int coalesce_spaces_P = 0; /* Coalesce multiple whitespace -> ' ' ? */ 242 static int allow_indented_paragraphs = 0; /* Can first line have diff. ind.? */ 243 static int tab_width = 8; /* Number of spaces per tab stop */ 244 static size_t output_tab_width = 8; /* Ditto, when squashing leading spaces */ 245 static const wchar_t *sentence_enders = L".?!"; /* Double-space after these */ 246 static int grok_mail_headers = 0; /* treat embedded mail headers magically? */ 247 static int format_troff = 0; /* Format troff? */ 248 249 static int n_errors = 0; /* Number of failed files. Return on exit. */ 250 static wchar_t *output_buffer = NULL; /* Output line will be built here */ 251 static size_t x; /* Horizontal position in output line */ 252 static size_t x0; /* Ditto, ignoring leading whitespace */ 253 static size_t output_buffer_length = 0; 254 static size_t pending_spaces; /* Spaces to add before next word */ 255 static int output_in_paragraph = 0; /* Any of current para written out yet? */ 256 257 /* Prototypes */ 258 259 static void process_named_file(const char *); 260 static void process_stream(FILE *, const char *); 261 static size_t indent_length(const wchar_t *, size_t); 262 static int might_be_header(const wchar_t *); 263 static void new_paragraph(size_t, size_t); 264 static void output_word(size_t, size_t, const wchar_t *, size_t, size_t); 265 static void output_indent(size_t); 266 static void center_stream(FILE *, const char *); 267 static wchar_t *get_line(FILE *, size_t *); 268 static void *xrealloc(void *, size_t); 269 270 #define XMALLOC(x) xrealloc(0,x) 271 272 /* Here is perhaps the right place to mention that this code is 273 * all in top-down order. Hence, |main| comes first. 274 */ 275 int 276 main(int argc, char *argv[]) 277 { 278 int ch; /* used for |getopt| processing */ 279 wchar_t *tmp; 280 size_t len; 281 const char *src; 282 283 (void)setlocale(LC_CTYPE, ""); 284 285 /* 1. Grok parameters. */ 286 287 while ((ch = getopt(argc, argv, "0123456789cd:hl:mnpst:w:")) != -1) 288 switch (ch) { 289 case 'c': 290 centerP = 1; 291 format_troff = 1; 292 continue; 293 case 'd': 294 src = optarg; 295 len = mbsrtowcs(NULL, &src, 0, NULL); 296 if (len == (size_t)-1) 297 err(EX_USAGE, "bad sentence-ending character set"); 298 tmp = XMALLOC((len + 1) * sizeof(wchar_t)); 299 mbsrtowcs(tmp, &src, len + 1, NULL); 300 sentence_enders = tmp; 301 continue; 302 case 'l': 303 output_tab_width 304 = get_nonnegative(optarg, "output tab width must be non-negative", 1); 305 continue; 306 case 'm': 307 grok_mail_headers = 1; 308 continue; 309 case 'n': 310 format_troff = 1; 311 continue; 312 case 'p': 313 allow_indented_paragraphs = 1; 314 continue; 315 case 's': 316 coalesce_spaces_P = 1; 317 continue; 318 case 't': 319 tab_width = get_positive(optarg, "tab width must be positive", 1); 320 continue; 321 case 'w': 322 goal_length = get_positive(optarg, "width must be positive", 1); 323 max_length = goal_length; 324 continue; 325 case '0': case '1': case '2': case '3': case '4': case '5': 326 case '6': case '7': case '8': case '9': 327 /* 328 * XXX this is not a stylistically approved use of 329 * getopt() 330 */ 331 if (goal_length == 0) { 332 char *p; 333 334 p = argv[optind - 1]; 335 if (p[0] == '-' && p[1] == ch && !p[2]) 336 goal_length = get_positive(++p, "width must be nonzero", 1); 337 else 338 goal_length = get_positive(argv[optind] + 1, 339 "width must be nonzero", 1); 340 max_length = goal_length; 341 } 342 continue; 343 case 'h': 344 default: 345 fprintf(stderr, 346 "usage: fmt [-cmps] [-d chars] [-l num] [-t num]\n" 347 " [-w width | -width | goal [maximum]] [file ...]\n" 348 "Options: -c center each line instead of formatting\n" 349 " -d <chars> double-space after <chars> at line end\n" 350 " -l <n> turn each <n> spaces at start of line into a tab\n" 351 " -m try to make sure mail header lines stay separate\n" 352 " -n format lines beginning with a dot\n" 353 " -p allow indented paragraphs\n" 354 " -s coalesce whitespace inside lines\n" 355 " -t <n> have tabs every <n> columns\n" 356 " -w <n> set maximum width to <n>\n" 357 " goal set target width to goal\n"); 358 exit(ch == 'h' ? 0 : EX_USAGE); 359 } 360 argc -= optind; 361 argv += optind; 362 363 /* [ goal [ maximum ] ] */ 364 365 if (argc > 0 && goal_length == 0 366 && (goal_length = get_positive(*argv, "goal length must be positive", 0)) 367 != 0) { 368 --argc; 369 ++argv; 370 if (argc > 0 371 && (max_length = get_positive(*argv, "max length must be positive", 0)) 372 != 0) { 373 --argc; 374 ++argv; 375 if (max_length < goal_length) 376 errx(EX_USAGE, "max length must be >= goal length"); 377 } 378 } 379 if (goal_length == 0) 380 goal_length = 65; 381 if (max_length == 0) 382 max_length = goal_length + 10; 383 if (max_length >= SIZE_T_MAX / sizeof(wchar_t)) 384 errx(EX_USAGE, "max length too large"); 385 /* really needn't be longer */ 386 output_buffer = XMALLOC((max_length + 1) * sizeof(wchar_t)); 387 388 /* 2. Process files. */ 389 390 if (argc > 0) { 391 while (argc-- > 0) 392 process_named_file(*argv++); 393 } else { 394 process_stream(stdin, "standard input"); 395 } 396 397 /* We're done. */ 398 399 return n_errors ? EX_NOINPUT : 0; 400 401 } 402 403 /* Process a single file, given its name. 404 */ 405 static void 406 process_named_file(const char *name) 407 { 408 FILE *f = fopen(name, "r"); 409 410 if (!f) { 411 warn("%s", name); 412 ++n_errors; 413 } else { 414 process_stream(f, name); 415 if (ferror(f)) { 416 warn("%s", name); 417 ++n_errors; 418 } 419 fclose(f); 420 } 421 } 422 423 /* Types of mail header continuation lines: 424 */ 425 typedef enum { 426 hdr_ParagraphStart = -1, 427 hdr_NonHeader = 0, 428 hdr_Header = 1, 429 hdr_Continuation = 2 430 } HdrType; 431 432 /* Process a stream. This is where the real work happens, 433 * except that centering is handled separately. 434 */ 435 static void 436 process_stream(FILE *stream, const char *name) 437 { 438 size_t last_indent = SILLY; /* how many spaces in last indent? */ 439 size_t para_line_number = 0; /* how many lines already read in this para? */ 440 size_t first_indent = SILLY; /* indentation of line 0 of paragraph */ 441 HdrType prev_header_type = hdr_ParagraphStart; 442 443 /* ^-- header_type of previous line; -1 at para start */ 444 wchar_t *line; 445 size_t length; 446 447 if (centerP) { 448 center_stream(stream, name); 449 return; 450 } 451 while ((line = get_line(stream, &length)) != NULL) { 452 size_t np = indent_length(line, length); 453 454 { 455 HdrType header_type = hdr_NonHeader; 456 457 if (grok_mail_headers && prev_header_type != hdr_NonHeader) { 458 if (np == 0 && might_be_header(line)) 459 header_type = hdr_Header; 460 else if (np > 0 && prev_header_type > hdr_NonHeader) 461 header_type = hdr_Continuation; 462 } 463 /* 464 * We need a new paragraph if and only if: this line 465 * is blank, OR it's a troff request (and we don't 466 * format troff), OR it's a mail header, OR it's not 467 * a mail header AND the last line was one, OR the 468 * indentation has changed AND the line isn't a mail 469 * header continuation line AND this isn't the 470 * second line of an indented paragraph. 471 */ 472 if (length == 0 473 || (line[0] == '.' && !format_troff) 474 || header_type == hdr_Header 475 || (header_type == hdr_NonHeader && prev_header_type > hdr_NonHeader) 476 || (np != last_indent 477 && header_type != hdr_Continuation 478 && (!allow_indented_paragraphs || para_line_number != 1))) { 479 new_paragraph(output_in_paragraph ? last_indent : first_indent, np); 480 para_line_number = 0; 481 first_indent = np; 482 last_indent = np; 483 if (header_type == hdr_Header) 484 last_indent = 2; /* for cont. lines */ 485 if (length == 0 || (line[0] == '.' && !format_troff)) { 486 if (length == 0) 487 putwchar('\n'); 488 else 489 wprintf(L"%.*ls\n", (int)length, 490 line); 491 prev_header_type = hdr_ParagraphStart; 492 continue; 493 } 494 } else { 495 /* 496 * If this is an indented paragraph other 497 * than a mail header continuation, set 498 * |last_indent|. 499 */ 500 if (np != last_indent && 501 header_type != hdr_Continuation) 502 last_indent = np; 503 } 504 prev_header_type = header_type; 505 } 506 507 { 508 size_t n = np; 509 510 while (n < length) { 511 /* Find word end and count spaces after it */ 512 size_t word_length = 0, space_length = 0; 513 514 while (n + word_length < length && 515 line[n + word_length] != ' ') 516 ++word_length; 517 space_length = word_length; 518 while (n + space_length < length && 519 line[n + space_length] == ' ') 520 ++space_length; 521 /* Send the word to the output machinery. */ 522 output_word(first_indent, last_indent, 523 line + n, word_length, 524 space_length - word_length); 525 n += space_length; 526 } 527 } 528 ++para_line_number; 529 } 530 new_paragraph(output_in_paragraph ? last_indent : first_indent, 0); 531 if (ferror(stream)) { 532 warn("%s", name); 533 ++n_errors; 534 } 535 } 536 537 /* How long is the indent on this line? 538 */ 539 static size_t 540 indent_length(const wchar_t *line, size_t length) 541 { 542 size_t n = 0; 543 544 while (n < length && *line++ == ' ') 545 ++n; 546 return n; 547 } 548 549 /* Might this line be a mail header? 550 * We deem a line to be a possible header if it matches the 551 * Perl regexp /^[A-Z][-A-Za-z0-9]*:\s/. This is *not* the same 552 * as in RFC whatever-number-it-is; we want to be gratuitously 553 * conservative to avoid mangling ordinary civilised text. 554 */ 555 static int 556 might_be_header(const wchar_t *line) 557 { 558 if (!iswupper(*line++)) 559 return 0; 560 while (*line && (iswalnum(*line) || *line == '-')) 561 ++line; 562 return (*line == ':' && iswspace(line[1])); 563 } 564 565 /* Begin a new paragraph with an indent of |indent| spaces. 566 */ 567 static void 568 new_paragraph(size_t old_indent, size_t indent) 569 { 570 if (output_buffer_length) { 571 if (old_indent > 0) 572 output_indent(old_indent); 573 wprintf(L"%.*ls\n", (int)output_buffer_length, output_buffer); 574 } 575 x = indent; 576 x0 = 0; 577 output_buffer_length = 0; 578 pending_spaces = 0; 579 output_in_paragraph = 0; 580 } 581 582 /* Output spaces or tabs for leading indentation. 583 */ 584 static void 585 output_indent(size_t n_spaces) 586 { 587 if (output_tab_width) { 588 while (n_spaces >= output_tab_width) { 589 putwchar('\t'); 590 n_spaces -= output_tab_width; 591 } 592 } 593 while (n_spaces-- > 0) 594 putwchar(' '); 595 } 596 597 /* Output a single word, or add it to the buffer. 598 * indent0 and indent1 are the indents to use on the first and subsequent 599 * lines of a paragraph. They'll often be the same, of course. 600 */ 601 static void 602 output_word(size_t indent0, size_t indent1, const wchar_t *word, size_t length, size_t spaces) 603 { 604 size_t new_x; 605 size_t indent = output_in_paragraph ? indent1 : indent0; 606 size_t width; 607 const wchar_t *p; 608 int cwidth; 609 610 for (p = word, width = 0; p < &word[length]; p++) 611 width += (cwidth = wcwidth(*p)) > 0 ? cwidth : 1; 612 613 new_x = x + pending_spaces + width; 614 615 /* 616 * If either |spaces==0| (at end of line) or |coalesce_spaces_P| 617 * (squashing internal whitespace), then add just one space; except 618 * that if the last character was a sentence-ender we actually add 619 * two spaces. 620 */ 621 if (coalesce_spaces_P || spaces == 0) 622 spaces = wcschr(sentence_enders, word[length - 1]) ? 2 : 1; 623 624 if (new_x <= goal_length) { 625 /* 626 * After adding the word we still aren't at the goal length, 627 * so clearly we add it to the buffer rather than outputing 628 * it. 629 */ 630 wmemset(output_buffer + output_buffer_length, L' ', 631 pending_spaces); 632 x0 += pending_spaces; 633 x += pending_spaces; 634 output_buffer_length += pending_spaces; 635 wmemcpy(output_buffer + output_buffer_length, word, length); 636 x0 += width; 637 x += width; 638 output_buffer_length += length; 639 pending_spaces = spaces; 640 } else { 641 /* 642 * Adding the word takes us past the goal. Print the 643 * line-so-far, and the word too iff either (1) the lsf is 644 * empty or (2) that makes us nearer the goal but doesn't 645 * take us over the limit, or (3) the word on its own takes 646 * us over the limit. In case (3) we put a newline in 647 * between. 648 */ 649 if (indent > 0) 650 output_indent(indent); 651 wprintf(L"%.*ls", (int)output_buffer_length, output_buffer); 652 if (x0 == 0 || (new_x <= max_length && 653 new_x - goal_length <= goal_length - x)) { 654 wprintf(L"%*ls", (int)pending_spaces, L""); 655 goto write_out_word; 656 } else { 657 /* 658 * If the word takes us over the limit on its own, 659 * just spit it out and don't bother buffering it. 660 */ 661 if (indent + width > max_length) { 662 putwchar('\n'); 663 if (indent > 0) 664 output_indent(indent); 665 write_out_word: 666 wprintf(L"%.*ls", (int)length, word); 667 x0 = 0; 668 x = indent1; 669 pending_spaces = 0; 670 output_buffer_length = 0; 671 } else { 672 wmemcpy(output_buffer, word, length); 673 x0 = width; 674 x = width + indent1; 675 pending_spaces = spaces; 676 output_buffer_length = length; 677 } 678 } 679 putwchar('\n'); 680 output_in_paragraph = 1; 681 } 682 } 683 684 /* Process a stream, but just center its lines rather than trying to 685 * format them neatly. 686 */ 687 static void 688 center_stream(FILE *stream, const char *name) 689 { 690 wchar_t *line, *p; 691 size_t length; 692 size_t width; 693 int cwidth; 694 695 while ((line = get_line(stream, &length)) != NULL) { 696 size_t l = length; 697 698 while (l > 0 && iswspace(*line)) { 699 ++line; 700 --l; 701 } 702 length = l; 703 for (p = line, width = 0; p < &line[length]; p++) 704 width += (cwidth = wcwidth(*p)) > 0 ? cwidth : 1; 705 l = width; 706 while (l < goal_length) { 707 putwchar(' '); 708 l += 2; 709 } 710 wprintf(L"%.*ls\n", (int)length, line); 711 } 712 if (ferror(stream)) { 713 warn("%s", name); 714 ++n_errors; 715 } 716 } 717 718 /* Get a single line from a stream. Expand tabs, strip control 719 * characters and trailing whitespace, and handle backspaces. 720 * Return the address of the buffer containing the line, and 721 * put the length of the line in |lengthp|. 722 * This can cope with arbitrarily long lines, and with lines 723 * without terminating \n. 724 * If there are no characters left or an error happens, we 725 * return 0. 726 * Don't confuse |spaces_pending| here with the global 727 * |pending_spaces|. 728 */ 729 static wchar_t * 730 get_line(FILE *stream, size_t *lengthp) 731 { 732 static wchar_t *buf = NULL; 733 static size_t length = 0; 734 size_t len = 0; 735 wint_t ch; 736 size_t spaces_pending = 0; 737 int troff = 0; 738 size_t col = 0; 739 int cwidth; 740 741 if (buf == NULL) { 742 length = 100; 743 buf = XMALLOC(length * sizeof(wchar_t)); 744 } 745 while ((ch = getwc(stream)) != '\n' && ch != WEOF) { 746 if (len + spaces_pending == 0 && ch == '.' && !format_troff) 747 troff = 1; 748 if (ch == ' ') 749 ++spaces_pending; 750 else if (troff || iswprint(ch)) { 751 while (len + spaces_pending >= length) { 752 length *= 2; 753 buf = xrealloc(buf, length * sizeof(wchar_t)); 754 } 755 while (spaces_pending > 0) { 756 --spaces_pending; 757 buf[len++] = ' '; 758 col++; 759 } 760 buf[len++] = ch; 761 col += (cwidth = wcwidth(ch)) > 0 ? cwidth : 1; 762 } else if (ch == '\t') 763 spaces_pending += tab_width - 764 (col + spaces_pending) % tab_width; 765 else if (ch == '\b') { 766 if (len) 767 --len; 768 if (col) 769 --col; 770 } 771 } 772 *lengthp = len; 773 return (len > 0 || ch != WEOF) ? buf : 0; 774 } 775 776 /* (Re)allocate some memory, exiting with an error if we can't. 777 */ 778 static void * 779 xrealloc(void *ptr, size_t nbytes) 780 { 781 void *p = realloc(ptr, nbytes); 782 783 if (p == NULL) 784 errx(EX_OSERR, "out of memory"); 785 return p; 786 } 787