Return-Path: From: Mikel Astiz To: linux-bluetooth@vger.kernel.org Cc: Mikel Astiz Subject: [PATCH obexd v1 08/11] client: open transfer file during creation Date: Tue, 10 Apr 2012 15:38:29 +0200 Message-Id: <1334065112-14966-9-git-send-email-mikel.astiz.oss@gmail.com> In-Reply-To: <1334065112-14966-1-git-send-email-mikel.astiz.oss@gmail.com> References: <1334065112-14966-1-git-send-email-mikel.astiz.oss@gmail.com> Sender: linux-bluetooth-owner@vger.kernel.org List-ID: From: Mikel Astiz For file-based transfers, it is safer to open the file during the creation of the transfer, instead of waiting until the transfer is actually started. This also simplifies the transfer api, by make obc_transfer_set_file unnecessary. This behavior is slightly more complicated when an agent is being used, since it can overwrite the filename for get operations. This will only be known once the transfer has been authorized, that is, when it's starting. --- client/session.c | 48 ++++++++++++------- client/transfer.c | 137 ++++++++++++++++++++++++++++++++++------------------ client/transfer.h | 12 +++-- 3 files changed, 126 insertions(+), 71 deletions(-) diff --git a/client/session.c b/client/session.c index 679627b..3abb342 100644 --- a/client/session.c +++ b/client/session.c @@ -688,8 +688,16 @@ static void session_request_reply(DBusPendingCall *call, gpointer user_data) if (strlen(name)) { if (obc_transfer_get_cmd(transfer) == OBC_TRANSFER_PUT) obc_transfer_set_name(transfer, name); - else - obc_transfer_set_filename(transfer, name); + else { + GError *gerr = NULL; + + if (!obc_transfer_set_filename(transfer, name, &gerr)) { + session_terminate_transfer(session, transfer, + gerr); + g_clear_error(&gerr); + return; + } + } } if (p->auth_complete) @@ -938,6 +946,7 @@ int obc_session_get(struct obc_session *session, const char *type, struct obc_transfer *transfer; struct obc_transfer_params *params = NULL; const char *agent; + GError *err = NULL; if (session->obex == NULL) return -ENOTCONN; @@ -958,15 +967,19 @@ int obc_session_get(struct obc_session *session, const char *type, transfer = obc_transfer_create(session->conn, agent, OBC_TRANSFER_GET, targetfile, name, type, - params); + params, &err); else transfer = obc_transfer_create_mem(session->conn, agent, OBC_TRANSFER_GET, NULL, 0, NULL, - name, type, params); + name, type, params, + &err); - if (transfer == NULL) - return -EIO; + if (transfer == NULL) { + int result = err->code; + g_error_free(err); + return result; + } return session_request(session, transfer, func, user_data); } @@ -976,7 +989,7 @@ int obc_session_send(struct obc_session *session, const char *filename, { struct obc_transfer *transfer; const char *agent; - int err; + GError *err = NULL; if (session->obex == NULL) return -ENOTCONN; @@ -986,14 +999,11 @@ int obc_session_send(struct obc_session *session, const char *filename, transfer = obc_transfer_create(session->conn, agent, OBC_TRANSFER_PUT, filename, - name, NULL, NULL); - if (transfer == NULL) - return -EINVAL; - - err = obc_transfer_set_file(transfer); - if (err < 0) { - obc_transfer_unregister(transfer); - return err; + name, NULL, NULL, &err); + if (transfer == NULL) { + int result = err->code; + g_error_free(err); + return result; } return session_request(session, transfer, NULL, NULL); @@ -1041,6 +1051,7 @@ int obc_session_put(struct obc_session *session, char *buf, const char *name) { struct obc_transfer *transfer; const char *agent; + GError *err = NULL; if (session->obex == NULL) { g_free(buf); @@ -1053,10 +1064,11 @@ int obc_session_put(struct obc_session *session, char *buf, const char *name) OBC_TRANSFER_PUT, buf, strlen(buf), g_free, name, NULL, - NULL); + NULL, &err); if (transfer == NULL) { - g_free(buf); - return -EIO; + int result = err->code; + g_error_free(err); + return result; } return session_request(session, transfer, NULL, NULL); diff --git a/client/transfer.c b/client/transfer.c index a177eb3..9e6aab2 100644 --- a/client/transfer.c +++ b/client/transfer.c @@ -28,6 +28,7 @@ #include #include #include +#include #include #include #include @@ -311,7 +312,8 @@ static struct obc_transfer *transfer_create(DBusConnection *conn, const char *type, struct obc_transfer_params *params, struct file_location *file_location, - struct mem_location *mem_location) + struct mem_location *mem_location, + GError **err) { struct obc_transfer *transfer; @@ -340,6 +342,8 @@ static struct obc_transfer *transfer_create(DBusConnection *conn, transfer->conn = dbus_bus_get(DBUS_BUS_SESSION, NULL); if (transfer->conn == NULL) { obc_transfer_free(transfer); + g_set_error(err, OBC_TRANSFER_ERROR, -EIO, + "Unable to register transfer"); return NULL; } @@ -348,6 +352,8 @@ static struct obc_transfer *transfer_create(DBusConnection *conn, obc_transfer_methods, NULL, NULL, transfer, NULL) == FALSE) { obc_transfer_free(transfer); + g_set_error(err, OBC_TRANSFER_ERROR, -EIO, + "Unable to register transfer"); return NULL; } @@ -369,23 +375,71 @@ void obc_transfer_unregister(struct obc_transfer *transfer) obc_transfer_free(transfer); } +static gboolean transfer_open_file(struct obc_transfer *transfer, GError **err) +{ + struct file_location *location = transfer->file_location; + int fd; + struct stat st; + + if (transfer->command == OBC_TRANSFER_PUT) { + DBG("opening file: %s", location->filename); + fd = open(location->filename, O_RDONLY); + } else { + fd = open(location->filename ? : transfer->name, + O_WRONLY | O_CREAT | O_TRUNC, 0600); + DBG("creating file: %s", location->filename); + } + + if (fd < 0) { + error("open(): %s (%d)", strerror(errno), errno); + g_set_error(err, OBC_TRANSFER_ERROR, -EIO, + "Cannot open file"); + return FALSE; + } + + if (transfer->command == OBC_TRANSFER_PUT) { + if (fstat(fd, &st) < 0) { + error("fstat(): %s (%d)", strerror(errno), errno); + g_set_error(err, OBC_TRANSFER_ERROR, -EIO, + "Cannot get file size"); + return FALSE; + } + + transfer->size = st.st_size; + } + + location->fd = fd; + return TRUE; +} + struct obc_transfer *obc_transfer_create(DBusConnection *conn, const char *agent, ObcTransferCommand cmd, const char *filename, const char *name, const char *type, - struct obc_transfer_params *params) + struct obc_transfer_params *params, + GError **err) { struct file_location *file_location; + struct obc_transfer *transfer; assert(filename != NULL); file_location = g_malloc0(sizeof(*file_location)); file_location->filename = g_strdup(filename); - return transfer_create(conn, agent, cmd, name, type, params, - file_location, NULL); + transfer = transfer_create(conn, agent, cmd, name, type, params, + file_location, NULL, err); + if (transfer == NULL) + return NULL; + + if (!transfer_open_file(transfer, err)) { + obc_transfer_free(transfer); + return NULL; + } + + return transfer; } struct obc_transfer *obc_transfer_create_mem(DBusConnection *conn, @@ -395,7 +449,8 @@ struct obc_transfer *obc_transfer_create_mem(DBusConnection *conn, GDestroyNotify buffer_destroy_func, const char *name, const char *type, - struct obc_transfer_params *params) + struct obc_transfer_params *params, + GError **err) { struct mem_location *mem_location; struct obc_transfer *transfer; @@ -419,7 +474,7 @@ struct obc_transfer *obc_transfer_create_mem(DBusConnection *conn, } transfer = transfer_create(conn, agent, cmd, name, type, params, - NULL, mem_location); + NULL, mem_location, err); if (transfer == NULL) return NULL; @@ -654,18 +709,6 @@ static gboolean transfer_start_get(struct obc_transfer *transfer, GError **err) if (transfer->mem_location != NULL) rsp_cb = get_buf_xfer_progress; else { - const char *filename = transfer->file_location->filename; - int fd = open(filename ? : transfer->name, - O_WRONLY | O_CREAT, 0600); - - if (fd < 0) { - error("open(): %s(%d)", strerror(errno), errno); - g_set_error(err, OBC_TRANSFER_ERROR, -EIO, - "Cannot open file"); - return FALSE; - } - - transfer->file_location->fd = fd; data_cb = get_xfer_progress; complete_cb = xfer_complete; } @@ -791,11 +834,35 @@ void obc_transfer_set_name(struct obc_transfer *transfer, const char *name) transfer->name = g_strdup(name); } -void obc_transfer_set_filename(struct obc_transfer *transfer, - const char *filename) +gboolean obc_transfer_set_filename(struct obc_transfer *transfer, + const char *filename, + GError **err) { - g_free(transfer->file_location->filename); - transfer->file_location->filename = g_strdup(filename); + struct file_location *location = transfer->file_location; + char *old = location->filename; + + location->filename = g_strdup(filename); + + if ((old == NULL) || (filename == NULL) || !g_strcmp0(old, filename)) + goto done; + + if (location->fd > 0) { + close(location->fd); + location->fd = 0; + + if ((transfer->command == OBC_TRANSFER_GET) && + (transfer->file_location != NULL)) + unlink(transfer->file_location->filename); + } + + if (!transfer_open_file(transfer, err)) { + g_free(old); + return FALSE; + } + +done: + g_free(old); + return TRUE; } const char *obc_transfer_get_path(struct obc_transfer *transfer) @@ -807,29 +874,3 @@ gint64 obc_transfer_get_size(struct obc_transfer *transfer) { return transfer->size; } - -int obc_transfer_set_file(struct obc_transfer *transfer) -{ - int fd; - struct stat st; - - if (transfer->file_location != NULL) - return 0; - - fd = open(transfer->file_location->filename, O_RDONLY); - if (fd < 0) { - error("open(): %s(%d)", strerror(errno), errno); - return -errno; - } - - if (fstat(fd, &st) < 0) { - error("fstat(): %s(%d)", strerror(errno), errno); - close(fd); - return -errno; - } - - transfer->file_location->fd = fd; - transfer->size = st.st_size; - - return 0; -} diff --git a/client/transfer.h b/client/transfer.h index 09c708d..166d08c 100644 --- a/client/transfer.h +++ b/client/transfer.h @@ -44,7 +44,8 @@ struct obc_transfer *obc_transfer_create(DBusConnection *conn, const char *filename, const char *name, const char *type, - struct obc_transfer_params *params); + struct obc_transfer_params *params, + GError **err); /* similar as above, but from memory. for get operations, buffer must be NULL */ struct obc_transfer *obc_transfer_create_mem(DBusConnection *conn, @@ -54,7 +55,8 @@ struct obc_transfer *obc_transfer_create_mem(DBusConnection *conn, GDestroyNotify buffer_destroy_func, const char *name, const char *type, - struct obc_transfer_params *params); + struct obc_transfer_params *params, + GError **err); void obc_transfer_unregister(struct obc_transfer *transfer); @@ -72,8 +74,8 @@ const void *obc_transfer_get_params(struct obc_transfer *transfer, const void *obc_transfer_get_buffer(struct obc_transfer *transfer, size_t *size); void obc_transfer_set_name(struct obc_transfer *transfer, const char *name); -void obc_transfer_set_filename(struct obc_transfer *transfer, - const char *filename); +gboolean obc_transfer_set_filename(struct obc_transfer *transfer, + const char *filename, + GError **err); const char *obc_transfer_get_path(struct obc_transfer *transfer); gint64 obc_transfer_get_size(struct obc_transfer *transfer); -int obc_transfer_set_file(struct obc_transfer *transfer); -- 1.7.7.6