159 lines
3.2 KiB
C
159 lines
3.2 KiB
C
#ifdef HAVE_CONFIG_H
|
|
#include <config.h>
|
|
#endif
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <pthread.h>
|
|
#include <string.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/un.h>
|
|
#include <assert.h>
|
|
#include <errno.h>
|
|
#include <signal.h>
|
|
#include <unistd.h>
|
|
#include <stdarg.h>
|
|
#include <assuan.h>
|
|
#include <opensc/opensc.h>
|
|
#include <opensc/pkcs15.h>
|
|
#include "openscd.h"
|
|
|
|
struct openscd_context *dctx = NULL;
|
|
|
|
int opt_pipe = 0;
|
|
|
|
void cleanup()
|
|
{
|
|
if (dctx == NULL)
|
|
return;
|
|
cleanup_cmd_stuff(dctx);
|
|
if (dctx->ctx != NULL)
|
|
sc_release_context(dctx->ctx);
|
|
if (dctx->socket_name != NULL) {
|
|
char *p;
|
|
|
|
if (dctx->socket_fd >= 0)
|
|
close(dctx->socket_fd);
|
|
unlink(dctx->socket_name);
|
|
if ((p = strrchr(dctx->socket_name, '/')) != NULL) {
|
|
*p = '\0';
|
|
rmdir(dctx->socket_name);
|
|
}
|
|
free(dctx->socket_name);
|
|
}
|
|
free(dctx);
|
|
dctx = NULL;
|
|
}
|
|
|
|
void die(int return_code, const char *errmsg, ...)
|
|
{
|
|
if (errmsg != NULL) {
|
|
va_list ap;
|
|
char *p = malloc(strlen(errmsg)+2);
|
|
|
|
strcpy(p, errmsg);
|
|
strcat(p, "\n");
|
|
va_start(ap, errmsg);
|
|
vfprintf(stderr, p, ap);
|
|
free(p);
|
|
va_end(ap);
|
|
}
|
|
cleanup();
|
|
|
|
exit(return_code);
|
|
}
|
|
|
|
void setup_socket(void)
|
|
{
|
|
int fd;
|
|
const char *socket_dir = "/tmp/openscd-XXXXXX";
|
|
const char *socket_file = "socket";
|
|
char *socket_name;
|
|
struct sockaddr_un serv_addr;
|
|
|
|
socket_name = malloc(strlen(socket_dir)+strlen(socket_file)+2);
|
|
assert(socket_name != NULL);
|
|
strcpy(socket_name, socket_dir);
|
|
|
|
if (mkdtemp(socket_name) == NULL)
|
|
die(1, "Unable to create directory '%s': %s\n",
|
|
socket_name, strerror(errno));
|
|
strcat(socket_name, "/");
|
|
strcat(socket_name, socket_file);
|
|
if (strlen(socket_name) + 1 >= sizeof(serv_addr.sun_path))
|
|
die(1, "Name of socket too long\n");
|
|
fd = socket(AF_UNIX, SOCK_STREAM, 0);
|
|
if (fd == -1)
|
|
die(1, "Can't create socket: %s\n", strerror(errno));
|
|
memset(&serv_addr, 0, sizeof(serv_addr));
|
|
serv_addr.sun_family = AF_UNIX;
|
|
strcpy(serv_addr.sun_path, socket_name);
|
|
if (bind(fd, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) == -1) {
|
|
close(fd);
|
|
die(1, "Error binding socket to '%s': %s\n",
|
|
serv_addr.sun_path, strerror(errno));
|
|
}
|
|
if (listen(fd, 5) == -1) {
|
|
close(fd);
|
|
die(1, "listen() failed: %s\n", strerror(errno));
|
|
}
|
|
dctx->socket_name = strdup(socket_name);
|
|
dctx->socket_fd = fd;
|
|
}
|
|
|
|
void do_fork()
|
|
{
|
|
int pid;
|
|
|
|
pid = fork();
|
|
if (pid == (pid_t) -1)
|
|
die(1, "fork() failed: %s\n", strerror(errno));
|
|
if (pid) {
|
|
char *infostr;
|
|
int n = strlen(dctx->socket_name) + 40;
|
|
|
|
infostr = malloc(n);
|
|
if (snprintf(infostr, n, "OPENSCD_INFO=%s:%lu:1",
|
|
dctx->socket_name, (ulong) pid) < 0) {
|
|
kill(pid, SIGTERM);
|
|
die(1, "Something went wrong\n");
|
|
}
|
|
printf("%s; export OPENSCD_INFO;\n", infostr);
|
|
free(infostr);
|
|
|
|
free(dctx->socket_name);
|
|
dctx->socket_name = NULL;
|
|
die(0, NULL);
|
|
}
|
|
}
|
|
|
|
void init_dctx()
|
|
{
|
|
int r;
|
|
|
|
r = sc_establish_context(&dctx->ctx, "openscd");
|
|
if (r != 0)
|
|
die(1, "Unable to establish context: %s", sc_strerror(r));
|
|
init_cmd_stuff(dctx);
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
int r;
|
|
|
|
dctx = malloc(sizeof(struct openscd_context));
|
|
assert(dctx != NULL);
|
|
memset(dctx, 0, sizeof(struct openscd_context));
|
|
|
|
if (!opt_pipe) {
|
|
setup_socket();
|
|
do_fork();
|
|
}
|
|
|
|
init_dctx();
|
|
|
|
atexit(cleanup);
|
|
command_handler(dctx);
|
|
|
|
return 0;
|
|
}
|