2 * Copyright (C) 2003 Sistina Software (UK) Limited.
3 * Copyright (C) 2004, 2010 Red Hat, Inc. All rights reserved.
5 * This file is released under the GPL.
8 #include <linux/device-mapper.h>
10 #include <linux/module.h>
11 #include <linux/init.h>
12 #include <linux/blkdev.h>
13 #include <linux/bio.h>
14 #include <linux/slab.h>
16 #define DM_MSG_PREFIX "flakey"
19 * Flakey: Used for testing only, simulates intermittent,
20 * catastrophic device failure.
24 unsigned long start_time
;
27 unsigned down_interval
;
30 static int parse_features(struct dm_arg_set
*as
, struct dm_target
*ti
)
36 static struct dm_arg _args
[] = {
37 {0, 0, "Invalid number of feature args"},
40 /* No feature arguments supplied. */
44 r
= dm_read_arg_group(_args
, as
, &argc
, &ti
->error
);
49 arg_name
= dm_shift_arg(as
);
52 ti
->error
= "Unrecognised flakey feature requested";
60 * Construct a flakey mapping:
61 * <dev_path> <offset> <up interval> <down interval> [<#feature args> [<arg>]*]
63 static int flakey_ctr(struct dm_target
*ti
, unsigned int argc
, char **argv
)
65 static struct dm_arg _args
[] = {
66 {0, UINT_MAX
, "Invalid up interval"},
67 {0, UINT_MAX
, "Invalid down interval"},
72 unsigned long long tmpll
;
80 ti
->error
= "Invalid argument count";
84 fc
= kmalloc(sizeof(*fc
), GFP_KERNEL
);
86 ti
->error
= "Cannot allocate linear context";
89 fc
->start_time
= jiffies
;
91 devname
= dm_shift_arg(&as
);
93 if (sscanf(dm_shift_arg(&as
), "%llu", &tmpll
) != 1) {
94 ti
->error
= "Invalid device sector";
99 r
= dm_read_arg(_args
, &as
, &fc
->up_interval
, &ti
->error
);
103 r
= dm_read_arg(_args
, &as
, &fc
->down_interval
, &ti
->error
);
107 if (!(fc
->up_interval
+ fc
->down_interval
)) {
108 ti
->error
= "Total (up + down) interval is zero";
112 if (fc
->up_interval
+ fc
->down_interval
< fc
->up_interval
) {
113 ti
->error
= "Interval overflow";
117 r
= parse_features(&as
, ti
);
121 if (dm_get_device(ti
, devname
, dm_table_get_mode(ti
->table
), &fc
->dev
)) {
122 ti
->error
= "Device lookup failed";
126 ti
->num_flush_requests
= 1;
127 ti
->num_discard_requests
= 1;
136 static void flakey_dtr(struct dm_target
*ti
)
138 struct flakey_c
*fc
= ti
->private;
140 dm_put_device(ti
, fc
->dev
);
144 static sector_t
flakey_map_sector(struct dm_target
*ti
, sector_t bi_sector
)
146 struct flakey_c
*fc
= ti
->private;
148 return fc
->start
+ dm_target_offset(ti
, bi_sector
);
151 static void flakey_map_bio(struct dm_target
*ti
, struct bio
*bio
)
153 struct flakey_c
*fc
= ti
->private;
155 bio
->bi_bdev
= fc
->dev
->bdev
;
156 if (bio_sectors(bio
))
157 bio
->bi_sector
= flakey_map_sector(ti
, bio
->bi_sector
);
160 static int flakey_map(struct dm_target
*ti
, struct bio
*bio
,
161 union map_info
*map_context
)
163 struct flakey_c
*fc
= ti
->private;
167 elapsed
= (jiffies
- fc
->start_time
) / HZ
;
168 if (elapsed
% (fc
->up_interval
+ fc
->down_interval
) >= fc
->up_interval
)
171 flakey_map_bio(ti
, bio
);
173 return DM_MAPIO_REMAPPED
;
176 static int flakey_status(struct dm_target
*ti
, status_type_t type
,
177 char *result
, unsigned int maxlen
)
179 struct flakey_c
*fc
= ti
->private;
182 case STATUSTYPE_INFO
:
186 case STATUSTYPE_TABLE
:
187 snprintf(result
, maxlen
, "%s %llu %u %u", fc
->dev
->name
,
188 (unsigned long long)fc
->start
, fc
->up_interval
,
195 static int flakey_ioctl(struct dm_target
*ti
, unsigned int cmd
, unsigned long arg
)
197 struct flakey_c
*fc
= ti
->private;
199 return __blkdev_driver_ioctl(fc
->dev
->bdev
, fc
->dev
->mode
, cmd
, arg
);
202 static int flakey_merge(struct dm_target
*ti
, struct bvec_merge_data
*bvm
,
203 struct bio_vec
*biovec
, int max_size
)
205 struct flakey_c
*fc
= ti
->private;
206 struct request_queue
*q
= bdev_get_queue(fc
->dev
->bdev
);
208 if (!q
->merge_bvec_fn
)
211 bvm
->bi_bdev
= fc
->dev
->bdev
;
212 bvm
->bi_sector
= flakey_map_sector(ti
, bvm
->bi_sector
);
214 return min(max_size
, q
->merge_bvec_fn(q
, bvm
, biovec
));
217 static int flakey_iterate_devices(struct dm_target
*ti
, iterate_devices_callout_fn fn
, void *data
)
219 struct flakey_c
*fc
= ti
->private;
221 return fn(ti
, fc
->dev
, fc
->start
, ti
->len
, data
);
224 static struct target_type flakey_target
= {
226 .version
= {1, 2, 0},
227 .module
= THIS_MODULE
,
231 .status
= flakey_status
,
232 .ioctl
= flakey_ioctl
,
233 .merge
= flakey_merge
,
234 .iterate_devices
= flakey_iterate_devices
,
237 static int __init
dm_flakey_init(void)
239 int r
= dm_register_target(&flakey_target
);
242 DMERR("register failed %d", r
);
247 static void __exit
dm_flakey_exit(void)
249 dm_unregister_target(&flakey_target
);
253 module_init(dm_flakey_init
);
254 module_exit(dm_flakey_exit
);
256 MODULE_DESCRIPTION(DM_NAME
" flakey target");
257 MODULE_AUTHOR("Joe Thornber <dm-devel@redhat.com>");
258 MODULE_LICENSE("GPL");