Add _sysconfigdata_*

Author: coolreader18

Cargo.lock

@@ -73,6 +73,7 @@ crossbeam-utils = "0.7"
 generational-arena = "0.2"
 parking_lot = { git = "https://github.com/Amanieu/parking_lot" } # TODO: use published version
 thread_local = "1.0"
+cfg-if = "0.1.10"
 
 ## unicode stuff
 unicode_names2 = "0.4"
@@ -128,3 +129,6 @@ features = [
 
 [target.'cfg(target_arch = "wasm32")'.dependencies]
 wasm-bindgen = "0.2"
+
+[build-dependencies]
+itertools = "0.8"

vm/Cargo.toml

@@ -1,3 +1,7 @@
+use itertools::Itertools;
+use std::env;
+use std::io::prelude::*;
+use std::path::PathBuf;
 use std::process::Command;
 
 fn main() {
@@ -8,6 +12,21 @@ fn main() {
     );
     println!("cargo:rustc-env=RUSTPYTHON_GIT_TAG={}", git_tag());
     println!("cargo:rustc-env=RUSTPYTHON_GIT_BRANCH={}", git_branch());
+
+    println!(
+        "cargo:rustc-env=RUSTPYTHON_TARGET_TRIPLE={}",
+        env::var("TARGET").unwrap()
+    );
+
+    let mut env_path = PathBuf::from(env::var_os("OUT_DIR").unwrap());
+    env_path.push("env_vars.rs");
+    let mut f = std::fs::File::create(env_path).unwrap();
+    write!(
+        f,
+        "hashmap! {{ {} }}",
+        std::env::vars_os().format_with(", ", |(k, v), f| f(&format_args!("{:?} => {:?}", k, v)))
+    )
+    .unwrap();
 }
 
 fn git_hash() -> String {

vm/build.rs

@@ -30,6 +30,7 @@ pub mod socket;
 mod string;
 #[cfg(feature = "rustpython-compiler")]
 mod symtable;
+mod sysconfigdata;
 #[cfg(not(target_arch = "wasm32"))]
 mod thread;
 mod time_module;
@@ -95,6 +96,7 @@ pub fn get_module_inits() -> HashMap<String, StdlibInitFunc> {
         "_imp".to_owned() => Box::new(imp::make_module),
         "unicodedata".to_owned() => Box::new(unicodedata::make_module),
         "_warnings".to_owned() => Box::new(warnings::make_module),
+        crate::sysmodule::sysconfigdata_name() => Box::new(sysconfigdata::make_module),
     };
 
     // Insert parser related modules:

vm/src/stdlib/mod.rs

@@ -0,0 +1,16 @@
+use crate::pyobject::{ItemProtocol, PyObjectRef};
+use crate::VirtualMachine;
+
+pub fn make_module(vm: &VirtualMachine) -> PyObjectRef {
+    let vars = vm.ctx.new_dict();
+    macro_rules! hashmap {
+        ($($key:literal => $value:literal),*) => {{
+            $(vars.set_item($key, vm.new_str($value.to_owned()), vm).unwrap();)*
+        }};
+    }
+    include!(concat!(env!("OUT_DIR"), "/env_vars.rs"));
+
+    py_module!(vm, "_sysconfigdata", {
+        "build_time_vars" => vars,
+    })
+}

vm/src/stdlib/sysconfigdata.rs

@@ -222,6 +222,32 @@ fn sys_displayhook(obj: PyObjectRef, vm: &VirtualMachine) -> PyResult<()> {
     Ok(())
 }
 
+const PLATFORM: &str = {
+    cfg_if::cfg_if! {
+        if #[cfg(any(target_os = "linux", target_os = "android"))] {
+            // Android is linux as well. see https://bugs.python.org/issue32637
+            "linux"
+        } else if #[cfg(target_os = "macos")] {
+            "darwin"
+        } else if #[cfg(windows)] {
+            "win32"
+        } else {
+            "unknown"
+        }
+    }
+};
+
+const ABIFLAGS: &str = "";
+
+// not the same as CPython (e.g. rust's x86_x64-unknown-linux-gnu is just x86_64-linux-gnu)
+// but hopefully that's just an implementation detail? TODO: copy CPython's multiarch exactly,
+// https://github.com/python/cpython/blob/3.8/configure.ac#L725
+const MULTIARCH: &str = env!("RUSTPYTHON_TARGET_TRIPLE");
+
+pub fn sysconfigdata_name() -> String {
+    format!("_sysconfigdata_{}_{}_{}", ABIFLAGS, PLATFORM, MULTIARCH)
+}
+
 pub fn make_module(vm: &VirtualMachine, module: PyObjectRef, builtins: PyObjectRef) {
     let ctx = &vm.ctx;
 
@@ -244,6 +270,7 @@ pub fn make_module(vm: &VirtualMachine, module: PyObjectRef, builtins: PyObjectR
     let implementation = py_namespace!(vm, {
         "name" => ctx.new_str("rustpython".to_owned()),
         "cache_tag" => ctx.new_str("rustpython-01".to_owned()),
+        "_multiarch" => ctx.new_str(MULTIARCH.to_owned()),
     });
 
     let path = ctx.new_list(
@@ -255,19 +282,6 @@ pub fn make_module(vm: &VirtualMachine, module: PyObjectRef, builtins: PyObjectR
             .collect(),
     );
 
-    let platform = if cfg!(target_os = "linux") {
-        "linux".to_owned()
-    } else if cfg!(target_os = "macos") {
-        "darwin".to_owned()
-    } else if cfg!(target_os = "windows") {
-        "win32".to_owned()
-    } else if cfg!(target_os = "android") {
-        // Linux as well. see https://bugs.python.org/issue32637
-        "linux".to_owned()
-    } else {
-        "unknown".to_owned()
-    };
-
     let framework = "".to_owned();
 
     // https://doc.rust-lang.org/reference/conditional-compilation.html#target_endian
@@ -395,7 +409,7 @@ settrace() -- set the global debug tracing function
       "_getframe" => ctx.new_function(getframe),
       "modules" => modules.clone(),
       "warnoptions" => ctx.new_list(vec![]),
-      "platform" => ctx.new_str(platform),
+      "platform" => ctx.new_str(PLATFORM.to_owned()),
       "_framework" => ctx.new_str(framework),
       "meta_path" => ctx.new_list(vec![]),
       "path_hooks" => ctx.new_list(vec![]),
@@ -414,7 +428,7 @@ settrace() -- set the global debug tracing function
       "exec_prefix" => ctx.new_str(exec_prefix.to_owned()),
       "base_exec_prefix" => ctx.new_str(base_exec_prefix.to_owned()),
       "exit" => ctx.new_function(sys_exit),
-      "abiflags" => ctx.new_str("".to_owned()),
+      "abiflags" => ctx.new_str(ABIFLAGS.to_owned()),
       "audit" => ctx.new_function(sys_audit),
       "displayhook" => ctx.new_function(sys_displayhook),
       "__displayhook__" => ctx.new_function(sys_displayhook),