githubmirror/shim

errlog.c
86 строк2 KB
Hans de Goede
console: Add console_print and console_print_at helpers
13 мар 2018, 01:00
13 мар 2018, 01:001fe31ee
/*
 * errlog.c
 * Copyright 2017 Peter Jones <pjones@redhat.com>
 *
 * Distributed under terms of the GPLv3 license.
 */

#include "shim.h"

static CHAR16 **errs = NULL;
static UINTN nerrs = 0;

EFI_STATUS
VLogError(const char *file, int line, const char *func, CHAR16 *fmt, va_list args)
{
	va_list args2;
	UINTN size = 0, size2;
	CHAR16 **newerrs;

	size = SPrint(NULL, 0, L"%a:%d %a() ", file, line, func);
	va_copy(args2, args);
	size2 = VSPrint(NULL, 0, fmt, args2);
	va_end(args2);

	newerrs = ReallocatePool(errs, (nerrs + 1) * sizeof(*errs),
				       (nerrs + 3) * sizeof(*errs));
	if (!newerrs)
		return EFI_OUT_OF_RESOURCES;

	newerrs[nerrs] = AllocatePool(size*2+2);
	if (!newerrs[nerrs])
		return EFI_OUT_OF_RESOURCES;
	newerrs[nerrs+1] = AllocatePool(size2*2+2);
	if (!newerrs[nerrs+1])
		return EFI_OUT_OF_RESOURCES;

	SPrint(newerrs[nerrs], size*2+2, L"%a:%d %a() ", file, line, func);
	va_copy(args2, args);
	VSPrint(newerrs[nerrs+1], size2*2+2, fmt, args2);
	va_end(args2);

	nerrs += 2;
	newerrs[nerrs] = NULL;
	errs = newerrs;

	return EFI_SUCCESS;
}

EFI_STATUS
LogError_(const char *file, int line, const char *func, CHAR16 *fmt, ...)
{
	va_list args;
	EFI_STATUS efi_status;

	va_start(args, fmt);
	efi_status = VLogError(file, line, func, fmt, args);
	va_end(args);

	return efi_status;
}

VOID
PrintErrors(VOID)
{
	UINTN i;

	if (!verbose)
		return;

	for (i = 0; i < nerrs; i++)
		console_print(L"%s", errs[i]);
}

VOID
ClearErrors(VOID)
{
	UINTN i;

	for (i = 0; i < nerrs; i++)
		FreePool(errs[i]);
	FreePool(errs);
	nerrs = 0;
	errs = NULL;
}

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