Message ID | 20230228154420.4141579-1-jack.wgm@gmail.com |
---|---|
State | New |
Headers | show |
Series | [FFmpeg-devel] libavformat/tcp: add local_addr/local_port for network option | expand |
Context | Check | Description |
---|---|---|
yinshiyou/make_loongarch64 | success | Make finished |
yinshiyou/make_fate_loongarch64 | success | Make fate finished |
andriy/make_x86 | success | Make finished |
andriy/make_fate_x86 | success | Make fate finished |
Quoting jackarain (2023-02-28 16:44:20) > -static void customize_fd(void *ctx, int fd) > +static int customize_fd(void *ctx, int fd) > { > TCPContext *s = ctx; > + > + if (s->local_addr || s->local_port) { > + struct addrinfo hints = { 0 }, *ai; > + int ret; > + > + hints.ai_family = AF_UNSPEC; > + hints.ai_socktype = SOCK_STREAM; > + > + ret = getaddrinfo(s->local_addr, s->local_port, &hints, &ai); > + if (ret) { > + av_log(ctx, AV_LOG_ERROR, > + "Failed to getaddrinfo local addr: %s port: %s err: %s\n", > + s->local_addr, s->local_port, gai_strerror(ret)); > + return ret; > + } else { nit: the else clause pointlessly adds an indentation level and serves no useful purpose > + struct addrinfo *cur_ai = ai; > + while (cur_ai) { > + ret = bind(fd, (struct sockaddr *)cur_ai->ai_addr, (int)cur_ai->ai_addrlen); > + if (ret) > + cur_ai = cur_ai->ai_next; > + else > + break; > + } > + freeaddrinfo(ai); > + > + if (ret) { > + av_log(ctx, AV_LOG_ERROR, > + "Failed to bind local addr: %s port: %s err: %s\n", > + s->local_addr, s->local_port, gai_strerror(ret)); > + return ret; > + } > + } > + } > /* Set the socket's send or receive buffer sizes, if specified. > If unspecified or setting fails, system default is used. */ > if (s->recv_buffer_size > 0) { > @@ -97,6 +134,8 @@ static void customize_fd(void *ctx, int fd) > } > } > #endif /* !HAVE_WINSOCK2_H */ > + > + return 0; > } > > /* return non zero if error */ > @@ -129,6 +168,14 @@ static int tcp_open(URLContext *h, const char *uri, int flags) > if (buf == endptr) > s->listen = 1; > } > + if (av_find_info_tag(buf, sizeof(buf), "local_port", p)) { > + av_freep(&s->local_port); > + s->local_port = av_strndup(buf, strlen(buf)); This does memory allocation, so the result should be checked. Also, it av_str_n_dup() gives you no advantages since you call strlen anyway. Just use av_strdup(). Same below.
Ok, thanks. Anton Khirnov <anton@khirnov.net> 于2023年3月1日周三 22:44写道: > Quoting jackarain (2023-02-28 16:44:20) > > -static void customize_fd(void *ctx, int fd) > > +static int customize_fd(void *ctx, int fd) > > { > > TCPContext *s = ctx; > > + > > + if (s->local_addr || s->local_port) { > > + struct addrinfo hints = { 0 }, *ai; > > + int ret; > > + > > + hints.ai_family = AF_UNSPEC; > > + hints.ai_socktype = SOCK_STREAM; > > + > > + ret = getaddrinfo(s->local_addr, s->local_port, &hints, &ai); > > + if (ret) { > > + av_log(ctx, AV_LOG_ERROR, > > + "Failed to getaddrinfo local addr: %s port: %s err: > %s\n", > > + s->local_addr, s->local_port, gai_strerror(ret)); > > + return ret; > > + } else { > > nit: the else clause pointlessly adds an indentation level and serves no > useful purpose > > > + struct addrinfo *cur_ai = ai; > > + while (cur_ai) { > > + ret = bind(fd, (struct sockaddr *)cur_ai->ai_addr, > (int)cur_ai->ai_addrlen); > > + if (ret) > > + cur_ai = cur_ai->ai_next; > > + else > > + break; > > + } > > + freeaddrinfo(ai); > > + > > + if (ret) { > > + av_log(ctx, AV_LOG_ERROR, > > + "Failed to bind local addr: %s port: %s err: %s\n", > > + s->local_addr, s->local_port, gai_strerror(ret)); > > + return ret; > > + } > > + } > > + } > > /* Set the socket's send or receive buffer sizes, if specified. > > If unspecified or setting fails, system default is used. */ > > if (s->recv_buffer_size > 0) { > > @@ -97,6 +134,8 @@ static void customize_fd(void *ctx, int fd) > > } > > } > > #endif /* !HAVE_WINSOCK2_H */ > > + > > + return 0; > > } > > > > /* return non zero if error */ > > @@ -129,6 +168,14 @@ static int tcp_open(URLContext *h, const char *uri, > int flags) > > if (buf == endptr) > > s->listen = 1; > > } > > + if (av_find_info_tag(buf, sizeof(buf), "local_port", p)) { > > + av_freep(&s->local_port); > > + s->local_port = av_strndup(buf, strlen(buf)); > > This does memory allocation, so the result should be checked. Also, it > av_str_n_dup() gives you no advantages since you call strlen anyway. > Just use av_strdup(). Same below. > > -- > Anton Khirnov >
diff --git a/doc/protocols.texi b/doc/protocols.texi index 21ae6181a0..b3fad55591 100644 --- a/doc/protocols.texi +++ b/doc/protocols.texi @@ -1882,6 +1882,12 @@ The list of supported options follows. Listen for an incoming connection. 0 disables listen, 1 enables listen in single client mode, 2 enables listen in multi-client mode. Default value is 0. +@item local_addr=@var{addr} +Local IP address of a network interface used for tcp socket connect. + +@item local_port=@var{port} +Local port used for tcp socket connect. + @item timeout=@var{microseconds} Set raise error timeout, expressed in microseconds. diff --git a/libavformat/network.c b/libavformat/network.c index 21e20b3e9a..de8b14be82 100644 --- a/libavformat/network.c +++ b/libavformat/network.c @@ -356,7 +356,7 @@ struct ConnectionAttempt { static int start_connect_attempt(struct ConnectionAttempt *attempt, struct addrinfo **ptr, int timeout_ms, URLContext *h, - void (*customize_fd)(void *, int), void *customize_ctx) + int (*customize_fd)(void *, int), void *customize_ctx) { struct addrinfo *ai = *ptr; int ret; @@ -371,8 +371,14 @@ static int start_connect_attempt(struct ConnectionAttempt *attempt, ff_socket_nonblock(attempt->fd, 1); - if (customize_fd) - customize_fd(customize_ctx, attempt->fd); + if (customize_fd) { + ret = customize_fd(customize_ctx, attempt->fd); + if (ret) { + closesocket(attempt->fd); + attempt->fd = -1; + return ret; + } + } while ((ret = connect(attempt->fd, ai->ai_addr, ai->ai_addrlen))) { ret = ff_neterrno(); @@ -402,7 +408,7 @@ static int start_connect_attempt(struct ConnectionAttempt *attempt, int ff_connect_parallel(struct addrinfo *addrs, int timeout_ms_per_address, int parallel, URLContext *h, int *fd, - void (*customize_fd)(void *, int), void *customize_ctx) + int (*customize_fd)(void *, int), void *customize_ctx) { struct ConnectionAttempt attempts[3]; struct pollfd pfd[3]; diff --git a/libavformat/network.h b/libavformat/network.h index 71c49a73fb..8a8cbe672e 100644 --- a/libavformat/network.h +++ b/libavformat/network.h @@ -336,6 +336,6 @@ void ff_log_net_error(void *ctx, int level, const char* prefix); */ int ff_connect_parallel(struct addrinfo *addrs, int timeout_ms_per_address, int parallel, URLContext *h, int *fd, - void (*customize_fd)(void *, int), void *customize_ctx); + int (*customize_fd)(void *, int), void *customize_ctx); #endif /* AVFORMAT_NETWORK_H */ diff --git a/libavformat/tcp.c b/libavformat/tcp.c index a11ccbb913..f9900c56a7 100644 --- a/libavformat/tcp.c +++ b/libavformat/tcp.c @@ -36,6 +36,8 @@ typedef struct TCPContext { const AVClass *class; int fd; int listen; + char *local_port; + char *local_addr; int open_timeout; int rw_timeout; int listen_timeout; @@ -52,6 +54,8 @@ typedef struct TCPContext { #define E AV_OPT_FLAG_ENCODING_PARAM static const AVOption options[] = { { "listen", "Listen for incoming connections", OFFSET(listen), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 2, .flags = D|E }, + { "local_port", "Local port", OFFSET(local_port), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, .flags = D|E }, + { "local_addr", "Local address", OFFSET(local_addr), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, .flags = D|E }, { "timeout", "set timeout (in microseconds) of socket I/O operations", OFFSET(rw_timeout), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, INT_MAX, .flags = D|E }, { "listen_timeout", "Connection awaiting timeout (in milliseconds)", OFFSET(listen_timeout), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, INT_MAX, .flags = D|E }, { "send_buffer_size", "Socket send buffer size (in bytes)", OFFSET(send_buffer_size), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, INT_MAX, .flags = D|E }, @@ -70,9 +74,42 @@ static const AVClass tcp_class = { .version = LIBAVUTIL_VERSION_INT, }; -static void customize_fd(void *ctx, int fd) +static int customize_fd(void *ctx, int fd) { TCPContext *s = ctx; + + if (s->local_addr || s->local_port) { + struct addrinfo hints = { 0 }, *ai; + int ret; + + hints.ai_family = AF_UNSPEC; + hints.ai_socktype = SOCK_STREAM; + + ret = getaddrinfo(s->local_addr, s->local_port, &hints, &ai); + if (ret) { + av_log(ctx, AV_LOG_ERROR, + "Failed to getaddrinfo local addr: %s port: %s err: %s\n", + s->local_addr, s->local_port, gai_strerror(ret)); + return ret; + } else { + struct addrinfo *cur_ai = ai; + while (cur_ai) { + ret = bind(fd, (struct sockaddr *)cur_ai->ai_addr, (int)cur_ai->ai_addrlen); + if (ret) + cur_ai = cur_ai->ai_next; + else + break; + } + freeaddrinfo(ai); + + if (ret) { + av_log(ctx, AV_LOG_ERROR, + "Failed to bind local addr: %s port: %s err: %s\n", + s->local_addr, s->local_port, gai_strerror(ret)); + return ret; + } + } + } /* Set the socket's send or receive buffer sizes, if specified. If unspecified or setting fails, system default is used. */ if (s->recv_buffer_size > 0) { @@ -97,6 +134,8 @@ static void customize_fd(void *ctx, int fd) } } #endif /* !HAVE_WINSOCK2_H */ + + return 0; } /* return non zero if error */ @@ -129,6 +168,14 @@ static int tcp_open(URLContext *h, const char *uri, int flags) if (buf == endptr) s->listen = 1; } + if (av_find_info_tag(buf, sizeof(buf), "local_port", p)) { + av_freep(&s->local_port); + s->local_port = av_strndup(buf, strlen(buf)); + } + if (av_find_info_tag(buf, sizeof(buf), "local_addr", p)) { + av_freep(&s->local_addr); + s->local_addr = av_strndup(buf, strlen(buf)); + } if (av_find_info_tag(buf, sizeof(buf), "timeout", p)) { s->rw_timeout = strtol(buf, NULL, 10); }
Signed-off-by: jackarain <jack.wgm@gmail.com> --- doc/protocols.texi | 6 ++++++ libavformat/network.c | 14 +++++++++---- libavformat/network.h | 2 +- libavformat/tcp.c | 49 ++++++++++++++++++++++++++++++++++++++++++- 4 files changed, 65 insertions(+), 6 deletions(-)