File: nullb

package info (click to toggle)
btrfs-progs 6.2-1%2Bdeb12u1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 17,244 kB
  • sloc: ansic: 114,376; sh: 9,576; python: 1,242; makefile: 820
file content (212 lines) | stat: -rwxr-xr-x 3,910 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
#!/bin/bash
# Simple management of null_blk devices, useful for zoned device testing
# Version: 0.1
#
# Commands:
#
# nullb setup
# - load required modules and mount configfs
#
# nullb create [-s size] [-z zonesize]
# - create a new device with given sizes, allocating the first free indes, #
#   device is /dev/nullb$index
#
# nullb ls
# nullb list
# - show table of created null_blk devices, size and zone sizes
#
# nullb rm NAME
# nullb delete NAME
# - delete existing null_blk device by name, must match the device node name
#   like 'nullb0'

CMD="$1"
DEBUG=false

# Defaults
SIZE='2048'
ZONESIZE='256'
SYSFS='/sys/kernel/config/nullb'

# create
# list
# delete
# setup

function _error() {
	echo "ERROR: $@"
	exit 1
}

function _warn() {
	echo "WARNING: $@"
}

function _msg() {
	echo "INFO: $@"
}

function _dbg() {
	$DEBUG && echo "DEBUG: $@" > /dev/tty
}

function _check_setup() {
	if ! grep -q null_blk /proc/modules; then
		_error "module not loaded"
	fi
	if ! grep -q configfs /proc/filesystems; then
		_error "configfs not mounted"
	fi
	if ! grep -q zoned "$SYSFS/features"; then
		_warn "null_blk module does not support zoned devices"
	fi
}

function _check_cd() {
	if ! [ -d "$1" ]; then
		_error "$1 not accessible"
	fi
	cd "$1"
}

function _find_free_index() {
	_check_cd "$SYSFS"
	found=-1
	for index in `seq 0 1 10`; do
		_dbg "index $index"
		ok=true
		for dir in $(ls -df1 * 2>/dev/null); do
			if ! [ -d "$dir" ]; then
				continue
			fi
			_dbg "found $dir"
			idx=$(cat "$dir/index")
			if [ "$idx" = "$index" ]; then
				ok=false
				break
			fi
		done
		if $ok; then
			found=$index
			break
		fi
	done
	if [ "$found" = "-1" ]; then
		_error "no free index found"
	fi
	_dbg "first free index: $found"
	echo -n "$found"
}

function _parse_device_size() {
	local size="$SIZE"
	_dbg "parse size $@"
	while [ $# -gt 0 ]; do
		_dbg "ARG: $1"
		if [ "$1" = '-s' ]; then
			size="$2"
			if [ -z "$size" ]; then
				_error "-s requires size"
			fi
			shift
		fi
		shift
	done

	echo -en "$size"
}

function _parse_zone_size() {
	local zonesize="$ZONESIZE"
	_dbg "parse zone size $@"
	while [ $# -gt 0 ]; do
		_dbg "ARG: $1"
		if [ "$1" = '-z' ]; then
			zonesize="$2"
			if [ -z "$size" ]; then
				_error "-z requires size"
			fi
			shift
		fi
		shift
	done

	echo -en "$zonesize"
}

# main()

if [ "$CMD" = 'setup' ]; then
	_msg "setup module and mounts"
	modprobe configfs
	modprobe null_blk nr_devices=0
	_check_setup
fi

if [ "$CMD" = 'create' ]; then
	_check_setup
	index=$(_find_free_index)
	name="nullb$index"
	# size in MB
	size=$(_parse_device_size "$@")
	# size in MB
	zonesize=$(_parse_zone_size "$@")
	_msg "Create nullb$index"
	_check_cd "$SYSFS"
	if mkdir "$name"; then
		_check_cd "$name"
		echo "$size" > size
		echo 1 > zoned
		echo 1 > memory_backed
		echo "$zonesize" > zone_size
		echo 1 > power
		node=$(cat "$SYSFS/$name/index")
		node="nullb${node}"
		_msg "name=$node"
		_msg "size=${size}M zone_size=${zonesize}M"
		# last printed line is the exact name for later use
		echo "/dev/$node"
	else
		_error "already exists"
	fi
fi

if [ "$CMD" = 'ls' -o "$CMD" = 'list' ]; then
	_msg "device nodes:"
	ls /dev/nullb* 2>/dev/null
	_msg "created devices:"
	_check_cd "$SYSFS"
	printf '%-2s  %-8s  %-16s  %11s  %11s  %-2s
'	\
		"No"				\
		"Name"				\
		"Device"			\
		"Size"				\
		"Zone size"			\
		"On"
	for dir in $(ls -df1 * 2>/dev/null); do
		[ -d "$dir" ] || continue
		printf '%2d  %-8s  %-16s  %10dM  %10dM  %2d
'	\
			$(cat "$dir/index")			\
			"$dir"					\
			"/dev/nullb"$(cat "$dir/index")		\
			$(cat "$dir/size")			\
			$(cat "$dir/zone_size")			\
			$(cat "$dir/power")
	done
fi

if [ "$CMD" = 'rm' -o "$CMD" = 'delete' ]; then
	_check_cd "$SYSFS"
	name="$2"
	_dbg "deleting $name"
	if [ -d "$name" ]; then
		_msg "check mounts"
		mount | grep -- "$name"
		_msg "removing $name"
		rmdir -- "$SYSFS/$name"
	else
		_error "no such device name: $name"
	fi
fi