mirror of
https://git.kernel.org/pub/scm/network/wireless/iwd.git
synced 2024-12-22 21:22:37 +01:00
15ade259e6
The code would assume that a command with argc >= 3 would always be composed of <family name> <entity> <command> and thus "known-networks forget <ssid>" could not be executed. Instead of checking argc >= 3 check whether command->entity is non-NULL.
525 lines
10 KiB
C
525 lines
10 KiB
C
/*
|
|
*
|
|
* Wireless daemon for Linux
|
|
*
|
|
* Copyright (C) 2017 Intel Corporation. All rights reserved.
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library 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
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include <config.h>
|
|
#endif
|
|
|
|
#include <stdio.h>
|
|
#include <ell/ell.h>
|
|
#include <readline/readline.h>
|
|
|
|
#include "command.h"
|
|
#include "display.h"
|
|
|
|
static struct l_queue *command_families;
|
|
|
|
static enum cmd_status cmd_version(const char *entity,
|
|
char **argv, int argc)
|
|
{
|
|
display("IWD version %s\n", VERSION);
|
|
|
|
return CMD_STATUS_OK;
|
|
}
|
|
|
|
static enum cmd_status cmd_quit(const char *entity,
|
|
char **argv, int argc)
|
|
{
|
|
display_quit();
|
|
|
|
l_main_quit();
|
|
|
|
return CMD_STATUS_OK;
|
|
}
|
|
|
|
static const struct command command_list[] = {
|
|
{ NULL, "version", NULL, cmd_version, "Display version" },
|
|
{ NULL, "quit", NULL, cmd_quit, "Quit program" },
|
|
{ NULL, "exit", NULL, cmd_quit },
|
|
{ NULL, "help" },
|
|
{ }
|
|
};
|
|
|
|
static char *cmd_generator(const char *text, int state)
|
|
{
|
|
static const struct l_queue_entry *entry;
|
|
static size_t index;
|
|
static size_t len;
|
|
const char *cmd;
|
|
|
|
if (!state) {
|
|
len = strlen(text);
|
|
index = 0;
|
|
entry = l_queue_get_entries(command_families);
|
|
}
|
|
|
|
while (entry) {
|
|
const struct command_family *family = entry->data;
|
|
|
|
entry = entry->next;
|
|
|
|
if (strncmp(family->name, text, len))
|
|
continue;
|
|
|
|
return l_strdup(family->name);
|
|
}
|
|
|
|
while ((cmd = command_list[index].cmd)) {
|
|
index++;
|
|
|
|
if (strncmp(cmd, text, len))
|
|
continue;
|
|
|
|
return l_strdup(cmd);
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
static bool cmd_completion_cmd_has_arg(const char *cmd,
|
|
const struct command_family *family)
|
|
{
|
|
size_t i;
|
|
char **matches = NULL;
|
|
bool status;
|
|
|
|
for (i = 0; family->command_list[i].cmd; i++) {
|
|
if (strcmp(family->command_list[i].cmd, cmd))
|
|
continue;
|
|
|
|
return family->command_list[i].arg ? true : false;
|
|
}
|
|
|
|
if (!family->family_arg_completion)
|
|
return false;
|
|
|
|
matches = rl_completion_matches(cmd, family->family_arg_completion);
|
|
if (!matches)
|
|
return false;
|
|
|
|
status = false;
|
|
|
|
for (i = 0; matches[i]; i++) {
|
|
if (strcmp(matches[i], cmd))
|
|
continue;
|
|
|
|
status = true;
|
|
|
|
break;
|
|
}
|
|
|
|
l_strfreev(matches);
|
|
|
|
return status;
|
|
}
|
|
|
|
static bool find_next_token(int *i, const char *token, int token_len)
|
|
{
|
|
char *line = rl_line_buffer;
|
|
|
|
while (*i && line[*i] == ' ')
|
|
(*i)--;
|
|
|
|
while (*i && line[*i] != ' ')
|
|
(*i)--;
|
|
|
|
return !strncmp(line + (line[*i] == ' ' ? *i + 1 : *i),
|
|
token, token_len);
|
|
}
|
|
|
|
bool command_line_find_token(const char *token, uint8_t num_to_inspect)
|
|
{
|
|
int i = rl_point - 1;
|
|
int len = strlen(token);
|
|
|
|
if (!len)
|
|
return false;
|
|
|
|
while (i && num_to_inspect) {
|
|
if (find_next_token(&i, token, len))
|
|
return true;
|
|
|
|
num_to_inspect--;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
static char **cmd_completion_match_entity_cmd(const char *cmd, const char *text,
|
|
const struct command *cmd_list)
|
|
{
|
|
char **matches = NULL;
|
|
size_t i;
|
|
|
|
for (i = 0; cmd_list[i].cmd; i++) {
|
|
if (strcmp(cmd_list[i].cmd, cmd))
|
|
continue;
|
|
|
|
if (!cmd_list[i].completion)
|
|
break;
|
|
|
|
matches = rl_completion_matches(text, cmd_list[i].completion);
|
|
|
|
break;
|
|
}
|
|
|
|
return matches;
|
|
}
|
|
|
|
static char **cmd_completion_match_family_cmd(const char *cmd_family,
|
|
char *args, const char *text,
|
|
bool ends_with_space)
|
|
{
|
|
const struct l_queue_entry *entry;
|
|
const char *arg1;
|
|
const char *arg2;
|
|
const char *arg3;
|
|
char **matches = NULL;
|
|
|
|
for (entry = l_queue_get_entries(command_families); entry;
|
|
entry = entry->next) {
|
|
const struct command_family *family = entry->data;
|
|
|
|
if (strcmp(family->name, cmd_family))
|
|
continue;
|
|
|
|
arg1 = strtok_r(NULL, " ", &args);
|
|
if (!arg1) {
|
|
if (!family->family_arg_completion)
|
|
break;
|
|
|
|
matches = rl_completion_matches(text,
|
|
family->family_arg_completion);
|
|
|
|
break;
|
|
}
|
|
|
|
arg2 = strtok_r(NULL, " ", &args);
|
|
if (!arg2 && !ends_with_space) {
|
|
if (!family->family_arg_completion)
|
|
break;
|
|
|
|
matches = rl_completion_matches(text,
|
|
family->family_arg_completion);
|
|
break;
|
|
} else if (!arg2 && ends_with_space) {
|
|
if (!cmd_completion_cmd_has_arg(arg1, family))
|
|
break;
|
|
|
|
if (!family->entity_arg_completion)
|
|
break;
|
|
|
|
matches = rl_completion_matches(text,
|
|
family->entity_arg_completion);
|
|
break;
|
|
}
|
|
|
|
arg3 = strtok_r(NULL, " ", &args);
|
|
if (!arg3 && !ends_with_space) {
|
|
if (!family->entity_arg_completion)
|
|
break;
|
|
|
|
matches = rl_completion_matches(text,
|
|
family->entity_arg_completion);
|
|
break;
|
|
}
|
|
|
|
if (family->set_default_entity)
|
|
family->set_default_entity(arg1);
|
|
|
|
matches = cmd_completion_match_entity_cmd(arg2, text,
|
|
family->command_list);
|
|
|
|
break;
|
|
}
|
|
|
|
return matches;
|
|
}
|
|
|
|
char **command_completion(const char *text, int start, int end)
|
|
{
|
|
char **matches = NULL;
|
|
const char *family;
|
|
char *args = NULL;
|
|
char *prompt = NULL;
|
|
bool ends_with_space = false;
|
|
|
|
if (display_agent_is_active()) {
|
|
rl_attempted_completion_over = 1;
|
|
return NULL;
|
|
}
|
|
|
|
if (!start) {
|
|
matches = rl_completion_matches(text, cmd_generator);
|
|
|
|
goto done;
|
|
}
|
|
|
|
prompt = rl_copy_text(0, rl_end);
|
|
|
|
family = strtok_r(prompt, " ", &args);
|
|
if (!family)
|
|
goto done;
|
|
|
|
if (args) {
|
|
int len = strlen(args);
|
|
|
|
if (len > 0 && args[len - 1] == ' ')
|
|
ends_with_space = true;
|
|
}
|
|
|
|
matches = cmd_completion_match_family_cmd(family, args, text,
|
|
ends_with_space);
|
|
|
|
done:
|
|
l_free(prompt);
|
|
|
|
if (!matches)
|
|
rl_attempted_completion_over = 1;
|
|
|
|
return matches;
|
|
}
|
|
|
|
char *command_entity_arg_completion(const char *text, int state,
|
|
const struct command *command_list)
|
|
{
|
|
static size_t index;
|
|
static size_t len;
|
|
const char *cmd;
|
|
|
|
if (!state) {
|
|
index = 0;
|
|
len = strlen(text);
|
|
}
|
|
|
|
while ((cmd = command_list[index].cmd)) {
|
|
if (!command_list[index++].entity)
|
|
continue;
|
|
|
|
if (strncmp(cmd, text, len))
|
|
continue;
|
|
|
|
return l_strdup(cmd);
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
static void execute_cmd(const char *family, const char *entity,
|
|
const struct command *cmd,
|
|
char **argv, int argc)
|
|
{
|
|
enum cmd_status status;
|
|
|
|
display_refresh_set_cmd(family, entity, cmd, argv, argc);
|
|
|
|
status = cmd->function(entity, argv, argc);
|
|
|
|
if (status != CMD_STATUS_OK)
|
|
goto error;
|
|
|
|
if (cmd->refreshable)
|
|
display_refresh_timeout_set();
|
|
|
|
return;
|
|
|
|
error:
|
|
switch (status) {
|
|
case CMD_STATUS_INVALID_ARGS:
|
|
display("Invalid command. Use the following pattern:\n");
|
|
display_command_line(family, cmd);
|
|
break;
|
|
|
|
case CMD_STATUS_INVALID_VALUE:
|
|
break;
|
|
|
|
case CMD_STATUS_UNSUPPORTED:
|
|
display_refresh_reset();
|
|
|
|
display("Unsupported command\n");
|
|
break;
|
|
|
|
case CMD_STATUS_FAILED:
|
|
l_main_quit();
|
|
break;
|
|
|
|
default:
|
|
l_error("Unknown command status.");
|
|
}
|
|
}
|
|
|
|
static bool match_cmd(const char *family, const char *param,
|
|
char **argv, int argc,
|
|
const struct command *command_list)
|
|
{
|
|
size_t i;
|
|
|
|
for (i = 0; command_list[i].cmd; i++) {
|
|
const char *entity;
|
|
const char *cmd;
|
|
int offset;
|
|
|
|
if (command_list[i].entity) {
|
|
if (argc < 1)
|
|
continue;
|
|
|
|
entity = param;
|
|
cmd = argv[0];
|
|
offset = 1;
|
|
} else {
|
|
entity = NULL;
|
|
cmd = param;
|
|
offset = 0;
|
|
}
|
|
|
|
if (strcmp(command_list[i].cmd, cmd))
|
|
continue;
|
|
|
|
if (!command_list[i].function)
|
|
return false;
|
|
|
|
execute_cmd(family, entity, &command_list[i],
|
|
argv + offset, argc - offset);
|
|
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
static bool match_cmd_family(char **argv, int argc)
|
|
{
|
|
const struct l_queue_entry *entry;
|
|
|
|
if (argc < 2)
|
|
return false;
|
|
|
|
for (entry = l_queue_get_entries(command_families); entry;
|
|
entry = entry->next) {
|
|
const struct command_family *family = entry->data;
|
|
|
|
if (strcmp(family->name, argv[0]))
|
|
continue;
|
|
|
|
return match_cmd(family->name, argv[1], argv + 2, argc - 2,
|
|
family->command_list);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
static void list_commands(const char *command_family,
|
|
const struct command *cmd_list)
|
|
{
|
|
size_t i;
|
|
|
|
for (i = 0; cmd_list[i].cmd; i++) {
|
|
if (!cmd_list[i].desc)
|
|
continue;
|
|
|
|
display_command_line(command_family, &cmd_list[i]);
|
|
}
|
|
}
|
|
|
|
static void list_cmd_families(void)
|
|
{
|
|
const struct l_queue_entry *entry;
|
|
|
|
for (entry = l_queue_get_entries(command_families); entry;
|
|
entry = entry->next) {
|
|
const struct command_family *family = entry->data;
|
|
|
|
display("\n%s:\n", family->caption);
|
|
list_commands(family->name, family->command_list);
|
|
}
|
|
}
|
|
|
|
void command_process_prompt(char **argv, int argc)
|
|
{
|
|
if (argc == 0)
|
|
return;
|
|
|
|
if (match_cmd_family(argv, argc))
|
|
return;
|
|
|
|
display_refresh_reset();
|
|
|
|
if (match_cmd(NULL, argv[0], argv + 1, argc - 1, command_list))
|
|
return;
|
|
|
|
if (strcmp(argv[0], "help")) {
|
|
display("Invalid command\n");
|
|
return;
|
|
}
|
|
|
|
display_table_header("Available commands", MARGIN "%-*s%-*s",
|
|
50, "Commands", 28, "Description");
|
|
|
|
list_cmd_families();
|
|
|
|
display("\nMiscellaneous:\n");
|
|
|
|
list_commands(NULL, command_list);
|
|
}
|
|
|
|
void command_family_register(const struct command_family *family)
|
|
{
|
|
l_queue_push_tail(command_families, (void *) family);
|
|
}
|
|
|
|
void command_family_unregister(const struct command_family *family)
|
|
{
|
|
l_queue_remove(command_families, (void *) family);
|
|
}
|
|
|
|
extern struct command_family_desc __start___command[];
|
|
extern struct command_family_desc __stop___command[];
|
|
|
|
void command_init(void)
|
|
{
|
|
struct command_family_desc *desc;
|
|
|
|
command_families = l_queue_new();
|
|
|
|
for (desc = __start___command; desc < __stop___command; desc++) {
|
|
if (!desc->init)
|
|
continue;
|
|
|
|
desc->init();
|
|
}
|
|
}
|
|
|
|
void command_exit(void)
|
|
{
|
|
struct command_family_desc *desc;
|
|
|
|
for (desc = __start___command; desc < __stop___command; desc++) {
|
|
if (!desc->exit)
|
|
continue;
|
|
|
|
desc->exit();
|
|
}
|
|
|
|
l_queue_destroy(command_families, NULL);
|
|
command_families = NULL;
|
|
}
|