diff options
Diffstat (limited to 'pw.c')
| -rw-r--r-- | pw.c | 30 |
1 files changed, 25 insertions, 5 deletions
@@ -28,6 +28,7 @@ #include <assert.h> #include <stdio.h> #include <stdlib.h> +#include <stdint.h> #include <ctype.h> #include <string.h> #include <limits.h> @@ -51,6 +52,7 @@ #define DEL 127 #define min(a, b) ((a) < (b) ? (a) : (b)) +#define max(a, b) ((a) > (b) ? (a) : (b)) #ifdef __GNUC__ #define printf_attr(fmtpos, vargpos) __attribute__ ((format (printf, \ @@ -151,8 +153,11 @@ static void dsdrop(char *str) static size_t dslen(const char *str) { - const dstr *ds = dstr_of(str); - return ds->len; + if (str) { + const dstr *ds = dstr_of(str); + return ds->len; + } + return 0; } static char *dsgrow(char *str, size_t len) @@ -691,6 +696,7 @@ int main(int argc, char **argv) unsigned stat = 0; FILE *tty = fopen("/dev/tty", "r+"); int maxlines = 15, nlines = 0, maxed = 0; + size_t maxlen = 2047; int opt; int ifd = fileno(stdin); int ttyfd = tty ? fileno(tty) : -1; @@ -728,7 +734,7 @@ int main(int argc, char **argv) } } - while ((opt = getopt(argc, argv, "n:i:l:dEBg:q:")) != -1) { + while ((opt = getopt(argc, argv, "n:i:l:dEBg:q:m:")) != -1) { switch (opt) { case 'n': { @@ -802,6 +808,20 @@ int main(int argc, char **argv) stat |= stat_grep; } break; + case 'm': + { + char *err; + int val = getzp(optarg, &err); + if (val < 0) { + error("-%c option: bad value %s: %s\n", opt, optarg, err); + return EXIT_FAILURE; + } + + if ((int) (size_t) val != val) + val = -1; + maxlen = max(72, val); + } + break; default: usage(); } @@ -865,7 +885,7 @@ int main(int argc, char **argv) if ((stat & stat_eof) == 0) { int ch; - while ((ch = getc(stdin)) != EOF && ch != '\n') + while ((ch = getc(stdin)) != EOF && ch != '\n' && dslen(line) < maxlen) line = addchesc(line, ch); if (ch == EOF) { if (feof(stdin) || (errno != EAGAIN && errno != EWOULDBLOCK)) { @@ -882,7 +902,7 @@ int main(int argc, char **argv) drawstatus(hist, columns, stat, curcmd); } clearerr(stdin); - } else { + } else if (ch == '\n') { nfds = 1; line = addch(line, 0); if ((stat & stat_grep)) { |