Handle possible null stream in TokenGenerator static ctor and throw e…

…xception when instance gets created.
0x0ade · Jul 20, 2024 · c994b96 · c994b96
1 parent 0af2345
commit c994b96
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/CelesteNet.Shared/TokenGenerator.cs b/CelesteNet.Shared/TokenGenerator.cs
@@ -22,7 +22,11 @@ public class TokenGenerator {
 
         static TokenGenerator() {
             // Read the list of polynomials
-            using Stream stream = typeof(TokenGenerator).Assembly.GetManifestResourceStream("polynomials.bin");
+            using Stream? stream = typeof(TokenGenerator).Assembly.GetManifestResourceStream("polynomials.bin");
+            if (stream == null) {
+                LFSRPolynomials = new uint[0];
+                return;
+            }
             using BinaryReader reader = new BinaryReader(stream);
 
             LFSRPolynomials = new uint[stream.Length / 4];
@@ -44,6 +48,9 @@ uint RandomUInt() {
                 return ((uint) rng.Next(1 << 30)) << 2 | ((uint) rng.Next(1 << 2));
             }
 
+            if (LFSRPolynomials.Length == 0)
+                throw new FileLoadException("Could not load LFSR polynomials", "polynomials.bin");
+
             lfsrStepRNG = rng;
 
             // Initialize the LFSR to a random non-zero state