aboutsummaryrefslogtreecommitdiffstats
path: root/src/cg-remove.c
diff options
context:
space:
mode:
authorMattias Andrée <maandree@kth.se>2016-08-06 18:47:51 +0200
committerMattias Andrée <maandree@kth.se>2016-08-06 18:47:51 +0200
commit0ba0c93fef27df59831f4320c6324da0c17ad952 (patch)
tree8268812b46023c0da31c78f874956cf7811e846a /src/cg-remove.c
parentm (diff)
downloadcg-tools-0ba0c93fef27df59831f4320c6324da0c17ad952.tar.gz
cg-tools-0ba0c93fef27df59831f4320c6324da0c17ad952.tar.bz2
cg-tools-0ba0c93fef27df59831f4320c6324da0c17ad952.tar.xz
Add cg-remove
Signed-off-by: Mattias Andrée <maandree@kth.se>
Diffstat (limited to 'src/cg-remove.c')
-rw-r--r--src/cg-remove.c404
1 files changed, 404 insertions, 0 deletions
diff --git a/src/cg-remove.c b/src/cg-remove.c
new file mode 100644
index 0000000..065e6eb
--- /dev/null
+++ b/src/cg-remove.c
@@ -0,0 +1,404 @@
+/**
+ * cg-tools -- Cooperative gamma-enabled tools
+ * Copyright (C) 2016 Mattias Andrée (maandree@kth.se)
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+#define _DEFAULT_SOURCE
+#define _BSD_SOURCE
+
+#include "arg.h"
+
+#include <libcoopgamma.h>
+
+#include <errno.h>
+#include <inttypes.h>
+#include <poll.h>
+#include <signal.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+
+
+/**
+ * The libcoopgamma context
+ */
+static libcoopgamma_context_t cg;
+
+
+
+/**
+ * Print usage information and exit
+ */
+void usage(void)
+{
+ fprintf(stderr,
+ "Usage: %s [-M method] [-S site] [-c crtc]... class...\n",
+ argv0);
+ exit(1);
+}
+
+
+/**
+ * Initialise the process, specifically
+ * reset the signal mask and signal handlers
+ *
+ * @return Zero on success, -1 on error
+ */
+static int initialise_proc(void)
+{
+ sigset_t sigmask;
+ int sig;
+
+ for (sig = 1; sig < _NSIG; sig++)
+ if (signal(sig, SIG_DFL) == SIG_ERR)
+ if (sig == SIGCHLD)
+ return -1;
+
+ if (sigemptyset(&sigmask) < 0)
+ return -1;
+ if (sigprocmask(SIG_SETMASK, &sigmask, NULL) < 0)
+ return -1;
+
+ return 0;
+}
+
+
+/**
+ * Print, to stdout, a list of all
+ * recognised adjustment methods
+ *
+ * @return Zero on success, -1 on error
+ */
+static int list_methods(void)
+{
+ char** list;
+ size_t i;
+
+ list = libcoopgamma_get_methods();
+ if (list == NULL)
+ return -1;
+ for (i = 0; list[i]; i++)
+ printf("%s\n", list[i]);
+ free(list);
+ if (fflush(stdout) < 0)
+ return -1;
+
+ return 0;
+}
+
+
+/**
+ * Print, to stdout, a list of all CRTC:s
+ *
+ * A connection to the coopgamma server
+ * must have been made
+ *
+ * @return Zero on success, -1 on error, -2
+ * on libcoopgamma error
+ */
+static int list_crtcs(void)
+{
+ char** list;
+ size_t i;
+
+ list = libcoopgamma_get_crtcs_sync(&cg);
+ if (list == NULL)
+ return -2;
+ for (i = 0; list[i]; i++)
+ printf("%s\n", list[i]);
+ free(list);
+ if (fflush(stdout) < 0)
+ return -1;
+
+ return 0;
+}
+
+
+/**
+ * Remove selected filters from selected CRTC:s
+ *
+ * @param crtcs `NULL`-terminated list of CRTC names
+ * @param classes `NULL`-terminated list of filter classes
+ * @return Zero on success, -1 on error, -2 on
+ * libcoopgamma error
+ */
+static int remove_filters(char* const* restrict crtcs, char* const* restrict classes)
+{
+ size_t n = 0, unsynced = 0, selected, i, j;
+ char* synced = NULL;
+ libcoopgamma_async_context_t* asyncs = NULL;
+ int saved_errno, need_flush = 0, ret = 0;
+ struct pollfd pollfd;
+ libcoopgamma_filter_t command;
+
+ for (i = 0; crtcs[i] != NULL; i++);
+ for (j = 0; classes[j] != NULL; j++);
+ synced = calloc(i, j * sizeof(*synced));
+ if (synced == NULL)
+ goto fail;
+ asyncs = calloc(i, j * sizeof(*asyncs));
+ if (asyncs == NULL)
+ goto fail;
+
+ i = j = 0;
+ command.lifespan = LIBCOOPGAMMA_REMOVE;
+ pollfd.fd = cg.fd;
+ pollfd.events = POLLIN | POLLRDNORM | POLLRDBAND | POLLPRI;
+
+ while ((unsynced > 0) || (crtcs[i] != NULL))
+ {
+ wait:
+ if (crtcs[i] != NULL)
+ pollfd.events |= POLLOUT;
+ else
+ pollfd.events &= ~POLLOUT;
+
+ pollfd.revents = 0;
+ if (poll(&pollfd, (nfds_t)1, -1) < 0)
+ goto fail;
+
+ if (pollfd.revents & (POLLOUT | POLLERR | POLLHUP | POLLNVAL))
+ {
+ if (need_flush && (libcoopgamma_flush(&cg) < 0))
+ goto send_fail;
+ need_flush = 0;
+ for (; crtcs[i] != NULL; i++, j = 0)
+ {
+ command.crtc = crtcs[i];
+ while (classes[j] != NULL)
+ {
+ command.class = classes[j++];
+ if (unsynced++, libcoopgamma_set_gamma_send(&command, &cg, asyncs + n++) < 0)
+ goto send_fail;
+ }
+ }
+ goto send_done;
+ send_fail:
+ switch (errno)
+ {
+ case EINTR:
+ case EAGAIN:
+#if EAGAIN != EWOULDBLOCK
+ case EWOULDBLOCK:
+#endif
+ need_flush = 1;
+ if (classes[j] == NULL)
+ i++, j = 0;
+ break;
+ default:
+ goto fail;
+ }
+ }
+ send_done:
+
+ if ((unsynced == 0) && (crtcs[i] == NULL))
+ break;
+
+ if (pollfd.revents & (POLLIN | POLLRDNORM | POLLRDBAND | POLLPRI))
+ while (unsynced > 0)
+ switch (libcoopgamma_synchronise(&cg, asyncs, n, &selected))
+ {
+ case 0:
+ if (synced[selected])
+ {
+ libcoopgamma_skip_message(&cg);
+ break;
+ }
+ synced[selected] = 1;
+ unsynced -= 1;
+ if (libcoopgamma_set_gamma_recv(&cg, asyncs + selected) < 0)
+ goto cg_fail;
+ break;
+ default:
+ switch (errno)
+ {
+ case 0:
+ break;
+ case EINTR:
+ case EAGAIN:
+#if EAGAIN != EWOULDBLOCK
+ case EWOULDBLOCK:
+#endif
+ goto wait;
+ default:
+ goto fail;
+ }
+ break;
+ }
+ }
+
+ done:
+ saved_errno = errno;
+ free(synced);
+ free(asyncs);
+ errno = saved_errno;
+ return ret;
+ fail:
+ ret = -1;
+ goto done;
+ cg_fail:
+ ret = -2;
+ goto done;
+}
+
+
+/**
+ * -M METHOD
+ * Select adjustment method. If METHOD is "?",
+ * available methods will be printed to stdout.
+ *
+ * -S SITE
+ * Select site (display server instance).
+ *
+ * -c CRTC
+ * Select CRT controller. If CRTC is "?", CRTC:s
+ * will be printed to stdout.
+ *
+ * Can be used multiple times. If not used, all
+ * CRTC:s are selected.
+ *
+ * @param argc The number of command line arguments
+ * @param argv The command line arguments
+ * @return 0 on success, 1 on error
+ */
+int main(int argc, char* argv[])
+{
+ int stage = 0;
+ int rc = 0;
+ char* method = NULL;
+ char* site = NULL;
+ char** crtcs_ = NULL;
+ char** crtcs = alloca(argc * sizeof(char*));
+ size_t i, crtc_n = 0;
+
+ ARGBEGIN
+ {
+ case 'M':
+ if (method != NULL)
+ usage();
+ method = EARGF(usage());
+ break;
+ case 'S':
+ if (site != NULL)
+ usage();
+ site = EARGF(usage());
+ break;
+ case 'c':
+ crtcs[crtc_n++] = EARGF(usage());
+ break;
+ }
+ ARGEND;
+
+ if (initialise_proc() < 0)
+ goto fail;
+
+ if ((method != NULL) && !strcmp(method, "?"))
+ {
+ if ((site != NULL) || (crtc_n > 0) || (argc > 0))
+ usage();
+ if (list_methods() < 0)
+ goto fail;
+ return 0;
+ }
+
+ if (libcoopgamma_context_initialise(&cg) < 0)
+ goto fail;
+ stage++;
+ if (libcoopgamma_connect(method, site, &cg) < 0)
+ {
+ fprintf(stderr, "%s: server failed to initialise\n", argv0);
+ goto custom_fail;
+ }
+ stage++;
+
+ for (i = 0; i < crtc_n; i++)
+ if (!strcmp(crtcs[i], "?"))
+ {
+ if (argc > 0)
+ usage();
+ switch (list_crtcs())
+ {
+ case 0:
+ goto done;
+ case -1:
+ goto fail;
+ default:
+ goto cg_fail;
+ }
+ }
+
+ if (argc == 0)
+ usage();
+
+ if (crtc_n == 0)
+ {
+ crtcs = crtcs_ = libcoopgamma_get_crtcs_sync(&cg);
+ if (crtcs == NULL)
+ goto cg_fail;
+ }
+ else
+ crtcs[crtc_n] = NULL;
+
+ if (libcoopgamma_set_nonblocking(&cg, 1) < 0)
+ goto fail;
+
+ switch (remove_filters(crtcs, argv))
+ {
+ case 0:
+ break;
+ case -1:
+ goto fail;
+ default:
+ goto cg_fail;
+ }
+
+ done:
+ if (stage >= 1)
+ libcoopgamma_context_destroy(&cg, stage >= 2);
+ free(crtcs_);
+ return rc;
+
+ custom_fail:
+ rc = 1;
+ goto done;
+
+ fail:
+ rc = 1;
+ perror(argv0);
+ goto done;
+
+ cg_fail:
+ rc = 1;
+ {
+ const char* side = cg.error.server_side ? "server" : "client";
+ if (cg.error.custom)
+ {
+ if ((cg.error.number != 0) || (cg.error.description != NULL))
+ fprintf(stderr, "%s: %s-side error number %" PRIu64 ": %s\n",
+ argv0, side, cg.error.number, cg.error.description);
+ else if (cg.error.number != 0)
+ fprintf(stderr, "%s: %s-side error number %" PRIu64 "\n", argv0, side, cg.error.number);
+ else if (cg.error.description != NULL)
+ fprintf(stderr, "%s: %s-side error: %s\n", argv0, side, cg.error.description);
+ }
+ else if (cg.error.description != NULL)
+ fprintf(stderr, "%s: %s-side error: %s\n", argv0, side, cg.error.description);
+ else
+ fprintf(stderr, "%s: %s-side error: %s\n", argv0, side, strerror(cg.error.number));
+ }
+ goto done;
+}
+