1 // SPDX-License-Identifier: MIT
2 /*
3 * Copyright © 2023 Intel Corporation
4 */
5
6 #define _GNU_SOURCE
7 #include <ctype.h>
8 #include <errno.h>
9 #include <stdbool.h>
10 #include <stdio.h>
11 #include <string.h>
12
13 #define HEADER \
14 "// SPDX-License-Identifier: MIT\n" \
15 "\n" \
16 "/*\n" \
17 " * DO NOT MODIFY.\n" \
18 " *\n" \
19 " * This file was generated from rules: %s\n" \
20 " */\n" \
21 "#ifndef _GENERATED_%s_\n" \
22 "#define _GENERATED_%s_\n" \
23 "\n" \
24 "enum {\n"
25
26 #define FOOTER \
27 "};\n" \
28 "\n" \
29 "#endif\n"
30
print_usage(FILE * f,const char * progname)31 static void print_usage(FILE *f, const char *progname)
32 {
33 fprintf(f, "usage: %s <input-rule-file> <generated-c-source-file> <generated-c-header-file>\n",
34 progname);
35 }
36
print_parse_error(const char * err_msg,const char * line,unsigned int lineno)37 static void print_parse_error(const char *err_msg, const char *line,
38 unsigned int lineno)
39 {
40 fprintf(stderr, "ERROR: %s\nERROR: %u: %.60s\n",
41 err_msg, lineno, line);
42 }
43
strip(char * line,size_t linelen)44 static char *strip(char *line, size_t linelen)
45 {
46 while (isspace(*(line + linelen)))
47 linelen--;
48
49 line[linelen - 1] = '\0';
50
51 return line + strspn(line, " \f\n\r\t\v");
52 }
53
54 #define MAX_LINE_LEN 4096
parse(FILE * input,FILE * csource,FILE * cheader,char * prefix)55 static int parse(FILE *input, FILE *csource, FILE *cheader, char *prefix)
56 {
57 char line[MAX_LINE_LEN + 1];
58 char *name, *prev_name = NULL, *rules;
59 unsigned int lineno = 0, idx = 0;
60
61 while (fgets(line, sizeof(line), input)) {
62 size_t linelen;
63 bool is_continuation;
64
65 if (line[0] == '\0' || line[0] == '#' || line[0] == '\n') {
66 lineno++;
67 continue;
68 }
69
70 linelen = strlen(line);
71 if (linelen == MAX_LINE_LEN) {
72 print_parse_error("line too long", line, lineno);
73 return -EINVAL;
74 }
75
76 is_continuation = isspace(line[0]);
77 name = strip(line, linelen);
78
79 if (!is_continuation) {
80 name = strtok(name, " \t");
81 rules = strtok(NULL, "");
82 } else {
83 if (!prev_name) {
84 print_parse_error("invalid rule continuation",
85 line, lineno);
86 return -EINVAL;
87 }
88
89 rules = name;
90 name = NULL;
91 }
92
93 if (rules[0] == '\0') {
94 print_parse_error("invalid empty rule\n", line, lineno);
95 return -EINVAL;
96 }
97
98 if (name) {
99 fprintf(cheader, "\t%s_%s = %u,\n", prefix, name, idx);
100
101 /* Close previous entry before starting a new one */
102 if (idx)
103 fprintf(csource, ") },\n");
104
105 fprintf(csource, "{ XE_RTP_NAME(\"%s\"),\n XE_RTP_RULES(%s",
106 name, rules);
107 idx++;
108 } else {
109 fprintf(csource, ", OR,\n\t%s", rules);
110 }
111
112 lineno++;
113 if (!is_continuation)
114 prev_name = name;
115 }
116
117 /* Close last entry */
118 if (idx)
119 fprintf(csource, ") },\n");
120
121 fprintf(cheader, "\t_%s_COUNT = %u\n", prefix, idx);
122
123 return 0;
124 }
125
fn_to_prefix(const char * fn,char * prefix,size_t size)126 static int fn_to_prefix(const char *fn, char *prefix, size_t size)
127 {
128 size_t len;
129
130 fn = basename(fn);
131 len = strlen(fn);
132
133 if (len > size - 1)
134 return -ENAMETOOLONG;
135
136 memcpy(prefix, fn, len + 1);
137
138 for (char *p = prefix; *p; p++) {
139 switch (*p) {
140 case '.':
141 *p = '\0';
142 return 0;
143 default:
144 *p = toupper(*p);
145 break;
146 }
147 }
148
149 return 0;
150 }
151
main(int argc,const char * argv[])152 int main(int argc, const char *argv[])
153 {
154 enum {
155 ARGS_INPUT,
156 ARGS_CSOURCE,
157 ARGS_CHEADER,
158 _ARGS_COUNT
159 };
160 struct {
161 const char *fn;
162 const char *mode;
163 FILE *f;
164 } args[] = {
165 [ARGS_INPUT] = { .fn = argv[1], .mode = "r" },
166 [ARGS_CSOURCE] = { .fn = argv[2], .mode = "w" },
167 [ARGS_CHEADER] = { .fn = argv[3], .mode = "w" },
168 };
169 int ret = 1;
170 char prefix[128];
171
172 if (argc < 3) {
173 fprintf(stderr, "ERROR: wrong arguments\n");
174 print_usage(stderr, argv[0]);
175 return 1;
176 }
177
178 if (fn_to_prefix(args[ARGS_CHEADER].fn, prefix, sizeof(prefix)) < 0)
179 return 1;
180
181 for (int i = 0; i < _ARGS_COUNT; i++) {
182 args[i].f = fopen(args[i].fn, args[i].mode);
183 if (!args[i].f) {
184 fprintf(stderr, "ERROR: Can't open %s: %m\n",
185 args[i].fn);
186 goto err;
187 }
188 }
189
190 fprintf(args[ARGS_CHEADER].f, HEADER, args[ARGS_INPUT].fn, prefix, prefix);
191
192 ret = parse(args[ARGS_INPUT].f, args[ARGS_CSOURCE].f,
193 args[ARGS_CHEADER].f, prefix);
194 if (!ret)
195 fprintf(args[ARGS_CHEADER].f, FOOTER);
196
197 err:
198 for (int i = 0; i < _ARGS_COUNT; i++) {
199 if (args[i].f)
200 fclose(args[i].f);
201 }
202
203 return ret;
204 }
205