/*****************************************************************************
 * mta_holder.c: Hold a MTA from another thread
 *****************************************************************************
 * Copyright (C) 2002-2017 the VideoLAN and AUTHORS
 *
 * Author: Hugo Beauzée-Luyssen <hugo@beauzee.fr>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
 *****************************************************************************/

#ifndef MTA_HOLDER_H
# define MTA_HOLDER_H

#include <vlc_common.h>
#include <vlc_threads.h>

#include <assert.h>
#include <windows.h>
#include <objbase.h>

typedef struct vlc_mta_holder
{
    vlc_thread_t thread;
    int          i_refcount;
    vlc_sem_t    ready_sem;
    vlc_sem_t    release_sem;
} vlc_mta_holder;

static inline void* MtaMainLoop( void* opaque )
{
    vlc_thread_set_name("vlc-mta");

    vlc_mta_holder* p_mta = (vlc_mta_holder*)opaque;
    CoInitializeEx( NULL, COINIT_MULTITHREADED | COINIT_DISABLE_OLE1DDE );

    vlc_sem_post( &p_mta->ready_sem );

    vlc_sem_wait( &p_mta->release_sem );

    CoUninitialize();
    return NULL;
}

/**
 * Ensure an MTA context will be available until vlc_mta_release gets called.
 *
 * In the background, this will create a thread that does nothing but to keep the MTA
 * refcount greater than 0.
 *
 * This is useful in order not to commit a thread to a specific concurrency model.
 * This function is win32 specific.
 */
static inline bool vlc_mta_acquire( vlc_object_t *p_parent )
{
    vlc_object_t *vlc = VLC_OBJECT(vlc_object_instance(p_parent));

    vlc_global_lock( VLC_MTA_MUTEX );
    vlc_mta_holder* p_mta = (vlc_mta_holder*)var_CreateGetAddress( vlc, "mta-holder" );
    if ( p_mta == NULL )
    {
        p_mta = (vlc_mta_holder*)malloc( sizeof( *p_mta ) );
        if ( unlikely( p_mta == NULL ) )
        {
            vlc_global_unlock( VLC_MTA_MUTEX );
            return false;
        }
        vlc_sem_init( &p_mta->ready_sem, 0 );
        vlc_sem_init( &p_mta->release_sem, 0 );
        p_mta->i_refcount = 1;
        if ( vlc_clone( &p_mta->thread, MtaMainLoop, p_mta ) )
        {
            free( p_mta );
            p_mta = NULL;
            vlc_global_unlock( VLC_MTA_MUTEX );
            return false;
        }
        var_SetAddress( vlc, "mta-holder", p_mta );
        vlc_sem_wait( &p_mta->ready_sem );
    }
    else
        ++p_mta->i_refcount;
    vlc_global_unlock( VLC_MTA_MUTEX );
    return true;
}

/**
 * Releases a reference to the MTA holder.
 *
 * When its refcount reaches 0, the thread created by
 */
static inline void vlc_mta_release( vlc_object_t* p_parent )
{
    vlc_object_t *vlc = VLC_OBJECT(vlc_object_instance(p_parent));

    vlc_global_lock( VLC_MTA_MUTEX );
    vlc_mta_holder *p_mta = (vlc_mta_holder*)var_InheritAddress( vlc, "mta-holder" );
    assert( p_mta != NULL );
    int i_refcount = --p_mta->i_refcount;
    if ( i_refcount == 0 )
        var_SetAddress( vlc, "mta-holder", NULL );
    vlc_global_unlock( VLC_MTA_MUTEX );
    if ( i_refcount == 0 )
    {
        vlc_sem_post( &p_mta->release_sem );

        vlc_join( p_mta->thread, NULL );

        free( p_mta );
    }
}

#endif