xref: /freebsd/contrib/bearssl/tools/xmem.c (revision 7ef62cebc2f965b0f640263e179276928885e33d)
1 /*
2  * Copyright (c) 2016 Thomas Pornin <pornin@bolet.org>
3  *
4  * Permission is hereby granted, free of charge, to any person obtaining
5  * a copy of this software and associated documentation files (the
6  * "Software"), to deal in the Software without restriction, including
7  * without limitation the rights to use, copy, modify, merge, publish,
8  * distribute, sublicense, and/or sell copies of the Software, and to
9  * permit persons to whom the Software is furnished to do so, subject to
10  * the following conditions:
11  *
12  * The above copyright notice and this permission notice shall be
13  * included in all copies or substantial portions of the Software.
14  *
15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
16  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
17  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
18  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
19  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
20  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
21  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22  * SOFTWARE.
23  */
24 
25 #include "brssl.h"
26 
27 /* see brssl.h */
28 void *
29 xmalloc(size_t len)
30 {
31 	void *buf;
32 
33 	if (len == 0) {
34 		return NULL;
35 	}
36 	buf = malloc(len);
37 	if (buf == NULL) {
38 #ifndef _STANDALONE
39 		fprintf(stderr, "ERROR: could not allocate %lu byte(s)\n",
40 			(unsigned long)len);
41 		exit(EXIT_FAILURE);
42 #else
43 ;
44 #endif
45 	}
46 	return buf;
47 }
48 
49 /* see brssl.h */
50 void
51 xfree(void *buf)
52 {
53 	if (buf != NULL) {
54 		free(buf);
55 	}
56 }
57 
58 /* see brssl.h */
59 void *
60 xblobdup(const void *src, size_t len)
61 {
62 	void *buf;
63 
64 	buf = xmalloc(len);
65 	memcpy(buf, src, len);
66 	return buf;
67 }
68 
69 /* see brssl.h */
70 char *
71 xstrdup(const void *src)
72 {
73 	return xblobdup(src, strlen(src) + 1);
74 }
75 
76 /* see brssl.h */
77 br_x509_pkey *
78 xpkeydup(const br_x509_pkey *pk)
79 {
80 	br_x509_pkey *pk2;
81 
82 	pk2 = xmalloc(sizeof *pk2);
83 	pk2->key_type = pk->key_type;
84 	switch (pk->key_type) {
85 	case BR_KEYTYPE_RSA:
86 		pk2->key.rsa.n = xblobdup(pk->key.rsa.n, pk->key.rsa.nlen);
87 		pk2->key.rsa.nlen = pk->key.rsa.nlen;
88 		pk2->key.rsa.e = xblobdup(pk->key.rsa.e, pk->key.rsa.elen);
89 		pk2->key.rsa.elen = pk->key.rsa.elen;
90 		break;
91 	case BR_KEYTYPE_EC:
92 		pk2->key.ec.curve = pk->key.ec.curve;
93 		pk2->key.ec.q = xblobdup(pk->key.ec.q, pk->key.ec.qlen);
94 		pk2->key.ec.qlen = pk->key.ec.qlen;
95 		break;
96 	default:
97 #ifndef _STANDALONE
98 		fprintf(stderr, "Unknown public key type: %u\n",
99 			(unsigned)pk->key_type);
100 		exit(EXIT_FAILURE);
101 #else
102 ;
103 #endif
104 	}
105 	return pk2;
106 }
107 
108 /* see brssl.h */
109 void
110 xfreepkey(br_x509_pkey *pk)
111 {
112 	if (pk != NULL) {
113 		switch (pk->key_type) {
114 		case BR_KEYTYPE_RSA:
115 			xfree(pk->key.rsa.n);
116 			xfree(pk->key.rsa.e);
117 			break;
118 		case BR_KEYTYPE_EC:
119 			xfree(pk->key.ec.q);
120 			break;
121 		default:
122 #ifndef _STANDALONE
123 			fprintf(stderr, "Unknown public key type: %u\n",
124 				(unsigned)pk->key_type);
125 			exit(EXIT_FAILURE);
126 #else
127 ;
128 #endif
129 		}
130 		xfree(pk);
131 	}
132 }
133