1#!/bin/bash
  2
  3efivarfs_mount=/sys/firmware/efi/efivars
  4test_guid=210be57c-9849-4fc7-a635-e6382d1aec27
  5
  6check_prereqs()
  7{
  8	local msg="skip all tests:"
  9
 10	if [ $UID != 0 ]; then
 11		echo $msg must be run as root >&2
 12		exit 0
 13	fi
 14
 15	if ! grep -q "^\S\+ $efivarfs_mount efivarfs" /proc/mounts; then
 16		echo $msg efivarfs is not mounted on $efivarfs_mount >&2
 17		exit 0
 18	fi
 19}
 20
 21run_test()
 22{
 23	local test="$1"
 24
 25	echo "--------------------"
 26	echo "running $test"
 27	echo "--------------------"
 28
 29	if [ "$(type -t $test)" = 'function' ]; then
 30		( $test )
 31	else
 32		( ./$test )
 33	fi
 34
 35	if [ $? -ne 0 ]; then
 36		echo "  [FAIL]"
 37		rc=1
 38	else
 39		echo "  [PASS]"
 40	fi
 41}
 42
 43test_create()
 44{
 45	local attrs='\x07\x00\x00\x00'
 46	local file=$efivarfs_mount/$FUNCNAME-$test_guid
 47
 48	printf "$attrs\x00" > $file
 49
 50	if [ ! -e $file ]; then
 51		echo "$file couldn't be created" >&2
 52		exit 1
 53	fi
 54
 55	if [ $(stat -c %s $file) -ne 5 ]; then
 56		echo "$file has invalid size" >&2
 57		exit 1
 58	fi
 59}
 60
 61test_create_empty()
 62{
 63	local file=$efivarfs_mount/$FUNCNAME-$test_guid
 64
 65	: > $file
 66
 67	if [ ! -e $file ]; then
 68		echo "$file can not be created without writing" >&2
 69		exit 1
 70	fi
 71}
 72
 73test_create_read()
 74{
 75	local file=$efivarfs_mount/$FUNCNAME-$test_guid
 76	./create-read $file
 77}
 78
 79test_delete()
 80{
 81	local attrs='\x07\x00\x00\x00'
 82	local file=$efivarfs_mount/$FUNCNAME-$test_guid
 83
 84	printf "$attrs\x00" > $file
 85
 86	if [ ! -e $file ]; then
 87		echo "$file couldn't be created" >&2
 88		exit 1
 89	fi
 90
 91	rm $file 2>/dev/null
 92	if [ $? -ne 0 ]; then
 93		chattr -i $file
 94		rm $file
 95	fi
 96
 97	if [ -e $file ]; then
 98		echo "$file couldn't be deleted" >&2
 99		exit 1
100	fi
101
102}
103
104# test that we can remove a variable by issuing a write with only
105# attributes specified
106test_zero_size_delete()
107{
108	local attrs='\x07\x00\x00\x00'
109	local file=$efivarfs_mount/$FUNCNAME-$test_guid
110
111	printf "$attrs\x00" > $file
112
113	if [ ! -e $file ]; then
114		echo "$file does not exist" >&2
115		exit 1
116	fi
117
118	chattr -i $file
119	printf "$attrs" > $file
120
121	if [ -e $file ]; then
122		echo "$file should have been deleted" >&2
123		exit 1
124	fi
125}
126
127test_open_unlink()
128{
129	local file=$efivarfs_mount/$FUNCNAME-$test_guid
130	./open-unlink $file
131}
132
133# test that we can create a range of filenames
134test_valid_filenames()
135{
136	local attrs='\x07\x00\x00\x00'
137	local ret=0
138
139	local file_list="abc dump-type0-11-1-1362436005 1234 -"
140	for f in $file_list; do
141		local file=$efivarfs_mount/$f-$test_guid
142
143		printf "$attrs\x00" > $file
144
145		if [ ! -e $file ]; then
146			echo "$file could not be created" >&2
147			ret=1
148		else
149			rm $file 2>/dev/null
150			if [ $? -ne 0 ]; then
151				chattr -i $file
152				rm $file
153			fi
154		fi
155	done
156
157	exit $ret
158}
159
160test_invalid_filenames()
161{
162	local attrs='\x07\x00\x00\x00'
163	local ret=0
164
165	local file_list="
166		-1234-1234-1234-123456789abc
167		foo
168		foo-bar
169		-foo-
170		foo-barbazba-foob-foob-foob-foobarbazfoo
171		foo-------------------------------------
172		-12345678-1234-1234-1234-123456789abc
173		a-12345678=1234-1234-1234-123456789abc
174		a-12345678-1234=1234-1234-123456789abc
175		a-12345678-1234-1234=1234-123456789abc
176		a-12345678-1234-1234-1234=123456789abc
177		1112345678-1234-1234-1234-123456789abc"
178
179	for f in $file_list; do
180		local file=$efivarfs_mount/$f
181
182		printf "$attrs\x00" 2>/dev/null > $file
183
184		if [ -e $file ]; then
185			echo "Creating $file should have failed" >&2
186			rm $file 2>/dev/null
187			if [ $? -ne 0 ]; then
188				chattr -i $file
189				rm $file
190			fi
191			ret=1
192		fi
193	done
194
195	exit $ret
196}
197
198check_prereqs
199
200rc=0
201
202run_test test_create
203run_test test_create_empty
204run_test test_create_read
205run_test test_delete
206run_test test_zero_size_delete
207run_test test_open_unlink
208run_test test_valid_filenames
209run_test test_invalid_filenames
210
211exit $rc