From f959236af00286b98be2666f0b7ee6f03fb97e55 Mon Sep 17 00:00:00 2001
From: Raymonf <raymonf@outlook.com>
Date: Wed, 10 May 2023 17:05:11 -0400
Subject: [PATCH] SUN encryption support

---
 titles/chuni/index.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/titles/chuni/index.py b/titles/chuni/index.py
index 811840a..5d185e9 100644
--- a/titles/chuni/index.py
+++ b/titles/chuni/index.py
@@ -98,15 +98,18 @@ class ChuniServlet:
             ]
             for method in method_list:
                 method_fixed = inflection.camelize(method)[6:-7]
+                # number of iterations was changed to 70 in SUN
+                iter_count = 70 if version >= ChuniConstants.VER_CHUNITHM_SUN else 44
                 hash = PBKDF2(
                     method_fixed,
                     bytes.fromhex(keys[2]),
                     128,
-                    count=44,
+                    count=iter_count,
                     hmac_hash_module=SHA1,
                 )
 
-                self.hash_table[version][hash.hex()] = method_fixed
+                hashed_name = hash.hex()[:32] # truncate unused bytes like the game does
+                self.hash_table[version][hashed_name] = method_fixed
 
                 self.logger.debug(
                     f"Hashed v{version} method {method_fixed} with {bytes.fromhex(keys[2])} to get {hash.hex()}"