xref: /openbmc/u-boot/test/py/tests/test_fpga.py (revision 77c07e7e)
1# SPDX-License-Identifier: GPL-2.0
2#
3# Copyright (c) 2018, Xilinx Inc.
4#
5# Michal Simek
6# Siva Durga Prasad Paladugu
7
8import pytest
9import re
10import random
11import u_boot_utils
12
13"""
14Note: This test relies on boardenv_* containing configuration values to define
15the network available and files to be used for testing. Without this, this test
16will be automatically skipped.
17
18For example:
19
20# True if a DHCP server is attached to the network, and should be tested.
21env__net_dhcp_server = True
22
23# A list of environment variables that should be set in order to configure a
24# static IP. In this test case we atleast need serverip for performing tftpb
25# to get required files.
26env__net_static_env_vars = [
27    ('ipaddr', '10.0.0.100'),
28    ('netmask', '255.255.255.0'),
29    ('serverip', '10.0.0.1'),
30]
31
32# Details regarding the files that may be read from a TFTP server. .
33env__fpga_secure_readable_file = {
34    'fn': 'auth_bhdr_ppk1_bit.bin',
35    'enckupfn': 'auth_bhdr_enc_kup_load_bit.bin',
36    'addr': 0x1000000,
37    'keyaddr': 0x100000,
38    'keyfn': 'key.txt',
39}
40
41env__fpga_under_test = {
42    'dev': 0,
43    'addr' : 0x1000000,
44    'bitstream_load': 'compress.bin',
45    'bitstream_load_size': 1831960,
46    'bitstream_loadp': 'compress_pr.bin',
47    'bitstream_loadp_size': 423352,
48    'bitstream_loadb': 'compress.bit',
49    'bitstream_loadb_size': 1832086,
50    'bitstream_loadbp': 'compress_pr.bit',
51    'bitstream_loadbp_size': 423491,
52    'mkimage_legacy': 'download.ub',
53    'mkimage_legacy_size': 13321468,
54    'mkimage_legacy_gz': 'download.gz.ub',
55    'mkimage_legacy_gz_size': 53632,
56    'mkimage_fit': 'download-fit.ub',
57    'mkimage_fit_size': 13322784,
58    'loadfs': 'mmc 0 compress.bin',
59    'loadfs_size': 1831960,
60    'loadfs_block_size': 0x10000,
61}
62"""
63
64import test_net
65
66def check_dev(u_boot_console):
67    f = u_boot_console.config.env.get('env__fpga_under_test', None)
68    if not f:
69        pytest.skip('No FPGA to test')
70
71    dev = f.get('dev', -1)
72    if dev < 0:
73        pytest.fail('No dev specified via env__fpga_under_test')
74
75    return dev, f
76
77def load_file_from_var(u_boot_console, name):
78    dev, f = check_dev(u_boot_console)
79
80    addr = f.get('addr', -1)
81    if addr < 0:
82        pytest.fail('No address specified via env__fpga_under_test')
83
84    test_net.test_net_dhcp(u_boot_console)
85    test_net.test_net_setup_static(u_boot_console)
86    bit = f['%s' % (name)]
87    bit_size = f['%s_size' % (name)]
88
89    expected_tftp = 'Bytes transferred = %d' % bit_size
90    output = u_boot_console.run_command('tftpboot %x %s' % (addr, bit))
91    assert expected_tftp in output
92
93    return f, dev, addr, bit, bit_size
94
95###### FPGA FAIL test ######
96expected_usage = 'fpga - loadable FPGA image support'
97
98@pytest.mark.xfail
99@pytest.mark.buildconfigspec('cmd_fpga')
100def test_fpga_fail(u_boot_console):
101    # Test non valid fpga subcommand
102    expected = 'fpga: non existing command'
103    output = u_boot_console.run_command('fpga broken 0')
104    #assert expected in output
105    assert expected_usage in output
106
107@pytest.mark.buildconfigspec('cmd_fpga')
108def test_fpga_help(u_boot_console):
109    # Just show help
110    output = u_boot_console.run_command('fpga')
111    assert expected_usage in output
112
113
114###### FPGA DUMP tests ######
115
116@pytest.mark.buildconfigspec('cmd_fpga')
117def test_fpga_dump(u_boot_console):
118    pytest.skip('Not implemented now')
119
120@pytest.mark.buildconfigspec('cmd_fpga')
121def test_fpga_dump_variable(u_boot_console):
122    # Same as above but via "fpga" variable
123    pytest.skip('Not implemented now')
124
125###### FPGA INFO tests ######
126
127@pytest.mark.buildconfigspec('cmd_fpga')
128def test_fpga_info_fail(u_boot_console):
129    # Maybe this can be skipped completely
130    dev, f = check_dev(u_boot_console)
131
132    # Multiple parameters to fpga info should fail
133    expected = 'fpga: more parameters passed'
134    output = u_boot_console.run_command('fpga info 0 0')
135    #assert expected in output
136    assert expected_usage in output
137
138@pytest.mark.buildconfigspec('cmd_fpga')
139def test_fpga_info_list(u_boot_console):
140    # Maybe this can be skipped completely
141    dev, f = check_dev(u_boot_console)
142
143    # Code is design in a way that if fpga dev is not passed it should
144    # return list of all fpga devices in the system
145    u_boot_console.run_command('setenv fpga')
146    output = u_boot_console.run_command('fpga info')
147    assert expected_usage not in output
148
149@pytest.mark.buildconfigspec('cmd_fpga')
150def test_fpga_info(u_boot_console):
151    dev, f = check_dev(u_boot_console)
152
153    output = u_boot_console.run_command('fpga info %x' % (dev))
154    assert expected_usage not in output
155
156@pytest.mark.buildconfigspec('cmd_fpga')
157def test_fpga_info_variable(u_boot_console):
158    dev, f = check_dev(u_boot_console)
159
160    #
161    # fpga variable is storing device number which doesn't need to be passed
162    #
163    u_boot_console.run_command('setenv fpga %x' % (dev))
164
165    output = u_boot_console.run_command('fpga info')
166    # Variable cleanup
167    u_boot_console.run_command('setenv fpga')
168    assert expected_usage not in output
169
170###### FPGA LOAD tests ######
171
172@pytest.mark.buildconfigspec('cmd_fpga')
173@pytest.mark.buildconfigspec('cmd_echo')
174def test_fpga_load_fail(u_boot_console):
175    f, dev, addr, bit, bit_size = load_file_from_var(u_boot_console, 'bitstream_load')
176
177    for cmd in ['dump', 'load', 'loadb']:
178	# missing dev parameter
179	expected = 'fpga: incorrect parameters passed'
180	output = u_boot_console.run_command('fpga %s %x $filesize' % (cmd, addr))
181	#assert expected in output
182	assert expected_usage in output
183
184	# more parameters - 0 at the end
185	expected = 'fpga: more parameters passed'
186	output = u_boot_console.run_command('fpga %s %x %x $filesize 0' % (cmd, dev, addr))
187	#assert expected in output
188	assert expected_usage in output
189
190	# 0 address
191	expected = 'fpga: zero fpga_data address'
192	output = u_boot_console.run_command('fpga %s %x 0 $filesize' % (cmd, dev))
193	#assert expected in output
194	assert expected_usage in output
195
196	# 0 filesize
197	expected = 'fpga: zero size'
198	output = u_boot_console.run_command('fpga %s %x %x 0' % (cmd, dev, addr))
199	#assert expected in output
200	assert expected_usage in output
201
202@pytest.mark.buildconfigspec('cmd_fpga')
203@pytest.mark.buildconfigspec('cmd_echo')
204def test_fpga_load(u_boot_console):
205    f, dev, addr, bit, bit_size = load_file_from_var(u_boot_console, 'bitstream_load')
206
207    expected_text = 'FPGA loaded successfully'
208    output = u_boot_console.run_command('fpga load %x %x $filesize && echo %s' % (dev, addr, expected_text))
209    assert expected_text in output
210
211@pytest.mark.buildconfigspec('cmd_fpga')
212@pytest.mark.buildconfigspec('cmd_fpga_loadp')
213@pytest.mark.buildconfigspec('cmd_echo')
214def test_fpga_loadp(u_boot_console):
215    f, dev, addr, bit, bit_size = load_file_from_var(u_boot_console, 'bitstream_load')
216
217    expected_text = 'FPGA loaded successfully'
218    output = u_boot_console.run_command('fpga load %x %x $filesize && echo %s' % (dev, addr, expected_text))
219    assert expected_text in output
220
221    # And load also partial bistream
222    f, dev, addr, bit, bit_size = load_file_from_var(u_boot_console, 'bitstream_loadp')
223
224    expected_text = 'FPGA loaded successfully'
225    output = u_boot_console.run_command('fpga loadp %x %x $filesize && echo %s' % (dev, addr, expected_text))
226    assert expected_text in output
227
228@pytest.mark.buildconfigspec('cmd_fpga')
229@pytest.mark.buildconfigspec('cmd_echo')
230def test_fpga_loadb(u_boot_console):
231    f, dev, addr, bit, bit_size = load_file_from_var(u_boot_console, 'bitstream_loadb')
232
233    expected_text = 'FPGA loaded successfully'
234    output = u_boot_console.run_command('fpga loadb %x %x $filesize && echo %s' % (dev, addr, expected_text))
235    assert expected_text in output
236
237@pytest.mark.buildconfigspec('cmd_fpga')
238@pytest.mark.buildconfigspec('cmd_fpga_loadbp')
239@pytest.mark.buildconfigspec('cmd_echo')
240def test_fpga_loadbp(u_boot_console):
241    f, dev, addr, bit, bit_size = load_file_from_var(u_boot_console, 'bitstream_loadb')
242
243    expected_text = 'FPGA loaded successfully'
244    output = u_boot_console.run_command('fpga loadb %x %x $filesize && echo %s' % (dev, addr, expected_text))
245    assert expected_text in output
246
247    # And load also partial bistream in bit format
248    f, dev, addr, bit, bit_size = load_file_from_var(u_boot_console, 'bitstream_loadbp')
249
250    expected_text = 'FPGA loaded successfully'
251    output = u_boot_console.run_command('fpga loadbp %x %x $filesize && echo %s' % (dev, addr, expected_text))
252    assert expected_text in output
253
254###### FPGA LOADMK tests ######
255
256@pytest.mark.buildconfigspec('cmd_fpga')
257@pytest.mark.buildconfigspec('cmd_fpga_loadmk')
258@pytest.mark.buildconfigspec('cmd_echo')
259@pytest.mark.buildconfigspec('image_format_legacy')
260def test_fpga_loadmk_fail(u_boot_console):
261    f, dev, addr, bit, bit_size = load_file_from_var(u_boot_console, 'mkimage_legacy')
262
263    u_boot_console.run_command('imi %x' % (addr))
264
265    # load image but pass incorrect address to show error message
266    expected = 'Unknown image type'
267    output = u_boot_console.run_command('fpga loadmk %x %x' % (dev, addr + 0x10))
268    assert expected in output
269
270    # Pass more parameters then command expects - 0 at the end
271    output = u_boot_console.run_command('fpga loadmk %x %x 0' % (dev, addr))
272    #assert expected in output
273    assert expected_usage in output
274
275@pytest.mark.buildconfigspec('cmd_fpga')
276@pytest.mark.buildconfigspec('cmd_fpga_loadmk')
277@pytest.mark.buildconfigspec('cmd_echo')
278@pytest.mark.buildconfigspec('image_format_legacy')
279def test_fpga_loadmk_legacy(u_boot_console):
280    f, dev, addr, bit, bit_size = load_file_from_var(u_boot_console, 'mkimage_legacy')
281
282    u_boot_console.run_command('imi %x' % (addr))
283
284    expected_text = 'FPGA loaded successfully'
285    output = u_boot_console.run_command('fpga loadmk %x %x && echo %s' % (dev, addr, expected_text))
286    assert expected_text in output
287
288@pytest.mark.xfail
289@pytest.mark.buildconfigspec('cmd_fpga')
290@pytest.mark.buildconfigspec('cmd_fpga_loadmk')
291@pytest.mark.buildconfigspec('cmd_echo')
292@pytest.mark.buildconfigspec('image_format_legacy')
293def test_fpga_loadmk_legacy_variable_fpga(u_boot_console):
294    f, dev, addr, bit, bit_size = load_file_from_var(u_boot_console, 'mkimage_legacy')
295
296    u_boot_console.run_command('imi %x' % (addr))
297
298    u_boot_console.run_command('setenv fpga %x' % (dev))
299
300    # this testcase should cover case which looks like it is supported but dev pointer is broken by loading mkimage address
301    expected_text = 'FPGA loaded successfully'
302    output = u_boot_console.run_command('fpga loadmk %x && echo %s' % (addr, expected_text))
303    u_boot_console.run_command('setenv fpga')
304    assert expected_text in output
305
306@pytest.mark.buildconfigspec('cmd_fpga')
307@pytest.mark.buildconfigspec('cmd_fpga_loadmk')
308@pytest.mark.buildconfigspec('cmd_echo')
309@pytest.mark.buildconfigspec('image_format_legacy')
310def test_fpga_loadmk_legacy_variable_fpgadata(u_boot_console):
311    f, dev, addr, bit, bit_size = load_file_from_var(u_boot_console, 'mkimage_legacy')
312
313    u_boot_console.run_command('imi %x' % (addr))
314
315    u_boot_console.run_command('setenv fpgadata %x' % (addr))
316
317    # this testcase should cover case which looks like it is supported but dev pointer is broken by loading mkimage address
318    expected_text = 'FPGA loaded successfully'
319    output = u_boot_console.run_command('fpga loadmk %x && echo %s' % (dev, expected_text))
320    u_boot_console.run_command('setenv fpgadata')
321    assert expected_text in output
322
323@pytest.mark.buildconfigspec('cmd_fpga')
324@pytest.mark.buildconfigspec('cmd_fpga_loadmk')
325@pytest.mark.buildconfigspec('cmd_echo')
326@pytest.mark.buildconfigspec('image_format_legacy')
327def test_fpga_loadmk_legacy_variable(u_boot_console):
328    f, dev, addr, bit, bit_size = load_file_from_var(u_boot_console, 'mkimage_legacy')
329
330    u_boot_console.run_command('imi %x' % (addr))
331
332    u_boot_console.run_command('setenv fpga %x' % (dev))
333    u_boot_console.run_command('setenv fpgadata %x' % (addr))
334
335    # this testcase should cover case which looks like it is supported but dev pointer is broken by loading mkimage address
336    expected_text = 'FPGA loaded successfully'
337    output = u_boot_console.run_command('fpga loadmk && echo %s' % (expected_text))
338    u_boot_console.run_command('setenv fpga')
339    u_boot_console.run_command('setenv fpgadata')
340    assert expected_text in output
341
342@pytest.mark.buildconfigspec('cmd_fpga')
343@pytest.mark.buildconfigspec('cmd_fpga_loadmk')
344@pytest.mark.buildconfigspec('cmd_echo')
345@pytest.mark.buildconfigspec('image_format_legacy')
346@pytest.mark.buildconfigspec('gzip')
347def test_fpga_loadmk_legacy_gz(u_boot_console):
348    f, dev, addr, bit, bit_size = load_file_from_var(u_boot_console, 'mkimage_legacy_gz')
349
350    u_boot_console.run_command('imi %x' % (addr))
351
352    expected_text = 'FPGA loaded successfully'
353    output = u_boot_console.run_command('fpga loadmk %x %x && echo %s' % (dev, addr, expected_text))
354    assert expected_text in output
355
356@pytest.mark.buildconfigspec('cmd_fpga')
357@pytest.mark.buildconfigspec('cmd_fpga_loadmk')
358@pytest.mark.buildconfigspec('fit')
359@pytest.mark.buildconfigspec('cmd_echo')
360def test_fpga_loadmk_fit(u_boot_console):
361    f, dev, addr, bit, bit_size = load_file_from_var(u_boot_console, 'mkimage_fit')
362
363    u_boot_console.run_command('imi %x' % (addr))
364
365    expected_text = 'FPGA loaded successfully'
366    output = u_boot_console.run_command('fpga loadmk %x %x:fpga && echo %s' % (dev, addr, expected_text))
367    assert expected_text in output
368
369@pytest.mark.buildconfigspec('cmd_fpga')
370@pytest.mark.buildconfigspec('cmd_fpga_loadmk')
371@pytest.mark.buildconfigspec('fit')
372@pytest.mark.buildconfigspec('cmd_echo')
373def test_fpga_loadmk_fit_variable_fpga(u_boot_console):
374    f, dev, addr, bit, bit_size = load_file_from_var(u_boot_console, 'mkimage_fit')
375
376    u_boot_console.run_command('imi %x' % (addr))
377    # FIXME this should fail - broken support in past
378    u_boot_console.run_command('setenv fpga %x' % (dev))
379
380    expected_text = 'FPGA loaded successfully'
381    output = u_boot_console.run_command('fpga loadmk %x:fpga && echo %s' % (addr, expected_text))
382    u_boot_console.run_command('setenv fpga')
383    assert expected_text in output
384
385@pytest.mark.buildconfigspec('cmd_fpga')
386@pytest.mark.buildconfigspec('cmd_fpga_loadmk')
387@pytest.mark.buildconfigspec('fit')
388@pytest.mark.buildconfigspec('cmd_echo')
389def test_fpga_loadmk_fit_variable_fpgadata(u_boot_console):
390    f, dev, addr, bit, bit_size = load_file_from_var(u_boot_console, 'mkimage_fit')
391
392    u_boot_console.run_command('imi %x' % (addr))
393    # FIXME this should fail - broken support in past
394    u_boot_console.run_command('setenv fpgadata %x:fpga' % (addr))
395
396    expected_text = 'FPGA loaded successfully'
397    output = u_boot_console.run_command('fpga loadmk %x && echo %s' % (dev, expected_text))
398    u_boot_console.run_command('setenv fpgadata')
399    assert expected_text in output
400
401@pytest.mark.buildconfigspec('cmd_fpga')
402@pytest.mark.buildconfigspec('cmd_fpga_loadmk')
403@pytest.mark.buildconfigspec('fit')
404@pytest.mark.buildconfigspec('cmd_echo')
405def test_fpga_loadmk_fit_variable(u_boot_console):
406    f, dev, addr, bit, bit_size = load_file_from_var(u_boot_console, 'mkimage_fit')
407
408    u_boot_console.run_command('imi %x' % (addr))
409
410    u_boot_console.run_command('setenv fpga %x' % (dev))
411    u_boot_console.run_command('setenv fpgadata %x:fpga' % (addr))
412
413    expected_text = 'FPGA loaded successfully'
414    output = u_boot_console.run_command('fpga loadmk && echo %s' % (expected_text))
415    u_boot_console.run_command('setenv fpga')
416    u_boot_console.run_command('setenv fpgadata')
417    assert expected_text in output
418
419###### FPGA LOAD tests ######
420
421@pytest.mark.buildconfigspec('cmd_fpga')
422def test_fpga_loadfs_fail(u_boot_console):
423    dev, f = check_dev(u_boot_console)
424
425    addr = f.get('addr', -1)
426    if addr < 0:
427        pytest.fail('No address specified via env__fpga_under_test')
428
429    bit = f['loadfs']
430    bit_size = f['loadfs_size']
431    block_size = f['loadfs_block_size']
432
433    # less params - dev number removed
434    expected = 'fpga: incorrect parameters passed'
435    output = u_boot_console.run_command('fpga loadfs %x %x %x %s' % (addr, bit_size, block_size, bit))
436    #assert expected in output
437    assert expected_usage in output
438
439    # one more param - 0 at the end
440    # This is the longest command that's why there is no message from cmd/fpga.c
441    output = u_boot_console.run_command('fpga loadfs %x %x %x %x %s 0' % (dev, addr, bit_size, block_size, bit))
442    assert expected_usage in output
443
444    # zero address 0
445    expected = 'fpga: zero fpga_data address'
446    output = u_boot_console.run_command('fpga loadfs %x %x %x %x %s' % (dev, 0, bit_size, block_size, bit))
447    #assert expected in output
448    assert expected_usage in output
449
450    # bit_size 0
451    expected = 'fpga: zero size'
452    output = u_boot_console.run_command('fpga loadfs %x %x %x %x %s' % (dev, addr, 0, block_size, bit))
453    #assert expected in output
454    assert expected_usage in output
455
456    # block size 0
457    # FIXME this should pass but it failing too
458    output = u_boot_console.run_command('fpga loadfs %x %x %x %x %s' % (dev, addr, bit_size, 0, bit))
459    assert expected_usage in output
460
461    # non existing bitstream name
462    expected = 'Unable to read file noname'
463    output = u_boot_console.run_command('fpga loadfs %x %x %x %x mmc 0 noname' % (dev, addr, bit_size, block_size))
464    assert expected in output
465    assert expected_usage in output
466
467    # -1 dev number
468    expected = 'fpga_fsload: Invalid device number -1'
469    output = u_boot_console.run_command('fpga loadfs %d %x %x %x mmc 0 noname' % (-1, addr, bit_size, block_size))
470    assert expected in output
471    assert expected_usage in output
472
473
474@pytest.mark.buildconfigspec('cmd_fpga')
475@pytest.mark.buildconfigspec('cmd_echo')
476def test_fpga_loadfs(u_boot_console):
477    dev, f = check_dev(u_boot_console)
478
479    addr = f.get('addr', -1)
480    if addr < 0:
481        pytest.fail('No address specified via env__fpga_under_test')
482
483    bit = f['loadfs']
484    bit_size = f['loadfs_size']
485    block_size = f['loadfs_block_size']
486
487    # This should be done better
488    expected_text = 'FPGA loaded successfully'
489    output = u_boot_console.run_command('fpga loadfs %x %x %x %x %s && echo %s' % (dev, addr, bit_size, block_size, bit, expected_text))
490    assert expected_text in output
491
492@pytest.mark.buildconfigspec('cmd_fpga')
493@pytest.mark.buildconfigspec('cmd_fpga_load_secure')
494@pytest.mark.buildconfigspec('cmd_net')
495@pytest.mark.buildconfigspec('cmd_dhcp')
496@pytest.mark.buildconfigspec('net')
497def test_fpga_secure_bit_auth(u_boot_console):
498
499    test_net.test_net_dhcp(u_boot_console)
500    test_net.test_net_setup_static(u_boot_console)
501
502    f = u_boot_console.config.env.get('env__fpga_secure_readable_file', None)
503    if not f:
504        pytest.skip('No TFTP readable file to read')
505
506    addr = f.get('addr', None)
507    if not addr:
508      addr = u_boot_utils.find_ram_base(u_boot_console)
509
510    expected_tftp = 'Bytes transferred = '
511    fn = f['fn']
512    output = u_boot_console.run_command('tftpboot %x %s' % (addr, fn))
513    assert expected_tftp in output
514
515    expected_zynqmpsecure = 'Bitstream successfully loaded'
516    output = u_boot_console.run_command('fpga loads 0 %x $filesize 0 2' % (addr))
517    assert expected_zynqmpsecure in output
518
519
520@pytest.mark.buildconfigspec('cmd_fpga')
521@pytest.mark.buildconfigspec('cmd_fpga_load_secure')
522@pytest.mark.buildconfigspec('cmd_net')
523@pytest.mark.buildconfigspec('cmd_dhcp')
524@pytest.mark.buildconfigspec('net')
525def test_fpga_secure_bit_img_auth_kup(u_boot_console):
526
527    test_net.test_net_dhcp(u_boot_console)
528    test_net.test_net_setup_static(u_boot_console)
529
530    f = u_boot_console.config.env.get('env__fpga_secure_readable_file', None)
531    if not f:
532        pytest.skip('No TFTP readable file to read')
533
534    keyaddr = f.get('keyaddr', None)
535    if not keyaddr:
536      addr = u_boot_utils.find_ram_base(u_boot_console)
537    expected_tftp = 'Bytes transferred = '
538    keyfn = f['keyfn']
539    output = u_boot_console.run_command('tftpboot %x %s' % (keyaddr, keyfn))
540    assert expected_tftp in output
541
542    addr = f.get('addr', None)
543    if not addr:
544      addr = u_boot_utils.find_ram_base(u_boot_console)
545    expected_tftp = 'Bytes transferred = '
546    fn = f['enckupfn']
547    output = u_boot_console.run_command('tftpboot %x %s' % (addr, fn))
548    assert expected_tftp in output
549
550    expected_zynqmpsecure = 'Bitstream successfully loaded'
551    output = u_boot_console.run_command('fpga loads 0 %x $filesize 0 1 %x' % (addr, keyaddr))
552    assert expected_zynqmpsecure in output
553