[thirdparty/openssl.git] / apps / rehash.c

/*
 * Copyright 2015-2016 The OpenSSL Project Authors. All Rights Reserved.
 *
 * Licensed under the OpenSSL license (the "License").  You may not use
 * this file except in compliance with the License.  You can obtain a copy
 * in the file LICENSE in the source distribution or at
 * https://www.openssl.org/source/license.html
 */

/*
 * C implementation based on the original Perl and shell versions
 *
 * Copyright (c) 2013-2014 Timo Teräs <timo.teras@iki.fi>
 */

#include "apps.h"

#if defined(OPENSSL_SYS_UNIX) || defined(__APPLE__) || \
    (defined(__VMS) && defined(__DECC) && __CTRL_VER >= 80300000)
# include <unistd.h>
# include <stdio.h>
# include <limits.h>
# include <errno.h>
# include <string.h>
# include <ctype.h>
# include <sys/stat.h>

# include "internal/o_dir.h"
# include <openssl/evp.h>
# include <openssl/pem.h>
# include <openssl/x509.h>


# ifndef PATH_MAX
#  define PATH_MAX 4096
# endif
# ifndef NAME_MAX
#  define NAME_MAX 255
# endif
# define MAX_COLLISIONS  256

typedef struct hentry_st {
    struct hentry_st *next;
    char *filename;
    unsigned short old_id;
    unsigned char need_symlink;
    unsigned char digest[EVP_MAX_MD_SIZE];
} HENTRY;

typedef struct bucket_st {
    struct bucket_st *next;
    HENTRY *first_entry, *last_entry;
    unsigned int hash;
    unsigned short type;
    unsigned short num_needed;
} BUCKET;

enum Type {
    /* Keep in sync with |suffixes|, below. */
    TYPE_CERT=0, TYPE_CRL=1
};

enum Hash {
    HASH_OLD, HASH_NEW, HASH_BOTH
};


static int evpmdsize;
static const EVP_MD *evpmd;
static int remove_links = 1;
static int verbose = 0;
static BUCKET *hash_table[257];

static const char *suffixes[] = { "", "r" };
static const char *extensions[] = { "pem", "crt", "cer", "crl" };


static void bit_set(unsigned char *set, unsigned int bit)
{
    set[bit >> 3] |= 1 << (bit & 0x7);
}

static int bit_isset(unsigned char *set, unsigned int bit)
{
    return set[bit >> 3] & (1 << (bit & 0x7));
}


/*
 * Process an entry; return number of errors.
 */
static int add_entry(enum Type type, unsigned int hash, const char *filename,
                      const unsigned char *digest, int need_symlink,
                      unsigned short old_id)
{
    static BUCKET nilbucket;
    static HENTRY nilhentry;
    BUCKET *bp;
    HENTRY *ep, *found = NULL;
    unsigned int ndx = (type + hash) % OSSL_NELEM(hash_table);

    for (bp = hash_table[ndx]; bp; bp = bp->next)
        if (bp->type == type && bp->hash == hash)
            break;
    if (bp == NULL) {
        bp = app_malloc(sizeof(*bp), "hash bucket");
        *bp = nilbucket;
        bp->next = hash_table[ndx];
        bp->type = type;
        bp->hash = hash;
        hash_table[ndx] = bp;
    }

    for (ep = bp->first_entry; ep; ep = ep->next) {
        if (digest && memcmp(digest, ep->digest, evpmdsize) == 0) {
            BIO_printf(bio_err,
                       "%s: skipping duplicate certificate in %s\n",
                       opt_getprog(), filename);
            return 1;
        }
        if (strcmp(filename, ep->filename) == 0) {
            found = ep;
            if (digest == NULL)
                break;
        }
    }
    ep = found;
    if (ep == NULL) {
        if (bp->num_needed >= MAX_COLLISIONS) {
            BIO_printf(bio_err,
                       "%s: hash table overflow for %s\n",
                       opt_getprog(), filename);
            return 1;
        }
        ep = app_malloc(sizeof(*ep), "collision bucket");
        *ep = nilhentry;
        ep->old_id = ~0;
        ep->filename = OPENSSL_strdup(filename);
        if (bp->last_entry)
            bp->last_entry->next = ep;
        if (bp->first_entry == NULL)
            bp->first_entry = ep;
        bp->last_entry = ep;
    }

    if (old_id < ep->old_id)
        ep->old_id = old_id;
    if (need_symlink && !ep->need_symlink) {
        ep->need_symlink = 1;
        bp->num_needed++;
        memcpy(ep->digest, digest, evpmdsize);
    }
    return 0;
}

