Merge pull request #20 from CAIDA/fix-pybgpstream-segfault

Fix segfault in `BGPStream.get_data_interfaces`
CAIDA · Apr 26, 2016 · 07dcf71 · 07dcf71
2 parents 4a35f92 + b835c72
commit 07dcf71
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pybgpstream/src/_pybgpstream_bgpstream.c b/pybgpstream/src/_pybgpstream_bgpstream.c
@@ -190,16 +190,20 @@ BGPStream_get_data_interfaces(BGPStreamObject *self)
       return NULL;
 
     info = bgpstream_get_data_interface_info(self->bs, ids[i]);
+    if(info == NULL) {
+      /* this data interface is not available */
+      continue;
+    }
 
     /* add info to dict */
 
     /* id */
 #if PY_MAJOR_VERSION > 2
     ADD_TO_DICT("id", PyLong_FromLong(ids[i]));
 #else
-	ADD_TO_DICT("id", PyInt_FromLong(ids[i]));
+    ADD_TO_DICT("id", PyInt_FromLong(ids[i]));
 #endif
-	
+
     /* name */
     ADD_TO_DICT("name", PYSTR_FROMSTR(info->name));