mirror of
https://github.com/ivabus/pantry
synced 2024-11-26 18:25:08 +03:00
fix(rustls) (#1022)
* fix(rustls) updated tests * we should provide llvm asan
This commit is contained in:
parent
b764300106
commit
f62e152b10
4 changed files with 60 additions and 64 deletions
|
@ -34,7 +34,7 @@ int
|
|||
make_conn(const char *hostname, const char *port)
|
||||
{
|
||||
int sockfd = 0;
|
||||
enum crustls_demo_result result = 0;
|
||||
enum demo_result result = 0;
|
||||
struct addrinfo *getaddrinfo_output = NULL, hints;
|
||||
|
||||
memset(&hints, 0, sizeof(hints));
|
||||
|
@ -64,7 +64,7 @@ make_conn(const char *hostname, const char *port)
|
|||
goto cleanup;
|
||||
}
|
||||
result = nonblock(sockfd);
|
||||
if(result != CRUSTLS_DEMO_OK) {
|
||||
if(result != DEMO_OK) {
|
||||
return 1;
|
||||
}
|
||||
|
||||
|
@ -85,13 +85,13 @@ cleanup:
|
|||
* Do one read from the socket, and process all resulting bytes into the
|
||||
* rustls_connection, then copy all plaintext bytes from the session to stdout.
|
||||
* Returns:
|
||||
* - CRUSTLS_DEMO_OK for success
|
||||
* - CRUSTLS_DEMO_AGAIN if we got an EAGAIN or EWOULDBLOCK reading from the
|
||||
* - DEMO_OK for success
|
||||
* - DEMO_AGAIN if we got an EAGAIN or EWOULDBLOCK reading from the
|
||||
* socket
|
||||
* - CRUSTLS_DEMO_EOF if we got EOF
|
||||
* - CRUSTLS_DEMO_ERROR for other errors.
|
||||
* - DEMO_EOF if we got EOF
|
||||
* - DEMO_ERROR for other errors.
|
||||
*/
|
||||
enum crustls_demo_result
|
||||
enum demo_result
|
||||
do_read(struct conndata *conn, struct rustls_connection *rconn)
|
||||
{
|
||||
int err = 1;
|
||||
|
@ -106,21 +106,21 @@ do_read(struct conndata *conn, struct rustls_connection *rconn)
|
|||
fprintf(stderr,
|
||||
"client: reading from socket: EAGAIN or EWOULDBLOCK: %s\n",
|
||||
strerror(errno));
|
||||
return CRUSTLS_DEMO_AGAIN;
|
||||
return DEMO_AGAIN;
|
||||
}
|
||||
else if(err != 0) {
|
||||
fprintf(stderr, "client: reading from socket: errno %d\n", err);
|
||||
return CRUSTLS_DEMO_ERROR;
|
||||
return DEMO_ERROR;
|
||||
}
|
||||
|
||||
result = rustls_connection_process_new_packets(rconn);
|
||||
if(result != RUSTLS_RESULT_OK) {
|
||||
print_error("server", "in process_new_packets", result);
|
||||
return CRUSTLS_DEMO_ERROR;
|
||||
return DEMO_ERROR;
|
||||
}
|
||||
|
||||
result = copy_plaintext_to_buffer(conn);
|
||||
if(result != CRUSTLS_DEMO_EOF) {
|
||||
if(result != DEMO_EOF) {
|
||||
return result;
|
||||
}
|
||||
|
||||
|
@ -131,15 +131,15 @@ do_read(struct conndata *conn, struct rustls_connection *rconn)
|
|||
fprintf(stderr,
|
||||
"client: error: read returned %zu bytes after receiving close_notify\n",
|
||||
n);
|
||||
return CRUSTLS_DEMO_ERROR;
|
||||
return DEMO_ERROR;
|
||||
}
|
||||
else if (signed_n < 0 && errno != EWOULDBLOCK) {
|
||||
fprintf(stderr,
|
||||
"client: error: read returned incorrect error after receiving close_notify: %s\n",
|
||||
strerror(errno));
|
||||
return CRUSTLS_DEMO_ERROR;
|
||||
return DEMO_ERROR;
|
||||
}
|
||||
return CRUSTLS_DEMO_EOF;
|
||||
return DEMO_EOF;
|
||||
}
|
||||
|
||||
static const char *CONTENT_LENGTH = "Content-Length";
|
||||
|
@ -224,13 +224,13 @@ send_request_and_read_response(struct conndata *conn,
|
|||
select awaiting the next bit of data. */
|
||||
for(;;) {
|
||||
result = do_read(conn, rconn);
|
||||
if(result == CRUSTLS_DEMO_AGAIN) {
|
||||
if(result == DEMO_AGAIN) {
|
||||
break;
|
||||
}
|
||||
else if(result == CRUSTLS_DEMO_EOF) {
|
||||
else if(result == DEMO_EOF) {
|
||||
goto drain_plaintext;
|
||||
}
|
||||
else if(result != CRUSTLS_DEMO_OK) {
|
||||
else if(result != DEMO_OK) {
|
||||
goto cleanup;
|
||||
}
|
||||
if(headers_len == 0) {
|
||||
|
@ -276,7 +276,7 @@ send_request_and_read_response(struct conndata *conn,
|
|||
stderr, "client: error in rustls_connection_write_tls: errno %d\n", err);
|
||||
goto cleanup;
|
||||
}
|
||||
if(result == CRUSTLS_DEMO_AGAIN) {
|
||||
if(result == DEMO_AGAIN) {
|
||||
break;
|
||||
}
|
||||
else if(n == 0) {
|
||||
|
@ -291,7 +291,7 @@ send_request_and_read_response(struct conndata *conn,
|
|||
|
||||
drain_plaintext:
|
||||
result = copy_plaintext_to_buffer(conn);
|
||||
if(result != CRUSTLS_DEMO_OK && result != CRUSTLS_DEMO_EOF) {
|
||||
if(result != DEMO_OK && result != DEMO_EOF) {
|
||||
goto cleanup;
|
||||
}
|
||||
fprintf(stderr, "client: writing %zu bytes to stdout\n", conn->data.len);
|
||||
|
@ -373,8 +373,8 @@ verify(void *userdata, const rustls_verify_server_cert_params *params)
|
|||
|
||||
fprintf(stderr,
|
||||
"client: custom certificate verifier called for %.*s\n",
|
||||
(int)params->dns_name.len,
|
||||
params->dns_name.data);
|
||||
(int)params->server_name.len,
|
||||
params->server_name.data);
|
||||
fprintf(stderr, "client: end entity len: %zu\n", params->end_entity_cert_der.len);
|
||||
fprintf(stderr, "client: intermediates:\n");
|
||||
for(i = 0; i < intermediates_len; i++) {
|
||||
|
@ -461,4 +461,4 @@ cleanup:
|
|||
#endif
|
||||
|
||||
return ret;
|
||||
}
|
||||
}
|
|
@ -74,9 +74,9 @@ write_all(int fd, const char *buf, int n)
|
|||
/*
|
||||
* Set a socket to be nonblocking.
|
||||
*
|
||||
* Returns CRUSTLS_DEMO_OK on success, CRUSTLS_DEMO_ERROR on error.
|
||||
* Returns DEMO_OK on success, DEMO_ERROR on error.
|
||||
*/
|
||||
enum crustls_demo_result
|
||||
enum demo_result
|
||||
nonblock(int sockfd)
|
||||
{
|
||||
#ifdef _WIN32
|
||||
|
@ -84,22 +84,22 @@ nonblock(int sockfd)
|
|||
|
||||
if(ioctlsocket(sockfd, FIONBIO, &nonblock) != 0) {
|
||||
perror("Error setting socket nonblocking");
|
||||
return CRUSTLS_DEMO_ERROR;
|
||||
return DEMO_ERROR;
|
||||
}
|
||||
#else
|
||||
int flags;
|
||||
flags = fcntl(sockfd, F_GETFL, 0);
|
||||
if(flags < 0) {
|
||||
perror("getting socket flags");
|
||||
return CRUSTLS_DEMO_ERROR;
|
||||
return DEMO_ERROR;
|
||||
}
|
||||
flags = fcntl(sockfd, F_SETFL, flags | O_NONBLOCK);
|
||||
if(flags < 0) {
|
||||
perror("setting socket nonblocking");
|
||||
return CRUSTLS_DEMO_ERROR;
|
||||
return DEMO_ERROR;
|
||||
}
|
||||
#endif
|
||||
return CRUSTLS_DEMO_OK;
|
||||
return DEMO_OK;
|
||||
}
|
||||
|
||||
int
|
||||
|
@ -183,8 +183,8 @@ bytevec_consume(struct bytevec *vec, size_t n)
|
|||
|
||||
// Ensure there are at least n bytes available between vec->len and
|
||||
// vec->capacity. If this requires reallocating, this may return
|
||||
// CRUSTLS_DEMO_ERROR.
|
||||
enum crustls_demo_result
|
||||
// DEMO_ERROR.
|
||||
enum demo_result
|
||||
bytevec_ensure_available(struct bytevec *vec, size_t n)
|
||||
{
|
||||
size_t available = vec->capacity - vec->len;
|
||||
|
@ -198,12 +198,12 @@ bytevec_ensure_available(struct bytevec *vec, size_t n)
|
|||
newdata = realloc(vec->data, newsize);
|
||||
if(newdata == NULL) {
|
||||
fprintf(stderr, "out of memory trying to get %zu bytes\n", newsize);
|
||||
return CRUSTLS_DEMO_ERROR;
|
||||
return DEMO_ERROR;
|
||||
}
|
||||
vec->data = newdata;
|
||||
vec->capacity = newsize;
|
||||
}
|
||||
return CRUSTLS_DEMO_OK;
|
||||
return DEMO_OK;
|
||||
}
|
||||
|
||||
/**
|
||||
|
@ -217,8 +217,8 @@ copy_plaintext_to_buffer(struct conndata *conn)
|
|||
size_t n;
|
||||
struct rustls_connection *rconn = conn->rconn;
|
||||
|
||||
if(bytevec_ensure_available(&conn->data, 1024) != CRUSTLS_DEMO_OK) {
|
||||
return CRUSTLS_DEMO_ERROR;
|
||||
if(bytevec_ensure_available(&conn->data, 1024) != DEMO_OK) {
|
||||
return DEMO_ERROR;
|
||||
}
|
||||
|
||||
for(;;) {
|
||||
|
@ -227,23 +227,23 @@ copy_plaintext_to_buffer(struct conndata *conn)
|
|||
result = rustls_connection_read(rconn, (uint8_t *)buf, avail, &n);
|
||||
if(result == RUSTLS_RESULT_PLAINTEXT_EMPTY) {
|
||||
/* This is expected. It just means "no more bytes for now." */
|
||||
return CRUSTLS_DEMO_OK;
|
||||
return DEMO_OK;
|
||||
}
|
||||
if(result != RUSTLS_RESULT_OK) {
|
||||
print_error(conn->program_name, "Error in rustls_connection_read", result);
|
||||
return CRUSTLS_DEMO_ERROR;
|
||||
return DEMO_ERROR;
|
||||
}
|
||||
if(n == 0) {
|
||||
fprintf(stderr, "got 0-byte read, cleanly ending connection\n");
|
||||
return CRUSTLS_DEMO_EOF;
|
||||
return DEMO_EOF;
|
||||
}
|
||||
bytevec_consume(&conn->data, n);
|
||||
if(bytevec_ensure_available(&conn->data, 1024) != CRUSTLS_DEMO_OK) {
|
||||
return CRUSTLS_DEMO_ERROR;
|
||||
if(bytevec_ensure_available(&conn->data, 1024) != DEMO_OK) {
|
||||
return DEMO_ERROR;
|
||||
}
|
||||
}
|
||||
|
||||
return CRUSTLS_DEMO_ERROR;
|
||||
return DEMO_ERROR;
|
||||
}
|
||||
|
||||
/**
|
||||
|
@ -352,4 +352,4 @@ log_cb(void *userdata, const struct rustls_log_params *params)
|
|||
struct rustls_str level_str = rustls_log_level_str(params->level);
|
||||
fprintf(stderr, "%s[fd %d][%.*s]: %.*s\n", conn->program_name, conn->fd,
|
||||
(int)level_str.len, level_str.data, (int)params->message.len, params->message.data);
|
||||
}
|
||||
}
|
|
@ -23,12 +23,12 @@ const char * ws_strerror(int err);
|
|||
#endif /* !STDOUT_FILENO */
|
||||
#endif /* _WIN32 */
|
||||
|
||||
enum crustls_demo_result
|
||||
enum demo_result
|
||||
{
|
||||
CRUSTLS_DEMO_OK,
|
||||
CRUSTLS_DEMO_ERROR,
|
||||
CRUSTLS_DEMO_AGAIN,
|
||||
CRUSTLS_DEMO_EOF,
|
||||
DEMO_OK,
|
||||
DEMO_ERROR,
|
||||
DEMO_AGAIN,
|
||||
DEMO_EOF,
|
||||
};
|
||||
|
||||
/* A growable vector of bytes. */
|
||||
|
@ -53,7 +53,7 @@ int
|
|||
write_all(int fd, const char *buf, int n);
|
||||
|
||||
/* Make a socket nonblocking. */
|
||||
enum crustls_demo_result
|
||||
enum demo_result
|
||||
nonblock(int sockfd);
|
||||
|
||||
/* A callback that reads bytes from the network. */
|
||||
|
@ -88,17 +88,17 @@ bytevec_consume(struct bytevec *vec, size_t n);
|
|||
|
||||
/* Ensure there are at least n bytes available between vec->len and
|
||||
* vec->capacity. If this requires reallocating, this may return
|
||||
* CRUSTLS_DEMO_ERROR. */
|
||||
enum crustls_demo_result
|
||||
* DEMO_ERROR. */
|
||||
enum demo_result
|
||||
bytevec_ensure_available(struct bytevec *vec, size_t n);
|
||||
|
||||
/* Read all available bytes from the rustls_connection until EOF.
|
||||
* Note that EOF here indicates "no more bytes until
|
||||
* process_new_packets", not "stream is closed".
|
||||
*
|
||||
* Returns CRUSTLS_DEMO_OK for success,
|
||||
* CRUSTLS_DEMO_ERROR for error,
|
||||
* CRUSTLS_DEMO_EOF for "connection cleanly terminated by peer"
|
||||
* Returns DEMO_OK for success,
|
||||
* DEMO_ERROR for error,
|
||||
* DEMO_EOF for "connection cleanly terminated by peer"
|
||||
*/
|
||||
int
|
||||
copy_plaintext_to_buffer(struct conndata *conn);
|
||||
|
@ -125,4 +125,4 @@ get_first_header_value(const char *headers, size_t headers_len,
|
|||
void
|
||||
log_cb(void *userdata, const struct rustls_log_params *params);
|
||||
|
||||
#endif /* COMMON_H */
|
||||
#endif /* COMMON_H */
|
|
@ -21,22 +21,18 @@ test:
|
|||
tea.xyz/gx/cc: c99
|
||||
openssl.org: '*' # needed for SSL_CERT_FILE
|
||||
env:
|
||||
LIBS:
|
||||
ARGS:
|
||||
- -lrustls
|
||||
- -lc
|
||||
linux:
|
||||
LIBS:
|
||||
- -lgcc_s
|
||||
- -lutil
|
||||
- -lrt
|
||||
ARGS:
|
||||
- -Wl,--gc-sections
|
||||
- -lpthread
|
||||
- -lm
|
||||
- -ldl
|
||||
darwin:
|
||||
LIBS:
|
||||
ARGS:
|
||||
- -Wl,-dead_strip
|
||||
- -framework Security
|
||||
- -liconv
|
||||
- -lSystem
|
||||
- -framework Foundation
|
||||
script: |
|
||||
cc client.c common.c -o client $LIBS
|
||||
cc client.c common.c -o client $ARGS
|
||||
CA_FILE=$SSL_CERT_FILE ./client tea.xyz 443 /
|
Loading…
Reference in a new issue