From patchwork Mon Jun 4 09:55:20 2018 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Xiao Guangrong X-Patchwork-Id: 10446271 Return-Path: Received: from mail.wl.linuxfoundation.org (pdx-wl-mail.web.codeaurora.org [172.30.200.125]) by pdx-korg-patchwork.web.codeaurora.org (Postfix) with ESMTP id 4B01B60375 for ; Mon, 4 Jun 2018 10:10:53 +0000 (UTC) Received: from mail.wl.linuxfoundation.org (localhost [127.0.0.1]) by mail.wl.linuxfoundation.org (Postfix) with ESMTP id 3902D28B95 for ; Mon, 4 Jun 2018 10:10:53 +0000 (UTC) Received: by mail.wl.linuxfoundation.org (Postfix, from userid 486) id 2AF7F28BCF; Mon, 4 Jun 2018 10:10:53 +0000 (UTC) X-Spam-Checker-Version: SpamAssassin 3.3.1 (2010-03-16) on pdx-wl-mail.web.codeaurora.org X-Spam-Level: X-Spam-Status: No, score=-7.8 required=2.0 tests=BAYES_00, DKIM_ADSP_CUSTOM_MED, DKIM_SIGNED, FREEMAIL_FROM, MAILING_LIST_MULTI, RCVD_IN_DNSWL_HI, T_DKIM_INVALID autolearn=unavailable version=3.3.1 Received: from lists.gnu.org (lists.gnu.org [208.118.235.17]) (using TLSv1 with cipher AES256-SHA (256/256 bits)) (No client certificate requested) by mail.wl.linuxfoundation.org (Postfix) with ESMTPS id 18B2628B95 for ; Mon, 4 Jun 2018 10:10:52 +0000 (UTC) Received: from localhost ([::1]:38692 helo=lists.gnu.org) by lists.gnu.org with esmtp (Exim 4.71) (envelope-from ) id 1fPmRv-0002qD-Bq for patchwork-qemu-devel@patchwork.kernel.org; Mon, 04 Jun 2018 06:10:51 -0400 Received: from eggs.gnu.org ([2001:4830:134:3::10]:46647) by lists.gnu.org with esmtp (Exim 4.71) (envelope-from ) id 1fPmEI-0007wh-SE for qemu-devel@nongnu.org; Mon, 04 Jun 2018 05:56:48 -0400 Received: from Debian-exim by eggs.gnu.org with spam-scanned (Exim 4.71) (envelope-from ) id 1fPmEF-0001N6-4z for qemu-devel@nongnu.org; Mon, 04 Jun 2018 05:56:46 -0400 Received: from mail-pg0-x232.google.com ([2607:f8b0:400e:c05::232]:37681) by eggs.gnu.org with esmtps (TLS1.0:RSA_AES_128_CBC_SHA1:16) (Exim 4.71) (envelope-from ) id 1fPmEE-0001Md-Rd for qemu-devel@nongnu.org; Mon, 04 Jun 2018 05:56:43 -0400 Received: by mail-pg0-x232.google.com with SMTP id a13-v6so14249069pgu.4 for ; Mon, 04 Jun 2018 02:56:42 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20161025; h=from:to:cc:subject:date:message-id:in-reply-to:references; bh=iZgOZ1AuEkabspkyAoobPcWdqMQd7mqyqIifUfGJFmo=; b=uBN+SjiVq9Wbpf6YzbsvDI4MjNA6b9hB8y8nH1lSXVWkfQQtqURZVKANsvrndYS9BO +NBGeFV1f9JNzemCE6cLMQflENJSp8i1NMN39jGZmYyDoteEK8F2/kYPWsBt44T4XtVa DbAphUwYL/vRw/8qafgSjVh5luTuHTxNKeSyie3c9YVAMbFvQqJn5CUNym6iYfm8iVfd ShhbWjGvJBmTt8r2MvspKoKStM6bhYahg/zPhLpRTqjZ40NkDKcTkwQEXmhOdTYVSUnR dn1XAEMAPT1N8Ba3f0hynEKXbsTQzLC5s7FBU8gB6E2DHcbwZDZbRqLghUKKczw9KkO4 f4cg== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20161025; h=x-gm-message-state:from:to:cc:subject:date:message-id:in-reply-to :references; bh=iZgOZ1AuEkabspkyAoobPcWdqMQd7mqyqIifUfGJFmo=; b=YL4FWtj98Eh6XGl82wJcTXieWQvbZRLUEP8lKIb7XUdLlRtExP4LWy4tAWDYui8CW/ WcsUK2eN29dVYXhfoUzbGodC4VoiwTy7VWvced2XtoA06Qj/apyzlF3/JyB4+CGQMk15 JnqIzV+2KjFsLpXje0Nz52tid5m7AzJdOb3GKx60hEwrdHOQJKrP052u7dX26swZ++a+ t4RqVMKYNKvTCKCavLRmaqEGwTRfNAEHLJdh17SWy1faEPrynY6w6VO2mXo5V6G4x2ZT egc7PvmZ8kDBfA9jUbO4LM/vx2bSqpuOX+uysrPwuvG3CUkMg/PZ3L+frLH4bPtok0DM amjA== X-Gm-Message-State: ALKqPwfZwMEdytdGSPUv7j4zwLDEEGE+MCKLBRWU1RPjoQNqanf4TWLl mwg2a/Q9ybt4rZmzkxeq9a4= X-Google-Smtp-Source: ADUXVKK4KgJP/9QYe9pJewDxmx5SUMiZaO/lQldN4FM0Vne5VFzrMfplyzLAXxnFkL56bgl6o0V32g== X-Received: by 2002:a65:438a:: with SMTP id m10-v6mr17054564pgp.315.1528106201916; Mon, 04 Jun 2018 02:56:41 -0700 (PDT) Received: from localhost.localdomain ([203.205.141.35]) by smtp.gmail.com with ESMTPSA id h130-v6sm124502105pfc.98.2018.06.04.02.56.38 (version=TLS1_2 cipher=ECDHE-RSA-CHACHA20-POLY1305 bits=256/256); Mon, 04 Jun 2018 02:56:41 -0700 (PDT) From: guangrong.xiao@gmail.com X-Google-Original-From: xiaoguangrong@tencent.com To: pbonzini@redhat.com, mst@redhat.com, mtosatti@redhat.com Date: Mon, 4 Jun 2018 17:55:20 +0800 Message-Id: <20180604095520.8563-13-xiaoguangrong@tencent.com> X-Mailer: git-send-email 2.14.4 In-Reply-To: <20180604095520.8563-1-xiaoguangrong@tencent.com> References: <20180604095520.8563-1-xiaoguangrong@tencent.com> X-detected-operating-system: by eggs.gnu.org: Genre and OS details not recognized. X-Received-From: 2607:f8b0:400e:c05::232 Subject: [Qemu-devel] [PATCH 12/12] migration: use lockless Multithread model for decompression X-BeenThere: qemu-devel@nongnu.org X-Mailman-Version: 2.1.21 Precedence: list List-Id: List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Cc: kvm@vger.kernel.org, Xiao Guangrong , qemu-devel@nongnu.org, peterx@redhat.com, dgilbert@redhat.com, wei.w.wang@intel.com, jiang.biao2@zte.com.cn Errors-To: qemu-devel-bounces+patchwork-qemu-devel=patchwork.kernel.org@nongnu.org Sender: "Qemu-devel" X-Virus-Scanned: ClamAV using ClamSMTP From: Xiao Guangrong Adapt the compression code to the lockless multithread model Signed-off-by: Xiao Guangrong --- migration/ram.c | 381 ++++++++++++++++++++++++++------------------------------ 1 file changed, 175 insertions(+), 206 deletions(-) diff --git a/migration/ram.c b/migration/ram.c index 58ecf5caa0..0a0ef0ee57 100644 --- a/migration/ram.c +++ b/migration/ram.c @@ -341,25 +341,9 @@ typedef struct PageSearchStatus PageSearchStatus; CompressionStats compression_counters; -struct DecompressParam { - bool done; - bool quit; - QemuMutex mutex; - QemuCond cond; - void *des; - uint8_t *compbuf; - int len; - z_stream stream; -}; -typedef struct DecompressParam DecompressParam; - static const QEMUFileOps empty_ops = { }; static QEMUFile *decomp_file; -static DecompressParam *decomp_param; -static QemuThread *decompress_threads; -static QemuMutex decomp_done_lock; -static QemuCond decomp_done_cond; /* Multiple fd's */ @@ -962,6 +946,178 @@ static int compress_threads_save_setup(void) return compress_threads ? 0 : -1; } +/* return the size after decompression, or negative value on error */ +static int +qemu_uncompress_data(z_stream *stream, uint8_t *dest, size_t dest_len, + const uint8_t *source, size_t source_len) +{ + int err; + + err = inflateReset(stream); + if (err != Z_OK) { + return -1; + } + + stream->avail_in = source_len; + stream->next_in = (uint8_t *)source; + stream->avail_out = dest_len; + stream->next_out = dest; + + err = inflate(stream, Z_NO_FLUSH); + if (err != Z_STREAM_END) { + return -1; + } + + return stream->total_out; +} + +struct DecompressData { + /* filled by migration thread.*/ + void *des; + uint8_t *compbuf; + size_t len; + + z_stream stream; + ThreadRequest request; +}; +typedef struct DecompressData DecompressData; + +static ThreadRequest *decompress_thread_data_init(void) +{ + DecompressData *dd = g_new0(DecompressData, 1); + + if (inflateInit(&dd->stream) != Z_OK) { + g_free(dd); + return NULL; + } + + dd->compbuf = g_malloc0(compressBound(TARGET_PAGE_SIZE)); + return &dd->request; +} + +static void decompress_thread_data_fini(ThreadRequest *request) +{ + DecompressData *dd = container_of(request, DecompressData, request); + + inflateEnd(&dd->stream); + g_free(dd->compbuf); + g_free(dd); +} + +static void decompress_thread_data_handler(ThreadRequest *request) +{ + DecompressData *dd = container_of(request, DecompressData, request); + unsigned long pagesize = TARGET_PAGE_SIZE; + int ret; + + ret = qemu_uncompress_data(&dd->stream, dd->des, pagesize, + dd->compbuf, dd->len); + if (ret < 0) { + error_report("decompress data failed"); + qemu_file_set_error(decomp_file, ret); + } +} + +static void decompress_thread_data_done(ThreadRequest *data) +{ +} + +struct CompressLoad { + Threads *decompress_threads; + + /* + * used to decompress data in migration thread if + * decompress threads are busy. + */ + z_stream stream; + uint8_t *compbuf; +}; +typedef struct CompressLoad CompressLoad; + +static CompressLoad compress_load; + +static int decompress_init(QEMUFile *f) +{ + Threads *threads; + + threads = threads_create(migrate_decompress_threads(), "decompress", + decompress_thread_data_init, + decompress_thread_data_fini, + decompress_thread_data_handler, + decompress_thread_data_done); + if (!threads) { + return -1; + } + + if (inflateInit(&compress_load.stream) != Z_OK) { + threads_destroy(threads); + return -1; + } + + compress_load.decompress_threads = threads; + compress_load.compbuf = g_malloc0(compressBound(TARGET_PAGE_SIZE)); + decomp_file = f; + return 0; +} + +static void decompress_fini(void) +{ + if (!compress_load.compbuf) { + return; + } + + threads_destroy(compress_load.decompress_threads); + compress_load.decompress_threads = NULL; + g_free(compress_load.compbuf); + compress_load.compbuf = NULL; + inflateEnd(&compress_load.stream); + decomp_file = NULL; +} + +static int flush_decompressed_data(void) +{ + if (!migrate_use_compression()) { + return 0; + } + + threads_wait_done(compress_load.decompress_threads); + return qemu_file_get_error(decomp_file); +} + +static void decompress_data_with_multi_threads(QEMUFile *f, + void *host, size_t len) +{ + ThreadRequest *request; + Threads *threads = compress_load.decompress_threads; + unsigned long pagesize = TARGET_PAGE_SIZE; + uint8_t *compbuf = compress_load.compbuf; + int ret; + + request = threads_submit_request_prepare(threads); + if (request) { + DecompressData *dd; + + dd = container_of(request, DecompressData, request); + dd->des = host; + dd->len = len; + qemu_get_buffer(f, dd->compbuf, len); + threads_submit_request_commit(threads, request); + return; + } + + /* load data and decompress in the main thread */ + + /* it can change compbuf to point to an internal buffer */ + qemu_get_buffer_in_place(f, &compbuf, len); + + ret = qemu_uncompress_data(&compress_load.stream, host, pagesize, + compbuf, len); + if (ret < 0) { + error_report("decompress data failed"); + qemu_file_set_error(decomp_file, ret); + } +} + /** * xbzrle_cache_zero_page: insert a zero page in the XBZRLE cache * @@ -2794,193 +2950,6 @@ void ram_handle_compressed(void *host, uint8_t ch, uint64_t size) } } -/* return the size after decompression, or negative value on error */ -static int -qemu_uncompress_data(z_stream *stream, uint8_t *dest, size_t dest_len, - const uint8_t *source, size_t source_len) -{ - int err; - - err = inflateReset(stream); - if (err != Z_OK) { - return -1; - } - - stream->avail_in = source_len; - stream->next_in = (uint8_t *)source; - stream->avail_out = dest_len; - stream->next_out = dest; - - err = inflate(stream, Z_NO_FLUSH); - if (err != Z_STREAM_END) { - return -1; - } - - return stream->total_out; -} - -static void *do_data_decompress(void *opaque) -{ - DecompressParam *param = opaque; - unsigned long pagesize; - uint8_t *des; - int len, ret; - - qemu_mutex_lock(¶m->mutex); - while (!param->quit) { - if (param->des) { - des = param->des; - len = param->len; - param->des = 0; - qemu_mutex_unlock(¶m->mutex); - - pagesize = TARGET_PAGE_SIZE; - - ret = qemu_uncompress_data(¶m->stream, des, pagesize, - param->compbuf, len); - if (ret < 0) { - error_report("decompress data failed"); - qemu_file_set_error(decomp_file, ret); - } - - qemu_mutex_lock(&decomp_done_lock); - param->done = true; - qemu_cond_signal(&decomp_done_cond); - qemu_mutex_unlock(&decomp_done_lock); - - qemu_mutex_lock(¶m->mutex); - } else { - qemu_cond_wait(¶m->cond, ¶m->mutex); - } - } - qemu_mutex_unlock(¶m->mutex); - - return NULL; -} - -static int wait_for_decompress_done(void) -{ - int idx, thread_count; - - if (!migrate_use_compression()) { - return 0; - } - - thread_count = migrate_decompress_threads(); - qemu_mutex_lock(&decomp_done_lock); - for (idx = 0; idx < thread_count; idx++) { - while (!decomp_param[idx].done) { - qemu_cond_wait(&decomp_done_cond, &decomp_done_lock); - } - } - qemu_mutex_unlock(&decomp_done_lock); - return qemu_file_get_error(decomp_file); -} - -static void compress_threads_load_cleanup(void) -{ - int i, thread_count; - - if (!migrate_use_compression()) { - return; - } - thread_count = migrate_decompress_threads(); - for (i = 0; i < thread_count; i++) { - /* - * we use it as a indicator which shows if the thread is - * properly init'd or not - */ - if (!decomp_param[i].compbuf) { - break; - } - - qemu_mutex_lock(&decomp_param[i].mutex); - decomp_param[i].quit = true; - qemu_cond_signal(&decomp_param[i].cond); - qemu_mutex_unlock(&decomp_param[i].mutex); - } - for (i = 0; i < thread_count; i++) { - if (!decomp_param[i].compbuf) { - break; - } - - qemu_thread_join(decompress_threads + i); - qemu_mutex_destroy(&decomp_param[i].mutex); - qemu_cond_destroy(&decomp_param[i].cond); - inflateEnd(&decomp_param[i].stream); - g_free(decomp_param[i].compbuf); - decomp_param[i].compbuf = NULL; - } - g_free(decompress_threads); - g_free(decomp_param); - decompress_threads = NULL; - decomp_param = NULL; - decomp_file = NULL; -} - -static int compress_threads_load_setup(QEMUFile *f) -{ - int i, thread_count; - - if (!migrate_use_compression()) { - return 0; - } - - thread_count = migrate_decompress_threads(); - decompress_threads = g_new0(QemuThread, thread_count); - decomp_param = g_new0(DecompressParam, thread_count); - qemu_mutex_init(&decomp_done_lock); - qemu_cond_init(&decomp_done_cond); - decomp_file = f; - for (i = 0; i < thread_count; i++) { - if (inflateInit(&decomp_param[i].stream) != Z_OK) { - goto exit; - } - - decomp_param[i].compbuf = g_malloc0(compressBound(TARGET_PAGE_SIZE)); - qemu_mutex_init(&decomp_param[i].mutex); - qemu_cond_init(&decomp_param[i].cond); - decomp_param[i].done = true; - decomp_param[i].quit = false; - qemu_thread_create(decompress_threads + i, "decompress", - do_data_decompress, decomp_param + i, - QEMU_THREAD_JOINABLE); - } - return 0; -exit: - compress_threads_load_cleanup(); - return -1; -} - -static void decompress_data_with_multi_threads(QEMUFile *f, - void *host, int len) -{ - int idx, thread_count; - - thread_count = migrate_decompress_threads(); - qemu_mutex_lock(&decomp_done_lock); - while (true) { - for (idx = 0; idx < thread_count; idx++) { - if (decomp_param[idx].done) { - decomp_param[idx].done = false; - qemu_mutex_lock(&decomp_param[idx].mutex); - qemu_get_buffer(f, decomp_param[idx].compbuf, len); - decomp_param[idx].des = host; - decomp_param[idx].len = len; - qemu_cond_signal(&decomp_param[idx].cond); - qemu_mutex_unlock(&decomp_param[idx].mutex); - break; - } - } - if (idx < thread_count) { - break; - } else { - qemu_cond_wait(&decomp_done_cond, &decomp_done_lock); - } - } - qemu_mutex_unlock(&decomp_done_lock); -} - /** * ram_load_setup: Setup RAM for migration incoming side * @@ -2991,7 +2960,7 @@ static void decompress_data_with_multi_threads(QEMUFile *f, */ static int ram_load_setup(QEMUFile *f, void *opaque) { - if (compress_threads_load_setup(f)) { + if (decompress_init(f)) { return -1; } @@ -3004,7 +2973,7 @@ static int ram_load_cleanup(void *opaque) { RAMBlock *rb; xbzrle_load_cleanup(); - compress_threads_load_cleanup(); + decompress_fini(); RAMBLOCK_FOREACH(rb) { g_free(rb->receivedmap); @@ -3346,7 +3315,7 @@ static int ram_load(QEMUFile *f, void *opaque, int version_id) } } - ret |= wait_for_decompress_done(); + ret |= flush_decompressed_data(); rcu_read_unlock(); trace_ram_load_complete(ret, seq_iter); return ret;