xref: /illumos-gate/usr/src/cmd/smbsrv/test-crypt/test_encrypt.c (revision 4f0ce1da2aeae875aeb13640b681d0ad37a3036b)
1 /*
2  * This file and its contents are supplied under the terms of the
3  * Common Development and Distribution License ("CDDL"), version 1.0.
4  * You may only use this file in accordance with the terms of version
5  * 1.0 of the CDDL.
6  *
7  * A full copy of the text of the CDDL should have accompanied this
8  * source.  A copy of the CDDL is also available via the Internet at
9  * http://www.illumos.org/license/CDDL.
10  */
11 
12 /*
13  * Copyright 2021 Tintri by DDN, Inc. All rights reserved.
14  */
15 
16 #include <sys/types.h>
17 #include <smbsrv/smb_kcrypt.h>
18 #include <security/cryptoki.h>
19 #include <security/pkcs11.h>
20 
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <strings.h>
25 #include <unistd.h>
26 
27 #include "test_data.h"
28 #include "utils.h"
29 
30 /*
31  * Test program for the interfaces used in
32  * smb3_encrypt_reply()
33  */
34 int
do_encrypt(uint8_t * outbuf,size_t * outlen,const char * inbuf,size_t inlen,int mid)35 do_encrypt(uint8_t *outbuf, size_t *outlen,
36     const char *inbuf, size_t inlen, int mid)
37 {
38 	smb_enc_ctx_t ctx;
39 	uio_t uio_in;
40 	uio_t uio_out;
41 	iovec_t iov_in[4];
42 	iovec_t iov_out[4];
43 	int rc;
44 
45 	bzero(&ctx, sizeof (ctx));
46 	ctx.mech.mechanism = mid; // CKM_AES_CCM or CKM_AES_GCM
47 
48 	switch (mid) {
49 
50 	case CKM_AES_CCM:
51 		smb3_crypto_init_ccm_param(&ctx,
52 		    (uint8_t *)nonce, 11,
53 		    (uint8_t *)authdata, 16,
54 		    inlen);
55 		break;
56 
57 	case CKM_AES_GCM:
58 		smb3_crypto_init_gcm_param(&ctx,
59 		    (uint8_t *)nonce, 12,
60 		    (uint8_t *)authdata, 16);
61 		break;
62 
63 	default:
64 		return (1);
65 	}
66 
67 	rc = smb3_encrypt_init(&ctx,
68 	    (uint8_t *)keydata, 16);
69 	if (rc != 0)
70 		return (rc);
71 
72 	make_uio((void *)inbuf, inlen, &uio_in, iov_in, 4);
73 	make_uio(outbuf, *outlen, &uio_out, iov_out, 4);
74 	*outlen = uio_out.uio_resid;
75 
76 	rc = smb3_encrypt_uio(&ctx, &uio_in, &uio_out);
77 	*outlen -= uio_out.uio_resid;
78 
79 	smb3_enc_ctx_done(&ctx);
80 
81 	return (rc);
82 }
83 
84 uint8_t outbuf[CIPHER_DATA_LEN];
85 
86 void
test_encrypt(const uint8_t * ref,int mid)87 test_encrypt(const uint8_t *ref, int mid)
88 {
89 	size_t outlen;
90 	int rc;
91 
92 	outlen = sizeof (outbuf);
93 	rc = do_encrypt(outbuf, &outlen,
94 	    clear_data_ref, clear_data_len, mid);
95 	if (rc != 0) {
96 		printf("FAIL: encrypt rc= %d\n");
97 		return;
98 	}
99 
100 	if (outlen != CIPHER_DATA_LEN) {
101 		printf("FAIL: out len = %d (want %d)\n",
102 		    outlen, CIPHER_DATA_LEN);
103 		return;
104 	}
105 
106 	if (memcmp(outbuf, ref, CIPHER_DATA_LEN) != 0) {
107 		printf("FAIL: ciphertext:\n");
108 		hexdump(outbuf, CIPHER_DATA_LEN);
109 		return;
110 	}
111 
112 	printf("PASS mid=0x%x\n", mid);
113 }
114 
115 int
main(int argc,char * argv[])116 main(int argc, char *argv[])
117 {
118 
119 	test_encrypt(cipher_data_ccm, CKM_AES_CCM);
120 	test_encrypt(cipher_data_gcm, CKM_AES_GCM);
121 
122 	return (0);
123 }
124