/*
 * Check if a symlink goes to the right spot; return 0 if okay.
 * This can be -1 if bad filename, or an error count.
 */
static int handle_symlink(const char *filename, const char *fullpath)
{
    unsigned int hash = 0;
    int i, type, id;
    unsigned char ch;
    char linktarget[PATH_MAX], *endptr;
    ossl_ssize_t n;

    for (i = 0; i < 8; i++) {
        ch = filename[i];
        if (!isxdigit(ch))
            return -1;
        hash <<= 4;
        hash += OPENSSL_hexchar2int(ch);
    }
    if (filename[i++] != '.')
        return -1;
    for (type = OSSL_NELEM(suffixes) - 1; type > 0; type--)
        if (strcasecmp(suffixes[type], &filename[i]) == 0)
            break;
    i += strlen(suffixes[type]);

    id = strtoul(&filename[i], &endptr, 10);
    if (*endptr != '\0')
        return -1;

    n = readlink(fullpath, linktarget, sizeof(linktarget));
    if (n < 0 || n >= (int)sizeof(linktarget))
        return -1;
    linktarget[n] = 0;

    return add_entry(type, hash, linktarget, NULL, 0, id);
}

/*
 * process a file, return number of errors.
 */
static int do_file(const char *filename, const char *fullpath, enum Hash h)
{
    STACK_OF (X509_INFO) *inf = NULL;
    X509_INFO *x;
    X509_NAME *name = NULL;
    BIO *b;
    const char *ext;
    unsigned char digest[EVP_MAX_MD_SIZE];
    int type, errs = 0;
    size_t i;

    /* Does it end with a recognized extension? */
    if ((ext = strrchr(filename, '.')) == NULL)
        goto end;
    for (i = 0; i < OSSL_NELEM(extensions); i++) {
        if (strcasecmp(extensions[i], ext + 1) == 0)
            break;
    }
    if (i >= OSSL_NELEM(extensions))
        goto end;

    /* Does it have X.509 data in it? */
    if ((b = BIO_new_file(fullpath, "r")) == NULL) {
        BIO_printf(bio_err, "%s: skipping %s, cannot open file\n",
                   opt_getprog(), filename);
        errs++;
        goto end;
    }
    inf = PEM_X509_INFO_read_bio(b, NULL, NULL, NULL);
    BIO_free(b);
    if (inf == NULL)
        goto end;

    if (sk_X509_INFO_num(inf) != 1) {
        BIO_printf(bio_err,
                   "%s: skipping %s,"
                   "it does not contain exactly one certificate or CRL\n",
                   opt_getprog(), filename);
        /* This is not an error. */
        goto end;
    }
    x = sk_X509_INFO_value(inf, 0);
    if (x->x509) {
        type = TYPE_CERT;
        name = X509_get_subject_name(x->x509);
        X509_digest(x->x509, evpmd, digest, NULL);
    } else if (x->crl) {
        type = TYPE_CRL;
        name = X509_CRL_get_issuer(x->crl);
        X509_CRL_digest(x->crl, evpmd, digest, NULL);
    } else {
        ++errs;
        goto end;
    }
    if (name) {
        if ((h == HASH_NEW) || (h == HASH_BOTH))
            errs += add_entry(type, X509_NAME_hash(name), filename, digest, 1, ~0);
        if ((h == HASH_OLD) || (h == HASH_BOTH))
            errs += add_entry(type, X509_NAME_hash_old(name), filename, digest, 1, ~0);
    }

end:
    sk_X509_INFO_pop_free(inf, X509_INFO_free);
    return errs;
}

/*
 * Process a directory; return number of errors found.
 */
