source: mainline/uspace/app/disp/disp.c

/*
 * Copyright (c) 2023 Jiri Svoboda
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/** @addtogroup disp
 * @{
 */
/** @file Display configuration utility.
 *
 * Configures the display service.
 */

#include <errno.h>
#include <dispcfg.h>
#include <io/table.h>
#include <loc.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <str.h>
#include <str_error.h>

#define NAME  "disp"

static void print_syntax(void)
{
        printf("%s: Display configuration utility.\n", NAME);
        printf("Syntax:\n");
        printf("  %s list-seat\n", NAME);
        printf("  %s create-seat <name>\n", NAME);
        printf("  %s delete-seat <name>\n", NAME);
        printf("  %s assign-dev <device> <seat>\n", NAME);
        printf("  %s unassign-dev <device>\n", NAME);
        printf("  %s list-dev <seat>\n", NAME);
}

/** Find seat by name.
 *
 * @param dispcfg Display configuration
 * @param name Seat name
 * @param rseat_id Place to store seat ID
 * @return EOK on success or an error code
 */
static errno_t seat_find_by_name(dispcfg_t *dispcfg, const char *name,
    sysarg_t *rseat_id)
{
        dispcfg_seat_list_t *seat_list;
        dispcfg_seat_info_t *sinfo;

        size_t i;
        errno_t rc;

        rc = dispcfg_get_seat_list(dispcfg, &seat_list);
        if (rc != EOK) {
                printf(NAME ": Failed getting seat list.\n");
                return rc;
        }

        for (i = 0; i < seat_list->nseats; i++) {
                rc = dispcfg_get_seat_info(dispcfg, seat_list->seats[i],
                    &sinfo);
                if (rc != EOK)
                        continue;

                if (str_cmp(sinfo->name, name) == 0) {
                        *rseat_id = seat_list->seats[i];
                        dispcfg_free_seat_info(sinfo);
                        return EOK;
                }

                dispcfg_free_seat_info(sinfo);
        }

        return ENOENT;
}

/** Create seat subcommand.
 *
 * @param dcfg_svc Display configuration service name
 * @param argc Number of arguments
 * @param argv Arguments
 * @return EOK on success or an erro code
 */
static errno_t create_seat(const char *dcfg_svc, int argc, char *argv[])
{
        dispcfg_t *dispcfg;
        char *seat_name;
        sysarg_t seat_id;
        errno_t rc;

        if (argc < 1) {
                printf(NAME ": Missing arguments.\n");
                print_syntax();
                return EINVAL;
        }

        if (argc > 1) {
                printf(NAME ": Too many arguments.\n");
                print_syntax();
                return EINVAL;
        }

        seat_name = argv[0];

        rc = dispcfg_open(dcfg_svc, NULL, NULL, &dispcfg);
        if (rc != EOK) {
                printf(NAME ": Failed connecting to display configuration "
                    "service: %s.\n", str_error(rc));
                return rc;
        }

        rc = dispcfg_seat_create(dispcfg, seat_name, &seat_id);
        if (rc != EOK) {
                printf(NAME ": Failed creating seat '%s' (%s)\n",
                    seat_name, str_error(rc));
                dispcfg_close(dispcfg);
                return EIO;
        }

        (void)seat_id;
        dispcfg_close(dispcfg);
        return EOK;
}

/** Delete seat subcommand.
 *
 * @param dcfg_svc Display configuration service name
 * @param argc Number of arguments
 * @param argv Arguments
 * @return EOK on success or an erro code
 */
static errno_t delete_seat(const char *dcfg_svc, int argc, char *argv[])
{
        dispcfg_t *dispcfg;
        char *seat_name;
        sysarg_t seat_id;
        errno_t rc;

        if (argc < 1) {
                printf(NAME ": Missing arguments.\n");
                print_syntax();
                return EINVAL;
        }

        if (argc > 1) {
                printf(NAME ": Too many arguments.\n");
                print_syntax();
                return EINVAL;
        }

        seat_name = argv[0];

        rc = dispcfg_open(dcfg_svc, NULL, NULL, &dispcfg);
        if (rc != EOK) {
                printf(NAME ": Failed connecting to display configuration "
                    "service: %s.\n", str_error(rc));
                return rc;
        }

        rc = seat_find_by_name(dispcfg, seat_name, &seat_id);
        if (rc != EOK) {
                printf(NAME ": Seat '%s' not found.\n", seat_name);
                dispcfg_close(dispcfg);
                return ENOENT;
        }

        rc = dispcfg_seat_delete(dispcfg, seat_id);
        if (rc != EOK) {
                printf(NAME ": Failed deleting seat '%s': %s\n", seat_name,
                    str_error(rc));
                dispcfg_close(dispcfg);
                return EIO;
        }

        dispcfg_close(dispcfg);
        return EOK;
}

