1// SPDX-License-Identifier: GPL-2.0-or-later
2/*
3 * Copyright (c) 2020 FUJITSU LIMITED. All rights reserved.
4 * Copyright (c) Linux Test Project, 2022
5 * Author: Yang Xu <xuyang2018.jy@cn.jujitsu.com>
6 */
7
8/*\
9 * [Description]
10 *
11 * Tests ioctl() on loopdevice with LOOP_SET_CAPACITY flag.
12 *
13 * Tests whether LOOP_SET_CAPACITY can update a live
14 * loop device size after change the size of the underlying
15 * backing file. Also checks sysfs value.
16 */
17
18#include <stdio.h>
19#include <unistd.h>
20#include <string.h>
21#include <stdlib.h>
22#include "lapi/loop.h"
23#include "tst_test.h"
24
25#define OLD_SIZE 10240
26#define NEW_SIZE 5120
27
28static char dev_path[1024], sys_loop_sizepath[1024];
29static char *wrbuf;
30static int dev_num, dev_fd, file_fd, attach_flag;
31
32static void verify_ioctl_loop(void)
33{
34	struct loop_info loopinfoget;
35
36	memset(&loopinfoget, 0, sizeof(loopinfoget));
37	tst_fill_file("test.img", 0, 1024, OLD_SIZE/1024);
38	tst_attach_device(dev_path, "test.img");
39	attach_flag = 1;
40
41	TST_ASSERT_INT(sys_loop_sizepath, OLD_SIZE/512);
42	file_fd = SAFE_OPEN("test.img", O_RDWR);
43	SAFE_IOCTL(dev_fd, LOOP_GET_STATUS, &loopinfoget);
44
45	if (loopinfoget.lo_flags & LO_FLAGS_READ_ONLY)
46		tst_brk(TCONF, "Current environment has unexpected LO_FLAGS_READ_ONLY flag");
47
48	SAFE_TRUNCATE("test.img", NEW_SIZE);
49	SAFE_IOCTL(dev_fd, LOOP_SET_CAPACITY);
50
51	SAFE_LSEEK(dev_fd, 0, SEEK_SET);
52
53	/*check that we can't write data beyond 5K into loop device*/
54	TEST(write(dev_fd, wrbuf, OLD_SIZE));
55	if (TST_RET == NEW_SIZE) {
56		tst_res(TPASS, "LOOP_SET_CAPACITY set loop size to %d", NEW_SIZE);
57	} else {
58		tst_res(TFAIL, "LOOP_SET_CAPACITY didn't set loop size to %d, its size is %ld",
59				NEW_SIZE, TST_RET);
60	}
61
62	TST_ASSERT_INT(sys_loop_sizepath, NEW_SIZE/512);
63
64	SAFE_CLOSE(file_fd);
65	tst_detach_device_by_fd(dev_path, dev_fd);
66	unlink("test.img");
67	attach_flag = 0;
68}
69
70static void setup(void)
71{
72	dev_num = tst_find_free_loopdev(dev_path, sizeof(dev_path));
73	if (dev_num < 0)
74		tst_brk(TBROK, "Failed to find free loop device");
75
76	wrbuf = SAFE_MALLOC(OLD_SIZE);
77	memset(wrbuf, 'x', OLD_SIZE);
78	sprintf(sys_loop_sizepath, "/sys/block/loop%d/size", dev_num);
79	dev_fd = SAFE_OPEN(dev_path, O_RDWR);
80}
81
82static void cleanup(void)
83{
84	if (dev_fd > 0)
85		SAFE_CLOSE(dev_fd);
86	if (file_fd > 0)
87		SAFE_CLOSE(file_fd);
88	if (wrbuf)
89		free(wrbuf);
90	if (attach_flag)
91		tst_detach_device(dev_path);
92}
93
94static struct tst_test test = {
95	.setup = setup,
96	.cleanup = cleanup,
97	.test_all = verify_ioctl_loop,
98	.needs_root = 1,
99	.needs_tmpdir = 1,
100	.needs_drivers = (const char *const []) {
101		"loop",
102		NULL
103	}
104};
105