mirror of
https://github.com/guanzhi/GmSSL.git
synced 2026-05-15 04:46:24 +08:00
103 lines
2.2 KiB
C
103 lines
2.2 KiB
C
/*
|
|
* Copyright 2014-2024 The GmSSL Project. All Rights Reserved.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the License); you may
|
|
* not use this file except in compliance with the License.
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*/
|
|
|
|
|
|
#include <gmssl/sm4.h>
|
|
#include <gmssl/mem.h>
|
|
#include <gmssl/error.h>
|
|
|
|
|
|
static void ctr_incr(uint8_t a[16])
|
|
{
|
|
int i;
|
|
for (i = 15; i >= 0; i--) {
|
|
a[i]++;
|
|
if (a[i]) break;
|
|
}
|
|
}
|
|
|
|
void sm4_ctr_encrypt(const SM4_KEY *key, uint8_t ctr[16], const uint8_t *in, size_t inlen, uint8_t *out)
|
|
{
|
|
uint8_t block[16];
|
|
size_t len;
|
|
|
|
while (inlen) {
|
|
len = inlen < 16 ? inlen : 16;
|
|
sm4_encrypt(key, ctr, block);
|
|
gmssl_memxor(out, in, block, len);
|
|
ctr_incr(ctr);
|
|
in += len;
|
|
out += len;
|
|
inlen -= len;
|
|
}
|
|
}
|
|
|
|
int sm4_ctr_encrypt_init(SM4_CTR_CTX *ctx,
|
|
const uint8_t key[SM4_BLOCK_SIZE], const uint8_t ctr[SM4_BLOCK_SIZE])
|
|
{
|
|
sm4_set_encrypt_key(&ctx->sm4_key, key);
|
|
memcpy(ctx->ctr, ctr, SM4_BLOCK_SIZE);
|
|
memset(ctx->block, 0, SM4_BLOCK_SIZE);
|
|
ctx->block_nbytes = 0;
|
|
return 1;
|
|
}
|
|
|
|
int sm4_ctr_encrypt_update(SM4_CTR_CTX *ctx,
|
|
const uint8_t *in, size_t inlen, uint8_t *out, size_t *outlen)
|
|
{
|
|
size_t left;
|
|
size_t nblocks;
|
|
size_t len;
|
|
|
|
if (ctx->block_nbytes >= SM4_BLOCK_SIZE) {
|
|
error_print();
|
|
return -1;
|
|
}
|
|
*outlen = 0;
|
|
if (ctx->block_nbytes) {
|
|
left = SM4_BLOCK_SIZE - ctx->block_nbytes;
|
|
if (inlen < left) {
|
|
memcpy(ctx->block + ctx->block_nbytes, in, inlen);
|
|
ctx->block_nbytes += inlen;
|
|
return 1;
|
|
}
|
|
memcpy(ctx->block + ctx->block_nbytes, in, left);
|
|
sm4_ctr_encrypt(&ctx->sm4_key, ctx->ctr, ctx->block, SM4_BLOCK_SIZE, out);
|
|
in += left;
|
|
inlen -= left;
|
|
out += SM4_BLOCK_SIZE;
|
|
*outlen += SM4_BLOCK_SIZE;
|
|
}
|
|
if (inlen >= SM4_BLOCK_SIZE) {
|
|
nblocks = inlen / SM4_BLOCK_SIZE;
|
|
len = nblocks * SM4_BLOCK_SIZE;
|
|
sm4_ctr_encrypt(&ctx->sm4_key, ctx->ctr, in, len, out);
|
|
in += len;
|
|
inlen -= len;
|
|
out += len;
|
|
*outlen += len;
|
|
}
|
|
if (inlen) {
|
|
memcpy(ctx->block, in, inlen);
|
|
}
|
|
ctx->block_nbytes = inlen;
|
|
return 1;
|
|
}
|
|
|
|
int sm4_ctr_encrypt_finish(SM4_CTR_CTX *ctx, uint8_t *out, size_t *outlen)
|
|
{
|
|
if (ctx->block_nbytes >= SM4_BLOCK_SIZE) {
|
|
error_print();
|
|
return -1;
|
|
}
|
|
sm4_ctr_encrypt(&ctx->sm4_key, ctx->ctr, ctx->block, ctx->block_nbytes, out);
|
|
*outlen = ctx->block_nbytes;
|
|
return 1;
|
|
}
|