123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326 |
- /*
- * px5g - Embedded x509 key and certificate generator based on PolarSSL
- *
- * Copyright (C) 2009 Steven Barth <steven@midlink.org>
- * Copyright (C) 2014 Felix Fietkau <nbd@nbd.name>
- *
- * This library is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Lesser General Public
- * License, version 2.1 as published by the Free Software Foundation.
- *
- * This library is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * Lesser General Public License for more details.
- *
- * You should have received a copy of the GNU Lesser General Public
- * License along with this library; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
- * MA 02110-1301 USA
- */
- #include <sys/types.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include <string.h>
- #include <time.h>
- #include <limits.h>
- #include <unistd.h>
- #include <fcntl.h>
- #include <stdbool.h>
- #include <mbedtls/bignum.h>
- #include <mbedtls/x509_crt.h>
- #include <mbedtls/ecp.h>
- #include <mbedtls/rsa.h>
- #include <mbedtls/pk.h>
- #define PX5G_VERSION "0.2"
- #define PX5G_COPY "Copyright (c) 2009 Steven Barth <steven@midlink.org>"
- #define PX5G_LICENSE "Licensed under the GNU Lesser General Public License v2.1"
- static int urandom_fd;
- static char buf[16384];
- static int _urandom(void *ctx, unsigned char *out, size_t len)
- {
- read(urandom_fd, out, len);
- return 0;
- }
- static void write_file(const char *path, int len, bool pem)
- {
- FILE *f = stdout;
- const char *buf_start = buf;
- if (!pem)
- buf_start += sizeof(buf) - len;
- if (!len) {
- fprintf(stderr, "No data to write\n");
- exit(1);
- }
- if (!f) {
- fprintf(stderr, "error: I/O error\n");
- exit(1);
- }
- if (path)
- f = fopen(path, "w");
- fwrite(buf_start, 1, len, f);
- fclose(f);
- }
- static mbedtls_ecp_group_id ecp_curve(const char *name)
- {
- const mbedtls_ecp_curve_info *curve_info;
- if (!strcmp(name, "P-256"))
- return MBEDTLS_ECP_DP_SECP256R1;
- else if (!strcmp(name, "P-384"))
- return MBEDTLS_ECP_DP_SECP384R1;
- else if (!strcmp(name, "P-521"))
- return MBEDTLS_ECP_DP_SECP521R1;
- curve_info = mbedtls_ecp_curve_info_from_name(name);
- if (curve_info == NULL)
- return MBEDTLS_ECP_DP_NONE;
- else
- return curve_info->grp_id;
- }
- static void write_key(mbedtls_pk_context *key, const char *path, bool pem)
- {
- int len = 0;
- if (pem) {
- if (mbedtls_pk_write_key_pem(key, (void *) buf, sizeof(buf)) == 0)
- len = strlen(buf);
- } else {
- len = mbedtls_pk_write_key_der(key, (void *) buf, sizeof(buf));
- if (len < 0)
- len = 0;
- }
- write_file(path, len, pem);
- }
- static void gen_key(mbedtls_pk_context *key, bool rsa, int ksize, int exp,
- mbedtls_ecp_group_id curve, bool pem)
- {
- mbedtls_pk_init(key);
- if (rsa) {
- fprintf(stderr, "Generating RSA private key, %i bit long modulus\n", ksize);
- mbedtls_pk_setup(key, mbedtls_pk_info_from_type(MBEDTLS_PK_RSA));
- if (!mbedtls_rsa_gen_key(mbedtls_pk_rsa(*key), _urandom, NULL, ksize, exp))
- return;
- } else {
- fprintf(stderr, "Generating EC private key\n");
- mbedtls_pk_setup(key, mbedtls_pk_info_from_type(MBEDTLS_PK_ECKEY));
- if (!mbedtls_ecp_gen_key(curve, mbedtls_pk_ec(*key), _urandom, NULL))
- return;
- }
- fprintf(stderr, "error: key generation failed\n");
- exit(1);
- }
- int dokey(bool rsa, char **arg)
- {
- mbedtls_pk_context key;
- unsigned int ksize = 512;
- int exp = 65537;
- char *path = NULL;
- bool pem = true;
- mbedtls_ecp_group_id curve = MBEDTLS_ECP_DP_SECP256R1;
- while (*arg && **arg == '-') {
- if (!strcmp(*arg, "-out") && arg[1]) {
- path = arg[1];
- arg++;
- } else if (!strcmp(*arg, "-3")) {
- exp = 3;
- } else if (!strcmp(*arg, "-der")) {
- pem = false;
- }
- arg++;
- }
- if (*arg && rsa) {
- ksize = (unsigned int)atoi(*arg);
- } else if (*arg) {
- curve = ecp_curve((const char *)*arg);
- if (curve == MBEDTLS_ECP_DP_NONE) {
- fprintf(stderr, "error: invalid curve name: %s\n", *arg);
- return 1;
- }
- }
- gen_key(&key, rsa, ksize, exp, curve, pem);
- write_key(&key, path, pem);
- mbedtls_pk_free(&key);
- return 0;
- }
- int selfsigned(char **arg)
- {
- mbedtls_pk_context key;
- mbedtls_x509write_cert cert;
- mbedtls_mpi serial;
- char *subject = "";
- unsigned int ksize = 512;
- int exp = 65537;
- unsigned int days = 30;
- char *keypath = NULL, *certpath = NULL;
- bool pem = true;
- time_t from = time(NULL), to;
- char fstr[20], tstr[20], sstr[17];
- int len;
- bool rsa = true;
- mbedtls_ecp_group_id curve = MBEDTLS_ECP_DP_SECP256R1;
- while (*arg && **arg == '-') {
- if (!strcmp(*arg, "-der")) {
- pem = false;
- } else if (!strcmp(*arg, "-newkey") && arg[1]) {
- if (!strncmp(arg[1], "rsa:", 4)) {
- rsa = true;
- ksize = (unsigned int)atoi(arg[1] + 4);
- } else if (!strcmp(arg[1], "ec")) {
- rsa = false;
- } else {
- fprintf(stderr, "error: invalid algorithm\n");
- return 1;
- }
- arg++;
- } else if (!strcmp(*arg, "-days") && arg[1]) {
- days = (unsigned int)atoi(arg[1]);
- arg++;
- } else if (!strcmp(*arg, "-pkeyopt") && arg[1]) {
- if (strncmp(arg[1], "ec_paramgen_curve:", 18)) {
- fprintf(stderr, "error: invalid pkey option: %s\n", arg[1]);
- return 1;
- }
- curve = ecp_curve((const char *)(arg[1] + 18));
- if (curve == MBEDTLS_ECP_DP_NONE) {
- fprintf(stderr, "error: invalid curve name: %s\n", arg[1] + 18);
- return 1;
- }
- arg++;
- } else if (!strcmp(*arg, "-keyout") && arg[1]) {
- keypath = arg[1];
- arg++;
- } else if (!strcmp(*arg, "-out") && arg[1]) {
- certpath = arg[1];
- arg++;
- } else if (!strcmp(*arg, "-subj") && arg[1]) {
- if (arg[1][0] != '/' || strchr(arg[1], ';')) {
- fprintf(stderr, "error: invalid subject");
- return 1;
- }
- subject = calloc(strlen(arg[1]) + 1, 1);
- char *oldc = arg[1] + 1, *newc = subject, *delim;
- do {
- delim = strchr(oldc, '=');
- if (!delim) {
- fprintf(stderr, "error: invalid subject");
- return 1;
- }
- memcpy(newc, oldc, delim - oldc + 1);
- newc += delim - oldc + 1;
- oldc = delim + 1;
- delim = strchr(oldc, '/');
- if (!delim) {
- delim = arg[1] + strlen(arg[1]);
- }
- memcpy(newc, oldc, delim - oldc);
- newc += delim - oldc;
- *newc++ = ',';
- oldc = delim + 1;
- } while(*delim);
- arg++;
- }
- arg++;
- }
- gen_key(&key, rsa, ksize, exp, curve, pem);
- if (keypath)
- write_key(&key, keypath, pem);
- from = (from < 1000000000) ? 1000000000 : from;
- strftime(fstr, sizeof(fstr), "%Y%m%d%H%M%S", gmtime(&from));
- to = from + 60 * 60 * 24 * days;
- if (to < from)
- to = INT_MAX;
- strftime(tstr, sizeof(tstr), "%Y%m%d%H%M%S", gmtime(&to));
- fprintf(stderr, "Generating selfsigned certificate with subject '%s'"
- " and validity %s-%s\n", subject, fstr, tstr);
- mbedtls_x509write_crt_init(&cert);
- mbedtls_x509write_crt_set_md_alg(&cert, MBEDTLS_MD_SHA256);
- mbedtls_x509write_crt_set_issuer_key(&cert, &key);
- mbedtls_x509write_crt_set_subject_key(&cert, &key);
- mbedtls_x509write_crt_set_subject_name(&cert, subject);
- mbedtls_x509write_crt_set_issuer_name(&cert, subject);
- mbedtls_x509write_crt_set_validity(&cert, fstr, tstr);
- mbedtls_x509write_crt_set_basic_constraints(&cert, 0, -1);
- mbedtls_x509write_crt_set_subject_key_identifier(&cert);
- mbedtls_x509write_crt_set_authority_key_identifier(&cert);
- _urandom(NULL, (void *) buf, 8);
- for (len = 0; len < 8; len++)
- sprintf(sstr + len*2, "%02x", (unsigned char) buf[len]);
- mbedtls_mpi_init(&serial);
- mbedtls_mpi_read_string(&serial, 16, sstr);
- mbedtls_x509write_crt_set_serial(&cert, &serial);
- if (pem) {
- if (mbedtls_x509write_crt_pem(&cert, (void *) buf, sizeof(buf), _urandom, NULL) < 0) {
- fprintf(stderr, "Failed to generate certificate\n");
- return 1;
- }
- len = strlen(buf);
- } else {
- len = mbedtls_x509write_crt_der(&cert, (void *) buf, sizeof(buf), _urandom, NULL);
- if (len < 0) {
- fprintf(stderr, "Failed to generate certificate: %d\n", len);
- return 1;
- }
- }
- write_file(certpath, len, pem);
- mbedtls_x509write_crt_free(&cert);
- mbedtls_mpi_free(&serial);
- mbedtls_pk_free(&key);
- return 0;
- }
- int main(int argc, char *argv[])
- {
- urandom_fd = open("/dev/urandom", O_RDONLY);
- if (!argv[1]) {
- //Usage
- } else if (!strcmp(argv[1], "eckey")) {
- return dokey(false, argv+2);
- } else if (!strcmp(argv[1], "rsakey")) {
- return dokey(true, argv+2);
- } else if (!strcmp(argv[1], "selfsigned")) {
- return selfsigned(argv+2);
- }
- fprintf(stderr,
- "PX5G X.509 Certificate Generator Utility v" PX5G_VERSION "\n" PX5G_COPY
- "\nbased on PolarSSL by Christophe Devine and Paul Bakker\n\n");
- fprintf(stderr, "Usage: %s [eckey|rsakey|selfsigned]\n", *argv);
- return 1;
- }
|