static int do_dir(const char *dirname, enum Hash h)
{
    BUCKET *bp, *nextbp;
    HENTRY *ep, *nextep;
    OPENSSL_DIR_CTX *d = NULL;
    struct stat st;
    unsigned char idmask[MAX_COLLISIONS / 8];
    int n, nextid, buflen, errs = 0;
    size_t i;
    const char *pathsep;
    const char *filename;
    char *buf;

    if (app_access(dirname, W_OK) < 0) {
        BIO_printf(bio_err, "Skipping %s, can't write\n", dirname);
        return 1;
    }
    buflen = strlen(dirname);
    pathsep = (buflen && dirname[buflen - 1] == '/') ? "" : "/";
    buflen += NAME_MAX + 1 + 1;
    buf = app_malloc(buflen, "filename buffer");

    if (verbose)
        BIO_printf(bio_out, "Doing %s\n", dirname);

    while ((filename = OPENSSL_DIR_read(&d, dirname)) != NULL) {
        if (snprintf(buf, buflen, "%s%s%s",
                    dirname, pathsep, filename) >= buflen)
            continue;
        if (lstat(buf, &st) < 0)
            continue;
        if (S_ISLNK(st.st_mode) && handle_symlink(filename, buf) == 0)
            continue;
        errs += do_file(filename, buf, h);
    }
    OPENSSL_DIR_end(&d);

    for (i = 0; i < OSSL_NELEM(hash_table); i++) {
        for (bp = hash_table[i]; bp; bp = nextbp) {
            nextbp = bp->next;
            nextid = 0;
            memset(idmask, 0, (bp->num_needed + 7) / 8);
            for (ep = bp->first_entry; ep; ep = ep->next)
                if (ep->old_id < bp->num_needed)
                    bit_set(idmask, ep->old_id);

            for (ep = bp->first_entry; ep; ep = nextep) {
                nextep = ep->next;
                if (ep->old_id < bp->num_needed) {
                    /* Link exists, and is used as-is */
                    snprintf(buf, buflen, "%08x.%s%d", bp->hash,
                             suffixes[bp->type], ep->old_id);
                    if (verbose)
                        BIO_printf(bio_out, "link %s -> %s\n",
                                   ep->filename, buf);
                } else if (ep->need_symlink) {
                    /* New link needed (it may replace something) */
                    while (bit_isset(idmask, nextid))
                        nextid++;

                    snprintf(buf, buflen, "%s%s%n%08x.%s%d",
                             dirname, pathsep, &n, bp->hash,
                             suffixes[bp->type], nextid);
                    if (verbose)
                        BIO_printf(bio_out, "link %s -> %s\n",
                                   ep->filename, &buf[n]);
                    if (unlink(buf) < 0 && errno != ENOENT) {
                        BIO_printf(bio_err,
                                   "%s: Can't unlink %s, %s\n",
                                   opt_getprog(), buf, strerror(errno));
                        errs++;
                    }
                    if (symlink(ep->filename, buf) < 0) {
                        BIO_printf(bio_err,
                                   "%s: Can't symlink %s, %s\n",
                                   opt_getprog(), ep->filename,
                                   strerror(errno));
                        errs++;
                    }
                } else if (remove_links) {
                    /* Link to be deleted */
                    snprintf(buf, buflen, "%s%s%n%08x.%s%d",
                             dirname, pathsep, &n, bp->hash,
                             suffixes[bp->type], ep->old_id);
                    if (verbose)
                        BIO_printf(bio_out, "unlink %s\n",
                                   &buf[n]);
                    if (unlink(buf) < 0 && errno != ENOENT) {
                        BIO_printf(bio_err,
                                   "%s: Can't unlink %s, %s\n",
                                   opt_getprog(), buf, strerror(errno));
                        errs++;
                    }
                }
                OPENSSL_free(ep->filename);
                OPENSSL_free(ep);
            }
            OPENSSL_free(bp);
        }
        hash_table[i] = NULL;
    }

    OPENSSL_free(buf);
    return errs;
}

typedef enum OPTION_choice {
    OPT_ERR = -1, OPT_EOF = 0, OPT_HELP,
    OPT_COMPAT, OPT_OLD, OPT_N, OPT_VERBOSE
} OPTION_CHOICE;

OPTIONS rehash_options[] = {
    {OPT_HELP_STR, 1, '-', "Usage: %s [options] [cert-directory...]\n"},
    {OPT_HELP_STR, 1, '-', "Valid options are:\n"},
    {"help", OPT_HELP, '-', "Display this summary"},
    {"compat", OPT_COMPAT, '-', "Create both new- and old-style hash links"},
    {"old", OPT_OLD, '-', "Use old-style hash to generate links"},
    {"n", OPT_N, '-', "Do not remove existing links"},
    {"v", OPT_VERBOSE, '-', "Verbose output"},
    {NULL}
};


