mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2024-12-23 12:43:46 +02:00
udp: pick params from AVOptions also
This updated version does not deviate from previous behavior on default value of 'buffer_size' I skipped porting 'sources', 'block' options for now as they're parsed seriously. So i added TODO remarks. Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
This commit is contained in:
parent
0da0b5e0f8
commit
aefed6ca87
@ -32,6 +32,8 @@
|
||||
#include "libavutil/fifo.h"
|
||||
#include "libavutil/intreadwrite.h"
|
||||
#include "libavutil/avstring.h"
|
||||
#include "libavutil/opt.h"
|
||||
#include "libavutil/log.h"
|
||||
#include "internal.h"
|
||||
#include "network.h"
|
||||
#include "os_support.h"
|
||||
@ -54,6 +56,7 @@
|
||||
#define UDP_MAX_PKT_SIZE 65536
|
||||
|
||||
typedef struct {
|
||||
const AVClass *class;
|
||||
int udp_fd;
|
||||
int ttl;
|
||||
int buffer_size;
|
||||
@ -77,8 +80,36 @@ typedef struct {
|
||||
#endif
|
||||
uint8_t tmp[UDP_MAX_PKT_SIZE+4];
|
||||
int remaining_in_dg;
|
||||
char *local_addr;
|
||||
int packet_size;
|
||||
int timeout;
|
||||
} UDPContext;
|
||||
|
||||
#define OFFSET(x) offsetof(UDPContext, x)
|
||||
#define D AV_OPT_FLAG_DECODING_PARAM
|
||||
#define E AV_OPT_FLAG_ENCODING_PARAM
|
||||
static const AVOption options[] = {
|
||||
{"buffer_size", "Socket buffer size in bytes", OFFSET(buffer_size), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, D|E },
|
||||
{"localport", "Set local port to bind to", OFFSET(local_port), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, D|E },
|
||||
{"localaddr", "Choose local IP address", OFFSET(local_addr), AV_OPT_TYPE_STRING, {.str = ""}, 0, 0, D|E },
|
||||
{"pkt_size", "Set size of UDP packets", OFFSET(packet_size), AV_OPT_TYPE_INT, {.i64 = 1472}, 0, INT_MAX, D|E },
|
||||
{"reuse", "Explicitly allow or disallow reusing UDP sockets", OFFSET(reuse_socket), AV_OPT_TYPE_INT, {.i64 = 0}, 0, 1, D|E },
|
||||
{"ttl", "Set the time to live value (for multicast only)", OFFSET(ttl), AV_OPT_TYPE_INT, {.i64 = 16}, 0, INT_MAX, E },
|
||||
{"connect", "Should connect() be called on socket", OFFSET(is_connected), AV_OPT_TYPE_INT, {.i64 = 0}, 0, 1, D|E },
|
||||
/* TODO 'sources', 'block' option */
|
||||
{"fifo_size", "Set the UDP receiving circular buffer size, expressed as a number of packets with size of 188 bytes", OFFSET(circular_buffer_size), AV_OPT_TYPE_INT, {.i64 = 7*4096}, 0, INT_MAX, D },
|
||||
{"overrun_nonfatal", "Survive in case of UDP receiving circular buffer overrun", OFFSET(overrun_nonfatal), AV_OPT_TYPE_INT, {.i64 = 0}, 0, 1, D },
|
||||
{"timeout", "In read mode: if no data arrived in more than this time interval, raise error", OFFSET(timeout), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, D },
|
||||
{NULL}
|
||||
};
|
||||
|
||||
static const AVClass udp_context_class = {
|
||||
.class_name = "udp",
|
||||
.item_name = av_default_item_name,
|
||||
.option = options,
|
||||
.version = LIBAVUTIL_VERSION_INT,
|
||||
};
|
||||
|
||||
static void log_net_error(void *ctx, int level, const char* prefix)
|
||||
{
|
||||
char errbuf[100];
|
||||
@ -475,14 +506,10 @@ static int udp_open(URLContext *h, const char *uri, int flags)
|
||||
char *sources[32];
|
||||
|
||||
h->is_streamed = 1;
|
||||
h->max_packet_size = 1472;
|
||||
|
||||
is_output = !(flags & AVIO_FLAG_READ);
|
||||
|
||||
s->ttl = 16;
|
||||
s->buffer_size = is_output ? UDP_TX_BUF_SIZE : UDP_MAX_PKT_SIZE;
|
||||
|
||||
s->circular_buffer_size = 7*188*4096;
|
||||
if (!s->buffer_size) /* if not set explicitly */
|
||||
s->buffer_size = is_output ? UDP_TX_BUF_SIZE : UDP_MAX_PKT_SIZE;
|
||||
|
||||
p = strchr(uri, '?');
|
||||
if (p) {
|
||||
@ -512,7 +539,7 @@ static int udp_open(URLContext *h, const char *uri, int flags)
|
||||
s->local_port = strtol(buf, NULL, 10);
|
||||
}
|
||||
if (av_find_info_tag(buf, sizeof(buf), "pkt_size", p)) {
|
||||
h->max_packet_size = strtol(buf, NULL, 10);
|
||||
s->packet_size = strtol(buf, NULL, 10);
|
||||
}
|
||||
if (av_find_info_tag(buf, sizeof(buf), "buffer_size", p)) {
|
||||
s->buffer_size = strtol(buf, NULL, 10);
|
||||
@ -521,7 +548,7 @@ static int udp_open(URLContext *h, const char *uri, int flags)
|
||||
s->is_connected = strtol(buf, NULL, 10);
|
||||
}
|
||||
if (av_find_info_tag(buf, sizeof(buf), "fifo_size", p)) {
|
||||
s->circular_buffer_size = strtol(buf, NULL, 10)*188;
|
||||
s->circular_buffer_size = strtol(buf, NULL, 10);
|
||||
if (!HAVE_PTHREAD_CANCEL)
|
||||
av_log(h, AV_LOG_WARNING,
|
||||
"'circular_buffer_size' option was set but it is not supported "
|
||||
@ -550,8 +577,12 @@ static int udp_open(URLContext *h, const char *uri, int flags)
|
||||
}
|
||||
}
|
||||
if (!is_output && av_find_info_tag(buf, sizeof(buf), "timeout", p))
|
||||
h->rw_timeout = strtol(buf, NULL, 10);
|
||||
s->timeout = strtol(buf, NULL, 10);
|
||||
}
|
||||
/* handling needed to support options picking from both AVOption and URL */
|
||||
s->circular_buffer_size *= 188;
|
||||
h->max_packet_size = s->packet_size;
|
||||
h->rw_timeout = s->timeout;
|
||||
|
||||
/* fill the dest addr */
|
||||
av_url_split(NULL, 0, NULL, 0, hostname, sizeof(hostname), &port, NULL, 0, uri);
|
||||
@ -568,7 +599,7 @@ static int udp_open(URLContext *h, const char *uri, int flags)
|
||||
|
||||
if ((s->is_multicast || !s->local_port) && (h->flags & AVIO_FLAG_READ))
|
||||
s->local_port = port;
|
||||
udp_fd = udp_socket_create(s, &my_addr, &len, localaddr);
|
||||
udp_fd = udp_socket_create(s, &my_addr, &len, localaddr[0] ? localaddr : s->local_addr);
|
||||
if (udp_fd < 0)
|
||||
goto fail;
|
||||
|
||||
@ -804,5 +835,6 @@ URLProtocol ff_udp_protocol = {
|
||||
.url_close = udp_close,
|
||||
.url_get_file_handle = udp_get_file_handle,
|
||||
.priv_data_size = sizeof(UDPContext),
|
||||
.priv_data_class = &udp_context_class,
|
||||
.flags = URL_PROTOCOL_FLAG_NETWORK,
|
||||
};
|
||||
|
Loading…
Reference in New Issue
Block a user