int
tls_accept_socket(struct tls *ctx, struct tls **cctx, int socket)
{
int rv;
rv = tls_accept_fds(ctx, cctx, socket, socket);
if (*cctx != NULL)
(*cctx)->socket = socket;
return (rv);
}
int
tls_accept_socket(struct tls *ctx, struct tls **cctx, int socket)
{
return (tls_accept_fds(ctx, cctx, socket, socket));
}
int
main(int argc, char *argv[])
{
struct tls *tls = NULL;
struct tls *cctx = NULL;
struct tls_config *tls_config = NULL;
char buf[BUFSIZ];
int ch;
int e;
if ((tls_config = tls_config_new()) == NULL)
err(EXIT_FAILURE, "tls_config_new");
while ((ch = getopt(argc, argv, "Cc:k:p:f:")) != -1) {
switch (ch) {
case 'C':
tls_config_verify_client(tls_config);
break;
case 'c':
if (tls_config_set_cert_file(tls_config, optarg) == -1)
err(EXIT_FAILURE, "tls_config_set_cert_file");
break;
case 'k':
if (tls_config_set_key_file(tls_config, optarg) == -1)
err(EXIT_FAILURE, "tls_config_set_key_file");
break;
case 'f':
if (tls_config_set_ca_file(tls_config, optarg) == -1)
err(EXIT_FAILURE, "tls_config_set_ca_file");
break;
case 'p':
if (tls_config_set_ca_path(tls_config, optarg) == -1)
err(EXIT_FAILURE, "tls_config_set_ca_path");
break;
default:
usage();
/* NOTREACHED */
}
}
argc -= optind;
argv += optind;
/* prepare libtls */
if (tls_init() == -1)
err(EXIT_FAILURE, "tls_init");
if ((tls = tls_server()) == NULL)
err(EXIT_FAILURE, "tls_server");
if (tls_configure(tls, tls_config) == -1)
goto err;
if (tls_accept_fds(tls, &cctx, STDIN_FILENO, STDOUT_FILENO) == -1)
goto err;
if (tls_handshake(cctx) == -1)
goto err;
if (setenv("PROTO", "SSL", 1) == -1)
err(EXIT_FAILURE, "setenv");
/* fork front end program */
char *prog = argv[0];
# define PIPE_READ 0
# define PIPE_WRITE 1
int pi[2]; /* input pipe */
int po[2]; /* output pipe */
if (pipe(pi) == -1) err(EXIT_FAILURE, "pipe");
if (pipe(po) == -1) err(EXIT_FAILURE, "pipe");
switch (fork()) {
case -1:
err(EXIT_FAILURE, "fork");
case 0: /* client program */
/* close non-using ends of pipes */
if (close(pi[PIPE_READ]) == -1) err(EXIT_FAILURE, "close");
if (close(po[PIPE_WRITE]) == -1) err(EXIT_FAILURE, "close");
/* move pipe end to ucspi defined fd numbers */
if (dup2(po[PIPE_READ], READ_FD) == -1)
err(EXIT_FAILURE, "dup2");
if (dup2(pi[PIPE_WRITE], WRITE_FD) == -1)
err(EXIT_FAILURE, "dup2");
if (close(po[PIPE_READ]) == -1) err(EXIT_FAILURE, "close");
if (close(pi[PIPE_WRITE]) == -1) err(EXIT_FAILURE, "close");
execv(prog, argv);
err(EXIT_FAILURE, "execve");
default: break; /* parent */
}
/* close non-using ends of pipes */
if (close(pi[PIPE_WRITE]) == -1) err(EXIT_FAILURE, "close");
if (close(po[PIPE_READ]) == -1) err(EXIT_FAILURE, "close");
int in = pi[PIPE_READ];
int out = po[PIPE_WRITE];
/* communication loop */
for (;;) {
int ret;
char buf[BUFSIZ];
ssize_t sn = 0;
fd_set readfds;
FD_ZERO(&readfds);
FD_SET(in, &readfds);
FD_SET(READ_FD, &readfds);
int max_fd = MAX(in, READ_FD);
ret = select(max_fd+1, &readfds, NULL, NULL, NULL);
if (ret == -1)
err(EXIT_FAILURE, "select");
if (FD_ISSET(READ_FD, &readfds)) {
do {
again:
sn = tls_read(cctx, buf, sizeof buf);
if (sn == TLS_WANT_POLLIN ||
sn == TLS_WANT_POLLOUT)
goto again;
if (sn == -1)
goto err;
if (sn == 0)
return EXIT_SUCCESS;
if (write(out, buf, sn) == -1)
err(EXIT_FAILURE, "write()");
} while (sn == sizeof buf);
} else if (FD_ISSET(in, &readfds)) {
if ((sn = read(in, buf, sizeof buf)) == -1)
err(EXIT_FAILURE, "read()");
if (sn == 0) /* EOF from inside */
goto out;
/* XXX: unable to detect disconnect here */
if (tls_write(cctx, buf, sn) == -1)
goto err;
}
}
out:
tls_close(cctx);
return EXIT_SUCCESS;
err:
while ((e = ERR_get_error())) {
ERR_error_string(e, buf);
fprintf(stderr, " %s\n", buf);
}
errx(EXIT_FAILURE, "tls_error: %s", tls_error(cctx));
errx(EXIT_FAILURE, "tls_error: %s", tls_error(tls));
}
