[Storage] Fix CRC64 extension with large data (Azure#39655)

Author: jalauzon-msft

sdk/storage/azure-storage-extensions/azure/storage/extensions/crc64/crc64module.c

@@ -15,7 +15,7 @@ PyObject* compute(PyObject* self, PyObject* args) {
     if (!PyArg_ParseTuple(args, "y#K", &src, &length, &uCrc))
         return NULL;
 
-    int pData = 0;
+    uint64_t pData = 0;
     uint64_t uSize = (uint64_t)length;
     uint64_t uBytes, uStop;
 
@@ -25,12 +25,12 @@ PyObject* compute(PyObject* self, PyObject* args) {
     uStop = uSize - (uSize % 32);
     if (uStop >= 2 * 32)
     {
-        uint64_t uCrc0 = 0ULL;
-        uint64_t uCrc1 = 0ULL;
-        uint64_t uCrc2 = 0ULL;
-        uint64_t uCrc3 = 0ULL;
+        uint64_t uCrc0 = 0;
+        uint64_t uCrc1 = 0;
+        uint64_t uCrc2 = 0;
+        uint64_t uCrc3 = 0;
 
-        int pLast = pData + (int)uStop - 32;
+        uint64_t pLast = pData + uStop - 32;
         uSize -= uStop;
         uCrc0 = uCrc;
 

sdk/storage/azure-storage-extensions/tests/test_crc64.py

@@ -49,8 +49,13 @@ def test_compute(data, initial, expected):
     actual = crc64.compute(data, initial)
     assert actual == expected
 
-def test_compute_chunks():
-    data = os.urandom(4 * 1024 * 1024)
+@pytest.mark.parametrize("size", [1024, 4 * 1024 * 1024, 1 * 1024 * 1024 * 1024, 3 * 1024 * 1024 * 1024])
+def test_compute_chunks(size):
+    data = b''
+    while len(data) < size:
+        length = min(size - len(data), 1 * 1024 * 1024 * 1024)
+        data += os.urandom(length)
+
     chunk_size = 1024 * 1024
     full_crc = crc64.compute(data, 0)