Client: Fix crashes happening on navigation in too large inventory (string).

[plomrogue] / src / client / control.c

/* src/client/control.c */

#include "control.h"
#include <stdint.h> /* uint8_t, uint16_t, uint32_t, UINT32_MAX */
#include <stdio.h> /* sprintf() */
#include <string.h> /* strlen() */
#include "../common/rexit.h" /* exit_err() */
#include "interface_conf.h" /* reload_interface_conf(), save_interface_conf() */
#include "io.h" /* send() */
#include "keybindings.h" /* get_command_to_keycode(), get_keycode_to_command(),
                          * mod_selected_keyb(), move_keyb_selection()
                          */
#include "map.h" /* for map_scroll(), map_center() */
#include "wincontrol.h" /* shift_active_win(), resize_active_win(),
                         * toggle_win_size_type(), toggle_window(),
                         * cycle_active_win(), scroll_v_screen(),
                         * toggle_linebreak_type(), toggle_winconfig()
                         */
#include "windows.h" /* get_win_by_id() */
#include "world.h" /* for global world */



/* Move world.inventory_sel up ("dir"="u") or down (else) as far as possible. */
static void nav_inventory(char dir);

/* If "command"'s .dsc_short fits "match", apply "f" with provided char
 * arguments and return 1; else, return 0.
 */
static uint8_t try_0args(struct Command * command, char * match, void (* f) ());
static uint8_t try_1args(struct Command * command, char * match,
                         void (* f) (char), char c);
static uint8_t try_2args(struct Command * command, char * match,
                         void (* f) (char, char), char c1, char c2);

/* Try if "command" matches a hard-coded list of client-only commands and, if
 * successful, execute the match and return 1. Else, return 0.
 */
static uint8_t try_client_commands(struct Command * command);

/* Try out "command" as one for server messaging; sending is .server_msg,
 * followed by either a string representing "command"'s .arg, or, if .arg is
 * 'i', world.player_inventory_select. Return 1 on success, 0 on failure.
 */
static uint8_t try_server_commands(struct Command * command);



static void nav_inventory(char dir)
{
    if ('u' == dir)
    {
        world.player_inventory_select = world.player_inventory_select
                                        - (world.player_inventory_select > 0);
        return;
    }
    uint8_t n_elems = 0;
    uint32_t i;
    char * err = "Inventory string is too large.";
    exit_err(UINT32_MAX <= strlen(world.player_inventory), err);
    for (i = 0; '\0' != world.player_inventory[i]; i++)
    {
        n_elems = n_elems + ('\n' == world.player_inventory[i]);
    }
    world.player_inventory_select = world.player_inventory_select
                                    + (world.player_inventory_select < n_elems);
}



static uint8_t try_0args(struct Command * command, char * match, void (* f) ())
{
    if (!strcmp(command->dsc_short, match))
    {
        f();
        return 1;
    }
    return 0;
}

static uint8_t try_1args(struct Command * command, char * match,
                             void (* f) (char), char c)
{
    if (!strcmp(command->dsc_short, match))
    {
        f(c);
        return 1;
    }
    return 0;
}



static uint8_t try_2args(struct Command * command, char * match,
                             void (* f) (char, char), char c1, char c2)
{
    if (!strcmp(command->dsc_short, match))
    {
        f(c1, c2);
        return 1;
    }
    return 0;
}