int rehash_main(int argc, char **argv)
{
    const char *env, *prog;
    char *e, *m;
    int errs = 0;
    OPTION_CHOICE o;
    enum Hash h = HASH_NEW;

    prog = opt_init(argc, argv, rehash_options);
    while ((o = opt_next()) != OPT_EOF) {
        switch (o) {
        case OPT_EOF:
        case OPT_ERR:
            BIO_printf(bio_err, "%s: Use -help for summary.\n", prog);
            goto end;
        case OPT_HELP:
            opt_help(rehash_options);
            goto end;
        case OPT_COMPAT:
            h = HASH_BOTH;
            break;
        case OPT_OLD:
            h = HASH_OLD;
            break;
        case OPT_N:
            remove_links = 0;
            break;
        case OPT_VERBOSE:
            verbose = 1;
            break;
        }
    }
    argc = opt_num_rest();
    argv = opt_rest();

    evpmd = EVP_sha1();
    evpmdsize = EVP_MD_size(evpmd);

    if (*argv) {
        while (*argv)
            errs += do_dir(*argv++, h);
    } else if ((env = getenv("SSL_CERT_DIR")) != NULL) {
        m = OPENSSL_strdup(env);
        for (e = strtok(m, ":"); e != NULL; e = strtok(NULL, ":"))
            errs += do_dir(e, h);
        OPENSSL_free(m);
    } else {
        errs += do_dir("/etc/ssl/certs", h);
    }

 end:
    return errs;
}

#else
OPTIONS rehash_options[] = {
    {NULL}
};

int rehash_main(int argc, char **argv)
{
    BIO_printf(bio_err, "Not available; use c_rehash script\n");
    return (1);
}

#endif /* defined(OPENSSL_SYS_UNIX) || defined(__APPLE__) */
Commit	Line	Data
8f6f1441	1	/*
846e33c7	2	* Copyright 2015-2016 The OpenSSL Project Authors. All Rights Reserved.
8f6f1441	3	*
846e33c7 RS	4	* Licensed under the OpenSSL license (the "License"). You may not use
	5	* this file except in compliance with the License. You can obtain a copy
	6	* in the file LICENSE in the source distribution or at
	7	* https://www.openssl.org/source/license.html
8f6f1441	8	*/
846e33c7 RS	9
	10	/*
	11	* C implementation based on the original Perl and shell versions
8f6f1441	12	*
846e33c7	13	* Copyright (c) 2013-2014 Timo Teräs <timo.teras@iki.fi>
8f6f1441 TT	14	*/
	15
	16	#include "apps.h"
	17
6616429d RL	18	#if defined(OPENSSL_SYS_UNIX) \|\| defined(__APPLE__) \|\| \
6616429d RL	19	(defined(__VMS) && defined(__DECC) && __CTRL_VER >= 80300000)
8f6f1441 TT	20	# include <unistd.h>
	21	# include <stdio.h>
	22	# include <limits.h>
	23	# include <errno.h>
	24	# include <string.h>
	25	# include <ctype.h>
	26	# include <sys/stat.h>
	27
	28	# include "internal/o_dir.h"
	29	# include <openssl/evp.h>
	30	# include <openssl/pem.h>
	31	# include <openssl/x509.h>
	32
	33
6616429d RL	34	# ifndef PATH_MAX
	35	# define PATH_MAX 4096
	36	# endif
c9c84a13 AG	37	# ifndef NAME_MAX
	38	# define NAME_MAX 255
	39	# endif
