qemu_aio_coroutine_enter() is (indirectly) called recursively when processing co_queue_wakeup. This can lead to stack exhaustion. This patch rewrites co_queue_wakeup in an iterative fashion (instead of recursive) with bounded memory usage to prevent stack exhaustion. qemu_co_queue_run_restart() is inlined into qemu_aio_coroutine_enter() and the qemu_coroutine_enter() call is turned into a loop to avoid recursion. There is one change that is worth mentioning: Previously, when coroutine A queued coroutine B, qemu_co_queue_run_restart() entered coroutine B from coroutine A. If A was terminating then it would still stay alive until B yielded. After this patch B is entered by A's parent so that a A can be deleted immediately if it is terminating. It is safe to make this change since B could never interact with A if it was terminating anyway. Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> Reviewed-by: Paolo Bonzini <pbonzini@redhat.com> Message-id: 20180322152834.12656-3-stefanha@redhat.com Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com>
		
			
				
	
	
		
			73 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			73 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*
 | 
						|
 * Coroutine internals
 | 
						|
 *
 | 
						|
 * Copyright (c) 2011 Kevin Wolf <kwolf@redhat.com>
 | 
						|
 *
 | 
						|
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 | 
						|
 * of this software and associated documentation files (the "Software"), to deal
 | 
						|
 * in the Software without restriction, including without limitation the rights
 | 
						|
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 | 
						|
 * copies of the Software, and to permit persons to whom the Software is
 | 
						|
 * furnished to do so, subject to the following conditions:
 | 
						|
 *
 | 
						|
 * The above copyright notice and this permission notice shall be included in
 | 
						|
 * all copies or substantial portions of the Software.
 | 
						|
 *
 | 
						|
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 | 
						|
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 | 
						|
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
 | 
						|
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 | 
						|
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 | 
						|
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 | 
						|
 * THE SOFTWARE.
 | 
						|
 */
 | 
						|
 | 
						|
#ifndef QEMU_COROUTINE_INT_H
 | 
						|
#define QEMU_COROUTINE_INT_H
 | 
						|
 | 
						|
#include "qemu/queue.h"
 | 
						|
#include "qemu/coroutine.h"
 | 
						|
 | 
						|
#define COROUTINE_STACK_SIZE (1 << 20)
 | 
						|
 | 
						|
typedef enum {
 | 
						|
    COROUTINE_YIELD = 1,
 | 
						|
    COROUTINE_TERMINATE = 2,
 | 
						|
    COROUTINE_ENTER = 3,
 | 
						|
} CoroutineAction;
 | 
						|
 | 
						|
struct Coroutine {
 | 
						|
    CoroutineEntry *entry;
 | 
						|
    void *entry_arg;
 | 
						|
    Coroutine *caller;
 | 
						|
 | 
						|
    /* Only used when the coroutine has terminated.  */
 | 
						|
    QSLIST_ENTRY(Coroutine) pool_next;
 | 
						|
 | 
						|
    size_t locks_held;
 | 
						|
 | 
						|
    /* Only used when the coroutine has yielded.  */
 | 
						|
    AioContext *ctx;
 | 
						|
 | 
						|
    /* Used to catch and abort on illegal co-routine entry.
 | 
						|
     * Will contain the name of the function that had first
 | 
						|
     * scheduled the coroutine. */
 | 
						|
    const char *scheduled;
 | 
						|
 | 
						|
    QSIMPLEQ_ENTRY(Coroutine) co_queue_next;
 | 
						|
 | 
						|
    /* Coroutines that should be woken up when we yield or terminate.
 | 
						|
     * Only used when the coroutine is running.
 | 
						|
     */
 | 
						|
    QSIMPLEQ_HEAD(, Coroutine) co_queue_wakeup;
 | 
						|
 | 
						|
    QSLIST_ENTRY(Coroutine) co_scheduled_next;
 | 
						|
};
 | 
						|
 | 
						|
Coroutine *qemu_coroutine_new(void);
 | 
						|
void qemu_coroutine_delete(Coroutine *co);
 | 
						|
CoroutineAction qemu_coroutine_switch(Coroutine *from, Coroutine *to,
 | 
						|
                                      CoroutineAction action);
 | 
						|
 | 
						|
#endif
 |