123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449 |
- #include <sys/poll.h>
- #include <openssl/ssl.h>
- typedef struct app_conn_st {
- SSL *ssl;
- BIO *ssl_bio;
- int rx_need_tx, tx_need_rx;
- } APP_CONN;
- SSL_CTX *create_ssl_ctx(void)
- {
- SSL_CTX *ctx;
- #ifdef USE_QUIC
- ctx = SSL_CTX_new(OSSL_QUIC_client_method());
- #else
- ctx = SSL_CTX_new(TLS_client_method());
- #endif
- if (ctx == NULL)
- return NULL;
-
- SSL_CTX_set_verify(ctx, SSL_VERIFY_PEER, NULL);
-
- if (SSL_CTX_set_default_verify_paths(ctx) == 0) {
- SSL_CTX_free(ctx);
- return NULL;
- }
- return ctx;
- }
- APP_CONN *new_conn(SSL_CTX *ctx, const char *hostname)
- {
- APP_CONN *conn;
- BIO *out, *buf;
- SSL *ssl = NULL;
- const char *bare_hostname;
- #ifdef USE_QUIC
- static const unsigned char alpn[] = {5, 'd', 'u', 'm', 'm', 'y'};
- #endif
- conn = calloc(1, sizeof(APP_CONN));
- if (conn == NULL)
- return NULL;
- out = BIO_new_ssl_connect(ctx);
- if (out == NULL) {
- free(conn);
- return NULL;
- }
- if (BIO_get_ssl(out, &ssl) == 0) {
- BIO_free_all(out);
- free(conn);
- return NULL;
- }
-
- buf = BIO_new(BIO_f_buffer());
- if (buf == NULL) {
- BIO_free_all(out);
- free(conn);
- return NULL;
- }
- BIO_push(out, buf);
- if (BIO_set_conn_hostname(out, hostname) == 0) {
- BIO_free_all(out);
- free(conn);
- return NULL;
- }
-
- bare_hostname = BIO_get_conn_hostname(out);
- if (bare_hostname == NULL) {
- BIO_free_all(out);
- free(conn);
- return NULL;
- }
-
- if (SSL_set1_host(ssl, bare_hostname) <= 0) {
- BIO_free_all(out);
- free(conn);
- return NULL;
- }
- #ifdef USE_QUIC
-
- if (SSL_set_alpn_protos(ssl, alpn, sizeof(alpn))) {
-
- BIO_free_all(out);
- return NULL;
- }
- #endif
-
- BIO_set_nbio(out, 1);
- conn->ssl_bio = out;
- return conn;
- }
- int tx(APP_CONN *conn, const void *buf, int buf_len)
- {
- int l;
- conn->tx_need_rx = 0;
- l = BIO_write(conn->ssl_bio, buf, buf_len);
- if (l <= 0) {
- if (BIO_should_retry(conn->ssl_bio)) {
- conn->tx_need_rx = BIO_should_read(conn->ssl_bio);
- return -2;
- } else {
- return -1;
- }
- }
- return l;
- }
- int rx(APP_CONN *conn, void *buf, int buf_len)
- {
- int l;
- conn->rx_need_tx = 0;
- l = BIO_read(conn->ssl_bio, buf, buf_len);
- if (l <= 0) {
- if (BIO_should_retry(conn->ssl_bio)) {
- conn->rx_need_tx = BIO_should_write(conn->ssl_bio);
- return -2;
- } else {
- return -1;
- }
- }
- return l;
- }
- int get_conn_fd(APP_CONN *conn)
- {
- #ifdef USE_QUIC
- BIO_POLL_DESCRIPTOR d;
- if (!BIO_get_rpoll_descriptor(conn->ssl_bio, &d))
- return -1;
- return d.value.fd;
- #else
- return BIO_get_fd(conn->ssl_bio, NULL);
- #endif
- }
- int get_conn_pending_tx(APP_CONN *conn)
- {
- #ifdef USE_QUIC
- return (SSL_net_read_desired(conn->ssl) ? POLLIN : 0)
- | (SSL_net_write_desired(conn->ssl) ? POLLOUT : 0)
- | POLLERR;
- #else
- return (conn->tx_need_rx ? POLLIN : 0) | POLLOUT | POLLERR;
- #endif
- }
- int get_conn_pending_rx(APP_CONN *conn)
- {
- #ifdef USE_QUIC
- return get_conn_pending_tx(conn);
- #else
- return (conn->rx_need_tx ? POLLOUT : 0) | POLLIN | POLLERR;
- #endif
- }
- #ifdef USE_QUIC
- static inline int timeval_to_ms(const struct timeval *t);
- int get_conn_pump_timeout(APP_CONN *conn)
- {
- struct timeval tv;
- int is_infinite;
- if (!SSL_get_event_timeout(conn->ssl, &tv, &is_infinite))
- return -1;
- return is_infinite ? -1 : timeval_to_ms(&tv);
- }
- void pump(APP_CONN *conn)
- {
- SSL_handle_events(conn->ssl);
- }
- #endif
- void teardown(APP_CONN *conn)
- {
- BIO_free_all(conn->ssl_bio);
- free(conn);
- }
- void teardown_ctx(SSL_CTX *ctx)
- {
- SSL_CTX_free(ctx);
- }
- #include <sys/time.h>
- static inline void ms_to_timeval(struct timeval *t, int ms)
- {
- t->tv_sec = ms < 0 ? -1 : ms/1000;
- t->tv_usec = ms < 0 ? 0 : (ms%1000)*1000;
- }
- static inline int timeval_to_ms(const struct timeval *t)
- {
- return t->tv_sec*1000 + t->tv_usec/1000;
- }
- int main(int argc, char **argv)
- {
- static char tx_msg[384], host_port[300];
- const char *tx_p = tx_msg;
- char rx_buf[2048];
- int res = 1, l, tx_len;
- #ifdef USE_QUIC
- struct timeval timeout;
- #else
- int timeout = 2000 ;
- #endif
- APP_CONN *conn = NULL;
- SSL_CTX *ctx = NULL;
- #ifdef USE_QUIC
- ms_to_timeval(&timeout, 2000);
- #endif
- if (argc < 3) {
- fprintf(stderr, "usage: %s host port\n", argv[0]);
- goto fail;
- }
- snprintf(host_port, sizeof(host_port), "%s:%s", argv[1], argv[2]);
- tx_len = snprintf(tx_msg, sizeof(tx_msg),
- "GET / HTTP/1.0\r\nHost: %s\r\n\r\n", argv[1]);
- ctx = create_ssl_ctx();
- if (ctx == NULL) {
- fprintf(stderr, "cannot create SSL context\n");
- goto fail;
- }
- conn = new_conn(ctx, host_port);
- if (conn == NULL) {
- fprintf(stderr, "cannot establish connection\n");
- goto fail;
- }
-
- while (tx_len != 0) {
- l = tx(conn, tx_p, tx_len);
- if (l > 0) {
- tx_p += l;
- tx_len -= l;
- } else if (l == -1) {
- fprintf(stderr, "tx error\n");
- } else if (l == -2) {
- #ifdef USE_QUIC
- struct timeval start, now, deadline, t;
- #endif
- struct pollfd pfd = {0};
- #ifdef USE_QUIC
- ms_to_timeval(&t, get_conn_pump_timeout(conn));
- if (t.tv_sec < 0 || timercmp(&t, &timeout, >))
- t = timeout;
- gettimeofday(&start, NULL);
- timeradd(&start, &timeout, &deadline);
- #endif
- pfd.fd = get_conn_fd(conn);
- pfd.events = get_conn_pending_tx(conn);
- #ifdef USE_QUIC
- if (poll(&pfd, 1, timeval_to_ms(&t)) == 0)
- #else
- if (poll(&pfd, 1, timeout) == 0)
- #endif
- {
- #ifdef USE_QUIC
- pump(conn);
- gettimeofday(&now, NULL);
- if (timercmp(&now, &deadline, >=))
- #endif
- {
- fprintf(stderr, "tx timeout\n");
- goto fail;
- }
- }
- }
- }
-
- for (;;) {
- l = rx(conn, rx_buf, sizeof(rx_buf));
- if (l > 0) {
- fwrite(rx_buf, 1, l, stdout);
- } else if (l == -1) {
- break;
- } else if (l == -2) {
- #ifdef USE_QUIC
- struct timeval start, now, deadline, t;
- #endif
- struct pollfd pfd = {0};
- #ifdef USE_QUIC
- ms_to_timeval(&t, get_conn_pump_timeout(conn));
- if (t.tv_sec < 0 || timercmp(&t, &timeout, >))
- t = timeout;
- gettimeofday(&start, NULL);
- timeradd(&start, &timeout, &deadline);
- #endif
- pfd.fd = get_conn_fd(conn);
- pfd.events = get_conn_pending_rx(conn);
- #ifdef USE_QUIC
- if (poll(&pfd, 1, timeval_to_ms(&t)) == 0)
- #else
- if (poll(&pfd, 1, timeout) == 0)
- #endif
- {
- #ifdef USE_QUIC
- pump(conn);
- gettimeofday(&now, NULL);
- if (timercmp(&now, &deadline, >=))
- #endif
- {
- fprintf(stderr, "rx timeout\n");
- goto fail;
- }
- }
- }
- }
- res = 0;
- fail:
- if (conn != NULL)
- teardown(conn);
- if (ctx != NULL)
- teardown_ctx(ctx);
- return res;
- }
|