369 lines
11 KiB
C
369 lines
11 KiB
C
/* This file is part of Clementine.
|
|
Copyright 2010, David Sansome <me@davidsansome.com>
|
|
|
|
Clementine 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 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
Clementine 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 Clementine. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "gstafcsrc.h"
|
|
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <gst/gst.h>
|
|
|
|
|
|
// Signals
|
|
enum {
|
|
LAST_SIGNAL
|
|
};
|
|
|
|
// Properties
|
|
enum {
|
|
PROP_0,
|
|
PROP_LOCATION,
|
|
};
|
|
|
|
GST_DEBUG_CATEGORY_STATIC(gst_afc_src_debug);
|
|
#define GST_CAT_DEFAULT gst_afc_src_debug
|
|
|
|
static GstStaticPadTemplate src_factory = GST_STATIC_PAD_TEMPLATE ("src",
|
|
GST_PAD_SRC,
|
|
GST_PAD_ALWAYS,
|
|
GST_STATIC_CAPS ("ANY")
|
|
);
|
|
|
|
static void gst_afc_src_interface_init(GType type);
|
|
static void gst_afc_src_uri_handler_init(gpointer iface, gpointer data);
|
|
static void gst_afc_src_set_property(GObject * object, guint prop_id, const GValue * value, GParamSpec * pspec);
|
|
static void gst_afc_src_get_property(GObject * object, guint prop_id, GValue * value, GParamSpec * pspec);
|
|
static void gst_afc_src_finalize(GObject* object);
|
|
static gboolean gst_afc_src_start(GstBaseSrc* src);
|
|
static gboolean gst_afc_src_stop(GstBaseSrc* src);
|
|
static GstFlowReturn gst_afc_src_create(GstBaseSrc* src, guint64 offset, guint length, GstBuffer** buffer);
|
|
static gboolean gst_afc_src_is_seekable(GstBaseSrc* src);
|
|
static gboolean gst_afc_src_get_size(GstBaseSrc* src, guint64* size);
|
|
|
|
GST_BOILERPLATE_FULL(GstAfcSrc, gst_afc_src, GstBaseSrc, GST_TYPE_BASE_SRC, gst_afc_src_interface_init);
|
|
|
|
|
|
static void gst_afc_src_interface_init(GType type) {
|
|
static const GInterfaceInfo urihandler_info = {
|
|
gst_afc_src_uri_handler_init,
|
|
NULL,
|
|
NULL
|
|
};
|
|
|
|
GST_DEBUG_CATEGORY_INIT (gst_afc_src_debug, "afcsrc", 0, "iPod/iPhone Source");
|
|
|
|
g_type_add_interface_static(type, GST_TYPE_URI_HANDLER, &urihandler_info);
|
|
}
|
|
|
|
static void gst_afc_src_base_init(gpointer gclass) {
|
|
GstElementClass *element_class = GST_ELEMENT_CLASS (gclass);
|
|
|
|
gst_element_class_set_details_simple(element_class,
|
|
"iPod/iPhone Source",
|
|
"Source/Afc",
|
|
"Read using libimobiledevice",
|
|
"David Sansome <me@davidsansome.com>");
|
|
|
|
gst_element_class_add_pad_template (element_class,
|
|
gst_static_pad_template_get (&src_factory));
|
|
}
|
|
|
|
static void gst_afc_src_class_init (GstAfcSrcClass* klass) {
|
|
GObjectClass* gobject_class = (GObjectClass*) klass;
|
|
GstBaseSrcClass* gstbasesrc_class = (GstBaseSrcClass*) klass;
|
|
|
|
gobject_class->set_property = gst_afc_src_set_property;
|
|
gobject_class->get_property = gst_afc_src_get_property;
|
|
gobject_class->finalize = gst_afc_src_finalize;
|
|
|
|
g_object_class_install_property(gobject_class, PROP_LOCATION,
|
|
g_param_spec_string(
|
|
"location", "URI",
|
|
"The URI of the file to read, must be of the form afc://uuid/filename", NULL,
|
|
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS | GST_PARAM_MUTABLE_READY));
|
|
|
|
gstbasesrc_class->start = gst_afc_src_start;
|
|
gstbasesrc_class->stop = gst_afc_src_stop;
|
|
gstbasesrc_class->create = gst_afc_src_create;
|
|
gstbasesrc_class->is_seekable = gst_afc_src_is_seekable;
|
|
gstbasesrc_class->get_size = gst_afc_src_get_size;
|
|
}
|
|
|
|
static GstURIType gst_afc_src_uri_get_type() {
|
|
return GST_URI_SRC;
|
|
}
|
|
|
|
static gchar** gst_afc_src_uri_get_protocols() {
|
|
static const gchar* protocols[] = { "afc", NULL };
|
|
return (char**) protocols;
|
|
}
|
|
|
|
static const gchar* gst_afc_src_uri_get_uri(GstURIHandler* handler) {
|
|
GstAfcSrc* self = GST_AFCSRC(handler);
|
|
return self->location_;
|
|
}
|
|
|
|
static gboolean gst_afc_src_uri_set_uri(GstURIHandler* handler, const gchar* uri) {
|
|
GstAfcSrc* self = GST_AFCSRC(handler);
|
|
self->location_ = g_strdup(uri);
|
|
return TRUE;
|
|
}
|
|
|
|
static void gst_afc_src_uri_handler_init(gpointer g_iface, gpointer data) {
|
|
GstURIHandlerInterface* iface = (GstURIHandlerInterface*) g_iface;
|
|
|
|
iface->get_type = gst_afc_src_uri_get_type;
|
|
iface->get_protocols = gst_afc_src_uri_get_protocols;
|
|
iface->set_uri = gst_afc_src_uri_set_uri;
|
|
iface->get_uri = gst_afc_src_uri_get_uri;
|
|
}
|
|
|
|
|
|
static void gst_afc_src_init(GstAfcSrc* element, GstAfcSrcClass* gclass) {
|
|
element->location_ = NULL;
|
|
element->uuid_ = NULL;
|
|
element->path_ = NULL;
|
|
element->connected_ = FALSE;
|
|
element->afc_ = NULL;
|
|
element->afc_port_ = 0;
|
|
element->device_ = NULL;
|
|
element->file_handle_ = 0;
|
|
element->buffer_ = NULL;
|
|
element->buffer_is_valid_ = FALSE;
|
|
element->buffer_length_ = 0;
|
|
element->buffer_offset_ = 0;
|
|
}
|
|
|
|
static void gst_afc_src_finalize(GObject* object) {
|
|
GstAfcSrc* self = GST_AFCSRC(object);
|
|
free(self->location_);
|
|
free(self->uuid_);
|
|
free(self->path_);
|
|
free(self->buffer_);
|
|
|
|
if (self->file_handle_)
|
|
afc_file_close(self->afc_, self->file_handle_);
|
|
|
|
if (self->afc_)
|
|
afc_client_free(self->afc_);
|
|
|
|
if (self->device_)
|
|
idevice_free(self->device_);
|
|
|
|
|
|
G_OBJECT_CLASS(parent_class)->finalize(object);
|
|
}
|
|
|
|
static void gst_afc_src_set_property(
|
|
GObject* object, guint prop_id, const GValue* value, GParamSpec* pspec) {
|
|
GstAfcSrc* self = GST_AFCSRC(object);
|
|
|
|
switch (prop_id) {
|
|
case PROP_LOCATION:
|
|
self->location_ = g_strdup(g_value_get_string(value));
|
|
break;
|
|
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
static void gst_afc_src_get_property(
|
|
GObject* object, guint prop_id, GValue* value, GParamSpec* pspec) {
|
|
GstAfcSrc* self = GST_AFCSRC(object);
|
|
|
|
switch (prop_id) {
|
|
case PROP_LOCATION:
|
|
g_value_set_string(value, self->location_);
|
|
break;
|
|
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
static gboolean gst_afc_src_start(GstBaseSrc* src) {
|
|
GstAfcSrc* self = GST_AFCSRC(src);
|
|
|
|
// Don't connect again
|
|
if (self->connected_)
|
|
return TRUE;
|
|
|
|
// Check that a URI has been passed
|
|
if (!self->location_ || self->location_[0] == '\0') {
|
|
GST_ELEMENT_ERROR(src, RESOURCE, NOT_FOUND, ("No URI specified"), (NULL));
|
|
return FALSE;
|
|
}
|
|
|
|
// Parse the URI
|
|
// HERE BE DRAGONS
|
|
gchar* location = gst_uri_get_location(self->location_);
|
|
char* path_pos = strstr(location, "/");
|
|
|
|
self->uuid_ = (char*) malloc(path_pos - location + 1);
|
|
memcpy(self->uuid_, location, path_pos - location);
|
|
self->uuid_[path_pos - location] = '\0';
|
|
self->path_ = g_strdup(path_pos);
|
|
|
|
g_free(location);
|
|
|
|
// Open the device
|
|
idevice_error_t err = idevice_new(&self->device_, self->uuid_);
|
|
if (err != IDEVICE_E_SUCCESS) {
|
|
GST_ELEMENT_ERROR(src, RESOURCE, NOT_FOUND, ("idevice error: %d", err), (NULL));
|
|
return FALSE;
|
|
}
|
|
|
|
lockdownd_client_t lockdown;
|
|
|
|
lockdownd_error_t lockdown_err =
|
|
lockdownd_client_new_with_handshake(self->device_, &lockdown, "GstAfcSrc");
|
|
if (lockdown_err != LOCKDOWN_E_SUCCESS) {
|
|
GST_ELEMENT_ERROR(src, RESOURCE, NOT_FOUND, ("lockdown error: %d", lockdown_err), (NULL));
|
|
return FALSE;
|
|
}
|
|
|
|
lockdown_err = lockdownd_start_service(lockdown, "com.apple.afc", &self->afc_port_);
|
|
if (lockdown_err != LOCKDOWN_E_SUCCESS) {
|
|
GST_ELEMENT_ERROR(src, RESOURCE, NOT_FOUND, ("lockdown error: %d", lockdown_err), (NULL));
|
|
lockdownd_client_free(lockdown);
|
|
return FALSE;
|
|
}
|
|
|
|
afc_error_t afc_err = afc_client_new(self->device_, self->afc_port_, &self->afc_);
|
|
if (afc_err != 0) {
|
|
GST_ELEMENT_ERROR(src, RESOURCE, NOT_FOUND, ("afc error: %d", afc_err), (NULL));
|
|
lockdownd_client_free(lockdown);
|
|
return FALSE;
|
|
}
|
|
|
|
lockdownd_client_free(lockdown);
|
|
|
|
// Try opening the file
|
|
afc_err = afc_file_open(self->afc_, self->path_, AFC_FOPEN_RDONLY, &self->file_handle_);
|
|
if (afc_err != 0) {
|
|
GST_ELEMENT_ERROR(src, RESOURCE, NOT_FOUND, ("afc error: %d", afc_err), (NULL));
|
|
return FALSE;
|
|
}
|
|
|
|
self->connected_ = TRUE;
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static gboolean gst_afc_src_stop(GstBaseSrc* src) {
|
|
return TRUE;
|
|
}
|
|
|
|
static GstFlowReturn gst_afc_src_create(GstBaseSrc* src, guint64 offset, guint length, GstBuffer** buffer) {
|
|
GstAfcSrc* self = GST_AFCSRC(src);
|
|
|
|
GstBuffer* buf = gst_buffer_try_new_and_alloc(length);
|
|
if (buf == NULL && length > 0) {
|
|
GST_ERROR_OBJECT(src, "Failed to allocate %u bytes", length);
|
|
return GST_FLOW_ERROR;
|
|
}
|
|
|
|
// Is this section within our cache?
|
|
if (!self->buffer_is_valid_ || offset < self->buffer_offset_ ||
|
|
offset + length > self->buffer_offset_ + self->buffer_length_) {
|
|
// No - read from the device to fill our internal cache.
|
|
// Always read twice the requested size so the next read(s) hit the cache too.
|
|
if (self->buffer_length_ != length * 2) {
|
|
self->buffer_ = (char*)realloc(self->buffer_, length * 2);
|
|
self->buffer_is_valid_ = TRUE;
|
|
self->buffer_length_ = length * 2;
|
|
}
|
|
self->buffer_offset_ = offset;
|
|
|
|
uint32_t bytes_read = 0;
|
|
afc_error_t err;
|
|
err = afc_file_seek(self->afc_, self->file_handle_, offset, SEEK_SET);
|
|
if (err == AFC_E_SUCCESS) {
|
|
err = afc_file_read(self->afc_, self->file_handle_, self->buffer_, length * 2, &bytes_read);
|
|
}
|
|
|
|
if (err != AFC_E_SUCCESS) {
|
|
gst_buffer_unref(buf);
|
|
return GST_FLOW_ERROR;
|
|
}
|
|
}
|
|
|
|
// Copy from our internal buffer to the output buffer
|
|
memcpy(GST_BUFFER_DATA(buf), self->buffer_ + offset - self->buffer_offset_, length);
|
|
|
|
*buffer = buf;
|
|
return GST_FLOW_OK;
|
|
}
|
|
|
|
static gboolean gst_afc_src_is_seekable(GstBaseSrc* src) {
|
|
return TRUE;
|
|
}
|
|
|
|
static gboolean gst_afc_src_get_size(GstBaseSrc* src, guint64* size) {
|
|
GstAfcSrc* self = GST_AFCSRC(src);
|
|
|
|
char** infolist = NULL;
|
|
afc_error_t err = afc_get_file_info(self->afc_, self->path_, &infolist);
|
|
if (err != AFC_E_SUCCESS || !infolist) {
|
|
GST_ELEMENT_ERROR(src, RESOURCE, NOT_FOUND, ("afc error: %d", err), (NULL));
|
|
return FALSE;
|
|
}
|
|
|
|
gboolean found_size = FALSE;
|
|
|
|
char** p = infolist;
|
|
while (*p != NULL) {
|
|
if (g_strcmp0(*p, "st_size") == 0) {
|
|
*size = strtoll(*(p+1), NULL, 0);
|
|
found_size = TRUE;
|
|
}
|
|
free(*p); ++p;
|
|
free(*p); ++p;
|
|
}
|
|
free(infolist);
|
|
|
|
if (!found_size) {
|
|
*size = 0;
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
#define PACKAGE "Clementine"
|
|
|
|
static gboolean afcsrc_init(GstPlugin* afcsrc) {
|
|
return gst_element_register(afcsrc, "afcsrc", GST_RANK_PRIMARY, GST_TYPE_AFCSRC);
|
|
}
|
|
|
|
void afcsrc_register_static() {
|
|
gst_plugin_register_static(
|
|
GST_VERSION_MAJOR,
|
|
GST_VERSION_MINOR,
|
|
"afcsrc",
|
|
"iPod/iPhone Source",
|
|
afcsrc_init,
|
|
"0.1",
|
|
"GPL",
|
|
"Clementine",
|
|
"Clementine",
|
|
"http://www.clementine-player.org/");
|
|
}
|
|
|