Catch more exceptions

insomniacslk · Aug 28, 2024 · 05a3042 · 05a3042
1 parent 00ca6f6
commit 05a3042
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/dublintraceroute/py3/_dublintraceroute.cc b/dublintraceroute/py3/_dublintraceroute.cc
@@ -99,6 +99,12 @@ static PyObject* DublinTraceroute_traceroute(PyObject *self, PyObject *args)
 	std::shared_ptr<TracerouteResults> results;
 	try {
 		results = dublintraceroute->traceroute();
+	} catch (Tins::malformed_packet &e) {
+		PyErr_SetString(PyExc_RuntimeError, e.what());
+		return NULL;
+	} catch (std::runtime_error &e) {
+		PyErr_SetString(PyExc_RuntimeError, e.what());
+		return NULL;
 	} catch (DublinTracerouteException &e) {
 		PyErr_SetString(PyExc_RuntimeError, e.what());
 		return NULL;