通过前面那些章节,我们已经实现了sm3算法,共分为init
、update
、done
三部分来实现的,调用的时候呢,就按以下方式调用即可:
针对大文件,可以使用这三部来一点一点的计算,直到文件全部读取完,但是针对一些小数据,比如就几百字节,也要分三次调用,就显得有点繁琐了,所以再定义一个函数,专门用于小数据计算的,定义如下:
/**
* 直接计算消息的摘要
* @param input 消息
* @param iLen 消息长度(字节)
* @param output 输出摘要结果
*/
void gm_sm3(const unsigned char * input, unsigned int iLen, unsigned char output[32]);
用户不需要知道,函数里面具体是如何实现的,只要输入待计算的消息,以及转入输出缓冲区即可,由函数自身来帮忙调用init
、update
、done
这三部,实现代码如下:
/**
* 直接计算消息的摘要
* @param input 消息
* @param iLen 消息长度(字节)
* @param output 输出摘要结果
*/
void gm_sm3(const unsigned char * input, unsigned int iLen, unsigned char output[32]) {
gm_sm3_context ctx;
gm_sm3_init(&ctx);
gm_sm3_update(&ctx, input, iLen);
gm_sm3_done(&ctx, output);
}
另外,可能还会有一些需要复用同一个上下文的场景,比如一个函数内需要计算多次sm3摘要,又不想分配多个上下文,那么用完后,需要对上下文进行重置,下次才能继续用。在sm3.h
增加如下代码:
/**
* 重置上下文,以达到复用此上下文的目的
* @param ctx 上下文
*/
inline void gm_sm3_reset(gm_sm3_context * ctx) {
gm_sm3_init(ctx);
}
无非就是重置一下那8个字寄存器,以及一些中间状态,当然除了内联函数,还可以用宏定义来实现,比如:
#define GM_SM3_RESET(ctx) gm_sm3_init(ctx)
大家自己按喜好选一种即可。