1 files changed, 36 insertions, 0 deletions
diff --git a/Library/UefiShellUfmCommandLib/tbi/screen.c b/Library/UefiShellUfmCommandLib/tbi/screen.c
new file mode 100644
index 0000000..ae4f2be
--- /dev/null
+++ b/Library/UefiShellUfmCommandLib/tbi/screen.c
@@ -0,0 +1,36 @@
+#include <Uefi.h>
+#include <Library/DebugLib.h>
+#include <Library/BaseMemoryLib.h>
+#include <Library/MemoryAllocationLib.h>
+#include <Library/UefiBootServicesTableLib.h>
+
+#include "screen.h"
+
+struct screen *prepare_screen(VOID)
+{
+	struct screen *scr;
+	scr = AllocateZeroPool(sizeof(struct screen));
+	if(!scr)
+		return NULL;
+
+	scr->stdin = gST->ConIn;
+	scr->stdout = gST->ConOut;
+	scr->stderr = gST->StdErr;
+
+	gST->ConOut->QueryMode(
+		gST->ConOut,
+		gST->ConOut->Mode->Mode,
+		&(scr->columns),
+		&(scr->lines)
+	);
+	scr->attr = gST->ConOut->Mode->Attribute;
+
+	return scr;
+}
+
+VOID forget_screen(struct screen *scr)
+{
+	ASSERT(scr != NULL);
+
+	FreePool(scr);
+}