@aztec/bb.js 0.0.1-commit.ee80a48 → 0.0.1-commit.f1df4d2

package/src/crs/net_crs.test.ts

@@ -0,0 +1,47 @@
+import { NetCrs, fetchWithFallback } from './net_crs.js';
+
+// Expected first G1 point from BN254 CRS (generator point with x=1, y=2 in big-endian)
+const BN254_G1_FIRST_ELEMENT = new Uint8Array([
+  0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0,
+  0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2,
+]);
+
+describe('NetCrs', () => {
+  it('should download CRS data from primary host', async () => {
+    const crs = new NetCrs(1);
+    await crs.init();
+
+    const g1Data = crs.getG1Data();
+    expect(g1Data.length).toBe(64); // 1 point * 64 bytes
+
+    // Verify first point matches expected generator
+    expect(g1Data).toEqual(BN254_G1_FIRST_ELEMENT);
+  }, 30000);
+
+  it('should download G2 data', async () => {
+    const crs = new NetCrs(1);
+    await crs.init();
+
+    const g2Data = crs.getG2Data();
+    expect(g2Data.length).toBe(128); // G2 point is 128 bytes
+  }, 30000);
+});
+
+describe('fetchWithFallback', () => {
+  it('should fallback to secondary URL when primary fails', async () => {
+    const badPrimaryUrl = 'https://nonexistent.invalid/g1.dat';
+    const goodFallbackUrl = 'https://crs.aztec-labs.com/g1.dat';
+    const options: RequestInit = {
+      headers: {
+        Range: 'bytes=0-63',
+      },
+    };
+
+    const response = await fetchWithFallback(badPrimaryUrl, goodFallbackUrl, options);
+    expect(response.ok || response.status === 206).toBe(true);
+
+    const data = new Uint8Array(await response.arrayBuffer());
+    expect(data.length).toBe(64);
+    expect(data).toEqual(BN254_G1_FIRST_ELEMENT);
+  }, 30000);
+});

package/src/crs/net_crs.ts

@@ -1,4 +1,30 @@
 import { retry, makeBackoff } from '../retry/index.js';
+
+// Primary CRS host (Cloudflare R2)
+const CRS_PRIMARY_HOST = 'https://crs.aztec-cdn.foundation';
+// Fallback CRS host (AWS S3)
+const CRS_FALLBACK_HOST = 'https://crs.aztec-labs.com';
+
+/**
+ * Fetches data from primary URL, falling back to secondary on failure
+ * @internal Exported for testing
+ */
+export async function fetchWithFallback(
+  primaryUrl: string,
+  fallbackUrl: string,
+  options: RequestInit,
+): Promise<Response> {
+  try {
+    const response = await fetch(primaryUrl, options);
+    if (response.ok || response.status === 206) {
+      return response;
+    }
+    throw new Error(`HTTP ${response.status}`);
+  } catch {
+    return await fetch(fallbackUrl, options);
+  }
+}
+
 /**
  * Downloader for CRS from the web or local.
  */
@@ -76,14 +102,14 @@ export class NetCrs {
     }
 
     const g1End = this.numPoints * 64 - 1;
+    const options: RequestInit = {
+      headers: {
+        Range: `bytes=0-${g1End}`,
+      },
+      cache: 'force-cache',
+    };
     return await retry(
-      () =>
-        fetch('https://crs.aztec.network/g1.dat', {
-          headers: {
-            Range: `bytes=0-${g1End}`,
-          },
-          cache: 'force-cache',
-        }),
+      () => fetchWithFallback(`${CRS_PRIMARY_HOST}/g1.dat`, `${CRS_FALLBACK_HOST}/g1.dat`, options),
       makeBackoff([5, 5, 5]),
     );
   }
@@ -92,11 +118,11 @@ export class NetCrs {
    * Fetches the appropriate range of points from a remote source
    */
   private async fetchG2Data(): Promise<Response> {
+    const options: RequestInit = {
+      cache: 'force-cache',
+    };
     return await retry(
-      () =>
-        fetch('https://crs.aztec.network/g2.dat', {
-          cache: 'force-cache',
-        }),
+      () => fetchWithFallback(`${CRS_PRIMARY_HOST}/g2.dat`, `${CRS_FALLBACK_HOST}/g2.dat`, options),
       makeBackoff([5, 5, 5]),
     );
   }
@@ -153,12 +179,17 @@ export class NetGrumpkinCrs {
     }
 
     const g1End = this.numPoints * 64 - 1;
-
-    return await fetch('https://crs.aztec.network/grumpkin_g1.dat', {
+    const options: RequestInit = {
       headers: {
         Range: `bytes=0-${g1End}`,
       },
       cache: 'force-cache',
-    });
+    };
+
+    return await fetchWithFallback(
+      `${CRS_PRIMARY_HOST}/grumpkin_g1.dat`,
+      `${CRS_FALLBACK_HOST}/grumpkin_g1.dat`,
+      options,
+    );
   }
 }