mirror of
git://git.suckless.org/sbase
synced 2024-12-25 08:22:44 +00:00
180 lines
4.0 KiB
C
180 lines
4.0 KiB
C
/* See LICENSE file for copyright and license details. */
|
|
#include <errno.h>
|
|
#include <limits.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
|
|
#include "text.h"
|
|
#include "util.h"
|
|
|
|
static void uudecode(FILE *, FILE *);
|
|
static void parseheader(FILE *, const char *, const char *, mode_t *, char **);
|
|
static FILE *parsefile(const char *);
|
|
|
|
static void
|
|
usage(void)
|
|
{
|
|
eprintf("usage: %s [file]\n", argv0);
|
|
}
|
|
|
|
int
|
|
main(int argc, char *argv[])
|
|
{
|
|
FILE *fp = NULL, *nfp = NULL;
|
|
char *fname;
|
|
mode_t mode = 0;
|
|
|
|
ARGBEGIN {
|
|
case 'm':
|
|
eprintf("-m not implemented\n");
|
|
default:
|
|
usage();
|
|
} ARGEND;
|
|
|
|
if (argc > 1)
|
|
usage();
|
|
|
|
if (argc == 0) {
|
|
parseheader(stdin, "<stdin>", "begin ", &mode, &fname);
|
|
if (!(nfp = parsefile(fname)))
|
|
eprintf("fopen %s:", fname);
|
|
uudecode(stdin, nfp);
|
|
} else {
|
|
if (!(fp = fopen(argv[0], "r")))
|
|
eprintf("fopen %s:", argv[0]);
|
|
parseheader(fp, argv[0], "begin ", &mode, &fname);
|
|
if (!(nfp = parsefile(fname)))
|
|
eprintf("fopen %s:", fname);
|
|
uudecode(fp, nfp);
|
|
}
|
|
|
|
if (chmod(fname, mode) < 0)
|
|
eprintf("chmod %s:", fname);
|
|
if (fp)
|
|
fclose(fp);
|
|
if (nfp)
|
|
fclose(nfp);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static FILE *
|
|
parsefile(const char *fname)
|
|
{
|
|
struct stat st;
|
|
int ret;
|
|
|
|
if (strcmp(fname, "/dev/stdout") == 0)
|
|
return stdout;
|
|
ret = lstat(fname, &st);
|
|
/* if it is a new file, try to open it */
|
|
if (ret < 0 && errno == ENOENT)
|
|
goto tropen;
|
|
if (ret < 0) {
|
|
weprintf("lstat %s:", fname);
|
|
return NULL;
|
|
}
|
|
if (!S_ISREG(st.st_mode)) {
|
|
weprintf("for safety uudecode operates only on regular files and /dev/stdout\n");
|
|
return NULL;
|
|
}
|
|
tropen:
|
|
return fopen(fname,"w");
|
|
}
|
|
|
|
static void
|
|
parseheader(FILE *fp, const char *s, const char *header, mode_t *mode, char **fname)
|
|
{
|
|
char bufs[PATH_MAX + 11]; /* len header + mode + maxname */
|
|
char *p, *q;
|
|
size_t n;
|
|
|
|
if (!fgets(bufs, sizeof(bufs), fp))
|
|
if (ferror(fp))
|
|
eprintf("%s: read error:", s);
|
|
if (bufs[0] == '\0' || feof(fp))
|
|
eprintf("empty or nil header string\n");
|
|
if (!(p = strchr(bufs, '\n')))
|
|
eprintf("header string too long or non-newline terminated file\n");
|
|
p = bufs;
|
|
if (strncmp(bufs, header, strlen(header)) != 0)
|
|
eprintf("malformed header prefix\n");
|
|
p += strlen(header);
|
|
if (!(q = strchr(p, ' ')))
|
|
eprintf("malformed mode string in header\n");
|
|
*q++ = '\0';
|
|
/* now mode should be null terminated, q points to fname */
|
|
*mode = parsemode(p, *mode, 0);
|
|
n = strlen(q);
|
|
while (n > 0 && (q[n - 1] == '\n' || q[n - 1] == '\r'))
|
|
q[--n] = '\0';
|
|
if (n > 0)
|
|
*fname = q;
|
|
}
|
|
|
|
static void
|
|
uudecode(FILE *fp, FILE *outfp)
|
|
{
|
|
char *bufb = NULL, *p;
|
|
size_t n = 0;
|
|
ssize_t len;
|
|
int ch, i;
|
|
|
|
#define DEC(c) (((c) - ' ') & 077) /* single character decode */
|
|
#define IS_DEC(c) ( (((c) - ' ') >= 0) && (((c) - ' ') <= 077 + 1) )
|
|
#define OUT_OF_RANGE(c) eprintf("character %c out of range: [%d-%d]", (c), 1 + ' ', 077 + ' ' + 1)
|
|
|
|
while ((len = getline(&bufb, &n, fp)) != -1) {
|
|
p = bufb;
|
|
/* trim newlines */
|
|
if (len && bufb[len - 1] != '\n')
|
|
bufb[len - 1] = '\0';
|
|
else
|
|
eprintf("no newline found, aborting\n");
|
|
/* check for last line */
|
|
if ((i = DEC(*p)) <= 0)
|
|
break;
|
|
for (++p; i > 0; p += 4, i -= 3) {
|
|
if (i >= 3) {
|
|
if (!(IS_DEC(*p) && IS_DEC(*(p + 1)) &&
|
|
IS_DEC(*(p + 2)) && IS_DEC(*(p + 3))))
|
|
OUT_OF_RANGE(*p);
|
|
|
|
ch = DEC(p[0]) << 2 | DEC(p[1]) >> 4;
|
|
putc(ch, outfp);
|
|
ch = DEC(p[1]) << 4 | DEC(p[2]) >> 2;
|
|
putc(ch, outfp);
|
|
ch = DEC(p[2]) << 6 | DEC(p[3]);
|
|
putc(ch, outfp);
|
|
} else {
|
|
if (i >= 1) {
|
|
if (!(IS_DEC(*p) && IS_DEC(*(p + 1))))
|
|
OUT_OF_RANGE(*p);
|
|
|
|
ch = DEC(p[0]) << 2 | DEC(p[1]) >> 4;
|
|
putc(ch, outfp);
|
|
}
|
|
if (i >= 2) {
|
|
if (!(IS_DEC(*(p + 1)) &&
|
|
IS_DEC(*(p + 2))))
|
|
OUT_OF_RANGE(*p);
|
|
|
|
ch = DEC(p[1]) << 4 | DEC(p[2]) >> 2;
|
|
putc(ch, outfp);
|
|
}
|
|
}
|
|
}
|
|
if (ferror(fp))
|
|
eprintf("read error:");
|
|
}
|
|
/* check for end or fail */
|
|
len = getline(&bufb, &n, fp);
|
|
if (len < 3 || strncmp(bufb, "end", 3) != 0 || bufb[3] != '\n')
|
|
eprintf("invalid uudecode footer \"end\" not found\n");
|
|
free(bufb);
|
|
}
|