Commit 57cde2b1 authored by wm4's avatar wm4 Committed by Martin Storsjö

lavf: move TLS-related ifdeffery to library specific files

There is no need to have this mess in network.c.
Signed-off-by: 's avatarMartin Storsjö <martin@martin.st>
parent d8ffb205
...@@ -20,102 +20,29 @@ ...@@ -20,102 +20,29 @@
#include <fcntl.h> #include <fcntl.h>
#include "network.h" #include "network.h"
#include "tls.h"
#include "url.h" #include "url.h"
#include "libavcodec/internal.h" #include "libavcodec/internal.h"
#include "libavutil/mem.h" #include "libavutil/mem.h"
#if HAVE_THREADS
#if HAVE_PTHREADS
#include <pthread.h>
#else
#include "compat/w32pthreads.h"
#endif
#endif
#if CONFIG_OPENSSL
#include <openssl/ssl.h>
static int openssl_init;
#if HAVE_THREADS
#include <openssl/crypto.h>
#include "libavutil/avutil.h"
pthread_mutex_t *openssl_mutexes;
static void openssl_lock(int mode, int type, const char *file, int line)
{
if (mode & CRYPTO_LOCK)
pthread_mutex_lock(&openssl_mutexes[type]);
else
pthread_mutex_unlock(&openssl_mutexes[type]);
}
#if !defined(WIN32) && OPENSSL_VERSION_NUMBER < 0x10000000
static unsigned long openssl_thread_id(void)
{
return (intptr_t) pthread_self();
}
#endif
#endif
#endif
#if CONFIG_GNUTLS
#include <gnutls/gnutls.h>
#if HAVE_THREADS && GNUTLS_VERSION_NUMBER <= 0x020b00
#include <gcrypt.h>
#include <errno.h>
GCRY_THREAD_OPTION_PTHREAD_IMPL;
#endif
#endif
void ff_tls_init(void) void ff_tls_init(void)
{ {
avpriv_lock_avformat(); #if CONFIG_TLS_OPENSSL_PROTOCOL
#if CONFIG_OPENSSL ff_openssl_init();
if (!openssl_init) {
SSL_library_init();
SSL_load_error_strings();
#if HAVE_THREADS
if (!CRYPTO_get_locking_callback()) {
int i;
openssl_mutexes = av_malloc(sizeof(pthread_mutex_t) * CRYPTO_num_locks());
for (i = 0; i < CRYPTO_num_locks(); i++)
pthread_mutex_init(&openssl_mutexes[i], NULL);
CRYPTO_set_locking_callback(openssl_lock);
#if !defined(WIN32) && OPENSSL_VERSION_NUMBER < 0x10000000
CRYPTO_set_id_callback(openssl_thread_id);
#endif
}
#endif
}
openssl_init++;
#endif
#if CONFIG_GNUTLS
#if HAVE_THREADS && GNUTLS_VERSION_NUMBER < 0x020b00
if (gcry_control(GCRYCTL_ANY_INITIALIZATION_P) == 0)
gcry_control(GCRYCTL_SET_THREAD_CBS, &gcry_threads_pthread);
#endif #endif
gnutls_global_init(); #if CONFIG_TLS_GNUTLS_PROTOCOL
ff_gnutls_init();
#endif #endif
avpriv_unlock_avformat();
} }
void ff_tls_deinit(void) void ff_tls_deinit(void)
{ {
avpriv_lock_avformat(); #if CONFIG_TLS_OPENSSL_PROTOCOL
#if CONFIG_OPENSSL ff_openssl_deinit();
openssl_init--;
if (!openssl_init) {
#if HAVE_THREADS
if (CRYPTO_get_locking_callback() == openssl_lock) {
int i;
CRYPTO_set_locking_callback(NULL);
for (i = 0; i < CRYPTO_num_locks(); i++)
pthread_mutex_destroy(&openssl_mutexes[i]);
av_free(openssl_mutexes);
}
#endif
}
#endif #endif
#if CONFIG_GNUTLS #if CONFIG_TLS_GNUTLS_PROTOCOL
gnutls_global_deinit(); ff_gnutls_deinit();
#endif #endif
avpriv_unlock_avformat();
} }
int ff_network_inited_globally; int ff_network_inited_globally;
......
...@@ -51,4 +51,10 @@ typedef struct TLSShared { ...@@ -51,4 +51,10 @@ typedef struct TLSShared {
int ff_tls_open_underlying(TLSShared *c, URLContext *parent, const char *uri, AVDictionary **options); int ff_tls_open_underlying(TLSShared *c, URLContext *parent, const char *uri, AVDictionary **options);
void ff_gnutls_init(void);
void ff_gnutls_deinit(void);
void ff_openssl_init(void);
void ff_openssl_deinit(void);
#endif /* AVFORMAT_TLS_H */ #endif /* AVFORMAT_TLS_H */
...@@ -19,6 +19,8 @@ ...@@ -19,6 +19,8 @@
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/ */
#include <errno.h>
#include <gnutls/gnutls.h> #include <gnutls/gnutls.h>
#include <gnutls/x509.h> #include <gnutls/x509.h>
...@@ -28,10 +30,16 @@ ...@@ -28,10 +30,16 @@
#include "os_support.h" #include "os_support.h"
#include "url.h" #include "url.h"
#include "tls.h" #include "tls.h"
#include "libavcodec/internal.h"
#include "libavutil/avstring.h" #include "libavutil/avstring.h"
#include "libavutil/opt.h" #include "libavutil/opt.h"
#include "libavutil/parseutils.h" #include "libavutil/parseutils.h"
#if HAVE_THREADS && GNUTLS_VERSION_NUMBER <= 0x020b00
#include <gcrypt.h>
GCRY_THREAD_OPTION_PTHREAD_IMPL;
#endif
typedef struct TLSContext { typedef struct TLSContext {
const AVClass *class; const AVClass *class;
TLSShared tls_shared; TLSShared tls_shared;
...@@ -40,6 +48,24 @@ typedef struct TLSContext { ...@@ -40,6 +48,24 @@ typedef struct TLSContext {
int need_shutdown; int need_shutdown;
} TLSContext; } TLSContext;
void ff_gnutls_init(void)
{
avpriv_lock_avformat();
#if HAVE_THREADS && GNUTLS_VERSION_NUMBER < 0x020b00
if (gcry_control(GCRYCTL_ANY_INITIALIZATION_P) == 0)
gcry_control(GCRYCTL_SET_THREAD_CBS, &gcry_threads_pthread);
#endif
gnutls_global_init();
avpriv_unlock_avformat();
}
void ff_gnutls_deinit(void)
{
avpriv_lock_avformat();
gnutls_global_deinit();
avpriv_unlock_avformat();
}
static int print_tls_error(URLContext *h, int ret) static int print_tls_error(URLContext *h, int ret)
{ {
switch (ret) { switch (ret) {
...@@ -67,7 +93,7 @@ static int tls_close(URLContext *h) ...@@ -67,7 +93,7 @@ static int tls_close(URLContext *h)
gnutls_certificate_free_credentials(c->cred); gnutls_certificate_free_credentials(c->cred);
if (c->tls_shared.tcp) if (c->tls_shared.tcp)
ffurl_close(c->tls_shared.tcp); ffurl_close(c->tls_shared.tcp);
ff_tls_deinit(); ff_gnutls_deinit();
return 0; return 0;
} }
...@@ -103,7 +129,7 @@ static int tls_open(URLContext *h, const char *uri, int flags, AVDictionary **op ...@@ -103,7 +129,7 @@ static int tls_open(URLContext *h, const char *uri, int flags, AVDictionary **op
TLSShared *c = &p->tls_shared; TLSShared *c = &p->tls_shared;
int ret; int ret;
ff_tls_init(); ff_gnutls_init();
if ((ret = ff_tls_open_underlying(c, h, uri, options)) < 0) if ((ret = ff_tls_open_underlying(c, h, uri, options)) < 0)
goto fail; goto fail;
......
...@@ -25,14 +25,19 @@ ...@@ -25,14 +25,19 @@
#include "os_support.h" #include "os_support.h"
#include "url.h" #include "url.h"
#include "tls.h" #include "tls.h"
#include "libavcodec/internal.h"
#include "libavutil/avstring.h" #include "libavutil/avstring.h"
#include "libavutil/avutil.h"
#include "libavutil/opt.h" #include "libavutil/opt.h"
#include "libavutil/parseutils.h" #include "libavutil/parseutils.h"
#include "libavutil/thread.h"
#include <openssl/bio.h> #include <openssl/bio.h>
#include <openssl/ssl.h> #include <openssl/ssl.h>
#include <openssl/err.h> #include <openssl/err.h>
static int openssl_init;
typedef struct TLSContext { typedef struct TLSContext {
const AVClass *class; const AVClass *class;
TLSShared tls_shared; TLSShared tls_shared;
...@@ -40,6 +45,65 @@ typedef struct TLSContext { ...@@ -40,6 +45,65 @@ typedef struct TLSContext {
SSL *ssl; SSL *ssl;
} TLSContext; } TLSContext;
#if HAVE_THREADS
#include <openssl/crypto.h>
pthread_mutex_t *openssl_mutexes;
static void openssl_lock(int mode, int type, const char *file, int line)
{
if (mode & CRYPTO_LOCK)
pthread_mutex_lock(&openssl_mutexes[type]);
else
pthread_mutex_unlock(&openssl_mutexes[type]);
}
#if !defined(WIN32) && OPENSSL_VERSION_NUMBER < 0x10000000
static unsigned long openssl_thread_id(void)
{
return (intptr_t) pthread_self();
}
#endif
#endif
void ff_openssl_init(void)
{
avpriv_lock_avformat();
if (!openssl_init) {
SSL_library_init();
SSL_load_error_strings();
#if HAVE_THREADS
if (!CRYPTO_get_locking_callback()) {
int i;
openssl_mutexes = av_malloc(sizeof(pthread_mutex_t) * CRYPTO_num_locks());
for (i = 0; i < CRYPTO_num_locks(); i++)
pthread_mutex_init(&openssl_mutexes[i], NULL);
CRYPTO_set_locking_callback(openssl_lock);
#if !defined(WIN32) && OPENSSL_VERSION_NUMBER < 0x10000000
CRYPTO_set_id_callback(openssl_thread_id);
#endif
}
#endif
}
openssl_init++;
avpriv_unlock_avformat();
}
void ff_openssl_deinit(void)
{
avpriv_lock_avformat();
openssl_init--;
if (!openssl_init) {
#if HAVE_THREADS
if (CRYPTO_get_locking_callback() == openssl_lock) {
int i;
CRYPTO_set_locking_callback(NULL);
for (i = 0; i < CRYPTO_num_locks(); i++)
pthread_mutex_destroy(&openssl_mutexes[i]);
av_free(openssl_mutexes);
}
#endif
}
avpriv_unlock_avformat();
}
static int print_tls_error(URLContext *h, int ret) static int print_tls_error(URLContext *h, int ret)
{ {
av_log(h, AV_LOG_ERROR, "%s\n", ERR_error_string(ERR_get_error(), NULL)); av_log(h, AV_LOG_ERROR, "%s\n", ERR_error_string(ERR_get_error(), NULL));
...@@ -57,7 +121,7 @@ static int tls_close(URLContext *h) ...@@ -57,7 +121,7 @@ static int tls_close(URLContext *h)
SSL_CTX_free(c->ctx); SSL_CTX_free(c->ctx);
if (c->tls_shared.tcp) if (c->tls_shared.tcp)
ffurl_close(c->tls_shared.tcp); ffurl_close(c->tls_shared.tcp);
ff_tls_deinit(); ff_openssl_deinit();
return 0; return 0;
} }
...@@ -131,7 +195,7 @@ static int tls_open(URLContext *h, const char *uri, int flags, AVDictionary **op ...@@ -131,7 +195,7 @@ static int tls_open(URLContext *h, const char *uri, int flags, AVDictionary **op
BIO *bio; BIO *bio;
int ret; int ret;
ff_tls_init(); ff_openssl_init();
if ((ret = ff_tls_open_underlying(c, h, uri, options)) < 0) if ((ret = ff_tls_open_underlying(c, h, uri, options)) < 0)
goto fail; goto fail;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment