more on firecfg --guide: fzenity

4 files changed, 205 insertions, 5 deletions

diff --git a/src/firecfg/firejail-welcome.sh b/src/firecfg/firejail-welcome.sh
index 7183b74e1..a7e74ebc3 100755
--- a/src/firecfg/firejail-welcome.sh
+++ b/src/firecfg/firejail-welcome.sh
@@ -36,12 +36,13 @@ enable_nonewprivs=false
 read -r -d $'\0' MSG_INTRO <<EOM
 <big><b>Welcome to Firejail!</b></big>
 
-This guide will walk you through some of the most common sandbox customizations. At the end of the
-guide you'll have the option to save your changes in Firejail's global config file at
-<b>/etc/firejail/firejail.config</b>. A copy of the original file is stored as
-<b>/etc/firejal/firejail.config-</b>.
+This guide will walk you through some of the most common sandbox customizations.
+At the end of the guide you'll have the option to save your changes in Firejail's
+global config file at <b>/etc/firejail/firejail.config</b>. A copy of the original file is saved
+as <b>/etc/firejal/firejail.config-</b>.
 
-Please note that running this script a second time can set new options, but does not clear options set in a previous run.
+Please note that running this script a second time can set new options, but does
+not clear options set in a previous run.
 
 Press OK to continue, or close this window to stop the program.
 
