From 4acf69e8ecd1ba82f3f7bbead4f5a3ab2a143352 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pawe=C5=82=20Marczewski?= <pawel@invisiblethingslab.com>
Date: Thu, 23 Jul 2020 13:32:16 +0200
Subject: [PATCH] Add power state to get_system_info

For qrexec policy, to implement 'autostart'
(see QubesOS/qubes-issues#5952).
---
 qubes/api/internal.py       | 1 +
 qubes/tests/api_internal.py | 4 ++++
 2 files changed, 5 insertions(+)

diff --git a/qubes/api/internal.py b/qubes/api/internal.py
index 19659a17..0b302582 100644
--- a/qubes/api/internal.py
+++ b/qubes/api/internal.py
@@ -42,6 +42,7 @@ def get_system_info(app):
             'icon': str(domain.label.icon),
             'guivm': (domain.guivm.name if getattr(domain, 'guivm', None)
                       else None),
+            'power_state': domain.get_power_state(),
         } for domain in app.domains
     }}
     return system_info
diff --git a/qubes/tests/api_internal.py b/qubes/tests/api_internal.py
index b5e73b15..3e51a6a2 100644
--- a/qubes/tests/api_internal.py
+++ b/qubes/tests/api_internal.py
@@ -150,6 +150,7 @@ class TC_00_API_Misc(qubes.tests.QubesTestCase):
         self.dom0.default_dispvm = None
         self.dom0.template_for_dispvms = False
         self.dom0.label.icon = 'icon-dom0'
+        self.dom0.get_power_state.return_value = 'Running'
         del self.dom0.guivm
 
         vm = mock.NonCallableMock(spec=qubes.vm.qubesvm.QubesVM)
@@ -159,6 +160,7 @@ class TC_00_API_Misc(qubes.tests.QubesTestCase):
         vm.template_for_dispvms = True
         vm.label.icon = 'icon-vm'
         vm.guivm = vm
+        vm.get_power_state.return_value = 'Halted'
         self.domains['vm'] = vm
 
         ret = json.loads(self.call_mgmt_func(b'internal.GetSystemInfo'))
@@ -171,6 +173,7 @@ class TC_00_API_Misc(qubes.tests.QubesTestCase):
                     'template_for_dispvms': False,
                     'icon': 'icon-dom0',
                     'guivm': None,
+                    'power_state': 'Running',
                 },
                 'vm': {
                     'tags': ['tag3', 'tag4'],
@@ -179,6 +182,7 @@ class TC_00_API_Misc(qubes.tests.QubesTestCase):
                     'template_for_dispvms': True,
                     'icon': 'icon-vm',
                     'guivm': 'vm',
+                    'power_state': 'Halted',
                 }
             }
         })