coreutils

// base64.c, version 1.0.1

// OMKOV coreutils base64

// Copyright (C) 2021, Jakob Wakeling

// MIT Licence

/*

	TODO Improve or replace fgetb64.

*/

#include "util/base64.h"

#include "util/error.h"

#include "util/optget.h"

#include <ctype.h>

#include <stdbool.h>

#include <stdint.h>

#include <stdio.h>

#define VERSION "1.0.1"

static struct lop lops[] = {

	{ "help",    ARG_NUL, 256 },

	{ "version", ARG_NUL, 257 },

	{ NULL, 0, 0 }

};

static bool dflag;

static inline int base64(const char *file);

static void encodeBase64(FILE *fi);

static void decodeBase64(FILE *fi);

static void wwrite(const uint8_t *p, size_t l, size_t c, size_t *cl, FILE *fi);

static inline size_t fgetb64(uint8_t *buf, size_t len, FILE *fi);

static void hlp(void);

static void ver(void);

int main(int ac, char *av[]) { A0 = av[0];

	struct opt opt = OPTGET_INIT; opt.str = "d"; opt.lops = lops;

	for (int o; (o = optget(&opt, av, 1)) != -1;) switch (o) {

	case 'd': { dflag = true; break; }

	case 256: { hlp(); return 0; }

	case 257: { ver(); return 0; }

	default: { return 1; }

	}

	if (opt.ind == ac) { base64(NULL); }

	else if (opt.ind != ac - 1) { error(1, "extra operand"); }

	else if (base64(av[opt.ind])) { error(1, "%s: %s", av[opt.ind], serr()); }

	return 0;

}

/*

	Base64 encode or decode a file.

	If the file path given is null, then use stdin.

*/

static inline int base64(const char *file) {

	FILE *fi;

	if (file == NULL) { fi = stdin; }

	else if (!(fi = fopen(file, "r"))) { return 1; }

	if (dflag) { decodeBase64(fi); } else { encodeBase64(fi); }

	if (fi != stdin) { fclose(fi); } return 0;

}

/* Encode base64 using fixed size buffers */

static void encodeBase64(FILE *fi) {

	uint8_t ibuf[BUFSIZ * 12]; uint8_t obuf[BUFSIZ * 16]; size_t cl = 0;

	for (register size_t c; (c = fread(ibuf, 1, sizeof (ibuf), fi));) {

		c = b64encode(obuf, ibuf, c); wwrite(obuf, c, 76, &cl, stdout);

	} fputc('\n', stdout); return;

}

/* Decode base64 using fixed size buffers */

static void decodeBase64(FILE *fi) {

	uint8_t ibuf[BUFSIZ * 16]; uint8_t obuf[BUFSIZ * 12];

	for (register size_t c; (c = fgetb64(ibuf, sizeof (ibuf), fi));) {

		c = b64decode(obuf, ibuf, c); fwrite(obuf, 1, c, stdout);

	} return;

}

/* Write string with wrapping */

static void wwrite(const uint8_t *p, size_t l, size_t c, size_t *cl, FILE *fi) {

	for (size_t w = 0; w < l;) {

		size_t cr = c - *cl < l - w ? c - *cl : l - w;

		if (!cr) { fputc('\n', fi); *cl = 0; }

		else { fwrite(p + w, 1, cr, fi); w += cr; *cl += cr; }

	}

}

/* Read valid Base64 characters from a file */

static inline size_t fgetb64(uint8_t *buf, size_t len, FILE *fi) {

	size_t i = 0; for (int c; (c = fgetc(fi)) != EOF;) {

		if (isalnum(c) || c == '+' || c == '/') { buf[i++] = c; }

		if (i == len) { break; }

	} return i;

}

/* Print help information */

static void hlp(void) {

	puts("base64 - base64 encode or decode data\n");

	puts("usage: base64 [-d] [file]\n");

	puts("options:");

	puts("  -d         Decode data");

	puts("  --help     Display help information");

	puts("  --version  Display version information");

}

/* Print version information */

static void ver(void) {

	puts("OMKOV coreutils base64, version " VERSION);

	puts("Copyright (C) 2021, Jakob Wakeling");

	puts("MIT Licence (https://opensource.org/licenses/MIT)");

}