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