/*
|
* Copyright (C) 2017 The Android Open Source Project
|
*
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
* you may not use this file except in compliance with the License.
|
* You may obtain a copy of the License at
|
*
|
* http://www.apache.org/licenses/LICENSE-2.0
|
*
|
* Unless required by applicable law or agreed to in writing, software
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
* See the License for the specific language governing permissions and
|
* limitations under the License.
|
*/
|
|
#include <stdio.h>
|
#include <stdlib.h>
|
#include <string.h>
|
#include <unistd.h>
|
|
|
extern void handle_usage_help(FILE *out_fp, const char *prog_name);
|
extern int handle_command_help(int argc, char *argv[], int arg_start);
|
extern void handle_usage_dump(FILE *out_fp, const char *prog_name);
|
extern int handle_command_dump(int argc, char *argv[], int arg_start);
|
extern void handle_usage_create(FILE *out_fp, const char *prog_name);
|
extern int handle_command_create(int argc, char *argv[], int arg_start);
|
extern void handle_usage_cfg_create(FILE *out_fp, const char *prog_name);
|
extern int handle_command_cfg_create(int argc, char *argv[], int arg_start);
|
|
|
struct command_info {
|
const char *command;
|
void (*usage)(FILE *out_fp, const char *prog_name);
|
int (*handler)(int argc, char *argv[], int arg_start);
|
};
|
|
static const struct command_info command_infos[] = {
|
{ "help", handle_usage_help, handle_command_help },
|
{ "dump", handle_usage_dump, handle_command_dump },
|
{ "create", handle_usage_create, handle_command_create },
|
{ "cfg_create", handle_usage_cfg_create, handle_command_cfg_create },
|
{ NULL, NULL, NULL }
|
};
|
|
static const struct command_info *search_command(const char *command) {
|
const struct command_info *info;
|
for (info = command_infos; info->command != NULL; info++) {
|
if (strcmp(command, info->command) == 0) {
|
break;
|
}
|
}
|
if (info->command == NULL) {
|
fprintf(stderr, "Unknown command: %s\n", command);
|
return NULL;
|
}
|
return info;
|
}
|
|
static void print_all_commands(FILE *out_fp) {
|
const struct command_info *info;
|
for (info = command_infos; info->command != NULL; info++) {
|
if (info != command_infos) {
|
fprintf(out_fp, ", ");
|
}
|
fprintf(out_fp, "%s", info->command);
|
}
|
fprintf(out_fp, "\n");
|
}
|
|
static void output_all_usage(FILE *out_fp, const char *prog_name) {
|
fprintf(out_fp, " %s <command>\n\n", prog_name);
|
fprintf(out_fp, " commands:\n ");
|
print_all_commands(out_fp);
|
fprintf(out_fp, "\n");
|
|
const struct command_info *info;
|
for (info = command_infos; info->command != NULL; info++) {
|
info->usage(out_fp, prog_name);
|
fprintf(out_fp, "\n");
|
}
|
}
|
|
void handle_usage_help(FILE *out_fp, const char *prog_name) {
|
fprintf(out_fp, " %s help all\n", prog_name);
|
fprintf(out_fp, " %s help <command>\n\n", prog_name);
|
fprintf(out_fp, " commands:\n ");
|
print_all_commands(out_fp);
|
}
|
|
int handle_command_help(int argc, char *argv[], int arg_start) {
|
const char *prog_name = argv[0];
|
|
if (argc - arg_start < 1) {
|
handle_usage_help(stderr, prog_name);
|
return 1;
|
}
|
|
if (strcmp(argv[arg_start], "all") == 0) {
|
output_all_usage(stdout, prog_name);
|
return 0;
|
}
|
|
const struct command_info *info = search_command(argv[arg_start]);
|
if (info == NULL) {
|
handle_usage_help(stderr, prog_name);
|
return 1;
|
}
|
|
info->usage(stdout, prog_name);
|
|
return 0;
|
}
|
|
int main(int argc, char *argv[]) {
|
if (argc <= 1) {
|
output_all_usage(stderr, argv[0]);
|
return 1;
|
}
|
|
const char *command = argv[1];
|
const struct command_info *info = search_command(command);
|
if (info == NULL) {
|
return 1;
|
}
|
|
/* skip 2 arguments, argv[0] and argv[1] */
|
return info->handler(argc, argv, 2);
|
}
|
