更新代码方案

inc/zuc256.h

@@ -1,36 +1,59 @@
-#ifndef ZUC256_H
-#define ZUC256_H
+#ifndef __ZUC256_H
+#define __ZUC256_H
 
 #include <stdint.h>
-#include <stddef.h>
+#include <stdlib.h>
 
-// ZUC256算法上下文结构
+// 上下文结构体:存储算法状态
 typedef struct {
-    uint32_t LFSR[16];  // 线性反馈移位寄存器
-    uint32_t NFSR[16];  // 非线性反馈移位寄存器
-    uint32_t R1, R2;    // 寄存器
-    uint32_t key[8];    // 256位密钥 (8*32)
-    uint32_t iv[4];     // 128位初始向量 (4*32)
-} zuc256_context;
+    uint32_t key[8];     // 256位密钥(拆分为8个32位字)
+    uint32_t iv[4];      // 128位IV(拆分为4个32位字)
+    uint32_t LFSR[16];   // 线性反馈移位寄存器
+    uint32_t NFSR[16];   // 非线性反馈移位寄存器
+    uint32_t R1, R2;     // 密钥流生成辅助寄存器
+    uint32_t ks_buf;     // 密钥流缓存(处理非4字节对齐数据)
+    size_t   ks_buf_len; // 缓存的密钥流字节数(0~3)
+} ZUC256_CTX;
+
+// 错误码定义
+#define ZUC256_SUCCESS    0   // 成功
+#define ZUC256_ERR_NULL   -1  // 空指针错误
+#define ZUC256_ERR_KEYLEN -2  // 密钥长度错误(需32字节)
+#define ZUC256_ERR_IVLEN  -3  // IV长度错误(需16字节)
+
+// 公共接口声明
+/**
+ * @brief 初始化ZUC256上下文
+ * @param ctx 上下文指针(需提前分配内存)
+ * @param key 256位密钥(32字节)
+ * @param iv 128位初始向量(16字节)
+ * @return 成功返回ZUC256_SUCCESS，失败返回错误码
+ */
+int ZUC256_Init(ZUC256_CTX *ctx, const uint8_t *key, const uint8_t *iv);
 
 /**
- * @brief 初始化ZUC256算法上下文
- * 
- * @param ctx 算法上下文指针
- * @param key 256位密钥（32字节）
- * @param iv 128位初始向量（16字节）
+ * @brief 分块处理数据(加密/解密)
+ * @param ctx 已初始化的上下文
+ * @param out 输出缓冲区(长度需≥in_len)
+ * @param in 输入数据
+ * @param in_len 输入数据长度(字节)
+ * @return 成功返回ZUC256_SUCCESS，失败返回错误码
  */
-void zuc256_init(zuc256_context *ctx, const uint8_t *key, const uint8_t *iv);
+int ZUC256_Update(ZUC256_CTX *ctx, uint8_t *out, const uint8_t *in, size_t in_len);
 
 /**
- * @brief 使用ZUC256算法进行加密或解密
- * 
- * @param ctx 算法上下文指针
- * @param input 输入数据
- * @param output 输出数据（与输入数据长度相同）
- * @param length 数据长度（字节数）
+ * @brief 结束处理(流密码无实际操作，仅对齐接口)
+ * @param ctx 上下文指针
+ * @param out 输出缓冲区(可为NULL)
+ * @return 成功返回ZUC256_SUCCESS，失败返回错误码
  */
-void zuc256_crypt(zuc256_context *ctx, const uint8_t *input, uint8_t *output, size_t length);
+int ZUC256_Final(ZUC256_CTX *ctx, uint8_t *out);
 
-#endif // ZUC256_H
-    
+/**
+ * @brief 清理上下文(清空敏感数据)
+ * @param ctx 上下文指针
+ */
+void ZUC256_Cleanup(ZUC256_CTX *ctx);
+
+#endif /*__ZUC256_H */
+