uuencode: add support for base64 and -o to stdout

This commit is contained in:
Tai Chi Minh Ralph Eastwood
2015-02-10 19:14:27 +00:00
committed by sin
parent 93d0178852
commit ec02816d3e
2 changed files with 175 additions and 23 deletions

View File

@@ -4,15 +4,19 @@
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include <assert.h>
#include "util.h"
static int mflag = 0;
static void uuencodeb64(FILE *, const char *, const char *);
static void uuencode(FILE *, const char *, const char *);
static void
usage(void)
{
eprintf("usage: %s [file] name\n", argv0);
eprintf("usage: %s [-m] [file] name\n", argv0);
}
int
@@ -22,7 +26,8 @@ main(int argc, char *argv[])
ARGBEGIN {
case 'm':
eprintf("-m not implemented\n");
mflag = 1;
break;
default:
usage();
} ARGEND;
@@ -31,16 +36,69 @@ main(int argc, char *argv[])
usage();
if (argc == 1) {
uuencode(stdin, argv[0], "<stdin>");
if (mflag)
uuencodeb64(stdin, argv[0], "<stdin>");
else
uuencode(stdin, argv[0], "<stdin>");
} else {
if (!(fp = fopen(argv[0], "r")))
eprintf("fopen %s:", argv[0]);
uuencode(fp, argv[1], argv[0]);
if (mflag)
uuencodeb64(fp, argv[1], argv[0]);
else
uuencode(fp, argv[1], argv[0]);
fclose(fp);
}
return 0;
}
static const char b64et[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
/* much faster */
static unsigned int
b64e(unsigned char b[2])
{
unsigned int o = 0;
unsigned int p = b[2] | (b[1] << 8) | (b[0] << 16);
o = b64et[p & 0x3f]; p >>= 6;
o = (o << 8) | b64et[p & 0x3f]; p >>= 6;
o = (o << 8) | b64et[p & 0x3f]; p >>= 6;
o = (o << 8) | b64et[p & 0x3f];
return o;
}
static void
uuencodeb64(FILE *fp, const char *name, const char *s)
{
struct stat st;
ssize_t n, m = 0;
unsigned char buf[45], *pb;
unsigned int out[sizeof(buf)/3 + 1], *po;
if (fstat(fileno(fp), &st) < 0)
eprintf("fstat %s:", s);
printf("begin-base64 %o %s\n", st.st_mode & 0777, name);
/* write line by line */
while ((n = fread(buf, 1, sizeof(buf), fp))) {
/* clear old buffer if converting with non-multiple of 3 */
if (n != sizeof(buf) && (m = n % 3) != 0) {
buf[n] = '\0'; /* m == 2 */
if (m == 1) buf[n+1] = '\0'; /* m == 1 */
}
for (pb = buf, po = out; pb < buf + n; pb += 3)
*po++ = b64e(pb);
if (m != 0) {
unsigned int mask = 0xffffffff, dest = 0x3d3d3d3d;
/* m==2 -> 0x00ffffff; m==1 -> 0x0000ffff */
mask >>= ((3-m) << 3);
po[-1] = (po[-1] & mask) | (dest & ~mask);
}
*po++ = '\n';
fwrite(out, 1, (po - out) * sizeof(unsigned int) - 3, stdout);
}
if (ferror(fp))
eprintf("'%s' read error:", s);
puts("====");
}
static void
uuencode(FILE *fp, const char *name, const char *s)
{