From 901d3533ec3285b58960d0af0937c8e05692ebe0 Mon Sep 17 00:00:00 2001 From: Mattias Andrée Date: Fri, 23 Jun 2023 10:40:22 +0200 Subject: Add argument checks to key2root-addkey and key2root-rmkey MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Mattias Andrée --- key2root-rmkey.c | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) (limited to 'key2root-rmkey.c') diff --git a/key2root-rmkey.c b/key2root-rmkey.c index 1050977..f2ab34d 100644 --- a/key2root-rmkey.c +++ b/key2root-rmkey.c @@ -19,6 +19,9 @@ usage(void) int main(int argc, char *argv[]) { + const char *user; + int i, failed = 0; + ARGBEGIN { default: usage(); @@ -27,5 +30,23 @@ main(int argc, char *argv[]) if (argc < 2) usage(); + user = *argv++; + argc--; + + if (!user[0] || user[0] == '.' || strchr(user, '/') || strchr(user, '~')) { + fprintf(stderr, "%s: bad user name specified: %s\n", argv0, user); + failed = 1; + } + for (i = 0; i < argc; i++) { + if (argv[i][strcspn(argv[i], " \t\f\n\r\v")]) { + fprintf(stderr, "%s: bad key name specified: %s, includes whitespace\n", argv0, argv[i]); + failed = 1; + } + } + if (failed) + return 1; + + /* TODO */ + return 0; } -- cgit v1.2.3-70-g09d2