Catch exceptions from iterator.MoveNext()

rickardraysearch · rickardraysearch · commit 8531755b006c · 2017-08-09T22:03:32.000+02:00
diff --git a/src/runtime/iterator.cs b/src/runtime/iterator.cs
@@ -23,9 +23,21 @@ public Iterator(IEnumerator e)
         public static IntPtr tp_iternext(IntPtr ob)
         {
             var self = GetManagedObject(ob) as Iterator;
-            if (!self.iter.MoveNext())
+            try
             {
-                Exceptions.SetError(Exceptions.StopIteration, Runtime.PyNone);
+                if (!self.iter.MoveNext())
+                {
+                    Exceptions.SetError(Exceptions.StopIteration, Runtime.PyNone);
+                    return IntPtr.Zero;
+                }
+            }
+            catch (Exception e)
+            {
+                if (e.InnerException != null)
+                {
+                    e = e.InnerException;
+                }
+                Exceptions.SetError(e);
                 return IntPtr.Zero;
             }
             object item = self.iter.Current;
