cdba/device_parser.c

#include <stdio.h>
#include <stdbool.h>
#include <yaml.h>

#include "device.h"
#include "cdb_assist.h"
#include "conmux.h"

struct device_parser {
	yaml_parser_t parser;
	yaml_event_t event;
};

static void nextsym(struct device_parser *dp)
{
	if (!yaml_parser_parse(&dp->parser, &dp->event)) {
		fprintf(stderr, "device parser: error %d\n", dp->parser.error);
		exit(1);
	}
}

static int accept(struct device_parser *dp, int type, char *scalar)
{
	if (dp->event.type == type) {
		if (scalar) {
			strncpy(scalar, (char *)dp->event.data.scalar.value, 79);
			scalar[79] = '\0';
		}

		yaml_event_delete(&dp->event);
		nextsym(dp);
		return true;
	} else {
		return false;
	}
}

static bool expect(struct device_parser *dp, int type, char *scalar)
{
	if (accept(dp, type, scalar)) {
		return true;
	}

	fprintf(stderr, "device parser: expected %d got %d\n", type, dp->event.type);
	exit(1);
}

static void parse_board(struct device_parser *dp)
{
	struct device *dev;
	char value[80];
	char key[80];

	dev = calloc(1, sizeof(*dev));

	while (accept(dp, YAML_SCALAR_EVENT, key)) {
		expect(dp, YAML_SCALAR_EVENT, value);

		if (!strcmp(key, "board")) {
			dev->board = strdup(value);
		} else if (!strcmp(key, "name")) {
			dev->name = strdup(value);
		} else if (!strcmp(key, "cdba")) {
			dev->cdb_serial = strdup(value);

			dev->open = cdb_assist_open;
			dev->power_on = cdb_assist_power_on;
			dev->power_off = cdb_assist_power_off;
			dev->print_status = cdb_assist_print_status;
			dev->vbus = cdb_assist_vbus;
			dev->write = cdb_target_write;
		} else if (!strcmp(key, "conmux")) {
			dev->cdb_serial = strdup(value);

			dev->open = conmux_open;
			dev->power_on = conmux_power_on;
			dev->power_off = conmux_power_off;
			dev->write = conmux_write;
		} else if (!strcmp(key, "voltage")) {
			dev->voltage = strtoul(value, NULL, 10);
		} else if (!strcmp(key, "fastboot")) {
			dev->serial = strdup(value);

			dev->boot = device_fastboot_boot;
		} else if (!strcmp(key, "fastboot_set_active")) {
			dev->set_active = !strcmp(value, "true");
		} else {
			fprintf(stderr, "device parser: unknown key \"%s\"\n", key);
			exit(1);
		}
	}

	if (!dev->board || !dev->serial || !dev->open) {
		fprintf(stderr, "device parser: insufficiently defined device\n");
		exit(1);
	}

	device_add(dev);
}

void device_parser(const char *path)
{
	struct device_parser dp;
	char key[80];
	FILE *fh;

	fh = fopen(path, "r");
	if (!fh) {
		fprintf(stderr, "device parser: unable to open %s\n", path);
		exit(1);
	}

	if(!yaml_parser_initialize(&dp.parser))
		fprintf(stderr, "device parser: failed to initialize parser\n");

	yaml_parser_set_input_file(&dp.parser, fh);

	nextsym(&dp);

	expect(&dp, YAML_STREAM_START_EVENT, NULL);

	expect(&dp, YAML_DOCUMENT_START_EVENT, NULL);
	expect(&dp, YAML_MAPPING_START_EVENT, NULL);

	if (accept(&dp, YAML_SCALAR_EVENT, key)) {
		expect(&dp, YAML_SEQUENCE_START_EVENT, NULL);

		while (accept(&dp, YAML_MAPPING_START_EVENT, NULL)) {
			parse_board(&dp);
			expect(&dp, YAML_MAPPING_END_EVENT, NULL);
		}

		expect(&dp, YAML_SEQUENCE_END_EVENT, NULL);
	}

	expect(&dp, YAML_MAPPING_END_EVENT, NULL);
	expect(&dp, YAML_DOCUMENT_END_EVENT, NULL);
	expect(&dp, YAML_STREAM_END_EVENT, NULL);

	yaml_event_delete(&dp.event);
	yaml_parser_delete(&dp.parser);

	fclose(fh);
}