diff --git a/src/firecfg/main.c b/src/firecfg/main.c
index 0c81f69bd..07e30415b 100644
--- a/src/firecfg/main.c
+++ b/src/firecfg/main.c
@@ -444,8 +444,14 @@ int main(int argc, char **argv) {
 
         if (arg_guide) {
                 char *cmd;
+if (arg_debug) {
+                if (asprintf(&cmd, "sudo %s/firejail/firejail-welcome.sh /usr/lib/firejail/fzenity %s %s", LIBDIR, SYSCONFDIR, user) == -1)
+                        errExit("asprintf");
+}
+else {
                 if (asprintf(&cmd, "sudo %s/firejail/firejail-welcome.sh /usr/bin/zenity %s %s", LIBDIR, SYSCONFDIR, user) == -1)
                         errExit("asprintf");
+}
                 int status = system(cmd);
                 if (status == -1) {
                         fprintf(stderr, "Error: cannot run firejail-welcome.sh\n");
diff --git a/src/fzenity/Makefile.in b/src/fzenity/Makefile.in
new file mode 100644
index 000000000..d9f976165
--- /dev/null
+++ b/src/fzenity/Makefile.in
@@ -0,0 +1,17 @@
+.PHONY: all
+all: fzenity
+
+include ../common.mk
+
+%.o : %.c $(H_FILE_LIST) ../include/common.h
+        $(CC) $(CFLAGS) $(EXTRA_CFLAGS) $(INCLUDE) -c $< -o $@
+
+fzenity: $(OBJS)
+        $(CC)  $(LDFLAGS) -o $@ $(OBJS) $(LIBS) $(EXTRA_LDFLAGS)
+
+.PHONY: clean
+clean:; rm -fr *.o fzenity *.gcov *.gcda *.gcno *.plist
+
+.PHONY: distclean
+distclean: clean
+        rm -fr Makefile
diff --git a/src/fzenity/main.c b/src/fzenity/main.c
new file mode 100644
index 000000000..4a0d3abac
--- /dev/null
+++ b/src/fzenity/main.c
@@ -0,0 +1,176 @@
+#include "../include/common.h"
+#include <sys/ioctl.h>
+
+static char *arg_title = NULL;
+static char *arg_text = NULL;
+static int arg_info = 0;
+static int arg_question = 0;
+
+static inline void ansi_topleft(void) {
+        char str[] = {0x1b, '[', '1', ';',  '1', 'H', '\0'};
+        printf("%s", str);
+        fflush(0);
+}
+
+static inline void ansi_clrscr(void) {
+        ansi_topleft();
+        char str[] = {0x1b, '[', '0', 'J', '\0'};
+        printf("%s", str);
+        fflush(0);
+}
+
+char *remove_markup(char *in) {
+        char *out = malloc(strlen(in) + 1);
+        if (!out)
+                errExit("malloc");
+        memset(out, 0, strlen(in) + 1);
+
+        char *ptr = in;
+        char *outptr = out;
+        while (*ptr != '\0') {
+                // skip <> markup
+                if (*ptr == '<') {
+                        while (*ptr != '\0' && *ptr != '>')
+                                ptr++;
+                        if (*ptr == '\0') {
+                                fprintf(stderr, "Error: invalid markup\n");
+                                exit(0);
+                        }
+                        ptr++;
+                }
+                // replace literal \n with char '\n'
+                else if (*ptr == '\\' && *(ptr + 1) == 'n') {
+                        ptr += 2;
+                        *outptr++ = '\n';
+                        continue;
+                }
+                // replace '/n' with ' '
+                else if (*ptr == '\n') {
+                        if (*(ptr + 1) == '\n') {
+                                *outptr++ = '\n';
+                                *outptr++ = '\n';
+                                ptr += 2;
+                        }
+                        else {
+                                *outptr++ = ' ';
+                                ptr++;
+                        }
+                }
+                else
+                        *outptr++ = *ptr++;
+        }
+
+        return out;
+}
+
+char *print_line(char *in, int col) {
+        char *ptr = in;
+        int i = 0;
+        while (*ptr !=  '\n' && *ptr != '\0' && i < col) {
+                ptr++;
+                i++;
+        }
+
+        if (*ptr == '\n') {
+                *ptr++ = '\0';
+                printf("%s\n", in);
+                return ptr++;
+        }
+        else if (i == col) {
+                while (*ptr != ' ' && ptr != in)
+                        ptr--;
+                *ptr++ = '\0';
+                printf("%s\n", in);
+                return ptr;
+        }
+        assert(0);
+        return NULL;
+}
+
+void paginate(char *in) {
+        struct winsize w;
+        int col = 80;
+        if (ioctl(0, TIOCGWINSZ, &w) == 0)
+                col = w.ws_col;
+
+        char *ptr = in;
+        while (*ptr != '\0') {
+                if (strlen(ptr) < col) {
+                        printf("%s", ptr);
+                        return;
+                }
+                ptr =print_line(ptr, col);
+        }
+
+        return;
+}
+
+static void info(void) {
+        ansi_clrscr();
+        if (arg_text == NULL) {
+                fprintf(stderr, "Error: --text argument required\n");
+                exit(1);
+        }
+
+        if (arg_title)
+                printf("%s\n\n", arg_title);
+
+        char *ptr = strstr(arg_text, "Press OK to continue");
+        if (ptr)
+                *ptr = '\0';
+        char *out = remove_markup(arg_text);
+        paginate(out);
+        free(out);
+
+        printf("\nContinue? (Y/N): ");
+
+        int c = getchar();
+        if (c == 'y' || c == 'Y')
+                exit(0);
+        exit(1);
+}
+
+static void question(void) {
+        ansi_clrscr();
+        if (arg_text == NULL) {
+                fprintf(stderr, "Error: --text argument required\n");
+                exit(1);
+        }
+
+        if (arg_title)
+                printf("%s\n\n", arg_title);
+
+        char *ptr = strstr(arg_text, "Press OK to continue");
+        if (ptr)
+                *ptr = '\0';
+        char *out = remove_markup(arg_text);
+        paginate(out);
+        free(out);
+
+        printf("\n\n(Y/N): ");
+
+        int c = getchar();
+        if (c == 'y' || c == 'Y')
+                exit(0);
+        exit(1);
+}
+
+int main(int argc, char **argv) {
+        int i;
+        for (i = 1; i < argc; i++) {
+//printf("argv %d: #%s#\n", i, argv[i]);
+                if (strcmp(argv[i], "--info") == 0)
+                        arg_info = 1;
+                else if (strcmp(argv[i], "--question") == 0)
+                        arg_question = 1;
+                else if (strncmp(argv[i], "--text=", 7) == 0)
+                        arg_text = argv[i] + 7;
+        }
+
+        if (arg_question)
+                question();
+        else if (arg_info)
+                info();
+
+        return 0;
+}