/* See LICENSE file for copyright and license details. */
#include "servers-crtc.h"
#include "servers-gamma.h"
#include "servers-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)
{
char *name;
if (!info->edid_error && info->edid) {
return libgamma_behex_edid(info->edid, info->edid_length);
} else if (!info->connector_name_error && info->connector_name) {
name = malloc(3 * sizeof(size_t) + strlen(info->connector_name) + 2);
if (name)
sprintf(name, "%zu.%s", crtc->partition->partition, info->connector_name);
return name;
} else {
name = malloc(2 * 3 * sizeof(size_t) + 2);
if (name)
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;
if (sitename && !(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:
sitename_dup = NULL;
libgamma_perror(argv0, gerror);
errno = 0;
fail:
free(sitename_dup);
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) {
partitions = calloc(site.partitions_available, sizeof(*partitions));
if (!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) {
crtcs = calloc(outputs_n, sizeof(*crtcs));
if (!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;
int cmp, is_same;
/* How many outputs does the system now have? */
i = j = new_outputs_n = 0;
while (i < old_outputs_n && j < outputs_n) {
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)
return -1;
}
/* Merge output states */
i = j = new_outputs_n = 0;
while (i < old_outputs_n && j < outputs_n) {
is_same = 0;
cmp = strcmp(old_outputs[i].name, outputs[j].name);
if (!cmp) {
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;
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:
for (i = 0; i < old_outputs_n; i++)
output_destroy(&old_outputs[i]);
free(old_outputs);
return -1;
}
