cracknuts-squirrel 0.1.0a1__tar.gz

cracknuts_squirrel-0.1.0a1/LICENSE

@@ -0,0 +1,175 @@
+
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.

cracknuts_squirrel-0.1.0a1/PKG-INFO

@@ -0,0 +1,31 @@
+Metadata-Version: 2.4
+Name: cracknuts-squirrel
+Version: 0.1.0a1
+Summary: A Python library for side-channel analysis
+Author-email: Dingzb <dingzibiao@gmail.com>
+Maintainer-email: Dingzb <dingzibiao@gmail.com>
+License-Expression: Apache-2.0
+Project-URL: homepage, https://cracknuts.io
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy~=1.26.4
+Requires-Dist: pycryptodome~=3.23.0
+Requires-Dist: zarr~=2.18.2
+Requires-Dist: packaging~=24.1
+Requires-Dist: numba~=0.61.2
+Requires-Dist: dask~=2025.5.1
+Provides-Extra: jupyter
+Requires-Dist: jupyter~=1.1.1; extra == "jupyter"
+Dynamic: license-file
+
+# CrackNuts Squirrel
+
+## For Developers
+
+In the virtual environment, run the following command to install the dependencies:
+
+```bash
+pip install -e .
+pip install -r requirements-dev.txt
+```

cracknuts_squirrel-0.1.0a1/README.md

@@ -0,0 +1,10 @@
+# CrackNuts Squirrel
+
+## For Developers
+
+In the virtual environment, run the following command to install the dependencies:
+
+```bash
+pip install -e .
+pip install -r requirements-dev.txt
+```

cracknuts_squirrel-0.1.0a1/pyproject.toml

