bindings: Better error handling in enumeration get() function.

[libsigrok.git] / bindings / cxx / enums.py

diff --git a/bindings/cxx/enums.py b/bindings/cxx/enums.py
index e942058cc3fb11e7b3748cd301d398d042e1cf98..9eddd1ad9631fe61344b7ec13068bf5cc0adb29c 100644 (file)
--- a/bindings/cxx/enums.py
+++ b/bindings/cxx/enums.py
@@ -95,6 +95,9 @@ code_template = """
 
 const {classname} *{classname}::get(int id)
 {{
+    if (_values.find(static_cast<{enumname}>(id)) == _values.end())
+        throw Error(SR_ERR_ARG);
+
     return {classname}::_values.at(static_cast<{enumname}>(id));
 }}
 """