（二）wireguard-tools源码解读之pubkey_main

---

前言

基于私钥生成公钥的模块

---

一、动手操作

1、生成私钥
wg genkey > privatekey
GOhFlzedvzYljeCIlbYLHiEVJqcfn4Oy1wiwDKE3YW0=
2、生成公钥
wg pubkey
输入：GOhFlzedvzYljeCIlbYLHiEVJqcfn4Oy1wiwDKE3YW0=
按CTRL+D，把私钥输入进去。
或者cat privatekey | wg pubkey
或者echo GOhFlzedvzYljeCIlbYLHiEVJqcfn4Oy1wiwDKE3YW0= | wg pubkey
等等方式。

二、代码解读

int pubkey_main(int argc, const char *argv[])
{
	/*
	unsigned long在64位编译器上为8个字节
	在32位编译器上为4个字节。
	*/
	uint8_t key[WG_KEY_LEN] __attribute__((aligned(sizeof(uintptr_t))));
	char base64[WG_KEY_LEN_BASE64];
	int trailing_char;

	if (argc != 1) {
		fprintf(stderr, "Usage: %s %s\n", PROG_NAME, argv[0]);
		return 1;
	}
	/*
	从输入中，读入私钥，并进行检查
	（1）检查私钥的长度
	（2）检查私钥输入流结尾标识：EOF，End of file
	 如果手工输入EOF：在windows下输入CTRL+Z； Linux/Unix/Mac操作系统输入CTRL+D。
	（3）检查base64格式
	*/
	if (fread(base64, 1, sizeof(base64) - 1, stdin) != sizeof(base64) - 1) {
		errno = EINVAL;
		fprintf(stderr, "%s: Key is not the correct length or format\n", PROG_NAME);
		return 1;
	}
	base64[WG_KEY_LEN_BASE64 - 1] = '\0';

	for (;;) {
		trailing_char = getc(stdin);
		if (!trailing_char || char_is_space(trailing_char))
			continue;
		if (trailing_char == EOF)
			break;
		fprintf(stderr, "%s: Trailing characters found after key\n", PROG_NAME);
		return 1;
	}

	if (!key_from_base64(key, base64)) {
		fprintf(stderr, "%s: Key is not the correct length or format\n", PROG_NAME);
		return 1;
	}
	/*
	调用curve25519算法，基于私钥生成公钥
	*/
	curve25519_generate_public(key, key);
	key_to_base64(base64, key);
	puts(base64);
	return 0;
}