1 /*
2 * CDDL HEADER START
3 *
4 * The contents of this file are subject to the terms of the
5 * Common Development and Distribution License (the "License").
6 * You may not use this file except in compliance with the License.
7 *
8 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9 * or http://www.opensolaris.org/os/licensing.
10 * See the License for the specific language governing permissions
11 * and limitations under the License.
12 *
13 * When distributing Covered Code, include this CDDL HEADER in each
14 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15 * If applicable, add the following below this CDDL HEADER, with the
16 * fields enclosed by brackets "[]" replaced with your own identifying
17 * information: Portions Copyright [yyyy] [name of copyright owner]
18 *
19 * CDDL HEADER END
20 */
21
22 /*
23 * Copyright (c) 2010, Oracle and/or its affiliates. All rights reserved.
24 */
25
26 /*
27 * Copyright 2019 Joyent, Inc.
28 */
29
30 #include <sys/types.h>
31
32 #include <stddef.h>
33 #include <stdlib.h>
34 #include <string.h>
35 #include <strings.h>
36 #include <alloca.h>
37 #include <stdio.h>
38 #include <unistd.h>
39 #include <dlfcn.h>
40 #include <thread.h>
41 #include <pthread.h>
42 #include <ctype.h>
43
44 #include <scsi/libsmp.h>
45 #include <scsi/libsmp_plugin.h>
46 #include "smp_impl.h"
47
48 __thread smp_errno_t _smp_errno;
49 __thread char _smp_errmsg[LIBSMP_ERRMSGLEN];
50
51 int
52 smp_assert(const char *expr, const char *file, int line)
53 {
54 char *msg;
55 size_t len;
56
57 len = snprintf(NULL, 0,
58 "ABORT: \"%s\", line %d: assertion failed: %s\n", file, line, expr);
59
60 msg = alloca(len + 1);
61
62 (void) snprintf(msg, len + 1,
63 "ABORT: \"%s\", line %d: assertion failed: %s\n", file, line, expr);
64
65 (void) write(STDERR_FILENO, msg, strlen(msg));
66
67 abort();
68 /*NOTREACHED*/
69 }
70
71 int
72 smp_set_errno(smp_errno_t err)
73 {
74 _smp_errno = err;
75 _smp_errmsg[0] = '\0';
76
77 return (-1);
78 }
79
80 /*
81 * Internal routine for setting both _smp_errno and _smp_errmsg. We save
82 * and restore the UNIX errno across this routing so the caller can use either
83 * smp_set_errno(), smp_error(), or smp_verror() without this value changing.
84 */
85 int
86 smp_verror(smp_errno_t err, const char *fmt, va_list ap)
87 {
88 size_t n;
89 char *errmsg;
90
91 /*
92 * To allow the existing error message to itself be used in an error
93 * message, we put the new error message into a buffer on the stack,
94 * and then copy it into lsh_errmsg. We also need to set the errno,
95 * but because the call to smp_set_errno() is destructive to
96 * lsh_errmsg, we do this after we print into our temporary buffer
97 * (in case _smp_errmsg is part of the error message) and before we
98 * copy the temporary buffer on to _smp_errmsg (to prevent our new
99 * message from being nuked by the call to smp_set_errno()).
100 */
101 errmsg = alloca(sizeof (_smp_errmsg));
102 (void) vsnprintf(errmsg, sizeof (_smp_errmsg), fmt, ap);
103 (void) smp_set_errno(err);
104
105 n = strlen(errmsg);
106
107 if (n != 0 && errmsg[n - 1] == '\n')
108 errmsg[n - 1] = '\0';
109
110 bcopy(errmsg, _smp_errmsg, n + 1);
111
112 return (-1);
113 }
114
115 int
116 smp_error(smp_errno_t err, const char *fmt, ...)
117 {
118 va_list ap;
119
120 if (fmt == NULL)
121 return (smp_set_errno(err));
122
123 va_start(ap, fmt);
124 err = smp_verror(err, fmt, ap);
125 va_end(ap);
126
127 return (err);
128 }
129
130 smp_errno_t
131 smp_errno(void)
132 {
133 return (_smp_errno);
134 }
135
136 const char *
137 smp_errmsg(void)
138 {
139 if (_smp_errmsg[0] == '\0')
140 (void) strlcpy(_smp_errmsg, smp_strerror(_smp_errno),
141 sizeof (_smp_errmsg));
142
143 return (_smp_errmsg);
144 }
145
146 /*ARGSUSED*/
147 void *
148 smp_alloc(size_t size)
149 {
150 void *mem;
151
152 if (size == 0) {
153 (void) smp_set_errno(ESMP_ZERO_LENGTH);
154 return (NULL);
155 }
156
157 if ((mem = malloc(size)) == NULL)
158 (void) smp_set_errno(ESMP_NOMEM);
159
160 return (mem);
161 }
162
163 void *
164 smp_zalloc(size_t size)
165 {
166 void *mem;
167
168 if ((mem = smp_alloc(size)) == NULL)
169 return (NULL);
170
171 bzero(mem, size);
172
173 return (mem);
174 }
175
176 char *
177 smp_strdup(const char *str)
178 {
179 size_t len = strlen(str);
180 char *dup = smp_alloc(len + 1);
181
182 if (dup == NULL)
183 return (NULL);
184
185 return (strcpy(dup, str));
186 }
187
188 void
189 smp_free(void *ptr)
190 {
191 free(ptr);
192 }
193
194 /*
195 * Trim any leading and/or trailing spaces from the fixed-length string
196 * argument and return a newly-allocated copy of it.
197 */
198 char *
199 smp_trim_strdup(const char *str, size_t len)
200 {
201 const char *p;
202 char *r;
203
204 for (p = str; p - str < len && isspace(*p); p++)
205 ;
206
207 len -= (p - str);
208
209 if (len == 0)
210 return (NULL);
211
212 for (str = p + len - 1; str > p && isspace(*str); str--, len--)
213 ;
214
215 if (len == 0)
216 return (NULL);
217
218 r = smp_alloc(len + 1);
219 if (r == NULL)
220 return (NULL);
221
222 bcopy(p, r, len);
223 r[len] = '\0';
224
225 return (r);
226 }
227
228 int
229 smp_init(int version)
230 {
231 if (version != LIBSMP_VERSION)
232 return (smp_error(ESMP_VERSION,
233 "library version %d does not match requested version %d",
234 LIBSMP_VERSION, version));
235
236 smp_engine_init();
237
238 return (0);
239 }
240
241 void
242 smp_fini(void)
243 {
244 smp_engine_fini();
245 }