git.sesse.net Git - ffmpeg/blob - libavformat/network.c

   1 /*
   2  * Copyright (c) 2007 The Libav Project
   3  *
   4  * This file is part of Libav.
   5  *
   6  * Libav is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * Libav is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with Libav; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19  */
  20
  21 #include "libavutil/avutil.h"
  22 #include "network.h"
  23 #include "libavcodec/internal.h"
  24 #include "libavutil/mem.h"
  25 #include "url.h"
  26 #include "libavutil/time.h"
  27
  28 #if HAVE_THREADS
  29 #if HAVE_PTHREADS
  30 #include <pthread.h>
  31 #else
  32 #include "libavcodec/w32pthreads.h"
  33 #endif
  34 #endif
  35
  36 #if CONFIG_OPENSSL
  37 #include <openssl/ssl.h>
  38 static int openssl_init;
  39 #if HAVE_THREADS
  40 #include <openssl/crypto.h>
  41 #include "libavutil/avutil.h"
  42 pthread_mutex_t *openssl_mutexes;
  43 static void openssl_lock(int mode, int type, const char *file, int line)
  44 {
  45     if (mode & CRYPTO_LOCK)
  46         pthread_mutex_lock(&openssl_mutexes[type]);
  47     else
  48         pthread_mutex_unlock(&openssl_mutexes[type]);
  49 }
  50 #if !defined(WIN32) && OPENSSL_VERSION_NUMBER < 0x10000000
  51 static unsigned long openssl_thread_id(void)
  52 {
  53     return (intptr_t) pthread_self();
  54 }
  55 #endif
  56 #endif
  57 #endif
  58 #if CONFIG_GNUTLS
  59 #include <gnutls/gnutls.h>
  60 #if HAVE_THREADS && GNUTLS_VERSION_NUMBER <= 0x020b00
  61 #include <gcrypt.h>
  62 #include <errno.h>
  63 #undef malloc
  64 #undef free
  65 GCRY_THREAD_OPTION_PTHREAD_IMPL;
  66 #endif
  67 #endif
  68
  69 void ff_tls_init(void)
  70 {
  71     avpriv_lock_avformat();
  72 #if CONFIG_OPENSSL
  73     if (!openssl_init) {
  74         SSL_library_init();
  75         SSL_load_error_strings();
  76 #if HAVE_THREADS
  77         if (!CRYPTO_get_locking_callback()) {
  78             int i;
  79             openssl_mutexes = av_malloc(sizeof(pthread_mutex_t) * CRYPTO_num_locks());
  80             for (i = 0; i < CRYPTO_num_locks(); i++)
  81                 pthread_mutex_init(&openssl_mutexes[i], NULL);
  82             CRYPTO_set_locking_callback(openssl_lock);
  83 #if !defined(WIN32) && OPENSSL_VERSION_NUMBER < 0x10000000
  84             CRYPTO_set_id_callback(openssl_thread_id);
  85 #endif
  86         }
  87 #endif
  88     }
  89     openssl_init++;
  90 #endif
  91 #if CONFIG_GNUTLS
  92 #if HAVE_THREADS && GNUTLS_VERSION_NUMBER < 0x020b00
  93     if (gcry_control(GCRYCTL_ANY_INITIALIZATION_P) == 0)
  94         gcry_control(GCRYCTL_SET_THREAD_CBS, &gcry_threads_pthread);
  95 #endif
  96     gnutls_global_init();
  97 #endif
  98     avpriv_unlock_avformat();
  99 }
 100
 101 void ff_tls_deinit(void)
 102 {
 103     avpriv_lock_avformat();
 104 #if CONFIG_OPENSSL
 105     openssl_init--;
 106     if (!openssl_init) {
 107 #if HAVE_THREADS
 108         if (CRYPTO_get_locking_callback() == openssl_lock) {
 109             int i;
 110             CRYPTO_set_locking_callback(NULL);
 111             for (i = 0; i < CRYPTO_num_locks(); i++)
 112                 pthread_mutex_destroy(&openssl_mutexes[i]);
 113             av_free(openssl_mutexes);
 114         }
 115 #endif
 116     }
 117 #endif
 118 #if CONFIG_GNUTLS
 119     gnutls_global_deinit();
 120 #endif
 121     avpriv_unlock_avformat();
 122 }
 123
 124 int ff_network_inited_globally;
 125
 126 int ff_network_init(void)
 127 {
 128 #if HAVE_WINSOCK2_H
 129     WSADATA wsaData;
 130 #endif
 131
 132     if (!ff_network_inited_globally)
 133         av_log(NULL, AV_LOG_WARNING, "Using network protocols without global "
 134                                      "network initialization. Please use "
 135                                      "avformat_network_init(), this will "
 136                                      "become mandatory later.\n");
 137 #if HAVE_WINSOCK2_H
 138     if (WSAStartup(MAKEWORD(1,1), &wsaData))
 139         return 0;
 140 #endif
 141     return 1;
 142 }
 143
 144 int ff_network_wait_fd(int fd, int write)
 145 {
 146     int ev = write ? POLLOUT : POLLIN;
 147     struct pollfd p = { .fd = fd, .events = ev, .revents = 0 };
 148     int ret;
 149     ret = poll(&p, 1, 100);
 150     return ret < 0 ? ff_neterrno() : p.revents & (ev | POLLERR | POLLHUP) ? 0 : AVERROR(EAGAIN);
 151 }
 152
 153 int ff_network_wait_fd_timeout(int fd, int write, int64_t timeout, AVIOInterruptCB *int_cb)
 154 {
 155     int ret;
 156     int64_t wait_start = 0;
 157
 158     while (1) {
 159         ret = ff_network_wait_fd(fd, write);
 160         if (ret != AVERROR(EAGAIN))
 161             return ret;
 162         if (ff_check_interrupt(int_cb))
 163             return AVERROR_EXIT;
 164         if (timeout) {
 165             if (!wait_start)
 166                 wait_start = av_gettime();
 167             else if (av_gettime() - wait_start > timeout)
 168                 return AVERROR(ETIMEDOUT);
 169         }
 170     }
 171 }
 172
 173 void ff_network_close(void)
 174 {
 175 #if HAVE_WINSOCK2_H
 176     WSACleanup();
 177 #endif
 178 }
 179
 180 #if HAVE_WINSOCK2_H
 181 int ff_neterrno(void)
 182 {
 183     int err = WSAGetLastError();
 184     switch (err) {
 185     case WSAEWOULDBLOCK:
 186         return AVERROR(EAGAIN);
 187     case WSAEINTR:
 188         return AVERROR(EINTR);
 189     case WSAEPROTONOSUPPORT:
 190         return AVERROR(EPROTONOSUPPORT);
 191     case WSAETIMEDOUT:
 192         return AVERROR(ETIMEDOUT);
 193     case WSAECONNREFUSED:
 194         return AVERROR(ECONNREFUSED);
 195     case WSAEINPROGRESS:
 196         return AVERROR(EINPROGRESS);
 197     }
 198     return -err;
 199 }
 200 #endif
 201
 202 int ff_is_multicast_address(struct sockaddr *addr)
 203 {
 204     if (addr->sa_family == AF_INET) {
 205         return IN_MULTICAST(ntohl(((struct sockaddr_in *)addr)->sin_addr.s_addr));
 206     }
 207 #if HAVE_STRUCT_SOCKADDR_IN6
 208     if (addr->sa_family == AF_INET6) {
 209         return IN6_IS_ADDR_MULTICAST(&((struct sockaddr_in6 *)addr)->sin6_addr);
 210     }
 211 #endif
 212
 213     return 0;
 214 }