xref: /freebsd/sys/kern/kern_fail.c (revision 0a713948)
1cfeb7489SZachary Loafman /*-
24d846d26SWarner Losh  * SPDX-License-Identifier: BSD-2-Clause
38a36da99SPedro F. Giffuni  *
4cfeb7489SZachary Loafman  * Copyright (c) 2009 Isilon Inc http://www.isilon.com/
5cfeb7489SZachary Loafman  *
6cfeb7489SZachary Loafman  * Redistribution and use in source and binary forms, with or without
7cfeb7489SZachary Loafman  * modification, are permitted provided that the following conditions
8cfeb7489SZachary Loafman  * are met:
9cfeb7489SZachary Loafman  * 1. Redistributions of source code must retain the above copyright
10cfeb7489SZachary Loafman  *    notice, this list of conditions and the following disclaimer.
11cfeb7489SZachary Loafman  * 2. Redistributions in binary form must reproduce the above copyright
12cfeb7489SZachary Loafman  *    notice, this list of conditions and the following disclaimer in the
13cfeb7489SZachary Loafman  *    documentation and/or other materials provided with the distribution.
14cfeb7489SZachary Loafman  *
15cfeb7489SZachary Loafman  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16cfeb7489SZachary Loafman  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17cfeb7489SZachary Loafman  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18cfeb7489SZachary Loafman  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19cfeb7489SZachary Loafman  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20cfeb7489SZachary Loafman  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21cfeb7489SZachary Loafman  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22cfeb7489SZachary Loafman  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23cfeb7489SZachary Loafman  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24cfeb7489SZachary Loafman  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25cfeb7489SZachary Loafman  * SUCH DAMAGE.
26cfeb7489SZachary Loafman  */
27cfeb7489SZachary Loafman /**
28cfeb7489SZachary Loafman  * @file
29cfeb7489SZachary Loafman  *
30cfeb7489SZachary Loafman  * fail(9) Facility.
31cfeb7489SZachary Loafman  *
32cfeb7489SZachary Loafman  * @ingroup failpoint_private
33cfeb7489SZachary Loafman  */
34cfeb7489SZachary Loafman /**
35cfeb7489SZachary Loafman  * @defgroup failpoint fail(9) Facility
36cfeb7489SZachary Loafman  *
37cfeb7489SZachary Loafman  * Failpoints allow for injecting fake errors into running code on the fly,
38cfeb7489SZachary Loafman  * without modifying code or recompiling with flags.  Failpoints are always
39cfeb7489SZachary Loafman  * present, and are very efficient when disabled.  Failpoints are described
40cfeb7489SZachary Loafman  * in man fail(9).
41cfeb7489SZachary Loafman  */
42cfeb7489SZachary Loafman /**
43cfeb7489SZachary Loafman  * @defgroup failpoint_private Private fail(9) Implementation functions
44cfeb7489SZachary Loafman  *
45cfeb7489SZachary Loafman  * Private implementations for the actual failpoint code.
46cfeb7489SZachary Loafman  *
47cfeb7489SZachary Loafman  * @ingroup failpoint
48cfeb7489SZachary Loafman  */
49cfeb7489SZachary Loafman /**
50cfeb7489SZachary Loafman  * @addtogroup failpoint_private
51cfeb7489SZachary Loafman  * @{
52cfeb7489SZachary Loafman  */
53cfeb7489SZachary Loafman 
54cfeb7489SZachary Loafman #include <sys/cdefs.h>
5582e17adcSConrad Meyer #include "opt_stack.h"
5682e17adcSConrad Meyer 
570894a9bcSMatthew D Fleming #include <sys/ctype.h>
58cfeb7489SZachary Loafman #include <sys/errno.h>
59cfeb7489SZachary Loafman #include <sys/fail.h>
60cfeb7489SZachary Loafman #include <sys/kernel.h>
61cfeb7489SZachary Loafman #include <sys/libkern.h>
6270e20d4eSConrad Meyer #include <sys/limits.h>
63cfeb7489SZachary Loafman #include <sys/lock.h>
64cfeb7489SZachary Loafman #include <sys/malloc.h>
65cfeb7489SZachary Loafman #include <sys/mutex.h>
66b5bd50aeSMatthew D Fleming #include <sys/proc.h>
67cfeb7489SZachary Loafman #include <sys/sbuf.h>
6870e20d4eSConrad Meyer #include <sys/sleepqueue.h>
6970e20d4eSConrad Meyer #include <sys/sx.h>
7070e20d4eSConrad Meyer #include <sys/sysctl.h>
7170e20d4eSConrad Meyer #include <sys/types.h>
72cfeb7489SZachary Loafman 
7370e20d4eSConrad Meyer #include <machine/atomic.h>
74cfeb7489SZachary Loafman #include <machine/stdarg.h>
75cfeb7489SZachary Loafman 
76cfeb7489SZachary Loafman #ifdef ILOG_DEFINE_FOR_FILE
77cfeb7489SZachary Loafman ILOG_DEFINE_FOR_FILE(L_ISI_FAIL_POINT, L_ILOG, fail_point);
78cfeb7489SZachary Loafman #endif
79cfeb7489SZachary Loafman 
80d745c852SEd Schouten static MALLOC_DEFINE(M_FAIL_POINT, "Fail Points", "fail points system");
81cfeb7489SZachary Loafman #define fp_free(ptr) free(ptr, M_FAIL_POINT)
82cfeb7489SZachary Loafman #define fp_malloc(size, flags) malloc((size), M_FAIL_POINT, (flags))
8370e20d4eSConrad Meyer #define fs_free(ptr) fp_free(ptr)
8470e20d4eSConrad Meyer #define fs_malloc() fp_malloc(sizeof(struct fail_point_setting), \
8570e20d4eSConrad Meyer     M_WAITOK | M_ZERO)
86cfeb7489SZachary Loafman 
8770e20d4eSConrad Meyer /**
8870e20d4eSConrad Meyer  * These define the wchans that are used for sleeping, pausing respectively.
8970e20d4eSConrad Meyer  * They are chosen arbitrarily but need to be distinct to the failpoint and
9070e20d4eSConrad Meyer  * the sleep/pause distinction.
9170e20d4eSConrad Meyer  */
9270e20d4eSConrad Meyer #define FP_SLEEP_CHANNEL(fp) (void*)(fp)
9370e20d4eSConrad Meyer #define FP_PAUSE_CHANNEL(fp) __DEVOLATILE(void*, &fp->fp_setting)
9470e20d4eSConrad Meyer 
9570e20d4eSConrad Meyer /**
9670e20d4eSConrad Meyer  * Don't allow more than this many entries in a fail point set by sysctl.
9770e20d4eSConrad Meyer  * The 99.99...% case is to have 1 entry.  I can't imagine having this many
9870e20d4eSConrad Meyer  * entries, so it should not limit us.  Saves on re-mallocs while holding
9970e20d4eSConrad Meyer  * a non-sleepable lock.
10070e20d4eSConrad Meyer  */
10170e20d4eSConrad Meyer #define FP_MAX_ENTRY_COUNT 20
10270e20d4eSConrad Meyer 
10370e20d4eSConrad Meyer /* Used to drain sbufs to the sysctl output */
10470e20d4eSConrad Meyer int fail_sysctl_drain_func(void *, const char *, int);
10570e20d4eSConrad Meyer 
10670e20d4eSConrad Meyer /* Head of tailq of struct fail_point_entry */
10770e20d4eSConrad Meyer TAILQ_HEAD(fail_point_entry_queue, fail_point_entry);
10870e20d4eSConrad Meyer 
10970e20d4eSConrad Meyer /**
11070e20d4eSConrad Meyer  * fp entries garbage list; outstanding entries are cleaned up in the
11170e20d4eSConrad Meyer  * garbage collector
11270e20d4eSConrad Meyer  */
11370e20d4eSConrad Meyer STAILQ_HEAD(fail_point_setting_garbage, fail_point_setting);
11470e20d4eSConrad Meyer static struct fail_point_setting_garbage fp_setting_garbage =
11570e20d4eSConrad Meyer         STAILQ_HEAD_INITIALIZER(fp_setting_garbage);
11670e20d4eSConrad Meyer static struct mtx mtx_garbage_list;
11770e20d4eSConrad Meyer MTX_SYSINIT(mtx_garbage_list, &mtx_garbage_list, "fail point garbage mtx",
11870e20d4eSConrad Meyer         MTX_SPIN);
11970e20d4eSConrad Meyer 
12070e20d4eSConrad Meyer static struct sx sx_fp_set;
12170e20d4eSConrad Meyer SX_SYSINIT(sx_fp_set, &sx_fp_set, "fail point set sx");
122cfeb7489SZachary Loafman 
123d26b7945SMatthew D Fleming /**
124d26b7945SMatthew D Fleming  * Failpoint types.
125d26b7945SMatthew D Fleming  * Don't change these without changing fail_type_strings in fail.c.
126d26b7945SMatthew D Fleming  * @ingroup failpoint_private
127d26b7945SMatthew D Fleming  */
128d26b7945SMatthew D Fleming enum fail_point_t {
129d26b7945SMatthew D Fleming 	FAIL_POINT_OFF,		/**< don't fail */
130d26b7945SMatthew D Fleming 	FAIL_POINT_PANIC,	/**< panic */
131d26b7945SMatthew D Fleming 	FAIL_POINT_RETURN,	/**< return an errorcode */
132d26b7945SMatthew D Fleming 	FAIL_POINT_BREAK,	/**< break into the debugger */
133d26b7945SMatthew D Fleming 	FAIL_POINT_PRINT,	/**< print a message */
134d26b7945SMatthew D Fleming 	FAIL_POINT_SLEEP,	/**< sleep for some msecs */
13570e20d4eSConrad Meyer 	FAIL_POINT_PAUSE,	/**< sleep until failpoint is set to off */
13670e20d4eSConrad Meyer 	FAIL_POINT_YIELD,	/**< yield the cpu */
13770e20d4eSConrad Meyer 	FAIL_POINT_DELAY,	/**< busy wait the cpu */
13870e20d4eSConrad Meyer 	FAIL_POINT_NUMTYPES,
13970e20d4eSConrad Meyer 	FAIL_POINT_INVALID = -1
140d26b7945SMatthew D Fleming };
141d26b7945SMatthew D Fleming 
1420894a9bcSMatthew D Fleming static struct {
1430894a9bcSMatthew D Fleming 	const char *name;
1440894a9bcSMatthew D Fleming 	int	nmlen;
1450894a9bcSMatthew D Fleming } fail_type_strings[] = {
1460894a9bcSMatthew D Fleming #define	FP_TYPE_NM_LEN(s)	{ s, sizeof(s) - 1 }
1470894a9bcSMatthew D Fleming 	[FAIL_POINT_OFF] =	FP_TYPE_NM_LEN("off"),
1480894a9bcSMatthew D Fleming 	[FAIL_POINT_PANIC] =	FP_TYPE_NM_LEN("panic"),
1490894a9bcSMatthew D Fleming 	[FAIL_POINT_RETURN] =	FP_TYPE_NM_LEN("return"),
1500894a9bcSMatthew D Fleming 	[FAIL_POINT_BREAK] =	FP_TYPE_NM_LEN("break"),
1510894a9bcSMatthew D Fleming 	[FAIL_POINT_PRINT] =	FP_TYPE_NM_LEN("print"),
1520894a9bcSMatthew D Fleming 	[FAIL_POINT_SLEEP] =	FP_TYPE_NM_LEN("sleep"),
15370e20d4eSConrad Meyer 	[FAIL_POINT_PAUSE] =	FP_TYPE_NM_LEN("pause"),
15470e20d4eSConrad Meyer 	[FAIL_POINT_YIELD] =	FP_TYPE_NM_LEN("yield"),
15570e20d4eSConrad Meyer 	[FAIL_POINT_DELAY] =	FP_TYPE_NM_LEN("delay"),
156d26b7945SMatthew D Fleming };
157d26b7945SMatthew D Fleming 
15870e20d4eSConrad Meyer #define FE_COUNT_UNTRACKED (INT_MIN)
15970e20d4eSConrad Meyer 
160d26b7945SMatthew D Fleming /**
161d26b7945SMatthew D Fleming  * Internal structure tracking a single term of a complete failpoint.
162d26b7945SMatthew D Fleming  * @ingroup failpoint_private
163d26b7945SMatthew D Fleming  */
164d26b7945SMatthew D Fleming struct fail_point_entry {
16570e20d4eSConrad Meyer 	volatile bool	fe_stale;
166d26b7945SMatthew D Fleming 	enum fail_point_t	fe_type;	/**< type of entry */
167d26b7945SMatthew D Fleming 	int		fe_arg;		/**< argument to type (e.g. return value) */
168d26b7945SMatthew D Fleming 	int		fe_prob;	/**< likelihood of firing in millionths */
169508d8569SMark Johnston 	int32_t		fe_count;	/**< number of times to fire, -1 means infinite */
170b5bd50aeSMatthew D Fleming 	pid_t		fe_pid;		/**< only fail for this process */
17170e20d4eSConrad Meyer 	struct fail_point	*fe_parent;	/**< backpointer to fp */
17270e20d4eSConrad Meyer 	TAILQ_ENTRY(fail_point_entry)	fe_entries; /**< next entry ptr */
173d26b7945SMatthew D Fleming };
174d26b7945SMatthew D Fleming 
17570e20d4eSConrad Meyer struct fail_point_setting {
17670e20d4eSConrad Meyer 	STAILQ_ENTRY(fail_point_setting) fs_garbage_link;
17770e20d4eSConrad Meyer 	struct fail_point_entry_queue fp_entry_queue;
17870e20d4eSConrad Meyer 	struct fail_point * fs_parent;
17970e20d4eSConrad Meyer 	struct mtx feq_mtx; /* Gives fail_point_pause something to do.  */
18070e20d4eSConrad Meyer };
181cfeb7489SZachary Loafman 
182cfeb7489SZachary Loafman /**
183cfeb7489SZachary Loafman  * Defines stating the equivalent of probablilty one (100%)
184cfeb7489SZachary Loafman  */
185cfeb7489SZachary Loafman enum {
186cfeb7489SZachary Loafman 	PROB_MAX = 1000000,	/* probability between zero and this number */
18770e20d4eSConrad Meyer 	PROB_DIGITS = 6		/* number of zero's in above number */
188cfeb7489SZachary Loafman };
189cfeb7489SZachary Loafman 
19070e20d4eSConrad Meyer /* Get a ref on an fp's fp_setting */
19170e20d4eSConrad Meyer static inline struct fail_point_setting *fail_point_setting_get_ref(
19270e20d4eSConrad Meyer         struct fail_point *fp);
19370e20d4eSConrad Meyer /* Release a ref on an fp_setting */
19470e20d4eSConrad Meyer static inline void fail_point_setting_release_ref(struct fail_point *fp);
19570e20d4eSConrad Meyer /* Allocate and initialize a struct fail_point_setting */
19670e20d4eSConrad Meyer static struct fail_point_setting *fail_point_setting_new(struct
19770e20d4eSConrad Meyer         fail_point *);
19870e20d4eSConrad Meyer /* Free a struct fail_point_setting */
19970e20d4eSConrad Meyer static void fail_point_setting_destroy(struct fail_point_setting *fp_setting);
20070e20d4eSConrad Meyer /* Allocate and initialize a struct fail_point_entry */
20170e20d4eSConrad Meyer static struct fail_point_entry *fail_point_entry_new(struct
20270e20d4eSConrad Meyer         fail_point_setting *);
20370e20d4eSConrad Meyer /* Free a struct fail_point_entry */
20470e20d4eSConrad Meyer static void fail_point_entry_destroy(struct fail_point_entry *fp_entry);
20570e20d4eSConrad Meyer /* Append fp setting to garbage list */
20670e20d4eSConrad Meyer static inline void fail_point_setting_garbage_append(
20770e20d4eSConrad Meyer         struct fail_point_setting *fp_setting);
20870e20d4eSConrad Meyer /* Swap fp's setting with fp_setting_new */
20970e20d4eSConrad Meyer static inline struct fail_point_setting *
21070e20d4eSConrad Meyer         fail_point_swap_settings(struct fail_point *fp,
21170e20d4eSConrad Meyer         struct fail_point_setting *fp_setting_new);
21270e20d4eSConrad Meyer /* Free up any zero-ref setting in the garbage queue */
21370e20d4eSConrad Meyer static void fail_point_garbage_collect(void);
21470e20d4eSConrad Meyer /* If this fail point's setting are empty, then swap it out to NULL. */
21570e20d4eSConrad Meyer static inline void fail_point_eval_swap_out(struct fail_point *fp,
21670e20d4eSConrad Meyer         struct fail_point_setting *fp_setting);
21770e20d4eSConrad Meyer 
21870e20d4eSConrad Meyer bool
fail_point_is_off(struct fail_point * fp)21970e20d4eSConrad Meyer fail_point_is_off(struct fail_point *fp)
22070e20d4eSConrad Meyer {
22170e20d4eSConrad Meyer 	bool return_val;
22270e20d4eSConrad Meyer 	struct fail_point_setting *fp_setting;
22370e20d4eSConrad Meyer 	struct fail_point_entry *ent;
22470e20d4eSConrad Meyer 
22570e20d4eSConrad Meyer 	return_val = true;
22670e20d4eSConrad Meyer 
22770e20d4eSConrad Meyer 	fp_setting = fail_point_setting_get_ref(fp);
22870e20d4eSConrad Meyer 	if (fp_setting != NULL) {
22970e20d4eSConrad Meyer 		TAILQ_FOREACH(ent, &fp_setting->fp_entry_queue,
23070e20d4eSConrad Meyer 		    fe_entries) {
23170e20d4eSConrad Meyer 			if (!ent->fe_stale) {
23270e20d4eSConrad Meyer 				return_val = false;
23370e20d4eSConrad Meyer 				break;
23470e20d4eSConrad Meyer 			}
23570e20d4eSConrad Meyer 		}
23670e20d4eSConrad Meyer 	}
23770e20d4eSConrad Meyer 	fail_point_setting_release_ref(fp);
23870e20d4eSConrad Meyer 
23970e20d4eSConrad Meyer 	return (return_val);
24070e20d4eSConrad Meyer }
24170e20d4eSConrad Meyer 
24270e20d4eSConrad Meyer /* Allocate and initialize a struct fail_point_setting */
24370e20d4eSConrad Meyer static struct fail_point_setting *
fail_point_setting_new(struct fail_point * fp)24470e20d4eSConrad Meyer fail_point_setting_new(struct fail_point *fp)
24570e20d4eSConrad Meyer {
24670e20d4eSConrad Meyer 	struct fail_point_setting *fs_new;
24770e20d4eSConrad Meyer 
24870e20d4eSConrad Meyer 	fs_new = fs_malloc();
24970e20d4eSConrad Meyer 	fs_new->fs_parent = fp;
25070e20d4eSConrad Meyer 	TAILQ_INIT(&fs_new->fp_entry_queue);
25170e20d4eSConrad Meyer 	mtx_init(&fs_new->feq_mtx, "fail point entries", NULL, MTX_SPIN);
25270e20d4eSConrad Meyer 
25370e20d4eSConrad Meyer 	fail_point_setting_garbage_append(fs_new);
25470e20d4eSConrad Meyer 
25570e20d4eSConrad Meyer 	return (fs_new);
25670e20d4eSConrad Meyer }
25770e20d4eSConrad Meyer 
25870e20d4eSConrad Meyer /* Free a struct fail_point_setting */
25970e20d4eSConrad Meyer static void
fail_point_setting_destroy(struct fail_point_setting * fp_setting)26070e20d4eSConrad Meyer fail_point_setting_destroy(struct fail_point_setting *fp_setting)
26170e20d4eSConrad Meyer {
26270e20d4eSConrad Meyer 	struct fail_point_entry *ent;
26370e20d4eSConrad Meyer 
26470e20d4eSConrad Meyer 	while (!TAILQ_EMPTY(&fp_setting->fp_entry_queue)) {
26570e20d4eSConrad Meyer 		ent = TAILQ_FIRST(&fp_setting->fp_entry_queue);
26670e20d4eSConrad Meyer 		TAILQ_REMOVE(&fp_setting->fp_entry_queue, ent, fe_entries);
26770e20d4eSConrad Meyer 		fail_point_entry_destroy(ent);
26870e20d4eSConrad Meyer 	}
26970e20d4eSConrad Meyer 
27070e20d4eSConrad Meyer 	fs_free(fp_setting);
27170e20d4eSConrad Meyer }
27270e20d4eSConrad Meyer 
27370e20d4eSConrad Meyer /* Allocate and initialize a struct fail_point_entry */
27470e20d4eSConrad Meyer static struct fail_point_entry *
fail_point_entry_new(struct fail_point_setting * fp_setting)27570e20d4eSConrad Meyer fail_point_entry_new(struct fail_point_setting *fp_setting)
27670e20d4eSConrad Meyer {
27770e20d4eSConrad Meyer 	struct fail_point_entry *fp_entry;
27870e20d4eSConrad Meyer 
27970e20d4eSConrad Meyer 	fp_entry = fp_malloc(sizeof(struct fail_point_entry),
28070e20d4eSConrad Meyer 	        M_WAITOK | M_ZERO);
28170e20d4eSConrad Meyer 	fp_entry->fe_parent = fp_setting->fs_parent;
28270e20d4eSConrad Meyer 	fp_entry->fe_prob = PROB_MAX;
28370e20d4eSConrad Meyer 	fp_entry->fe_pid = NO_PID;
28470e20d4eSConrad Meyer 	fp_entry->fe_count = FE_COUNT_UNTRACKED;
28570e20d4eSConrad Meyer 	TAILQ_INSERT_TAIL(&fp_setting->fp_entry_queue, fp_entry,
28670e20d4eSConrad Meyer 	        fe_entries);
28770e20d4eSConrad Meyer 
28870e20d4eSConrad Meyer 	return (fp_entry);
28970e20d4eSConrad Meyer }
29070e20d4eSConrad Meyer 
29170e20d4eSConrad Meyer /* Free a struct fail_point_entry */
29270e20d4eSConrad Meyer static void
fail_point_entry_destroy(struct fail_point_entry * fp_entry)29370e20d4eSConrad Meyer fail_point_entry_destroy(struct fail_point_entry *fp_entry)
29470e20d4eSConrad Meyer {
29570e20d4eSConrad Meyer 
29670e20d4eSConrad Meyer 	fp_free(fp_entry);
29770e20d4eSConrad Meyer }
29870e20d4eSConrad Meyer 
29970e20d4eSConrad Meyer /* Get a ref on an fp's fp_setting */
30070e20d4eSConrad Meyer static inline struct fail_point_setting *
fail_point_setting_get_ref(struct fail_point * fp)30170e20d4eSConrad Meyer fail_point_setting_get_ref(struct fail_point *fp)
30270e20d4eSConrad Meyer {
30370e20d4eSConrad Meyer 	struct fail_point_setting *fp_setting;
30470e20d4eSConrad Meyer 
30570e20d4eSConrad Meyer 	/* Invariant: if we have a ref, our pointer to fp_setting is safe */
30670e20d4eSConrad Meyer 	atomic_add_acq_32(&fp->fp_ref_cnt, 1);
30770e20d4eSConrad Meyer 	fp_setting = fp->fp_setting;
30870e20d4eSConrad Meyer 
30970e20d4eSConrad Meyer 	return (fp_setting);
31070e20d4eSConrad Meyer }
31170e20d4eSConrad Meyer 
31270e20d4eSConrad Meyer /* Release a ref on an fp_setting */
31370e20d4eSConrad Meyer static inline void
fail_point_setting_release_ref(struct fail_point * fp)31470e20d4eSConrad Meyer fail_point_setting_release_ref(struct fail_point *fp)
31570e20d4eSConrad Meyer {
31670e20d4eSConrad Meyer 
31770e20d4eSConrad Meyer 	KASSERT(&fp->fp_ref_cnt > 0, ("Attempting to deref w/no refs"));
31870e20d4eSConrad Meyer 	atomic_subtract_rel_32(&fp->fp_ref_cnt, 1);
31970e20d4eSConrad Meyer }
32070e20d4eSConrad Meyer 
32170e20d4eSConrad Meyer /* Append fp entries to fp garbage list */
32270e20d4eSConrad Meyer static inline void
fail_point_setting_garbage_append(struct fail_point_setting * fp_setting)32370e20d4eSConrad Meyer fail_point_setting_garbage_append(struct fail_point_setting *fp_setting)
32470e20d4eSConrad Meyer {
32570e20d4eSConrad Meyer 
32670e20d4eSConrad Meyer 	mtx_lock_spin(&mtx_garbage_list);
32770e20d4eSConrad Meyer 	STAILQ_INSERT_TAIL(&fp_setting_garbage, fp_setting,
32870e20d4eSConrad Meyer 	        fs_garbage_link);
32970e20d4eSConrad Meyer 	mtx_unlock_spin(&mtx_garbage_list);
33070e20d4eSConrad Meyer }
33170e20d4eSConrad Meyer 
33270e20d4eSConrad Meyer /* Swap fp's entries with fp_setting_new */
33370e20d4eSConrad Meyer static struct fail_point_setting *
fail_point_swap_settings(struct fail_point * fp,struct fail_point_setting * fp_setting_new)33470e20d4eSConrad Meyer fail_point_swap_settings(struct fail_point *fp,
33570e20d4eSConrad Meyer         struct fail_point_setting *fp_setting_new)
33670e20d4eSConrad Meyer {
33770e20d4eSConrad Meyer 	struct fail_point_setting *fp_setting_old;
33870e20d4eSConrad Meyer 
33970e20d4eSConrad Meyer 	fp_setting_old = fp->fp_setting;
34070e20d4eSConrad Meyer 	fp->fp_setting = fp_setting_new;
34170e20d4eSConrad Meyer 
34270e20d4eSConrad Meyer 	return (fp_setting_old);
34370e20d4eSConrad Meyer }
34470e20d4eSConrad Meyer 
34570e20d4eSConrad Meyer static inline void
fail_point_eval_swap_out(struct fail_point * fp,struct fail_point_setting * fp_setting)34670e20d4eSConrad Meyer fail_point_eval_swap_out(struct fail_point *fp,
34770e20d4eSConrad Meyer         struct fail_point_setting *fp_setting)
34870e20d4eSConrad Meyer {
34970e20d4eSConrad Meyer 
35070e20d4eSConrad Meyer 	/* We may have already been swapped out and replaced; ignore. */
35170e20d4eSConrad Meyer 	if (fp->fp_setting == fp_setting)
35270e20d4eSConrad Meyer 		fail_point_swap_settings(fp, NULL);
35370e20d4eSConrad Meyer }
35470e20d4eSConrad Meyer 
35570e20d4eSConrad Meyer /* Free up any zero-ref entries in the garbage queue */
35670e20d4eSConrad Meyer static void
fail_point_garbage_collect(void)357508d8569SMark Johnston fail_point_garbage_collect(void)
35870e20d4eSConrad Meyer {
35970e20d4eSConrad Meyer 	struct fail_point_setting *fs_current, *fs_next;
36070e20d4eSConrad Meyer 	struct fail_point_setting_garbage fp_ents_free_list;
36170e20d4eSConrad Meyer 
36270e20d4eSConrad Meyer 	/**
36370e20d4eSConrad Meyer 	  * We will transfer the entries to free to fp_ents_free_list while holding
36470e20d4eSConrad Meyer 	  * the spin mutex, then free it after we drop the lock. This avoids
36570e20d4eSConrad Meyer 	  * triggering witness due to sleepable mutexes in the memory
36670e20d4eSConrad Meyer 	  * allocator.
36770e20d4eSConrad Meyer 	  */
36870e20d4eSConrad Meyer 	STAILQ_INIT(&fp_ents_free_list);
36970e20d4eSConrad Meyer 
37070e20d4eSConrad Meyer 	mtx_lock_spin(&mtx_garbage_list);
37170e20d4eSConrad Meyer 	STAILQ_FOREACH_SAFE(fs_current, &fp_setting_garbage, fs_garbage_link,
37270e20d4eSConrad Meyer 	    fs_next) {
37370e20d4eSConrad Meyer 		if (fs_current->fs_parent->fp_setting != fs_current &&
37470e20d4eSConrad Meyer 		        fs_current->fs_parent->fp_ref_cnt == 0) {
37570e20d4eSConrad Meyer 			STAILQ_REMOVE(&fp_setting_garbage, fs_current,
37670e20d4eSConrad Meyer 			        fail_point_setting, fs_garbage_link);
37770e20d4eSConrad Meyer 			STAILQ_INSERT_HEAD(&fp_ents_free_list, fs_current,
37870e20d4eSConrad Meyer 			        fs_garbage_link);
37970e20d4eSConrad Meyer 		}
38070e20d4eSConrad Meyer 	}
38170e20d4eSConrad Meyer 	mtx_unlock_spin(&mtx_garbage_list);
38270e20d4eSConrad Meyer 
38370e20d4eSConrad Meyer 	STAILQ_FOREACH_SAFE(fs_current, &fp_ents_free_list, fs_garbage_link,
38470e20d4eSConrad Meyer 	        fs_next)
38570e20d4eSConrad Meyer 		fail_point_setting_destroy(fs_current);
38670e20d4eSConrad Meyer }
38770e20d4eSConrad Meyer 
38870e20d4eSConrad Meyer /* Drain out all refs from this fail point */
38970e20d4eSConrad Meyer static inline void
fail_point_drain(struct fail_point * fp,int expected_ref)39070e20d4eSConrad Meyer fail_point_drain(struct fail_point *fp, int expected_ref)
39170e20d4eSConrad Meyer {
39270e20d4eSConrad Meyer 	struct fail_point_setting *entries;
39370e20d4eSConrad Meyer 
39470e20d4eSConrad Meyer 	entries = fail_point_swap_settings(fp, NULL);
39570e20d4eSConrad Meyer 	/**
39670e20d4eSConrad Meyer 	 * We have unpaused all threads; so we will wait no longer
39770e20d4eSConrad Meyer 	 * than the time taken for the longest remaining sleep, or
39870e20d4eSConrad Meyer 	 * the length of time of a long-running code block.
39970e20d4eSConrad Meyer 	 */
40070e20d4eSConrad Meyer 	while (fp->fp_ref_cnt > expected_ref) {
40170e20d4eSConrad Meyer 		wakeup(FP_PAUSE_CHANNEL(fp));
40270e20d4eSConrad Meyer 		tsleep(&fp, PWAIT, "fail_point_drain", hz / 100);
40370e20d4eSConrad Meyer 	}
404bf2276f3SJohn Baldwin 	if (fp->fp_callout)
405bf2276f3SJohn Baldwin 		callout_drain(fp->fp_callout);
40670e20d4eSConrad Meyer 	fail_point_swap_settings(fp, entries);
40770e20d4eSConrad Meyer }
40870e20d4eSConrad Meyer 
40970e20d4eSConrad Meyer static inline void
fail_point_pause(struct fail_point * fp,enum fail_point_return_code * pret,struct mtx * mtx_sleep)41070e20d4eSConrad Meyer fail_point_pause(struct fail_point *fp, enum fail_point_return_code *pret,
41170e20d4eSConrad Meyer         struct mtx *mtx_sleep)
41270e20d4eSConrad Meyer {
41370e20d4eSConrad Meyer 
41470e20d4eSConrad Meyer 	if (fp->fp_pre_sleep_fn)
41570e20d4eSConrad Meyer 		fp->fp_pre_sleep_fn(fp->fp_pre_sleep_arg);
41670e20d4eSConrad Meyer 
41770e20d4eSConrad Meyer 	msleep_spin(FP_PAUSE_CHANNEL(fp), mtx_sleep, "failpt", 0);
41870e20d4eSConrad Meyer 
41970e20d4eSConrad Meyer 	if (fp->fp_post_sleep_fn)
42070e20d4eSConrad Meyer 		fp->fp_post_sleep_fn(fp->fp_post_sleep_arg);
42170e20d4eSConrad Meyer }
42270e20d4eSConrad Meyer 
42370e20d4eSConrad Meyer static inline void
fail_point_sleep(struct fail_point * fp,int msecs,enum fail_point_return_code * pret)42470e20d4eSConrad Meyer fail_point_sleep(struct fail_point *fp, int msecs,
42570e20d4eSConrad Meyer         enum fail_point_return_code *pret)
42670e20d4eSConrad Meyer {
42770e20d4eSConrad Meyer 	int timo;
42870e20d4eSConrad Meyer 
42970e20d4eSConrad Meyer 	/* Convert from millisecs to ticks, rounding up */
4305abb3b74SConrad Meyer 	timo = howmany((int64_t)msecs * hz, 1000L);
43170e20d4eSConrad Meyer 
43270e20d4eSConrad Meyer 	if (timo > 0) {
43370e20d4eSConrad Meyer 		if (!(fp->fp_flags & FAIL_POINT_USE_TIMEOUT_PATH)) {
43470e20d4eSConrad Meyer 			if (fp->fp_pre_sleep_fn)
43570e20d4eSConrad Meyer 				fp->fp_pre_sleep_fn(fp->fp_pre_sleep_arg);
43670e20d4eSConrad Meyer 
43770e20d4eSConrad Meyer 			tsleep(FP_SLEEP_CHANNEL(fp), PWAIT, "failpt", timo);
43870e20d4eSConrad Meyer 
43970e20d4eSConrad Meyer 			if (fp->fp_post_sleep_fn)
44070e20d4eSConrad Meyer 				fp->fp_post_sleep_fn(fp->fp_post_sleep_arg);
44170e20d4eSConrad Meyer 		} else {
44270e20d4eSConrad Meyer 			if (fp->fp_pre_sleep_fn)
44370e20d4eSConrad Meyer 				fp->fp_pre_sleep_fn(fp->fp_pre_sleep_arg);
44470e20d4eSConrad Meyer 
445bf2276f3SJohn Baldwin 			callout_reset(fp->fp_callout, timo,
446bf2276f3SJohn Baldwin 			    fp->fp_post_sleep_fn, fp->fp_post_sleep_arg);
44770e20d4eSConrad Meyer 			*pret = FAIL_POINT_RC_QUEUED;
44870e20d4eSConrad Meyer 		}
44970e20d4eSConrad Meyer 	}
45070e20d4eSConrad Meyer }
45170e20d4eSConrad Meyer 
45270e20d4eSConrad Meyer static char *parse_fail_point(struct fail_point_setting *, char *);
45370e20d4eSConrad Meyer static char *parse_term(struct fail_point_setting *, char *);
454cfeb7489SZachary Loafman static char *parse_number(int *out_units, int *out_decimal, char *);
455cfeb7489SZachary Loafman static char *parse_type(struct fail_point_entry *, char *);
456cfeb7489SZachary Loafman 
457cfeb7489SZachary Loafman /**
458cfeb7489SZachary Loafman  * Initialize a fail_point.  The name is formed in a printf-like fashion
459cfeb7489SZachary Loafman  * from "fmt" and subsequent arguments.  This function is generally used
460cfeb7489SZachary Loafman  * for custom failpoints located at odd places in the sysctl tree, and is
461cfeb7489SZachary Loafman  * not explicitly needed for standard in-line-declared failpoints.
462cfeb7489SZachary Loafman  *
463cfeb7489SZachary Loafman  * @ingroup failpoint
464cfeb7489SZachary Loafman  */
465cfeb7489SZachary Loafman void
fail_point_init(struct fail_point * fp,const char * fmt,...)466cfeb7489SZachary Loafman fail_point_init(struct fail_point *fp, const char *fmt, ...)
467cfeb7489SZachary Loafman {
468cfeb7489SZachary Loafman 	va_list ap;
469cfeb7489SZachary Loafman 	char *name;
470cfeb7489SZachary Loafman 	int n;
471cfeb7489SZachary Loafman 
47270e20d4eSConrad Meyer 	fp->fp_setting = NULL;
473cfeb7489SZachary Loafman 	fp->fp_flags = 0;
474cfeb7489SZachary Loafman 
475cfeb7489SZachary Loafman 	/* Figure out the size of the name. */
476cfeb7489SZachary Loafman 	va_start(ap, fmt);
477cfeb7489SZachary Loafman 	n = vsnprintf(NULL, 0, fmt, ap);
478cfeb7489SZachary Loafman 	va_end(ap);
479cfeb7489SZachary Loafman 
480cfeb7489SZachary Loafman 	/* Allocate the name and fill it in. */
481cfeb7489SZachary Loafman 	name = fp_malloc(n + 1, M_WAITOK);
482cfeb7489SZachary Loafman 	if (name != NULL) {
483cfeb7489SZachary Loafman 		va_start(ap, fmt);
484cfeb7489SZachary Loafman 		vsnprintf(name, n + 1, fmt, ap);
485cfeb7489SZachary Loafman 		va_end(ap);
486cfeb7489SZachary Loafman 	}
487cfeb7489SZachary Loafman 	fp->fp_name = name;
4884b7c6844SMatthew D Fleming 	fp->fp_location = "";
489cfeb7489SZachary Loafman 	fp->fp_flags |= FAIL_POINT_DYNAMIC_NAME;
49070e20d4eSConrad Meyer 	fp->fp_pre_sleep_fn = NULL;
49170e20d4eSConrad Meyer 	fp->fp_pre_sleep_arg = NULL;
49270e20d4eSConrad Meyer 	fp->fp_post_sleep_fn = NULL;
49370e20d4eSConrad Meyer 	fp->fp_post_sleep_arg = NULL;
494cfeb7489SZachary Loafman }
495cfeb7489SZachary Loafman 
496bf2276f3SJohn Baldwin void
fail_point_alloc_callout(struct fail_point * fp)497bf2276f3SJohn Baldwin fail_point_alloc_callout(struct fail_point *fp)
498bf2276f3SJohn Baldwin {
499bf2276f3SJohn Baldwin 
500bf2276f3SJohn Baldwin 	/**
501bf2276f3SJohn Baldwin 	 * This assumes that calls to fail_point_use_timeout_path()
502bf2276f3SJohn Baldwin 	 * will not race.
503bf2276f3SJohn Baldwin 	 */
504bf2276f3SJohn Baldwin 	if (fp->fp_callout != NULL)
505bf2276f3SJohn Baldwin 		return;
506bf2276f3SJohn Baldwin 	fp->fp_callout = fp_malloc(sizeof(*fp->fp_callout), M_WAITOK);
507bf2276f3SJohn Baldwin 	callout_init(fp->fp_callout, CALLOUT_MPSAFE);
508bf2276f3SJohn Baldwin }
509bf2276f3SJohn Baldwin 
510cfeb7489SZachary Loafman /**
51170e20d4eSConrad Meyer  * Free the resources held by a fail_point, and wake any paused threads.
51270e20d4eSConrad Meyer  * Thou shalt not allow threads to hit this fail point after you enter this
51370e20d4eSConrad Meyer  * function, nor shall you call this multiple times for a given fp.
514cfeb7489SZachary Loafman  * @ingroup failpoint
515cfeb7489SZachary Loafman  */
516cfeb7489SZachary Loafman void
fail_point_destroy(struct fail_point * fp)517cfeb7489SZachary Loafman fail_point_destroy(struct fail_point *fp)
518cfeb7489SZachary Loafman {
519cfeb7489SZachary Loafman 
52070e20d4eSConrad Meyer 	fail_point_drain(fp, 0);
52170e20d4eSConrad Meyer 
5220894a9bcSMatthew D Fleming 	if ((fp->fp_flags & FAIL_POINT_DYNAMIC_NAME) != 0) {
5230894a9bcSMatthew D Fleming 		fp_free(__DECONST(void *, fp->fp_name));
524cfeb7489SZachary Loafman 		fp->fp_name = NULL;
525cfeb7489SZachary Loafman 	}
526cfeb7489SZachary Loafman 	fp->fp_flags = 0;
527bf2276f3SJohn Baldwin 	if (fp->fp_callout) {
528bf2276f3SJohn Baldwin 		fp_free(fp->fp_callout);
529bf2276f3SJohn Baldwin 		fp->fp_callout = NULL;
530bf2276f3SJohn Baldwin 	}
53170e20d4eSConrad Meyer 
53270e20d4eSConrad Meyer 	sx_xlock(&sx_fp_set);
53370e20d4eSConrad Meyer 	fail_point_garbage_collect();
53470e20d4eSConrad Meyer 	sx_xunlock(&sx_fp_set);
535cfeb7489SZachary Loafman }
536cfeb7489SZachary Loafman 
537cfeb7489SZachary Loafman /**
538cfeb7489SZachary Loafman  * This does the real work of evaluating a fail point. If the fail point tells
539cfeb7489SZachary Loafman  * us to return a value, this function returns 1 and fills in 'return_value'
540cfeb7489SZachary Loafman  * (return_value is allowed to be null). If the fail point tells us to panic,
541cfeb7489SZachary Loafman  * we never return. Otherwise we just return 0 after doing some work, which
542cfeb7489SZachary Loafman  * means "keep going".
543cfeb7489SZachary Loafman  */
544cfeb7489SZachary Loafman enum fail_point_return_code
fail_point_eval_nontrivial(struct fail_point * fp,int * return_value)545cfeb7489SZachary Loafman fail_point_eval_nontrivial(struct fail_point *fp, int *return_value)
546cfeb7489SZachary Loafman {
54770e20d4eSConrad Meyer 	bool execute = false;
54870e20d4eSConrad Meyer 	struct fail_point_entry *ent;
54970e20d4eSConrad Meyer 	struct fail_point_setting *fp_setting;
55070e20d4eSConrad Meyer 	enum fail_point_return_code ret;
55170e20d4eSConrad Meyer 	int cont;
55270e20d4eSConrad Meyer 	int count;
553cfeb7489SZachary Loafman 	int msecs;
55470e20d4eSConrad Meyer 	int usecs;
555cfeb7489SZachary Loafman 
55670e20d4eSConrad Meyer 	ret = FAIL_POINT_RC_CONTINUE;
55770e20d4eSConrad Meyer 	cont = 0; /* don't continue by default */
558cfeb7489SZachary Loafman 
55970e20d4eSConrad Meyer 	fp_setting = fail_point_setting_get_ref(fp);
56070e20d4eSConrad Meyer 	if (fp_setting == NULL)
56170e20d4eSConrad Meyer 		goto abort;
56270e20d4eSConrad Meyer 
56370e20d4eSConrad Meyer 	TAILQ_FOREACH(ent, &fp_setting->fp_entry_queue, fe_entries) {
56470e20d4eSConrad Meyer 		if (ent->fe_stale)
56570e20d4eSConrad Meyer 			continue;
566cfeb7489SZachary Loafman 
567cfeb7489SZachary Loafman 		if (ent->fe_prob < PROB_MAX &&
5680894a9bcSMatthew D Fleming 		    ent->fe_prob < random() % PROB_MAX)
5690894a9bcSMatthew D Fleming 			continue;
57070e20d4eSConrad Meyer 
571b5bd50aeSMatthew D Fleming 		if (ent->fe_pid != NO_PID && ent->fe_pid != curproc->p_pid)
572b5bd50aeSMatthew D Fleming 			continue;
573cfeb7489SZachary Loafman 
57470e20d4eSConrad Meyer 		if (ent->fe_count != FE_COUNT_UNTRACKED) {
57570e20d4eSConrad Meyer 			count = ent->fe_count;
57670e20d4eSConrad Meyer 			while (count > 0) {
57770e20d4eSConrad Meyer 				if (atomic_cmpset_32(&ent->fe_count, count, count - 1)) {
57870e20d4eSConrad Meyer 					count--;
57970e20d4eSConrad Meyer 					execute = true;
58070e20d4eSConrad Meyer 					break;
58170e20d4eSConrad Meyer 				}
58270e20d4eSConrad Meyer 				count = ent->fe_count;
58370e20d4eSConrad Meyer 			}
58470e20d4eSConrad Meyer 			if (execute == false)
58570e20d4eSConrad Meyer 				/* We lost the race; consider the entry stale and bail now */
58670e20d4eSConrad Meyer 				continue;
58770e20d4eSConrad Meyer 			if (count == 0)
58870e20d4eSConrad Meyer 				ent->fe_stale = true;
58970e20d4eSConrad Meyer 		}
59070e20d4eSConrad Meyer 
591cfeb7489SZachary Loafman 		switch (ent->fe_type) {
592cfeb7489SZachary Loafman 		case FAIL_POINT_PANIC:
593cfeb7489SZachary Loafman 			panic("fail point %s panicking", fp->fp_name);
594cfeb7489SZachary Loafman 			/* NOTREACHED */
595cfeb7489SZachary Loafman 
596cfeb7489SZachary Loafman 		case FAIL_POINT_RETURN:
5970894a9bcSMatthew D Fleming 			if (return_value != NULL)
598cfeb7489SZachary Loafman 				*return_value = ent->fe_arg;
599cfeb7489SZachary Loafman 			ret = FAIL_POINT_RC_RETURN;
600cfeb7489SZachary Loafman 			break;
601cfeb7489SZachary Loafman 
602cfeb7489SZachary Loafman 		case FAIL_POINT_BREAK:
6030894a9bcSMatthew D Fleming 			printf("fail point %s breaking to debugger\n",
6040894a9bcSMatthew D Fleming 			        fp->fp_name);
605cfeb7489SZachary Loafman 			breakpoint();
606cfeb7489SZachary Loafman 			break;
607cfeb7489SZachary Loafman 
608cfeb7489SZachary Loafman 		case FAIL_POINT_PRINT:
609cfeb7489SZachary Loafman 			printf("fail point %s executing\n", fp->fp_name);
610cfeb7489SZachary Loafman 			cont = ent->fe_arg;
611cfeb7489SZachary Loafman 			break;
612cfeb7489SZachary Loafman 
613cfeb7489SZachary Loafman 		case FAIL_POINT_SLEEP:
614cfeb7489SZachary Loafman 			msecs = ent->fe_arg;
615cfeb7489SZachary Loafman 			if (msecs)
61670e20d4eSConrad Meyer 				fail_point_sleep(fp, msecs, &ret);
61770e20d4eSConrad Meyer 			break;
61870e20d4eSConrad Meyer 
61970e20d4eSConrad Meyer 		case FAIL_POINT_PAUSE:
62070e20d4eSConrad Meyer 			/**
62170e20d4eSConrad Meyer 			 * Pausing is inherently strange with multiple
62270e20d4eSConrad Meyer 			 * entries given our design.  That is because some
62370e20d4eSConrad Meyer 			 * entries could be unreachable, for instance in cases like:
62470e20d4eSConrad Meyer 			 * pause->return. We can never reach the return entry.
62570e20d4eSConrad Meyer 			 * The sysctl layer actually truncates all entries after
62670e20d4eSConrad Meyer 			 * a pause for this reason.
62770e20d4eSConrad Meyer 			 */
62870e20d4eSConrad Meyer 			mtx_lock_spin(&fp_setting->feq_mtx);
62970e20d4eSConrad Meyer 			fail_point_pause(fp, &ret, &fp_setting->feq_mtx);
63070e20d4eSConrad Meyer 			mtx_unlock_spin(&fp_setting->feq_mtx);
63170e20d4eSConrad Meyer 			break;
63270e20d4eSConrad Meyer 
63370e20d4eSConrad Meyer 		case FAIL_POINT_YIELD:
6343bd485f9SMark Johnston 			kern_yield(PRI_UNCHANGED);
63570e20d4eSConrad Meyer 			break;
63670e20d4eSConrad Meyer 
63770e20d4eSConrad Meyer 		case FAIL_POINT_DELAY:
63870e20d4eSConrad Meyer 			usecs = ent->fe_arg;
63970e20d4eSConrad Meyer 			DELAY(usecs);
640cfeb7489SZachary Loafman 			break;
641cfeb7489SZachary Loafman 
642cfeb7489SZachary Loafman 		default:
643cfeb7489SZachary Loafman 			break;
644cfeb7489SZachary Loafman 		}
645cfeb7489SZachary Loafman 
6460894a9bcSMatthew D Fleming 		if (cont == 0)
647cfeb7489SZachary Loafman 			break;
648cfeb7489SZachary Loafman 	}
649cfeb7489SZachary Loafman 
65070e20d4eSConrad Meyer 	if (fail_point_is_off(fp))
65170e20d4eSConrad Meyer 		fail_point_eval_swap_out(fp, fp_setting);
652cfeb7489SZachary Loafman 
65370e20d4eSConrad Meyer abort:
65470e20d4eSConrad Meyer 	fail_point_setting_release_ref(fp);
655cfeb7489SZachary Loafman 
6560894a9bcSMatthew D Fleming 	return (ret);
657cfeb7489SZachary Loafman }
658cfeb7489SZachary Loafman 
659cfeb7489SZachary Loafman /**
660cfeb7489SZachary Loafman  * Translate internal fail_point structure into human-readable text.
661cfeb7489SZachary Loafman  */
662cfeb7489SZachary Loafman static void
fail_point_get(struct fail_point * fp,struct sbuf * sb,bool verbose)66370e20d4eSConrad Meyer fail_point_get(struct fail_point *fp, struct sbuf *sb,
66470e20d4eSConrad Meyer         bool verbose)
665cfeb7489SZachary Loafman {
666cfeb7489SZachary Loafman 	struct fail_point_entry *ent;
66770e20d4eSConrad Meyer 	struct fail_point_setting *fp_setting;
66870e20d4eSConrad Meyer 	struct fail_point_entry *fp_entry_cpy;
66970e20d4eSConrad Meyer 	int cnt_sleeping;
67070e20d4eSConrad Meyer 	int idx;
67170e20d4eSConrad Meyer 	int printed_entry_count;
672cfeb7489SZachary Loafman 
67370e20d4eSConrad Meyer 	cnt_sleeping = 0;
67470e20d4eSConrad Meyer 	idx = 0;
67570e20d4eSConrad Meyer 	printed_entry_count = 0;
676cfeb7489SZachary Loafman 
67770e20d4eSConrad Meyer 	fp_entry_cpy = fp_malloc(sizeof(struct fail_point_entry) *
67870e20d4eSConrad Meyer 	        (FP_MAX_ENTRY_COUNT + 1), M_WAITOK);
67970e20d4eSConrad Meyer 
68070e20d4eSConrad Meyer 	fp_setting = fail_point_setting_get_ref(fp);
68170e20d4eSConrad Meyer 
68270e20d4eSConrad Meyer 	if (fp_setting != NULL) {
68370e20d4eSConrad Meyer 		TAILQ_FOREACH(ent, &fp_setting->fp_entry_queue, fe_entries) {
68470e20d4eSConrad Meyer 			if (ent->fe_stale)
68570e20d4eSConrad Meyer 				continue;
68670e20d4eSConrad Meyer 
68770e20d4eSConrad Meyer 			KASSERT(printed_entry_count < FP_MAX_ENTRY_COUNT,
68870e20d4eSConrad Meyer 			        ("FP entry list larger than allowed"));
68970e20d4eSConrad Meyer 
69070e20d4eSConrad Meyer 			fp_entry_cpy[printed_entry_count] = *ent;
69170e20d4eSConrad Meyer 			++printed_entry_count;
69270e20d4eSConrad Meyer 		}
69370e20d4eSConrad Meyer 	}
69470e20d4eSConrad Meyer 	fail_point_setting_release_ref(fp);
69570e20d4eSConrad Meyer 
69670e20d4eSConrad Meyer 	/* This is our equivalent of a NULL terminator */
69770e20d4eSConrad Meyer 	fp_entry_cpy[printed_entry_count].fe_type = FAIL_POINT_INVALID;
69870e20d4eSConrad Meyer 
69970e20d4eSConrad Meyer 	while (idx < printed_entry_count) {
70070e20d4eSConrad Meyer 		ent = &fp_entry_cpy[idx];
70170e20d4eSConrad Meyer 		++idx;
702cfeb7489SZachary Loafman 		if (ent->fe_prob < PROB_MAX) {
703cfeb7489SZachary Loafman 			int decimal = ent->fe_prob % (PROB_MAX / 100);
704cfeb7489SZachary Loafman 			int units = ent->fe_prob / (PROB_MAX / 100);
705cfeb7489SZachary Loafman 			sbuf_printf(sb, "%d", units);
706cfeb7489SZachary Loafman 			if (decimal) {
707cfeb7489SZachary Loafman 				int digits = PROB_DIGITS - 2;
708cfeb7489SZachary Loafman 				while (!(decimal % 10)) {
709cfeb7489SZachary Loafman 					digits--;
710cfeb7489SZachary Loafman 					decimal /= 10;
711cfeb7489SZachary Loafman 				}
712cfeb7489SZachary Loafman 				sbuf_printf(sb, ".%0*d", digits, decimal);
713cfeb7489SZachary Loafman 			}
714cfeb7489SZachary Loafman 			sbuf_printf(sb, "%%");
715cfeb7489SZachary Loafman 		}
71670e20d4eSConrad Meyer 		if (ent->fe_count >= 0)
717cfeb7489SZachary Loafman 			sbuf_printf(sb, "%d*", ent->fe_count);
7180894a9bcSMatthew D Fleming 		sbuf_printf(sb, "%s", fail_type_strings[ent->fe_type].name);
719cfeb7489SZachary Loafman 		if (ent->fe_arg)
720cfeb7489SZachary Loafman 			sbuf_printf(sb, "(%d)", ent->fe_arg);
721b5bd50aeSMatthew D Fleming 		if (ent->fe_pid != NO_PID)
722b5bd50aeSMatthew D Fleming 			sbuf_printf(sb, "[pid %d]", ent->fe_pid);
723cfeb7489SZachary Loafman 		if (TAILQ_NEXT(ent, fe_entries))
7240a713948SAlexander Motin 			sbuf_cat(sb, "->");
725cfeb7489SZachary Loafman 	}
72670e20d4eSConrad Meyer 	if (!printed_entry_count)
7270a713948SAlexander Motin 		sbuf_cat(sb, "off");
728cfeb7489SZachary Loafman 
72970e20d4eSConrad Meyer 	fp_free(fp_entry_cpy);
73070e20d4eSConrad Meyer 	if (verbose) {
73182e17adcSConrad Meyer #ifdef STACK
73270e20d4eSConrad Meyer 		/* Print number of sleeping threads. queue=0 is the argument
73370e20d4eSConrad Meyer 		 * used by msleep when sending our threads to sleep. */
7340a713948SAlexander Motin 		sbuf_cat(sb, "\nsleeping_thread_stacks = {\n");
73570e20d4eSConrad Meyer 		sleepq_sbuf_print_stacks(sb, FP_SLEEP_CHANNEL(fp), 0,
73670e20d4eSConrad Meyer 		        &cnt_sleeping);
73770e20d4eSConrad Meyer 
7380a713948SAlexander Motin 		sbuf_cat(sb, "},\n");
73982e17adcSConrad Meyer #endif
74070e20d4eSConrad Meyer 		sbuf_printf(sb, "sleeping_thread_count = %d,\n",
74170e20d4eSConrad Meyer 		        cnt_sleeping);
74270e20d4eSConrad Meyer 
74382e17adcSConrad Meyer #ifdef STACK
7440a713948SAlexander Motin 		sbuf_cat(sb, "paused_thread_stacks = {\n");
74570e20d4eSConrad Meyer 		sleepq_sbuf_print_stacks(sb, FP_PAUSE_CHANNEL(fp), 0,
74670e20d4eSConrad Meyer 		        &cnt_sleeping);
74770e20d4eSConrad Meyer 
7480a713948SAlexander Motin 		sbuf_cat(sb, "},\n");
74982e17adcSConrad Meyer #endif
75070e20d4eSConrad Meyer 		sbuf_printf(sb, "paused_thread_count = %d\n",
75170e20d4eSConrad Meyer 		        cnt_sleeping);
75270e20d4eSConrad Meyer 	}
753cfeb7489SZachary Loafman }
754cfeb7489SZachary Loafman 
755cfeb7489SZachary Loafman /**
756cfeb7489SZachary Loafman  * Set an internal fail_point structure from a human-readable failpoint string
757cfeb7489SZachary Loafman  * in a lock-safe manner.
758cfeb7489SZachary Loafman  */
759cfeb7489SZachary Loafman static int
fail_point_set(struct fail_point * fp,char * buf)760cfeb7489SZachary Loafman fail_point_set(struct fail_point *fp, char *buf)
761cfeb7489SZachary Loafman {
762cfeb7489SZachary Loafman 	struct fail_point_entry *ent, *ent_next;
76370e20d4eSConrad Meyer 	struct fail_point_setting *entries;
76470e20d4eSConrad Meyer 	bool should_wake_paused;
76570e20d4eSConrad Meyer 	bool should_truncate;
76670e20d4eSConrad Meyer 	int error;
76770e20d4eSConrad Meyer 
76870e20d4eSConrad Meyer 	error = 0;
76970e20d4eSConrad Meyer 	should_wake_paused = false;
77070e20d4eSConrad Meyer 	should_truncate = false;
771cfeb7489SZachary Loafman 
772cfeb7489SZachary Loafman 	/* Parse new entries. */
77370e20d4eSConrad Meyer 	/**
77470e20d4eSConrad Meyer 	 * ref protects our new malloc'd stuff from being garbage collected
77570e20d4eSConrad Meyer 	 * before we link it.
77670e20d4eSConrad Meyer 	 */
77770e20d4eSConrad Meyer 	fail_point_setting_get_ref(fp);
77870e20d4eSConrad Meyer 	entries = fail_point_setting_new(fp);
77970e20d4eSConrad Meyer 	if (parse_fail_point(entries, buf) == NULL) {
78070e20d4eSConrad Meyer 		STAILQ_REMOVE(&fp_setting_garbage, entries,
78170e20d4eSConrad Meyer 		        fail_point_setting, fs_garbage_link);
78270e20d4eSConrad Meyer 		fail_point_setting_destroy(entries);
783cfeb7489SZachary Loafman 		error = EINVAL;
784cfeb7489SZachary Loafman 		goto end;
785cfeb7489SZachary Loafman 	}
786cfeb7489SZachary Loafman 
78770e20d4eSConrad Meyer 	/**
78870e20d4eSConrad Meyer 	 * Transfer the entries we are going to keep to a new list.
78970e20d4eSConrad Meyer 	 * Get rid of useless zero probability entries, and entries with hit
79070e20d4eSConrad Meyer 	 * count 0.
79170e20d4eSConrad Meyer 	 * If 'off' is present, and it has no hit count set, then all entries
79270e20d4eSConrad Meyer 	 *       after it are discarded since they are unreachable.
79370e20d4eSConrad Meyer 	 */
79470e20d4eSConrad Meyer 	TAILQ_FOREACH_SAFE(ent, &entries->fp_entry_queue, fe_entries, ent_next) {
79570e20d4eSConrad Meyer 		if (ent->fe_prob == 0 || ent->fe_count == 0) {
79670e20d4eSConrad Meyer 			printf("Discarding entry which cannot execute %s\n",
79770e20d4eSConrad Meyer 			        fail_type_strings[ent->fe_type].name);
79870e20d4eSConrad Meyer 			TAILQ_REMOVE(&entries->fp_entry_queue, ent,
79970e20d4eSConrad Meyer 			        fe_entries);
80070e20d4eSConrad Meyer 			fp_free(ent);
80170e20d4eSConrad Meyer 			continue;
80270e20d4eSConrad Meyer 		} else if (should_truncate) {
80370e20d4eSConrad Meyer 			printf("Discarding unreachable entry %s\n",
80470e20d4eSConrad Meyer 			        fail_type_strings[ent->fe_type].name);
80570e20d4eSConrad Meyer 			TAILQ_REMOVE(&entries->fp_entry_queue, ent,
80670e20d4eSConrad Meyer 			        fe_entries);
80770e20d4eSConrad Meyer 			fp_free(ent);
80870e20d4eSConrad Meyer 			continue;
809cfeb7489SZachary Loafman 		}
810cfeb7489SZachary Loafman 
81170e20d4eSConrad Meyer 		if (ent->fe_type == FAIL_POINT_OFF) {
81270e20d4eSConrad Meyer 			should_wake_paused = true;
81370e20d4eSConrad Meyer 			if (ent->fe_count == FE_COUNT_UNTRACKED) {
81470e20d4eSConrad Meyer 				should_truncate = true;
81570e20d4eSConrad Meyer 				TAILQ_REMOVE(&entries->fp_entry_queue, ent,
81670e20d4eSConrad Meyer 				        fe_entries);
81770e20d4eSConrad Meyer 				fp_free(ent);
81870e20d4eSConrad Meyer 			}
81970e20d4eSConrad Meyer 		} else if (ent->fe_type == FAIL_POINT_PAUSE) {
82070e20d4eSConrad Meyer 			should_truncate = true;
82170e20d4eSConrad Meyer 		} else if (ent->fe_type == FAIL_POINT_SLEEP && (fp->fp_flags &
82270e20d4eSConrad Meyer 		        FAIL_POINT_NONSLEEPABLE)) {
82370e20d4eSConrad Meyer 			/**
82470e20d4eSConrad Meyer 			 * If this fail point is annotated as being in a
82570e20d4eSConrad Meyer 			 * non-sleepable ctx, convert sleep to delay and
82670e20d4eSConrad Meyer 			 * convert the msec argument to usecs.
82770e20d4eSConrad Meyer 			 */
82870e20d4eSConrad Meyer 			printf("Sleep call request on fail point in "
82970e20d4eSConrad Meyer 			        "non-sleepable context; using delay instead "
83070e20d4eSConrad Meyer 			        "of sleep\n");
83170e20d4eSConrad Meyer 			ent->fe_type = FAIL_POINT_DELAY;
83270e20d4eSConrad Meyer 			ent->fe_arg *= 1000;
83370e20d4eSConrad Meyer 		}
83470e20d4eSConrad Meyer 	}
835cfeb7489SZachary Loafman 
83670e20d4eSConrad Meyer 	if (TAILQ_EMPTY(&entries->fp_entry_queue)) {
83770e20d4eSConrad Meyer 		entries = fail_point_swap_settings(fp, NULL);
83870e20d4eSConrad Meyer 		if (entries != NULL)
83970e20d4eSConrad Meyer 			wakeup(FP_PAUSE_CHANNEL(fp));
84070e20d4eSConrad Meyer 	} else {
84170e20d4eSConrad Meyer 		if (should_wake_paused)
84270e20d4eSConrad Meyer 			wakeup(FP_PAUSE_CHANNEL(fp));
84370e20d4eSConrad Meyer 		fail_point_swap_settings(fp, entries);
84470e20d4eSConrad Meyer 	}
845cfeb7489SZachary Loafman 
846cfeb7489SZachary Loafman end:
847cfeb7489SZachary Loafman #ifdef IWARNING
848cfeb7489SZachary Loafman 	if (error)
8494b7c6844SMatthew D Fleming 		IWARNING("Failed to set %s %s to %s",
850cfeb7489SZachary Loafman 		    fp->fp_name, fp->fp_location, buf);
851cfeb7489SZachary Loafman 	else
8524b7c6844SMatthew D Fleming 		INOTICE("Set %s %s to %s",
853cfeb7489SZachary Loafman 		    fp->fp_name, fp->fp_location, buf);
854cfeb7489SZachary Loafman #endif /* IWARNING */
855cfeb7489SZachary Loafman 
85670e20d4eSConrad Meyer 	fail_point_setting_release_ref(fp);
8570894a9bcSMatthew D Fleming 	return (error);
858cfeb7489SZachary Loafman }
859cfeb7489SZachary Loafman 
860cfeb7489SZachary Loafman #define MAX_FAIL_POINT_BUF	1023
861cfeb7489SZachary Loafman 
862cfeb7489SZachary Loafman /**
863cfeb7489SZachary Loafman  * Handle kernel failpoint set/get.
864cfeb7489SZachary Loafman  */
865cfeb7489SZachary Loafman int
fail_point_sysctl(SYSCTL_HANDLER_ARGS)866cfeb7489SZachary Loafman fail_point_sysctl(SYSCTL_HANDLER_ARGS)
867cfeb7489SZachary Loafman {
86870e20d4eSConrad Meyer 	struct fail_point *fp;
86970e20d4eSConrad Meyer 	char *buf;
870508d8569SMark Johnston 	struct sbuf sb, *sb_check;
871cfeb7489SZachary Loafman 	int error;
872cfeb7489SZachary Loafman 
873508d8569SMark Johnston 	buf = NULL;
87470e20d4eSConrad Meyer 	error = 0;
87570e20d4eSConrad Meyer 	fp = arg1;
87670e20d4eSConrad Meyer 
87770e20d4eSConrad Meyer 	sb_check = sbuf_new(&sb, NULL, 1024, SBUF_AUTOEXTEND);
87870e20d4eSConrad Meyer 	if (sb_check != &sb)
87970e20d4eSConrad Meyer 		return (ENOMEM);
88070e20d4eSConrad Meyer 
88170e20d4eSConrad Meyer 	sbuf_set_drain(&sb, (sbuf_drain_func *)fail_sysctl_drain_func, req);
882cfeb7489SZachary Loafman 
883cfeb7489SZachary Loafman 	/* Setting */
88470e20d4eSConrad Meyer 	/**
88570e20d4eSConrad Meyer 	 * Lock protects any new entries from being garbage collected before we
88670e20d4eSConrad Meyer 	 * can link them to the fail point.
88770e20d4eSConrad Meyer 	 */
88870e20d4eSConrad Meyer 	sx_xlock(&sx_fp_set);
88970e20d4eSConrad Meyer 	if (req->newptr) {
890cfeb7489SZachary Loafman 		if (req->newlen > MAX_FAIL_POINT_BUF) {
891cfeb7489SZachary Loafman 			error = EINVAL;
892cfeb7489SZachary Loafman 			goto out;
893cfeb7489SZachary Loafman 		}
894cfeb7489SZachary Loafman 
895cfeb7489SZachary Loafman 		buf = fp_malloc(req->newlen + 1, M_WAITOK);
896cfeb7489SZachary Loafman 
897cfeb7489SZachary Loafman 		error = SYSCTL_IN(req, buf, req->newlen);
898cfeb7489SZachary Loafman 		if (error)
899cfeb7489SZachary Loafman 			goto out;
900cfeb7489SZachary Loafman 		buf[req->newlen] = '\0';
901cfeb7489SZachary Loafman 
902cfeb7489SZachary Loafman 		error = fail_point_set(fp, buf);
903cfeb7489SZachary Loafman 	}
904cfeb7489SZachary Loafman 
90570e20d4eSConrad Meyer 	fail_point_garbage_collect();
90670e20d4eSConrad Meyer 	sx_xunlock(&sx_fp_set);
90770e20d4eSConrad Meyer 
90870e20d4eSConrad Meyer 	/* Retrieving. */
90970e20d4eSConrad Meyer 	fail_point_get(fp, &sb, false);
91070e20d4eSConrad Meyer 
911cfeb7489SZachary Loafman out:
91270e20d4eSConrad Meyer 	sbuf_finish(&sb);
91370e20d4eSConrad Meyer 	sbuf_delete(&sb);
91470e20d4eSConrad Meyer 
91570e20d4eSConrad Meyer 	if (buf)
916cfeb7489SZachary Loafman 		fp_free(buf);
91770e20d4eSConrad Meyer 
9180894a9bcSMatthew D Fleming 	return (error);
919cfeb7489SZachary Loafman }
920cfeb7489SZachary Loafman 
92170e20d4eSConrad Meyer int
fail_point_sysctl_status(SYSCTL_HANDLER_ARGS)92270e20d4eSConrad Meyer fail_point_sysctl_status(SYSCTL_HANDLER_ARGS)
92370e20d4eSConrad Meyer {
92470e20d4eSConrad Meyer 	struct fail_point *fp;
92570e20d4eSConrad Meyer 	struct sbuf sb, *sb_check;
92670e20d4eSConrad Meyer 
92770e20d4eSConrad Meyer 	fp = arg1;
92870e20d4eSConrad Meyer 
92970e20d4eSConrad Meyer 	sb_check = sbuf_new(&sb, NULL, 1024, SBUF_AUTOEXTEND);
93070e20d4eSConrad Meyer 	if (sb_check != &sb)
93170e20d4eSConrad Meyer 		return (ENOMEM);
93270e20d4eSConrad Meyer 
93370e20d4eSConrad Meyer 	sbuf_set_drain(&sb, (sbuf_drain_func *)fail_sysctl_drain_func, req);
93470e20d4eSConrad Meyer 
93570e20d4eSConrad Meyer 	/* Retrieving. */
93670e20d4eSConrad Meyer 	fail_point_get(fp, &sb, true);
93770e20d4eSConrad Meyer 
93870e20d4eSConrad Meyer 	sbuf_finish(&sb);
93970e20d4eSConrad Meyer 	sbuf_delete(&sb);
94070e20d4eSConrad Meyer 
94170e20d4eSConrad Meyer 	/**
94270e20d4eSConrad Meyer 	 * Lock protects any new entries from being garbage collected before we
94370e20d4eSConrad Meyer 	 * can link them to the fail point.
94470e20d4eSConrad Meyer 	 */
94570e20d4eSConrad Meyer 	sx_xlock(&sx_fp_set);
94670e20d4eSConrad Meyer 	fail_point_garbage_collect();
94770e20d4eSConrad Meyer 	sx_xunlock(&sx_fp_set);
94870e20d4eSConrad Meyer 
94970e20d4eSConrad Meyer 	return (0);
95070e20d4eSConrad Meyer }
95170e20d4eSConrad Meyer 
95270e20d4eSConrad Meyer int
fail_sysctl_drain_func(void * sysctl_args,const char * buf,int len)95370e20d4eSConrad Meyer fail_sysctl_drain_func(void *sysctl_args, const char *buf, int len)
95470e20d4eSConrad Meyer {
95570e20d4eSConrad Meyer 	struct sysctl_req *sa;
95670e20d4eSConrad Meyer 	int error;
95770e20d4eSConrad Meyer 
95870e20d4eSConrad Meyer 	sa = sysctl_args;
95970e20d4eSConrad Meyer 
96070e20d4eSConrad Meyer 	error = SYSCTL_OUT(sa, buf, len);
96170e20d4eSConrad Meyer 
96270e20d4eSConrad Meyer 	if (error == ENOMEM)
96370e20d4eSConrad Meyer 		return (-1);
96470e20d4eSConrad Meyer 	else
96570e20d4eSConrad Meyer 		return (len);
96670e20d4eSConrad Meyer }
96770e20d4eSConrad Meyer 
968cfeb7489SZachary Loafman /**
969cfeb7489SZachary Loafman  * Internal helper function to translate a human-readable failpoint string
970cfeb7489SZachary Loafman  * into a internally-parsable fail_point structure.
971cfeb7489SZachary Loafman  */
972cfeb7489SZachary Loafman static char *
parse_fail_point(struct fail_point_setting * ents,char * p)97370e20d4eSConrad Meyer parse_fail_point(struct fail_point_setting *ents, char *p)
974cfeb7489SZachary Loafman {
975cfeb7489SZachary Loafman 	/*  <fail_point> ::
976cfeb7489SZachary Loafman 	 *      <term> ( "->" <term> )*
977cfeb7489SZachary Loafman 	 */
97870e20d4eSConrad Meyer 	uint8_t term_count;
97970e20d4eSConrad Meyer 
98070e20d4eSConrad Meyer 	term_count = 1;
98170e20d4eSConrad Meyer 
9820894a9bcSMatthew D Fleming 	p = parse_term(ents, p);
9830894a9bcSMatthew D Fleming 	if (p == NULL)
9840894a9bcSMatthew D Fleming 		return (NULL);
98570e20d4eSConrad Meyer 
9860894a9bcSMatthew D Fleming 	while (*p != '\0') {
98770e20d4eSConrad Meyer 		term_count++;
98870e20d4eSConrad Meyer 		if (p[0] != '-' || p[1] != '>' ||
98970e20d4eSConrad Meyer 		        (p = parse_term(ents, p+2)) == NULL ||
99070e20d4eSConrad Meyer 		        term_count > FP_MAX_ENTRY_COUNT)
9910894a9bcSMatthew D Fleming 			return (NULL);
9920894a9bcSMatthew D Fleming 	}
9930894a9bcSMatthew D Fleming 	return (p);
994cfeb7489SZachary Loafman }
995cfeb7489SZachary Loafman 
996cfeb7489SZachary Loafman /**
997cfeb7489SZachary Loafman  * Internal helper function to parse an individual term from a failpoint.
998cfeb7489SZachary Loafman  */
999cfeb7489SZachary Loafman static char *
parse_term(struct fail_point_setting * ents,char * p)100070e20d4eSConrad Meyer parse_term(struct fail_point_setting *ents, char *p)
1001cfeb7489SZachary Loafman {
1002cfeb7489SZachary Loafman 	struct fail_point_entry *ent;
1003cfeb7489SZachary Loafman 
100470e20d4eSConrad Meyer 	ent = fail_point_entry_new(ents);
1005cfeb7489SZachary Loafman 
1006cfeb7489SZachary Loafman 	/*
1007cfeb7489SZachary Loafman 	 * <term> ::
1008cfeb7489SZachary Loafman 	 *     ( (<float> "%") | (<integer> "*" ) )*
1009cfeb7489SZachary Loafman 	 *     <type>
1010cfeb7489SZachary Loafman 	 *     [ "(" <integer> ")" ]
1011b5bd50aeSMatthew D Fleming 	 *     [ "[pid " <integer> "]" ]
1012cfeb7489SZachary Loafman 	 */
1013cfeb7489SZachary Loafman 
1014cfeb7489SZachary Loafman 	/* ( (<float> "%") | (<integer> "*" ) )* */
10150894a9bcSMatthew D Fleming 	while (isdigit(*p) || *p == '.') {
1016cfeb7489SZachary Loafman 		int units, decimal;
1017cfeb7489SZachary Loafman 
10180894a9bcSMatthew D Fleming 		p = parse_number(&units, &decimal, p);
10190894a9bcSMatthew D Fleming 		if (p == NULL)
10200894a9bcSMatthew D Fleming 			return (NULL);
1021cfeb7489SZachary Loafman 
1022cfeb7489SZachary Loafman 		if (*p == '%') {
1023cfeb7489SZachary Loafman 			if (units > 100) /* prevent overflow early */
1024cfeb7489SZachary Loafman 				units = 100;
1025cfeb7489SZachary Loafman 			ent->fe_prob = units * (PROB_MAX / 100) + decimal;
1026cfeb7489SZachary Loafman 			if (ent->fe_prob > PROB_MAX)
1027cfeb7489SZachary Loafman 				ent->fe_prob = PROB_MAX;
1028cfeb7489SZachary Loafman 		} else if (*p == '*') {
102970e20d4eSConrad Meyer 			if (!units || units < 0 || decimal)
10300894a9bcSMatthew D Fleming 				return (NULL);
1031c2ede4b3SMartin Blapp 			ent->fe_count = units;
10320894a9bcSMatthew D Fleming 		} else
10330894a9bcSMatthew D Fleming 			return (NULL);
1034cfeb7489SZachary Loafman 		p++;
1035cfeb7489SZachary Loafman 	}
1036cfeb7489SZachary Loafman 
1037cfeb7489SZachary Loafman 	/* <type> */
10380894a9bcSMatthew D Fleming 	p = parse_type(ent, p);
10390894a9bcSMatthew D Fleming 	if (p == NULL)
10400894a9bcSMatthew D Fleming 		return (NULL);
1041cfeb7489SZachary Loafman 	if (*p == '\0')
10420894a9bcSMatthew D Fleming 		return (p);
1043cfeb7489SZachary Loafman 
1044cfeb7489SZachary Loafman 	/* [ "(" <integer> ")" ] */
1045cfeb7489SZachary Loafman 	if (*p != '(')
104670e20d4eSConrad Meyer 		return (p);
1047cfeb7489SZachary Loafman 	p++;
10480894a9bcSMatthew D Fleming 	if (!isdigit(*p) && *p != '-')
10490894a9bcSMatthew D Fleming 		return (NULL);
1050cfeb7489SZachary Loafman 	ent->fe_arg = strtol(p, &p, 0);
1051cfeb7489SZachary Loafman 	if (*p++ != ')')
10520894a9bcSMatthew D Fleming 		return (NULL);
1053cfeb7489SZachary Loafman 
1054b5bd50aeSMatthew D Fleming 	/* [ "[pid " <integer> "]" ] */
1055b5bd50aeSMatthew D Fleming #define PID_STRING "[pid "
1056b5bd50aeSMatthew D Fleming 	if (strncmp(p, PID_STRING, sizeof(PID_STRING) - 1) != 0)
1057b5bd50aeSMatthew D Fleming 		return (p);
1058b5bd50aeSMatthew D Fleming 	p += sizeof(PID_STRING) - 1;
1059b5bd50aeSMatthew D Fleming 	if (!isdigit(*p))
1060b5bd50aeSMatthew D Fleming 		return (NULL);
1061b5bd50aeSMatthew D Fleming 	ent->fe_pid = strtol(p, &p, 0);
1062b5bd50aeSMatthew D Fleming 	if (*p++ != ']')
1063b5bd50aeSMatthew D Fleming 		return (NULL);
1064b5bd50aeSMatthew D Fleming 
10650894a9bcSMatthew D Fleming 	return (p);
1066cfeb7489SZachary Loafman }
1067cfeb7489SZachary Loafman 
1068cfeb7489SZachary Loafman /**
1069cfeb7489SZachary Loafman  * Internal helper function to parse a numeric for a failpoint term.
1070cfeb7489SZachary Loafman  */
1071cfeb7489SZachary Loafman static char *
parse_number(int * out_units,int * out_decimal,char * p)1072cfeb7489SZachary Loafman parse_number(int *out_units, int *out_decimal, char *p)
1073cfeb7489SZachary Loafman {
1074cfeb7489SZachary Loafman 	char *old_p;
1075cfeb7489SZachary Loafman 
107670e20d4eSConrad Meyer 	/**
1077cfeb7489SZachary Loafman 	 *  <number> ::
1078cfeb7489SZachary Loafman 	 *      <integer> [ "." <integer> ] |
1079cfeb7489SZachary Loafman 	 *      "." <integer>
1080cfeb7489SZachary Loafman 	 */
1081cfeb7489SZachary Loafman 
1082cfeb7489SZachary Loafman 	/* whole part */
1083cfeb7489SZachary Loafman 	old_p = p;
1084c2ede4b3SMartin Blapp 	*out_units = strtol(p, &p, 10);
1085cfeb7489SZachary Loafman 	if (p == old_p && *p != '.')
10860894a9bcSMatthew D Fleming 		return (NULL);
1087cfeb7489SZachary Loafman 
1088cfeb7489SZachary Loafman 	/* fractional part */
1089cfeb7489SZachary Loafman 	*out_decimal = 0;
1090cfeb7489SZachary Loafman 	if (*p == '.') {
1091cfeb7489SZachary Loafman 		int digits = 0;
1092cfeb7489SZachary Loafman 		p++;
10930894a9bcSMatthew D Fleming 		while (isdigit(*p)) {
1094cfeb7489SZachary Loafman 			int digit = *p - '0';
1095cfeb7489SZachary Loafman 			if (digits < PROB_DIGITS - 2)
1096cfeb7489SZachary Loafman 				*out_decimal = *out_decimal * 10 + digit;
1097cfeb7489SZachary Loafman 			else if (digits == PROB_DIGITS - 2 && digit >= 5)
1098cfeb7489SZachary Loafman 				(*out_decimal)++;
1099cfeb7489SZachary Loafman 			digits++;
1100cfeb7489SZachary Loafman 			p++;
1101cfeb7489SZachary Loafman 		}
1102cfeb7489SZachary Loafman 		if (!digits) /* need at least one digit after '.' */
11030894a9bcSMatthew D Fleming 			return (NULL);
1104cfeb7489SZachary Loafman 		while (digits++ < PROB_DIGITS - 2) /* add implicit zeros */
1105cfeb7489SZachary Loafman 			*out_decimal *= 10;
1106cfeb7489SZachary Loafman 	}
1107cfeb7489SZachary Loafman 
11080894a9bcSMatthew D Fleming 	return (p); /* success */
1109cfeb7489SZachary Loafman }
1110cfeb7489SZachary Loafman 
1111cfeb7489SZachary Loafman /**
1112cfeb7489SZachary Loafman  * Internal helper function to parse an individual type for a failpoint term.
1113cfeb7489SZachary Loafman  */
1114cfeb7489SZachary Loafman static char *
parse_type(struct fail_point_entry * ent,char * beg)1115cfeb7489SZachary Loafman parse_type(struct fail_point_entry *ent, char *beg)
1116cfeb7489SZachary Loafman {
1117cfeb7489SZachary Loafman 	enum fail_point_t type;
11180894a9bcSMatthew D Fleming 	int len;
11190894a9bcSMatthew D Fleming 
11200894a9bcSMatthew D Fleming 	for (type = FAIL_POINT_OFF; type < FAIL_POINT_NUMTYPES; type++) {
11210894a9bcSMatthew D Fleming 		len = fail_type_strings[type].nmlen;
11220894a9bcSMatthew D Fleming 		if (strncmp(fail_type_strings[type].name, beg, len) == 0) {
1123cfeb7489SZachary Loafman 			ent->fe_type = type;
11240894a9bcSMatthew D Fleming 			return (beg + len);
1125cfeb7489SZachary Loafman 		}
1126cfeb7489SZachary Loafman 	}
11270894a9bcSMatthew D Fleming 	return (NULL);
1128cfeb7489SZachary Loafman }
1129cfeb7489SZachary Loafman 
1130cfeb7489SZachary Loafman /* The fail point sysctl tree. */
11317029da5cSPawel Biernacki SYSCTL_NODE(_debug, OID_AUTO, fail_point, CTLFLAG_RW | CTLFLAG_MPSAFE, 0,
11327029da5cSPawel Biernacki     "fail points");
113370e20d4eSConrad Meyer 
113470e20d4eSConrad Meyer /* Debugging/testing stuff for fail point */
113570e20d4eSConrad Meyer static int
sysctl_test_fail_point(SYSCTL_HANDLER_ARGS)113670e20d4eSConrad Meyer sysctl_test_fail_point(SYSCTL_HANDLER_ARGS)
113770e20d4eSConrad Meyer {
113870e20d4eSConrad Meyer 
113970e20d4eSConrad Meyer 	KFAIL_POINT_RETURN(DEBUG_FP, test_fail_point);
114070e20d4eSConrad Meyer 	return (0);
114170e20d4eSConrad Meyer }
114270e20d4eSConrad Meyer SYSCTL_OID(_debug_fail_point, OID_AUTO, test_trigger_fail_point,
11437029da5cSPawel Biernacki     CTLTYPE_STRING | CTLFLAG_RD | CTLFLAG_NEEDGIANT, NULL, 0,
11447029da5cSPawel Biernacki     sysctl_test_fail_point, "A",
114570e20d4eSConrad Meyer     "Trigger test fail points");
1146