1 // SPDX-License-Identifier: GPL-2.0 OR MIT
2 /*
3 * Copyright (C) 2015-2020 Jason A. Donenfeld <Jason@zx2c4.com>. All Rights Reserved.
4 */
5
6 #include <errno.h>
7 #include <stdio.h>
8
9 #include "curve25519.h"
10 #include "encoding.h"
11 #include "subcommands.h"
12 #include "ctype.h"
13
pubkey_main(int argc,const char * argv[])14 int pubkey_main(int argc, const char *argv[])
15 {
16 uint8_t key[WG_KEY_LEN] __attribute__((aligned(sizeof(uintptr_t))));
17 char base64[WG_KEY_LEN_BASE64];
18 int trailing_char;
19
20 if (argc != 1) {
21 fprintf(stderr, "Usage: %s %s\n", PROG_NAME, argv[0]);
22 return 1;
23 }
24
25 if (fread(base64, 1, sizeof(base64) - 1, stdin) != sizeof(base64) - 1) {
26 errno = EINVAL;
27 fprintf(stderr, "%s: Key is not the correct length or format\n", PROG_NAME);
28 return 1;
29 }
30 base64[WG_KEY_LEN_BASE64 - 1] = '\0';
31
32 for (;;) {
33 trailing_char = getc(stdin);
34 if (!trailing_char || char_is_space(trailing_char))
35 continue;
36 if (trailing_char == EOF)
37 break;
38 fprintf(stderr, "%s: Trailing characters found after key\n", PROG_NAME);
39 return 1;
40 }
41
42 if (!key_from_base64(key, base64)) {
43 fprintf(stderr, "%s: Key is not the correct length or format\n", PROG_NAME);
44 return 1;
45 }
46 curve25519_generate_public(key, key);
47 key_to_base64(base64, key);
48 puts(base64);
49 return 0;
50 }
51