mirror of
https://github.com/guanzhi/GmSSL.git
synced 2026-05-15 04:46:24 +08:00
153 lines
3.5 KiB
C
153 lines
3.5 KiB
C
/*
|
||
* Copyright 2014-2021 The GmSSL Project Authors. 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.
|
||
* You may obtain a copy of the License at
|
||
*
|
||
* http://www.apache.org/licenses/LICENSE-2.0
|
||
*
|
||
* Unless required by applicable law or agreed to in writing, software
|
||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
* See the License for the specific language governing permissions and
|
||
* limitations under the License.
|
||
*/
|
||
|
||
#include <stdio.h>
|
||
#include <string.h>
|
||
#include <stdlib.h>
|
||
#include <gmssl/oid.h>
|
||
#include <gmssl/x509.h>
|
||
#include <gmssl/rand.h>
|
||
#include <gmssl/error.h>
|
||
#include <gmssl/sm4.h>
|
||
#include <gmssl/cms.h>
|
||
|
||
static int test_cms_data(void)
|
||
{
|
||
uint8_t data[20];
|
||
uint8_t content_info[512];
|
||
size_t content_info_len = sizeof(content_info);
|
||
size_t len = 0;
|
||
int i;
|
||
|
||
memset(data, 'A', sizeof(data));
|
||
|
||
cms_content_info_set_data(content_info, &content_info_len, data, sizeof(data));
|
||
cms_content_info_print(stdout, content_info, content_info_len, 0, 0);
|
||
return 1;
|
||
}
|
||
|
||
static int test_cms_sign(void)
|
||
{
|
||
SM2_KEY sign_key;
|
||
X509_CERTIFICATE sign_cert;
|
||
uint8_t data[20];
|
||
uint8_t content_info[1024];
|
||
size_t content_info_len = 0;
|
||
|
||
FILE *key_fp = fopen("sign_key.pem", "r");
|
||
FILE *cert_fp = fopen("sign_cert.pem", "r");
|
||
|
||
if (sm2_private_key_from_pem(&sign_key, key_fp) != 1) {
|
||
error_print();
|
||
return -1;
|
||
}
|
||
if (x509_certificate_from_pem(&sign_cert, cert_fp) != 1) {
|
||
error_print();
|
||
return -1;
|
||
}
|
||
if (cms_sign(&sign_key, &sign_cert, 1, CMS_data, data, 20, NULL, NULL, 0, content_info, &content_info_len) != 1) {
|
||
error_print();
|
||
return -1;
|
||
}
|
||
cms_content_info_print(stdout, content_info, content_info_len, 0, 0);
|
||
|
||
return 1;
|
||
}
|
||
|
||
|
||
static int test_cms_enced_content_info(void)
|
||
{
|
||
uint8_t buf[512];
|
||
uint8_t *p = buf;
|
||
const uint8_t *cp = buf;
|
||
size_t len = 0;
|
||
uint8_t iv[16];
|
||
uint8_t enced_content[30];
|
||
|
||
if (cms_enced_content_info_to_der(OID_sm4_cbc, iv, sizeof(iv),
|
||
CMS_data, enced_content, sizeof(enced_content),
|
||
NULL, 0,
|
||
NULL, 0,
|
||
&p, &len) != 1) {
|
||
error_print();
|
||
return -1;
|
||
}
|
||
|
||
int content_type;
|
||
int enc_algor;
|
||
const uint8_t *enc_iv;
|
||
size_t enc_iv_len;
|
||
const uint8_t *penced_content;
|
||
size_t enced_content_len;
|
||
const uint8_t *shared_info1, *shared_info2;
|
||
size_t shared_info1_len, shared_info2_len;
|
||
|
||
if (cms_enced_content_info_from_der(&content_type,
|
||
&enc_algor, &enc_iv, &enc_iv_len,
|
||
&penced_content, &enced_content_len,
|
||
&shared_info1, &shared_info1_len,
|
||
&shared_info2, &shared_info2_len,
|
||
&cp, &len) != 1) {
|
||
error_print();
|
||
return -1;
|
||
}
|
||
|
||
|
||
|
||
return 1;
|
||
}
|
||
|
||
|
||
static int test_cms_encrypt(void)
|
||
{
|
||
uint8_t key[16];
|
||
uint8_t msg[] = "Hello world!";
|
||
uint8_t cbuf[512];
|
||
uint8_t mbuf[512];
|
||
size_t clen, mlen;
|
||
int content_type = 0;
|
||
const uint8_t *shared_info1 = NULL;
|
||
const uint8_t *shared_info2 = NULL;
|
||
size_t shared_info1_len, shared_info2_len;
|
||
|
||
if (cms_encrypt(key, msg, sizeof(msg), cbuf, &clen) != 1) {
|
||
error_print();
|
||
return -1;
|
||
}
|
||
|
||
format_bytes(stderr, 0, 0, "EncryptedData\n", cbuf, clen);
|
||
|
||
|
||
if (cms_decrypt(key, cbuf, clen, &content_type, mbuf, &mlen,
|
||
&shared_info1, &shared_info1_len,
|
||
&shared_info2, &shared_info2_len) != 1) {
|
||
error_print();
|
||
return -1;
|
||
}
|
||
|
||
return 1;
|
||
}
|
||
|
||
int main(void)
|
||
{
|
||
// 很可能x509_algor.c中有错误！
|
||
test_cms_enced_content_info();
|
||
//test_cms_encrypt();
|
||
//test_cms_data();
|
||
//test_cms_sign();
|
||
return 0;
|
||
}
|