Enum converter: write enums based on variant, instead of casting to uint

Signed-off-by: Martynas Gurskas <[email protected]>
NordSecurity · Feb 12, 2024 · 9a1e9f6 · 9a1e9f6
1 parent 24751f8
commit 9a1e9f6
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/bindgen/src/bindings/cpp/templates/enum_tmpl.cpp b/bindgen/src/bindings/cpp/templates/enum_tmpl.cpp
@@ -32,7 +32,15 @@ RustBuffer {{ ffi_converter_name }}::lower(const {{ type_name }} &val) {
 }
 
 void {{ ffi_converter_name }}::write(RustStream &stream, const {{ type_name }} &val) {
-    stream << static_cast<int32_t>(val);
+    switch (val) {
+        {% for variant in e.variants() %}
+    case {{ type_name }}::{{ variant|variant_name }}:
+        stream << static_cast<int32_t>({{ loop.index }});
+        break;
+        {% endfor %}
+    default:
+        throw std::runtime_error("No matchig {{ type_name }} variant");
+    }
 }
 
 int32_t {{ ffi_converter_name }}::allocation_size(const {{ type_name|class_name }} &) {
@@ -122,4 +130,4 @@ int32_t {{ ffi_converter_name }}::allocation_size(const {{ type_name|class_name
 
     return size;
 }
-{%- endif %}
+{%- endif %}