/*
 * Convert Intel microcode.dat into a single binary microcode.bin file
 *
 * Based on code by Kay Sievers <kay.sievers@vrfy.org>
 * Changed to create a single file by Thomas Bächler <thomas@archlinux.org>
 */


#ifndef _GNU_SOURCE
# define _GNU_SOURCE 1
#endif

#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <limits.h>
#include <stdbool.h>
#include <inttypes.h>
#include <fcntl.h>
#include <errno.h>
#include <sys/stat.h>

struct microcode_header_intel {
	unsigned int hdrver;
	unsigned int rev;
	unsigned int date;
	unsigned int sig;
	unsigned int cksum;
	unsigned int ldrver;
	unsigned int pf;
	unsigned int datasize;
	unsigned int totalsize;
	unsigned int reserved[3];
};

union mcbuf {
	struct microcode_header_intel hdr;
	unsigned int i[0];
	char c[0];
};

int main(int argc, char *argv[])
{
	const char *filename = "/lib/firmware/microcode.dat";
	FILE *f;
	char line[LINE_MAX];
	char buf[4000000];
	union mcbuf *mc;
	size_t bufsize, count, start;
	int rc = EXIT_SUCCESS;

	if (argv[1] != NULL)
		filename = argv[1];

	count = 0;
	mc = (union mcbuf *) buf;
	f = fopen(filename, "re");
	if (f == NULL) {
		printf("open %s: %m\n", filename);
		rc = EXIT_FAILURE;
		goto out;
	}

	while (fgets(line, sizeof(line), f) != NULL) {
		if (sscanf(line, "%x, %x, %x, %x",
		    &mc->i[count],
		    &mc->i[count + 1],
		    &mc->i[count + 2],
		    &mc->i[count + 3]) != 4)
			continue;
		count += 4;
	}
	fclose(f);

	bufsize = count * sizeof(int);
	printf("%s: %lu(%luk) bytes, %zu integers\n",
	       filename,
	       bufsize,
	       bufsize / 1024,
	       count);

	if (bufsize < sizeof(struct microcode_header_intel))
		goto out;

	f = fopen("microcode.bin", "we");
	if (f == NULL) {
		printf("open microcode.bin: %m\n");
		rc = EXIT_FAILURE;
		goto out;
	}

	start = 0;
	for (;;) {
		size_t size;
		unsigned int family, model, stepping, type;
		unsigned int year, month, day;

		mc = (union mcbuf *) &buf[start];

		if (mc->hdr.totalsize)
			size = mc->hdr.totalsize;
		else
			size = 2000 + sizeof(struct microcode_header_intel);

		if (mc->hdr.ldrver != 1 || mc->hdr.hdrver != 1) {
			printf("unknown version/format:\n");
			rc = EXIT_FAILURE;
			break;
		}

		/*
		 *  0- 3 stepping
		 *  4- 7 model
		 *  8-11 family
		 * 12-13 type
		 * 16-19 extended model
		 * 20-27 extended family
		 */
		stepping = mc->hdr.sig & 0x0f;
		model = (mc->hdr.sig >> 4) & 0x0f;
		family = (mc->hdr.sig >> 8) & 0x0f;
		type = (mc->hdr.sig >> 12) & 0x0f;
		if (family == 0x06)
			model += ((mc->hdr.sig >> 16) & 0x0f) << 4;
		if (family == 0x0f)
			family += (mc->hdr.sig >> 20) & 0xff;

		year = mc->hdr.date & 0xffff;
		month = mc->hdr.date >> 24;
		day = (mc->hdr.date >> 16) & 0xff;

		printf("\n");
		printf("signature: 0x%02x (stepping %d, model %d, family %d, type %d)\n",
			mc->hdr.sig, stepping, model, family, type);
		printf("flags:     0x%02x\n", mc->hdr.pf);
		printf("revision:  0x%02x\n", mc->hdr.rev);
		printf("date:      %04x-%02x-%02x\n", year, month, day);
		printf("size:      %zu\n", size);

		if (fwrite(mc, size, 1, f) != 1) {
			printf("write microcode.bin: %m\n");
			rc = EXIT_FAILURE;
			goto out;
		}

		start += size;
		if (start >= bufsize)
			break;
	}
	fclose(f);
	printf("\n");
out:
	return rc;
}