M7350/external/bluetooth/bluez/obexd/client/transfer.c

943 lines
22 KiB
C
Raw Normal View History

2024-09-09 08:57:42 +00:00
/*
*
* OBEX Client
*
* Copyright (C) 2007-2010 Marcel Holtmann <marcel@holtmann.org>
* Copyright (C) 2011-2012 BMW Car IT GmbH. All rights reserved.
*
*
* 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 St, Fifth Floor, Boston, MA 02110-1301 USA
*
*/
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <fcntl.h>
#include <unistd.h>
#include <string.h>
#include <sys/stat.h>
#include <inttypes.h>
#include <glib.h>
#include <gdbus/gdbus.h>
#include <gobex/gobex.h>
#include "dbus.h"
#include "log.h"
#include "transfer.h"
#define TRANSFER_INTERFACE "org.bluez.obex.Transfer1"
#define ERROR_INTERFACE "org.bluez.obex.Error"
#define OBC_TRANSFER_ERROR obc_transfer_error_quark()
#define FIRST_PACKET_TIMEOUT 60
static guint64 counter = 0;
struct transfer_callback {
transfer_callback_t func;
void *data;
};
enum {
TRANSFER_STATUS_QUEUED = 0,
TRANSFER_STATUS_ACTIVE,
TRANSFER_STATUS_SUSPENDED,
TRANSFER_STATUS_COMPLETE,
TRANSFER_STATUS_ERROR
};
struct obc_transfer {
GObex *obex;
uint8_t status;
GObexApparam *apparam;
guint8 op;
struct transfer_callback *callback;
DBusConnection *conn;
DBusMessage *msg;
char *session; /* Session path */
char *owner; /* Transfer initiator */
char *path; /* Transfer path */
char *filename; /* Transfer file location */
char *name; /* Transfer object name */
char *type; /* Transfer object type */
int fd;
guint req;
guint xfer;
gint64 size;
gint64 transferred;
gint64 progress;
guint progress_id;
};
static GQuark obc_transfer_error_quark(void)
{
return g_quark_from_static_string("obc-transfer-error-quark");
}
DBusMessage *obc_transfer_create_dbus_reply(struct obc_transfer *transfer,
DBusMessage *message)
{
DBusMessage *reply;
DBusMessageIter iter;
reply = dbus_message_new_method_return(message);
if (reply == NULL)
return NULL;
dbus_message_iter_init_append(reply, &iter);
dbus_message_iter_append_basic(&iter, DBUS_TYPE_OBJECT_PATH,
&transfer->path);
g_dbus_get_properties(transfer->conn, transfer->path,
TRANSFER_INTERFACE, &iter);
return reply;
}
static void abort_complete(GObex *obex, GError *err, gpointer user_data)
{
struct obc_transfer *transfer = user_data;
struct transfer_callback *callback = transfer->callback;
DBusMessage *reply;
transfer->xfer = 0;
reply = dbus_message_new_method_return(transfer->msg);
if (reply)
g_dbus_send_message(transfer->conn, reply);
dbus_message_unref(transfer->msg);
transfer->msg = NULL;
if (callback == NULL)
return;
if (err) {
callback->func(transfer, err, callback->data);
} else {
GError *abort_err;
abort_err = g_error_new(OBC_TRANSFER_ERROR, -ECANCELED, "%s",
"Transfer cancelled by user");
callback->func(transfer, abort_err, callback->data);
g_error_free(abort_err);
}
}
static DBusMessage *obc_transfer_cancel(DBusConnection *connection,
DBusMessage *message, void *user_data)
{
struct obc_transfer *transfer = user_data;
const char *sender;
sender = dbus_message_get_sender(message);
if (g_strcmp0(transfer->owner, sender) != 0)
return g_dbus_create_error(message,
ERROR_INTERFACE ".NotAuthorized",
"Not Authorized");
if (transfer->msg != NULL)
return g_dbus_create_error(message,
ERROR_INTERFACE ".InProgress",
"Cancellation already in progress");
if (transfer->req > 0) {
if (!g_obex_cancel_req(transfer->obex, transfer->req, TRUE))
return g_dbus_create_error(message,
ERROR_INTERFACE ".Failed",
"Failed");
transfer->req = 0;
}
if (transfer->xfer == 0) {
struct transfer_callback *callback = transfer->callback;
if (callback != NULL) {
GError *err;
err = g_error_new(OBC_TRANSFER_ERROR, -ECANCELED, "%s",
"Transfer cancelled by user");
callback->func(transfer, err, callback->data);
g_error_free(err);
}
return dbus_message_new_method_return(message);
}
if (transfer->progress_id != 0) {
g_source_remove(transfer->progress_id);
transfer->progress_id = 0;
}
if (!g_obex_cancel_transfer(transfer->xfer, abort_complete, transfer))
return g_dbus_create_error(message,
ERROR_INTERFACE ".Failed",
"Failed");
transfer->msg = dbus_message_ref(message);
return NULL;
}
static void transfer_set_status(struct obc_transfer *transfer, uint8_t status)
{
if (transfer->status == status)
return;
transfer->status = status;
g_dbus_emit_property_changed(transfer->conn, transfer->path,
TRANSFER_INTERFACE, "Status");
}
static DBusMessage *obc_transfer_suspend(DBusConnection *connection,
DBusMessage *message, void *user_data)
{
struct obc_transfer *transfer = user_data;
const char *sender;
sender = dbus_message_get_sender(message);
if (g_strcmp0(transfer->owner, sender) != 0)
return g_dbus_create_error(message,
ERROR_INTERFACE ".NotAuthorized",
"Not Authorized");
if (transfer->xfer == 0)
return g_dbus_create_error(message,
ERROR_INTERFACE ".NotInProgress",
"Not in progress");
g_obex_suspend(transfer->obex);
transfer_set_status(transfer, TRANSFER_STATUS_SUSPENDED);
return g_dbus_create_reply(message, DBUS_TYPE_INVALID);
}
static DBusMessage *obc_transfer_resume(DBusConnection *connection,
DBusMessage *message, void *user_data)
{
struct obc_transfer *transfer = user_data;
const char *sender;
sender = dbus_message_get_sender(message);
if (g_strcmp0(transfer->owner, sender) != 0)
return g_dbus_create_error(message,
ERROR_INTERFACE ".NotAuthorized",
"Not Authorized");
if (transfer->xfer == 0)
return g_dbus_create_error(message,
ERROR_INTERFACE ".NotInProgress",
"Not in progress");
g_obex_resume(transfer->obex);
transfer_set_status(transfer, TRANSFER_STATUS_ACTIVE);
return g_dbus_create_reply(message, DBUS_TYPE_INVALID);
}
static gboolean name_exists(const GDBusPropertyTable *property, void *data)
{
struct obc_transfer *transfer = data;
return transfer->name != NULL;
}
static gboolean get_name(const GDBusPropertyTable *property,
DBusMessageIter *iter, void *data)
{
struct obc_transfer *transfer = data;
if (transfer->name == NULL)
return FALSE;
dbus_message_iter_append_basic(iter, DBUS_TYPE_STRING,
&transfer->name);
return TRUE;
}
static gboolean get_size(const GDBusPropertyTable *property,
DBusMessageIter *iter, void *data)
{
struct obc_transfer *transfer = data;
dbus_message_iter_append_basic(iter, DBUS_TYPE_UINT64,
&transfer->size);
return TRUE;
}
static gboolean filename_exists(const GDBusPropertyTable *property, void *data)
{
struct obc_transfer *transfer = data;
return transfer->filename != NULL;
}
static gboolean get_filename(const GDBusPropertyTable *property,
DBusMessageIter *iter, void *data)
{
struct obc_transfer *transfer = data;
if (transfer->filename == NULL)
return FALSE;
dbus_message_iter_append_basic(iter, DBUS_TYPE_STRING,
&transfer->filename);
return TRUE;
}
static gboolean transferred_exists(const GDBusPropertyTable *property,
void *data)
{
struct obc_transfer *transfer = data;
return transfer->obex != NULL;
}
static gboolean get_transferred(const GDBusPropertyTable *property,
DBusMessageIter *iter, void *data)
{
struct obc_transfer *transfer = data;
if (transfer->obex == NULL)
return FALSE;
dbus_message_iter_append_basic(iter, DBUS_TYPE_UINT64,
&transfer->progress);
return TRUE;
}
static const char *status2str(uint8_t status)
{
switch (status) {
case TRANSFER_STATUS_QUEUED:
return "queued";
case TRANSFER_STATUS_ACTIVE:
return "active";
case TRANSFER_STATUS_SUSPENDED:
return "suspended";
case TRANSFER_STATUS_COMPLETE:
return "complete";
case TRANSFER_STATUS_ERROR:
default:
return "error";
}
}
static gboolean get_status(const GDBusPropertyTable *property,
DBusMessageIter *iter, void *data)
{
struct obc_transfer *transfer = data;
const char *status = status2str(transfer->status);
dbus_message_iter_append_basic(iter, DBUS_TYPE_STRING, &status);
return TRUE;
}
static gboolean get_session(const GDBusPropertyTable *property,
DBusMessageIter *iter, void *data)
{
struct obc_transfer *transfer = data;
dbus_message_iter_append_basic(iter, DBUS_TYPE_OBJECT_PATH,
&transfer->session);
return TRUE;
}
static const GDBusMethodTable obc_transfer_methods[] = {
{ GDBUS_METHOD("Suspend", NULL, NULL, obc_transfer_suspend) },
{ GDBUS_METHOD("Resume", NULL, NULL, obc_transfer_resume) },
{ GDBUS_ASYNC_METHOD("Cancel", NULL, NULL,
obc_transfer_cancel) },
{ }
};
static const GDBusPropertyTable obc_transfer_properties[] = {
{ "Status", "s", get_status },
{ "Name", "s", get_name, NULL, name_exists },
{ "Size", "t", get_size },
{ "Filename", "s", get_filename, NULL, filename_exists },
{ "Transferred", "t", get_transferred, NULL, transferred_exists },
{ "Session", "o", get_session },
{ }
};
static void obc_transfer_free(struct obc_transfer *transfer)
{
DBG("%p", transfer);
if (transfer->req > 0)
g_obex_cancel_req(transfer->obex, transfer->req, TRUE);
if (transfer->xfer)
g_obex_cancel_transfer(transfer->xfer, NULL, NULL);
if (transfer->progress_id != 0) {
g_source_remove(transfer->progress_id);
transfer->progress_id = 0;
}
if (transfer->op == G_OBEX_OP_GET &&
transfer->status != TRANSFER_STATUS_COMPLETE &&
transfer->filename)
remove(transfer->filename);
if (transfer->fd > 0)
close(transfer->fd);
if (transfer->apparam != NULL)
g_obex_apparam_free(transfer->apparam);
if (transfer->conn)
dbus_connection_unref(transfer->conn);
if (transfer->msg)
dbus_message_unref(transfer->msg);
if (transfer->obex)
g_obex_unref(transfer->obex);
g_free(transfer->callback);
g_free(transfer->owner);
g_free(transfer->filename);
g_free(transfer->name);
g_free(transfer->type);
g_free(transfer->session);
g_free(transfer->path);
g_free(transfer);
}
static struct obc_transfer *obc_transfer_create(guint8 op,
const char *filename,
const char *name,
const char *type)
{
struct obc_transfer *transfer;
transfer = g_new0(struct obc_transfer, 1);
transfer->op = op;
transfer->filename = g_strdup(filename);
transfer->name = g_strdup(name);
transfer->type = g_strdup(type);
return transfer;
}
gboolean obc_transfer_register(struct obc_transfer *transfer,
DBusConnection *conn,
const char *session,
const char *owner,
GError **err)
{
transfer->owner = g_strdup(owner);
transfer->session = g_strdup(session);
transfer->path = g_strdup_printf("%s/transfer%ju", session, counter++);
transfer->conn = dbus_connection_ref(conn);
if (transfer->conn == NULL) {
g_set_error(err, OBC_TRANSFER_ERROR, -EFAULT,
"Unable to connect to D-Bus");
return FALSE;
}
if (g_dbus_register_interface(transfer->conn, transfer->path,
TRANSFER_INTERFACE,
obc_transfer_methods, NULL,
obc_transfer_properties, transfer,
NULL) == FALSE) {
g_set_error(err, OBC_TRANSFER_ERROR, -EFAULT,
"Unable to register to D-Bus");
return FALSE;
}
DBG("%p registered %s", transfer, transfer->path);
return TRUE;
}
static gboolean transfer_open(struct obc_transfer *transfer, int flags,
mode_t mode, GError **err)
{
int fd;
char *filename;
if (transfer->filename != NULL && strcmp(transfer->filename, "") != 0) {
fd = open(transfer->filename, flags, mode);
if (fd < 0) {
error("open(): %s(%d)", strerror(errno), errno);
g_set_error(err, OBC_TRANSFER_ERROR, -errno,
"Unable to open file");
return FALSE;
}
goto done;
}
fd = g_file_open_tmp("obex-clientXXXXXX", &filename, err);
if (fd < 0) {
error("g_file_open_tmp(): %s", (*err)->message);
return FALSE;
}
if (transfer->filename == NULL) {
remove(filename); /* remove always only if NULL was given */
g_free(filename);
} else {
g_free(transfer->filename);
transfer->filename = filename;
}
done:
transfer->fd = fd;
return TRUE;
}
struct obc_transfer *obc_transfer_get(const char *type, const char *name,
const char *filename, GError **err)
{
struct obc_transfer *transfer;
int perr;
transfer = obc_transfer_create(G_OBEX_OP_GET, filename, name, type);
perr = transfer_open(transfer, O_WRONLY | O_CREAT | O_TRUNC, 0600, err);
if (perr < 0) {
obc_transfer_free(transfer);
return NULL;
}
return transfer;
}
struct obc_transfer *obc_transfer_put(const char *type, const char *name,
const char *filename,
const void *contents, size_t size,
GError **err)
{
struct obc_transfer *transfer;
struct stat st;
int perr;
if ((filename == NULL || strcmp(filename, "") == 0) &&
contents == NULL) {
g_set_error(err, OBC_TRANSFER_ERROR, -EINVAL,
"Invalid filename given");
return NULL;
}
transfer = obc_transfer_create(G_OBEX_OP_PUT, filename, name, type);
if (contents != NULL) {
ssize_t w;
if (!transfer_open(transfer, O_RDWR, 0, err))
goto fail;
w = write(transfer->fd, contents, size);
if (w < 0) {
perr = errno;
error("write(): %s(%d)", strerror(perr), perr);
g_set_error(err, OBC_TRANSFER_ERROR, -perr,
"Writing to file failed");
goto fail;
} else if ((size_t) w != size) {
error("Unable to write all contents to file");
g_set_error(err, OBC_TRANSFER_ERROR, -EFAULT,
"Writing all contents to file failed");
goto fail;
}
lseek(transfer->fd, 0, SEEK_SET);
} else {
if (!transfer_open(transfer, O_RDONLY, 0, err))
goto fail;
}
if (fstat(transfer->fd, &st) < 0) {
perr = errno;
error("fstat(): %s(%d)", strerror(perr), perr);
g_set_error(err, OBC_TRANSFER_ERROR, -perr,
"Unable to get file status");
goto fail;
}
transfer->size = st.st_size;
return transfer;
fail:
obc_transfer_free(transfer);
return NULL;
}
void obc_transfer_unregister(struct obc_transfer *transfer)
{
if (transfer->path) {
g_dbus_unregister_interface(transfer->conn,
transfer->path, TRANSFER_INTERFACE);
}
DBG("%p unregistered %s", transfer, transfer->path);
obc_transfer_free(transfer);
}
static gboolean get_xfer_progress(const void *buf, gsize len,
gpointer user_data)
{
struct obc_transfer *transfer = user_data;
if (transfer->fd > 0) {
int w;
w = write(transfer->fd, buf, len);
if (w < 0)
return FALSE;
transfer->transferred += w;
}
return TRUE;
}
static void xfer_complete(GObex *obex, GError *err, gpointer user_data)
{
struct obc_transfer *transfer = user_data;
struct transfer_callback *callback = transfer->callback;
transfer->xfer = 0;
if (transfer->progress_id != 0) {
g_source_remove(transfer->progress_id);
transfer->progress_id = 0;
}
if (err)
transfer_set_status(transfer, TRANSFER_STATUS_ERROR);
else
transfer_set_status(transfer, TRANSFER_STATUS_COMPLETE);
if (callback)
callback->func(transfer, err, callback->data);
}
static void get_xfer_progress_first(GObex *obex, GError *err, GObexPacket *rsp,
gpointer user_data)
{
struct obc_transfer *transfer = user_data;
GObexPacket *req;
GObexHeader *hdr;
GObexApparam *apparam;
const guint8 *buf;
gsize len;
guint8 rspcode;
gboolean final;
if (err != NULL) {
xfer_complete(obex, err, transfer);
return;
}
rspcode = g_obex_packet_get_operation(rsp, &final);
if (rspcode != G_OBEX_RSP_SUCCESS && rspcode != G_OBEX_RSP_CONTINUE) {
err = g_error_new(OBC_TRANSFER_ERROR, rspcode, "%s",
g_obex_strerror(rspcode));
xfer_complete(obex, err, transfer);
g_error_free(err);
return;
}
hdr = g_obex_packet_get_header(rsp, G_OBEX_HDR_LENGTH);
if (hdr) {
uint32_t len;
if (g_obex_header_get_uint32(hdr, &len)) {
transfer->size = len;
g_dbus_emit_property_changed(transfer->conn,
transfer->path,
TRANSFER_INTERFACE, "Size");
}
}
hdr = g_obex_packet_get_header(rsp, G_OBEX_HDR_APPARAM);
if (hdr) {
apparam = g_obex_header_get_apparam(hdr);
if (apparam != NULL)
obc_transfer_set_apparam(transfer, apparam);
}
hdr = g_obex_packet_get_body(rsp);
if (hdr) {
g_obex_header_get_bytes(hdr, &buf, &len);
if (len != 0)
get_xfer_progress(buf, len, transfer);
}
if (rspcode == G_OBEX_RSP_SUCCESS) {
transfer->req = 0;
xfer_complete(obex, err, transfer);
return;
}
if (g_obex_srm_active(obex) ||
transfer->status == TRANSFER_STATUS_SUSPENDED)
return;
transfer->req = 0;
req = g_obex_packet_new(G_OBEX_OP_GET, TRUE, G_OBEX_HDR_INVALID);
transfer->xfer = g_obex_get_req_pkt(obex, req, get_xfer_progress,
xfer_complete, transfer,
&err);
}
static gssize put_xfer_progress(void *buf, gsize len, gpointer user_data)
{
struct obc_transfer *transfer = user_data;
gssize size;
size = read(transfer->fd, buf, len);
if (size <= 0)
return size;
transfer->transferred += size;
return size;
}
gboolean obc_transfer_set_callback(struct obc_transfer *transfer,
transfer_callback_t func,
void *user_data)
{
struct transfer_callback *callback;
if (transfer->callback != NULL)
return FALSE;
callback = g_new0(struct transfer_callback, 1);
callback->func = func;
callback->data = user_data;
transfer->callback = callback;
return TRUE;
}
static gboolean report_progress(gpointer data)
{
struct obc_transfer *transfer = data;
if (transfer->transferred == transfer->progress)
return TRUE;
transfer->progress = transfer->transferred;
if (transfer->transferred == transfer->size) {
transfer->progress_id = 0;
return FALSE;
}
if (transfer->status != TRANSFER_STATUS_ACTIVE &&
transfer->status != TRANSFER_STATUS_SUSPENDED)
transfer_set_status(transfer, TRANSFER_STATUS_ACTIVE);
g_dbus_emit_property_changed(transfer->conn, transfer->path,
TRANSFER_INTERFACE, "Transferred");
return TRUE;
}
static gboolean transfer_start_get(struct obc_transfer *transfer, GError **err)
{
GObexPacket *req;
GObexHeader *hdr;
if (transfer->xfer > 0) {
g_set_error(err, OBC_TRANSFER_ERROR, -EALREADY,
"Transfer already started");
return FALSE;
}
req = g_obex_packet_new(G_OBEX_OP_GET, TRUE, G_OBEX_HDR_INVALID);
if (transfer->name != NULL)
g_obex_packet_add_unicode(req, G_OBEX_HDR_NAME,
transfer->name);
if (transfer->type != NULL)
g_obex_packet_add_bytes(req, G_OBEX_HDR_TYPE, transfer->type,
strlen(transfer->type) + 1);
if (transfer->apparam != NULL) {
hdr = g_obex_header_new_apparam(transfer->apparam);
g_obex_packet_add_header(req, hdr);
}
transfer->req = g_obex_send_req(transfer->obex, req,
FIRST_PACKET_TIMEOUT,
get_xfer_progress_first,
transfer, err);
if (transfer->req == 0)
return FALSE;
if (transfer->path == NULL)
return TRUE;
transfer->progress_id = g_timeout_add_seconds(1, report_progress,
transfer);
return TRUE;
}
static gboolean transfer_start_put(struct obc_transfer *transfer, GError **err)
{
GObexPacket *req;
GObexHeader *hdr;
if (transfer->xfer > 0) {
g_set_error(err, OBC_TRANSFER_ERROR, -EALREADY,
"Transfer already started");
return FALSE;
}
req = g_obex_packet_new(G_OBEX_OP_PUT, FALSE, G_OBEX_HDR_INVALID);
if (transfer->name != NULL)
g_obex_packet_add_unicode(req, G_OBEX_HDR_NAME,
transfer->name);
if (transfer->type != NULL)
g_obex_packet_add_bytes(req, G_OBEX_HDR_TYPE, transfer->type,
strlen(transfer->type) + 1);
if (transfer->size < UINT32_MAX)
g_obex_packet_add_uint32(req, G_OBEX_HDR_LENGTH, transfer->size);
if (transfer->apparam != NULL) {
hdr = g_obex_header_new_apparam(transfer->apparam);
g_obex_packet_add_header(req, hdr);
}
transfer->xfer = g_obex_put_req_pkt(transfer->obex, req,
put_xfer_progress, xfer_complete,
transfer, err);
if (transfer->xfer == 0)
return FALSE;
if (transfer->path == NULL)
return TRUE;
transfer->progress_id = g_timeout_add_seconds(1, report_progress,
transfer);
return TRUE;
}
gboolean obc_transfer_start(struct obc_transfer *transfer, void *obex,
GError **err)
{
transfer->obex = g_obex_ref(obex);
switch (transfer->op) {
case G_OBEX_OP_GET:
return transfer_start_get(transfer, err);
case G_OBEX_OP_PUT:
return transfer_start_put(transfer, err);
}
g_set_error(err, OBC_TRANSFER_ERROR, -ENOTSUP, "Not supported");
return FALSE;
}
guint8 obc_transfer_get_operation(struct obc_transfer *transfer)
{
return transfer->op;
}
void obc_transfer_set_apparam(struct obc_transfer *transfer, void *data)
{
if (transfer->apparam != NULL)
g_obex_apparam_free(transfer->apparam);
if (data == NULL)
return;
transfer->apparam = data;
}
void *obc_transfer_get_apparam(struct obc_transfer *transfer)
{
return transfer->apparam;
}
int obc_transfer_get_contents(struct obc_transfer *transfer, char **contents,
size_t *size)
{
struct stat st;
ssize_t ret;
if (contents == NULL)
return -EINVAL;
if (fstat(transfer->fd, &st) < 0) {
error("fstat(): %s(%d)", strerror(errno), errno);
return -errno;
}
if (lseek(transfer->fd, 0, SEEK_SET) < 0) {
error("lseek(): %s(%d)", strerror(errno), errno);
return -errno;
}
*contents = g_malloc(st.st_size + 1);
ret = read(transfer->fd, *contents, st.st_size);
if (ret < 0) {
error("read(): %s(%d)", strerror(errno), errno);
g_free(*contents);
return -errno;
}
(*contents)[ret] = '\0';
if (size)
*size = ret;
return 0;
}
const char *obc_transfer_get_path(struct obc_transfer *transfer)
{
return transfer->path;
}
gint64 obc_transfer_get_size(struct obc_transfer *transfer)
{
return transfer->size;
}