From 0568d2ae3bfa27782455f45eb00c512e4c0c20a7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marek=20Marczykowski-G=C3=B3recki?=
 <marmarek@invisiblethingslab.com>
Date: Thu, 11 Feb 2016 02:16:01 +0100
Subject: [PATCH] qrexec: initial convert qrexec-policy to core3 API

---
 qrexec/qrexec-policy | 45 +++++++++++++++++++++++++++++++++-----------
 1 file changed, 34 insertions(+), 11 deletions(-)

diff --git a/qrexec/qrexec-policy b/qrexec/qrexec-policy
index 0d236ae..eaacd04 100755
--- a/qrexec/qrexec-policy
+++ b/qrexec/qrexec-policy
@@ -3,12 +3,12 @@ import sys
 import os
 import os.path
 import subprocess
-from qubes.qubes import vmm
-from qubes.qubes import QubesVmCollection
-import qubes.guihelpers
+import qubes
 import libvirt
 from optparse import OptionParser
 import fcntl
+from PyQt4.QtGui import QApplication,QMessageBox
+
 
 POLICY_FILE_DIR="/etc/qubes-rpc/policy"
 # XXX: Backward compatibility, to be removed soon
@@ -21,6 +21,31 @@ class UserChoice:
     DENY=1
     ALWAYS_ALLOW=2
 
+def prepare_app():
+    app = QApplication(sys.argv)
+    app.setOrganizationName("The Qubes Project")
+    app.setOrganizationDomain("http://qubes-os.org")
+    app.setApplicationName("Qubes")
+    return app
+
+def ask(text, title="Question", yestoall=False):
+    prepare_app()
+
+    buttons = QMessageBox.Yes | QMessageBox.No
+    if yestoall:
+        buttons |= QMessageBox.YesToAll
+
+    reply = QMessageBox.question(None, title, text, buttons, defaultButton=QMessageBox.Yes)
+    if reply == QMessageBox.Yes:
+        return 0
+    elif reply == QMessageBox.No:
+        return 1
+    elif reply == QMessageBox.YesToAll:
+        return 2
+    else:
+        #?!
+        return 127
+
 def line_to_dict(line):
     tokens=line.split()
     if len(tokens) < 3:
@@ -92,12 +117,9 @@ def validate_target(target):
     if target in ['$dispvm']:
         return True
 
-    qc = QubesVmCollection()
-    qc.lock_db_for_reading()
-    qc.load()
-    qc.unlock_db()
+    app = qubes.Qubes()
 
-    return qc.get_vm_by_name(target)
+    return app.domains[target]
 
 def spawn_target_if_necessary(vm):
     if vm.is_running():
@@ -115,7 +137,7 @@ def do_execute(domain, target, user, service_name, process_ident, vm=None):
         os.execl(QREXEC_CLIENT, "qrexec-client",
                 "-d", "dom0", "-c", process_ident, cmd)
     else:
-        if isinstance(vm, qubes.qubes.QubesVm):
+        if isinstance(vm, qubes.vm.qubesvm.QubesVM):
             spawn_target_if_necessary(vm)
         if target == "dom0":
             cmd = QUBES_RPC_MULTIPLEXER_PATH + " " + service_name + " " + domain
@@ -186,8 +208,9 @@ def main():
     # connection
     process_ident+=","+domain+","+domain_id
 
-    vm = validate_target(target)
-    if vm is None:
+    try:
+        vm = validate_target(target)
+    except KeyError:
         print >> sys.stderr, "Rpc failed (unknown domain):", domain, target, service_name
         text = "Domain '%s' doesn't exist (service %s called by domain %s)." % (
                 target, service_name, domain)