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 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