27 #define _DEFAULT_SOURCE 46 #include "TargetConditionals.h" 55 #define UDPLITE_SEND_CSCOV 10 56 #define UDPLITE_RECV_CSCOV 11 59 #ifndef IPPROTO_UDPLITE 60 #define IPPROTO_UDPLITE 136 63 #if HAVE_PTHREAD_CANCEL 67 #ifndef HAVE_PTHREAD_CANCEL 68 #define HAVE_PTHREAD_CANCEL 0 71 #ifndef IPV6_ADD_MEMBERSHIP 72 #define IPV6_ADD_MEMBERSHIP IPV6_JOIN_GROUP 73 #define IPV6_DROP_MEMBERSHIP IPV6_LEAVE_GROUP 76 #define UDP_TX_BUF_SIZE 32768 77 #define UDP_MAX_PKT_SIZE 65536 78 #define UDP_HEADER_SIZE 8 103 #if HAVE_PTHREAD_CANCEL 118 #define OFFSET(x) offsetof(UDPContext, x) 119 #define D AV_OPT_FLAG_DECODING_PARAM 120 #define E AV_OPT_FLAG_ENCODING_PARAM 161 av_log(ctx, level,
"%s: %s\n", prefix, errbuf);
165 struct sockaddr *addr)
167 #ifdef IP_MULTICAST_TTL 168 if (addr->sa_family == AF_INET) {
169 if (setsockopt(sockfd, IPPROTO_IP, IP_MULTICAST_TTL, &mcastTTL,
sizeof(mcastTTL)) < 0) {
175 #if defined(IPPROTO_IPV6) && defined(IPV6_MULTICAST_HOPS) 176 if (addr->sa_family == AF_INET6) {
177 if (setsockopt(sockfd, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, &mcastTTL,
sizeof(mcastTTL)) < 0) {
188 #ifdef IP_ADD_MEMBERSHIP 189 if (addr->sa_family == AF_INET) {
192 mreq.imr_multiaddr.s_addr = ((
struct sockaddr_in *)addr)->sin_addr.s_addr;
194 mreq.imr_interface= ((
struct sockaddr_in *)local_addr)->sin_addr;
196 mreq.imr_interface.s_addr= INADDR_ANY;
197 if (setsockopt(sockfd, IPPROTO_IP, IP_ADD_MEMBERSHIP, (
const void *)&mreq,
sizeof(mreq)) < 0) {
203 #if HAVE_STRUCT_IPV6_MREQ && defined(IPPROTO_IPV6) 204 if (addr->sa_family == AF_INET6) {
205 struct ipv6_mreq mreq6;
207 memcpy(&mreq6.ipv6mr_multiaddr, &(((
struct sockaddr_in6 *)addr)->sin6_addr),
sizeof(
struct in6_addr));
208 mreq6.ipv6mr_interface= 0;
220 #ifdef IP_DROP_MEMBERSHIP 221 if (addr->sa_family == AF_INET) {
224 mreq.imr_multiaddr.s_addr = ((
struct sockaddr_in *)addr)->sin_addr.s_addr;
226 mreq.imr_interface= ((
struct sockaddr_in *)local_addr)->sin_addr;
228 mreq.imr_interface.s_addr= INADDR_ANY;
229 if (setsockopt(sockfd, IPPROTO_IP, IP_DROP_MEMBERSHIP, (
const void *)&mreq,
sizeof(mreq)) < 0) {
235 #if HAVE_STRUCT_IPV6_MREQ && defined(IPPROTO_IPV6) 236 if (addr->sa_family == AF_INET6) {
237 struct ipv6_mreq mreq6;
239 memcpy(&mreq6.ipv6mr_multiaddr, &(((
struct sockaddr_in6 *)addr)->sin6_addr),
sizeof(
struct in6_addr));
240 mreq6.ipv6mr_interface= 0;
251 const char *hostname,
int port,
252 int type,
int family,
int flags)
254 struct addrinfo hints = { 0 }, *res = 0;
257 const char *node = 0, *service =
"0";
260 snprintf(sport,
sizeof(sport),
"%d", port);
263 if ((hostname) && (hostname[0] !=
'\0') && (hostname[0] !=
'?')) {
269 if ((error =
getaddrinfo(node, service, &hints, &res))) {
272 node ? node :
"unknown",
281 int sockfd,
struct sockaddr *addr,
283 int nb_sources,
int include)
285 #if HAVE_STRUCT_GROUP_SOURCE_REQ && defined(MCAST_BLOCK_SOURCE) && !defined(_WIN32) && (!defined(TARGET_OS_TV) || !TARGET_OS_TV) 289 for (i = 0; i < nb_sources; i++) {
290 struct group_source_req mreqs;
291 int level = addr->sa_family == AF_INET ? IPPROTO_IP : IPPROTO_IPV6;
293 SOCK_DGRAM, AF_UNSPEC,
298 mreqs.gsr_interface = 0;
299 memcpy(&mreqs.gsr_group, addr, addr_len);
303 if (setsockopt(sockfd, level,
304 include ? MCAST_JOIN_SOURCE_GROUP : MCAST_BLOCK_SOURCE,
305 (
const void *)&mreqs,
sizeof(mreqs)) < 0) {
313 #elif HAVE_STRUCT_IP_MREQ_SOURCE && defined(IP_BLOCK_SOURCE) 315 if (addr->sa_family != AF_INET) {
317 "Setting multicast sources only supported for IPv4\n");
320 for (i = 0; i < nb_sources; i++) {
321 struct ip_mreq_source mreqs;
323 SOCK_DGRAM, AF_UNSPEC,
327 if (sourceaddr->
ai_addr->sa_family != AF_INET) {
334 mreqs.imr_multiaddr.s_addr = ((
struct sockaddr_in *)addr)->sin_addr.s_addr;
335 mreqs.imr_interface.s_addr = INADDR_ANY;
336 mreqs.imr_sourceaddr.s_addr = ((
struct sockaddr_in *)sourceaddr->
ai_addr)->sin_addr.s_addr;
339 if (setsockopt(sockfd, IPPROTO_IP,
340 include ? IP_ADD_SOURCE_MEMBERSHIP : IP_BLOCK_SOURCE,
341 (
const void *)&mreqs,
sizeof(mreqs)) < 0) {
356 const char *hostname,
int port)
362 if (!res0)
return AVERROR(EIO);
371 socklen_t *addr_len,
const char *
localaddr)
376 int family = AF_UNSPEC;
378 if (((
struct sockaddr *) &s->
dest_addr)->sa_family)
379 family = ((
struct sockaddr *) &s->
dest_addr)->sa_family;
385 for (res = res0; res; res=res->
ai_next) {
390 if (udp_fd != -1)
break;
414 char sbuf[
sizeof(
int)*3+1];
422 return strtol(sbuf,
NULL, 10);
445 char hostname[256],
buf[10];
457 p = strchr(uri,
'?');
498 #if HAVE_PTHREAD_CANCEL 499 static void *circular_buffer_task_rx(
void *_URLContext)
505 pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &old_cancelstate);
519 pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, &old_cancelstate);
521 pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &old_cancelstate);
536 "Surviving due to overrun_nonfatal option\n");
540 "To avoid, increase fifo_size URL option. " 541 "To survive in such case, use overrun_nonfatal option\n");
556 static void *circular_buffer_task_tx(
void *_URLContext)
563 int64_t sent_bits = 0;
567 pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &old_cancelstate);
602 pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, &old_cancelstate);
606 if (timestamp < target_timestamp) {
607 int64_t delay = target_timestamp - timestamp;
608 if (delay > max_delay) {
610 start_timestamp = timestamp + delay;
615 if (timestamp - burst_interval > target_timestamp) {
616 start_timestamp = timestamp - burst_interval;
620 sent_bits += len * 8;
621 target_timestamp = start_timestamp + sent_bits * 1000000 / s->
bitrate;
629 ret = sendto (s->
udp_fd, p, len, 0,
633 ret = send(s->
udp_fd, p, len, 0);
648 pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &old_cancelstate);
667 char *next = strchr(source_start,
',');
670 sources[*num_sources] =
av_strdup(source_start);
671 if (!sources[*num_sources])
673 source_start = next + 1;
675 if (*num_sources >= max_sources || !next)
685 char hostname[1024],
localaddr[1024] =
"";
686 int port,
udp_fd = -1,
tmp, bind_ret = -1, dscp = -1;
693 int i, num_include_sources = 0, num_exclude_sources = 0;
694 char *include_sources[32], *exclude_sources[32];
704 &num_include_sources,
718 p = strchr(uri,
'?');
735 "'overrun_nonfatal' option was set but it is not supported " 736 "on this build (pthread support is required)\n");
739 s->
ttl = strtol(buf,
NULL, 10);
757 dscp = strtol(buf,
NULL, 10);
763 "'circular_buffer_size' option was set but it is not supported " 764 "on this build (pthread support is required)\n");
770 "'bitrate' option was set but it is not supported " 771 "on this build (pthread support is required)\n");
777 av_strlcpy(localaddr, buf,
sizeof(localaddr));
807 if (hostname[0] ==
'\0' || hostname[0] ==
'?') {
858 if (setsockopt (udp_fd, IPPROTO_IP, IP_TOS, &dscp,
sizeof(dscp)) != 0)
867 bind_ret = bind(udp_fd,(
struct sockaddr *)&s->
dest_addr, len);
872 if (bind_ret < 0 && bind(udp_fd,(
struct sockaddr *)&my_addr, len) < 0) {
877 len =
sizeof(my_addr);
878 getsockname(udp_fd, (
struct sockaddr *)&my_addr, &len);
882 if (h->
flags & AVIO_FLAG_WRITE) {
889 if (num_include_sources && num_exclude_sources) {
890 av_log(h,
AV_LOG_ERROR,
"Simultaneously including and excluding multicast sources is not supported\n");
893 if (num_include_sources) {
898 num_include_sources, 1) < 0)
904 if (num_exclude_sources) {
909 num_exclude_sources, 0) < 0)
918 if (setsockopt(udp_fd, SOL_SOCKET, SO_SNDBUF, &
tmp,
sizeof(
tmp)) < 0) {
925 if (setsockopt(udp_fd, SOL_SOCKET, SO_RCVBUF, &
tmp,
sizeof(
tmp)) < 0) {
929 if (getsockopt(udp_fd, SOL_SOCKET, SO_RCVBUF, &
tmp, &len) < 0) {
947 for (i = 0; i < num_include_sources; i++)
949 for (i = 0; i < num_exclude_sources; i++)
954 #if HAVE_PTHREAD_CANCEL 963 av_log(h,
AV_LOG_WARNING,
"'bitrate' option was set but 'circular_buffer_size' is not, but required\n");
981 ret =
pthread_create(&s->circular_buffer_thread,
NULL, is_output?circular_buffer_task_tx:circular_buffer_task_rx, h);
986 s->thread_started = 1;
991 #if HAVE_PTHREAD_CANCEL 1001 for (i = 0; i < num_include_sources; i++)
1003 for (i = 0; i < num_exclude_sources; i++)
1022 #if HAVE_PTHREAD_CANCEL 1047 }
else if(nonblock) {
1055 struct timespec tv = { .tv_sec = t / 1000000,
1056 .tv_nsec = (t % 1000000) * 1000 };
1057 if (pthread_cond_timedwait(&s->cond, &s->mutex, &tv) < 0) {
1059 return AVERROR(errno == ETIMEDOUT ? EAGAIN : errno);
1072 ret = recv(s->
udp_fd, buf, size, 0);
1082 #if HAVE_PTHREAD_CANCEL 1118 ret = sendto (s->
udp_fd, buf, size, 0,
1122 ret = send(s->
udp_fd, buf, size, 0);
1131 #if HAVE_PTHREAD_CANCEL 1143 #if HAVE_PTHREAD_CANCEL 1144 if (s->thread_started) {
1148 pthread_cancel(s->circular_buffer_thread);
1169 .priv_data_class = &udp_class,
1181 .priv_data_class = &udplite_context_class,
static int udp_open(URLContext *h, const char *uri, int flags)
void av_url_split(char *proto, int proto_size, char *authorization, int authorization_size, char *hostname, int hostname_size, int *port_ptr, char *path, int path_size, const char *url)
Split a URL string into components.
static int udp_set_multicast_sources(URLContext *h, int sockfd, struct sockaddr *addr, int addr_len, char **sources, int nb_sources, int include)
static const AVClass udplite_context_class
static void log_net_error(void *ctx, int level, const char *prefix)
static av_always_inline int pthread_mutex_destroy(pthread_mutex_t *mutex)
#define URL_PROTOCOL_FLAG_NETWORK
#define pthread_mutex_lock(a)
#define HAVE_PTHREAD_CANCEL
static av_always_inline int pthread_cond_wait(pthread_cond_t *cond, pthread_mutex_t *mutex)
#define AV_LOG_WARNING
Something somehow does not look correct.
#define LIBAVUTIL_VERSION_INT
const URLProtocol ff_udplite_protocol
int is_streamed
true if streamed (no seek possible), default = false
static int parse_source_list(char *buf, char **sources, int *num_sources, int max_sources)
#define AVIO_FLAG_READ
read-only
int64_t rw_timeout
maximum time to wait for (network) read/write operation completion, in mcs
static int udp_set_url(URLContext *h, struct sockaddr_storage *addr, const char *hostname, int port)
#define AVIO_FLAG_WRITE
write-only
#define UDPLITE_SEND_CSCOV
int av_usleep(unsigned usec)
Sleep for a period of time.
static av_always_inline int pthread_cond_destroy(pthread_cond_t *cond)
int av_fifo_generic_write(AVFifoBuffer *f, void *src, int size, int(*func)(void *, void *, int))
Feed data from a user-supplied callback to an AVFifoBuffer.
int ff_socket(int af, int type, int proto)
const char * class_name
The name of the class; usually it is the same name as the context structure type to which the AVClass...
#define av_assert0(cond)
assert() equivalent, that is always enabled.
#define UDPLITE_RECV_CSCOV
miscellaneous OS support macros and functions.
int ff_udp_get_local_port(URLContext *h)
Return the local port used by the UDP connection.
static av_cold int end(AVCodecContext *avctx)
int av_fifo_space(const AVFifoBuffer *f)
Return the amount of space in bytes in the AVFifoBuffer, that is the amount of data you can write int...
static int udp_leave_multicast_group(int sockfd, struct sockaddr *addr, struct sockaddr *local_addr)
static int udp_port(struct sockaddr_storage *addr, int addr_len)
int av_find_info_tag(char *arg, int arg_size, const char *tag1, const char *info)
Attempt to find a specific tag in a URL.
static av_always_inline int pthread_cond_signal(pthread_cond_t *cond)
static int udp_close(URLContext *h)
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
int ff_udp_set_remote_url(URLContext *h, const char *uri)
If no filename is given to av_open_input_file because you want to get the local port first...
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
int av_fifo_generic_read(AVFifoBuffer *f, void *dest, int buf_size, void(*func)(void *, void *, int))
Feed data from an AVFifoBuffer to a user-supplied callback.
simple assert() macros that are a bit more flexible than ISO C assert().
static struct addrinfo * udp_resolve_host(URLContext *h, const char *hostname, int port, int type, int family, int flags)
int ff_is_multicast_address(struct sockaddr *addr)
size_t av_strlcpy(char *dst, const char *src, size_t size)
Copy the string src to dst, but no more than size - 1 bytes, and null-terminate dst.
static const AVOption options[]
uint8_t tmp[UDP_MAX_PKT_SIZE+4]
static av_always_inline int pthread_join(pthread_t thread, void **value_ptr)
static av_always_inline int pthread_mutex_init(pthread_mutex_t *mutex, const pthread_mutexattr_t *attr)
static int udp_get_file_handle(URLContext *h)
Return the udp file handle for select() usage to wait for several RTP streams at the same time...
#define pthread_mutex_unlock(a)
struct sockaddr_storage local_addr_storage
static pthread_mutex_t * mutex
int ff_socket_nonblock(int socket, int enable)
static void error(const char *err)
#define FF_ARRAY_ELEMS(a)
int64_t av_gettime(void)
Get the current time in microseconds.
static av_always_inline int pthread_create(pthread_t *thread, const pthread_attr_t *attr, void *(*start_routine)(void *), void *arg)
#define IPV6_ADD_MEMBERSHIP
char * av_strdup(const char *s)
Duplicate a string.
int av_fifo_size(const AVFifoBuffer *f)
Return the amount of data in bytes in the AVFifoBuffer, that is the amount of data you can read from ...
#define AVIO_FLAG_NONBLOCK
Use non-blocking mode.
#define IPV6_DROP_MEMBERSHIP
a very simple circular buffer FIFO implementation
Describe the class of an AVClass context structure.
int64_t av_gettime_relative(void)
Get the current time in microseconds since some unspecified starting point.
int av_strerror(int errnum, char *errbuf, size_t errbuf_size)
Put a description of the AVERROR code errnum in errbuf.
int(* cond)(enum AVPixelFormat pix_fmt)
struct sockaddr_storage dest_addr
static int udp_read(URLContext *h, uint8_t *buf, int size)
struct addrinfo * ai_next
static av_always_inline int pthread_cond_init(pthread_cond_t *cond, const pthread_condattr_t *attr)
static int udp_set_multicast_ttl(int sockfd, int mcastTTL, struct sockaddr *addr)
AVFifoBuffer * av_fifo_alloc(unsigned int size)
Initialize an AVFifoBuffer.
static int udplite_open(URLContext *h, const char *uri, int flags)
static int udp_join_multicast_group(int sockfd, struct sockaddr *addr, struct sockaddr *local_addr)
int circular_buffer_error
static int udp_socket_create(URLContext *h, struct sockaddr_storage *addr, socklen_t *addr_len, const char *localaddr)
int max_packet_size
if non zero, the stream is packetized with this max packet size
static const AVClass udp_class
int ff_network_wait_fd(int fd, int write)
unbuffered private I/O API
void av_fifo_freep(AVFifoBuffer **f)
Free an AVFifoBuffer and reset pointer to NULL.
void av_fifo_drain(AVFifoBuffer *f, int size)
Discard data from the FIFO.
struct sockaddr * ai_addr
static int udp_write(URLContext *h, const uint8_t *buf, int size)
const URLProtocol ff_udp_protocol