[libc] work on puts in uefi

Author: RossComputerGuy

libc/config/uefi/entrypoints.txt

@@ -152,6 +152,19 @@ set(TARGET_LIBC_ENTRYPOINTS
     libc.src.stdbit.stdc_trailing_zeros_ull
     libc.src.stdbit.stdc_trailing_zeros_us
 
+    # stdio.h entrypoints
+    libc.src.stdio.getchar
+    libc.src.stdio.printf
+    libc.src.stdio.putchar
+    libc.src.stdio.puts
+    libc.src.stdio.snprintf
+    libc.src.stdio.sprintf
+    libc.src.stdio.asprintf
+    libc.src.stdio.vprintf
+    libc.src.stdio.vsnprintf
+    libc.src.stdio.vsprintf
+    libc.src.stdio.vasprintf
+
     # stdlib.h entrypoints
     libc.src.stdlib._Exit
     libc.src.stdlib.abort

libc/config/uefi/headers.txt

@@ -10,6 +10,7 @@ set(TARGET_PUBLIC_HEADERS
     libc.include.setjmp
     libc.include.stdfix
     libc.include.stdint
+    libc.include.stdio
     libc.include.stdlib
     libc.include.string
     libc.include.strings

libc/include/llvm-libc-types/CMakeLists.txt

@@ -217,7 +217,9 @@ add_header(EFI_SIMPLE_TEXT_INPUT_PROTOCOL
   HDR
     EFI_SIMPLE_TEXT_INPUT_PROTOCOL.h
   DEPENDS
+    libc.include.llvm-libc-macros.EFIAPI_macros
     libc.include.llvm-libc-macros.stdint_macros
+    .EFI_EVENT
     .EFI_STATUS
     .char16_t
 )

libc/include/llvm-libc-types/EFI_SIMPLE_TEXT_INPUT_PROTOCOL.h

@@ -9,7 +9,9 @@
 #ifndef LLVM_LIBC_TYPES_EFI_SIMPLE_TEXT_INPUT_PROTOCOL_H
 #define LLVM_LIBC_TYPES_EFI_SIMPLE_TEXT_INPUT_PROTOCOL_H
 
+#include "../llvm-libc-macros/EFIAPI-macros.h"
 #include "../llvm-libc-macros/stdint-macros.h"
+#include "EFI_EVENT.h"
 #include "EFI_STATUS.h"
 #include "char16_t.h"
 

libc/src/__support/CMakeLists.txt

@@ -359,6 +359,10 @@ add_subdirectory(StringUtil)
 add_subdirectory(GPU)
 add_subdirectory(RPC)
 
+if(LIBC_TARGET_OS_IS_UEFI)
+  add_subdirectory(UEFI)
+endif()
+
 # Thread support is used by other "File". So, we add the "threads"
 # before "File".
 add_subdirectory(threads)

libc/src/__support/UEFI/CMakeLists.txt

@@ -0,0 +1,11 @@
+add_object_library(
+  file
+  SRCS
+    file.cpp
+  HDRS
+    file.h
+  DEPENDS
+    libc.include.uefi
+    libc.hdr.types.FILE
+    libc.src.__support.CPP.new
+)

libc/src/__support/UEFI/file.cpp

@@ -0,0 +1,67 @@
+#include "file.h"
+#include "hdr/types/FILE.h"
+#include "src/__support/macros/config.h"
+#include <Uefi.h>
+
+#define STDIN_FILENO 0
+#define STDOUT_FILENO 1
+#define STDERR_FILENO 2
+
+namespace LIBC_NAMESPACE_DECL {
+bool File::needsReset() { return handle_type == FileHandleId; }
+
+void File::reset() {
+  if (handle_type != FileHandleId)
+    return;
+
+  if (handle.id == STDIN_FILENO) {
+    handle = (FileHandle){
+        .simple_text_input = efi_system_table->ConIn,
+    };
+    handle_type = FileHandleSimpleTextInput;
+  } else {
+    handle = (FileHandle){
+        .simple_text_output = handle.id == STDERR_FILENO
+                                  ? efi_system_table->StdErr
+                                  : efi_system_table->ConOut,
+    };
+    handle_type = FileHandleSimpleTextOutput;
+  }
+}
+
+size_t File::write(const void *data, size_t data_len) {
+  if (needsReset())
+    reset();
+
+  if (handle_type == FileHandleSimpleTextOutput) {
+    handle.simple_text_output->OutputString(
+        handle.simple_text_output, reinterpret_cast<const char16_t *>(data));
+    return data_len;
+  }
+  return 0;
+}
+
+File stdin(
+    (FileHandle){
+        .id = STDIN_FILENO,
+    },
+    FileHandleId);
+
+File stdout(
+    (FileHandle){
+        .id = STDOUT_FILENO,
+    },
+    FileHandleId);
+
+File stderr(
+    (FileHandle){
+        .id = STDERR_FILENO,
+    },
+    FileHandleId);
+} // namespace LIBC_NAMESPACE_DECL
+
+extern "C" {
+FILE *stdin = reinterpret_cast<FILE *>(&LIBC_NAMESPACE::stdin);
+FILE *stdout = reinterpret_cast<FILE *>(&LIBC_NAMESPACE::stdout);
+FILE *stderr = reinterpret_cast<FILE *>(&LIBC_NAMESPACE::stderr);
+}

libc/src/__support/UEFI/file.h

@@ -0,0 +1,46 @@
+#ifndef LLVM_LIBC_SRC___SUPPORT_UEFI_FILE_H
+#define LLVM_LIBC_SRC___SUPPORT_UEFI_FILE_H
+
+#include "include/llvm-libc-types/EFI_SIMPLE_TEXT_INPUT_PROTOCOL.h"
+#include "include/llvm-libc-types/EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL.h"
+#include "src/__support/CPP/new.h"
+#include "src/__support/macros/config.h"
+
+namespace LIBC_NAMESPACE_DECL {
+
+enum FileHandleType {
+  FileHandleId,
+  FileHandleSimpleTextInput,
+  FileHandleSimpleTextOutput,
+};
+
+union FileHandle {
+  int id;
+  EFI_SIMPLE_TEXT_INPUT_PROTOCOL *simple_text_input;
+  EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL *simple_text_output;
+};
+
+class File {
+  FileHandle handle;
+  FileHandleType handle_type;
+
+private:
+  bool needsReset();
+
+public:
+  constexpr File(FileHandle handle, FileHandleType handle_type)
+      : handle(handle), handle_type(handle_type) {}
+
+  void reset();
+
+  size_t read(void *data, size_t len);
+  size_t write(const void *data, size_t len);
+};
+
+extern File stdin;
+extern File stdout;
+extern File stderr;
+
+} // namespace LIBC_NAMESPACE_DECL
+
+#endif // LLVM_LIBC_SRC___SUPPORT_UEFI_FILE_H

libc/src/stdio/CMakeLists.txt

@@ -240,26 +240,28 @@ add_entrypoint_object(
 add_subdirectory(printf_core)
 add_subdirectory(scanf_core)
 
-add_entrypoint_object(
-  remove
-  ALIAS
-  DEPENDS
-    .${LIBC_TARGET_OS}.remove
-)
+if(NOT LIBC_TARGET_OS_IS_UEFI)
+  add_entrypoint_object(
+    remove
+    ALIAS
+    DEPENDS
+      .${LIBC_TARGET_OS}.remove
+  )
 
-add_entrypoint_object(
-  rename
-  ALIAS
-  DEPENDS
-    .${LIBC_TARGET_OS}.rename
-)
+  add_entrypoint_object(
+    rename
+    ALIAS
+    DEPENDS
+      .${LIBC_TARGET_OS}.rename
+  )
 
-add_entrypoint_object(
-  fdopen
-  ALIAS
-  DEPENDS
-    .${LIBC_TARGET_OS}.fdopen
-)
+  add_entrypoint_object(
+    fdopen
+    ALIAS
+    DEPENDS
+      .${LIBC_TARGET_OS}.fdopen
+  )
+endif()
 
 # These entrypoints have multiple potential implementations.
 add_stdio_entrypoint_object(feof)

libc/src/stdio/uefi/CMakeLists.txt

@@ -0,0 +1,59 @@
+add_entrypoint_object(
+  getchar
+  SRCS
+    getchar.cpp
+  HDRS
+    ../getchar.h
+  DEPENDS
+    libc.hdr.stdio_macros
+    libc.src.__support.UEFI.file
+)
+
+add_entrypoint_object(
+  printf
+  SRCS
+    printf.cpp
+  HDRS
+    ../printf.h
+  DEPENDS
+    libc.src.stdio.printf_core.printf_main
+    libc.src.stdio.printf_core.writer
+    libc.src.__support.arg_list
+    libc.src.__support.OSUtil.osutil
+)
+
+add_entrypoint_object(
+  putchar
+  SRCS
+    putchar.cpp
+  HDRS
+    ../putchar.h
+  DEPENDS
+    libc.src.__support.OSUtil.osutil
+    libc.src.__support.CPP.string_view
+    libc.src.__support.UEFI.file
+)
+
+add_entrypoint_object(
+  puts
+  SRCS
+    puts.cpp
+  HDRS
+    ../puts.h
+  DEPENDS
+    libc.src.__support.UEFI.file
+    libc.src.string.strlen
+)
+
+add_entrypoint_object(
+  vprintf
+  SRCS
+    vprintf.cpp
+  HDRS
+    ../vprintf.h
+  DEPENDS
+    libc.src.stdio.printf_core.printf_main
+    libc.src.stdio.printf_core.writer
+    libc.src.__support.arg_list
+    libc.src.__support.OSUtil.osutil
+)