8f6f1441 TT	40	# define MAX_COLLISIONS 256
	41
	42	typedef struct hentry_st {
	43	struct hentry_st *next;
	44	char *filename;
	45	unsigned short old_id;
	46	unsigned char need_symlink;
	47	unsigned char digest[EVP_MAX_MD_SIZE];
	48	} HENTRY;
	49
	50	typedef struct bucket_st {
	51	struct bucket_st *next;
	52	HENTRY first_entry, last_entry;
	53	unsigned int hash;
	54	unsigned short type;
	55	unsigned short num_needed;
	56	} BUCKET;
	57
	58	enum Type {
	59	/* Keep in sync with \|suffixes\|, below. */
	60	TYPE_CERT=0, TYPE_CRL=1
	61	};
	62
	63	enum Hash {
	64	HASH_OLD, HASH_NEW, HASH_BOTH
	65	};
	66
	67
	68	static int evpmdsize;
	69	static const EVP_MD *evpmd;
	70	static int remove_links = 1;
	71	static int verbose = 0;
	72	static BUCKET *hash_table[257];
	73
	74	static const char *suffixes[] = { "", "r" };
	75	static const char *extensions[] = { "pem", "crt", "cer", "crl" };
	76
	77
	78	static void bit_set(unsigned char *set, unsigned int bit)
	79	{
	80	set[bit >> 3] \|= 1 << (bit & 0x7);
	81	}
	82
	83	static int bit_isset(unsigned char *set, unsigned int bit)
	84	{
	85	return set[bit >> 3] & (1 << (bit & 0x7));
	86	}
	87
	88
ce249fac RS	89	/*
	90	* Process an entry; return number of errors.
	91	*/
	92	static int add_entry(enum Type type, unsigned int hash, const char *filename,
8f6f1441 TT	93	const unsigned char *digest, int need_symlink,
	94	unsigned short old_id)
	95	{
	96	static BUCKET nilbucket;
	97	static HENTRY nilhentry;
	98	BUCKET *bp;
	99	HENTRY ep, found = NULL;
	100	unsigned int ndx = (type + hash) % OSSL_NELEM(hash_table);
	101
	102	for (bp = hash_table[ndx]; bp; bp = bp->next)
	103	if (bp->type == type && bp->hash == hash)
	104	break;
	105	if (bp == NULL) {
	106	bp = app_malloc(sizeof(*bp), "hash bucket");
	107	*bp = nilbucket;
	108	bp->next = hash_table[ndx];
	109	bp->type = type;
	110	bp->hash = hash;
	111	hash_table[ndx] = bp;
	112	}
	113
	114	for (ep = bp->first_entry; ep; ep = ep->next) {
	115	if (digest && memcmp(digest, ep->digest, evpmdsize) == 0) {
	116	BIO_printf(bio_err,
	117	"%s: skipping duplicate certificate in %s\n",
	118	opt_getprog(), filename);
ce249fac	119	return 1;
8f6f1441 TT	120	}
	121	if (strcmp(filename, ep->filename) == 0) {
	122	found = ep;
	123	if (digest == NULL)
	124	break;
	125	}
	126	}
	127	ep = found;
	128	if (ep == NULL) {
ce249fac RS	129	if (bp->num_needed >= MAX_COLLISIONS) {
	130	BIO_printf(bio_err,
	131	"%s: hash table overflow for %s\n",
	132	opt_getprog(), filename);
	133	return 1;
	134	}
8f6f1441 TT	135	ep = app_malloc(sizeof(*ep), "collision bucket");
	136	*ep = nilhentry;
	137	ep->old_id = ~0;
7644a9ae	138	ep->filename = OPENSSL_strdup(filename);
8f6f1441 TT	139	if (bp->last_entry)
	140	bp->last_entry->next = ep;
	141	if (bp->first_entry == NULL)
	142	bp->first_entry = ep;
	143	bp->last_entry = ep;
	144	}
	145
	146	if (old_id < ep->old_id)
	147	ep->old_id = old_id;
	148	if (need_symlink && !ep->need_symlink) {
	149	ep->need_symlink = 1;
	150	bp->num_needed++;
	151	memcpy(ep->digest, digest, evpmdsize);
	152	}
ce249fac	153	return 0;
8f6f1441 TT	154	}
8f6f1441 TT	155
ce249fac RS	156	/*
	157	* Check if a symlink goes to the right spot; return 0 if okay.
	158	* This can be -1 if bad filename, or an error count.
	159	*/
8f6f1441 TT	160	static int handle_symlink(const char filename, const char fullpath)
	161	{
	162	unsigned int hash = 0;
	163	int i, type, id;
	164	unsigned char ch;
e7a68985	165	char linktarget[PATH_MAX], *endptr;
6616429d	166	ossl_ssize_t n;
8f6f1441 TT	167
	168	for (i = 0; i < 8; i++) {
	169	ch = filename[i];
	170	if (!isxdigit(ch))
	171	return -1;
	172	hash <<= 4;
14f051a0	173	hash += OPENSSL_hexchar2int(ch);
8f6f1441 TT	174	}
	175	if (filename[i++] != '.')
	176	return -1;
	177	for (type = OSSL_NELEM(suffixes) - 1; type > 0; type--)
	178	if (strcasecmp(suffixes[type], &filename[i]) == 0)
	179	break;
	180	i += strlen(suffixes[type]);
	181
	182	id = strtoul(&filename[i], &endptr, 10);
	183	if (*endptr != '\0')
	184	return -1;
	185
	186	n = readlink(fullpath, linktarget, sizeof(linktarget));
	187	if (n < 0 \|\| n >= (int)sizeof(linktarget))
	188	return -1;
	189	linktarget[n] = 0;
	190
ce249fac	191	return add_entry(type, hash, linktarget, NULL, 0, id);
8f6f1441 TT	192	}
8f6f1441 TT	193
ce249fac RS	194	/*
	195	* process a file, return number of errors.
	196	*/
8f6f1441 TT	197	static int do_file(const char filename, const char fullpath, enum Hash h)
8f6f1441 TT	198	{
ce249fac	199	STACK_OF (X509_INFO) *inf = NULL;
8f6f1441 TT	200	X509_INFO *x;
	201	X509_NAME *name = NULL;
	202	BIO *b;
	203	const char *ext;
	204	unsigned char digest[EVP_MAX_MD_SIZE];
8c82de99 BL	205	int type, errs = 0;
8c82de99 BL	206	size_t i;
8f6f1441	207
ce249fac	208	/* Does it end with a recognized extension? */
8f6f1441	209	if ((ext = strrchr(filename, '.')) == NULL)
ce249fac	210	goto end;
8c82de99	211	for (i = 0; i < OSSL_NELEM(extensions); i++) {
8f6f1441 TT	212	if (strcasecmp(extensions[i], ext + 1) == 0)
	213	break;
	214	}
8c82de99	215	if (i >= OSSL_NELEM(extensions))
ce249fac	216	goto end;
8f6f1441	217
ce249fac RS	218	/* Does it have X.509 data in it? */
	219	if ((b = BIO_new_file(fullpath, "r")) == NULL) {
	220	BIO_printf(bio_err, "%s: skipping %s, cannot open file\n",
	221	opt_getprog(), filename);
	222	errs++;
	223	goto end;
	224	}
8f6f1441 TT	225	inf = PEM_X509_INFO_read_bio(b, NULL, NULL, NULL);
	226	BIO_free(b);
	227	if (inf == NULL)
ce249fac	228	goto end;
8f6f1441 TT	229
	230	if (sk_X509_INFO_num(inf) != 1) {
	231	BIO_printf(bio_err,
	232	"%s: skipping %s,"
	233	"it does not contain exactly one certificate or CRL\n",
	234	opt_getprog(), filename);
ce249fac	235	/* This is not an error. */
8f6f1441 TT	236	goto end;
	237	}
	238	x = sk_X509_INFO_value(inf, 0);
	239	if (x->x509) {
	240	type = TYPE_CERT;
	241	name = X509_get_subject_name(x->x509);
	242	X509_digest(x->x509, evpmd, digest, NULL);
	243	} else if (x->crl) {
	244	type = TYPE_CRL;
	245	name = X509_CRL_get_issuer(x->crl);
	246	X509_CRL_digest(x->crl, evpmd, digest, NULL);
8c82de99 BL	247	} else {
	248	++errs;
	249	goto end;
8f6f1441 TT	250	}
	251	if (name) {
	252	if ((h == HASH_NEW) \|\| (h == HASH_BOTH))
ce249fac	253	errs += add_entry(type, X509_NAME_hash(name), filename, digest, 1, ~0);
8f6f1441	254	if ((h == HASH_OLD) \|\| (h == HASH_BOTH))
ce249fac	255	errs += add_entry(type, X509_NAME_hash_old(name), filename, digest, 1, ~0);
8f6f1441 TT	256	}
	257
	258	end:
	259	sk_X509_INFO_pop_free(inf, X509_INFO_free);
ce249fac	260	return errs;
8f6f1441 TT	261	}
8f6f1441 TT	262
ce249fac RS	263	/*
	264	* Process a directory; return number of errors found.
	265	*/
8f6f1441 TT	266	static int do_dir(const char *dirname, enum Hash h)
	267	{
	268	BUCKET bp, nextbp;
	269	HENTRY ep, nextep;
	270	OPENSSL_DIR_CTX *d = NULL;
	271	struct stat st;
	272	unsigned char idmask[MAX_COLLISIONS / 8];
8c82de99 BL	273	int n, nextid, buflen, errs = 0;
8c82de99 BL	274	size_t i;
8f6f1441 TT	275	const char *pathsep;
	276	const char *filename;
	277	char *buf;
	278
ff2f6bb0 RS	279	if (app_access(dirname, W_OK) < 0) {
ff2f6bb0 RS	280	BIO_printf(bio_err, "Skipping %s, can't write\n", dirname);
4c7103a5	281	return 1;
ff2f6bb0	282	}
8f6f1441 TT	283	buflen = strlen(dirname);
8f6f1441 TT	284	pathsep = (buflen && dirname[buflen - 1] == '/') ? "" : "/";
c9c84a13 AG	285	buflen += NAME_MAX + 1 + 1;
c9c84a13 AG	286	buf = app_malloc(buflen, "filename buffer");
8f6f1441 TT	287
	288	if (verbose)
	289	BIO_printf(bio_out, "Doing %s\n", dirname);
	290
	291	while ((filename = OPENSSL_DIR_read(&d, dirname)) != NULL) {
	292	if (snprintf(buf, buflen, "%s%s%s",
	293	dirname, pathsep, filename) >= buflen)
	294	continue;
	295	if (lstat(buf, &st) < 0)
	296	continue;
	297	if (S_ISLNK(st.st_mode) && handle_symlink(filename, buf) == 0)
	298	continue;
ce249fac	299	errs += do_file(filename, buf, h);
8f6f1441 TT	300	}
	301	OPENSSL_DIR_end(&d);
	302
8c82de99	303	for (i = 0; i < OSSL_NELEM(hash_table); i++) {
8f6f1441 TT	304	for (bp = hash_table[i]; bp; bp = nextbp) {
	305	nextbp = bp->next;
	306	nextid = 0;
	307	memset(idmask, 0, (bp->num_needed + 7) / 8);
	308	for (ep = bp->first_entry; ep; ep = ep->next)
	309	if (ep->old_id < bp->num_needed)
	310	bit_set(idmask, ep->old_id);
	311
	312	for (ep = bp->first_entry; ep; ep = nextep) {
	313	nextep = ep->next;
	314	if (ep->old_id < bp->num_needed) {
	315	/* Link exists, and is used as-is */
	316	snprintf(buf, buflen, "%08x.%s%d", bp->hash,
	317	suffixes[bp->type], ep->old_id);
	318	if (verbose)
	319	BIO_printf(bio_out, "link %s -> %s\n",
	320	ep->filename, buf);
	321	} else if (ep->need_symlink) {
	322	/* New link needed (it may replace something) */
	323	while (bit_isset(idmask, nextid))
	324	nextid++;
	325
	326	snprintf(buf, buflen, "%s%s%n%08x.%s%d",
	327	dirname, pathsep, &n, bp->hash,
	328	suffixes[bp->type], nextid);
	329	if (verbose)
	330	BIO_printf(bio_out, "link %s -> %s\n",
	331	ep->filename, &buf[n]);
ce249fac	332	if (unlink(buf) < 0 && errno != ENOENT) {
8f6f1441 TT	333	BIO_printf(bio_err,
	334	"%s: Can't unlink %s, %s\n",
	335	opt_getprog(), buf, strerror(errno));
ce249fac RS	336	errs++;
	337	}
	338	if (symlink(ep->filename, buf) < 0) {
8f6f1441 TT	339	BIO_printf(bio_err,
	340	"%s: Can't symlink %s, %s\n",
	341	opt_getprog(), ep->filename,
	342	strerror(errno));
ce249fac RS	343	errs++;
ce249fac RS	344	}
8f6f1441 TT	345	} else if (remove_links) {
	346	/* Link to be deleted */
	347	snprintf(buf, buflen, "%s%s%n%08x.%s%d",
	348	dirname, pathsep, &n, bp->hash,
	349	suffixes[bp->type], ep->old_id);
	350	if (verbose)
	351	BIO_printf(bio_out, "unlink %s\n",
	352	&buf[n]);
ce249fac	353	if (unlink(buf) < 0 && errno != ENOENT) {
8f6f1441 TT	354	BIO_printf(bio_err,
	355	"%s: Can't unlink %s, %s\n",
	356	opt_getprog(), buf, strerror(errno));
ce249fac RS	357	errs++;
ce249fac RS	358	}
8f6f1441 TT	359	}
	360	OPENSSL_free(ep->filename);
	361	OPENSSL_free(ep);
	362	}
	363	OPENSSL_free(bp);
	364	}
	365	hash_table[i] = NULL;
	366	}
8f6f1441 TT	367
8f6f1441 TT	368	OPENSSL_free(buf);
ce249fac	369	return errs;
8f6f1441 TT	370	}
	371
	372	typedef enum OPTION_choice {
	373	OPT_ERR = -1, OPT_EOF = 0, OPT_HELP,
	374	OPT_COMPAT, OPT_OLD, OPT_N, OPT_VERBOSE
	375	} OPTION_CHOICE;
	376
	377	OPTIONS rehash_options[] = {
	378	{OPT_HELP_STR, 1, '-', "Usage: %s [options] [cert-directory...]\n"},
	379	{OPT_HELP_STR, 1, '-', "Valid options are:\n"},
	380	{"help", OPT_HELP, '-', "Display this summary"},
	381	{"compat", OPT_COMPAT, '-', "Create both new- and old-style hash links"},
	382	{"old", OPT_OLD, '-', "Use old-style hash to generate links"},
	383	{"n", OPT_N, '-', "Do not remove existing links"},
	384	{"v", OPT_VERBOSE, '-', "Verbose output"},
	385	{NULL}
	386	};
	387
	388
	389	int rehash_main(int argc, char **argv)
	390	{
	391	const char env, prog;
	392	char e, m;
ce249fac	393	int errs = 0;
8f6f1441 TT	394	OPTION_CHOICE o;
	395	enum Hash h = HASH_NEW;
	396
	397	prog = opt_init(argc, argv, rehash_options);
	398	while ((o = opt_next()) != OPT_EOF) {
	399	switch (o) {
	400	case OPT_EOF:
	401	case OPT_ERR:
	402	BIO_printf(bio_err, "%s: Use -help for summary.\n", prog);
	403	goto end;
	404	case OPT_HELP:
	405	opt_help(rehash_options);
	406	goto end;
	407	case OPT_COMPAT:
	408	h = HASH_BOTH;
	409	break;
	410	case OPT_OLD:
	411	h = HASH_OLD;
	412	break;
	413	case OPT_N:
	414	remove_links = 0;
	415	break;
	416	case OPT_VERBOSE:
	417	verbose = 1;
	418	break;
	419	}
	420	}
	421	argc = opt_num_rest();
	422	argv = opt_rest();
	423
	424	evpmd = EVP_sha1();
	425	evpmdsize = EVP_MD_size(evpmd);
	426
	427	if (*argv) {
	428	while (*argv)
ce249fac	429	errs += do_dir(*argv++, h);
8f6f1441	430	} else if ((env = getenv("SSL_CERT_DIR")) != NULL) {
7644a9ae	431	m = OPENSSL_strdup(env);
8f6f1441	432	for (e = strtok(m, ":"); e != NULL; e = strtok(NULL, ":"))
ce249fac	433	errs += do_dir(e, h);
8f6f1441 TT	434	OPENSSL_free(m);
8f6f1441 TT	435	} else {
ce249fac	436	errs += do_dir("/etc/ssl/certs", h);
8f6f1441 TT	437	}
	438
	439	end:
ce249fac	440	return errs;
8f6f1441 TT	441	}
	442
	443	#else
62fdf4ee RS	444	OPTIONS rehash_options[] = {
	445	{NULL}
	446	};
8f6f1441 TT	447
	448	int rehash_main(int argc, char **argv)
	449	{
9e0da060	450	BIO_printf(bio_err, "Not available; use c_rehash script\n");
8f6f1441 TT	451	return (1);
	452	}
	453
568b8020	454	#endif /* defined(OPENSSL_SYS_UNIX) \|\| defined(__APPLE__) */