summaryrefslogtreecommitdiff
path: root/csv.c
blob: f6b37f150f09eaaccd352d1b2df60fbc83b45521 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
// SPDX-License-Identifier: BSD-2-Clause-Patent
/*
 * csv.c - CSV parser
 */

#include "shim.h"

void NONNULL(1, 3, 4)
parse_csv_line(char * line, size_t max, size_t *n_columns, const char *columns[])
{
	char *next = line;
	size_t n = 0, new_n = n;
	const char * const delims = ",";
	char state = 0;
	char *token = NULL;

	bool valid = true;
	for (n = 0; n < *n_columns; n++) {

		if (valid) {
			valid = strntoken(next, max, delims, &token, &state);
		}
		if (valid) {
			next += strlen(token) + 1;
			max -= strlen(token) + 1;
			columns[n] = token;
			new_n = n + 1;
		} else {
			columns[n] = NULL;
			continue;
		}
	}
	*n_columns = new_n;
}

void
free_csv_list(list_t *list)
{
	list_t *pos = NULL, *tmp = NULL;
	list_for_each_safe(pos, tmp, list) {
		struct csv_row *row;

		row = list_entry(pos, struct csv_row, list);
		list_del(&row->list);
		FreePool(row);
	}
}

EFI_STATUS
parse_csv_data(char *data, char *data_end, size_t n_columns, list_t *list)
{
	EFI_STATUS efi_status = EFI_OUT_OF_RESOURCES;
	char delims[] = "\r\n";
	char *line = data;
	size_t max = 0;
	char *end = data_end;

	if (!data || !end || end <= data || !n_columns || !list) {
		dprint(L"data:0x%lx end:0x%lx n_columns:%lu list:0x%lx\n",
		       data, end, n_columns, list);
		return EFI_INVALID_PARAMETER;
	}

	max = (uintptr_t)end - (uintptr_t)line + (end > line ? 1 : 0);

	if (line && is_utf8_bom(line, max))
		line += UTF8_BOM_SIZE;

	while (line && line <= data_end) {
		size_t entrysz = sizeof(char *) * n_columns + sizeof(struct csv_row);
		struct csv_row *entry;
		size_t m_columns = n_columns;
		char *delim;
		bool found = true;

		end = data_end;
		max = (uintptr_t)end - (uintptr_t)line + (end > line ? 1 : 0);
		while (max && found) {
			found = false;
			for (delim = &delims[0]; max && *delim; delim++) {
				if (line[0] == *delim) {
					line++;
					max--;
					found = true;
				}
			}
		}
		for (delim = &delims[0]; *delim; delim++) {
			char *tmp = strnchrnul(line, max, *delim);
			if (tmp < end)
				end = tmp;
		}
		max = (uintptr_t)end - (uintptr_t)line + (end > line ? 1 : 0);
		*end = '\0';

		if (line == data_end || max == 0) {
			line = end + 1;
			continue;
		}

		entry = AllocateZeroPool(entrysz);
		if (!entry) {
			efi_status = EFI_OUT_OF_RESOURCES;
			goto err_oom;
		}

		INIT_LIST_HEAD(&entry->list);
		list_add_tail(&entry->list, list);

		for (delim = &delims[0]; *delim; delim++) {
			char *tmp = strnchrnul((const char *)line, max, *delim);
			if (tmp < end)
				end = tmp;
		}

		parse_csv_line(line, max, &m_columns, (const char **)entry->columns);
		entry->n_columns = m_columns;
		line = end + 1;
	}

	return EFI_SUCCESS;
err_oom:
	free_csv_list(list);
	return efi_status;
}

// vim:fenc=utf-8:tw=75:noet