Add cg-remove

Signed-off-by: Mattias Andrée <maandree@kth.se>

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;
+}
+