/** List seats subcommand.
 *
 * @param dcfg_svc Display configuration service name
 * @param argc Number of arguments
 * @param argv Arguments
 * @return EOK on success or an erro code
 */
static errno_t list_seat(const char *dcfg_svc, int argc, char *argv[])
{
        dispcfg_t *dispcfg;
        dispcfg_seat_list_t *seat_list;
        dispcfg_seat_info_t *sinfo;
        table_t *table = NULL;
        size_t i;
        errno_t rc;

        if (argc > 0) {
                printf(NAME ": Too many arguments.\n");
                print_syntax();
                return EINVAL;
        }

        rc = dispcfg_open(dcfg_svc, NULL, NULL, &dispcfg);
        if (rc != EOK) {
                printf(NAME ": Failed connecting to display configuration "
                    "service: %s.\n", str_error(rc));
                return rc;
        }

        rc = dispcfg_get_seat_list(dispcfg, &seat_list);
        if (rc != EOK) {
                printf(NAME ": Failed getting seat list.\n");
                dispcfg_close(dispcfg);
                return rc;
        }

        rc = table_create(&table);
        if (rc != EOK) {
                printf("Memory allocation failed.\n");
                dispcfg_close(dispcfg);
                goto out;
        }

        table_header_row(table);
        table_printf(table, "Seat Name\n");

        for (i = 0; i < seat_list->nseats; i++) {
                rc = dispcfg_get_seat_info(dispcfg, seat_list->seats[i],
                    &sinfo);
                if (rc != EOK) {
                        printf("Failed getting properties of seat %zu.\n",
                            (size_t)seat_list->seats[i]);
                        continue;
                }

                table_printf(table, "%s\n", sinfo->name);

                dispcfg_free_seat_info(sinfo);
        }

        if (seat_list->nseats != 0) {
                rc = table_print_out(table, stdout);
                if (rc != EOK) {
                        printf("Error printing table.\n");
                        goto out;
                }
        }

        rc = EOK;
out:
        table_destroy(table);
        dispcfg_close(dispcfg);
        return rc;
}

/** Assign device subcommand.
 *
 * @param dcfg_svc Display configuration service name
 * @param argc Number of arguments
 * @param argv Arguments
 * @return EOK on success or an erro code
 */
static errno_t dev_assign(const char *dcfg_svc, int argc, char *argv[])
{
        dispcfg_t *dispcfg;
        char *dev_name;
        service_id_t svc_id;
        char *seat_name;
        sysarg_t seat_id;
        errno_t rc;

        if (argc < 2) {
                printf(NAME ": Missing arguments.\n");
                print_syntax();
                return EINVAL;
        }

        if (argc > 2) {
                printf(NAME ": Too many arguments.\n");
                print_syntax();
                return EINVAL;
        }

        dev_name = argv[0];
        seat_name = argv[1];

        rc = loc_service_get_id(dev_name, &svc_id, 0);
        if (rc != EOK) {
                printf(NAME ": Device service '%s' not found.\n",
                    dev_name);
                return ENOENT;
        }

        rc = dispcfg_open(dcfg_svc, NULL, NULL, &dispcfg);
        if (rc != EOK) {
                printf(NAME ": Failed connecting to display configuration "
                    "service: %s.\n", str_error(rc));
                return rc;
        }

        rc = seat_find_by_name(dispcfg, seat_name, &seat_id);
        if (rc != EOK) {
                printf(NAME ": Seat '%s' not found.\n", seat_name);
                dispcfg_close(dispcfg);
                return ENOENT;
        }

        rc = dispcfg_dev_assign(dispcfg, svc_id, seat_id);
        if (rc != EOK) {
                printf(NAME ": Failed assigning device '%s' to seat '%s': "
                    "%s\n", dev_name, seat_name, str_error(rc));
                dispcfg_close(dispcfg);
                return EIO;
        }

        dispcfg_close(dispcfg);
        return EOK;
}

/** Unassign device subcommand.
 *
 * @param dcfg_svc Display configuration service name
 * @param argc Number of arguments
 * @param argv Arguments
 * @return EOK on success or an erro code
 */
static errno_t dev_unassign(const char *dcfg_svc, int argc, char *argv[])
{
        dispcfg_t *dispcfg;
        char *dev_name;
        service_id_t svc_id;
        errno_t rc;

        if (argc < 1) {
                printf(NAME ": Missing arguments.\n");
                print_syntax();
                return EINVAL;
        }

        if (argc > 1) {
                printf(NAME ": Too many arguments.\n");
                print_syntax();
                return EINVAL;
        }

        dev_name = argv[0];

        rc = loc_service_get_id(dev_name, &svc_id, 0);
        if (rc != EOK) {
                printf(NAME ": Device service '%s' not found.\n",
                    dev_name);
                return ENOENT;
        }

        rc = dispcfg_open(dcfg_svc, NULL, NULL, &dispcfg);
        if (rc != EOK) {
                printf(NAME ": Failed connecting to display configuration "
                    "service: %s.\n", str_error(rc));
                return rc;
        }

        rc = dispcfg_dev_unassign(dispcfg, svc_id);
        if (rc != EOK) {
                printf(NAME ": Failed unassigning device '%s': %s\n",
                    dev_name, str_error(rc));
                dispcfg_close(dispcfg);
                return EIO;
        }

        dispcfg_close(dispcfg);
        return EOK;
}

