int yasm_lookup_label(const struct yasm *y, uint64_t *addr,
const char *labelname)
{
if (bug_on(!y))
return -err_internal;
return l_lookup(y->l, addr, labelname);
}
static int lookup_section_label(struct label *l, const char *name,
const char *attribute, uint64_t *value)
{
char label[255];
int errcode;
errcode = create_section_label_name(label, sizeof(label), name,
attribute);
if (errcode < 0)
return errcode;
return l_lookup(l, value, label);
}
/* Processes the current directive.
* If the encoder returns an error, a message including current file and
* line number together with the pt error string is printed on stderr.
*
* Returns 0 on success; a negative enum errcode otherwise.
* Returns -err_internal if @p or @e is the NULL pointer.
* Returns -err_parse_missing_directive if there was a pt directive marker,
* but no directive.
* Returns -stop_process if the .exp directive was encountered.
* Returns -err_pt_lib if the pt encoder returned an error.
* Returns -err_parse if a general parsing error was encountered.
* Returns -err_parse_unknown_directive if there was an unknown pt directive.
*/
static int p_process(struct parser *p, struct pt_encoder *e)
{
int bytes_written;
int errcode;
char *directive, *payload, *pt_label_name, *tmp;
struct pt_directive *pd;
struct pt_packet packet;
if (bug_on(!p))
return -err_internal;
if (bug_on(!e))
return -err_internal;
pd = p->pd;
if (!pd)
return -err_internal;
directive = pd->name;
payload = pd->payload;
pt_label_name = NULL;
bytes_written = 0;
errcode = 0;
/* find a label name. */
tmp = strchr(directive, ':');
if (tmp) {
uint64_t x;
pt_label_name = directive;
directive = tmp+1;
*tmp = '\0';
/* ignore whitespace between label and directive. */
while (isspace(*directive))
directive += 1;
/* if we can lookup a yasm label with the same name, the
* current pt directive label is invalid. */
errcode = yasm_lookup_label(p->y, &x, pt_label_name);
if (errcode == 0)
errcode = -err_label_not_unique;
if (errcode != -err_no_label)
return yasm_print_err(p->y, "label lookup",
errcode);
/* if we can lookup a pt directive label with the same
* name, the current pt directive label is invalid. */
errcode = l_lookup(p->pt_labels, &x, pt_label_name);
if (errcode == 0)
errcode = -err_label_not_unique;
if (errcode != -err_no_label)
return yasm_print_err(p->y, "label lookup",
-err_label_not_unique);
}
/* now try to match the directive string and call the
* corresponding function that parses the payload and emits an
* according packet.
*/
if (strcmp(directive, "") == 0)
return yasm_print_err(p->y, "invalid syntax",
-err_parse_missing_directive);
else if (strcmp(directive, ".exp") == 0) {
/* this is the end of processing pt directives, so we
* add a p_last label to the pt directive labels.
*/
errcode = l_append(p->pt_labels, "eos", p->pt_bytes_written);
if (errcode < 0)
return yasm_print_err(p->y, "append label", errcode);
return -stop_process;
}
if (strcmp(directive, "psb") == 0) {
errcode = parse_empty(payload);
if (errcode < 0) {
yasm_print_err(p->y, "psb: parsing failed", errcode);
goto error;
}
packet.type = ppt_psb;
} else if (strcmp(directive, "psbend") == 0) {
errcode = parse_empty(payload);
if (errcode < 0) {
yasm_print_err(p->y, "psbend: parsing failed", errcode);
goto error;
}
packet.type = ppt_psbend;
} else if (strcmp(directive, "pad") == 0) {
errcode = parse_empty(payload);
if (errcode < 0) {
yasm_print_err(p->y, "pad: parsing failed", errcode);
goto error;
}
packet.type = ppt_pad;
} else if (strcmp(directive, "ovf") == 0) {
errcode = parse_empty(payload);
if (errcode < 0) {
yasm_print_err(p->y, "ovf: parsing failed", errcode);
goto error;
}
packet.type = ppt_ovf;
} else if (strcmp(directive, "tnt") == 0) {
errcode = parse_tnt(&packet.payload.tnt.payload,
&packet.payload.tnt.bit_size, payload);
if (errcode < 0) {
yasm_print_err(p->y, "tnt: parsing failed", errcode);
goto error;
}
packet.type = ppt_tnt_8;
} else if (strcmp(directive, "tnt64") == 0) {
errcode = parse_tnt(&packet.payload.tnt.payload,
&packet.payload.tnt.bit_size, payload);
if (errcode < 0) {
yasm_print_err(p->y, "tnt64: parsing failed", errcode);
goto error;
}
packet.type = ppt_tnt_64;
} else if (strcmp(directive, "tip") == 0) {
errcode = parse_ip(p, &packet.payload.ip.ip,
&packet.payload.ip.ipc, payload);
if (errcode < 0) {
yasm_print_err(p->y, "tip: parsing failed", errcode);
goto error;
}
packet.type = ppt_tip;
} else if (strcmp(directive, "tip.pge") == 0) {
errcode = parse_ip(p, &packet.payload.ip.ip,
&packet.payload.ip.ipc, payload);
if (errcode < 0) {
yasm_print_err(p->y, "tip.pge: parsing failed",
errcode);
goto error;
}
packet.type = ppt_tip_pge;
} else if (strcmp(directive, "tip.pgd") == 0) {
errcode = parse_ip(p, &packet.payload.ip.ip,
&packet.payload.ip.ipc, payload);
if (errcode < 0) {
yasm_print_err(p->y, "tip.pgd: parsing failed",
errcode);
goto error;
}
packet.type = ppt_tip_pgd;
} else if (strcmp(directive, "fup") == 0) {
errcode = parse_ip(p, &packet.payload.ip.ip,
&packet.payload.ip.ipc, payload);
if (errcode < 0) {
yasm_print_err(p->y, "fup: parsing failed", errcode);
goto error;
}
packet.type = ppt_fup;
} else if (strcmp(directive, "mode.exec") == 0) {
if (strcmp(payload, "16bit") == 0) {
packet.payload.mode.bits.exec.csl = 0;
packet.payload.mode.bits.exec.csd = 0;
} else if (strcmp(payload, "64bit") == 0) {
packet.payload.mode.bits.exec.csl = 1;
packet.payload.mode.bits.exec.csd = 0;
} else if (strcmp(payload, "32bit") == 0) {
packet.payload.mode.bits.exec.csl = 0;
packet.payload.mode.bits.exec.csd = 1;
} else {
errcode = yasm_print_err(p->y,
"mode.exec: argument must be one of \"16bit\", \"64bit\" or \"32bit\"",
-err_parse);
goto error;
}
packet.payload.mode.leaf = pt_mol_exec;
packet.type = ppt_mode;
} else if (strcmp(directive, "mode.tsx") == 0) {
if (strcmp(payload, "begin") == 0) {
packet.payload.mode.bits.tsx.intx = 1;
packet.payload.mode.bits.tsx.abrt = 0;
} else if (strcmp(payload, "abort") == 0) {
packet.payload.mode.bits.tsx.intx = 0;
packet.payload.mode.bits.tsx.abrt = 1;
} else if (strcmp(payload, "commit") == 0) {
packet.payload.mode.bits.tsx.intx = 0;
packet.payload.mode.bits.tsx.abrt = 0;
} else {
errcode = yasm_print_err(p->y,
"mode.tsx: argument must be one of \"begin\", \"abort\" or \"commit\"",
-err_parse);
goto error;
}
packet.payload.mode.leaf = pt_mol_tsx;
packet.type = ppt_mode;
} else if (strcmp(directive, "pip") == 0) {
errcode = parse_uint64(&packet.payload.pip.cr3, payload);
if (errcode < 0) {
yasm_print_err(p->y, "pip: parsing failed", errcode);
goto error;
}
packet.type = ppt_pip;
} else if (strcmp(directive, "tsc") == 0) {
errcode = parse_uint64(&packet.payload.tsc.tsc, payload);
if (errcode < 0) {
yasm_print_err(p->y, "tsc: parsing failed", errcode);
goto error;
}
packet.type = ppt_tsc;
} else if (strcmp(directive, "cbr") == 0) {
errcode = parse_uint8(&packet.payload.cbr.ratio, payload);
if (errcode < 0) {
yasm_print_err(p->y, "cbr: parsing cbr failed",
errcode);
goto error;
}
packet.type = ppt_cbr;
} else {
errcode = yasm_print_err(p->y, "invalid syntax",
-err_parse_unknown_directive);
goto error;
}
bytes_written = pt_enc_next(e, &packet);
if (bytes_written < 0) {
const char *errstr, *format;
char *msg;
size_t n;
errstr = pt_errstr(pt_errcode(bytes_written));
format = "encoder error in directive %s (status %s)";
/* the length of format includes the "%s" (-2)
* characters, we add errstr (+-0) and then we need
* space for a terminating null-byte (+1).
*/
n = strlen(format)-4 + strlen(directive) + strlen(errstr) + 1;
msg = malloc(n);
if (!msg)
errcode = yasm_print_err(p->y,
"encoder error not enough memory to show error code",
-err_pt_lib);
else {
sprintf(msg, format, directive, errstr);
errcode = yasm_print_err(p->y, msg, -err_pt_lib);
free(msg);
}
} else {
if (pt_label_name) {
errcode = l_append(p->pt_labels, pt_label_name,
p->pt_bytes_written);
if (errcode < 0)
goto error;
}
p->pt_bytes_written += bytes_written;
}
error:
if (errcode < 0)
bytes_written = errcode;
return bytes_written;
}
/* Generates the content of the .exp file by printing all lines with
* everything up to and including the first comment semicolon removed.
*
* Returns 0 on success; a negative enum errcode otherwise.
* Returns -err_internal if @p is the NULL pointer.
* Returns -err_file_write if the .exp file could not be fully written.
*/
static int p_gen_expfile(struct parser *p)
{
int errcode;
enum { slen = 1024 };
char s[slen];
struct pt_directive *pd;
char *filename;
FILE *f;
if (bug_on(!p))
return -err_internal;
pd = p->pd;
/* the directive in the current line must be the .exp directive. */
errcode = yasm_pd_parse(p->y, pd);
if (bug_on(errcode < 0))
return -err_internal;
if (bug_on(strcmp(pd->name, ".exp") != 0))
return -err_internal;
filename = expfilename(p, pd->payload);
if (!filename)
return -err_no_mem;
f = fopen(filename, "w");
if (!f) {
free(filename);
return -err_file_open;
}
for (;;) {
int i;
char *line, *comment;
errcode = yasm_next_line(p->y, s, slen);
if (errcode < 0)
break;
errcode = yasm_pd_parse(p->y, pd);
if (errcode < 0 && errcode != -err_no_directive)
break;
if (errcode == 0 && strcmp(pd->name, ".exp") == 0) {
fclose(f);
printf("%s\n", filename);
free(filename);
filename = expfilename(p, pd->payload);
if (!filename)
return -err_no_mem;
f = fopen(filename, "w");
if (!f) {
free(filename);
return -err_file_open;
}
continue;
}
line = strchr(s, ';');
if (!line)
continue;
line += 1;
comment = strchr(line, '#');
if (comment)
*comment = '\0';
/* remove trailing spaces. */
for (i = (int) strlen(line)-1; i >= 0 && isspace(line[i]); i--)
line[i] = '\0';
for (;;) {
char *tmp, label[256];
uint64_t addr;
int i, zero_padding, qmark_padding, qmark_size, status;
zero_padding = 0;
qmark_padding = 0;
qmark_size = 0;
/* find the label character in the string.
* if there is no label character, we just print
* the rest of the line and end.
*/
tmp = strchr(line, '%');
if (!tmp) {
if (fprintf(f, "%s", line) < 0) {
errcode = -err_file_write;
goto error;
}
break;
}
/* make the label character a null byte and
* print the first portion, which does not
* belong to the label into the file.
*/
*tmp = '\0';
if (fprintf(f, "%s", line) < 0) {
errcode = -err_file_write;
goto error;
}
/* test if there is a valid label name after the %. */
line = tmp+1;
if (*line == '\0' || isspace(*line)) {
errcode = -err_no_label;
goto error;
}
/* check if zero padding is requested. */
if (*line == '0') {
zero_padding = 1;
line += 1;
}
/* chek if ? padding is requested. */
else if (*line == '?') {
qmark_padding = 1;
zero_padding = 1;
qmark_size = 0;
line += 1;
}
/* advance i to the first non alpha-numeric
* character. all characters everything from
* line[0] to line[i-1] belongs to the label
* name.
*/
for (i = 0; islabelchar(line[i]); i++)
;
if (i > 255) {
errcode = -err_label_name;
goto error;
}
strncpy(label, line, i);
label[i] = '\0';
/* advance to next character. */
line = &line[i];
/* lookup the label name and print it to the
* output file.
*/
errcode = yasm_lookup_label(p->y, &addr, label);
if (errcode < 0) {
errcode = l_lookup(p->pt_labels, &addr, label);
if (errcode < 0)
goto error;
if (zero_padding)
status = fprintf(f, "%016" PRIx64, addr);
else
status = fprintf(f, "%" PRIx64, addr);
if (status < 0) {
errcode = -err_file_write;
goto error;
}
continue;
}
/* check if masking is requested. */
if (*line == '.') {
char *endptr;
long int n;
line += 1;
n = strtol(line, &endptr, 0);
/* check if strtol made progress and
* stops on a space or null byte.
* otherwise the int could not be
* parsed.
*/
if (line == endptr ||
(*endptr != '\0' && !isspace(*endptr)
&& !ispunct(*endptr))) {
errcode = -err_parse_int;
goto error;
}
addr &= (1 << (n << 3)) - 1;
line = endptr;
qmark_size = 8 - n;
}
if (qmark_padding) {
int i;
status = fprintf(f, "0x");
if (status < 0) {
errcode = -err_file_write;
goto error;
}
for (i = 0; i < qmark_size; ++i) {
status = fprintf(f, "??");
if (status < 0) {
errcode = -err_file_write;
goto error;
}
}
for (; i < 8; ++i) {
uint8_t byte;
byte = (uint8_t)(addr >> ((7 - i) * 8));
status = fprintf(f, "%02" PRIx8, byte);
if (status < 0) {
errcode = -err_file_write;
goto error;
}
}
} else if (zero_padding)
status = fprintf(f, "0x%016" PRIx64, addr);
else
status = fprintf(f, "0x%" PRIx64, addr);
if (status < 0) {
errcode = -err_file_write;
goto error;
}
}
if (fprintf(f, "\n") < 0) {
errcode = -err_file_write;
goto error;
}
}
