From 9ed56198f91e7921543e3381bf2523e4702d492b Mon Sep 17 00:00:00 2001
From: Quinten Stokkink <goqs@hotmail.com>
Date: Thu, 26 Sep 2024 15:32:44 +0200
Subject: [PATCH] Fixed rendering b64 total pieces decoding

---
 src/tribler/ui/src/pages/Downloads/Details.tsx |  2 +-
 src/tribler/ui/src/pages/Downloads/Pieces.tsx  | 14 +++++++-------
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/src/tribler/ui/src/pages/Downloads/Details.tsx b/src/tribler/ui/src/pages/Downloads/Details.tsx
index 54a33b3a8c..ca5b6649ab 100644
--- a/src/tribler/ui/src/pages/Downloads/Details.tsx
+++ b/src/tribler/ui/src/pages/Downloads/Details.tsx
@@ -40,7 +40,7 @@ export default function DownloadDetails({ selectedDownloads }: { selectedDownloa
                     <div className="flex flex-col h-full p-4 text-sm">
                         <div className="flex flex-row">
                             <div className="basis-1/4">{t('Progress')}</div>
-                            <div className="basis-3/4 m-auto"><Pieces pieces64={download.pieces || ''} /></div>
+                            <div className="basis-3/4 m-auto"><Pieces numpieces={download.total_pieces} pieces64={download.pieces || ''} /></div>
                         </div>
                         <div className="flex flex-row">
                             <div className="basis-1/4">{t('Name')}</div>
diff --git a/src/tribler/ui/src/pages/Downloads/Pieces.tsx b/src/tribler/ui/src/pages/Downloads/Pieces.tsx
index f2da2f0d47..1ec723b5f9 100644
--- a/src/tribler/ui/src/pages/Downloads/Pieces.tsx
+++ b/src/tribler/ui/src/pages/Downloads/Pieces.tsx
@@ -1,10 +1,10 @@
 import { useEffect, useRef } from 'react'
 
 
-export default function Pieces({ pieces64 }: { pieces64: string }) {
+export default function Pieces({ pieces64, numpieces }: { pieces64: string, numpieces: number }) {
     const ref = useRef<HTMLCanvasElement>(null)
 
-    const convertPieces = (pieces64: string) => {
+    const convertPieces = (pieces64: string, numpieces: number) => {
         if (pieces64 === undefined) { return [] }
 
         // Remove trailing '='
@@ -12,7 +12,7 @@ export default function Pieces({ pieces64 }: { pieces64: string }) {
 
         const pieces = [];
         const pieceString = atob(pieces64);
-        for (let i = 0; i < pieceString.length; ++i) {
+        for (let i = 0; i < Math.min(numpieces, pieceString.length); ++i) {
             const pieceNumber = pieceString[i].charCodeAt(0);
             for (let j = 8 - 1; j >= 0; --j) {
                 pieces.push(pieceNumber & 1 << j ? 1 : 0);
@@ -23,14 +23,14 @@ export default function Pieces({ pieces64 }: { pieces64: string }) {
 
     useEffect(() => {
         if (ref.current) {
-            const canvas = ref.current.getContext('2d')
-            const pieces = convertPieces(pieces64)
+            const canvas = ref.current.getContext('2d');
+            const pieces = convertPieces(pieces64, numpieces);
             if (!canvas || !pieces || pieces.length === 0) { return; }
 
             // Get size from the HTML canvas element
             const width = canvas.canvas.width;
             const height = canvas.canvas.height;
-            const numPieces = pieces.length;
+            const numPieces = numpieces;
 
             if (numPieces <= width) {
                 const pieceWidth = width / numPieces;
@@ -55,7 +55,7 @@ export default function Pieces({ pieces64 }: { pieces64: string }) {
                 }
             }
         }
-    }, [pieces64])
+    }, [pieces64, numpieces])
 
     return <canvas ref={ref} style={{ height: '20px', width: '97%', background: 'white', border: '1px solid #2f2f2f' }} />
 }