@@ -0,0 +1,51 @@
+[build-system]
+requires = ["setuptools >= 61.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "cracknuts-squirrel"
+description = "A Python library for side-channel analysis"
+dynamic = ["version"]
+requires-python = ">=3.12"
+readme = "README.md"
+license = "Apache-2.0"
+authors = [{ name = "Dingzb", email = "dingzibiao@gmail.com" }]
+maintainers = [{ name = "Dingzb", email = "dingzibiao@gmail.com" }]
+dependencies = [
+    "numpy~=1.26.4",
+    "pycryptodome~=3.23.0",
+    "zarr~=2.18.2",
+    "packaging~=24.1",
+    "numba~=0.61.2",
+    "dask~=2025.5.1"
+]
+
+[project.optional-dependencies]
+jupyter = [
+    "jupyter~=1.1.1",
+]
+
+[project.urls]
+homepage = "https://cracknuts.io"
+
+[tool.setuptools.dynamic]
+version = { attr = "cracknuts_squirrel.__version__" }
+
+[tool.ruff]
+line-length = 120
+target-version = "py312"
+exclude = ["demo", "tests", "docs"]
+
+[tool.ruff.lint]
+extend-select = [
+    "F",
+    "E",
+    "W",
+    "UP",
+]
+
+[tool.typos.files]
+extend-exclude = [
+    "demo",
+    "tests"
+]

cracknuts_squirrel-0.1.0a1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

cracknuts_squirrel-0.1.0a1/src/cracknuts_squirrel/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.0-alpha.1"

cracknuts_squirrel-0.1.0a1/src/cracknuts_squirrel/cpa_analysis.py

@@ -0,0 +1,189 @@
+import dask.array as da
+import numpy as np
+import numba as nb
+from preprocessingBasic import PPBasic
+from dask.diagnostics import ProgressBar
+import zarr
+from Crypto.Cipher import AES
+
+# AES-128 sbox used to compute model values
+AES_SBOX = np.array([99,124,119,123,242,107,111,197,48,1,103,43,254,215,171,118,
+                    202,130,201,125,250,89,71,240,173,212,162,175,156,164,114,192,
+                    183,253,147,38,54,63,247,204,52,165,229,241,113,216,49,21,
+                    4,199,35,195,24,150,5,154,7,18,128,226,235,39,178,117,
+                    9,131,44,26,27,110,90,160,82,59,214,179,41,227,47,132,
+                    83,209,0,237,32,252,177,91,106,203,190,57,74,76,88,207,
+                    208,239,170,251,67,77,51,133,69,249,2,127,80,60,159,168,
+                    81,163,64,143,146,157,56,245,188,182,218,33,16,255,243,210,
+                    205,12,19,236,95,151,68,23,196,167,126,61,100,93,25,115,
+                    96,129,79,220,34,42,144,136,70,238,184,20,222,94,11,219,
+                    224,50,58,10,73,6,36,92,194,211,172,98,145,149,228,121,
+                    231,200,55,109,141,213,78,169,108,86,244,234,101,122,174,8,
+                    186,120,37,46,28,166,180,198,232,221,116,31,75,189,139,138,
+                    112,62,181,102,72,3,246,14,97,53,87,185,134,193,29,158,
+                    225,248,152,17,105,217,142,148,155,30,135,233,206,85,40,223,
+                    140,161,137,13,191,230,66,104,65,153,45,15,176,84,187,22])
+
+# Hamming weights of the values 0-255 used for model values
+WEIGHTS = np.array([0,1,1,2,1,2,2,3,1,2,2,3,2,3,3,4,
+                    1,2,2,3,2,3,3,4,2,3,3,4,3,4,4,5,
+                    1,2,2,3,2,3,3,4,2,3,3,4,3,4,4,5,
+                    2,3,3,4,3,4,4,5,3,4,4,5,4,5,5,6,
+                    1,2,2,3,2,3,3,4,2,3,3,4,3,4,4,5,
+                    2,3,3,4,3,4,4,5,3,4,4,5,4,5,5,6,
+                    2,3,3,4,3,4,4,5,3,4,4,5,4,5,5,6,
+                    3,4,4,5,4,5,5,6,4,5,5,6,5,6,6,7,
+                    1,2,2,3,2,3,3,4,2,3,3,4,3,4,4,5,
+                    2,3,3,4,3,4,4,5,3,4,4,5,4,5,5,6,
+                    2,3,3,4,3,4,4,5,3,4,4,5,4,5,5,6,
+                    3,4,4,5,4,5,5,6,4,5,5,6,5,6,6,7,
+                    2,3,3,4,3,4,4,5,3,4,4,5,4,5,5,6,
+                    3,4,4,5,4,5,5,6,4,5,5,6,5,6,6,7,
+                    3,4,4,5,4,5,5,6,4,5,5,6,5,6,6,7,
+                    4,5,5,6,5,6,6,7,5,6,6,7,6,7,7,8],
+                    np.float32)
+
+class CPAAnalysis(PPBasic):
+    """
+    AES-CPA分析类，继承自预处理基类PPBasic
+    """
+    def __init__(self, input_path=None, output_path=None, byte_pos=list(range(16)), sample_range=(0, None), **kwargs):
+        super().__init__(input_path=input_path, output_path=output_path, **kwargs)
+        self.byte_pos = byte_pos if isinstance(byte_pos, (list, tuple)) else [byte_pos] 
+        self.sample_range = sample_range
+
+
+
+    def update(self, traces: np.ndarray, data: np.ndarray):
+        # Update the number of rows processed
+        self.trace_count += traces.shape[0]
+        # Update sample accumulator
+        self.sample_sum += np.sum(traces, axis=0)
+        # Update sample squared accumulator
+        self.sample_sq_sum += np.sum(np.square(traces), axis=0)
+        # Update model accumulator
+        self.model_sum += np.sum(data, axis=0)
+        # Update model squared accumulator
+        self.model_sq_sum += np.sum(np.square(data), axis=0)
+        data = data.reshape((data.shape[0], -1))
+        # Update product accumulator
+        self.prod_sum += np.matmul(data.T, traces)
+
+    def calculate(self):
+        # Sample mean computation
+        sample_mean = np.divide(self.sample_sum, self.trace_count)
+        # Model mean computation
+        model_mean = np.divide(self.model_sum, self.trace_count)
+
+        prod_mean = np.divide(self.prod_sum, self.trace_count)
+        # Calculate correlation coefficient numerator
+        numerator = np.subtract(prod_mean, model_mean*sample_mean)
+        # Calculate correlation coeefficient denominator sample part
+        to_sqrt = np.subtract(np.divide(self.sample_sq_sum, self.trace_count), np.square(sample_mean))
+        to_sqrt[to_sqrt < 0] = 0
+        denom_sample = np.sqrt(to_sqrt)
+        # Calculate correlation coefficient denominator model part
+        to_sqrt = np.subtract(np.divide(self.model_sq_sum, self.trace_count), np.square(model_mean))
+        to_sqrt = np.maximum(to_sqrt, 0)
+        denom_model = np.sqrt(to_sqrt)
+
+        denominator = denom_model*denom_sample
+
+        denominator[denominator == 0] = 1
+
+        return np.divide(numerator, denominator)
+
+    def perform_cpa(self):
+        """执行相关系数分析"""
+        store = zarr.DirectoryStore(self.output_path)
+        root = zarr.group(store=store, overwrite=True)
+        
+        # correlation = root.create_dataset(
+        #     '/0/0/correlation',
+        #     shape=(256, 16, self.sel_num_samples),
+        #     chunks=(256, 16, 1000)
+        # )
+        correlation = np.zeros((256, 16, self.sel_num_samples), dtype=np.float32)
+        
+
+        traces = self.t[:, self.sample_range[0]:self.sample_range[1]]
+        
+        # plaintext_bytes = np.array([pt[self.byte_pos] for pt in self.plaintext[:self.sel_num_traces]])
+        plaintext_bytes = self.plaintext[:self.sel_num_traces, :16]
+        # key_bytes = np.arange(256)[:, np.newaxis]
+        # xor_result = np.bitwise_xor(plaintext_bytes, key_bytes)
+        # sbox_output = AES_SBOX[xor_result]
+        # hw_matrix = WEIGHTS[sbox_output]
+        
+        # 创建Dask延迟计算任务
+        def compute_correlation(key_byte):
+            key_byte_array = np.full((self.sel_num_traces, 16), key_byte, dtype=np.uint8)
+            xor_result = np.bitwise_xor(plaintext_bytes, key_byte_array)
+            sbox_output = AES_SBOX[xor_result]
+            hw_matrix = WEIGHTS[sbox_output]
+            
+            # 使用Dask进行延迟计算
+            trace_count = da.from_array([self.sel_num_traces], chunks=1)
+            sample_sum = da.sum(traces, axis=0)
+            model_sum = da.sum(hw_matrix, axis=0)
+            prod_sum = da.dot(hw_matrix.T, traces)
+            
+            # 构建计算图
+            model_mean = model_sum / trace_count
+            sample_mean = sample_sum / trace_count
+            
+            # 调整维度以便广播
+            numerator = (prod_sum / trace_count) - (model_mean.reshape(-1, 1) * sample_mean)
+            
+            denom = da.sqrt(
+                da.maximum(0, (da.sum(hw_matrix**2)/trace_count - model_mean**2).reshape(-1, 1)) * 
+                da.maximum(0, (da.sum(traces**2, axis=0)/trace_count - sample_mean**2))
+            )
+            denom = da.where(denom == 0, 1, denom)
+            
+            return da.divide(numerator, denom)
+
+        # 并行计算所有密钥字节
+        delayed_results = [compute_correlation(kb) for kb in range(256)]
+        with ProgressBar():
+            futures = da.compute(*delayed_results)
+        
+        # 收集结果
+        # correlation = np.zeros((256, self.sel_num_samples), dtype=np.float32)
+        for key_byte, result in enumerate(futures):
+            correlation[key_byte] = result
+
+        # 优化后的候选值分析（向量化操作）
+        max_indices = np.argmax(np.abs(correlation), axis=2)
+        candidates = np.take_along_axis(correlation, max_indices[:, :, np.newaxis], axis=2).squeeze()
+        
+        for j in self.byte_pos:
+            # 获取当前字节的最优候选
+            best_key = np.abs(candidates[:, j]).argmax()
+            print(f'第{j+1}字节密钥: {hex(best_key)} 相关系数: {candidates[best_key, j]:.4f}')
+            
+            # 获取前5候选（向量化版本）
+            top5_indices = np.argsort(np.abs(candidates[:, j]))[::-1][:5]
+            for rank, idx in enumerate(top5_indices, 1):
+                print(f'第{rank}候选值：{hex(idx)}，相关系数：{candidates[idx, j]:.4f}')
+            print('\n')
+
+        root.create_dataset(
+            '/0/0/correlation',
+            data = correlation[:, self.byte_pos, :]
+        )
+        # 添加元数据
+        root.attrs.update({
+            "cpa_metadata": {
+                "analyzed_byte": self.byte_pos,
+                "sample_range": self.sample_range,
+                "trace_count": self.sel_num_traces
+            }
+        })
+
+if __name__ == "__main__":
+    # 示例用法
+    cpa = CPAAnalysis(input_path=r'D:\project\cracknuts\demo\jupyter\dataset\20250521110621.zarr')
+    cpa.auto_out_filename()
+    cpa.set_range(sample_range=(500, 10000))  # 设置分析的采样点范围
+    cpa.perform_cpa()

cracknuts_squirrel-0.1.0a1/src/cracknuts_squirrel/preprocessingBasic.py

@@ -0,0 +1,21 @@
+from traceIO import DaskZarrIO
+import dask.array as da
+import os
+
+class PPBasic(DaskZarrIO):
+    def __init__(self, input_path=None, output_path=None, tile='/0/0/', **kwargs):
+        super().__init__(input_path=input_path, output_path=output_path, tile=tile, **kwargs)
+        self.out_traces = da.zeros((self.num_traces, self.num_samples), chunks=self.t.chunks)
+
+    # @staticmethod    
+    def auto_out_filename(self):
+        base_name, ext = os.path.splitext(self.input_path)
+        self.output_path = f"{base_name}_{self.__class__.__name__}{ext}"
+
+
+if __name__ == "__main__":
+    # 示例用法
+    input_path = "E:\\codes\\template\\dataset\\20250221202721.zarr"
+    testPP = PPBasic(input_path=input_path)
+    testPP.auto_out_filename()
+    print(testPP.output_path)

cracknuts_squirrel-0.1.0a1/src/cracknuts_squirrel/staticalgin.py

@@ -0,0 +1,105 @@
+from enum import auto
+import dask.array as da
+from preprocessingBasic import PPBasic
+import numba as nb
+import numpy as np
+import scipy as sp
+from tqdm import tqdm
+import zarr
+
+class Staticalign(PPBasic):
+    """
+    用于侧信道曲线对齐的类，继承自曲线预处理的基类PPBasic
+    """
+    
+    def __init__(self, input_path=None, output_path=None, ref_path=None, ref_index=0, tile='/0/0/', threshold=0, **kwargs):
+        """
+        初始化参数
+        :param input_paths: 多个输入Zarr文件路径列表
+        :param output_path: 输出Zarr文件路径
+        :param reference_index: 参考曲线索引
+        :param kwargs: 传递给父类的其他参数
+        """
+        super().__init__(input_path=input_path, output_path=output_path, tile='/0/0/',**kwargs)
+        self.threshold = threshold
+        self.ref_path = ref_path
+        self.ref_index = ref_index
+        
+    def set_ref(self, ref_range=None, max_shift=0):
+        """
+        设置参考曲线
+        :param ref_range: 参考曲线范围，默认为(0, num_samples)
+        :param max_shift: 最大偏移量
+        """
+        if ref_range is None:
+            ref_range = (0, self.num_samples)
+        if self.ref_path is None:
+            ref_path = self.input_path
+        self.ref_range = ref_range
+        self.max_shift = max_shift
+        self.reference_trs = da.from_zarr(ref_path+'/0/0/traces')[self.ref_index, :].compute()
+        self.reference = self.reference_trs[ref_range[0]:ref_range[1]]
+        self.reference = self.reference[::-1]
+        cross = sp.signal.fftconvolve(self.reference_trs, self.reference, mode='valid')
+        self._refmaxloc = np.argmax(cross[ref_range[0]:ref_range[1]])
+        self._refmaxsize = max(cross[ref_range[0]:ref_range[1]])
+    
+
+    # @nb.njit(parallel=True)
+    def align_curves(self, method='correlation'):
+        """
+        :param method: 对齐方法，可选'correlation'(相关性)或'feature'(特征点)
+        :return: 对齐后的Dask数组列表
+        """
+        store = zarr.DirectoryStore(self.output_path)
+        root = zarr.group(store=store, overwrite=True)
+        
+        # 创建traces数据集
+        traces = root.create_dataset(
+            '/0/0/traces', 
+            shape=(self.sel_num_traces, self.num_samples)
+        )
+        plaintext = root.create_dataset(
+            '/0/0/plaintext', 
+            shape=(self.plaintext.shape[0], self.plaintext.shape[1])
+        )
+        plaintext[self.trace_range[0]:self.trace_range[1]] = self.plaintext[self.trace_range[0]:self.trace_range[1]].compute()
+        
+        # 添加attrs元数据
+        root.attrs.update({
+            "metadata": {
+                "channel_names": ["1"],
+                "create_time": 1740140842,
+                "data_length": self.plaintext.shape[1],
+                "sample_count": self.sel_num_samples,
+                "trace_count": self.sel_num_traces,
+                "version": "((0, '0.0.1'), (0, '0.0.1'))"
+            }
+        })
+
+        for trace in tqdm(range(self.trace_range[0], self.trace_range[1]), desc="对齐进度", unit="trace"):
+
+            curr_trace = self.t[trace, :]
+            # 归一化处理
+            norm_trace = (curr_trace - np.mean(curr_trace)) / (np.std(curr_trace) * len(curr_trace))
+            norm_ref = (self.reference - np.mean(self.reference)) / np.std(self.reference)
+            # 计算互相关并归一化为相关系数
+            cross = sp.signal.fftconvolve(norm_trace, norm_ref[::-1], mode='valid') 
+
+            newmaxloc = np.argmax(cross[self.ref_range[0]:self.ref_range[1]])
+                    
+            diff = newmaxloc-self._refmaxloc
+            if diff < 0:
+                newtrace = np.append(np.zeros(-diff), curr_trace[:diff])
+            elif diff > 0:
+                newtrace = np.append(curr_trace[diff:], np.zeros(diff))
+            traces[trace, :] = newtrace
+            
+
+if __name__ == "__main__":
+    # 示例用法    
+    aligner = Staticalign(input_path='E:\\codes\\template\\dataset\\20250221202721.zarr')
+    aligner.auto_out_filename()
+    aligner.set_ref(ref_range=(100, 1000))  # 设置参考曲线范围
+    aligner.set_range(trace_range=(0, 100))  # 设置读取范围，可选
+    aligner.align_curves(method='correlation')  # 使用相关性方法对齐曲线

cracknuts_squirrel-0.1.0a1/src/cracknuts_squirrel/traceIO.py

@@ -0,0 +1,167 @@
+import dask.array as da
+import zarr
+import os
+
+class DaskZarrIO:
+    """
+    用于高效读写Zarr文件的Dask封装类
+    """
+    
+    def __init__(self, input_path=None, output_path=None, expected_shape=None, chunks=None, tile='/0/0/'):
+        """
+        初始化参数
+        :param input_path: 输入Zarr文件路径
+        :param output_path: 输出Zarr文件路径
+        :param expected_shape: 预期数组形状
+        :param chunks: 分块大小
+        """
+        self.input_path = input_path
+        self.output_path = output_path
+        self.expected_shape = expected_shape
+        self.chunks = chunks
+        self.tile = tile
+        
+    
+        if not os.path.exists(self.input_path):
+            raise FileNotFoundError(f"输入文件不存在: {self.input_path}")
+            
+        if not os.path.exists(self.input_path + tile + 'traces'):
+            raise FileNotFoundError(f"traces数据不存在: {self.input_path + tile + 'traces'}")
+        self.t = da.from_zarr(self.input_path + tile + 'traces')
+        self.num_samples = self.t.shape[1]
+        self.sel_num_samples = self.num_samples
+        self.num_traces = self.t.shape[0]
+        self.sel_num_traces = self.num_traces
+        self.trace_range = (0, self.num_traces)
+        self.sample_range = (0, self.num_samples)
+        if os.path.exists(self.input_path+tile+'plaintext'):
+            self.plaintext = da.from_zarr(self.input_path+tile+'plaintext')
+        else:
+            self.plaintext = None
+        if os.path.exists(self.input_path+tile+'key'):
+            self.key = da.from_zarr(self.input_path+tile+'key')
+        else:
+            self.key = None
+        if os.path.exists(self.input_path+tile+'ciphertext'):
+            self.metadata = da.from_zarr(self.input_path+tile+'ciphertext')
+        else:
+            self.metadata = None
+
+    def set_range(self, trace_range=None, sample_range=None):
+        """
+        设置读取范围
+        :param trace_range: 读取的trace范围，默认为全范围
+        :param sample_range: 读取的sample范围，默认为全范围
+        """
+        if trace_range is None:
+            self.trace_range = (0, self.num_traces)
+        else:
+            self.trace_range = trace_range
+            self.sel_num_traces = trace_range[1] - trace_range[0]
+        if sample_range is None:
+            self.sample_range = (0, self.num_samples)
+        else:
+            self.sample_range = sample_range
+            self.sel_num_samples = sample_range[1] - sample_range[0]
+
+    
+    def read(self):
+        """
+        读取Zarr文件为Dask数组
+        :return: Dask数组
+        """
+        if not os.path.exists(self.input_path):
+            raise FileNotFoundError(f"输入文件不存在: {self.input_path}")
+            
+        dask_array = da.from_zarr(self.input_path)
+        
+        # if self.expected_shape and dask_array.shape != tuple(self.expected_shape):
+        #     raise ValueError(f"数组形状不匹配，预期: {self.expected_shape}, 实际: {dask_array.shape}")
+            
+        print(f"Dask数组形状: {dask_array.shape}")
+        print(f"Dask数组块大小: {dask_array.chunks}")
+        return dask_array
+    
+    def write(self, dask_array):
+        """
+        将Dask数组写入Zarr文件
+        :param dask_array: 要写入的Dask数组
+        """
+        if not self.output_path:
+            raise ValueError("未指定输出路径")
+            
+        if self.chunks:
+            dask_array = dask_array.rechunk(self.chunks)
+            
+        dask_array.to_zarr(self.output_path, overwrite=True)
+        print(f"数据已写入: {self.output_path}")
+        
+        
+
+# 保留原有函数作为兼容接口
+def read_zarr_with_dask(zarr_path):
+    """兼容旧版读取函数"""
+    return DaskZarrIO(input_path=zarr_path).read()
+
+def write_dask_to_zarr(dask_array, output_path):
+    """兼容旧版写入函数"""
+    DaskZarrIO(output_path=output_path).write(dask_array)
+
+
+
+def merge_zarr_files(file1, file2, output_file, tile='/0/0/', axis=0):
+    """
+    合并两个zarr文件并保存到输出文件
+    :param file1: 第一个zarr文件路径
+    :param file2: 第二个zarr文件路径
+    :param output_file: 输出文件路径
+    :param axis: 合并轴，0为垂直合并，1为水平合并
+    :return: 合并后的Dask数组
+    """
+
+    store = zarr.DirectoryStore(output_file)
+    root = zarr.group(store=store, overwrite=True)
+    # 加载两个文件的曲线
+    arr1 = da.from_zarr(file1+tile+'traces')
+    arr2 = da.from_zarr(file2+tile+'traces')  
+    # 检查形状是否兼容
+    if axis == 0 and arr1.shape[1:] != arr2.shape[1:]:
+        raise ValueError("垂直合并时除第一维外其他维度必须相同")
+    if axis == 1 and arr1.shape[0] != arr2.shape[0]:
+        raise ValueError("水平合并时第一维必须相同")
+    # 合并数组
+    merged = da.concatenate([arr1, arr2], axis=axis)
+    # 创建traces数据集
+    root.create_dataset(
+        '/0/0/traces', 
+        shape=(merged.shape[0], merged.shape[1]),
+        data = merged.compute()
+    )
+    #TODO: 合并明文、密文和密钥
+
+    # 加载file1的traces数据集
+    file1_store = zarr.DirectoryStore(file1)
+    file1_root = zarr.group(store=file1_store)
+
+    # 将输出文件的attrs设为和file1一致
+    root.attrs.update(file1_root.attrs)
+    
+
+
+if __name__ == "__main__":
+    # 示例用法
+    input_path = "E:\\codes\\template\\dataset\\20250221202721.zarr"
+    output_path = "dataset/dask_output.zarr"
+    
+    test = DaskZarrIO(input_path=input_path)
+    print(test.t[0].compute())
+    
+    # 读取示例
+    # dask_data = read_zarr_with_dask(input_path)
+    
+    # # 对数据进行一些操作（示例：计算均值）
+    # mean_value = dask_data.mean().compute()
+    # print(f"数据均值: {mean_value}")
+    
+    # 写入示例
+    # write_dask_to_zarr(dask_data, output_path)

cracknuts_squirrel-0.1.0a1/src/cracknuts_squirrel.egg-info/PKG-INFO

@@ -0,0 +1,31 @@
+Metadata-Version: 2.4
+Name: cracknuts-squirrel
+Version: 0.1.0a1
+Summary: A Python library for side-channel analysis
+Author-email: Dingzb <dingzibiao@gmail.com>
+Maintainer-email: Dingzb <dingzibiao@gmail.com>
+License-Expression: Apache-2.0
+Project-URL: homepage, https://cracknuts.io
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy~=1.26.4
+Requires-Dist: pycryptodome~=3.23.0
+Requires-Dist: zarr~=2.18.2
+Requires-Dist: packaging~=24.1
+Requires-Dist: numba~=0.61.2
+Requires-Dist: dask~=2025.5.1
+Provides-Extra: jupyter
+Requires-Dist: jupyter~=1.1.1; extra == "jupyter"
+Dynamic: license-file
+
+# CrackNuts Squirrel
+
+## For Developers
+
+In the virtual environment, run the following command to install the dependencies:
+
+```bash
+pip install -e .
+pip install -r requirements-dev.txt
+```

cracknuts_squirrel-0.1.0a1/src/cracknuts_squirrel.egg-info/SOURCES.txt

@@ -0,0 +1,13 @@
+LICENSE
+README.md
+pyproject.toml
+src/cracknuts_squirrel/__init__.py
+src/cracknuts_squirrel/cpa_analysis.py
+src/cracknuts_squirrel/preprocessingBasic.py
+src/cracknuts_squirrel/staticalgin.py
+src/cracknuts_squirrel/traceIO.py
+src/cracknuts_squirrel.egg-info/PKG-INFO
+src/cracknuts_squirrel.egg-info/SOURCES.txt
+src/cracknuts_squirrel.egg-info/dependency_links.txt
+src/cracknuts_squirrel.egg-info/requires.txt
+src/cracknuts_squirrel.egg-info/top_level.txt

cracknuts_squirrel-0.1.0a1/src/cracknuts_squirrel.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

cracknuts_squirrel-0.1.0a1/src/cracknuts_squirrel.egg-info/requires.txt

@@ -0,0 +1,9 @@
+numpy~=1.26.4
+pycryptodome~=3.23.0
+zarr~=2.18.2
+packaging~=24.1
+numba~=0.61.2
+dask~=2025.5.1
+
+[jupyter]
+jupyter~=1.1.1

cracknuts_squirrel-0.1.0a1/src/cracknuts_squirrel.egg-info/top_level.txt

@@ -0,0 +1 @@
+cracknuts_squirrel