[gcc r14-7652] gccrs: expand: Add procmacro loading function

[gcc r14-7652] gccrs: expand: Add procmacro loading function

[Arthur Cohen]

https://gcc.gnu.org/g:1f4660d5dba3a6e02aec5f95e8ec7b223ebb9730

commit r14-7652-g1f4660d5dba3a6e02aec5f95e8ec7b223ebb9730
Author: Pierre-Emmanuel Patry <pierre-emmanuel.patry@embecosm.com>
Date:   Mon May 15 15:23:50 2023 +0200

    gccrs: expand: Add procmacro loading function
    
    Add a function to load a shared object procmacro and retrieve the list
    of available procedural macros.
    
    gcc/rust/ChangeLog:
    
            * Make-lang.in: Add rust-proc-macro object to list.
            * expand/rust-proc-macro.cc: New file.
            * expand/rust-proc-macro.h: New file.
    
    Signed-off-by: Pierre-Emmanuel Patry <pierre-emmanuel.patry@embecosm.com>

Diff:
---
 gcc/rust/Make-lang.in              |  1 +
 gcc/rust/expand/rust-proc-macro.cc | 64 ++++++++++++++++++++++++++++++++++++++
 gcc/rust/expand/rust-proc-macro.h  | 34 ++++++++++++++++++++
 3 files changed, 99 insertions(+)

diff --git a/gcc/rust/Make-lang.in b/gcc/rust/Make-lang.in
index 03a92ac8874..2f1807912fc 100644
--- a/gcc/rust/Make-lang.in
+++ b/gcc/rust/Make-lang.in
@@ -91,6 +91,7 @@ GRS_OBJS = \
 	rust/rust-derive.o \
 	rust/rust-derive-clone.o \
 	rust/rust-derive-copy.o \
+	rust/rust-proc-macro.o \
     rust/rust-macro-invoc-lexer.o \
     rust/rust-macro-substitute-ctx.o \
     rust/rust-macro-builtins.o \
diff --git a/gcc/rust/expand/rust-proc-macro.cc b/gcc/rust/expand/rust-proc-macro.cc
new file mode 100644
index 00000000000..22744cb547d
--- /dev/null
+++ b/gcc/rust/expand/rust-proc-macro.cc
@@ -0,0 +1,64 @@
+// This file is part of GCC.
+
+// GCC is free software; you can redistribute it and/or modify it under
+// the terms of the GNU General Public License as published by the Free
+// Software Foundation; either version 3, or (at your option) any later
+// version.
+
+// GCC is distributed in the hope that it will be useful, but WITHOUT ANY
+// WARRANTY; without even the implied warranty of MERCHANTABILITY or
+// FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+// for more details.
+
+// You should have received a copy of the GNU General Public License
+// along with GCC; see the file COPYING3.  If not see
+// <http://www.gnu.org/licenses/>.
+
+#include "rust-diagnostics.h"
+#include "rust-proc-macro.h"
+#ifndef _WIN32
+#include <dlfcn.h>
+#endif
+
+namespace Rust {
+
+const std::string PROC_MACRO_DECL_PREFIX = "__gccrs_proc_macro_decls_";
+
+const ProcMacro::ProcmacroArray *
+load_macros_array (std::string path)
+{
+#ifndef _WIN32
+  void *handle = dlopen (path.c_str (), RTLD_LAZY | RTLD_LOCAL);
+  // We're leaking the handle since we can't ever unload it
+  if (!handle)
+    {
+      rust_debug ("Error whilst opening procedural macro: %s", dlerror ());
+      return nullptr;
+    }
+
+  // FIXME: Add CrateStableId handling, right now all versions may be loaded,
+  // even incompatible ones.
+  return *reinterpret_cast<const ProcMacro::ProcmacroArray **> (
+    dlsym (handle, PROC_MACRO_DECL_PREFIX.c_str ()));
+#else
+  rust_sorry_at (Location (),
+		 "Procedural macros are not yet supported on windows host");
+  gcc_unreachable ();
+#endif
+}
+
+const std::vector<ProcMacro::Procmacro>
+load_macros (std::string path)
+{
+  const ProcMacro::ProcmacroArray *array = load_macros_array (path);
+  // Did not load the proc macro
+  if (array == nullptr)
+    gcc_unreachable ();
+
+  rust_debug ("Found %lu procedural macros", array->length);
+
+  return std::vector<ProcMacro::Procmacro> (array->macros,
+					    array->macros + array->length);
+}
+
+} // namespace Rust
diff --git a/gcc/rust/expand/rust-proc-macro.h b/gcc/rust/expand/rust-proc-macro.h
new file mode 100644
index 00000000000..779d3c70fbc
--- /dev/null
+++ b/gcc/rust/expand/rust-proc-macro.h
@@ -0,0 +1,34 @@
+// This file is part of GCC.
+
+// GCC is free software; you can redistribute it and/or modify it under
+// the terms of the GNU General Public License as published by the Free
+// Software Foundation; either version 3, or (at your option) any later
+// version.
+
+// GCC is distributed in the hope that it will be useful, but WITHOUT ANY
+// WARRANTY; without even the implied warranty of MERCHANTABILITY or
+// FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+// for more details.
+
+// You should have received a copy of the GNU General Public License
+// along with GCC; see the file COPYING3.  If not see
+// <http://www.gnu.org/licenses/>.
+
+#ifndef RUST_PROC_MACRO_H
+#define RUST_PROC_MACRO_H
+
+#include "libproc_macro/proc_macro.h"
+
+namespace Rust {
+
+/**
+ * Load a procedural macro library and return a pointer to it's entrypoint.
+ *
+ * @param The path to the shared object file to load.
+ */
+const std::vector<ProcMacro::Procmacro>
+load_macros (std::string path);
+
+} // namespace Rust
+
+#endif /* ! RUST_PROC_MACRO_H */