ShuriZma
/
xemu
mirror of https://github.com/xemu-project/xemu.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

usb-mtp: refactor the flow of usb_mtp_write_data 

There's no functional change but the flow is (hopefully)
more consistent for both file and folder object types.

Signed-off-by: Bandan Das <bsd@redhat.com>
Message-Id: <20190401211712.19012-4-bsd@redhat.com>
Signed-off-by: Gerd Hoffmann <kraxel@redhat.com>
This commit is contained in:
Bandan Das 2019-04-01 17:17:12 -04:00 committed by Gerd Hoffmann
parent 442bac16a6
commit e39b8a66d2
1 changed files with 30 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
hw/usb/dev-mtp.c
View File

@ -1599,7 +1599,7 @@ static int usb_mtp_update_object(MTPObject *parent, char *name)
return ret; return ret;
} }
static int usb_mtp_write_data(MTPState *s) static void usb_mtp_write_data(MTPState *s, uint32_t handle)
{ {
MTPData *d = s->data_out; MTPData *d = s->data_out;
MTPObject *parent = MTPObject *parent =
@ -1616,26 +1616,33 @@ static int usb_mtp_write_data(MTPState *s)
if (!parent || !s->write_pending) { if (!parent || !s->write_pending) {
usb_mtp_queue_result(s, RES_INVALID_OBJECTINFO, d->trans, usb_mtp_queue_result(s, RES_INVALID_OBJECTINFO, d->trans,
0, 0, 0, 0); 0, 0, 0, 0);
return 1; return;
} }
if (s->dataset.filename) { if (s->dataset.filename) {
path = g_strdup_printf("%s/%s", parent->path, s->dataset.filename); path = g_strdup_printf("%s/%s", parent->path, s->dataset.filename);
if (s->dataset.format == FMT_ASSOCIATION) { if (s->dataset.format == FMT_ASSOCIATION) {
ret = mkdir(path, mask); ret = mkdir(path, mask);
goto free; if (!ret) {
usb_mtp_queue_result(s, RES_OK, d->trans, 3,
QEMU_STORAGE_ID,
s->dataset.parent_handle,
handle);
goto close;
}
goto done;
} }
d->fd = open(path, O_CREAT | O_WRONLY | d->fd = open(path, O_CREAT | O_WRONLY |
O_CLOEXEC | O_NOFOLLOW, mask); O_CLOEXEC | O_NOFOLLOW, mask);
if (d->fd == -1) { if (d->fd == -1) {
usb_mtp_queue_result(s, RES_STORE_FULL, d->trans, ret = 1;
0, 0, 0, 0);
goto done; goto done;
} }
/* Return success if initiator sent 0 sized data */ /* Return success if initiator sent 0 sized data */
if (!s->dataset.size) { if (!s->dataset.size) {
goto success; goto done;
} }
if (d->length != MTP_WRITE_BUF_SZ && !d->pending) { if (d->length != MTP_WRITE_BUF_SZ && !d->pending) {
d->write_status = WRITE_END; d->write_status = WRITE_END;
@ -1647,13 +1654,12 @@ static int usb_mtp_write_data(MTPState *s)
rc = write_retry(d->fd, d->data, d->data_offset, rc = write_retry(d->fd, d->data, d->data_offset,
d->offset - d->data_offset); d->offset - d->data_offset);
if (rc != d->data_offset) { if (rc != d->data_offset) {
usb_mtp_queue_result(s, RES_STORE_FULL, d->trans, ret = 1;
0, 0, 0, 0);
goto done; goto done;
} }
if (d->write_status != WRITE_END) { if (d->write_status != WRITE_END) {
g_free(path); g_free(path);
return ret; return;
} else { } else {
/* /*
* Return an incomplete transfer if file size doesn't match * Return an incomplete transfer if file size doesn't match
@ -1665,16 +1671,20 @@ static int usb_mtp_write_data(MTPState *s)
usb_mtp_update_object(parent, s->dataset.filename)) { usb_mtp_update_object(parent, s->dataset.filename)) {
usb_mtp_queue_result(s, RES_INCOMPLETE_TRANSFER, d->trans, usb_mtp_queue_result(s, RES_INCOMPLETE_TRANSFER, d->trans,
0, 0, 0, 0); 0, 0, 0, 0);
goto done; goto close;
} }
} }
} }
success:
usb_mtp_queue_result(s, RES_OK, d->trans,
0, 0, 0, 0);
done: done:
if (ret) {
usb_mtp_queue_result(s, RES_STORE_FULL, d->trans,
0, 0, 0, 0);
} else {
usb_mtp_queue_result(s, RES_OK, d->trans,
0, 0, 0, 0);
}
close:
/* /*
* The write dataset is kept around and freed only * The write dataset is kept around and freed only
* on success or if another write request comes in * on success or if another write request comes in
@ -1683,12 +1693,10 @@ done:
close(d->fd); close(d->fd);
d->fd = -1; d->fd = -1;
} }
free:
g_free(s->dataset.filename); g_free(s->dataset.filename);
s->dataset.size = 0; s->dataset.size = 0;
g_free(path); g_free(path);
s->write_pending = false; s->write_pending = false;
return ret;
} }
static void usb_mtp_write_metadata(MTPState *s, uint64_t dlen) static void usb_mtp_write_metadata(MTPState *s, uint64_t dlen)
@ -1732,16 +1740,11 @@ static void usb_mtp_write_metadata(MTPState *s, uint64_t dlen)
s->write_pending = true; s->write_pending = true;
if (s->dataset.format == FMT_ASSOCIATION) { if (s->dataset.format == FMT_ASSOCIATION) {
if (usb_mtp_write_data(s)) { usb_mtp_write_data(s, next_handle);
/* next_handle will be allocated to the newly created dir */ } else {
usb_mtp_queue_result(s, RES_STORE_FULL, d->trans, usb_mtp_queue_result(s, RES_OK, d->trans, 3, QEMU_STORAGE_ID,
0, 0, 0, 0); s->dataset.parent_handle, next_handle);
return;
}
} }
usb_mtp_queue_result(s, RES_OK, d->trans, 3, QEMU_STORAGE_ID,
s->dataset.parent_handle, next_handle);
} }
static void usb_mtp_get_data(MTPState *s, mtp_container *container, static void usb_mtp_get_data(MTPState *s, mtp_container *container,
@ -1821,14 +1824,14 @@ static void usb_mtp_get_data(MTPState *s, mtp_container *container,
} else { } else {
d->write_status = WRITE_START; d->write_status = WRITE_START;
} }
usb_mtp_write_data(s); usb_mtp_write_data(s, 0);
usb_mtp_data_free(s->data_out); usb_mtp_data_free(s->data_out);
s->data_out = NULL; s->data_out = NULL;
return; return;
} }
if (d->data_offset == d->length) { if (d->data_offset == d->length) {
d->pending = true; d->pending = true;
usb_mtp_write_data(s); usb_mtp_write_data(s, 0);
} }
break; break;
default: default: