/*

 * Copyright (C) 2014 Nikos Mavrogiannopoulos

 *

 * This file is part of GnuTLS.

 *

 * GnuTLS is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License as published by

 * the Free Software Foundation; either version 3 of the License, or

 * (at your option) any later version.

 *

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

 * General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with GnuTLS; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA

 */

#ifndef SOFTHSM_H

# define SOFTHSM_H

#include <sys/stat.h>

#define SOFTHSM_V2

#ifdef SOFTHSM_V1

# define SOFTHSM_URL "pkcs11:model=SoftHSM;manufacturer=SoftHSM;serial=1;token=test"

# define LIB1 "/usr/lib64/pkcs11/libsofthsm.so"

# define LIB2 "/usr/lib/pkcs11/libsofthsm.so"

# define LIB3 "/usr/lib/softhsm/libsofthsm.so"

# define LIB4 "/usr/local/lib/softhsm/libsofthsm.so"

# define SOFTHSM_BIN1 "/usr/bin/softhsm"

# define SOFTHSM_BIN2 "/usr/local/bin/softhsm"

# define SOFTHSM_ENV "SOFTHSM_CONF"

#else

# define SOFTHSM_URL "pkcs11:model=SoftHSM%20v2;manufacturer=SoftHSM%20project;token=test"

# define LIB1 "/usr/lib64/pkcs11/libsofthsm2.so"

# define LIB2 "/usr/lib/pkcs11/libsofthsm2.so"

# define LIB3 "/usr/lib/softhsm/libsofthsm2.so"

# define LIB4 "/usr/lib/x86_64-linux-gnu/softhsm/libsofthsm2.so"

# define SOFTHSM_BIN1 "/usr/bin/softhsm2-util"

# define SOFTHSM_BIN2 "/usr/local/bin/softhsm2-util"

# define SOFTHSM_ENV "SOFTHSM2_CONF"

#endif

inline static const char *softhsm_lib(void) 

{

	const char *lib;

	if (sizeof(long) == 8 && access(LIB1, R_OK) == 0) {

		lib = LIB1;

	} else if (access(LIB2, R_OK) == 0) {

		lib = LIB2;

	} else if (access(LIB3, R_OK) == 0) {

		lib = LIB3;

	} else if (sizeof(long) == 8 && access(LIB4, R_OK) == 0) {

		lib = LIB4;

	} else {

		fprintf(stderr, "cannot find softhsm module\n");

		exit(77);

	}

	return lib;

}

inline static const char *softhsm_bin(void) 

{

	const char *bin;

	if (access(SOFTHSM_BIN1, X_OK) == 0) {

		bin = SOFTHSM_BIN1;

	} else if (access(SOFTHSM_BIN2, X_OK) == 0) {

		bin = SOFTHSM_BIN2;

	} else {

		fprintf(stderr, "cannot find softhsm bin\n");

		exit(77);

	}

	return bin;

}

static

void set_softhsm_conf(const char *config)

{

	char buf[128];

	char db_dir[128];

	FILE *fp;

	snprintf(db_dir, sizeof(db_dir), "%s.db", config);

	unsetenv(SOFTHSM_ENV);

	remove(config);

	fp = fopen(config, "w");

	if (fp == NULL) {

		fprintf(stderr, "error writing %s\n", config);

		exit(1);

	}

#ifdef SOFTHSM_V1

	remove(db_dir);

	snprintf(buf, sizeof(buf), "0:./%s\n", db_dir);

	fputs(buf, fp);

#else

	fputs("directories.tokendir = ", fp);

	fputs(db_dir, fp);

	fputs("\n", fp);

	fputs("objectstore.backend = file\n", fp);

	if (strlen(db_dir) < 6) {

		fprintf(stderr, "too short name for db: %s\n", db_dir);

		exit(1);

	}

	snprintf(buf, sizeof(buf), "rm -rf %s\n", db_dir);

	system(buf);

	mkdir(db_dir, 0755);

#endif

	fclose(fp);

	setenv(SOFTHSM_ENV, config, 0);

}

#endif