/** List dev subcommand.
 *
 * @param dcfg_svc Display configuration service name
 * @param argc Number of arguments
 * @param argv Arguments
 * @return EOK on success or an erro code
 */
static errno_t list_dev(const char *dcfg_svc, int argc, char *argv[])
{
        dispcfg_t *dispcfg;
        char *seat_name;
        sysarg_t seat_id;
        dispcfg_dev_list_t *dev_list;
        size_t i;
        char *svc_name;
        table_t *table = NULL;
        errno_t rc;

        if (argc < 1) {
                printf(NAME ": Missing arguments.\n");
                print_syntax();
                return EINVAL;
        }

        if (argc > 1) {
                printf(NAME ": Too many arguments.\n");
                print_syntax();
                return EINVAL;
        }

        seat_name = argv[0];

        rc = dispcfg_open(dcfg_svc, NULL, NULL, &dispcfg);
        if (rc != EOK) {
                printf(NAME ": Failed connecting to display configuration "
                    "service: %s.\n", str_error(rc));
                return rc;
        }

        rc = seat_find_by_name(dispcfg, seat_name, &seat_id);
        if (rc != EOK) {
                printf(NAME ": Seat '%s' not found.\n", seat_name);
                dispcfg_close(dispcfg);
                return ENOENT;
        }

        rc = dispcfg_get_asgn_dev_list(dispcfg, seat_id, &dev_list);
        if (rc != EOK) {
                printf(NAME ": Failed getting seat list.\n");
                dispcfg_close(dispcfg);
                return rc;
        }

        rc = table_create(&table);
        if (rc != EOK) {
                printf("Memory allocation failed.\n");
                dispcfg_free_dev_list(dev_list);
                dispcfg_close(dispcfg);
                return rc;
        }

        table_header_row(table);
        table_printf(table, "Device Name\n");

        for (i = 0; i < dev_list->ndevs; i++) {
                rc = loc_service_get_name(dev_list->devs[i], &svc_name);
                if (rc != EOK) {
                        printf("Failed getting name of service %zu\n",
                            (size_t)dev_list->devs[i]);
                        continue;
                }

                table_printf(table, "%s\n", svc_name);
                free(svc_name);
        }

        if (dev_list->ndevs != 0) {
                rc = table_print_out(table, stdout);
                if (rc != EOK) {
                        printf("Error printing table.\n");
                        table_destroy(table);
                        dispcfg_free_dev_list(dev_list);
                        dispcfg_close(dispcfg);
                        return rc;
                }
        }

        dispcfg_close(dispcfg);
        return EOK;
}

int main(int argc, char *argv[])
{
        const char *dispcfg_svc = DISPCFG_DEFAULT;
        errno_t rc;

        if (argc < 2 || str_cmp(argv[1], "-h") == 0) {
                print_syntax();
                return 0;
        }

        if (str_cmp(argv[1], "list-seat") == 0) {
                rc = list_seat(dispcfg_svc, argc - 2, argv + 2);
                if (rc != EOK)
                        return 1;
        } else if (str_cmp(argv[1], "create-seat") == 0) {
                rc = create_seat(dispcfg_svc, argc - 2, argv + 2);
                if (rc != EOK)
                        return 1;
        } else if (str_cmp(argv[1], "delete-seat") == 0) {
                rc = delete_seat(dispcfg_svc, argc - 2, argv + 2);
                if (rc != EOK)
                        return 1;
        } else if (str_cmp(argv[1], "assign-dev") == 0) {
                rc = dev_assign(dispcfg_svc, argc - 2, argv + 2);
                if (rc != EOK)
                        return 1;
        } else if (str_cmp(argv[1], "unassign-dev") == 0) {
                rc = dev_unassign(dispcfg_svc, argc - 2, argv + 2);
                if (rc != EOK)
                        return 1;
        } else if (str_cmp(argv[1], "list-dev") == 0) {
                rc = list_dev(dispcfg_svc, argc - 2, argv + 2);
                if (rc != EOK)
                        return 1;
        } else {
                printf(NAME ": Unknown command '%s'.\n", argv[1]);
                print_syntax();
                return 1;
        }

        return 0;
}

/** @}
 */