You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
248 lines
6.4 KiB
248 lines
6.4 KiB
/* test.c */ |
|
/* $Id: test.c,v 1.1 2001/09/17 19:06:59 bodo Exp $ */ |
|
|
|
#define L_PORT 9999 |
|
#define C_PORT 443 |
|
|
|
#include <arpa/inet.h> |
|
#include <assert.h> |
|
#include <errno.h> |
|
#include <fcntl.h> |
|
#include <netinet/in.h> |
|
#include <netinet/tcp.h> |
|
#include <stdlib.h> |
|
#include <stdio.h> |
|
#include <string.h> |
|
#include <sys/select.h> |
|
#include <sys/socket.h> |
|
#include <unistd.h> |
|
|
|
#include "test.h" |
|
#include "easy-tls.h" |
|
|
|
void test_process_init(int fd, int client_p, void *apparg) |
|
{ |
|
fprintf(stderr, |
|
"test_process_init(fd = %d, client_p = %d, apparg = %p)\n", fd, |
|
client_p, apparg); |
|
} |
|
|
|
void test_errflush(int child_p, char *errbuf, size_t num, void *apparg) |
|
{ |
|
fputs(errbuf, stderr); |
|
} |
|
|
|
int main(int argc, char *argv[]) |
|
{ |
|
int s, fd, r; |
|
FILE *conn_in; |
|
FILE *conn_out; |
|
char buf[256]; |
|
SSL_CTX *ctx; |
|
int client_p = 0; |
|
int port; |
|
int tls = 0; |
|
char infobuf[TLS_INFO_SIZE + 1]; |
|
|
|
if (argc > 1 && argv[1][0] == '-') { |
|
fputs("Usage: test [port] -- server\n" |
|
" test num.num.num.num [port] -- client\n", stderr); |
|
exit(1); |
|
} |
|
|
|
if (argc > 1) { |
|
if (strchr(argv[1], '.')) { |
|
client_p = 1; |
|
} |
|
} |
|
|
|
fputs(client_p ? "Client\n" : "Server\n", stderr); |
|
|
|
{ |
|
struct tls_create_ctx_args a = tls_create_ctx_defaultargs(); |
|
a.client_p = client_p; |
|
a.certificate_file = "cert.pem"; |
|
a.key_file = "cert.pem"; |
|
a.ca_file = "cacerts.pem"; |
|
|
|
ctx = tls_create_ctx(a, NULL); |
|
if (ctx == NULL) |
|
exit(1); |
|
} |
|
|
|
s = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); |
|
if (s == -1) { |
|
perror("socket"); |
|
exit(1); |
|
} |
|
|
|
if (client_p) { |
|
struct sockaddr_in addr; |
|
size_t addr_len = sizeof addr; |
|
|
|
addr.sin_family = AF_INET; |
|
assert(argc > 1); |
|
if (argc > 2) |
|
sscanf(argv[2], "%d", &port); |
|
else |
|
port = C_PORT; |
|
addr.sin_port = htons(port); |
|
addr.sin_addr.s_addr = inet_addr(argv[1]); |
|
|
|
r = connect(s, &addr, addr_len); |
|
if (r != 0) { |
|
perror("connect"); |
|
exit(1); |
|
} |
|
fd = s; |
|
fprintf(stderr, "Connect (fd = %d).\n", fd); |
|
} else { |
|
/* server */ |
|
{ |
|
int i = 1; |
|
|
|
r = setsockopt(s, SOL_SOCKET, SO_REUSEADDR, (void *)&i, sizeof i); |
|
if (r == -1) { |
|
perror("setsockopt"); |
|
exit(1); |
|
} |
|
} |
|
|
|
{ |
|
struct sockaddr_in addr; |
|
size_t addr_len = sizeof addr; |
|
|
|
if (argc > 1) |
|
sscanf(argv[1], "%d", &port); |
|
else |
|
port = L_PORT; |
|
addr.sin_family = AF_INET; |
|
addr.sin_port = htons(port); |
|
addr.sin_addr.s_addr = INADDR_ANY; |
|
|
|
r = bind(s, &addr, addr_len); |
|
if (r != 0) { |
|
perror("bind"); |
|
exit(1); |
|
} |
|
} |
|
|
|
r = listen(s, 1); |
|
if (r == -1) { |
|
perror("listen"); |
|
exit(1); |
|
} |
|
|
|
fprintf(stderr, "Listening at port %i.\n", port); |
|
|
|
fd = accept(s, NULL, 0); |
|
if (fd == -1) { |
|
perror("accept"); |
|
exit(1); |
|
} |
|
|
|
fprintf(stderr, "Accept (fd = %d).\n", fd); |
|
} |
|
|
|
conn_in = fdopen(fd, "r"); |
|
if (conn_in == NULL) { |
|
perror("fdopen"); |
|
exit(1); |
|
} |
|
conn_out = fdopen(fd, "w"); |
|
if (conn_out == NULL) { |
|
perror("fdopen"); |
|
exit(1); |
|
} |
|
|
|
setvbuf(conn_in, NULL, _IOLBF, 256); |
|
setvbuf(conn_out, NULL, _IOLBF, 256); |
|
|
|
while (fgets(buf, sizeof buf, stdin) != NULL) { |
|
if (buf[0] == 'W') { |
|
fprintf(conn_out, "%.*s\r\n", (int)(strlen(buf + 1) - 1), |
|
buf + 1); |
|
fprintf(stderr, ">>> %.*s\n", (int)(strlen(buf + 1) - 1), |
|
buf + 1); |
|
} else if (buf[0] == 'C') { |
|
fprintf(stderr, "Closing.\n"); |
|
fclose(conn_in); |
|
fclose(conn_out); |
|
exit(0); |
|
} else if (buf[0] == 'R') { |
|
int lines = 0; |
|
|
|
sscanf(buf + 1, "%d", &lines); |
|
do { |
|
if (fgets(buf, sizeof buf, conn_in) == NULL) { |
|
if (ferror(conn_in)) { |
|
fprintf(stderr, "ERROR\n"); |
|
exit(1); |
|
} |
|
fprintf(stderr, "CLOSED\n"); |
|
return 0; |
|
} |
|
fprintf(stderr, "<<< %s", buf); |
|
} while (--lines > 0); |
|
} else if (buf[0] == 'T') { |
|
int infofd; |
|
|
|
tls++; |
|
{ |
|
struct tls_start_proxy_args a = tls_start_proxy_defaultargs(); |
|
a.fd = fd; |
|
a.client_p = client_p; |
|
a.ctx = ctx; |
|
a.infofd = &infofd; |
|
r = tls_start_proxy(a, NULL); |
|
} |
|
assert(r != 1); |
|
if (r != 0) { |
|
fprintf(stderr, "tls_start_proxy failed: %d\n", r); |
|
switch (r) { |
|
case -1: |
|
fputs("socketpair", stderr); |
|
break; |
|
case 2: |
|
fputs("FD_SETSIZE exceeded", stderr); |
|
break; |
|
case -3: |
|
fputs("pipe", stderr); |
|
break; |
|
case -4: |
|
fputs("fork", stderr); |
|
break; |
|
case -5: |
|
fputs("dup2", stderr); |
|
break; |
|
default: |
|
fputs("?", stderr); |
|
} |
|
if (r < 0) |
|
perror(""); |
|
else |
|
fputc('\n', stderr); |
|
exit(1); |
|
} |
|
|
|
r = read(infofd, infobuf, sizeof infobuf - 1); |
|
if (r > 0) { |
|
const char *info = infobuf; |
|
const char *eol; |
|
|
|
infobuf[r] = '\0'; |
|
while ((eol = strchr(info, '\n')) != NULL) { |
|
fprintf(stderr, "+++ `%.*s'\n", eol - info, info); |
|
info = eol + 1; |
|
} |
|
close(infofd); |
|
} |
|
} else { |
|
fprintf(stderr, "W... write line to network\n" |
|
"R[n] read line (n lines) from network\n" |
|
"C close\n" |
|
"T start %sTLS proxy\n", tls ? "another " : ""); |
|
} |
|
} |
|
return 0; |
|
}
|
|
|