mirror of
https://code.videolan.org/videolan/vlc
synced 2024-09-12 13:44:56 +02:00
298 lines
9.6 KiB
C
298 lines
9.6 KiB
C
/*****************************************************************************
|
|
* mtp.c : MTP interface module
|
|
*****************************************************************************
|
|
* Copyright (C) 2009 the VideoLAN team
|
|
*
|
|
* Authors: Fabio Ritrovato <exsephiroth87@gmail.com>
|
|
*
|
|
* 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.
|
|
*****************************************************************************/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
# include "config.h"
|
|
#endif
|
|
|
|
#include <vlc_common.h>
|
|
#include <vlc_plugin.h>
|
|
#include <vlc_services_discovery.h>
|
|
|
|
#include <libmtp.h>
|
|
|
|
/*****************************************************************************
|
|
* Module descriptor
|
|
*****************************************************************************/
|
|
static int Open( vlc_object_t * );
|
|
static void Close( vlc_object_t * );
|
|
|
|
VLC_SD_PROBE_HELPER("mtp", "MTP devices", SD_CAT_DEVICES)
|
|
|
|
vlc_module_begin()
|
|
set_shortname( "MTP" )
|
|
set_description( N_( "MTP devices" ) )
|
|
set_category( CAT_PLAYLIST )
|
|
set_subcategory( SUBCAT_PLAYLIST_SD )
|
|
set_capability( "services_discovery", 0 )
|
|
set_callbacks( Open, Close )
|
|
cannot_unload_broken_library()
|
|
|
|
VLC_SD_PROBE_SUBMODULE
|
|
vlc_module_end()
|
|
|
|
|
|
/*****************************************************************************
|
|
* Local prototypes
|
|
*****************************************************************************/
|
|
|
|
static void *Run( void * );
|
|
|
|
static int AddDevice( services_discovery_t *, LIBMTP_raw_device_t * );
|
|
static void AddTrack( services_discovery_t *, LIBMTP_track_t *);
|
|
static void CloseDevice( services_discovery_t * );
|
|
static int CountTracks( uint64_t const, uint64_t const, void const * const );
|
|
|
|
/*****************************************************************************
|
|
* Local structures
|
|
*****************************************************************************/
|
|
|
|
struct services_discovery_sys_t
|
|
{
|
|
int i_tracks_num;
|
|
input_item_t **pp_items;
|
|
int i_count;
|
|
char *psz_name;
|
|
uint32_t i_bus;
|
|
uint8_t i_dev;
|
|
uint16_t i_product_id;
|
|
vlc_thread_t thread;
|
|
};
|
|
|
|
static vlc_mutex_t mtp_lock = VLC_STATIC_MUTEX;
|
|
static bool b_mtp_initialized = false;
|
|
|
|
/*****************************************************************************
|
|
* Open: initialize and create stuff
|
|
*****************************************************************************/
|
|
static int Open( vlc_object_t *p_this )
|
|
{
|
|
services_discovery_t *p_sd = ( services_discovery_t * )p_this;
|
|
services_discovery_sys_t *p_sys;
|
|
|
|
if( !( p_sys = malloc( sizeof( services_discovery_sys_t ) ) ) )
|
|
return VLC_ENOMEM;
|
|
p_sd->p_sys = p_sys;
|
|
p_sys->psz_name = NULL;
|
|
|
|
vlc_mutex_lock( &mtp_lock );
|
|
if( !b_mtp_initialized )
|
|
{
|
|
LIBMTP_Init();
|
|
b_mtp_initialized = true;
|
|
}
|
|
vlc_mutex_unlock( &mtp_lock );
|
|
|
|
if (vlc_clone (&p_sys->thread, Run, p_sd, VLC_THREAD_PRIORITY_LOW))
|
|
{
|
|
free (p_sys);
|
|
return VLC_EGENERIC;
|
|
}
|
|
return VLC_SUCCESS;
|
|
}
|
|
|
|
/*****************************************************************************
|
|
* Close: cleanup
|
|
*****************************************************************************/
|
|
static void Close( vlc_object_t *p_this )
|
|
{
|
|
services_discovery_t *p_sd = ( services_discovery_t * )p_this;
|
|
|
|
free( p_sd->p_sys->psz_name );
|
|
vlc_cancel( p_sd->p_sys->thread );
|
|
vlc_join( p_sd->p_sys->thread, NULL );
|
|
free( p_sd->p_sys );
|
|
}
|
|
|
|
/*****************************************************************************
|
|
* Run: main thread
|
|
*****************************************************************************/
|
|
static void *Run( void *data )
|
|
{
|
|
LIBMTP_raw_device_t *p_rawdevices;
|
|
int i_numrawdevices;
|
|
int i_ret;
|
|
int i_status = 0;
|
|
services_discovery_t *p_sd = data;
|
|
|
|
for(;;)
|
|
{
|
|
int canc = vlc_savecancel();
|
|
i_ret = LIBMTP_Detect_Raw_Devices( &p_rawdevices, &i_numrawdevices );
|
|
if ( i_ret == 0 && i_numrawdevices > 0 && p_rawdevices != NULL &&
|
|
i_status == 0 )
|
|
{
|
|
/* Found a new device, add it */
|
|
msg_Dbg( p_sd, "New device found" );
|
|
if( AddDevice( p_sd, &p_rawdevices[0] ) == VLC_SUCCESS )
|
|
i_status = 1;
|
|
else
|
|
i_status = 2;
|
|
}
|
|
else
|
|
{
|
|
if ( ( i_ret != 0 || i_numrawdevices == 0 || p_rawdevices == NULL )
|
|
&& i_status == 1)
|
|
{
|
|
/* The device is not connected anymore, delete it */
|
|
msg_Info( p_sd, "Device disconnected" );
|
|
CloseDevice( p_sd );
|
|
i_status = 0;
|
|
}
|
|
}
|
|
free( p_rawdevices );
|
|
vlc_restorecancel(canc);
|
|
if( i_status == 2 )
|
|
{
|
|
msleep( 5000000 );
|
|
i_status = 0;
|
|
}
|
|
else
|
|
msleep( 500000 );
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
/*****************************************************************************
|
|
* Everything else
|
|
*****************************************************************************/
|
|
static int AddDevice( services_discovery_t *p_sd,
|
|
LIBMTP_raw_device_t *p_raw_device )
|
|
{
|
|
char *psz_name = NULL;
|
|
LIBMTP_mtpdevice_t *p_device;
|
|
LIBMTP_track_t *p_track, *p_tmp;
|
|
|
|
if( ( p_device = LIBMTP_Open_Raw_Device( p_raw_device ) ) != NULL )
|
|
{
|
|
if( !( psz_name = LIBMTP_Get_Friendlyname( p_device ) ) )
|
|
if( !( psz_name = LIBMTP_Get_Modelname( p_device ) ) )
|
|
if( !( psz_name = strdup( N_( "MTP Device" ) ) ) )
|
|
return VLC_ENOMEM;
|
|
msg_Info( p_sd, "Found device: %s", psz_name );
|
|
p_sd->p_sys->i_bus = p_raw_device->bus_location;
|
|
p_sd->p_sys->i_dev = p_raw_device->devnum;
|
|
p_sd->p_sys->i_product_id = p_raw_device->device_entry.product_id;
|
|
if( ( p_track = LIBMTP_Get_Tracklisting_With_Callback( p_device,
|
|
CountTracks, p_sd ) ) == NULL )
|
|
{
|
|
msg_Warn( p_sd, "No tracks on the device" );
|
|
}
|
|
else
|
|
{
|
|
if( !( p_sd->p_sys->pp_items = calloc( p_sd->p_sys->i_tracks_num,
|
|
sizeof( input_item_t * ) ) ) )
|
|
{
|
|
free( psz_name );
|
|
return VLC_ENOMEM;
|
|
}
|
|
p_sd->p_sys->i_count = 0;
|
|
while( p_track != NULL )
|
|
{
|
|
msg_Dbg( p_sd, "Track found: %s - %s", p_track->artist,
|
|
p_track->title );
|
|
AddTrack( p_sd, p_track );
|
|
p_tmp = p_track;
|
|
p_track = p_track->next;
|
|
LIBMTP_destroy_track_t( p_tmp );
|
|
}
|
|
}
|
|
p_sd->p_sys->psz_name = psz_name;
|
|
LIBMTP_Release_Device( p_device );
|
|
return VLC_SUCCESS;
|
|
}
|
|
else
|
|
{
|
|
msg_Info( p_sd, "The device seems to be mounted, unmount it first" );
|
|
return VLC_EGENERIC;
|
|
}
|
|
}
|
|
|
|
static void AddTrack( services_discovery_t *p_sd, LIBMTP_track_t *p_track )
|
|
{
|
|
input_item_t *p_input;
|
|
char *psz_string;
|
|
char *extension;
|
|
|
|
extension = rindex( p_track->filename, '.' );
|
|
if( asprintf( &psz_string, "mtp://%"PRIu32":%"PRIu8":%"PRIu16":%d%s",
|
|
p_sd->p_sys->i_bus, p_sd->p_sys->i_dev,
|
|
p_sd->p_sys->i_product_id, p_track->item_id,
|
|
extension ) == -1 )
|
|
{
|
|
msg_Err( p_sd, "Error adding %s, skipping it", p_track->filename );
|
|
return;
|
|
}
|
|
if( ( p_input = input_item_New( psz_string, p_track->title ) ) == NULL )
|
|
{
|
|
msg_Err( p_sd, "Error adding %s, skipping it", p_track->filename );
|
|
free( psz_string );
|
|
return;
|
|
}
|
|
free( psz_string );
|
|
|
|
input_item_SetArtist( p_input, p_track->artist );
|
|
input_item_SetGenre( p_input, p_track->genre );
|
|
input_item_SetAlbum( p_input, p_track->album );
|
|
if( asprintf( &psz_string, "%d", p_track->tracknumber ) != -1 )
|
|
{
|
|
input_item_SetTrackNum( p_input, psz_string );
|
|
free( psz_string );
|
|
}
|
|
if( asprintf( &psz_string, "%d", p_track->rating ) != -1 )
|
|
{
|
|
input_item_SetRating( p_input, psz_string );
|
|
free( psz_string );
|
|
}
|
|
input_item_SetDate( p_input, p_track->date );
|
|
input_item_SetDuration( p_input, p_track->duration * 1000 );
|
|
services_discovery_AddItem( p_sd, p_input, NULL );
|
|
p_sd->p_sys->pp_items[p_sd->p_sys->i_count++] = p_input;
|
|
}
|
|
|
|
static void CloseDevice( services_discovery_t *p_sd )
|
|
{
|
|
input_item_t **pp_items = p_sd->p_sys->pp_items;
|
|
|
|
if( pp_items != NULL )
|
|
{
|
|
for( int i_i = 0; i_i < p_sd->p_sys->i_count; i_i++ )
|
|
{
|
|
if( pp_items[i_i] != NULL )
|
|
{
|
|
services_discovery_RemoveItem( p_sd, pp_items[i_i] );
|
|
vlc_gc_decref( pp_items[i_i] );
|
|
}
|
|
}
|
|
free( pp_items );
|
|
}
|
|
}
|
|
|
|
static int CountTracks( uint64_t const sent, uint64_t const total,
|
|
void const * const data )
|
|
{
|
|
VLC_UNUSED( sent );
|
|
services_discovery_t *p_sd = (services_discovery_t *)data;
|
|
p_sd->p_sys->i_tracks_num = total;
|
|
return 0;
|
|
}
|