add linewrap option when encoding (76 columns)

Makefile

@@ -1,11 +1,12 @@
 CFLAGS = -g -Wall -Wextra -Werror
 
-objects = main.o input.o encode.o
+objects = main.o input.o output.o encode.o
 
 b64 : $(objects)
 	cc -o b64 $(objects)
 
 main.o input.o : input.h
+main.o output.o : output.h
 main.o encode.o : encode.h
 
 .PHONY : clean

main.c

@@ -1,6 +1,7 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include "input.h"
+#include "output.h"
 #include "encode.h"
 
 #define OBUFSIZE 4
@@ -12,11 +13,12 @@ unsigned char sbuf[SBUFSIZE];
 int
 main(int argc, char *argv[])
 {
-    int c, n, dec, url, hlp;
+    int c, n, l;
+    int dec, url, hlp, wrp;
     char *prog = *argv;
     FILE *in, *out;
 
-    dec = url = hlp = 0;
+    dec = url = hlp = wrp = 0;
     while (--argc > 0 && (*++argv)[0] == '-')
         while ((c = *++argv[0]))
             switch (c) {
@@ -29,6 +31,9 @@ main(int argc, char *argv[])
                 case 'h':
                     hlp = 1;
                     break;
+                case 'w':
+                    wrp = 1;
+                    break;
                 default:
                     fprintf(stderr, "%s: illegal option %c\n", prog, c);
                     exit(EXIT_FAILURE);
@@ -52,9 +57,15 @@ main(int argc, char *argv[])
             }
         } else {
             while ((n = getocts((argc >= 1) ? in : stdin, obuf, OBUFSIZE))) {
+                if (wrp) {
+                    n = encode(obuf, n, sbuf, url);
+                    l = printw((argc == 2) ? out : stdout, sbuf, n);
+                } else {
                     encode(obuf, n, sbuf, url);
                     fprintf((argc == 2) ? out : stdout, "%s", sbuf);
                 }
+            }
+            if (!wrp || l != '\n')
                 fprintf((argc == 2) ? out : stdout, "\n");
         }
         if (in)

output.c

@@ -0,0 +1,19 @@
+#include "output.h"
+
+#define WRAPCOL 76
+
+char
+printw(FILE *fp, unsigned char *s, int slen)
+{
+    static int col;
+    int c;
+
+    while (slen--) {
+        fputc((c = *s++), fp);
+        ++col;
+        if (!(col %= WRAPCOL))
+            fputc((c = '\n'), fp);
+    }
+
+    return c;
+}

output.h

@@ -0,0 +1,9 @@
+#ifndef HEADER_OUTPUT
+#define HEADER_OUTPUT
+
+#include <stdio.h>
+
+char
+printw(FILE *fp, unsigned char *s, int slen);
+
+#endif