xref: /freebsd/sys/kern/kern_fail.c (revision 6a2a385507c79abaa9db9eabfdd827362f3dc7ed)
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 	va_start(ap, fmt);
483cfeb7489SZachary Loafman 	vsnprintf(name, n + 1, fmt, ap);
484cfeb7489SZachary Loafman 	va_end(ap);
485*6a2a3855SZhenlei Huang 
486cfeb7489SZachary Loafman 	fp->fp_name = name;
4874b7c6844SMatthew D Fleming 	fp->fp_location = "";
488cfeb7489SZachary Loafman 	fp->fp_flags |= FAIL_POINT_DYNAMIC_NAME;
48970e20d4eSConrad Meyer 	fp->fp_pre_sleep_fn = NULL;
49070e20d4eSConrad Meyer 	fp->fp_pre_sleep_arg = NULL;
49170e20d4eSConrad Meyer 	fp->fp_post_sleep_fn = NULL;
49270e20d4eSConrad Meyer 	fp->fp_post_sleep_arg = NULL;
493cfeb7489SZachary Loafman }
494cfeb7489SZachary Loafman 
495bf2276f3SJohn Baldwin void
fail_point_alloc_callout(struct fail_point * fp)496bf2276f3SJohn Baldwin fail_point_alloc_callout(struct fail_point *fp)
497bf2276f3SJohn Baldwin {
498bf2276f3SJohn Baldwin 
499bf2276f3SJohn Baldwin 	/**
500bf2276f3SJohn Baldwin 	 * This assumes that calls to fail_point_use_timeout_path()
501bf2276f3SJohn Baldwin 	 * will not race.
502bf2276f3SJohn Baldwin 	 */
503bf2276f3SJohn Baldwin 	if (fp->fp_callout != NULL)
504bf2276f3SJohn Baldwin 		return;
505bf2276f3SJohn Baldwin 	fp->fp_callout = fp_malloc(sizeof(*fp->fp_callout), M_WAITOK);
506bf2276f3SJohn Baldwin 	callout_init(fp->fp_callout, CALLOUT_MPSAFE);
507bf2276f3SJohn Baldwin }
508bf2276f3SJohn Baldwin 
509cfeb7489SZachary Loafman /**
51070e20d4eSConrad Meyer  * Free the resources held by a fail_point, and wake any paused threads.
51170e20d4eSConrad Meyer  * Thou shalt not allow threads to hit this fail point after you enter this
51270e20d4eSConrad Meyer  * function, nor shall you call this multiple times for a given fp.
513cfeb7489SZachary Loafman  * @ingroup failpoint
514cfeb7489SZachary Loafman  */
515cfeb7489SZachary Loafman void
fail_point_destroy(struct fail_point * fp)516cfeb7489SZachary Loafman fail_point_destroy(struct fail_point *fp)
517cfeb7489SZachary Loafman {
518cfeb7489SZachary Loafman 
51970e20d4eSConrad Meyer 	fail_point_drain(fp, 0);
52070e20d4eSConrad Meyer 
5210894a9bcSMatthew D Fleming 	if ((fp->fp_flags & FAIL_POINT_DYNAMIC_NAME) != 0) {
5220894a9bcSMatthew D Fleming 		fp_free(__DECONST(void *, fp->fp_name));
523cfeb7489SZachary Loafman 		fp->fp_name = NULL;
524cfeb7489SZachary Loafman 	}
525cfeb7489SZachary Loafman 	fp->fp_flags = 0;
526bf2276f3SJohn Baldwin 	if (fp->fp_callout) {
527bf2276f3SJohn Baldwin 		fp_free(fp->fp_callout);
528bf2276f3SJohn Baldwin 		fp->fp_callout = NULL;
529bf2276f3SJohn Baldwin 	}
53070e20d4eSConrad Meyer 
53170e20d4eSConrad Meyer 	sx_xlock(&sx_fp_set);
53270e20d4eSConrad Meyer 	fail_point_garbage_collect();
53370e20d4eSConrad Meyer 	sx_xunlock(&sx_fp_set);
534cfeb7489SZachary Loafman }
535cfeb7489SZachary Loafman 
536cfeb7489SZachary Loafman /**
537cfeb7489SZachary Loafman  * This does the real work of evaluating a fail point. If the fail point tells
538cfeb7489SZachary Loafman  * us to return a value, this function returns 1 and fills in 'return_value'
539cfeb7489SZachary Loafman  * (return_value is allowed to be null). If the fail point tells us to panic,
540cfeb7489SZachary Loafman  * we never return. Otherwise we just return 0 after doing some work, which
541cfeb7489SZachary Loafman  * means "keep going".
542cfeb7489SZachary Loafman  */
543cfeb7489SZachary Loafman enum fail_point_return_code
fail_point_eval_nontrivial(struct fail_point * fp,int * return_value)544cfeb7489SZachary Loafman fail_point_eval_nontrivial(struct fail_point *fp, int *return_value)
545cfeb7489SZachary Loafman {
54670e20d4eSConrad Meyer 	bool execute = false;
54770e20d4eSConrad Meyer 	struct fail_point_entry *ent;
54870e20d4eSConrad Meyer 	struct fail_point_setting *fp_setting;
54970e20d4eSConrad Meyer 	enum fail_point_return_code ret;
55070e20d4eSConrad Meyer 	int cont;
55170e20d4eSConrad Meyer 	int count;
552cfeb7489SZachary Loafman 	int msecs;
55370e20d4eSConrad Meyer 	int usecs;
554cfeb7489SZachary Loafman 
55570e20d4eSConrad Meyer 	ret = FAIL_POINT_RC_CONTINUE;
55670e20d4eSConrad Meyer 	cont = 0; /* don't continue by default */
557cfeb7489SZachary Loafman 
55870e20d4eSConrad Meyer 	fp_setting = fail_point_setting_get_ref(fp);
55970e20d4eSConrad Meyer 	if (fp_setting == NULL)
56070e20d4eSConrad Meyer 		goto abort;
56170e20d4eSConrad Meyer 
56270e20d4eSConrad Meyer 	TAILQ_FOREACH(ent, &fp_setting->fp_entry_queue, fe_entries) {
56370e20d4eSConrad Meyer 		if (ent->fe_stale)
56470e20d4eSConrad Meyer 			continue;
565cfeb7489SZachary Loafman 
566cfeb7489SZachary Loafman 		if (ent->fe_prob < PROB_MAX &&
5670894a9bcSMatthew D Fleming 		    ent->fe_prob < random() % PROB_MAX)
5680894a9bcSMatthew D Fleming 			continue;
56970e20d4eSConrad Meyer 
570b5bd50aeSMatthew D Fleming 		if (ent->fe_pid != NO_PID && ent->fe_pid != curproc->p_pid)
571b5bd50aeSMatthew D Fleming 			continue;
572cfeb7489SZachary Loafman 
57370e20d4eSConrad Meyer 		if (ent->fe_count != FE_COUNT_UNTRACKED) {
57470e20d4eSConrad Meyer 			count = ent->fe_count;
57570e20d4eSConrad Meyer 			while (count > 0) {
57670e20d4eSConrad Meyer 				if (atomic_cmpset_32(&ent->fe_count, count, count - 1)) {
57770e20d4eSConrad Meyer 					count--;
57870e20d4eSConrad Meyer 					execute = true;
57970e20d4eSConrad Meyer 					break;
58070e20d4eSConrad Meyer 				}
58170e20d4eSConrad Meyer 				count = ent->fe_count;
58270e20d4eSConrad Meyer 			}
58370e20d4eSConrad Meyer 			if (execute == false)
58470e20d4eSConrad Meyer 				/* We lost the race; consider the entry stale and bail now */
58570e20d4eSConrad Meyer 				continue;
58670e20d4eSConrad Meyer 			if (count == 0)
58770e20d4eSConrad Meyer 				ent->fe_stale = true;
58870e20d4eSConrad Meyer 		}
58970e20d4eSConrad Meyer 
590cfeb7489SZachary Loafman 		switch (ent->fe_type) {
591cfeb7489SZachary Loafman 		case FAIL_POINT_PANIC:
592cfeb7489SZachary Loafman 			panic("fail point %s panicking", fp->fp_name);
593cfeb7489SZachary Loafman 			/* NOTREACHED */
594cfeb7489SZachary Loafman 
595cfeb7489SZachary Loafman 		case FAIL_POINT_RETURN:
5960894a9bcSMatthew D Fleming 			if (return_value != NULL)
597cfeb7489SZachary Loafman 				*return_value = ent->fe_arg;
598cfeb7489SZachary Loafman 			ret = FAIL_POINT_RC_RETURN;
599cfeb7489SZachary Loafman 			break;
600cfeb7489SZachary Loafman 
601cfeb7489SZachary Loafman 		case FAIL_POINT_BREAK:
6020894a9bcSMatthew D Fleming 			printf("fail point %s breaking to debugger\n",
6030894a9bcSMatthew D Fleming 			        fp->fp_name);
604cfeb7489SZachary Loafman 			breakpoint();
605cfeb7489SZachary Loafman 			break;
606cfeb7489SZachary Loafman 
607cfeb7489SZachary Loafman 		case FAIL_POINT_PRINT:
608cfeb7489SZachary Loafman 			printf("fail point %s executing\n", fp->fp_name);
609cfeb7489SZachary Loafman 			cont = ent->fe_arg;
610cfeb7489SZachary Loafman 			break;
611cfeb7489SZachary Loafman 
612cfeb7489SZachary Loafman 		case FAIL_POINT_SLEEP:
613cfeb7489SZachary Loafman 			msecs = ent->fe_arg;
614cfeb7489SZachary Loafman 			if (msecs)
61570e20d4eSConrad Meyer 				fail_point_sleep(fp, msecs, &ret);
61670e20d4eSConrad Meyer 			break;
61770e20d4eSConrad Meyer 
61870e20d4eSConrad Meyer 		case FAIL_POINT_PAUSE:
61970e20d4eSConrad Meyer 			/**
62070e20d4eSConrad Meyer 			 * Pausing is inherently strange with multiple
62170e20d4eSConrad Meyer 			 * entries given our design.  That is because some
62270e20d4eSConrad Meyer 			 * entries could be unreachable, for instance in cases like:
62370e20d4eSConrad Meyer 			 * pause->return. We can never reach the return entry.
62470e20d4eSConrad Meyer 			 * The sysctl layer actually truncates all entries after
62570e20d4eSConrad Meyer 			 * a pause for this reason.
62670e20d4eSConrad Meyer 			 */
62770e20d4eSConrad Meyer 			mtx_lock_spin(&fp_setting->feq_mtx);
62870e20d4eSConrad Meyer 			fail_point_pause(fp, &ret, &fp_setting->feq_mtx);
62970e20d4eSConrad Meyer 			mtx_unlock_spin(&fp_setting->feq_mtx);
63070e20d4eSConrad Meyer 			break;
63170e20d4eSConrad Meyer 
63270e20d4eSConrad Meyer 		case FAIL_POINT_YIELD:
6333bd485f9SMark Johnston 			kern_yield(PRI_UNCHANGED);
63470e20d4eSConrad Meyer 			break;
63570e20d4eSConrad Meyer 
63670e20d4eSConrad Meyer 		case FAIL_POINT_DELAY:
63770e20d4eSConrad Meyer 			usecs = ent->fe_arg;
63870e20d4eSConrad Meyer 			DELAY(usecs);
639cfeb7489SZachary Loafman 			break;
640cfeb7489SZachary Loafman 
641cfeb7489SZachary Loafman 		default:
642cfeb7489SZachary Loafman 			break;
643cfeb7489SZachary Loafman 		}
644cfeb7489SZachary Loafman 
6450894a9bcSMatthew D Fleming 		if (cont == 0)
646cfeb7489SZachary Loafman 			break;
647cfeb7489SZachary Loafman 	}
648cfeb7489SZachary Loafman 
64970e20d4eSConrad Meyer 	if (fail_point_is_off(fp))
65070e20d4eSConrad Meyer 		fail_point_eval_swap_out(fp, fp_setting);
651cfeb7489SZachary Loafman 
65270e20d4eSConrad Meyer abort:
65370e20d4eSConrad Meyer 	fail_point_setting_release_ref(fp);
654cfeb7489SZachary Loafman 
6550894a9bcSMatthew D Fleming 	return (ret);
656cfeb7489SZachary Loafman }
657cfeb7489SZachary Loafman 
658cfeb7489SZachary Loafman /**
659cfeb7489SZachary Loafman  * Translate internal fail_point structure into human-readable text.
660cfeb7489SZachary Loafman  */
661cfeb7489SZachary Loafman static void
fail_point_get(struct fail_point * fp,struct sbuf * sb,bool verbose)66270e20d4eSConrad Meyer fail_point_get(struct fail_point *fp, struct sbuf *sb,
66370e20d4eSConrad Meyer         bool verbose)
664cfeb7489SZachary Loafman {
665cfeb7489SZachary Loafman 	struct fail_point_entry *ent;
66670e20d4eSConrad Meyer 	struct fail_point_setting *fp_setting;
66770e20d4eSConrad Meyer 	struct fail_point_entry *fp_entry_cpy;
66870e20d4eSConrad Meyer 	int cnt_sleeping;
66970e20d4eSConrad Meyer 	int idx;
67070e20d4eSConrad Meyer 	int printed_entry_count;
671cfeb7489SZachary Loafman 
67270e20d4eSConrad Meyer 	cnt_sleeping = 0;
67370e20d4eSConrad Meyer 	idx = 0;
67470e20d4eSConrad Meyer 	printed_entry_count = 0;
675cfeb7489SZachary Loafman 
67670e20d4eSConrad Meyer 	fp_entry_cpy = fp_malloc(sizeof(struct fail_point_entry) *
67770e20d4eSConrad Meyer 	        (FP_MAX_ENTRY_COUNT + 1), M_WAITOK);
67870e20d4eSConrad Meyer 
67970e20d4eSConrad Meyer 	fp_setting = fail_point_setting_get_ref(fp);
68070e20d4eSConrad Meyer 
68170e20d4eSConrad Meyer 	if (fp_setting != NULL) {
68270e20d4eSConrad Meyer 		TAILQ_FOREACH(ent, &fp_setting->fp_entry_queue, fe_entries) {
68370e20d4eSConrad Meyer 			if (ent->fe_stale)
68470e20d4eSConrad Meyer 				continue;
68570e20d4eSConrad Meyer 
68670e20d4eSConrad Meyer 			KASSERT(printed_entry_count < FP_MAX_ENTRY_COUNT,
68770e20d4eSConrad Meyer 			        ("FP entry list larger than allowed"));
68870e20d4eSConrad Meyer 
68970e20d4eSConrad Meyer 			fp_entry_cpy[printed_entry_count] = *ent;
69070e20d4eSConrad Meyer 			++printed_entry_count;
69170e20d4eSConrad Meyer 		}
69270e20d4eSConrad Meyer 	}
69370e20d4eSConrad Meyer 	fail_point_setting_release_ref(fp);
69470e20d4eSConrad Meyer 
69570e20d4eSConrad Meyer 	/* This is our equivalent of a NULL terminator */
69670e20d4eSConrad Meyer 	fp_entry_cpy[printed_entry_count].fe_type = FAIL_POINT_INVALID;
69770e20d4eSConrad Meyer 
69870e20d4eSConrad Meyer 	while (idx < printed_entry_count) {
69970e20d4eSConrad Meyer 		ent = &fp_entry_cpy[idx];
70070e20d4eSConrad Meyer 		++idx;
701cfeb7489SZachary Loafman 		if (ent->fe_prob < PROB_MAX) {
702cfeb7489SZachary Loafman 			int decimal = ent->fe_prob % (PROB_MAX / 100);
703cfeb7489SZachary Loafman 			int units = ent->fe_prob / (PROB_MAX / 100);
704cfeb7489SZachary Loafman 			sbuf_printf(sb, "%d", units);
705cfeb7489SZachary Loafman 			if (decimal) {
706cfeb7489SZachary Loafman 				int digits = PROB_DIGITS - 2;
707cfeb7489SZachary Loafman 				while (!(decimal % 10)) {
708cfeb7489SZachary Loafman 					digits--;
709cfeb7489SZachary Loafman 					decimal /= 10;
710cfeb7489SZachary Loafman 				}
711cfeb7489SZachary Loafman 				sbuf_printf(sb, ".%0*d", digits, decimal);
712cfeb7489SZachary Loafman 			}
713cfeb7489SZachary Loafman 			sbuf_printf(sb, "%%");
714cfeb7489SZachary Loafman 		}
71570e20d4eSConrad Meyer 		if (ent->fe_count >= 0)
716cfeb7489SZachary Loafman 			sbuf_printf(sb, "%d*", ent->fe_count);
7170894a9bcSMatthew D Fleming 		sbuf_printf(sb, "%s", fail_type_strings[ent->fe_type].name);
718cfeb7489SZachary Loafman 		if (ent->fe_arg)
719cfeb7489SZachary Loafman 			sbuf_printf(sb, "(%d)", ent->fe_arg);
720b5bd50aeSMatthew D Fleming 		if (ent->fe_pid != NO_PID)
721b5bd50aeSMatthew D Fleming 			sbuf_printf(sb, "[pid %d]", ent->fe_pid);
722cfeb7489SZachary Loafman 		if (TAILQ_NEXT(ent, fe_entries))
7230a713948SAlexander Motin 			sbuf_cat(sb, "->");
724cfeb7489SZachary Loafman 	}
72570e20d4eSConrad Meyer 	if (!printed_entry_count)
7260a713948SAlexander Motin 		sbuf_cat(sb, "off");
727cfeb7489SZachary Loafman 
72870e20d4eSConrad Meyer 	fp_free(fp_entry_cpy);
72970e20d4eSConrad Meyer 	if (verbose) {
73082e17adcSConrad Meyer #ifdef STACK
73170e20d4eSConrad Meyer 		/* Print number of sleeping threads. queue=0 is the argument
73270e20d4eSConrad Meyer 		 * used by msleep when sending our threads to sleep. */
7330a713948SAlexander Motin 		sbuf_cat(sb, "\nsleeping_thread_stacks = {\n");
73470e20d4eSConrad Meyer 		sleepq_sbuf_print_stacks(sb, FP_SLEEP_CHANNEL(fp), 0,
73570e20d4eSConrad Meyer 		        &cnt_sleeping);
73670e20d4eSConrad Meyer 
7370a713948SAlexander Motin 		sbuf_cat(sb, "},\n");
73882e17adcSConrad Meyer #endif
73970e20d4eSConrad Meyer 		sbuf_printf(sb, "sleeping_thread_count = %d,\n",
74070e20d4eSConrad Meyer 		        cnt_sleeping);
74170e20d4eSConrad Meyer 
74282e17adcSConrad Meyer #ifdef STACK
7430a713948SAlexander Motin 		sbuf_cat(sb, "paused_thread_stacks = {\n");
74470e20d4eSConrad Meyer 		sleepq_sbuf_print_stacks(sb, FP_PAUSE_CHANNEL(fp), 0,
74570e20d4eSConrad Meyer 		        &cnt_sleeping);
74670e20d4eSConrad Meyer 
7470a713948SAlexander Motin 		sbuf_cat(sb, "},\n");
74882e17adcSConrad Meyer #endif
74970e20d4eSConrad Meyer 		sbuf_printf(sb, "paused_thread_count = %d\n",
75070e20d4eSConrad Meyer 		        cnt_sleeping);
75170e20d4eSConrad Meyer 	}
752cfeb7489SZachary Loafman }
753cfeb7489SZachary Loafman 
754cfeb7489SZachary Loafman /**
755cfeb7489SZachary Loafman  * Set an internal fail_point structure from a human-readable failpoint string
756cfeb7489SZachary Loafman  * in a lock-safe manner.
757cfeb7489SZachary Loafman  */
758cfeb7489SZachary Loafman static int
fail_point_set(struct fail_point * fp,char * buf)759cfeb7489SZachary Loafman fail_point_set(struct fail_point *fp, char *buf)
760cfeb7489SZachary Loafman {
761cfeb7489SZachary Loafman 	struct fail_point_entry *ent, *ent_next;
76270e20d4eSConrad Meyer 	struct fail_point_setting *entries;
76370e20d4eSConrad Meyer 	bool should_wake_paused;
76470e20d4eSConrad Meyer 	bool should_truncate;
76570e20d4eSConrad Meyer 	int error;
76670e20d4eSConrad Meyer 
76770e20d4eSConrad Meyer 	error = 0;
76870e20d4eSConrad Meyer 	should_wake_paused = false;
76970e20d4eSConrad Meyer 	should_truncate = false;
770cfeb7489SZachary Loafman 
771cfeb7489SZachary Loafman 	/* Parse new entries. */
77270e20d4eSConrad Meyer 	/**
77370e20d4eSConrad Meyer 	 * ref protects our new malloc'd stuff from being garbage collected
77470e20d4eSConrad Meyer 	 * before we link it.
77570e20d4eSConrad Meyer 	 */
77670e20d4eSConrad Meyer 	fail_point_setting_get_ref(fp);
77770e20d4eSConrad Meyer 	entries = fail_point_setting_new(fp);
77870e20d4eSConrad Meyer 	if (parse_fail_point(entries, buf) == NULL) {
77970e20d4eSConrad Meyer 		STAILQ_REMOVE(&fp_setting_garbage, entries,
78070e20d4eSConrad Meyer 		        fail_point_setting, fs_garbage_link);
78170e20d4eSConrad Meyer 		fail_point_setting_destroy(entries);
782cfeb7489SZachary Loafman 		error = EINVAL;
783cfeb7489SZachary Loafman 		goto end;
784cfeb7489SZachary Loafman 	}
785cfeb7489SZachary Loafman 
78670e20d4eSConrad Meyer 	/**
78770e20d4eSConrad Meyer 	 * Transfer the entries we are going to keep to a new list.
78870e20d4eSConrad Meyer 	 * Get rid of useless zero probability entries, and entries with hit
78970e20d4eSConrad Meyer 	 * count 0.
79070e20d4eSConrad Meyer 	 * If 'off' is present, and it has no hit count set, then all entries
79170e20d4eSConrad Meyer 	 *       after it are discarded since they are unreachable.
79270e20d4eSConrad Meyer 	 */
79370e20d4eSConrad Meyer 	TAILQ_FOREACH_SAFE(ent, &entries->fp_entry_queue, fe_entries, ent_next) {
79470e20d4eSConrad Meyer 		if (ent->fe_prob == 0 || ent->fe_count == 0) {
79570e20d4eSConrad Meyer 			printf("Discarding entry which cannot execute %s\n",
79670e20d4eSConrad Meyer 			        fail_type_strings[ent->fe_type].name);
79770e20d4eSConrad Meyer 			TAILQ_REMOVE(&entries->fp_entry_queue, ent,
79870e20d4eSConrad Meyer 			        fe_entries);
79970e20d4eSConrad Meyer 			fp_free(ent);
80070e20d4eSConrad Meyer 			continue;
80170e20d4eSConrad Meyer 		} else if (should_truncate) {
80270e20d4eSConrad Meyer 			printf("Discarding unreachable entry %s\n",
80370e20d4eSConrad Meyer 			        fail_type_strings[ent->fe_type].name);
80470e20d4eSConrad Meyer 			TAILQ_REMOVE(&entries->fp_entry_queue, ent,
80570e20d4eSConrad Meyer 			        fe_entries);
80670e20d4eSConrad Meyer 			fp_free(ent);
80770e20d4eSConrad Meyer 			continue;
808cfeb7489SZachary Loafman 		}
809cfeb7489SZachary Loafman 
81070e20d4eSConrad Meyer 		if (ent->fe_type == FAIL_POINT_OFF) {
81170e20d4eSConrad Meyer 			should_wake_paused = true;
81270e20d4eSConrad Meyer 			if (ent->fe_count == FE_COUNT_UNTRACKED) {
81370e20d4eSConrad Meyer 				should_truncate = true;
81470e20d4eSConrad Meyer 				TAILQ_REMOVE(&entries->fp_entry_queue, ent,
81570e20d4eSConrad Meyer 				        fe_entries);
81670e20d4eSConrad Meyer 				fp_free(ent);
81770e20d4eSConrad Meyer 			}
81870e20d4eSConrad Meyer 		} else if (ent->fe_type == FAIL_POINT_PAUSE) {
81970e20d4eSConrad Meyer 			should_truncate = true;
82070e20d4eSConrad Meyer 		} else if (ent->fe_type == FAIL_POINT_SLEEP && (fp->fp_flags &
82170e20d4eSConrad Meyer 		        FAIL_POINT_NONSLEEPABLE)) {
82270e20d4eSConrad Meyer 			/**
82370e20d4eSConrad Meyer 			 * If this fail point is annotated as being in a
82470e20d4eSConrad Meyer 			 * non-sleepable ctx, convert sleep to delay and
82570e20d4eSConrad Meyer 			 * convert the msec argument to usecs.
82670e20d4eSConrad Meyer 			 */
82770e20d4eSConrad Meyer 			printf("Sleep call request on fail point in "
82870e20d4eSConrad Meyer 			        "non-sleepable context; using delay instead "
82970e20d4eSConrad Meyer 			        "of sleep\n");
83070e20d4eSConrad Meyer 			ent->fe_type = FAIL_POINT_DELAY;
83170e20d4eSConrad Meyer 			ent->fe_arg *= 1000;
83270e20d4eSConrad Meyer 		}
83370e20d4eSConrad Meyer 	}
834cfeb7489SZachary Loafman 
83570e20d4eSConrad Meyer 	if (TAILQ_EMPTY(&entries->fp_entry_queue)) {
83670e20d4eSConrad Meyer 		entries = fail_point_swap_settings(fp, NULL);
83770e20d4eSConrad Meyer 		if (entries != NULL)
83870e20d4eSConrad Meyer 			wakeup(FP_PAUSE_CHANNEL(fp));
83970e20d4eSConrad Meyer 	} else {
84070e20d4eSConrad Meyer 		if (should_wake_paused)
84170e20d4eSConrad Meyer 			wakeup(FP_PAUSE_CHANNEL(fp));
84270e20d4eSConrad Meyer 		fail_point_swap_settings(fp, entries);
84370e20d4eSConrad Meyer 	}
844cfeb7489SZachary Loafman 
845cfeb7489SZachary Loafman end:
846cfeb7489SZachary Loafman #ifdef IWARNING
847cfeb7489SZachary Loafman 	if (error)
8484b7c6844SMatthew D Fleming 		IWARNING("Failed to set %s %s to %s",
849cfeb7489SZachary Loafman 		    fp->fp_name, fp->fp_location, buf);
850cfeb7489SZachary Loafman 	else
8514b7c6844SMatthew D Fleming 		INOTICE("Set %s %s to %s",
852cfeb7489SZachary Loafman 		    fp->fp_name, fp->fp_location, buf);
853cfeb7489SZachary Loafman #endif /* IWARNING */
854cfeb7489SZachary Loafman 
85570e20d4eSConrad Meyer 	fail_point_setting_release_ref(fp);
8560894a9bcSMatthew D Fleming 	return (error);
857cfeb7489SZachary Loafman }
858cfeb7489SZachary Loafman 
859cfeb7489SZachary Loafman #define MAX_FAIL_POINT_BUF	1023
860cfeb7489SZachary Loafman 
861cfeb7489SZachary Loafman /**
862cfeb7489SZachary Loafman  * Handle kernel failpoint set/get.
863cfeb7489SZachary Loafman  */
864cfeb7489SZachary Loafman int
fail_point_sysctl(SYSCTL_HANDLER_ARGS)865cfeb7489SZachary Loafman fail_point_sysctl(SYSCTL_HANDLER_ARGS)
866cfeb7489SZachary Loafman {
86770e20d4eSConrad Meyer 	struct fail_point *fp;
86870e20d4eSConrad Meyer 	char *buf;
869508d8569SMark Johnston 	struct sbuf sb, *sb_check;
870cfeb7489SZachary Loafman 	int error;
871cfeb7489SZachary Loafman 
872508d8569SMark Johnston 	buf = NULL;
87370e20d4eSConrad Meyer 	error = 0;
87470e20d4eSConrad Meyer 	fp = arg1;
87570e20d4eSConrad Meyer 
87670e20d4eSConrad Meyer 	sb_check = sbuf_new(&sb, NULL, 1024, SBUF_AUTOEXTEND);
87770e20d4eSConrad Meyer 	if (sb_check != &sb)
87870e20d4eSConrad Meyer 		return (ENOMEM);
87970e20d4eSConrad Meyer 
88070e20d4eSConrad Meyer 	sbuf_set_drain(&sb, (sbuf_drain_func *)fail_sysctl_drain_func, req);
881cfeb7489SZachary Loafman 
882cfeb7489SZachary Loafman 	/* Setting */
88370e20d4eSConrad Meyer 	/**
88470e20d4eSConrad Meyer 	 * Lock protects any new entries from being garbage collected before we
88570e20d4eSConrad Meyer 	 * can link them to the fail point.
88670e20d4eSConrad Meyer 	 */
88770e20d4eSConrad Meyer 	sx_xlock(&sx_fp_set);
88870e20d4eSConrad Meyer 	if (req->newptr) {
889cfeb7489SZachary Loafman 		if (req->newlen > MAX_FAIL_POINT_BUF) {
890cfeb7489SZachary Loafman 			error = EINVAL;
891cfeb7489SZachary Loafman 			goto out;
892cfeb7489SZachary Loafman 		}
893cfeb7489SZachary Loafman 
894cfeb7489SZachary Loafman 		buf = fp_malloc(req->newlen + 1, M_WAITOK);
895cfeb7489SZachary Loafman 
896cfeb7489SZachary Loafman 		error = SYSCTL_IN(req, buf, req->newlen);
897cfeb7489SZachary Loafman 		if (error)
898cfeb7489SZachary Loafman 			goto out;
899cfeb7489SZachary Loafman 		buf[req->newlen] = '\0';
900cfeb7489SZachary Loafman 
901cfeb7489SZachary Loafman 		error = fail_point_set(fp, buf);
902cfeb7489SZachary Loafman 	}
903cfeb7489SZachary Loafman 
90470e20d4eSConrad Meyer 	fail_point_garbage_collect();
90570e20d4eSConrad Meyer 	sx_xunlock(&sx_fp_set);
90670e20d4eSConrad Meyer 
90770e20d4eSConrad Meyer 	/* Retrieving. */
90870e20d4eSConrad Meyer 	fail_point_get(fp, &sb, false);
90970e20d4eSConrad Meyer 
910cfeb7489SZachary Loafman out:
91170e20d4eSConrad Meyer 	sbuf_finish(&sb);
91270e20d4eSConrad Meyer 	sbuf_delete(&sb);
91370e20d4eSConrad Meyer 
91470e20d4eSConrad Meyer 	if (buf)
915cfeb7489SZachary Loafman 		fp_free(buf);
91670e20d4eSConrad Meyer 
9170894a9bcSMatthew D Fleming 	return (error);
918cfeb7489SZachary Loafman }
919cfeb7489SZachary Loafman 
92070e20d4eSConrad Meyer int
fail_point_sysctl_status(SYSCTL_HANDLER_ARGS)92170e20d4eSConrad Meyer fail_point_sysctl_status(SYSCTL_HANDLER_ARGS)
92270e20d4eSConrad Meyer {
92370e20d4eSConrad Meyer 	struct fail_point *fp;
92470e20d4eSConrad Meyer 	struct sbuf sb, *sb_check;
92570e20d4eSConrad Meyer 
92670e20d4eSConrad Meyer 	fp = arg1;
92770e20d4eSConrad Meyer 
92870e20d4eSConrad Meyer 	sb_check = sbuf_new(&sb, NULL, 1024, SBUF_AUTOEXTEND);
92970e20d4eSConrad Meyer 	if (sb_check != &sb)
93070e20d4eSConrad Meyer 		return (ENOMEM);
93170e20d4eSConrad Meyer 
93270e20d4eSConrad Meyer 	sbuf_set_drain(&sb, (sbuf_drain_func *)fail_sysctl_drain_func, req);
93370e20d4eSConrad Meyer 
93470e20d4eSConrad Meyer 	/* Retrieving. */
93570e20d4eSConrad Meyer 	fail_point_get(fp, &sb, true);
93670e20d4eSConrad Meyer 
93770e20d4eSConrad Meyer 	sbuf_finish(&sb);
93870e20d4eSConrad Meyer 	sbuf_delete(&sb);
93970e20d4eSConrad Meyer 
94070e20d4eSConrad Meyer 	/**
94170e20d4eSConrad Meyer 	 * Lock protects any new entries from being garbage collected before we
94270e20d4eSConrad Meyer 	 * can link them to the fail point.
94370e20d4eSConrad Meyer 	 */
94470e20d4eSConrad Meyer 	sx_xlock(&sx_fp_set);
94570e20d4eSConrad Meyer 	fail_point_garbage_collect();
94670e20d4eSConrad Meyer 	sx_xunlock(&sx_fp_set);
94770e20d4eSConrad Meyer 
94870e20d4eSConrad Meyer 	return (0);
94970e20d4eSConrad Meyer }
95070e20d4eSConrad Meyer 
95170e20d4eSConrad Meyer int
fail_sysctl_drain_func(void * sysctl_args,const char * buf,int len)95270e20d4eSConrad Meyer fail_sysctl_drain_func(void *sysctl_args, const char *buf, int len)
95370e20d4eSConrad Meyer {
95470e20d4eSConrad Meyer 	struct sysctl_req *sa;
95570e20d4eSConrad Meyer 	int error;
95670e20d4eSConrad Meyer 
95770e20d4eSConrad Meyer 	sa = sysctl_args;
95870e20d4eSConrad Meyer 
95970e20d4eSConrad Meyer 	error = SYSCTL_OUT(sa, buf, len);
96070e20d4eSConrad Meyer 
96170e20d4eSConrad Meyer 	if (error == ENOMEM)
96270e20d4eSConrad Meyer 		return (-1);
96370e20d4eSConrad Meyer 	else
96470e20d4eSConrad Meyer 		return (len);
96570e20d4eSConrad Meyer }
96670e20d4eSConrad Meyer 
967cfeb7489SZachary Loafman /**
968cfeb7489SZachary Loafman  * Internal helper function to translate a human-readable failpoint string
969cfeb7489SZachary Loafman  * into a internally-parsable fail_point structure.
970cfeb7489SZachary Loafman  */
971cfeb7489SZachary Loafman static char *
parse_fail_point(struct fail_point_setting * ents,char * p)97270e20d4eSConrad Meyer parse_fail_point(struct fail_point_setting *ents, char *p)
973cfeb7489SZachary Loafman {
974cfeb7489SZachary Loafman 	/*  <fail_point> ::
975cfeb7489SZachary Loafman 	 *      <term> ( "->" <term> )*
976cfeb7489SZachary Loafman 	 */
97770e20d4eSConrad Meyer 	uint8_t term_count;
97870e20d4eSConrad Meyer 
97970e20d4eSConrad Meyer 	term_count = 1;
98070e20d4eSConrad Meyer 
9810894a9bcSMatthew D Fleming 	p = parse_term(ents, p);
9820894a9bcSMatthew D Fleming 	if (p == NULL)
9830894a9bcSMatthew D Fleming 		return (NULL);
98470e20d4eSConrad Meyer 
9850894a9bcSMatthew D Fleming 	while (*p != '\0') {
98670e20d4eSConrad Meyer 		term_count++;
98770e20d4eSConrad Meyer 		if (p[0] != '-' || p[1] != '>' ||
98870e20d4eSConrad Meyer 		        (p = parse_term(ents, p+2)) == NULL ||
98970e20d4eSConrad Meyer 		        term_count > FP_MAX_ENTRY_COUNT)
9900894a9bcSMatthew D Fleming 			return (NULL);
9910894a9bcSMatthew D Fleming 	}
9920894a9bcSMatthew D Fleming 	return (p);
993cfeb7489SZachary Loafman }
994cfeb7489SZachary Loafman 
995cfeb7489SZachary Loafman /**
996cfeb7489SZachary Loafman  * Internal helper function to parse an individual term from a failpoint.
997cfeb7489SZachary Loafman  */
998cfeb7489SZachary Loafman static char *
parse_term(struct fail_point_setting * ents,char * p)99970e20d4eSConrad Meyer parse_term(struct fail_point_setting *ents, char *p)
1000cfeb7489SZachary Loafman {
1001cfeb7489SZachary Loafman 	struct fail_point_entry *ent;
1002cfeb7489SZachary Loafman 
100370e20d4eSConrad Meyer 	ent = fail_point_entry_new(ents);
1004cfeb7489SZachary Loafman 
1005cfeb7489SZachary Loafman 	/*
1006cfeb7489SZachary Loafman 	 * <term> ::
1007cfeb7489SZachary Loafman 	 *     ( (<float> "%") | (<integer> "*" ) )*
1008cfeb7489SZachary Loafman 	 *     <type>
1009cfeb7489SZachary Loafman 	 *     [ "(" <integer> ")" ]
1010b5bd50aeSMatthew D Fleming 	 *     [ "[pid " <integer> "]" ]
1011cfeb7489SZachary Loafman 	 */
1012cfeb7489SZachary Loafman 
1013cfeb7489SZachary Loafman 	/* ( (<float> "%") | (<integer> "*" ) )* */
10140894a9bcSMatthew D Fleming 	while (isdigit(*p) || *p == '.') {
1015cfeb7489SZachary Loafman 		int units, decimal;
1016cfeb7489SZachary Loafman 
10170894a9bcSMatthew D Fleming 		p = parse_number(&units, &decimal, p);
10180894a9bcSMatthew D Fleming 		if (p == NULL)
10190894a9bcSMatthew D Fleming 			return (NULL);
1020cfeb7489SZachary Loafman 
1021cfeb7489SZachary Loafman 		if (*p == '%') {
1022cfeb7489SZachary Loafman 			if (units > 100) /* prevent overflow early */
1023cfeb7489SZachary Loafman 				units = 100;
1024cfeb7489SZachary Loafman 			ent->fe_prob = units * (PROB_MAX / 100) + decimal;
1025cfeb7489SZachary Loafman 			if (ent->fe_prob > PROB_MAX)
1026cfeb7489SZachary Loafman 				ent->fe_prob = PROB_MAX;
1027cfeb7489SZachary Loafman 		} else if (*p == '*') {
102870e20d4eSConrad Meyer 			if (!units || units < 0 || decimal)
10290894a9bcSMatthew D Fleming 				return (NULL);
1030c2ede4b3SMartin Blapp 			ent->fe_count = units;
10310894a9bcSMatthew D Fleming 		} else
10320894a9bcSMatthew D Fleming 			return (NULL);
1033cfeb7489SZachary Loafman 		p++;
1034cfeb7489SZachary Loafman 	}
1035cfeb7489SZachary Loafman 
1036cfeb7489SZachary Loafman 	/* <type> */
10370894a9bcSMatthew D Fleming 	p = parse_type(ent, p);
10380894a9bcSMatthew D Fleming 	if (p == NULL)
10390894a9bcSMatthew D Fleming 		return (NULL);
1040cfeb7489SZachary Loafman 	if (*p == '\0')
10410894a9bcSMatthew D Fleming 		return (p);
1042cfeb7489SZachary Loafman 
1043cfeb7489SZachary Loafman 	/* [ "(" <integer> ")" ] */
1044cfeb7489SZachary Loafman 	if (*p != '(')
104570e20d4eSConrad Meyer 		return (p);
1046cfeb7489SZachary Loafman 	p++;
10470894a9bcSMatthew D Fleming 	if (!isdigit(*p) && *p != '-')
10480894a9bcSMatthew D Fleming 		return (NULL);
1049cfeb7489SZachary Loafman 	ent->fe_arg = strtol(p, &p, 0);
1050cfeb7489SZachary Loafman 	if (*p++ != ')')
10510894a9bcSMatthew D Fleming 		return (NULL);
1052cfeb7489SZachary Loafman 
1053b5bd50aeSMatthew D Fleming 	/* [ "[pid " <integer> "]" ] */
1054b5bd50aeSMatthew D Fleming #define PID_STRING "[pid "
1055b5bd50aeSMatthew D Fleming 	if (strncmp(p, PID_STRING, sizeof(PID_STRING) - 1) != 0)
1056b5bd50aeSMatthew D Fleming 		return (p);
1057b5bd50aeSMatthew D Fleming 	p += sizeof(PID_STRING) - 1;
1058b5bd50aeSMatthew D Fleming 	if (!isdigit(*p))
1059b5bd50aeSMatthew D Fleming 		return (NULL);
1060b5bd50aeSMatthew D Fleming 	ent->fe_pid = strtol(p, &p, 0);
1061b5bd50aeSMatthew D Fleming 	if (*p++ != ']')
1062b5bd50aeSMatthew D Fleming 		return (NULL);
1063b5bd50aeSMatthew D Fleming 
10640894a9bcSMatthew D Fleming 	return (p);
1065cfeb7489SZachary Loafman }
1066cfeb7489SZachary Loafman 
1067cfeb7489SZachary Loafman /**
1068cfeb7489SZachary Loafman  * Internal helper function to parse a numeric for a failpoint term.
1069cfeb7489SZachary Loafman  */
1070cfeb7489SZachary Loafman static char *
parse_number(int * out_units,int * out_decimal,char * p)1071cfeb7489SZachary Loafman parse_number(int *out_units, int *out_decimal, char *p)
1072cfeb7489SZachary Loafman {
1073cfeb7489SZachary Loafman 	char *old_p;
1074cfeb7489SZachary Loafman 
107570e20d4eSConrad Meyer 	/**
1076cfeb7489SZachary Loafman 	 *  <number> ::
1077cfeb7489SZachary Loafman 	 *      <integer> [ "." <integer> ] |
1078cfeb7489SZachary Loafman 	 *      "." <integer>
1079cfeb7489SZachary Loafman 	 */
1080cfeb7489SZachary Loafman 
1081cfeb7489SZachary Loafman 	/* whole part */
1082cfeb7489SZachary Loafman 	old_p = p;
1083c2ede4b3SMartin Blapp 	*out_units = strtol(p, &p, 10);
1084cfeb7489SZachary Loafman 	if (p == old_p && *p != '.')
10850894a9bcSMatthew D Fleming 		return (NULL);
1086cfeb7489SZachary Loafman 
1087cfeb7489SZachary Loafman 	/* fractional part */
1088cfeb7489SZachary Loafman 	*out_decimal = 0;
1089cfeb7489SZachary Loafman 	if (*p == '.') {
1090cfeb7489SZachary Loafman 		int digits = 0;
1091cfeb7489SZachary Loafman 		p++;
10920894a9bcSMatthew D Fleming 		while (isdigit(*p)) {
1093cfeb7489SZachary Loafman 			int digit = *p - '0';
1094cfeb7489SZachary Loafman 			if (digits < PROB_DIGITS - 2)
1095cfeb7489SZachary Loafman 				*out_decimal = *out_decimal * 10 + digit;
1096cfeb7489SZachary Loafman 			else if (digits == PROB_DIGITS - 2 && digit >= 5)
1097cfeb7489SZachary Loafman 				(*out_decimal)++;
1098cfeb7489SZachary Loafman 			digits++;
1099cfeb7489SZachary Loafman 			p++;
1100cfeb7489SZachary Loafman 		}
1101cfeb7489SZachary Loafman 		if (!digits) /* need at least one digit after '.' */
11020894a9bcSMatthew D Fleming 			return (NULL);
1103cfeb7489SZachary Loafman 		while (digits++ < PROB_DIGITS - 2) /* add implicit zeros */
1104cfeb7489SZachary Loafman 			*out_decimal *= 10;
1105cfeb7489SZachary Loafman 	}
1106cfeb7489SZachary Loafman 
11070894a9bcSMatthew D Fleming 	return (p); /* success */
1108cfeb7489SZachary Loafman }
1109cfeb7489SZachary Loafman 
1110cfeb7489SZachary Loafman /**
1111cfeb7489SZachary Loafman  * Internal helper function to parse an individual type for a failpoint term.
1112cfeb7489SZachary Loafman  */
1113cfeb7489SZachary Loafman static char *
parse_type(struct fail_point_entry * ent,char * beg)1114cfeb7489SZachary Loafman parse_type(struct fail_point_entry *ent, char *beg)
1115cfeb7489SZachary Loafman {
1116cfeb7489SZachary Loafman 	enum fail_point_t type;
11170894a9bcSMatthew D Fleming 	int len;
11180894a9bcSMatthew D Fleming 
11190894a9bcSMatthew D Fleming 	for (type = FAIL_POINT_OFF; type < FAIL_POINT_NUMTYPES; type++) {
11200894a9bcSMatthew D Fleming 		len = fail_type_strings[type].nmlen;
11210894a9bcSMatthew D Fleming 		if (strncmp(fail_type_strings[type].name, beg, len) == 0) {
1122cfeb7489SZachary Loafman 			ent->fe_type = type;
11230894a9bcSMatthew D Fleming 			return (beg + len);
1124cfeb7489SZachary Loafman 		}
1125cfeb7489SZachary Loafman 	}
11260894a9bcSMatthew D Fleming 	return (NULL);
1127cfeb7489SZachary Loafman }
1128cfeb7489SZachary Loafman 
1129cfeb7489SZachary Loafman /* The fail point sysctl tree. */
11307029da5cSPawel Biernacki SYSCTL_NODE(_debug, OID_AUTO, fail_point, CTLFLAG_RW | CTLFLAG_MPSAFE, 0,
11317029da5cSPawel Biernacki     "fail points");
113270e20d4eSConrad Meyer 
113370e20d4eSConrad Meyer /* Debugging/testing stuff for fail point */
113470e20d4eSConrad Meyer static int
sysctl_test_fail_point(SYSCTL_HANDLER_ARGS)113570e20d4eSConrad Meyer sysctl_test_fail_point(SYSCTL_HANDLER_ARGS)
113670e20d4eSConrad Meyer {
113770e20d4eSConrad Meyer 
113870e20d4eSConrad Meyer 	KFAIL_POINT_RETURN(DEBUG_FP, test_fail_point);
113970e20d4eSConrad Meyer 	return (0);
114070e20d4eSConrad Meyer }
114170e20d4eSConrad Meyer SYSCTL_OID(_debug_fail_point, OID_AUTO, test_trigger_fail_point,
11427029da5cSPawel Biernacki     CTLTYPE_STRING | CTLFLAG_RD | CTLFLAG_NEEDGIANT, NULL, 0,
11437029da5cSPawel Biernacki     sysctl_test_fail_point, "A",
114470e20d4eSConrad Meyer     "Trigger test fail points");
1145