1 files changed, 19 insertions, 0 deletions
diff --git a/fdisk.c b/fdisk.c
index d7d3fe4..d78e6e2 100644
--- a/fdisk.c
+++ b/fdisk.c
@@ -14,6 +14,25 @@
 
 #include "fdisk.h"
 
+void *set_PyErr_from_rc(int e)
+{
+	switch (e) {
+	case ENOMEM:
+		PyErr_SetString(PyExc_MemoryError, strerror(e));
+		break;
+	case EINVAL:
+		PyErr_SetString(PyExc_TypeError, strerror(e));
+		break;
+	case ENOSYS:
+		PyErr_SetString(PyExc_NotImplementedError, strerror(e));
+		break;
+	default:
+		PyErr_SetString(PyExc_Exception, strerror(e));
+	}
+
+	return NULL;
+}
+
 PyObject *PyObjectResultStr(const char *s)
 {
 	PyObject *result;