xref: /openbmc/qemu/tests/avocado/vnc.py (revision 54fde4ff) 
1# Simple functional tests for VNC functionality
2#
3# Copyright (c) 2018 Red Hat, Inc.
4#
5# Author:
6#  Cleber Rosa <crosa@redhat.com>
7#
8# This work is licensed under the terms of the GNU GPL, version 2 or
9# later.  See the COPYING file in the top-level directory.
10
11import socket
12from typing import List
13
14from avocado_qemu import QemuSystemTest
15
16
17VNC_ADDR = '127.0.0.1'
18VNC_PORT_START = 32768
19VNC_PORT_END = VNC_PORT_START + 1024
20
21
22def check_bind(port: int) -> bool:
23    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:
24        try:
25            sock.bind((VNC_ADDR, port))
26        except OSError:
27            return False
28
29    return True
30
31
32def check_connect(port: int) -> bool:
33    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:
34        try:
35            sock.connect((VNC_ADDR, port))
36        except ConnectionRefusedError:
37            return False
38
39    return True
40
41
42def find_free_ports(count: int) -> List[int]:
43    result = []
44    for port in range(VNC_PORT_START, VNC_PORT_END):
45        if check_bind(port):
46            result.append(port)
47            if len(result) >= count:
48                break
49    assert len(result) == count
50    return result
51
52
53class Vnc(QemuSystemTest):
54    """
55    :avocado: tags=vnc,quick
56    :avocado: tags=machine:none
57    """
58    def test_no_vnc(self):
59        self.vm.add_args('-nodefaults', '-S')
60        self.vm.launch()
61        self.assertFalse(self.vm.qmp('query-vnc')['return']['enabled'])
62
63    def test_no_vnc_change_password(self):
64        self.vm.add_args('-nodefaults', '-S')
65        self.vm.launch()
66        self.assertFalse(self.vm.qmp('query-vnc')['return']['enabled'])
67        set_password_response = self.vm.qmp('change-vnc-password',
68                                            password='new_password')
69        self.assertIn('error', set_password_response)
70        self.assertEqual(set_password_response['error']['class'],
71                         'GenericError')
72        self.assertEqual(set_password_response['error']['desc'],
73                         'Could not set password')
74
75    def test_change_password_requires_a_password(self):
76        self.vm.add_args('-nodefaults', '-S', '-vnc', ':0')
77        self.vm.launch()
78        self.assertTrue(self.vm.qmp('query-vnc')['return']['enabled'])
79        set_password_response = self.vm.qmp('change-vnc-password',
80                                            password='new_password')
81        self.assertIn('error', set_password_response)
82        self.assertEqual(set_password_response['error']['class'],
83                         'GenericError')
84        self.assertEqual(set_password_response['error']['desc'],
85                         'Could not set password')
86
87    def test_change_password(self):
88        self.vm.add_args('-nodefaults', '-S', '-vnc', ':0,password=on')
89        self.vm.launch()
90        self.assertTrue(self.vm.qmp('query-vnc')['return']['enabled'])
91        set_password_response = self.vm.qmp('change-vnc-password',
92                                            password='new_password')
93        self.assertEqual(set_password_response['return'], {})
94
95    def test_change_listen(self):
96        a, b, c = find_free_ports(3)
97        self.assertFalse(check_connect(a))
98        self.assertFalse(check_connect(b))
99        self.assertFalse(check_connect(c))
100
101        self.vm.add_args('-nodefaults', '-S', '-vnc', f'{VNC_ADDR}:{a - 5900}')
102        self.vm.launch()
103        self.assertEqual(self.vm.qmp('query-vnc')['return']['service'], str(a))
104        self.assertTrue(check_connect(a))
105        self.assertFalse(check_connect(b))
106        self.assertFalse(check_connect(c))
107
108        res = self.vm.qmp('display-update', type='vnc',
109                          addresses=[{'type': 'inet', 'host': VNC_ADDR,
110                                      'port': str(b)},
111                                     {'type': 'inet', 'host': VNC_ADDR,
112                                      'port': str(c)}])
113        self.assertEqual(res['return'], {})
114        self.assertEqual(self.vm.qmp('query-vnc')['return']['service'], str(b))
115        self.assertFalse(check_connect(a))
116        self.assertTrue(check_connect(b))
117        self.assertTrue(check_connect(c))
118