xref: /NextBSD/cddl/contrib/opensolaris/lib/libzpool/common/kernel.c (revision 84d351007654069f9643c8e4b4802a7f5f08ee42)
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  * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
23  * Copyright (c) 2012, 2015 by Delphix. All rights reserved.
24  * Copyright (c) 2013, Joyent, Inc.  All rights reserved.
25  */
26 
27 #include <assert.h>
28 #include <fcntl.h>
29 #include <poll.h>
30 #include <stdio.h>
31 #include <stdlib.h>
32 #include <string.h>
33 #include <zlib.h>
34 #include <libgen.h>
35 #include <sys/spa.h>
36 #include <sys/stat.h>
37 #include <sys/processor.h>
38 #include <sys/zfs_context.h>
39 #include <sys/rrwlock.h>
40 #include <sys/zmod.h>
41 #include <sys/utsname.h>
42 #include <sys/systeminfo.h>
43 
44 /*
45  * Emulation of kernel services in userland.
46  */
47 
48 #ifndef __FreeBSD__
49 int aok;
50 #endif
51 uint64_t physmem;
52 vnode_t *rootdir = (vnode_t *)0xabcd1234;
53 char hw_serial[HW_HOSTID_LEN];
54 #ifdef illumos
55 kmutex_t cpu_lock;
56 #endif
57 
58 /* If set, all blocks read will be copied to the specified directory. */
59 char *vn_dumpdir = NULL;
60 
61 struct utsname utsname = {
62 	"userland", "libzpool", "1", "1", "na"
63 };
64 
65 /* this only exists to have its address taken */
66 struct proc p0;
67 
68 /*
69  * =========================================================================
70  * threads
71  * =========================================================================
72  */
73 /*ARGSUSED*/
74 kthread_t *
zk_thread_create(void (* func)(),void * arg)75 zk_thread_create(void (*func)(), void *arg)
76 {
77 	thread_t tid;
78 
79 	VERIFY(thr_create(0, 0, (void *(*)(void *))func, arg, THR_DETACHED,
80 	    &tid) == 0);
81 
82 	return ((void *)(uintptr_t)tid);
83 }
84 
85 /*
86  * =========================================================================
87  * kstats
88  * =========================================================================
89  */
90 /*ARGSUSED*/
91 kstat_t *
kstat_create(char * module,int instance,char * name,char * class,uchar_t type,ulong_t ndata,uchar_t ks_flag)92 kstat_create(char *module, int instance, char *name, char *class,
93     uchar_t type, ulong_t ndata, uchar_t ks_flag)
94 {
95 	return (NULL);
96 }
97 
98 /*ARGSUSED*/
99 void
kstat_install(kstat_t * ksp)100 kstat_install(kstat_t *ksp)
101 {}
102 
103 /*ARGSUSED*/
104 void
kstat_delete(kstat_t * ksp)105 kstat_delete(kstat_t *ksp)
106 {}
107 
108 /*
109  * =========================================================================
110  * mutexes
111  * =========================================================================
112  */
113 void
zmutex_init(kmutex_t * mp)114 zmutex_init(kmutex_t *mp)
115 {
116 	mp->m_owner = NULL;
117 	mp->initialized = B_TRUE;
118 	(void) _mutex_init(&mp->m_lock, USYNC_THREAD, NULL);
119 }
120 
121 void
zmutex_destroy(kmutex_t * mp)122 zmutex_destroy(kmutex_t *mp)
123 {
124 	ASSERT(mp->initialized == B_TRUE);
125 	ASSERT(mp->m_owner == NULL);
126 	(void) _mutex_destroy(&(mp)->m_lock);
127 	mp->m_owner = (void *)-1UL;
128 	mp->initialized = B_FALSE;
129 }
130 
131 int
zmutex_owned(kmutex_t * mp)132 zmutex_owned(kmutex_t *mp)
133 {
134 	ASSERT(mp->initialized == B_TRUE);
135 
136 	return (mp->m_owner == curthread);
137 }
138 
139 void
mutex_enter(kmutex_t * mp)140 mutex_enter(kmutex_t *mp)
141 {
142 	ASSERT(mp->initialized == B_TRUE);
143 	ASSERT(mp->m_owner != (void *)-1UL);
144 	ASSERT(mp->m_owner != curthread);
145 	VERIFY(mutex_lock(&mp->m_lock) == 0);
146 	ASSERT(mp->m_owner == NULL);
147 	mp->m_owner = curthread;
148 }
149 
150 int
mutex_tryenter(kmutex_t * mp)151 mutex_tryenter(kmutex_t *mp)
152 {
153 	ASSERT(mp->initialized == B_TRUE);
154 	ASSERT(mp->m_owner != (void *)-1UL);
155 	if (0 == mutex_trylock(&mp->m_lock)) {
156 		ASSERT(mp->m_owner == NULL);
157 		mp->m_owner = curthread;
158 		return (1);
159 	} else {
160 		return (0);
161 	}
162 }
163 
164 void
mutex_exit(kmutex_t * mp)165 mutex_exit(kmutex_t *mp)
166 {
167 	ASSERT(mp->initialized == B_TRUE);
168 	ASSERT(mutex_owner(mp) == curthread);
169 	mp->m_owner = NULL;
170 	VERIFY(mutex_unlock(&mp->m_lock) == 0);
171 }
172 
173 void *
mutex_owner(kmutex_t * mp)174 mutex_owner(kmutex_t *mp)
175 {
176 	ASSERT(mp->initialized == B_TRUE);
177 	return (mp->m_owner);
178 }
179 
180 /*
181  * =========================================================================
182  * rwlocks
183  * =========================================================================
184  */
185 /*ARGSUSED*/
186 void
rw_init(krwlock_t * rwlp,char * name,int type,void * arg)187 rw_init(krwlock_t *rwlp, char *name, int type, void *arg)
188 {
189 	rwlock_init(&rwlp->rw_lock, USYNC_THREAD, NULL);
190 	rwlp->rw_owner = NULL;
191 	rwlp->initialized = B_TRUE;
192 	rwlp->rw_count = 0;
193 }
194 
195 void
rw_destroy(krwlock_t * rwlp)196 rw_destroy(krwlock_t *rwlp)
197 {
198 	ASSERT(rwlp->rw_count == 0);
199 	rwlock_destroy(&rwlp->rw_lock);
200 	rwlp->rw_owner = (void *)-1UL;
201 	rwlp->initialized = B_FALSE;
202 }
203 
204 void
rw_enter(krwlock_t * rwlp,krw_t rw)205 rw_enter(krwlock_t *rwlp, krw_t rw)
206 {
207 	//ASSERT(!RW_LOCK_HELD(rwlp));
208 	ASSERT(rwlp->initialized == B_TRUE);
209 	ASSERT(rwlp->rw_owner != (void *)-1UL);
210 	ASSERT(rwlp->rw_owner != curthread);
211 
212 	if (rw == RW_READER) {
213 		VERIFY(rw_rdlock(&rwlp->rw_lock) == 0);
214 		ASSERT(rwlp->rw_count >= 0);
215 		atomic_add_int(&rwlp->rw_count, 1);
216 	} else {
217 		VERIFY(rw_wrlock(&rwlp->rw_lock) == 0);
218 		ASSERT(rwlp->rw_count == 0);
219 		rwlp->rw_count = -1;
220 		rwlp->rw_owner = curthread;
221 	}
222 }
223 
224 void
rw_exit(krwlock_t * rwlp)225 rw_exit(krwlock_t *rwlp)
226 {
227 	ASSERT(rwlp->initialized == B_TRUE);
228 	ASSERT(rwlp->rw_owner != (void *)-1UL);
229 
230 	if (rwlp->rw_owner == curthread) {
231 		/* Write locked. */
232 		ASSERT(rwlp->rw_count == -1);
233 		rwlp->rw_count = 0;
234 		rwlp->rw_owner = NULL;
235 	} else {
236 		/* Read locked. */
237 		ASSERT(rwlp->rw_count > 0);
238 		atomic_add_int(&rwlp->rw_count, -1);
239 	}
240 	VERIFY(rw_unlock(&rwlp->rw_lock) == 0);
241 }
242 
243 int
rw_tryenter(krwlock_t * rwlp,krw_t rw)244 rw_tryenter(krwlock_t *rwlp, krw_t rw)
245 {
246 	int rv;
247 
248 	ASSERT(rwlp->initialized == B_TRUE);
249 	ASSERT(rwlp->rw_owner != (void *)-1UL);
250 	ASSERT(rwlp->rw_owner != curthread);
251 
252 	if (rw == RW_READER)
253 		rv = rw_tryrdlock(&rwlp->rw_lock);
254 	else
255 		rv = rw_trywrlock(&rwlp->rw_lock);
256 
257 	if (rv == 0) {
258 		ASSERT(rwlp->rw_owner == NULL);
259 		if (rw == RW_READER) {
260 			ASSERT(rwlp->rw_count >= 0);
261 			atomic_add_int(&rwlp->rw_count, 1);
262 		} else {
263 			ASSERT(rwlp->rw_count == 0);
264 			rwlp->rw_count = -1;
265 			rwlp->rw_owner = curthread;
266 		}
267 		return (1);
268 	}
269 
270 	return (0);
271 }
272 
273 /*ARGSUSED*/
274 int
rw_tryupgrade(krwlock_t * rwlp)275 rw_tryupgrade(krwlock_t *rwlp)
276 {
277 	ASSERT(rwlp->initialized == B_TRUE);
278 	ASSERT(rwlp->rw_owner != (void *)-1UL);
279 
280 	return (0);
281 }
282 
283 int
rw_lock_held(krwlock_t * rwlp)284 rw_lock_held(krwlock_t *rwlp)
285 {
286 
287 	return (rwlp->rw_count != 0);
288 }
289 
290 /*
291  * =========================================================================
292  * condition variables
293  * =========================================================================
294  */
295 /*ARGSUSED*/
296 void
cv_init(kcondvar_t * cv,char * name,int type,void * arg)297 cv_init(kcondvar_t *cv, char *name, int type, void *arg)
298 {
299 	VERIFY(cond_init(cv, name, NULL) == 0);
300 }
301 
302 void
cv_destroy(kcondvar_t * cv)303 cv_destroy(kcondvar_t *cv)
304 {
305 	VERIFY(cond_destroy(cv) == 0);
306 }
307 
308 void
cv_wait(kcondvar_t * cv,kmutex_t * mp)309 cv_wait(kcondvar_t *cv, kmutex_t *mp)
310 {
311 	ASSERT(mutex_owner(mp) == curthread);
312 	mp->m_owner = NULL;
313 	int ret = cond_wait(cv, &mp->m_lock);
314 	VERIFY(ret == 0 || ret == EINTR);
315 	mp->m_owner = curthread;
316 }
317 
318 clock_t
cv_timedwait(kcondvar_t * cv,kmutex_t * mp,clock_t abstime)319 cv_timedwait(kcondvar_t *cv, kmutex_t *mp, clock_t abstime)
320 {
321 	int error;
322 	struct timespec ts;
323 	struct timeval tv;
324 	clock_t delta;
325 
326 	abstime += ddi_get_lbolt();
327 top:
328 	delta = abstime - ddi_get_lbolt();
329 	if (delta <= 0)
330 		return (-1);
331 
332 	if (gettimeofday(&tv, NULL) != 0)
333 		assert(!"gettimeofday() failed");
334 
335 	ts.tv_sec = tv.tv_sec + delta / hz;
336 	ts.tv_nsec = tv.tv_usec * 1000 + (delta % hz) * (NANOSEC / hz);
337 	ASSERT(ts.tv_nsec >= 0);
338 
339 	if (ts.tv_nsec >= NANOSEC) {
340 		ts.tv_sec++;
341 		ts.tv_nsec -= NANOSEC;
342 	}
343 
344 	ASSERT(mutex_owner(mp) == curthread);
345 	mp->m_owner = NULL;
346 	error = pthread_cond_timedwait(cv, &mp->m_lock, &ts);
347 	mp->m_owner = curthread;
348 
349 	if (error == EINTR)
350 		goto top;
351 
352 	if (error == ETIMEDOUT)
353 		return (-1);
354 
355 	ASSERT(error == 0);
356 
357 	return (1);
358 }
359 
360 /*ARGSUSED*/
361 clock_t
cv_timedwait_hires(kcondvar_t * cv,kmutex_t * mp,hrtime_t tim,hrtime_t res,int flag)362 cv_timedwait_hires(kcondvar_t *cv, kmutex_t *mp, hrtime_t tim, hrtime_t res,
363     int flag)
364 {
365 	int error;
366 	timestruc_t ts;
367 	hrtime_t delta;
368 
369 	ASSERT(flag == 0);
370 
371 top:
372 	delta = tim - gethrtime();
373 	if (delta <= 0)
374 		return (-1);
375 
376 	ts.tv_sec = delta / NANOSEC;
377 	ts.tv_nsec = delta % NANOSEC;
378 
379 	ASSERT(mutex_owner(mp) == curthread);
380 	mp->m_owner = NULL;
381 	error = pthread_cond_timedwait(cv, &mp->m_lock, &ts);
382 	mp->m_owner = curthread;
383 
384 	if (error == ETIMEDOUT)
385 		return (-1);
386 
387 	if (error == EINTR)
388 		goto top;
389 
390 	ASSERT(error == 0);
391 
392 	return (1);
393 }
394 
395 void
cv_signal(kcondvar_t * cv)396 cv_signal(kcondvar_t *cv)
397 {
398 	VERIFY(cond_signal(cv) == 0);
399 }
400 
401 void
cv_broadcast(kcondvar_t * cv)402 cv_broadcast(kcondvar_t *cv)
403 {
404 	VERIFY(cond_broadcast(cv) == 0);
405 }
406 
407 /*
408  * =========================================================================
409  * vnode operations
410  * =========================================================================
411  */
412 /*
413  * Note: for the xxxat() versions of these functions, we assume that the
414  * starting vp is always rootdir (which is true for spa_directory.c, the only
415  * ZFS consumer of these interfaces).  We assert this is true, and then emulate
416  * them by adding '/' in front of the path.
417  */
418 
419 /*ARGSUSED*/
420 int
vn_open(char * path,int x1,int flags,int mode,vnode_t ** vpp,int x2,int x3)421 vn_open(char *path, int x1, int flags, int mode, vnode_t **vpp, int x2, int x3)
422 {
423 	int fd;
424 	int dump_fd;
425 	vnode_t *vp;
426 	int old_umask;
427 	char realpath[MAXPATHLEN];
428 	struct stat64 st;
429 
430 	/*
431 	 * If we're accessing a real disk from userland, we need to use
432 	 * the character interface to avoid caching.  This is particularly
433 	 * important if we're trying to look at a real in-kernel storage
434 	 * pool from userland, e.g. via zdb, because otherwise we won't
435 	 * see the changes occurring under the segmap cache.
436 	 * On the other hand, the stupid character device returns zero
437 	 * for its size.  So -- gag -- we open the block device to get
438 	 * its size, and remember it for subsequent VOP_GETATTR().
439 	 */
440 	if (strncmp(path, "/dev/", 5) == 0) {
441 		char *dsk;
442 		fd = open64(path, O_RDONLY);
443 		if (fd == -1)
444 			return (errno);
445 		if (fstat64(fd, &st) == -1) {
446 			close(fd);
447 			return (errno);
448 		}
449 		close(fd);
450 		(void) sprintf(realpath, "%s", path);
451 		dsk = strstr(path, "/dsk/");
452 		if (dsk != NULL)
453 			(void) sprintf(realpath + (dsk - path) + 1, "r%s",
454 			    dsk + 1);
455 	} else {
456 		(void) sprintf(realpath, "%s", path);
457 		if (!(flags & FCREAT) && stat64(realpath, &st) == -1)
458 			return (errno);
459 	}
460 
461 	if (flags & FCREAT)
462 		old_umask = umask(0);
463 
464 	/*
465 	 * The construct 'flags - FREAD' conveniently maps combinations of
466 	 * FREAD and FWRITE to the corresponding O_RDONLY, O_WRONLY, and O_RDWR.
467 	 */
468 	fd = open64(realpath, flags - FREAD, mode);
469 
470 	if (flags & FCREAT)
471 		(void) umask(old_umask);
472 
473 	if (vn_dumpdir != NULL) {
474 		char dumppath[MAXPATHLEN];
475 		(void) snprintf(dumppath, sizeof (dumppath),
476 		    "%s/%s", vn_dumpdir, basename(realpath));
477 		dump_fd = open64(dumppath, O_CREAT | O_WRONLY, 0666);
478 		if (dump_fd == -1)
479 			return (errno);
480 	} else {
481 		dump_fd = -1;
482 	}
483 
484 	if (fd == -1)
485 		return (errno);
486 
487 	if (fstat64(fd, &st) == -1) {
488 		close(fd);
489 		return (errno);
490 	}
491 
492 	(void) fcntl(fd, F_SETFD, FD_CLOEXEC);
493 
494 	*vpp = vp = umem_zalloc(sizeof (vnode_t), UMEM_NOFAIL);
495 
496 	vp->v_fd = fd;
497 	vp->v_size = st.st_size;
498 	vp->v_path = spa_strdup(path);
499 	vp->v_dump_fd = dump_fd;
500 
501 	return (0);
502 }
503 
504 /*ARGSUSED*/
505 int
vn_openat(char * path,int x1,int flags,int mode,vnode_t ** vpp,int x2,int x3,vnode_t * startvp,int fd)506 vn_openat(char *path, int x1, int flags, int mode, vnode_t **vpp, int x2,
507     int x3, vnode_t *startvp, int fd)
508 {
509 	char *realpath = umem_alloc(strlen(path) + 2, UMEM_NOFAIL);
510 	int ret;
511 
512 	ASSERT(startvp == rootdir);
513 	(void) sprintf(realpath, "/%s", path);
514 
515 	/* fd ignored for now, need if want to simulate nbmand support */
516 	ret = vn_open(realpath, x1, flags, mode, vpp, x2, x3);
517 
518 	umem_free(realpath, strlen(path) + 2);
519 
520 	return (ret);
521 }
522 
523 /*ARGSUSED*/
524 int
vn_rdwr(int uio,vnode_t * vp,void * addr,ssize_t len,offset_t offset,int x1,int x2,rlim64_t x3,void * x4,ssize_t * residp)525 vn_rdwr(int uio, vnode_t *vp, void *addr, ssize_t len, offset_t offset,
526     int x1, int x2, rlim64_t x3, void *x4, ssize_t *residp)
527 {
528 	ssize_t iolen, split;
529 
530 	if (uio == UIO_READ) {
531 		iolen = pread64(vp->v_fd, addr, len, offset);
532 		if (vp->v_dump_fd != -1) {
533 			int status =
534 			    pwrite64(vp->v_dump_fd, addr, iolen, offset);
535 			ASSERT(status != -1);
536 		}
537 	} else {
538 		/*
539 		 * To simulate partial disk writes, we split writes into two
540 		 * system calls so that the process can be killed in between.
541 		 */
542 		int sectors = len >> SPA_MINBLOCKSHIFT;
543 		split = (sectors > 0 ? rand() % sectors : 0) <<
544 		    SPA_MINBLOCKSHIFT;
545 		iolen = pwrite64(vp->v_fd, addr, split, offset);
546 		iolen += pwrite64(vp->v_fd, (char *)addr + split,
547 		    len - split, offset + split);
548 	}
549 
550 	if (iolen == -1)
551 		return (errno);
552 	if (residp)
553 		*residp = len - iolen;
554 	else if (iolen != len)
555 		return (EIO);
556 	return (0);
557 }
558 
559 void
vn_close(vnode_t * vp,int openflag,cred_t * cr,kthread_t * td)560 vn_close(vnode_t *vp, int openflag, cred_t *cr, kthread_t *td)
561 {
562 	close(vp->v_fd);
563 	if (vp->v_dump_fd != -1)
564 		close(vp->v_dump_fd);
565 	spa_strfree(vp->v_path);
566 	umem_free(vp, sizeof (vnode_t));
567 }
568 
569 /*
570  * At a minimum we need to update the size since vdev_reopen()
571  * will no longer call vn_openat().
572  */
573 int
fop_getattr(vnode_t * vp,vattr_t * vap)574 fop_getattr(vnode_t *vp, vattr_t *vap)
575 {
576 	struct stat64 st;
577 
578 	if (fstat64(vp->v_fd, &st) == -1) {
579 		close(vp->v_fd);
580 		return (errno);
581 	}
582 
583 	vap->va_size = st.st_size;
584 	return (0);
585 }
586 
587 #ifdef ZFS_DEBUG
588 
589 /*
590  * =========================================================================
591  * Figure out which debugging statements to print
592  * =========================================================================
593  */
594 
595 static char *dprintf_string;
596 static int dprintf_print_all;
597 
598 int
dprintf_find_string(const char * string)599 dprintf_find_string(const char *string)
600 {
601 	char *tmp_str = dprintf_string;
602 	int len = strlen(string);
603 
604 	/*
605 	 * Find out if this is a string we want to print.
606 	 * String format: file1.c,function_name1,file2.c,file3.c
607 	 */
608 
609 	while (tmp_str != NULL) {
610 		if (strncmp(tmp_str, string, len) == 0 &&
611 		    (tmp_str[len] == ',' || tmp_str[len] == '\0'))
612 			return (1);
613 		tmp_str = strchr(tmp_str, ',');
614 		if (tmp_str != NULL)
615 			tmp_str++; /* Get rid of , */
616 	}
617 	return (0);
618 }
619 
620 void
dprintf_setup(int * argc,char ** argv)621 dprintf_setup(int *argc, char **argv)
622 {
623 	int i, j;
624 
625 	/*
626 	 * Debugging can be specified two ways: by setting the
627 	 * environment variable ZFS_DEBUG, or by including a
628 	 * "debug=..."  argument on the command line.  The command
629 	 * line setting overrides the environment variable.
630 	 */
631 
632 	for (i = 1; i < *argc; i++) {
633 		int len = strlen("debug=");
634 		/* First look for a command line argument */
635 		if (strncmp("debug=", argv[i], len) == 0) {
636 			dprintf_string = argv[i] + len;
637 			/* Remove from args */
638 			for (j = i; j < *argc; j++)
639 				argv[j] = argv[j+1];
640 			argv[j] = NULL;
641 			(*argc)--;
642 		}
643 	}
644 
645 	if (dprintf_string == NULL) {
646 		/* Look for ZFS_DEBUG environment variable */
647 		dprintf_string = getenv("ZFS_DEBUG");
648 	}
649 
650 	/*
651 	 * Are we just turning on all debugging?
652 	 */
653 	if (dprintf_find_string("on"))
654 		dprintf_print_all = 1;
655 
656 	if (dprintf_string != NULL)
657 		zfs_flags |= ZFS_DEBUG_DPRINTF;
658 }
659 
660 int
sysctl_handle_64(SYSCTL_HANDLER_ARGS)661 sysctl_handle_64(SYSCTL_HANDLER_ARGS)
662 {
663 	return (0);
664 }
665 
666 /*
667  * =========================================================================
668  * debug printfs
669  * =========================================================================
670  */
671 void
__dprintf(const char * file,const char * func,int line,const char * fmt,...)672 __dprintf(const char *file, const char *func, int line, const char *fmt, ...)
673 {
674 	const char *newfile;
675 	va_list adx;
676 
677 	/*
678 	 * Get rid of annoying "../common/" prefix to filename.
679 	 */
680 	newfile = strrchr(file, '/');
681 	if (newfile != NULL) {
682 		newfile = newfile + 1; /* Get rid of leading / */
683 	} else {
684 		newfile = file;
685 	}
686 
687 	if (dprintf_print_all ||
688 	    dprintf_find_string(newfile) ||
689 	    dprintf_find_string(func)) {
690 		/* Print out just the function name if requested */
691 		flockfile(stdout);
692 		if (dprintf_find_string("pid"))
693 			(void) printf("%d ", getpid());
694 		if (dprintf_find_string("tid"))
695 			(void) printf("%lu ", thr_self());
696 #if 0
697 		if (dprintf_find_string("cpu"))
698 			(void) printf("%u ", getcpuid());
699 #endif
700 		if (dprintf_find_string("time"))
701 			(void) printf("%llu ", gethrtime());
702 		if (dprintf_find_string("long"))
703 			(void) printf("%s, line %d: ", newfile, line);
704 		(void) printf("%s: ", func);
705 		va_start(adx, fmt);
706 		(void) vprintf(fmt, adx);
707 		va_end(adx);
708 		funlockfile(stdout);
709 	}
710 }
711 
712 #endif /* ZFS_DEBUG */
713 
714 /*
715  * =========================================================================
716  * cmn_err() and panic()
717  * =========================================================================
718  */
719 static char ce_prefix[CE_IGNORE][10] = { "", "NOTICE: ", "WARNING: ", "" };
720 static char ce_suffix[CE_IGNORE][2] = { "", "\n", "\n", "" };
721 
722 void
vpanic(const char * fmt,va_list adx)723 vpanic(const char *fmt, va_list adx)
724 {
725 	(void) fprintf(stderr, "error: ");
726 	(void) vfprintf(stderr, fmt, adx);
727 	(void) fprintf(stderr, "\n");
728 
729 	abort();	/* think of it as a "user-level crash dump" */
730 }
731 
732 void
panic(const char * fmt,...)733 panic(const char *fmt, ...)
734 {
735 	va_list adx;
736 
737 	va_start(adx, fmt);
738 	vpanic(fmt, adx);
739 	va_end(adx);
740 }
741 
742 void
vcmn_err(int ce,const char * fmt,va_list adx)743 vcmn_err(int ce, const char *fmt, va_list adx)
744 {
745 	if (ce == CE_PANIC)
746 		vpanic(fmt, adx);
747 	if (ce != CE_NOTE) {	/* suppress noise in userland stress testing */
748 		(void) fprintf(stderr, "%s", ce_prefix[ce]);
749 		(void) vfprintf(stderr, fmt, adx);
750 		(void) fprintf(stderr, "%s", ce_suffix[ce]);
751 	}
752 }
753 
754 /*PRINTFLIKE2*/
755 void
cmn_err(int ce,const char * fmt,...)756 cmn_err(int ce, const char *fmt, ...)
757 {
758 	va_list adx;
759 
760 	va_start(adx, fmt);
761 	vcmn_err(ce, fmt, adx);
762 	va_end(adx);
763 }
764 
765 /*
766  * =========================================================================
767  * kobj interfaces
768  * =========================================================================
769  */
770 struct _buf *
kobj_open_file(char * name)771 kobj_open_file(char *name)
772 {
773 	struct _buf *file;
774 	vnode_t *vp;
775 
776 	/* set vp as the _fd field of the file */
777 	if (vn_openat(name, UIO_SYSSPACE, FREAD, 0, &vp, 0, 0, rootdir,
778 	    -1) != 0)
779 		return ((void *)-1UL);
780 
781 	file = umem_zalloc(sizeof (struct _buf), UMEM_NOFAIL);
782 	file->_fd = (intptr_t)vp;
783 	return (file);
784 }
785 
786 int
kobj_read_file(struct _buf * file,char * buf,unsigned size,unsigned off)787 kobj_read_file(struct _buf *file, char *buf, unsigned size, unsigned off)
788 {
789 	ssize_t resid;
790 
791 	vn_rdwr(UIO_READ, (vnode_t *)file->_fd, buf, size, (offset_t)off,
792 	    UIO_SYSSPACE, 0, 0, 0, &resid);
793 
794 	return (size - resid);
795 }
796 
797 void
kobj_close_file(struct _buf * file)798 kobj_close_file(struct _buf *file)
799 {
800 	vn_close((vnode_t *)file->_fd, 0, NULL, NULL);
801 	umem_free(file, sizeof (struct _buf));
802 }
803 
804 int
kobj_get_filesize(struct _buf * file,uint64_t * size)805 kobj_get_filesize(struct _buf *file, uint64_t *size)
806 {
807 	struct stat64 st;
808 	vnode_t *vp = (vnode_t *)file->_fd;
809 
810 	if (fstat64(vp->v_fd, &st) == -1) {
811 		vn_close(vp, 0, NULL, NULL);
812 		return (errno);
813 	}
814 	*size = st.st_size;
815 	return (0);
816 }
817 
818 /*
819  * =========================================================================
820  * misc routines
821  * =========================================================================
822  */
823 
824 void
delay(clock_t ticks)825 delay(clock_t ticks)
826 {
827 	poll(0, 0, ticks * (1000 / hz));
828 }
829 
830 #if 0
831 /*
832  * Find highest one bit set.
833  *	Returns bit number + 1 of highest bit that is set, otherwise returns 0.
834  */
835 int
836 highbit64(uint64_t i)
837 {
838 	int h = 1;
839 
840 	if (i == 0)
841 		return (0);
842 	if (i & 0xffffffff00000000ULL) {
843 		h += 32; i >>= 32;
844 	}
845 	if (i & 0xffff0000) {
846 		h += 16; i >>= 16;
847 	}
848 	if (i & 0xff00) {
849 		h += 8; i >>= 8;
850 	}
851 	if (i & 0xf0) {
852 		h += 4; i >>= 4;
853 	}
854 	if (i & 0xc) {
855 		h += 2; i >>= 2;
856 	}
857 	if (i & 0x2) {
858 		h += 1;
859 	}
860 	return (h);
861 }
862 #endif
863 
864 static int random_fd = -1, urandom_fd = -1;
865 
866 static int
random_get_bytes_common(uint8_t * ptr,size_t len,int fd)867 random_get_bytes_common(uint8_t *ptr, size_t len, int fd)
868 {
869 	size_t resid = len;
870 	ssize_t bytes;
871 
872 	ASSERT(fd != -1);
873 
874 	while (resid != 0) {
875 		bytes = read(fd, ptr, resid);
876 		ASSERT3S(bytes, >=, 0);
877 		ptr += bytes;
878 		resid -= bytes;
879 	}
880 
881 	return (0);
882 }
883 
884 int
random_get_bytes(uint8_t * ptr,size_t len)885 random_get_bytes(uint8_t *ptr, size_t len)
886 {
887 	return (random_get_bytes_common(ptr, len, random_fd));
888 }
889 
890 int
random_get_pseudo_bytes(uint8_t * ptr,size_t len)891 random_get_pseudo_bytes(uint8_t *ptr, size_t len)
892 {
893 	return (random_get_bytes_common(ptr, len, urandom_fd));
894 }
895 
896 int
ddi_strtoul(const char * hw_serial,char ** nptr,int base,unsigned long * result)897 ddi_strtoul(const char *hw_serial, char **nptr, int base, unsigned long *result)
898 {
899 	char *end;
900 
901 	*result = strtoul(hw_serial, &end, base);
902 	if (*result == 0)
903 		return (errno);
904 	return (0);
905 }
906 
907 int
ddi_strtoull(const char * str,char ** nptr,int base,u_longlong_t * result)908 ddi_strtoull(const char *str, char **nptr, int base, u_longlong_t *result)
909 {
910 	char *end;
911 
912 	*result = strtoull(str, &end, base);
913 	if (*result == 0)
914 		return (errno);
915 	return (0);
916 }
917 
918 #ifdef illumos
919 /* ARGSUSED */
920 cyclic_id_t
cyclic_add(cyc_handler_t * hdlr,cyc_time_t * when)921 cyclic_add(cyc_handler_t *hdlr, cyc_time_t *when)
922 {
923 	return (1);
924 }
925 
926 /* ARGSUSED */
927 void
cyclic_remove(cyclic_id_t id)928 cyclic_remove(cyclic_id_t id)
929 {
930 }
931 
932 /* ARGSUSED */
933 int
cyclic_reprogram(cyclic_id_t id,hrtime_t expiration)934 cyclic_reprogram(cyclic_id_t id, hrtime_t expiration)
935 {
936 	return (1);
937 }
938 #endif
939 
940 /*
941  * =========================================================================
942  * kernel emulation setup & teardown
943  * =========================================================================
944  */
945 static int
umem_out_of_memory(void)946 umem_out_of_memory(void)
947 {
948 	char errmsg[] = "out of memory -- generating core dump\n";
949 
950 	write(fileno(stderr), errmsg, sizeof (errmsg));
951 	abort();
952 	return (0);
953 }
954 
955 void
kernel_init(int mode)956 kernel_init(int mode)
957 {
958 	extern uint_t rrw_tsd_key;
959 
960 	umem_nofail_callback(umem_out_of_memory);
961 
962 	physmem = sysconf(_SC_PHYS_PAGES);
963 
964 	dprintf("physmem = %llu pages (%.2f GB)\n", physmem,
965 	    (double)physmem * sysconf(_SC_PAGE_SIZE) / (1ULL << 30));
966 
967 	(void) snprintf(hw_serial, sizeof (hw_serial), "%lu",
968 	    (mode & FWRITE) ? (unsigned long)gethostid() : 0);
969 
970 	VERIFY((random_fd = open("/dev/random", O_RDONLY)) != -1);
971 	VERIFY((urandom_fd = open("/dev/urandom", O_RDONLY)) != -1);
972 
973 	system_taskq_init();
974 
975 #ifdef illumos
976 	mutex_init(&cpu_lock, NULL, MUTEX_DEFAULT, NULL);
977 #endif
978 
979 	spa_init(mode);
980 
981 	tsd_create(&rrw_tsd_key, rrw_tsd_destroy);
982 }
983 
984 void
kernel_fini(void)985 kernel_fini(void)
986 {
987 	spa_fini();
988 
989 	system_taskq_fini();
990 
991 	close(random_fd);
992 	close(urandom_fd);
993 
994 	random_fd = -1;
995 	urandom_fd = -1;
996 }
997 
998 int
z_uncompress(void * dst,size_t * dstlen,const void * src,size_t srclen)999 z_uncompress(void *dst, size_t *dstlen, const void *src, size_t srclen)
1000 {
1001 	int ret;
1002 	uLongf len = *dstlen;
1003 
1004 	if ((ret = uncompress(dst, &len, src, srclen)) == Z_OK)
1005 		*dstlen = (size_t)len;
1006 
1007 	return (ret);
1008 }
1009 
1010 int
z_compress_level(void * dst,size_t * dstlen,const void * src,size_t srclen,int level)1011 z_compress_level(void *dst, size_t *dstlen, const void *src, size_t srclen,
1012     int level)
1013 {
1014 	int ret;
1015 	uLongf len = *dstlen;
1016 
1017 	if ((ret = compress2(dst, &len, src, srclen, level)) == Z_OK)
1018 		*dstlen = (size_t)len;
1019 
1020 	return (ret);
1021 }
1022 
1023 uid_t
crgetuid(cred_t * cr)1024 crgetuid(cred_t *cr)
1025 {
1026 	return (0);
1027 }
1028 
1029 uid_t
crgetruid(cred_t * cr)1030 crgetruid(cred_t *cr)
1031 {
1032 	return (0);
1033 }
1034 
1035 gid_t
crgetgid(cred_t * cr)1036 crgetgid(cred_t *cr)
1037 {
1038 	return (0);
1039 }
1040 
1041 int
crgetngroups(cred_t * cr)1042 crgetngroups(cred_t *cr)
1043 {
1044 	return (0);
1045 }
1046 
1047 gid_t *
crgetgroups(cred_t * cr)1048 crgetgroups(cred_t *cr)
1049 {
1050 	return (NULL);
1051 }
1052 
1053 int
zfs_secpolicy_snapshot_perms(const char * name,cred_t * cr)1054 zfs_secpolicy_snapshot_perms(const char *name, cred_t *cr)
1055 {
1056 	return (0);
1057 }
1058 
1059 int
zfs_secpolicy_rename_perms(const char * from,const char * to,cred_t * cr)1060 zfs_secpolicy_rename_perms(const char *from, const char *to, cred_t *cr)
1061 {
1062 	return (0);
1063 }
1064 
1065 int
zfs_secpolicy_destroy_perms(const char * name,cred_t * cr)1066 zfs_secpolicy_destroy_perms(const char *name, cred_t *cr)
1067 {
1068 	return (0);
1069 }
1070 
1071 ksiddomain_t *
ksid_lookupdomain(const char * dom)1072 ksid_lookupdomain(const char *dom)
1073 {
1074 	ksiddomain_t *kd;
1075 
1076 	kd = umem_zalloc(sizeof (ksiddomain_t), UMEM_NOFAIL);
1077 	kd->kd_name = spa_strdup(dom);
1078 	return (kd);
1079 }
1080 
1081 void
ksiddomain_rele(ksiddomain_t * ksid)1082 ksiddomain_rele(ksiddomain_t *ksid)
1083 {
1084 	spa_strfree(ksid->kd_name);
1085 	umem_free(ksid, sizeof (ksiddomain_t));
1086 }
1087 
1088 /*
1089  * Do not change the length of the returned string; it must be freed
1090  * with strfree().
1091  */
1092 char *
kmem_asprintf(const char * fmt,...)1093 kmem_asprintf(const char *fmt, ...)
1094 {
1095 	int size;
1096 	va_list adx;
1097 	char *buf;
1098 
1099 	va_start(adx, fmt);
1100 	size = vsnprintf(NULL, 0, fmt, adx) + 1;
1101 	va_end(adx);
1102 
1103 	buf = kmem_alloc(size, KM_SLEEP);
1104 
1105 	va_start(adx, fmt);
1106 	size = vsnprintf(buf, size, fmt, adx);
1107 	va_end(adx);
1108 
1109 	return (buf);
1110 }
1111 
1112 /* ARGSUSED */
1113 int
zfs_onexit_fd_hold(int fd,minor_t * minorp)1114 zfs_onexit_fd_hold(int fd, minor_t *minorp)
1115 {
1116 	*minorp = 0;
1117 	return (0);
1118 }
1119 
1120 /* ARGSUSED */
1121 void
zfs_onexit_fd_rele(int fd)1122 zfs_onexit_fd_rele(int fd)
1123 {
1124 }
1125 
1126 /* ARGSUSED */
1127 int
zfs_onexit_add_cb(minor_t minor,void (* func)(void *),void * data,uint64_t * action_handle)1128 zfs_onexit_add_cb(minor_t minor, void (*func)(void *), void *data,
1129     uint64_t *action_handle)
1130 {
1131 	return (0);
1132 }
1133 
1134 /* ARGSUSED */
1135 int
zfs_onexit_del_cb(minor_t minor,uint64_t action_handle,boolean_t fire)1136 zfs_onexit_del_cb(minor_t minor, uint64_t action_handle, boolean_t fire)
1137 {
1138 	return (0);
1139 }
1140 
1141 /* ARGSUSED */
1142 int
zfs_onexit_cb_data(minor_t minor,uint64_t action_handle,void ** data)1143 zfs_onexit_cb_data(minor_t minor, uint64_t action_handle, void **data)
1144 {
1145 	return (0);
1146 }
1147 
1148 #ifdef __FreeBSD__
1149 /* ARGSUSED */
1150 int
zvol_create_minors(const char * name)1151 zvol_create_minors(const char *name)
1152 {
1153 	return (0);
1154 }
1155 #endif
1156 
1157 #ifdef illumos
1158 void
bioinit(buf_t * bp)1159 bioinit(buf_t *bp)
1160 {
1161 	bzero(bp, sizeof (buf_t));
1162 }
1163 
1164 void
biodone(buf_t * bp)1165 biodone(buf_t *bp)
1166 {
1167 	if (bp->b_iodone != NULL) {
1168 		(*(bp->b_iodone))(bp);
1169 		return;
1170 	}
1171 	ASSERT((bp->b_flags & B_DONE) == 0);
1172 	bp->b_flags |= B_DONE;
1173 }
1174 
1175 void
bioerror(buf_t * bp,int error)1176 bioerror(buf_t *bp, int error)
1177 {
1178 	ASSERT(bp != NULL);
1179 	ASSERT(error >= 0);
1180 
1181 	if (error != 0) {
1182 		bp->b_flags |= B_ERROR;
1183 	} else {
1184 		bp->b_flags &= ~B_ERROR;
1185 	}
1186 	bp->b_error = error;
1187 }
1188 
1189 
1190 int
geterror(struct buf * bp)1191 geterror(struct buf *bp)
1192 {
1193 	int error = 0;
1194 
1195 	if (bp->b_flags & B_ERROR) {
1196 		error = bp->b_error;
1197 		if (!error)
1198 			error = EIO;
1199 	}
1200 	return (error);
1201 }
1202 #endif
1203