From f4a5b6623bd4d1bfc2c0fd50af505b35b86599c9 Mon Sep 17 00:00:00 2001 From: Thibault Saunier Date: Wed, 4 Oct 2023 11:09:37 -0300 Subject: [PATCH] adaptivedemux2: Do not submit_transfer when cancelled There is a race condition where transfer has not been submitted yet while the request is cancelled which leads to the transfer state going back to `DOWNLOAD_REQUEST_STATE_OPEN` and the user of the request to get signalled about its completion (and the task actually happening after it was cancelled) leading to assertions and misbehaviours. To ensure that this race can't happen, we start differentiating between the UNSENT and CANCELLED states as in the normal case, when entering `submit_request` the state is UNSENT and at that point we need to know that it is not because the request has been cancelled. In practice this case lead to an assertion in `gst_adaptive_demux2_stream_begin_download_uri` because in a previous call to `gst_adaptive_demux2_stream_stop_default` we cancelled the previous request and setup a new one while it had not been submitted yet and then got a `on_download_complete` callback called from that previous cancelled request and then we tried to do `download_request_set_uri` on a request that was still `in_use`, leading to something like: ``` #0: 0x0000000186655ec8 g_assert (request->in_use == FALSE)assert.c:0 #1: 0x00000001127236b8 libgstadaptivedemux2.dylib`download_request_set_uri(request=0x000060000017cc00, uri="https://XXX/chunk-stream1-00002.webm", range_start=0, range_end=-1) at downloadrequest.c:361 #2: 0x000000011271cee8 libgstadaptivedemux2.dylib`gst_adaptive_demux2_stream_begin_download_uri(stream=0x00000001330f1800, uri="https://XXX/chunk-stream1-00002.webm", start=0, end=-1) at gstadaptivedemux-stream.c:1447 #3: 0x0000000112719898 libgstadaptivedemux2.dylib`gst_adaptive_demux2_stream_load_a_fragment [inlined] gst_adaptive_demux2_stream_download_fragment(stream=0x00000001330f1800) at gstadaptivedemux-stream.c:0 #4: 0x00000001127197f8 libgstadaptivedemux2.dylib`gst_adaptive_demux2_stream_load_a_fragment(stream=0x00000001330f1800) at gstadaptivedemux-stream.c:1969 #5: 0x000000011271c2a4 libgstadaptivedemux2.dylib`gst_adaptive_demux2_stream_next_download(stream=0x00000001330f1800) at gstadaptivedemux-stream.c:2112 ``` Part-of: --- .../ext/adaptivedemux2/downloadhelper.c | 37 +++++++++++-------- .../ext/adaptivedemux2/downloadrequest.c | 1 + .../ext/adaptivedemux2/downloadrequest.h | 1 + 3 files changed, 24 insertions(+), 15 deletions(-) diff --git a/subprojects/gst-plugins-good/ext/adaptivedemux2/downloadhelper.c b/subprojects/gst-plugins-good/ext/adaptivedemux2/downloadhelper.c index 3e7197eecf..1dd9a29755 100644 --- a/subprojects/gst-plugins-good/ext/adaptivedemux2/downloadhelper.c +++ b/subprojects/gst-plugins-good/ext/adaptivedemux2/downloadhelper.c @@ -268,7 +268,7 @@ on_read_ready (GObject * source, GAsyncResult * result, gpointer user_data) if (!g_cancellable_is_cancelled (transfer->cancellable)) { GST_ERROR ("Failed to read stream: %s", error->message); - if (request->state != DOWNLOAD_REQUEST_STATE_UNSENT) + if (request->state != DOWNLOAD_REQUEST_STATE_CANCELLED) request->state = DOWNLOAD_REQUEST_STATE_ERROR; finish_transfer_task (dh, transfer_task, error); } else { @@ -316,9 +316,8 @@ on_read_ready (GObject * source, GAsyncResult * result, gpointer user_data) } if (gst_buffer != NULL) { - /* Unsent means cancellation is in progress, so don't override - * the state. Otherwise make sure it is LOADING */ - if (request->state != DOWNLOAD_REQUEST_STATE_UNSENT) + /* Don't override CANCELLED state. Otherwise make sure it is LOADING */ + if (request->state != DOWNLOAD_REQUEST_STATE_CANCELLED) request->state = DOWNLOAD_REQUEST_STATE_LOADING; GST_LOG ("Adding %u bytes to buffer", @@ -354,12 +353,13 @@ finish_transfer: G_GINT64_FORMAT, status_code, request->uri, request->range_start, request->range_end); - if (request->state != DOWNLOAD_REQUEST_STATE_UNSENT) { + if (request->state != DOWNLOAD_REQUEST_STATE_CANCELLED) { if (SOUP_STATUS_IS_SUCCESSFUL (status_code) - || SOUP_STATUS_IS_REDIRECTION (status_code)) + || SOUP_STATUS_IS_REDIRECTION (status_code)) { request->state = DOWNLOAD_REQUEST_STATE_COMPLETE; - else + } else { request->state = DOWNLOAD_REQUEST_STATE_ERROR; + } } } request->download_end_time = now; @@ -529,7 +529,7 @@ on_request_sent (GObject * source, GAsyncResult * result, gpointer user_data) G_GINT64_FORMAT, request->status_code, request->uri, request->range_start, request->range_end); - if (request->state != DOWNLOAD_REQUEST_STATE_UNSENT) + if (request->state != DOWNLOAD_REQUEST_STATE_CANCELLED) request->state = DOWNLOAD_REQUEST_STATE_ERROR; finish_transfer_task (dh, transfer_task, error); } else { @@ -544,8 +544,8 @@ on_request_sent (GObject * source, GAsyncResult * result, gpointer user_data) return; } - /* If the state went back to UNSENT, we were cancelled so don't override it */ - if (request->state != DOWNLOAD_REQUEST_STATE_UNSENT && + /* If the state is cancelled don't override it */ + if (request->state != DOWNLOAD_REQUEST_STATE_CANCELLED && request->state != DOWNLOAD_REQUEST_STATE_HEADERS_RECEIVED) { request->state = DOWNLOAD_REQUEST_STATE_HEADERS_RECEIVED; @@ -586,7 +586,9 @@ finish_transfer_error: GST_LOG ("request complete. Code %d URI %s range %" G_GINT64_FORMAT " %" G_GINT64_FORMAT, _soup_message_get_status (msg), request->uri, request->range_start, request->range_end); - if (request->state != DOWNLOAD_REQUEST_STATE_UNSENT) + + /* If the state is cancelled don't override it */ + if (request->state != DOWNLOAD_REQUEST_STATE_CANCELLED) request->state = DOWNLOAD_REQUEST_STATE_ERROR; } @@ -690,6 +692,13 @@ submit_transfer (DownloadHelper * dh, GTask * transfer_task) DownloadRequest *request = transfer->request; download_request_lock (request); + if (request->state == DOWNLOAD_REQUEST_STATE_CANCELLED) { + download_request_unlock (request); + + GST_DEBUG ("Don't submit already cancelled transfer"); + return; + } + request->state = DOWNLOAD_REQUEST_STATE_OPEN; request->download_request_time = gst_adaptive_demux_clock_get_time (dh->clock); @@ -800,8 +809,7 @@ downloadhelper_stop (DownloadHelper * dh) DownloadRequest *request = transfer->request; download_request_lock (request); - /* Reset the state to UNSENT, to indicate cancellation, like an XMLHttpRequest does */ - request->state = DOWNLOAD_REQUEST_STATE_UNSENT; + request->state = DOWNLOAD_REQUEST_STATE_CANCELLED; download_request_unlock (request); transfer->complete = TRUE; @@ -967,8 +975,7 @@ downloadhelper_cancel_request (DownloadHelper * dh, DownloadRequest * request) GST_DEBUG ("Cancelling request for URI %s range %" G_GINT64_FORMAT " %" G_GINT64_FORMAT, request->uri, request->range_start, request->range_end); - request->state = DOWNLOAD_REQUEST_STATE_UNSENT; - + request->state = DOWNLOAD_REQUEST_STATE_CANCELLED; for (i = dh->active_transfers->len - 1; i >= 0; i--) { GTask *transfer_task = g_array_index (dh->active_transfers, GTask *, i); DownloadHelperTransfer *transfer = g_task_get_task_data (transfer_task); diff --git a/subprojects/gst-plugins-good/ext/adaptivedemux2/downloadrequest.c b/subprojects/gst-plugins-good/ext/adaptivedemux2/downloadrequest.c index a40878c2ec..175e8ef9c3 100644 --- a/subprojects/gst-plugins-good/ext/adaptivedemux2/downloadrequest.c +++ b/subprojects/gst-plugins-good/ext/adaptivedemux2/downloadrequest.c @@ -161,6 +161,7 @@ download_request_despatch_completion (DownloadRequest * request) DownloadRequestPrivate *priv = DOWNLOAD_REQUEST_PRIVATE (request); switch (request->state) { case DOWNLOAD_REQUEST_STATE_UNSENT: + case DOWNLOAD_REQUEST_STATE_CANCELLED: if (priv->cancellation_cb) priv->cancellation_cb (request, request->state, priv->cb_data); break; diff --git a/subprojects/gst-plugins-good/ext/adaptivedemux2/downloadrequest.h b/subprojects/gst-plugins-good/ext/adaptivedemux2/downloadrequest.h index b814d921b3..a3cf8cb2a2 100644 --- a/subprojects/gst-plugins-good/ext/adaptivedemux2/downloadrequest.h +++ b/subprojects/gst-plugins-good/ext/adaptivedemux2/downloadrequest.h @@ -41,6 +41,7 @@ enum _DownloadRequestState { DOWNLOAD_REQUEST_STATE_LOADING, /* Content loading in progress */ DOWNLOAD_REQUEST_STATE_COMPLETE, /* Request processing finished successfully - check status_code for completion 200-399 codes */ DOWNLOAD_REQUEST_STATE_ERROR, /* Request generated an http error - check status_code */ + DOWNLOAD_REQUEST_STATE_CANCELLED, /* Request has been cancelled by the user */ }; struct _DownloadRequest