684 lines
20 KiB
C
684 lines
20 KiB
C
|
/***************************************************************************
|
||
|
|
* _ _ ____ _
|
||
|
|
* Project ___| | | | _ \| |
|
||
|
|
* / __| | | | |_) | |
|
||
|
|
* | (__| |_| | _ <| |___
|
||
|
|
* \___|\___/|_| \_\_____|
|
||
|
|
*
|
||
|
|
* Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al.
|
||
|
|
*
|
||
|
|
* This software is licensed as described in the file COPYING, which
|
||
|
|
* you should have received as part of this distribution. The terms
|
||
|
|
* are also available at https://curl.se/docs/copyright.html.
|
||
|
|
*
|
||
|
|
* You may opt to use, copy, modify, merge, publish, distribute and/or sell
|
||
|
|
* copies of the Software, and permit persons to whom the Software is
|
||
|
|
* furnished to do so, under the terms of the COPYING file.
|
||
|
|
*
|
||
|
|
* This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
|
||
|
|
* KIND, either express or implied.
|
||
|
|
*
|
||
|
|
* SPDX-License-Identifier: curl
|
||
|
|
*
|
||
|
|
***************************************************************************/
|
||
|
|
|
||
|
|
/* WIP, experimental: use recvmmsg() on linux
|
||
|
|
* we have no configure check, yet
|
||
|
|
* and also it is only available for _GNU_SOURCE, which
|
||
|
|
* we do not use otherwise.
|
||
|
|
#define HAVE_SENDMMSG
|
||
|
|
*/
|
||
|
|
#if defined(HAVE_SENDMMSG)
|
||
|
|
#define _GNU_SOURCE
|
||
|
|
#include <sys/socket.h>
|
||
|
|
#undef _GNU_SOURCE
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#include "curl_setup.h"
|
||
|
|
|
||
|
|
#ifdef HAVE_FCNTL_H
|
||
|
|
#include <fcntl.h>
|
||
|
|
#endif
|
||
|
|
#include "urldata.h"
|
||
|
|
#include "bufq.h"
|
||
|
|
#include "dynbuf.h"
|
||
|
|
#include "cfilters.h"
|
||
|
|
#include "curl_trc.h"
|
||
|
|
#include "curl_msh3.h"
|
||
|
|
#include "curl_ngtcp2.h"
|
||
|
|
#include "curl_osslq.h"
|
||
|
|
#include "curl_quiche.h"
|
||
|
|
#include "rand.h"
|
||
|
|
#include "vquic.h"
|
||
|
|
#include "vquic_int.h"
|
||
|
|
#include "strerror.h"
|
||
|
|
|
||
|
|
/* The last 3 #include files should be in this order */
|
||
|
|
#include "curl_printf.h"
|
||
|
|
#include "curl_memory.h"
|
||
|
|
#include "memdebug.h"
|
||
|
|
|
||
|
|
|
||
|
|
#ifdef ENABLE_QUIC
|
||
|
|
|
||
|
|
#ifdef O_BINARY
|
||
|
|
#define QLOGMODE O_WRONLY|O_CREAT|O_BINARY
|
||
|
|
#else
|
||
|
|
#define QLOGMODE O_WRONLY|O_CREAT
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#define NW_CHUNK_SIZE (64 * 1024)
|
||
|
|
#define NW_SEND_CHUNKS 2
|
||
|
|
|
||
|
|
|
||
|
|
void Curl_quic_ver(char *p, size_t len)
|
||
|
|
{
|
||
|
|
#if defined(USE_NGTCP2) && defined(USE_NGHTTP3)
|
||
|
|
Curl_ngtcp2_ver(p, len);
|
||
|
|
#elif defined(USE_OPENSSL_QUIC) && defined(USE_NGHTTP3)
|
||
|
|
Curl_osslq_ver(p, len);
|
||
|
|
#elif defined(USE_QUICHE)
|
||
|
|
Curl_quiche_ver(p, len);
|
||
|
|
#elif defined(USE_MSH3)
|
||
|
|
Curl_msh3_ver(p, len);
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
CURLcode vquic_ctx_init(struct cf_quic_ctx *qctx)
|
||
|
|
{
|
||
|
|
Curl_bufq_init2(&qctx->sendbuf, NW_CHUNK_SIZE, NW_SEND_CHUNKS,
|
||
|
|
BUFQ_OPT_SOFT_LIMIT);
|
||
|
|
#if defined(__linux__) && defined(UDP_SEGMENT) && defined(HAVE_SENDMSG)
|
||
|
|
qctx->no_gso = FALSE;
|
||
|
|
#else
|
||
|
|
qctx->no_gso = TRUE;
|
||
|
|
#endif
|
||
|
|
#ifdef DEBUGBUILD
|
||
|
|
{
|
||
|
|
char *p = getenv("CURL_DBG_QUIC_WBLOCK");
|
||
|
|
if(p) {
|
||
|
|
long l = strtol(p, NULL, 10);
|
||
|
|
if(l >= 0 && l <= 100)
|
||
|
|
qctx->wblock_percent = (int)l;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
vquic_ctx_update_time(qctx);
|
||
|
|
|
||
|
|
return CURLE_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
void vquic_ctx_free(struct cf_quic_ctx *qctx)
|
||
|
|
{
|
||
|
|
Curl_bufq_free(&qctx->sendbuf);
|
||
|
|
}
|
||
|
|
|
||
|
|
void vquic_ctx_update_time(struct cf_quic_ctx *qctx)
|
||
|
|
{
|
||
|
|
qctx->last_op = Curl_now();
|
||
|
|
}
|
||
|
|
|
||
|
|
static CURLcode send_packet_no_gso(struct Curl_cfilter *cf,
|
||
|
|
struct Curl_easy *data,
|
||
|
|
struct cf_quic_ctx *qctx,
|
||
|
|
const uint8_t *pkt, size_t pktlen,
|
||
|
|
size_t gsolen, size_t *psent);
|
||
|
|
|
||
|
|
static CURLcode do_sendmsg(struct Curl_cfilter *cf,
|
||
|
|
struct Curl_easy *data,
|
||
|
|
struct cf_quic_ctx *qctx,
|
||
|
|
const uint8_t *pkt, size_t pktlen, size_t gsolen,
|
||
|
|
size_t *psent)
|
||
|
|
{
|
||
|
|
#ifdef HAVE_SENDMSG
|
||
|
|
struct iovec msg_iov;
|
||
|
|
struct msghdr msg = {0};
|
||
|
|
ssize_t sent;
|
||
|
|
#if defined(__linux__) && defined(UDP_SEGMENT)
|
||
|
|
uint8_t msg_ctrl[32];
|
||
|
|
struct cmsghdr *cm;
|
||
|
|
#endif
|
||
|
|
|
||
|
|
*psent = 0;
|
||
|
|
msg_iov.iov_base = (uint8_t *)pkt;
|
||
|
|
msg_iov.iov_len = pktlen;
|
||
|
|
msg.msg_iov = &msg_iov;
|
||
|
|
msg.msg_iovlen = 1;
|
||
|
|
|
||
|
|
#if defined(__linux__) && defined(UDP_SEGMENT)
|
||
|
|
if(pktlen > gsolen) {
|
||
|
|
/* Only set this, when we need it. macOS, for example,
|
||
|
|
* does not seem to like a msg_control of length 0. */
|
||
|
|
msg.msg_control = msg_ctrl;
|
||
|
|
assert(sizeof(msg_ctrl) >= CMSG_SPACE(sizeof(uint16_t)));
|
||
|
|
msg.msg_controllen = CMSG_SPACE(sizeof(uint16_t));
|
||
|
|
cm = CMSG_FIRSTHDR(&msg);
|
||
|
|
cm->cmsg_level = SOL_UDP;
|
||
|
|
cm->cmsg_type = UDP_SEGMENT;
|
||
|
|
cm->cmsg_len = CMSG_LEN(sizeof(uint16_t));
|
||
|
|
*(uint16_t *)(void *)CMSG_DATA(cm) = gsolen & 0xffff;
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
|
||
|
|
while((sent = sendmsg(qctx->sockfd, &msg, 0)) == -1 && SOCKERRNO == EINTR)
|
||
|
|
;
|
||
|
|
|
||
|
|
if(sent == -1) {
|
||
|
|
switch(SOCKERRNO) {
|
||
|
|
case EAGAIN:
|
||
|
|
#if EAGAIN != EWOULDBLOCK
|
||
|
|
case EWOULDBLOCK:
|
||
|
|
#endif
|
||
|
|
return CURLE_AGAIN;
|
||
|
|
case EMSGSIZE:
|
||
|
|
/* UDP datagram is too large; caused by PMTUD. Just let it be lost. */
|
||
|
|
break;
|
||
|
|
case EIO:
|
||
|
|
if(pktlen > gsolen) {
|
||
|
|
/* GSO failure */
|
||
|
|
failf(data, "sendmsg() returned %zd (errno %d); disable GSO", sent,
|
||
|
|
SOCKERRNO);
|
||
|
|
qctx->no_gso = TRUE;
|
||
|
|
return send_packet_no_gso(cf, data, qctx, pkt, pktlen, gsolen, psent);
|
||
|
|
}
|
||
|
|
FALLTHROUGH();
|
||
|
|
default:
|
||
|
|
failf(data, "sendmsg() returned %zd (errno %d)", sent, SOCKERRNO);
|
||
|
|
return CURLE_SEND_ERROR;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
assert(pktlen == (size_t)sent);
|
||
|
|
}
|
||
|
|
#else
|
||
|
|
ssize_t sent;
|
||
|
|
(void)gsolen;
|
||
|
|
|
||
|
|
*psent = 0;
|
||
|
|
|
||
|
|
while((sent = send(qctx->sockfd,
|
||
|
|
(const char *)pkt, (SEND_TYPE_ARG3)pktlen, 0)) == -1 &&
|
||
|
|
SOCKERRNO == EINTR)
|
||
|
|
;
|
||
|
|
|
||
|
|
if(sent == -1) {
|
||
|
|
if(SOCKERRNO == EAGAIN || SOCKERRNO == EWOULDBLOCK) {
|
||
|
|
return CURLE_AGAIN;
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
failf(data, "send() returned %zd (errno %d)", sent, SOCKERRNO);
|
||
|
|
if(SOCKERRNO != EMSGSIZE) {
|
||
|
|
return CURLE_SEND_ERROR;
|
||
|
|
}
|
||
|
|
/* UDP datagram is too large; caused by PMTUD. Just let it be
|
||
|
|
lost. */
|
||
|
|
}
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
(void)cf;
|
||
|
|
*psent = pktlen;
|
||
|
|
|
||
|
|
return CURLE_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
static CURLcode send_packet_no_gso(struct Curl_cfilter *cf,
|
||
|
|
struct Curl_easy *data,
|
||
|
|
struct cf_quic_ctx *qctx,
|
||
|
|
const uint8_t *pkt, size_t pktlen,
|
||
|
|
size_t gsolen, size_t *psent)
|
||
|
|
{
|
||
|
|
const uint8_t *p, *end = pkt + pktlen;
|
||
|
|
size_t sent;
|
||
|
|
|
||
|
|
*psent = 0;
|
||
|
|
|
||
|
|
for(p = pkt; p < end; p += gsolen) {
|
||
|
|
size_t len = CURLMIN(gsolen, (size_t)(end - p));
|
||
|
|
CURLcode curlcode = do_sendmsg(cf, data, qctx, p, len, len, &sent);
|
||
|
|
if(curlcode != CURLE_OK) {
|
||
|
|
return curlcode;
|
||
|
|
}
|
||
|
|
*psent += sent;
|
||
|
|
}
|
||
|
|
|
||
|
|
return CURLE_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
static CURLcode vquic_send_packets(struct Curl_cfilter *cf,
|
||
|
|
struct Curl_easy *data,
|
||
|
|
struct cf_quic_ctx *qctx,
|
||
|
|
const uint8_t *pkt, size_t pktlen,
|
||
|
|
size_t gsolen, size_t *psent)
|
||
|
|
{
|
||
|
|
CURLcode result;
|
||
|
|
#ifdef DEBUGBUILD
|
||
|
|
/* simulate network blocking/partial writes */
|
||
|
|
if(qctx->wblock_percent > 0) {
|
||
|
|
unsigned char c;
|
||
|
|
Curl_rand(data, &c, 1);
|
||
|
|
if(c >= ((100-qctx->wblock_percent)*256/100)) {
|
||
|
|
CURL_TRC_CF(data, cf, "vquic_flush() simulate EWOULDBLOCK");
|
||
|
|
return CURLE_AGAIN;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
if(qctx->no_gso && pktlen > gsolen) {
|
||
|
|
result = send_packet_no_gso(cf, data, qctx, pkt, pktlen, gsolen, psent);
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
result = do_sendmsg(cf, data, qctx, pkt, pktlen, gsolen, psent);
|
||
|
|
}
|
||
|
|
if(!result)
|
||
|
|
qctx->last_io = qctx->last_op;
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
CURLcode vquic_flush(struct Curl_cfilter *cf, struct Curl_easy *data,
|
||
|
|
struct cf_quic_ctx *qctx)
|
||
|
|
{
|
||
|
|
const unsigned char *buf;
|
||
|
|
size_t blen, sent;
|
||
|
|
CURLcode result;
|
||
|
|
size_t gsolen;
|
||
|
|
|
||
|
|
while(Curl_bufq_peek(&qctx->sendbuf, &buf, &blen)) {
|
||
|
|
gsolen = qctx->gsolen;
|
||
|
|
if(qctx->split_len) {
|
||
|
|
gsolen = qctx->split_gsolen;
|
||
|
|
if(blen > qctx->split_len)
|
||
|
|
blen = qctx->split_len;
|
||
|
|
}
|
||
|
|
|
||
|
|
result = vquic_send_packets(cf, data, qctx, buf, blen, gsolen, &sent);
|
||
|
|
CURL_TRC_CF(data, cf, "vquic_send(len=%zu, gso=%zu) -> %d, sent=%zu",
|
||
|
|
blen, gsolen, result, sent);
|
||
|
|
if(result) {
|
||
|
|
if(result == CURLE_AGAIN) {
|
||
|
|
Curl_bufq_skip(&qctx->sendbuf, sent);
|
||
|
|
if(qctx->split_len)
|
||
|
|
qctx->split_len -= sent;
|
||
|
|
}
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
Curl_bufq_skip(&qctx->sendbuf, sent);
|
||
|
|
if(qctx->split_len)
|
||
|
|
qctx->split_len -= sent;
|
||
|
|
}
|
||
|
|
return CURLE_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
CURLcode vquic_send(struct Curl_cfilter *cf, struct Curl_easy *data,
|
||
|
|
struct cf_quic_ctx *qctx, size_t gsolen)
|
||
|
|
{
|
||
|
|
qctx->gsolen = gsolen;
|
||
|
|
return vquic_flush(cf, data, qctx);
|
||
|
|
}
|
||
|
|
|
||
|
|
CURLcode vquic_send_tail_split(struct Curl_cfilter *cf, struct Curl_easy *data,
|
||
|
|
struct cf_quic_ctx *qctx, size_t gsolen,
|
||
|
|
size_t tail_len, size_t tail_gsolen)
|
||
|
|
{
|
||
|
|
DEBUGASSERT(Curl_bufq_len(&qctx->sendbuf) > tail_len);
|
||
|
|
qctx->split_len = Curl_bufq_len(&qctx->sendbuf) - tail_len;
|
||
|
|
qctx->split_gsolen = gsolen;
|
||
|
|
qctx->gsolen = tail_gsolen;
|
||
|
|
CURL_TRC_CF(data, cf, "vquic_send_tail_split: [%zu gso=%zu][%zu gso=%zu]",
|
||
|
|
qctx->split_len, qctx->split_gsolen,
|
||
|
|
tail_len, qctx->gsolen);
|
||
|
|
return vquic_flush(cf, data, qctx);
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifdef HAVE_SENDMMSG
|
||
|
|
static CURLcode recvmmsg_packets(struct Curl_cfilter *cf,
|
||
|
|
struct Curl_easy *data,
|
||
|
|
struct cf_quic_ctx *qctx,
|
||
|
|
size_t max_pkts,
|
||
|
|
vquic_recv_pkt_cb *recv_cb, void *userp)
|
||
|
|
{
|
||
|
|
#define MMSG_NUM 64
|
||
|
|
struct iovec msg_iov[MMSG_NUM];
|
||
|
|
struct mmsghdr mmsg[MMSG_NUM];
|
||
|
|
uint8_t bufs[MMSG_NUM][2*1024];
|
||
|
|
struct sockaddr_storage remote_addr[MMSG_NUM];
|
||
|
|
size_t total_nread, pkts;
|
||
|
|
int mcount, i, n;
|
||
|
|
char errstr[STRERROR_LEN];
|
||
|
|
CURLcode result = CURLE_OK;
|
||
|
|
|
||
|
|
DEBUGASSERT(max_pkts > 0);
|
||
|
|
pkts = 0;
|
||
|
|
total_nread = 0;
|
||
|
|
while(pkts < max_pkts) {
|
||
|
|
n = (int)CURLMIN(MMSG_NUM, max_pkts);
|
||
|
|
memset(&mmsg, 0, sizeof(mmsg));
|
||
|
|
for(i = 0; i < n; ++i) {
|
||
|
|
msg_iov[i].iov_base = bufs[i];
|
||
|
|
msg_iov[i].iov_len = (int)sizeof(bufs[i]);
|
||
|
|
mmsg[i].msg_hdr.msg_iov = &msg_iov[i];
|
||
|
|
mmsg[i].msg_hdr.msg_iovlen = 1;
|
||
|
|
mmsg[i].msg_hdr.msg_name = &remote_addr[i];
|
||
|
|
mmsg[i].msg_hdr.msg_namelen = sizeof(remote_addr[i]);
|
||
|
|
}
|
||
|
|
|
||
|
|
while((mcount = recvmmsg(qctx->sockfd, mmsg, n, 0, NULL)) == -1 &&
|
||
|
|
SOCKERRNO == EINTR)
|
||
|
|
;
|
||
|
|
if(mcount == -1) {
|
||
|
|
if(SOCKERRNO == EAGAIN || SOCKERRNO == EWOULDBLOCK) {
|
||
|
|
CURL_TRC_CF(data, cf, "ingress, recvmmsg -> EAGAIN");
|
||
|
|
goto out;
|
||
|
|
}
|
||
|
|
if(!cf->connected && SOCKERRNO == ECONNREFUSED) {
|
||
|
|
const char *r_ip = NULL;
|
||
|
|
int r_port = 0;
|
||
|
|
Curl_cf_socket_peek(cf->next, data, NULL, NULL,
|
||
|
|
&r_ip, &r_port, NULL, NULL);
|
||
|
|
failf(data, "QUIC: connection to %s port %u refused",
|
||
|
|
r_ip, r_port);
|
||
|
|
result = CURLE_COULDNT_CONNECT;
|
||
|
|
goto out;
|
||
|
|
}
|
||
|
|
Curl_strerror(SOCKERRNO, errstr, sizeof(errstr));
|
||
|
|
failf(data, "QUIC: recvmsg() unexpectedly returned %d (errno=%d; %s)",
|
||
|
|
mcount, SOCKERRNO, errstr);
|
||
|
|
result = CURLE_RECV_ERROR;
|
||
|
|
goto out;
|
||
|
|
}
|
||
|
|
|
||
|
|
CURL_TRC_CF(data, cf, "recvmmsg() -> %d packets", mcount);
|
||
|
|
pkts += mcount;
|
||
|
|
for(i = 0; i < mcount; ++i) {
|
||
|
|
total_nread += mmsg[i].msg_len;
|
||
|
|
result = recv_cb(bufs[i], mmsg[i].msg_len,
|
||
|
|
mmsg[i].msg_hdr.msg_name, mmsg[i].msg_hdr.msg_namelen,
|
||
|
|
0, userp);
|
||
|
|
if(result)
|
||
|
|
goto out;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
out:
|
||
|
|
if(total_nread || result)
|
||
|
|
CURL_TRC_CF(data, cf, "recvd %zu packets with %zu bytes -> %d",
|
||
|
|
pkts, total_nread, result);
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
#elif defined(HAVE_SENDMSG)
|
||
|
|
static CURLcode recvmsg_packets(struct Curl_cfilter *cf,
|
||
|
|
struct Curl_easy *data,
|
||
|
|
struct cf_quic_ctx *qctx,
|
||
|
|
size_t max_pkts,
|
||
|
|
vquic_recv_pkt_cb *recv_cb, void *userp)
|
||
|
|
{
|
||
|
|
struct iovec msg_iov;
|
||
|
|
struct msghdr msg;
|
||
|
|
uint8_t buf[64*1024];
|
||
|
|
struct sockaddr_storage remote_addr;
|
||
|
|
size_t total_nread, pkts;
|
||
|
|
ssize_t nread;
|
||
|
|
char errstr[STRERROR_LEN];
|
||
|
|
CURLcode result = CURLE_OK;
|
||
|
|
|
||
|
|
msg_iov.iov_base = buf;
|
||
|
|
msg_iov.iov_len = (int)sizeof(buf);
|
||
|
|
|
||
|
|
memset(&msg, 0, sizeof(msg));
|
||
|
|
msg.msg_iov = &msg_iov;
|
||
|
|
msg.msg_iovlen = 1;
|
||
|
|
|
||
|
|
DEBUGASSERT(max_pkts > 0);
|
||
|
|
for(pkts = 0, total_nread = 0; pkts < max_pkts;) {
|
||
|
|
msg.msg_name = &remote_addr;
|
||
|
|
msg.msg_namelen = sizeof(remote_addr);
|
||
|
|
while((nread = recvmsg(qctx->sockfd, &msg, 0)) == -1 &&
|
||
|
|
SOCKERRNO == EINTR)
|
||
|
|
;
|
||
|
|
if(nread == -1) {
|
||
|
|
if(SOCKERRNO == EAGAIN || SOCKERRNO == EWOULDBLOCK) {
|
||
|
|
goto out;
|
||
|
|
}
|
||
|
|
if(!cf->connected && SOCKERRNO == ECONNREFUSED) {
|
||
|
|
const char *r_ip = NULL;
|
||
|
|
int r_port = 0;
|
||
|
|
Curl_cf_socket_peek(cf->next, data, NULL, NULL,
|
||
|
|
&r_ip, &r_port, NULL, NULL);
|
||
|
|
failf(data, "QUIC: connection to %s port %u refused",
|
||
|
|
r_ip, r_port);
|
||
|
|
result = CURLE_COULDNT_CONNECT;
|
||
|
|
goto out;
|
||
|
|
}
|
||
|
|
Curl_strerror(SOCKERRNO, errstr, sizeof(errstr));
|
||
|
|
failf(data, "QUIC: recvmsg() unexpectedly returned %zd (errno=%d; %s)",
|
||
|
|
nread, SOCKERRNO, errstr);
|
||
|
|
result = CURLE_RECV_ERROR;
|
||
|
|
goto out;
|
||
|
|
}
|
||
|
|
|
||
|
|
++pkts;
|
||
|
|
total_nread += (size_t)nread;
|
||
|
|
result = recv_cb(buf, (size_t)nread, msg.msg_name, msg.msg_namelen,
|
||
|
|
0, userp);
|
||
|
|
if(result)
|
||
|
|
goto out;
|
||
|
|
}
|
||
|
|
|
||
|
|
out:
|
||
|
|
if(total_nread || result)
|
||
|
|
CURL_TRC_CF(data, cf, "recvd %zu packets with %zu bytes -> %d",
|
||
|
|
pkts, total_nread, result);
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
#else /* HAVE_SENDMMSG || HAVE_SENDMSG */
|
||
|
|
static CURLcode recvfrom_packets(struct Curl_cfilter *cf,
|
||
|
|
struct Curl_easy *data,
|
||
|
|
struct cf_quic_ctx *qctx,
|
||
|
|
size_t max_pkts,
|
||
|
|
vquic_recv_pkt_cb *recv_cb, void *userp)
|
||
|
|
{
|
||
|
|
uint8_t buf[64*1024];
|
||
|
|
int bufsize = (int)sizeof(buf);
|
||
|
|
struct sockaddr_storage remote_addr;
|
||
|
|
socklen_t remote_addrlen = sizeof(remote_addr);
|
||
|
|
size_t total_nread, pkts;
|
||
|
|
ssize_t nread;
|
||
|
|
char errstr[STRERROR_LEN];
|
||
|
|
CURLcode result = CURLE_OK;
|
||
|
|
|
||
|
|
DEBUGASSERT(max_pkts > 0);
|
||
|
|
for(pkts = 0, total_nread = 0; pkts < max_pkts;) {
|
||
|
|
while((nread = recvfrom(qctx->sockfd, (char *)buf, bufsize, 0,
|
||
|
|
(struct sockaddr *)&remote_addr,
|
||
|
|
&remote_addrlen)) == -1 &&
|
||
|
|
SOCKERRNO == EINTR)
|
||
|
|
;
|
||
|
|
if(nread == -1) {
|
||
|
|
if(SOCKERRNO == EAGAIN || SOCKERRNO == EWOULDBLOCK) {
|
||
|
|
CURL_TRC_CF(data, cf, "ingress, recvfrom -> EAGAIN");
|
||
|
|
goto out;
|
||
|
|
}
|
||
|
|
if(!cf->connected && SOCKERRNO == ECONNREFUSED) {
|
||
|
|
const char *r_ip = NULL;
|
||
|
|
int r_port = 0;
|
||
|
|
Curl_cf_socket_peek(cf->next, data, NULL, NULL,
|
||
|
|
&r_ip, &r_port, NULL, NULL);
|
||
|
|
failf(data, "QUIC: connection to %s port %u refused",
|
||
|
|
r_ip, r_port);
|
||
|
|
result = CURLE_COULDNT_CONNECT;
|
||
|
|
goto out;
|
||
|
|
}
|
||
|
|
Curl_strerror(SOCKERRNO, errstr, sizeof(errstr));
|
||
|
|
failf(data, "QUIC: recvfrom() unexpectedly returned %zd (errno=%d; %s)",
|
||
|
|
nread, SOCKERRNO, errstr);
|
||
|
|
result = CURLE_RECV_ERROR;
|
||
|
|
goto out;
|
||
|
|
}
|
||
|
|
|
||
|
|
++pkts;
|
||
|
|
total_nread += (size_t)nread;
|
||
|
|
result = recv_cb(buf, (size_t)nread, &remote_addr, remote_addrlen,
|
||
|
|
0, userp);
|
||
|
|
if(result)
|
||
|
|
goto out;
|
||
|
|
}
|
||
|
|
|
||
|
|
out:
|
||
|
|
if(total_nread || result)
|
||
|
|
CURL_TRC_CF(data, cf, "recvd %zu packets with %zu bytes -> %d",
|
||
|
|
pkts, total_nread, result);
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
#endif /* !HAVE_SENDMMSG && !HAVE_SENDMSG */
|
||
|
|
|
||
|
|
CURLcode vquic_recv_packets(struct Curl_cfilter *cf,
|
||
|
|
struct Curl_easy *data,
|
||
|
|
struct cf_quic_ctx *qctx,
|
||
|
|
size_t max_pkts,
|
||
|
|
vquic_recv_pkt_cb *recv_cb, void *userp)
|
||
|
|
{
|
||
|
|
CURLcode result;
|
||
|
|
#if defined(HAVE_SENDMMSG)
|
||
|
|
result = recvmmsg_packets(cf, data, qctx, max_pkts, recv_cb, userp);
|
||
|
|
#elif defined(HAVE_SENDMSG)
|
||
|
|
result = recvmsg_packets(cf, data, qctx, max_pkts, recv_cb, userp);
|
||
|
|
#else
|
||
|
|
result = recvfrom_packets(cf, data, qctx, max_pkts, recv_cb, userp);
|
||
|
|
#endif
|
||
|
|
if(!result) {
|
||
|
|
if(!qctx->got_first_byte) {
|
||
|
|
qctx->got_first_byte = TRUE;
|
||
|
|
qctx->first_byte_at = qctx->last_op;
|
||
|
|
}
|
||
|
|
qctx->last_io = qctx->last_op;
|
||
|
|
}
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
* If the QLOGDIR environment variable is set, open and return a file
|
||
|
|
* descriptor to write the log to.
|
||
|
|
*
|
||
|
|
* This function returns error if something failed outside of failing to
|
||
|
|
* create the file. Open file success is deemed by seeing if the returned fd
|
||
|
|
* is != -1.
|
||
|
|
*/
|
||
|
|
CURLcode Curl_qlogdir(struct Curl_easy *data,
|
||
|
|
unsigned char *scid,
|
||
|
|
size_t scidlen,
|
||
|
|
int *qlogfdp)
|
||
|
|
{
|
||
|
|
const char *qlog_dir = getenv("QLOGDIR");
|
||
|
|
*qlogfdp = -1;
|
||
|
|
if(qlog_dir) {
|
||
|
|
struct dynbuf fname;
|
||
|
|
CURLcode result;
|
||
|
|
unsigned int i;
|
||
|
|
Curl_dyn_init(&fname, DYN_QLOG_NAME);
|
||
|
|
result = Curl_dyn_add(&fname, qlog_dir);
|
||
|
|
if(!result)
|
||
|
|
result = Curl_dyn_add(&fname, "/");
|
||
|
|
for(i = 0; (i < scidlen) && !result; i++) {
|
||
|
|
char hex[3];
|
||
|
|
msnprintf(hex, 3, "%02x", scid[i]);
|
||
|
|
result = Curl_dyn_add(&fname, hex);
|
||
|
|
}
|
||
|
|
if(!result)
|
||
|
|
result = Curl_dyn_add(&fname, ".sqlog");
|
||
|
|
|
||
|
|
if(!result) {
|
||
|
|
int qlogfd = open(Curl_dyn_ptr(&fname), QLOGMODE,
|
||
|
|
data->set.new_file_perms);
|
||
|
|
if(qlogfd != -1)
|
||
|
|
*qlogfdp = qlogfd;
|
||
|
|
}
|
||
|
|
Curl_dyn_free(&fname);
|
||
|
|
if(result)
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
return CURLE_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
CURLcode Curl_cf_quic_create(struct Curl_cfilter **pcf,
|
||
|
|
struct Curl_easy *data,
|
||
|
|
struct connectdata *conn,
|
||
|
|
const struct Curl_addrinfo *ai,
|
||
|
|
int transport)
|
||
|
|
{
|
||
|
|
(void)transport;
|
||
|
|
DEBUGASSERT(transport == TRNSPRT_QUIC);
|
||
|
|
#if defined(USE_NGTCP2) && defined(USE_NGHTTP3)
|
||
|
|
return Curl_cf_ngtcp2_create(pcf, data, conn, ai);
|
||
|
|
#elif defined(USE_OPENSSL_QUIC) && defined(USE_NGHTTP3)
|
||
|
|
return Curl_cf_osslq_create(pcf, data, conn, ai);
|
||
|
|
#elif defined(USE_QUICHE)
|
||
|
|
return Curl_cf_quiche_create(pcf, data, conn, ai);
|
||
|
|
#elif defined(USE_MSH3)
|
||
|
|
return Curl_cf_msh3_create(pcf, data, conn, ai);
|
||
|
|
#else
|
||
|
|
*pcf = NULL;
|
||
|
|
(void)data;
|
||
|
|
(void)conn;
|
||
|
|
(void)ai;
|
||
|
|
return CURLE_NOT_BUILT_IN;
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
bool Curl_conn_is_http3(const struct Curl_easy *data,
|
||
|
|
const struct connectdata *conn,
|
||
|
|
int sockindex)
|
||
|
|
{
|
||
|
|
#if defined(USE_NGTCP2) && defined(USE_NGHTTP3)
|
||
|
|
return Curl_conn_is_ngtcp2(data, conn, sockindex);
|
||
|
|
#elif defined(USE_OPENSSL_QUIC) && defined(USE_NGHTTP3)
|
||
|
|
return Curl_conn_is_osslq(data, conn, sockindex);
|
||
|
|
#elif defined(USE_QUICHE)
|
||
|
|
return Curl_conn_is_quiche(data, conn, sockindex);
|
||
|
|
#elif defined(USE_MSH3)
|
||
|
|
return Curl_conn_is_msh3(data, conn, sockindex);
|
||
|
|
#else
|
||
|
|
return ((conn->handler->protocol & PROTO_FAMILY_HTTP) &&
|
||
|
|
(conn->httpversion == 30));
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
CURLcode Curl_conn_may_http3(struct Curl_easy *data,
|
||
|
|
const struct connectdata *conn)
|
||
|
|
{
|
||
|
|
if(conn->transport == TRNSPRT_UNIX) {
|
||
|
|
/* cannot do QUIC over a unix domain socket */
|
||
|
|
return CURLE_QUIC_CONNECT_ERROR;
|
||
|
|
}
|
||
|
|
if(!(conn->handler->flags & PROTOPT_SSL)) {
|
||
|
|
failf(data, "HTTP/3 requested for non-HTTPS URL");
|
||
|
|
return CURLE_URL_MALFORMAT;
|
||
|
|
}
|
||
|
|
#ifndef CURL_DISABLE_PROXY
|
||
|
|
if(conn->bits.socksproxy) {
|
||
|
|
failf(data, "HTTP/3 is not supported over a SOCKS proxy");
|
||
|
|
return CURLE_URL_MALFORMAT;
|
||
|
|
}
|
||
|
|
if(conn->bits.httpproxy && conn->bits.tunnel_proxy) {
|
||
|
|
failf(data, "HTTP/3 is not supported over a HTTP proxy");
|
||
|
|
return CURLE_URL_MALFORMAT;
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
return CURLE_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
#else /* ENABLE_QUIC */
|
||
|
|
|
||
|
|
CURLcode Curl_conn_may_http3(struct Curl_easy *data,
|
||
|
|
const struct connectdata *conn)
|
||
|
|
{
|
||
|
|
(void)conn;
|
||
|
|
(void)data;
|
||
|
|
DEBUGF(infof(data, "QUIC is not supported in this build"));
|
||
|
|
return CURLE_NOT_BUILT_IN;
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif /* !ENABLE_QUIC */
|