546 	struct dm_cache_migration *mg;  in alloc_migration()  local
548 	mg = mempool_alloc(&cache->migration_pool, GFP_NOIO);  in alloc_migration()
550 	memset(mg, 0, sizeof(*mg));  in alloc_migration()
552 	mg->cache = cache;  in alloc_migration()
555 	return mg;  in alloc_migration()
558 static void free_migration(struct dm_cache_migration *mg)  in free_migration()  argument
560 	struct cache *cache = mg->cache;  in free_migration()
565 	mempool_free(mg, &cache->migration_pool);  in free_migration()
1137 static void quiesce(struct dm_cache_migration *mg,  in quiesce()  argument
1140 	init_continuation(&mg->k, continuation);  in quiesce()
1141 	dm_cell_quiesce_v2(mg->cache->prison, mg->cell, &mg->k.ws);  in quiesce()
1152 	struct dm_cache_migration *mg = container_of(context, struct dm_cache_migration, k);  in copy_complete()  local
1155 		mg->k.input = BLK_STS_IOERR;  in copy_complete()
1157 	queue_continuation(mg->cache->wq, &mg->k);  in copy_complete()
1160 static void copy(struct dm_cache_migration *mg, bool promote)  in copy()  argument
1163 	struct cache *cache = mg->cache;  in copy()
1166 	o_region.sector = from_oblock(mg->op->oblock) * cache->sectors_per_block;  in copy()
1170 	c_region.sector = from_cblock(mg->op->cblock) * cache->sectors_per_block;  in copy()
1174 		dm_kcopyd_copy(cache->copier, &o_region, 1, &c_region, 0, copy_complete, &mg->k);  in copy()
1176 		dm_kcopyd_copy(cache->copier, &c_region, 1, &o_region, 0, copy_complete, &mg->k);  in copy()
1190 	struct dm_cache_migration *mg = bio->bi_private;  in overwrite_endio()  local
1191 	struct cache *cache = mg->cache;  in overwrite_endio()
1197 		mg->k.input = bio->bi_status;  in overwrite_endio()
1199 	queue_continuation(cache->wq, &mg->k);  in overwrite_endio()
1202 static void overwrite(struct dm_cache_migration *mg,  in overwrite()  argument
1205 	struct bio *bio = mg->overwrite_bio;  in overwrite()
1208 	dm_hook_bio(&pb->hook_info, bio, overwrite_endio, mg);  in overwrite()
1214 	if (mg->op->op == POLICY_PROMOTE)  in overwrite()
1215 		remap_to_cache(mg->cache, bio, mg->op->cblock);  in overwrite()
1217 		remap_to_origin(mg->cache, bio);  in overwrite()
1219 	init_continuation(&mg->k, continuation);  in overwrite()
1220 	accounted_request(mg->cache, bio);  in overwrite()
1234 static void mg_complete(struct dm_cache_migration *mg, bool success)  in mg_complete()  argument
1237 	struct cache *cache = mg->cache;  in mg_complete()
1238 	struct policy_work *op = mg->op;  in mg_complete()
1249 		if (mg->overwrite_bio) {  in mg_complete()
1252 			else if (mg->k.input)  in mg_complete()
1253 				mg->overwrite_bio->bi_status = mg->k.input;  in mg_complete()
1255 				mg->overwrite_bio->bi_status = BLK_STS_IOERR;  in mg_complete()
1256 			bio_endio(mg->overwrite_bio);  in mg_complete()
1283 	if (mg->cell) {  in mg_complete()
1284 		if (dm_cell_unlock_v2(cache->prison, mg->cell, &bios))  in mg_complete()
1285 			free_prison_cell(cache, mg->cell);  in mg_complete()
1288 	free_migration(mg);  in mg_complete()
1297 	struct dm_cache_migration *mg = ws_to_mg(ws);  in mg_success()  local
1298 	mg_complete(mg, mg->k.input == 0);  in mg_success()
1304 	struct dm_cache_migration *mg = ws_to_mg(ws);  in mg_update_metadata()  local
1305 	struct cache *cache = mg->cache;  in mg_update_metadata()
1306 	struct policy_work *op = mg->op;  in mg_update_metadata()
1316 			mg_complete(mg, false);  in mg_update_metadata()
1319 		mg_complete(mg, true);  in mg_update_metadata()
1329 			mg_complete(mg, false);  in mg_update_metadata()
1352 		init_continuation(&mg->k, mg_success);  in mg_update_metadata()
1353 		continue_after_commit(&cache->committer, &mg->k);  in mg_update_metadata()
1358 		mg_complete(mg, true);  in mg_update_metadata()
1365 	struct dm_cache_migration *mg = ws_to_mg(ws);  in mg_update_metadata_after_copy()  local
1370 	if (mg->k.input)  in mg_update_metadata_after_copy()
1371 		mg_complete(mg, false);  in mg_update_metadata_after_copy()
1379 	struct dm_cache_migration *mg = ws_to_mg(ws);  in mg_upgrade_lock()  local
1384 	if (mg->k.input)  in mg_upgrade_lock()
1385 		mg_complete(mg, false);  in mg_upgrade_lock()
1391 		r = dm_cell_lock_promote_v2(mg->cache->prison, mg->cell,  in mg_upgrade_lock()
1394 			mg_complete(mg, false);  in mg_upgrade_lock()
1397 			quiesce(mg, mg_update_metadata);  in mg_upgrade_lock()
1406 	struct dm_cache_migration *mg = ws_to_mg(ws);  in mg_full_copy()  local
1407 	struct cache *cache = mg->cache;  in mg_full_copy()
1408 	struct policy_work *op = mg->op;  in mg_full_copy()
1417 	init_continuation(&mg->k, mg_upgrade_lock);  in mg_full_copy()
1418 	copy(mg, is_policy_promote);  in mg_full_copy()
1423 	struct dm_cache_migration *mg = ws_to_mg(ws);  in mg_copy()  local
1425 	if (mg->overwrite_bio) {  in mg_copy()
1431 		if (!optimisable_bio(mg->cache, mg->overwrite_bio, mg->op->oblock)) {  in mg_copy()
1435 			bool rb = bio_detain_shared(mg->cache, mg->op->oblock, mg->overwrite_bio);  in mg_copy()
1437 			mg->overwrite_bio = NULL;  in mg_copy()
1438 			inc_io_migrations(mg->cache);  in mg_copy()
1450 		overwrite(mg, mg_update_metadata_after_copy);  in mg_copy()
1456 static int mg_lock_writes(struct dm_cache_migration *mg)  in mg_lock_writes()  argument
1460 	struct cache *cache = mg->cache;  in mg_lock_writes()
1470 	build_key(mg->op->oblock, oblock_succ(mg->op->oblock), &key);  in mg_lock_writes()
1472 			    mg->overwrite_bio ?  READ_WRITE_LOCK_LEVEL : WRITE_LOCK_LEVEL,  in mg_lock_writes()
1473 			    prealloc, &mg->cell);  in mg_lock_writes()
1476 		mg_complete(mg, false);  in mg_lock_writes()
1480 	if (mg->cell != prealloc)  in mg_lock_writes()
1484 		mg_copy(&mg->k.ws);  in mg_lock_writes()
1486 		quiesce(mg, mg_copy);  in mg_lock_writes()
1493 	struct dm_cache_migration *mg;  in mg_start()  local
1500 	mg = alloc_migration(cache);  in mg_start()
1502 	mg->op = op;  in mg_start()
1503 	mg->overwrite_bio = bio;  in mg_start()
1508 	return mg_lock_writes(mg);  in mg_start()
1515 static void invalidate_complete(struct dm_cache_migration *mg, bool success)  in invalidate_complete()  argument
1518 	struct cache *cache = mg->cache;  in invalidate_complete()
1521 	if (dm_cell_unlock_v2(cache->prison, mg->cell, &bios))  in invalidate_complete()
1522 		free_prison_cell(cache, mg->cell);  in invalidate_complete()
1524 	if (!success && mg->overwrite_bio)  in invalidate_complete()
1525 		bio_io_error(mg->overwrite_bio);  in invalidate_complete()
1527 	free_migration(mg);  in invalidate_complete()
1535 	struct dm_cache_migration *mg = ws_to_mg(ws);  in invalidate_completed()  local
1536 	invalidate_complete(mg, !mg->k.input);  in invalidate_completed()
1565 	struct dm_cache_migration *mg = ws_to_mg(ws);  in invalidate_remove()  local
1566 	struct cache *cache = mg->cache;  in invalidate_remove()
1568 	r = invalidate_cblock(cache, mg->invalidate_cblock);  in invalidate_remove()
1570 		invalidate_complete(mg, false);  in invalidate_remove()
1574 	init_continuation(&mg->k, invalidate_completed);  in invalidate_remove()
1575 	continue_after_commit(&cache->committer, &mg->k);  in invalidate_remove()
1576 	remap_to_origin_clear_discard(cache, mg->overwrite_bio, mg->invalidate_oblock);  in invalidate_remove()
1577 	mg->overwrite_bio = NULL;  in invalidate_remove()
1581 static int invalidate_lock(struct dm_cache_migration *mg)  in invalidate_lock()  argument
1585 	struct cache *cache = mg->cache;  in invalidate_lock()
1590 	build_key(mg->invalidate_oblock, oblock_succ(mg->invalidate_oblock), &key);  in invalidate_lock()
1592 			    READ_WRITE_LOCK_LEVEL, prealloc, &mg->cell);  in invalidate_lock()
1595 		invalidate_complete(mg, false);  in invalidate_lock()
1599 	if (mg->cell != prealloc)  in invalidate_lock()
1603 		quiesce(mg, invalidate_remove);  in invalidate_lock()
1610 		init_continuation(&mg->k, invalidate_remove);  in invalidate_lock()
1611 		queue_work(cache->wq, &mg->k.ws);  in invalidate_lock()
1620 	struct dm_cache_migration *mg;  in invalidate_start()  local
1625 	mg = alloc_migration(cache);  in invalidate_start()
1627 	mg->overwrite_bio = bio;  in invalidate_start()
1628 	mg->invalidate_cblock = cblock;  in invalidate_start()
1629 	mg->invalidate_oblock = oblock;  in invalidate_start()
1631 	return invalidate_lock(mg);  in invalidate_start()