瀏覽代碼

Avoid generating zero byte chunks at the end of files

Jonas Borgström 14 年之前
父節點
當前提交
d130754a10
共有 1 個文件被更改,包括 7 次插入1 次删除
  1. 7 1
      darc/_speedups.c

+ 7 - 1
darc/_speedups.c

@@ -86,7 +86,13 @@ ChunkifyIter_iternext(PyObject *self)
     }
     if(c->remaining < c->window_size) {
         c->done = 1;
-        return PyBuffer_FromMemory(c->data + c->position, c->remaining);
+        if(c->remaining) {
+            return PyBuffer_FromMemory(c->data + c->position, c->remaining);
+        }
+        else {
+            PyErr_SetNone(PyExc_StopIteration);
+            return NULL;
+        }
     }
     sum = checksum(c->data + c->position, c->window_size, 0);
     c->remaining -= c->window_size;