Implement invoking `sway` as IPC client

As an alternative to invoking swaymsg.

1 files changed, 25 insertions, 0 deletions

diff --git a/sway/main.c b/sway/main.c
index c7696e2e..9a5e351c 100644
--- a/sway/main.c
+++ b/sway/main.c
@@ -6,6 +6,7 @@
 #include <sys/types.h>
 #include <sys/un.h>
 #include <signal.h>
+#include <unistd.h>
 #include <getopt.h>
 #include "extensions.h"
 #include "layout.h"
@@ -14,6 +15,7 @@
 #include "log.h"
 #include "readline.h"
 #include "handlers.h"
+#include "ipc-client.h"
 #include "ipc-server.h"
 #include "sway.h"
 
@@ -51,6 +53,14 @@ void detect_nvidia() {
         fclose(f);
 }
 
+void run_as_ipc_client(char *command, char *socket_path) {
+        int socketfd = ipc_open_socket(socket_path);
+        uint32_t len = strlen(command);
+        char *resp = ipc_single_command(socketfd, IPC_COMMAND, command, &len);
+        printf("%s\n", resp);
+        close(socketfd);
+}
+
 int main(int argc, char **argv) {
         static int verbose = 0, debug = 0, validate = 0;
 
@@ -126,6 +136,21 @@ int main(int argc, char **argv) {
                 }
         }
 
+        if (optind < argc) { // Behave as IPC client
+                if (getuid() != geteuid() || getgid() != getegid()) {
+                        if (setgid(getgid()) != 0 || setuid(getuid()) != 0) {
+                                sway_abort("Unable to drop root");
+                        }
+                }
+                char *socket_path = getenv("SWAYSOCK");
+                if (!socket_path) {
+                        sway_abort("Unable to retrieve socket path");
+                }
+                char *command = join_args(argv + optind, argc - optind);
+                run_as_ipc_client(command, socket_path);
+                return 0;
+        }
+
         // we need to setup logging before wlc_init in case it fails.
         if (debug) {
                 init_log(L_DEBUG);