static uint8_t try_client_commands(struct Command * command)
{
    return (   try_0args(command, "map_c", map_center)
            || try_1args(command, "map_u", map_scroll, '8')
            || try_1args(command, "map_d", map_scroll, '2')
            || try_1args(command, "map_r", map_scroll, '6')
            || try_1args(command, "map_l", map_scroll, '4')
            || try_1args(command, "inv_u", nav_inventory, 'u')
            || try_1args(command, "inv_d", nav_inventory, 'd')
            || try_1args(command, "cyc_win_f", cycle_active_win, 'f')
            || try_1args(command, "cyc_win_b", cycle_active_win, 'b')
            || try_1args(command, "scrl_r", scroll_v_screen, '+')
            || try_1args(command, "scrl_l", scroll_v_screen, '-')
            || try_1args(command, "to_a_keywin", toggle_window, 'k')
            || try_1args(command, "to_g_keywin", toggle_window, '0')
            || try_1args(command, "to_wg_keywin", toggle_window, '1')
            || try_1args(command, "to_wk_keywin", toggle_window, '2')
            || try_1args(command, "to_mapwin", toggle_window, 'm')
            || try_1args(command, "to_infowin", toggle_window, 'i')
            || try_1args(command, "to_inv", toggle_window, 'c')
            || try_1args(command, "to_logwin", toggle_window, 'l')
            || try_0args(command, "winconf", toggle_winconfig)
            || try_1args(command, "grow_h", resize_active_win, '*')
            || try_1args(command, "shri_h", resize_active_win, '_')
            || try_1args(command, "grow_v", resize_active_win, '+')
            || try_1args(command, "shri_v", resize_active_win, '-')
            || try_0args(command, "to_break", toggle_linebreak_type)
            || try_1args(command, "to_height_t", toggle_win_size_type, 'y')
            || try_1args(command, "to_width_t", toggle_win_size_type, 'x')
            || try_1args(command, "shift_f", shift_active_win, 'f')
            || try_1args(command, "shift_b", shift_active_win, 'b')
            || try_0args(command, "reload_conf", reload_interface_conf)
            || try_0args(command, "save_conf", save_interface_conf)
            || try_1args(command, "g_keys_m", mod_selected_keyb, 'G')
            || try_2args(command, "g_keys_u", move_keyb_selection, 'G', 'u')
            || try_2args(command, "g_keys_d", move_keyb_selection, 'G', 'd')
            || try_1args(command, "w_keys_m", mod_selected_keyb, 'w')
            || try_2args(command, "w_keys_u", move_keyb_selection, 'w', 'u')
            || try_2args(command, "w_keys_d", move_keyb_selection, 'w', 'd')
            || try_1args(command, "wg_keys_m", mod_selected_keyb, 'g')
            || try_2args(command, "wg_keys_u", move_keyb_selection, 'g', 'u')
            || try_2args(command, "wg_keys_d", move_keyb_selection, 'g', 'd')
            || try_1args(command, "wk_keys_m", mod_selected_keyb, 'k')
            || try_2args(command, "wk_keys_u", move_keyb_selection, 'k', 'u')
            || try_2args(command, "wk_keys_d", move_keyb_selection, 'k', 'd'));
}



static uint8_t try_server_commands(struct Command * command)
{
    if (command->server_msg)
    {
        uint8_t arg = (uint8_t) command->arg;
        if ('i' == arg)
        {
            arg = world.player_inventory_select;
        }
        uint8_t command_size = strlen(command->server_msg);
        uint8_t arg_size = 3;
        char msg[command_size + 1 + arg_size + 1];
        sprintf(msg, "%s %d", command->server_msg, arg);
        send(msg);
        return 1;
    }
    return 0;
}



extern uint8_t try_key(uint16_t key)
{
    struct Command * command = get_command_to_keycode(&world.kb_global, key);
    if (!command && world.winDB.active)
    {
        struct Win * w = get_win_by_id(world.winDB.active);
        if      (0 == w->view)
        {
            command = get_command_to_keycode(&w->kb, key);
        }
        else if (1 == w->view)
        {
            command = get_command_to_keycode(&world.kb_wingeom, key);
        }
        else if (2 == w->view)
        {
            command = get_command_to_keycode(&world.kb_winkeys, key);
        }
    }
    if (command)
    {
        if      (try_server_commands(command))
        {
            return 1;
        }
        else if (try_client_commands(command))
        {
            return 1;
        }
        else if (!strcmp("quit", command->dsc_short))
        {
            return 2;
        }
    }
    return 0;
}