[mypyc] Add primitive for librt.base64.b64decode

mypyc/lib-rt/librt_base64.c

@@ -259,16 +259,20 @@ static PyMethodDef librt_base64_module_methods[] = {
     {NULL, NULL, 0, NULL}
 };
 
+#ifdef MYPYC_EXPERIMENTAL
+
 static int
 base64_abi_version(void) {
-    return 0;
+    return LIBRT_BASE64_ABI_VERSION;
 }
 
 static int
 base64_api_version(void) {
-    return 0;
+    return LIBRT_BASE64_API_VERSION;
 }
 
+#endif
+
 static int
 librt_base64_module_exec(PyObject *m)
 {
@@ -278,6 +282,7 @@ librt_base64_module_exec(PyObject *m)
         (void *)base64_abi_version,
         (void *)base64_api_version,
         (void *)b64encode_internal,
+        (void *)b64decode_internal,
     };
     PyObject *c_api_object = PyCapsule_New((void *)base64_api, "librt.base64._C_API", NULL);
     if (PyModule_Add(m, "_C_API", c_api_object) < 0) {

mypyc/lib-rt/librt_base64.h

@@ -13,14 +13,15 @@ import_librt_base64(void)
 #else  // MYPYC_EXPERIMENTAL
 
 #define LIBRT_BASE64_ABI_VERSION 0
-#define LIBRT_BASE64_API_VERSION 0
-#define LIBRT_BASE64_API_LEN 3
+#define LIBRT_BASE64_API_VERSION 1
+#define LIBRT_BASE64_API_LEN 4
 
 static void *LibRTBase64_API[LIBRT_BASE64_API_LEN];
 
 #define LibRTBase64_ABIVersion (*(int (*)(void)) LibRTBase64_API[0])
 #define LibRTBase64_APIVersion (*(int (*)(void)) LibRTBase64_API[1])
 #define LibRTBase64_b64encode_internal (*(PyObject* (*)(PyObject *source)) LibRTBase64_API[2])
+#define LibRTBase64_b64decode_internal (*(PyObject* (*)(PyObject *source)) LibRTBase64_API[3])
 
 static int
 import_librt_base64(void)

mypyc/primitives/misc_ops.py

@@ -5,6 +5,7 @@
 from mypyc.ir.ops import ERR_FALSE, ERR_MAGIC, ERR_MAGIC_OVERLAPPING, ERR_NEVER
 from mypyc.ir.rtypes import (
     KNOWN_NATIVE_TYPES,
+    RUnion,
     bit_rprimitive,
     bool_rprimitive,
     bytes_rprimitive,
@@ -475,3 +476,13 @@
     experimental=True,
     capsule="librt.base64",
 )
+
+function_op(
+    name="librt.base64.b64decode",
+    arg_types=[RUnion([bytes_rprimitive, str_rprimitive])],
+    return_type=bytes_rprimitive,
+    c_function_name="LibRTBase64_b64decode_internal",
+    error_kind=ERR_MAGIC,
+    experimental=True,
+    capsule="librt.base64",
+)

mypyc/test-data/irbuild-base64.test

@@ -1,14 +1,31 @@
 [case testBase64_experimental]
-from librt.base64 import b64encode
+from librt.base64 import b64encode, b64decode
 
 def enc(b: bytes) -> bytes:
     return b64encode(b)
+
+def dec_bytes(b: bytes) -> bytes:
+    return b64decode(b)
+
+def dec_str(b: str) -> bytes:
+    return b64decode(b)
 [out]
 def enc(b):
     b, r0 :: bytes
 L0:
     r0 = LibRTBase64_b64encode_internal(b)
     return r0
+def dec_bytes(b):
+    b, r0 :: bytes
+L0:
+    r0 = LibRTBase64_b64decode_internal(b)
+    return r0
+def dec_str(b):
+    b :: str
+    r0 :: bytes
+L0:
+    r0 = LibRTBase64_b64decode_internal(b)
+    return r0
 
 [case testBase64ExperimentalDisabled]
 from librt.base64 import b64encode