From d2f4b5d694642f2b776441d458aaae5bb80945a2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marek=20Marczykowski-G=C3=B3recki?=
 <marmarek@invisiblethingslab.com>
Date: Sat, 14 Sep 2019 23:32:31 +0200
Subject: [PATCH] Make qubesd-query-fast support keyword targets

Specifically, allow to use `@default` and `@adminvm` as a target.
---
 qubes-rpc/qubesd-query-fast.c | 25 ++++++++++++++++++++++---
 1 file changed, 22 insertions(+), 3 deletions(-)

diff --git a/qubes-rpc/qubesd-query-fast.c b/qubes-rpc/qubesd-query-fast.c
index 7505fc23..eeb02cf0 100644
--- a/qubes-rpc/qubesd-query-fast.c
+++ b/qubes-rpc/qubesd-query-fast.c
@@ -1,3 +1,4 @@
+#include <assert.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <unistd.h>
@@ -22,7 +23,10 @@ void write_wrapper(int fd, char *data, size_t len) {
 
 int main(int argc, char **argv) {
     char *source_domain = getenv("QREXEC_REMOTE_DOMAIN");
-    char *target_domain = getenv("QREXEC_REQUESTED_TARGET");
+    char *target_domain = NULL;
+    char *target_domain_type = getenv("QREXEC_REQUESTED_TARGET_TYPE");
+    char *target_domain_name = getenv("QREXEC_REQUESTED_TARGET");
+    char *target_domain_keyword = getenv("QREXEC_REQUESTED_TARGET_KEYWORD");
     char *service_name = strrchr(argv[0], '/');
     int fd;
     char buf[4096];
@@ -35,12 +39,12 @@ int main(int argc, char **argv) {
     if (service_name)
         service_name++;
 
-    if (!source_domain || !target_domain || !service_name || argc > 2) {
+    if (!source_domain || !target_domain_type || !service_name || argc > 2) {
         fprintf(stderr, "Usage: %s [service-argument]\n", argv[0]);
         fprintf(stderr, "\n");
         fprintf(stderr, "Expected environment variables:\n");
         fprintf(stderr, " - QREXEC_REMOTE_DOMAIN - source domain for the call\n");
-        fprintf(stderr, " - QREXEC_REQUESTED_TARGET - target domain for the call\n");
+        fprintf(stderr, " - QREXEC_REQUESTED_TARGET_TYPE - target domain for the call\n");
         fprintf(stderr, "\n");
         fprintf(stderr, "Additionally, this program assumes being called with desired service name as argv[0] (use symlink)\n");
         return 1;
@@ -57,6 +61,21 @@ int main(int argc, char **argv) {
         return 1;
     }
 
+    if (strcmp(target_domain_type, "keyword") == 0) {
+        // translate @adminvm back to dom0
+        if (strcmp(target_domain_keyword, "adminvm") == 0)
+            target_domain = "dom0";
+        else if (strcmp(target_domain_keyword, "default") == 0)
+            target_domain = "";
+        else {
+            fprintf(stderr, "Unsupported keyword: %s", target_domain_keyword);
+            return 1;
+        }
+    } else
+        target_domain = target_domain_name;
+
+    assert(target_domain);
+
     // write parameters, including trailing zero as separator
     write_wrapper(fd, source_domain, strlen(source_domain) + 1);
     write_wrapper(fd, service_name, strlen(service_name) + 1);