/**
* coopgammad -- Cooperative gamma server
* 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/>.
*/
#include "crtc.h"
#include "gamma.h"
#include "coopgamma.h"
#include "../state.h"
#include "../communication.h"
#include "../util.h"
#include <errno.h>
#include <string.h>
/**
* Handle a ‘Command: enumerate-crtcs’ message
*
* @param conn The index of the connection
* @param message_id The value of the ‘Message ID’ header
* @return Zero on success (even if ignored), -1 on error,
* 1 if connection closed
*/
int handle_enumerate_crtcs(size_t conn, const char* restrict message_id)
{
size_t i, n = 0, len;
char* restrict buf;
for (i = 0; i < outputs_n; i++)
n += strlen(outputs[i].name) + 1;
MAKE_MESSAGE(&buf, &n, n,
"Command: crtc-enumeration\n"
"In response to: %s\n"
"Length: %zu\n"
"\n",
message_id, n);
for (i = 0; i < outputs_n; i++)
{
len = strlen(outputs[i].name);
memcpy(buf + n, outputs[i].name, len);
buf[n + len] = '\n';
n += len + 1;
}
return send_message(conn, buf, n);
}
/**
* Get the name of a CRTC
*
* @param info Information about the CRTC
* @param crtc libgamma's state for the CRTC
* @return The name of the CRTC, `NULL` on error
*/
char* get_crtc_name(const libgamma_crtc_information_t* restrict info,
const libgamma_crtc_state_t* restrict crtc)
{
if ((info->edid_error == 0) && (info->edid != NULL))
return libgamma_behex_edid(info->edid, info->edid_length);
else if ((info->connector_name_error == 0) && (info->connector_name != NULL))
{
char* name = malloc(3 * sizeof(size_t) + strlen(info->connector_name) + 2);
if (name != NULL)
sprintf(name, "%zu.%s", crtc->partition->partition, info->connector_name);
return name;
}
else
{
char* name = malloc(2 * 3 * sizeof(size_t) + 2);
if (name != NULL)
sprintf(name, "%zu.%zu", crtc->partition->partition, crtc->crtc);
return name;
}
}
/**
* Initialise the site
*
* @return Zero on success, -1 on error
*/
int initialise_site(void)
{
char* restrict sitename_dup = NULL;
int gerror, saved_errno;
if ((sitename != NULL) && !(sitename_dup = memdup(sitename, strlen(sitename) + 1)))
goto fail;
if ((gerror = libgamma_site_initialise(&site, method, sitename_dup)))
goto fail_libgamma;
return 0;
fail_libgamma:
libgamma_perror(argv0, gerror);
errno = 0;
fail:
saved_errno = errno;
free(sitename_dup);
errno = saved_errno;
return -1;
}
/**
* Get partitions and CRTC:s
*
* @return Zero on success, -1 on error
*/
int initialise_crtcs(void)
{
size_t i, j, n, n0;
int gerror;
/* Get partitions */
outputs_n = 0;
if (site.partitions_available)
if (!(partitions = calloc(site.partitions_available, sizeof(*partitions))))
goto fail;
for (i = 0; i < site.partitions_available; i++)
{
if ((gerror = libgamma_partition_initialise(partitions + i, &site, i)))
goto fail_libgamma;
outputs_n += partitions[i].crtcs_available;
}
/* Get CRTC:s */
if (outputs_n)
if (!(crtcs = calloc(outputs_n, sizeof(*crtcs))))
goto fail;
for (i = 0, j = n = 0; i < site.partitions_available; i++)
for (n0 = n, n += partitions[i].crtcs_available; j < n; j++)
if ((gerror = libgamma_crtc_initialise(crtcs + j, partitions + i, j - n0)))
goto fail_libgamma;
return 0;
fail_libgamma:
libgamma_perror(argv0, gerror);
errno = 0;
fail:
return -1;
}
/**
* Merge the new state with an old state
*
* @param old_outputs The old `outputs`
* @param old_outputs_n The old `outputs_n`
* @return Zero on success, -1 on error
*/
int merge_state(struct output* restrict old_outputs, size_t old_outputs_n)
{
struct output* restrict new_outputs = NULL;
size_t new_outputs_n;
size_t i, j;
/* How many outputs does the system now have? */
i = j = new_outputs_n = 0;
while ((i < old_outputs_n) && (j < outputs_n))
{
int cmp = strcmp(old_outputs[i].name, outputs[j].name);
if (cmp <= 0)
new_outputs_n++;
i += cmp >= 0;
j += cmp <= 0;
}
new_outputs_n += outputs_n - j;
/* Allocate output state array */
if (new_outputs_n > 0)
{
new_outputs = calloc(new_outputs_n, sizeof(*new_outputs));
if (new_outputs == NULL)
return -1;
}
/* Merge output states */
i = j = new_outputs_n = 0;
while ((i < old_outputs_n) && (j < outputs_n))
{
int is_same = 0, cmp = strcmp(old_outputs[i].name, outputs[j].name);
if (cmp == 0)
is_same = (old_outputs[i].depth == outputs[j].depth &&
old_outputs[i].red_size == outputs[j].red_size &&
old_outputs[i].green_size == outputs[j].green_size &&
old_outputs[i].blue_size == outputs[j].blue_size);
if (is_same)
{
new_outputs[new_outputs_n] = old_outputs[i];
new_outputs[new_outputs_n].crtc = outputs[j].crtc;
memset(old_outputs + i, 0, sizeof(*old_outputs));
outputs[j].crtc = NULL;
output_destroy(outputs + j);
new_outputs_n++;
}
else if (cmp <= 0)
new_outputs[new_outputs_n++] = outputs[j];
i += cmp >= 0;
j += cmp <= 0;
}
while (j < outputs_n)
new_outputs[new_outputs_n++] = outputs[j++];
/* Commit merge */
free(outputs);
outputs = new_outputs;
outputs_n = new_outputs_n;
return 0;
}
/**
* Disconnect from the site
*
* @return Zero on success, -1 on error
*/
int disconnect(void)
{
size_t i;
if (!connected)
return 0;
connected = 0;
for (i = 0; i < outputs_n; i++)
{
outputs[i].crtc = NULL;
libgamma_crtc_destroy(crtcs + i);
}
free(crtcs), crtcs = NULL;
for (i = 0; i < site.partitions_available; i++)
libgamma_partition_destroy(partitions + i);
free(partitions), partitions = NULL;
libgamma_site_destroy(&site);
memset(&site, 0, sizeof(site));
return 0;
}
/**
* Reconnect to the site
*
* @return Zero on success, -1 on error
*/
int reconnect(void)
{
struct output* restrict old_outputs;
size_t i, old_outputs_n;
int saved_errno;
if (connected)
return 0;
connected = 1;
/* Remember old state */
old_outputs = outputs, outputs = NULL;
old_outputs_n = outputs_n, outputs_n = 0;
/* Get site */
if (initialise_site() < 0)
goto fail;
/* Get partitions and CRTC:s */
if (initialise_crtcs() < 0)
goto fail;
/* Get CRTC information */
if (outputs_n && !(outputs = calloc(outputs_n, sizeof(*outputs))))
goto fail;
if (initialise_gamma_info() < 0)
goto fail;
/* Sort outputs */
qsort(outputs, outputs_n, sizeof(*outputs), output_cmp_by_name);
/* Load current gamma ramps */
store_gamma();
/* Preserve current gamma ramps at priority=0 if -p */
if (preserve && (preserve_gamma() < 0))
goto fail;
/* Merge state */
if (merge_state(old_outputs, old_outputs_n) < 0)
goto fail;
for (i = 0; i < old_outputs_n; i++)
output_destroy(old_outputs + i);
free(old_outputs), old_outputs = NULL, old_outputs_n = 0;
/* Reapply gamma ramps */
reapply_gamma();
return 0;
fail:
saved_errno = errno;
for (i = 0; i < old_outputs_n; i++)
output_destroy(old_outputs + i);
free(old_outputs);
errno = saved_errno;
return -1;
}
