Statistics
| Branch: | Revision:

root / dump.c @ feature-archipelago

History | View | Annotate | Download (51.8 kB)

1 783e9b48 Wen Congyang
/*
2 783e9b48 Wen Congyang
 * QEMU dump
3 783e9b48 Wen Congyang
 *
4 783e9b48 Wen Congyang
 * Copyright Fujitsu, Corp. 2011, 2012
5 783e9b48 Wen Congyang
 *
6 783e9b48 Wen Congyang
 * Authors:
7 783e9b48 Wen Congyang
 *     Wen Congyang <wency@cn.fujitsu.com>
8 783e9b48 Wen Congyang
 *
9 352666e2 Stefan Weil
 * This work is licensed under the terms of the GNU GPL, version 2 or later.
10 352666e2 Stefan Weil
 * See the COPYING file in the top-level directory.
11 783e9b48 Wen Congyang
 *
12 783e9b48 Wen Congyang
 */
13 783e9b48 Wen Congyang
14 783e9b48 Wen Congyang
#include "qemu-common.h"
15 783e9b48 Wen Congyang
#include "elf.h"
16 783e9b48 Wen Congyang
#include "cpu.h"
17 022c62cb Paolo Bonzini
#include "exec/cpu-all.h"
18 022c62cb Paolo Bonzini
#include "exec/hwaddr.h"
19 83c9089e Paolo Bonzini
#include "monitor/monitor.h"
20 9c17d615 Paolo Bonzini
#include "sysemu/kvm.h"
21 9c17d615 Paolo Bonzini
#include "sysemu/dump.h"
22 9c17d615 Paolo Bonzini
#include "sysemu/sysemu.h"
23 9c17d615 Paolo Bonzini
#include "sysemu/memory_mapping.h"
24 1b3509ca Andreas Färber
#include "sysemu/cpus.h"
25 7b1b5d19 Paolo Bonzini
#include "qapi/error.h"
26 783e9b48 Wen Congyang
#include "qmp-commands.h"
27 783e9b48 Wen Congyang
28 d12f57ec qiaonuohan
#include <zlib.h>
29 d12f57ec qiaonuohan
#ifdef CONFIG_LZO
30 d12f57ec qiaonuohan
#include <lzo/lzo1x.h>
31 d12f57ec qiaonuohan
#endif
32 d12f57ec qiaonuohan
#ifdef CONFIG_SNAPPY
33 d12f57ec qiaonuohan
#include <snappy-c.h>
34 d12f57ec qiaonuohan
#endif
35 4ab23a91 qiaonuohan
#ifndef ELF_MACHINE_UNAME
36 4ab23a91 qiaonuohan
#define ELF_MACHINE_UNAME "Unknown"
37 4ab23a91 qiaonuohan
#endif
38 d12f57ec qiaonuohan
39 783e9b48 Wen Congyang
static uint16_t cpu_convert_to_target16(uint16_t val, int endian)
40 783e9b48 Wen Congyang
{
41 783e9b48 Wen Congyang
    if (endian == ELFDATA2LSB) {
42 783e9b48 Wen Congyang
        val = cpu_to_le16(val);
43 783e9b48 Wen Congyang
    } else {
44 783e9b48 Wen Congyang
        val = cpu_to_be16(val);
45 783e9b48 Wen Congyang
    }
46 783e9b48 Wen Congyang
47 783e9b48 Wen Congyang
    return val;
48 783e9b48 Wen Congyang
}
49 783e9b48 Wen Congyang
50 783e9b48 Wen Congyang
static uint32_t cpu_convert_to_target32(uint32_t val, int endian)
51 783e9b48 Wen Congyang
{
52 783e9b48 Wen Congyang
    if (endian == ELFDATA2LSB) {
53 783e9b48 Wen Congyang
        val = cpu_to_le32(val);
54 783e9b48 Wen Congyang
    } else {
55 783e9b48 Wen Congyang
        val = cpu_to_be32(val);
56 783e9b48 Wen Congyang
    }
57 783e9b48 Wen Congyang
58 783e9b48 Wen Congyang
    return val;
59 783e9b48 Wen Congyang
}
60 783e9b48 Wen Congyang
61 783e9b48 Wen Congyang
static uint64_t cpu_convert_to_target64(uint64_t val, int endian)
62 783e9b48 Wen Congyang
{
63 783e9b48 Wen Congyang
    if (endian == ELFDATA2LSB) {
64 783e9b48 Wen Congyang
        val = cpu_to_le64(val);
65 783e9b48 Wen Congyang
    } else {
66 783e9b48 Wen Congyang
        val = cpu_to_be64(val);
67 783e9b48 Wen Congyang
    }
68 783e9b48 Wen Congyang
69 783e9b48 Wen Congyang
    return val;
70 783e9b48 Wen Congyang
}
71 783e9b48 Wen Congyang
72 783e9b48 Wen Congyang
typedef struct DumpState {
73 5ee163e8 Laszlo Ersek
    GuestPhysBlockList guest_phys_blocks;
74 783e9b48 Wen Congyang
    ArchDumpInfo dump_info;
75 783e9b48 Wen Congyang
    MemoryMappingList list;
76 783e9b48 Wen Congyang
    uint16_t phdr_num;
77 783e9b48 Wen Congyang
    uint32_t sh_info;
78 783e9b48 Wen Congyang
    bool have_section;
79 783e9b48 Wen Congyang
    bool resume;
80 bb6b6843 Aneesh Kumar K.V
    ssize_t note_size;
81 a8170e5e Avi Kivity
    hwaddr memory_offset;
82 783e9b48 Wen Congyang
    int fd;
83 783e9b48 Wen Congyang
84 56c4bfb3 Laszlo Ersek
    GuestPhysBlock *next_block;
85 783e9b48 Wen Congyang
    ram_addr_t start;
86 783e9b48 Wen Congyang
    bool has_filter;
87 783e9b48 Wen Congyang
    int64_t begin;
88 783e9b48 Wen Congyang
    int64_t length;
89 783e9b48 Wen Congyang
    Error **errp;
90 4835ef77 qiaonuohan
91 4835ef77 qiaonuohan
    uint8_t *note_buf;          /* buffer for notes */
92 4835ef77 qiaonuohan
    size_t note_buf_offset;     /* the writing place in note_buf */
93 7aad248d qiaonuohan
    uint32_t nr_cpus;           /* number of guest's cpu */
94 7aad248d qiaonuohan
    size_t page_size;           /* guest's page size */
95 7aad248d qiaonuohan
    uint32_t page_shift;        /* guest's page shift */
96 7aad248d qiaonuohan
    uint64_t max_mapnr;         /* the biggest guest's phys-mem's number */
97 7aad248d qiaonuohan
    size_t len_dump_bitmap;     /* the size of the place used to store
98 7aad248d qiaonuohan
                                   dump_bitmap in vmcore */
99 7aad248d qiaonuohan
    off_t offset_dump_bitmap;   /* offset of dump_bitmap part in vmcore */
100 7aad248d qiaonuohan
    off_t offset_page;          /* offset of page part in vmcore */
101 7aad248d qiaonuohan
    size_t num_dumpable;        /* number of page that can be dumped */
102 7aad248d qiaonuohan
    uint32_t flag_compress;     /* indicate the compression format */
103 783e9b48 Wen Congyang
} DumpState;
104 783e9b48 Wen Congyang
105 783e9b48 Wen Congyang
static int dump_cleanup(DumpState *s)
106 783e9b48 Wen Congyang
{
107 783e9b48 Wen Congyang
    int ret = 0;
108 783e9b48 Wen Congyang
109 5ee163e8 Laszlo Ersek
    guest_phys_blocks_free(&s->guest_phys_blocks);
110 783e9b48 Wen Congyang
    memory_mapping_list_free(&s->list);
111 783e9b48 Wen Congyang
    if (s->fd != -1) {
112 783e9b48 Wen Congyang
        close(s->fd);
113 783e9b48 Wen Congyang
    }
114 783e9b48 Wen Congyang
    if (s->resume) {
115 783e9b48 Wen Congyang
        vm_start();
116 783e9b48 Wen Congyang
    }
117 783e9b48 Wen Congyang
118 783e9b48 Wen Congyang
    return ret;
119 783e9b48 Wen Congyang
}
120 783e9b48 Wen Congyang
121 783e9b48 Wen Congyang
static void dump_error(DumpState *s, const char *reason)
122 783e9b48 Wen Congyang
{
123 783e9b48 Wen Congyang
    dump_cleanup(s);
124 783e9b48 Wen Congyang
}
125 783e9b48 Wen Congyang
126 b5ba1cc6 qiaonuohan
static int fd_write_vmcore(const void *buf, size_t size, void *opaque)
127 783e9b48 Wen Congyang
{
128 783e9b48 Wen Congyang
    DumpState *s = opaque;
129 2f61652d Luiz Capitulino
    size_t written_size;
130 2f61652d Luiz Capitulino
131 2f61652d Luiz Capitulino
    written_size = qemu_write_full(s->fd, buf, size);
132 2f61652d Luiz Capitulino
    if (written_size != size) {
133 2f61652d Luiz Capitulino
        return -1;
134 783e9b48 Wen Congyang
    }
135 783e9b48 Wen Congyang
136 783e9b48 Wen Congyang
    return 0;
137 783e9b48 Wen Congyang
}
138 783e9b48 Wen Congyang
139 783e9b48 Wen Congyang
static int write_elf64_header(DumpState *s)
140 783e9b48 Wen Congyang
{
141 783e9b48 Wen Congyang
    Elf64_Ehdr elf_header;
142 783e9b48 Wen Congyang
    int ret;
143 783e9b48 Wen Congyang
    int endian = s->dump_info.d_endian;
144 783e9b48 Wen Congyang
145 783e9b48 Wen Congyang
    memset(&elf_header, 0, sizeof(Elf64_Ehdr));
146 783e9b48 Wen Congyang
    memcpy(&elf_header, ELFMAG, SELFMAG);
147 783e9b48 Wen Congyang
    elf_header.e_ident[EI_CLASS] = ELFCLASS64;
148 783e9b48 Wen Congyang
    elf_header.e_ident[EI_DATA] = s->dump_info.d_endian;
149 783e9b48 Wen Congyang
    elf_header.e_ident[EI_VERSION] = EV_CURRENT;
150 783e9b48 Wen Congyang
    elf_header.e_type = cpu_convert_to_target16(ET_CORE, endian);
151 783e9b48 Wen Congyang
    elf_header.e_machine = cpu_convert_to_target16(s->dump_info.d_machine,
152 783e9b48 Wen Congyang
                                                   endian);
153 783e9b48 Wen Congyang
    elf_header.e_version = cpu_convert_to_target32(EV_CURRENT, endian);
154 783e9b48 Wen Congyang
    elf_header.e_ehsize = cpu_convert_to_target16(sizeof(elf_header), endian);
155 783e9b48 Wen Congyang
    elf_header.e_phoff = cpu_convert_to_target64(sizeof(Elf64_Ehdr), endian);
156 783e9b48 Wen Congyang
    elf_header.e_phentsize = cpu_convert_to_target16(sizeof(Elf64_Phdr),
157 783e9b48 Wen Congyang
                                                     endian);
158 783e9b48 Wen Congyang
    elf_header.e_phnum = cpu_convert_to_target16(s->phdr_num, endian);
159 783e9b48 Wen Congyang
    if (s->have_section) {
160 783e9b48 Wen Congyang
        uint64_t shoff = sizeof(Elf64_Ehdr) + sizeof(Elf64_Phdr) * s->sh_info;
161 783e9b48 Wen Congyang
162 783e9b48 Wen Congyang
        elf_header.e_shoff = cpu_convert_to_target64(shoff, endian);
163 783e9b48 Wen Congyang
        elf_header.e_shentsize = cpu_convert_to_target16(sizeof(Elf64_Shdr),
164 783e9b48 Wen Congyang
                                                         endian);
165 783e9b48 Wen Congyang
        elf_header.e_shnum = cpu_convert_to_target16(1, endian);
166 783e9b48 Wen Congyang
    }
167 783e9b48 Wen Congyang
168 783e9b48 Wen Congyang
    ret = fd_write_vmcore(&elf_header, sizeof(elf_header), s);
169 783e9b48 Wen Congyang
    if (ret < 0) {
170 783e9b48 Wen Congyang
        dump_error(s, "dump: failed to write elf header.\n");
171 783e9b48 Wen Congyang
        return -1;
172 783e9b48 Wen Congyang
    }
173 783e9b48 Wen Congyang
174 783e9b48 Wen Congyang
    return 0;
175 783e9b48 Wen Congyang
}
176 783e9b48 Wen Congyang
177 783e9b48 Wen Congyang
static int write_elf32_header(DumpState *s)
178 783e9b48 Wen Congyang
{
179 783e9b48 Wen Congyang
    Elf32_Ehdr elf_header;
180 783e9b48 Wen Congyang
    int ret;
181 783e9b48 Wen Congyang
    int endian = s->dump_info.d_endian;
182 783e9b48 Wen Congyang
183 783e9b48 Wen Congyang
    memset(&elf_header, 0, sizeof(Elf32_Ehdr));
184 783e9b48 Wen Congyang
    memcpy(&elf_header, ELFMAG, SELFMAG);
185 783e9b48 Wen Congyang
    elf_header.e_ident[EI_CLASS] = ELFCLASS32;
186 783e9b48 Wen Congyang
    elf_header.e_ident[EI_DATA] = endian;
187 783e9b48 Wen Congyang
    elf_header.e_ident[EI_VERSION] = EV_CURRENT;
188 783e9b48 Wen Congyang
    elf_header.e_type = cpu_convert_to_target16(ET_CORE, endian);
189 783e9b48 Wen Congyang
    elf_header.e_machine = cpu_convert_to_target16(s->dump_info.d_machine,
190 783e9b48 Wen Congyang
                                                   endian);
191 783e9b48 Wen Congyang
    elf_header.e_version = cpu_convert_to_target32(EV_CURRENT, endian);
192 783e9b48 Wen Congyang
    elf_header.e_ehsize = cpu_convert_to_target16(sizeof(elf_header), endian);
193 783e9b48 Wen Congyang
    elf_header.e_phoff = cpu_convert_to_target32(sizeof(Elf32_Ehdr), endian);
194 783e9b48 Wen Congyang
    elf_header.e_phentsize = cpu_convert_to_target16(sizeof(Elf32_Phdr),
195 783e9b48 Wen Congyang
                                                     endian);
196 783e9b48 Wen Congyang
    elf_header.e_phnum = cpu_convert_to_target16(s->phdr_num, endian);
197 783e9b48 Wen Congyang
    if (s->have_section) {
198 783e9b48 Wen Congyang
        uint32_t shoff = sizeof(Elf32_Ehdr) + sizeof(Elf32_Phdr) * s->sh_info;
199 783e9b48 Wen Congyang
200 783e9b48 Wen Congyang
        elf_header.e_shoff = cpu_convert_to_target32(shoff, endian);
201 783e9b48 Wen Congyang
        elf_header.e_shentsize = cpu_convert_to_target16(sizeof(Elf32_Shdr),
202 783e9b48 Wen Congyang
                                                         endian);
203 783e9b48 Wen Congyang
        elf_header.e_shnum = cpu_convert_to_target16(1, endian);
204 783e9b48 Wen Congyang
    }
205 783e9b48 Wen Congyang
206 783e9b48 Wen Congyang
    ret = fd_write_vmcore(&elf_header, sizeof(elf_header), s);
207 783e9b48 Wen Congyang
    if (ret < 0) {
208 783e9b48 Wen Congyang
        dump_error(s, "dump: failed to write elf header.\n");
209 783e9b48 Wen Congyang
        return -1;
210 783e9b48 Wen Congyang
    }
211 783e9b48 Wen Congyang
212 783e9b48 Wen Congyang
    return 0;
213 783e9b48 Wen Congyang
}
214 783e9b48 Wen Congyang
215 783e9b48 Wen Congyang
static int write_elf64_load(DumpState *s, MemoryMapping *memory_mapping,
216 2cac2607 Laszlo Ersek
                            int phdr_index, hwaddr offset,
217 2cac2607 Laszlo Ersek
                            hwaddr filesz)
218 783e9b48 Wen Congyang
{
219 783e9b48 Wen Congyang
    Elf64_Phdr phdr;
220 783e9b48 Wen Congyang
    int ret;
221 783e9b48 Wen Congyang
    int endian = s->dump_info.d_endian;
222 783e9b48 Wen Congyang
223 783e9b48 Wen Congyang
    memset(&phdr, 0, sizeof(Elf64_Phdr));
224 783e9b48 Wen Congyang
    phdr.p_type = cpu_convert_to_target32(PT_LOAD, endian);
225 783e9b48 Wen Congyang
    phdr.p_offset = cpu_convert_to_target64(offset, endian);
226 783e9b48 Wen Congyang
    phdr.p_paddr = cpu_convert_to_target64(memory_mapping->phys_addr, endian);
227 2cac2607 Laszlo Ersek
    phdr.p_filesz = cpu_convert_to_target64(filesz, endian);
228 783e9b48 Wen Congyang
    phdr.p_memsz = cpu_convert_to_target64(memory_mapping->length, endian);
229 783e9b48 Wen Congyang
    phdr.p_vaddr = cpu_convert_to_target64(memory_mapping->virt_addr, endian);
230 783e9b48 Wen Congyang
231 2cac2607 Laszlo Ersek
    assert(memory_mapping->length >= filesz);
232 2cac2607 Laszlo Ersek
233 783e9b48 Wen Congyang
    ret = fd_write_vmcore(&phdr, sizeof(Elf64_Phdr), s);
234 783e9b48 Wen Congyang
    if (ret < 0) {
235 783e9b48 Wen Congyang
        dump_error(s, "dump: failed to write program header table.\n");
236 783e9b48 Wen Congyang
        return -1;
237 783e9b48 Wen Congyang
    }
238 783e9b48 Wen Congyang
239 783e9b48 Wen Congyang
    return 0;
240 783e9b48 Wen Congyang
}
241 783e9b48 Wen Congyang
242 783e9b48 Wen Congyang
static int write_elf32_load(DumpState *s, MemoryMapping *memory_mapping,
243 2cac2607 Laszlo Ersek
                            int phdr_index, hwaddr offset,
244 2cac2607 Laszlo Ersek
                            hwaddr filesz)
245 783e9b48 Wen Congyang
{
246 783e9b48 Wen Congyang
    Elf32_Phdr phdr;
247 783e9b48 Wen Congyang
    int ret;
248 783e9b48 Wen Congyang
    int endian = s->dump_info.d_endian;
249 783e9b48 Wen Congyang
250 783e9b48 Wen Congyang
    memset(&phdr, 0, sizeof(Elf32_Phdr));
251 783e9b48 Wen Congyang
    phdr.p_type = cpu_convert_to_target32(PT_LOAD, endian);
252 783e9b48 Wen Congyang
    phdr.p_offset = cpu_convert_to_target32(offset, endian);
253 783e9b48 Wen Congyang
    phdr.p_paddr = cpu_convert_to_target32(memory_mapping->phys_addr, endian);
254 2cac2607 Laszlo Ersek
    phdr.p_filesz = cpu_convert_to_target32(filesz, endian);
255 783e9b48 Wen Congyang
    phdr.p_memsz = cpu_convert_to_target32(memory_mapping->length, endian);
256 783e9b48 Wen Congyang
    phdr.p_vaddr = cpu_convert_to_target32(memory_mapping->virt_addr, endian);
257 783e9b48 Wen Congyang
258 2cac2607 Laszlo Ersek
    assert(memory_mapping->length >= filesz);
259 2cac2607 Laszlo Ersek
260 783e9b48 Wen Congyang
    ret = fd_write_vmcore(&phdr, sizeof(Elf32_Phdr), s);
261 783e9b48 Wen Congyang
    if (ret < 0) {
262 783e9b48 Wen Congyang
        dump_error(s, "dump: failed to write program header table.\n");
263 783e9b48 Wen Congyang
        return -1;
264 783e9b48 Wen Congyang
    }
265 783e9b48 Wen Congyang
266 783e9b48 Wen Congyang
    return 0;
267 783e9b48 Wen Congyang
}
268 783e9b48 Wen Congyang
269 783e9b48 Wen Congyang
static int write_elf64_note(DumpState *s)
270 783e9b48 Wen Congyang
{
271 783e9b48 Wen Congyang
    Elf64_Phdr phdr;
272 783e9b48 Wen Congyang
    int endian = s->dump_info.d_endian;
273 a8170e5e Avi Kivity
    hwaddr begin = s->memory_offset - s->note_size;
274 783e9b48 Wen Congyang
    int ret;
275 783e9b48 Wen Congyang
276 783e9b48 Wen Congyang
    memset(&phdr, 0, sizeof(Elf64_Phdr));
277 783e9b48 Wen Congyang
    phdr.p_type = cpu_convert_to_target32(PT_NOTE, endian);
278 783e9b48 Wen Congyang
    phdr.p_offset = cpu_convert_to_target64(begin, endian);
279 783e9b48 Wen Congyang
    phdr.p_paddr = 0;
280 783e9b48 Wen Congyang
    phdr.p_filesz = cpu_convert_to_target64(s->note_size, endian);
281 783e9b48 Wen Congyang
    phdr.p_memsz = cpu_convert_to_target64(s->note_size, endian);
282 783e9b48 Wen Congyang
    phdr.p_vaddr = 0;
283 783e9b48 Wen Congyang
284 783e9b48 Wen Congyang
    ret = fd_write_vmcore(&phdr, sizeof(Elf64_Phdr), s);
285 783e9b48 Wen Congyang
    if (ret < 0) {
286 783e9b48 Wen Congyang
        dump_error(s, "dump: failed to write program header table.\n");
287 783e9b48 Wen Congyang
        return -1;
288 783e9b48 Wen Congyang
    }
289 783e9b48 Wen Congyang
290 783e9b48 Wen Congyang
    return 0;
291 783e9b48 Wen Congyang
}
292 783e9b48 Wen Congyang
293 0bc3cd62 Paolo Bonzini
static inline int cpu_index(CPUState *cpu)
294 0bc3cd62 Paolo Bonzini
{
295 0bc3cd62 Paolo Bonzini
    return cpu->cpu_index + 1;
296 0bc3cd62 Paolo Bonzini
}
297 0bc3cd62 Paolo Bonzini
298 6a519918 qiaonuohan
static int write_elf64_notes(WriteCoreDumpFunction f, DumpState *s)
299 783e9b48 Wen Congyang
{
300 0d34282f Andreas Färber
    CPUState *cpu;
301 783e9b48 Wen Congyang
    int ret;
302 783e9b48 Wen Congyang
    int id;
303 783e9b48 Wen Congyang
304 bdc44640 Andreas Färber
    CPU_FOREACH(cpu) {
305 0d34282f Andreas Färber
        id = cpu_index(cpu);
306 6a519918 qiaonuohan
        ret = cpu_write_elf64_note(f, cpu, id, s);
307 783e9b48 Wen Congyang
        if (ret < 0) {
308 783e9b48 Wen Congyang
            dump_error(s, "dump: failed to write elf notes.\n");
309 783e9b48 Wen Congyang
            return -1;
310 783e9b48 Wen Congyang
        }
311 783e9b48 Wen Congyang
    }
312 783e9b48 Wen Congyang
313 bdc44640 Andreas Färber
    CPU_FOREACH(cpu) {
314 6a519918 qiaonuohan
        ret = cpu_write_elf64_qemunote(f, cpu, s);
315 783e9b48 Wen Congyang
        if (ret < 0) {
316 783e9b48 Wen Congyang
            dump_error(s, "dump: failed to write CPU status.\n");
317 783e9b48 Wen Congyang
            return -1;
318 783e9b48 Wen Congyang
        }
319 783e9b48 Wen Congyang
    }
320 783e9b48 Wen Congyang
321 783e9b48 Wen Congyang
    return 0;
322 783e9b48 Wen Congyang
}
323 783e9b48 Wen Congyang
324 783e9b48 Wen Congyang
static int write_elf32_note(DumpState *s)
325 783e9b48 Wen Congyang
{
326 a8170e5e Avi Kivity
    hwaddr begin = s->memory_offset - s->note_size;
327 783e9b48 Wen Congyang
    Elf32_Phdr phdr;
328 783e9b48 Wen Congyang
    int endian = s->dump_info.d_endian;
329 783e9b48 Wen Congyang
    int ret;
330 783e9b48 Wen Congyang
331 783e9b48 Wen Congyang
    memset(&phdr, 0, sizeof(Elf32_Phdr));
332 783e9b48 Wen Congyang
    phdr.p_type = cpu_convert_to_target32(PT_NOTE, endian);
333 783e9b48 Wen Congyang
    phdr.p_offset = cpu_convert_to_target32(begin, endian);
334 783e9b48 Wen Congyang
    phdr.p_paddr = 0;
335 783e9b48 Wen Congyang
    phdr.p_filesz = cpu_convert_to_target32(s->note_size, endian);
336 783e9b48 Wen Congyang
    phdr.p_memsz = cpu_convert_to_target32(s->note_size, endian);
337 783e9b48 Wen Congyang
    phdr.p_vaddr = 0;
338 783e9b48 Wen Congyang
339 783e9b48 Wen Congyang
    ret = fd_write_vmcore(&phdr, sizeof(Elf32_Phdr), s);
340 783e9b48 Wen Congyang
    if (ret < 0) {
341 783e9b48 Wen Congyang
        dump_error(s, "dump: failed to write program header table.\n");
342 783e9b48 Wen Congyang
        return -1;
343 783e9b48 Wen Congyang
    }
344 783e9b48 Wen Congyang
345 783e9b48 Wen Congyang
    return 0;
346 783e9b48 Wen Congyang
}
347 783e9b48 Wen Congyang
348 6a519918 qiaonuohan
static int write_elf32_notes(WriteCoreDumpFunction f, DumpState *s)
349 783e9b48 Wen Congyang
{
350 0d34282f Andreas Färber
    CPUState *cpu;
351 783e9b48 Wen Congyang
    int ret;
352 783e9b48 Wen Congyang
    int id;
353 783e9b48 Wen Congyang
354 bdc44640 Andreas Färber
    CPU_FOREACH(cpu) {
355 0d34282f Andreas Färber
        id = cpu_index(cpu);
356 6a519918 qiaonuohan
        ret = cpu_write_elf32_note(f, cpu, id, s);
357 783e9b48 Wen Congyang
        if (ret < 0) {
358 783e9b48 Wen Congyang
            dump_error(s, "dump: failed to write elf notes.\n");
359 783e9b48 Wen Congyang
            return -1;
360 783e9b48 Wen Congyang
        }
361 783e9b48 Wen Congyang
    }
362 783e9b48 Wen Congyang
363 bdc44640 Andreas Färber
    CPU_FOREACH(cpu) {
364 6a519918 qiaonuohan
        ret = cpu_write_elf32_qemunote(f, cpu, s);
365 783e9b48 Wen Congyang
        if (ret < 0) {
366 783e9b48 Wen Congyang
            dump_error(s, "dump: failed to write CPU status.\n");
367 783e9b48 Wen Congyang
            return -1;
368 783e9b48 Wen Congyang
        }
369 783e9b48 Wen Congyang
    }
370 783e9b48 Wen Congyang
371 783e9b48 Wen Congyang
    return 0;
372 783e9b48 Wen Congyang
}
373 783e9b48 Wen Congyang
374 783e9b48 Wen Congyang
static int write_elf_section(DumpState *s, int type)
375 783e9b48 Wen Congyang
{
376 783e9b48 Wen Congyang
    Elf32_Shdr shdr32;
377 783e9b48 Wen Congyang
    Elf64_Shdr shdr64;
378 783e9b48 Wen Congyang
    int endian = s->dump_info.d_endian;
379 783e9b48 Wen Congyang
    int shdr_size;
380 783e9b48 Wen Congyang
    void *shdr;
381 783e9b48 Wen Congyang
    int ret;
382 783e9b48 Wen Congyang
383 783e9b48 Wen Congyang
    if (type == 0) {
384 783e9b48 Wen Congyang
        shdr_size = sizeof(Elf32_Shdr);
385 783e9b48 Wen Congyang
        memset(&shdr32, 0, shdr_size);
386 783e9b48 Wen Congyang
        shdr32.sh_info = cpu_convert_to_target32(s->sh_info, endian);
387 783e9b48 Wen Congyang
        shdr = &shdr32;
388 783e9b48 Wen Congyang
    } else {
389 783e9b48 Wen Congyang
        shdr_size = sizeof(Elf64_Shdr);
390 783e9b48 Wen Congyang
        memset(&shdr64, 0, shdr_size);
391 783e9b48 Wen Congyang
        shdr64.sh_info = cpu_convert_to_target32(s->sh_info, endian);
392 783e9b48 Wen Congyang
        shdr = &shdr64;
393 783e9b48 Wen Congyang
    }
394 783e9b48 Wen Congyang
395 783e9b48 Wen Congyang
    ret = fd_write_vmcore(&shdr, shdr_size, s);
396 783e9b48 Wen Congyang
    if (ret < 0) {
397 783e9b48 Wen Congyang
        dump_error(s, "dump: failed to write section header table.\n");
398 783e9b48 Wen Congyang
        return -1;
399 783e9b48 Wen Congyang
    }
400 783e9b48 Wen Congyang
401 783e9b48 Wen Congyang
    return 0;
402 783e9b48 Wen Congyang
}
403 783e9b48 Wen Congyang
404 783e9b48 Wen Congyang
static int write_data(DumpState *s, void *buf, int length)
405 783e9b48 Wen Congyang
{
406 783e9b48 Wen Congyang
    int ret;
407 783e9b48 Wen Congyang
408 783e9b48 Wen Congyang
    ret = fd_write_vmcore(buf, length, s);
409 783e9b48 Wen Congyang
    if (ret < 0) {
410 783e9b48 Wen Congyang
        dump_error(s, "dump: failed to save memory.\n");
411 783e9b48 Wen Congyang
        return -1;
412 783e9b48 Wen Congyang
    }
413 783e9b48 Wen Congyang
414 783e9b48 Wen Congyang
    return 0;
415 783e9b48 Wen Congyang
}
416 783e9b48 Wen Congyang
417 783e9b48 Wen Congyang
/* write the memroy to vmcore. 1 page per I/O. */
418 56c4bfb3 Laszlo Ersek
static int write_memory(DumpState *s, GuestPhysBlock *block, ram_addr_t start,
419 783e9b48 Wen Congyang
                        int64_t size)
420 783e9b48 Wen Congyang
{
421 783e9b48 Wen Congyang
    int64_t i;
422 783e9b48 Wen Congyang
    int ret;
423 783e9b48 Wen Congyang
424 783e9b48 Wen Congyang
    for (i = 0; i < size / TARGET_PAGE_SIZE; i++) {
425 56c4bfb3 Laszlo Ersek
        ret = write_data(s, block->host_addr + start + i * TARGET_PAGE_SIZE,
426 783e9b48 Wen Congyang
                         TARGET_PAGE_SIZE);
427 783e9b48 Wen Congyang
        if (ret < 0) {
428 783e9b48 Wen Congyang
            return ret;
429 783e9b48 Wen Congyang
        }
430 783e9b48 Wen Congyang
    }
431 783e9b48 Wen Congyang
432 783e9b48 Wen Congyang
    if ((size % TARGET_PAGE_SIZE) != 0) {
433 56c4bfb3 Laszlo Ersek
        ret = write_data(s, block->host_addr + start + i * TARGET_PAGE_SIZE,
434 783e9b48 Wen Congyang
                         size % TARGET_PAGE_SIZE);
435 783e9b48 Wen Congyang
        if (ret < 0) {
436 783e9b48 Wen Congyang
            return ret;
437 783e9b48 Wen Congyang
        }
438 783e9b48 Wen Congyang
    }
439 783e9b48 Wen Congyang
440 783e9b48 Wen Congyang
    return 0;
441 783e9b48 Wen Congyang
}
442 783e9b48 Wen Congyang
443 2cac2607 Laszlo Ersek
/* get the memory's offset and size in the vmcore */
444 2cac2607 Laszlo Ersek
static void get_offset_range(hwaddr phys_addr,
445 2cac2607 Laszlo Ersek
                             ram_addr_t mapping_length,
446 2cac2607 Laszlo Ersek
                             DumpState *s,
447 2cac2607 Laszlo Ersek
                             hwaddr *p_offset,
448 2cac2607 Laszlo Ersek
                             hwaddr *p_filesz)
449 783e9b48 Wen Congyang
{
450 56c4bfb3 Laszlo Ersek
    GuestPhysBlock *block;
451 a8170e5e Avi Kivity
    hwaddr offset = s->memory_offset;
452 783e9b48 Wen Congyang
    int64_t size_in_block, start;
453 783e9b48 Wen Congyang
454 2cac2607 Laszlo Ersek
    /* When the memory is not stored into vmcore, offset will be -1 */
455 2cac2607 Laszlo Ersek
    *p_offset = -1;
456 2cac2607 Laszlo Ersek
    *p_filesz = 0;
457 2cac2607 Laszlo Ersek
458 783e9b48 Wen Congyang
    if (s->has_filter) {
459 783e9b48 Wen Congyang
        if (phys_addr < s->begin || phys_addr >= s->begin + s->length) {
460 2cac2607 Laszlo Ersek
            return;
461 783e9b48 Wen Congyang
        }
462 783e9b48 Wen Congyang
    }
463 783e9b48 Wen Congyang
464 56c4bfb3 Laszlo Ersek
    QTAILQ_FOREACH(block, &s->guest_phys_blocks.head, next) {
465 783e9b48 Wen Congyang
        if (s->has_filter) {
466 56c4bfb3 Laszlo Ersek
            if (block->target_start >= s->begin + s->length ||
467 56c4bfb3 Laszlo Ersek
                block->target_end <= s->begin) {
468 783e9b48 Wen Congyang
                /* This block is out of the range */
469 783e9b48 Wen Congyang
                continue;
470 783e9b48 Wen Congyang
            }
471 783e9b48 Wen Congyang
472 56c4bfb3 Laszlo Ersek
            if (s->begin <= block->target_start) {
473 56c4bfb3 Laszlo Ersek
                start = block->target_start;
474 783e9b48 Wen Congyang
            } else {
475 783e9b48 Wen Congyang
                start = s->begin;
476 783e9b48 Wen Congyang
            }
477 783e9b48 Wen Congyang
478 56c4bfb3 Laszlo Ersek
            size_in_block = block->target_end - start;
479 56c4bfb3 Laszlo Ersek
            if (s->begin + s->length < block->target_end) {
480 56c4bfb3 Laszlo Ersek
                size_in_block -= block->target_end - (s->begin + s->length);
481 783e9b48 Wen Congyang
            }
482 783e9b48 Wen Congyang
        } else {
483 56c4bfb3 Laszlo Ersek
            start = block->target_start;
484 56c4bfb3 Laszlo Ersek
            size_in_block = block->target_end - block->target_start;
485 783e9b48 Wen Congyang
        }
486 783e9b48 Wen Congyang
487 783e9b48 Wen Congyang
        if (phys_addr >= start && phys_addr < start + size_in_block) {
488 2cac2607 Laszlo Ersek
            *p_offset = phys_addr - start + offset;
489 2cac2607 Laszlo Ersek
490 2cac2607 Laszlo Ersek
            /* The offset range mapped from the vmcore file must not spill over
491 56c4bfb3 Laszlo Ersek
             * the GuestPhysBlock, clamp it. The rest of the mapping will be
492 2cac2607 Laszlo Ersek
             * zero-filled in memory at load time; see
493 2cac2607 Laszlo Ersek
             * <http://refspecs.linuxbase.org/elf/gabi4+/ch5.pheader.html>.
494 2cac2607 Laszlo Ersek
             */
495 2cac2607 Laszlo Ersek
            *p_filesz = phys_addr + mapping_length <= start + size_in_block ?
496 2cac2607 Laszlo Ersek
                        mapping_length :
497 2cac2607 Laszlo Ersek
                        size_in_block - (phys_addr - start);
498 2cac2607 Laszlo Ersek
            return;
499 783e9b48 Wen Congyang
        }
500 783e9b48 Wen Congyang
501 783e9b48 Wen Congyang
        offset += size_in_block;
502 783e9b48 Wen Congyang
    }
503 783e9b48 Wen Congyang
}
504 783e9b48 Wen Congyang
505 783e9b48 Wen Congyang
static int write_elf_loads(DumpState *s)
506 783e9b48 Wen Congyang
{
507 2cac2607 Laszlo Ersek
    hwaddr offset, filesz;
508 783e9b48 Wen Congyang
    MemoryMapping *memory_mapping;
509 783e9b48 Wen Congyang
    uint32_t phdr_index = 1;
510 783e9b48 Wen Congyang
    int ret;
511 783e9b48 Wen Congyang
    uint32_t max_index;
512 783e9b48 Wen Congyang
513 783e9b48 Wen Congyang
    if (s->have_section) {
514 783e9b48 Wen Congyang
        max_index = s->sh_info;
515 783e9b48 Wen Congyang
    } else {
516 783e9b48 Wen Congyang
        max_index = s->phdr_num;
517 783e9b48 Wen Congyang
    }
518 783e9b48 Wen Congyang
519 783e9b48 Wen Congyang
    QTAILQ_FOREACH(memory_mapping, &s->list.head, next) {
520 2cac2607 Laszlo Ersek
        get_offset_range(memory_mapping->phys_addr,
521 2cac2607 Laszlo Ersek
                         memory_mapping->length,
522 2cac2607 Laszlo Ersek
                         s, &offset, &filesz);
523 783e9b48 Wen Congyang
        if (s->dump_info.d_class == ELFCLASS64) {
524 2cac2607 Laszlo Ersek
            ret = write_elf64_load(s, memory_mapping, phdr_index++, offset,
525 2cac2607 Laszlo Ersek
                                   filesz);
526 783e9b48 Wen Congyang
        } else {
527 2cac2607 Laszlo Ersek
            ret = write_elf32_load(s, memory_mapping, phdr_index++, offset,
528 2cac2607 Laszlo Ersek
                                   filesz);
529 783e9b48 Wen Congyang
        }
530 783e9b48 Wen Congyang
531 783e9b48 Wen Congyang
        if (ret < 0) {
532 783e9b48 Wen Congyang
            return -1;
533 783e9b48 Wen Congyang
        }
534 783e9b48 Wen Congyang
535 783e9b48 Wen Congyang
        if (phdr_index >= max_index) {
536 783e9b48 Wen Congyang
            break;
537 783e9b48 Wen Congyang
        }
538 783e9b48 Wen Congyang
    }
539 783e9b48 Wen Congyang
540 783e9b48 Wen Congyang
    return 0;
541 783e9b48 Wen Congyang
}
542 783e9b48 Wen Congyang
543 783e9b48 Wen Congyang
/* write elf header, PT_NOTE and elf note to vmcore. */
544 783e9b48 Wen Congyang
static int dump_begin(DumpState *s)
545 783e9b48 Wen Congyang
{
546 783e9b48 Wen Congyang
    int ret;
547 783e9b48 Wen Congyang
548 783e9b48 Wen Congyang
    /*
549 783e9b48 Wen Congyang
     * the vmcore's format is:
550 783e9b48 Wen Congyang
     *   --------------
551 783e9b48 Wen Congyang
     *   |  elf header |
552 783e9b48 Wen Congyang
     *   --------------
553 783e9b48 Wen Congyang
     *   |  PT_NOTE    |
554 783e9b48 Wen Congyang
     *   --------------
555 783e9b48 Wen Congyang
     *   |  PT_LOAD    |
556 783e9b48 Wen Congyang
     *   --------------
557 783e9b48 Wen Congyang
     *   |  ......     |
558 783e9b48 Wen Congyang
     *   --------------
559 783e9b48 Wen Congyang
     *   |  PT_LOAD    |
560 783e9b48 Wen Congyang
     *   --------------
561 783e9b48 Wen Congyang
     *   |  sec_hdr    |
562 783e9b48 Wen Congyang
     *   --------------
563 783e9b48 Wen Congyang
     *   |  elf note   |
564 783e9b48 Wen Congyang
     *   --------------
565 783e9b48 Wen Congyang
     *   |  memory     |
566 783e9b48 Wen Congyang
     *   --------------
567 783e9b48 Wen Congyang
     *
568 783e9b48 Wen Congyang
     * we only know where the memory is saved after we write elf note into
569 783e9b48 Wen Congyang
     * vmcore.
570 783e9b48 Wen Congyang
     */
571 783e9b48 Wen Congyang
572 783e9b48 Wen Congyang
    /* write elf header to vmcore */
573 783e9b48 Wen Congyang
    if (s->dump_info.d_class == ELFCLASS64) {
574 783e9b48 Wen Congyang
        ret = write_elf64_header(s);
575 783e9b48 Wen Congyang
    } else {
576 783e9b48 Wen Congyang
        ret = write_elf32_header(s);
577 783e9b48 Wen Congyang
    }
578 783e9b48 Wen Congyang
    if (ret < 0) {
579 783e9b48 Wen Congyang
        return -1;
580 783e9b48 Wen Congyang
    }
581 783e9b48 Wen Congyang
582 783e9b48 Wen Congyang
    if (s->dump_info.d_class == ELFCLASS64) {
583 783e9b48 Wen Congyang
        /* write PT_NOTE to vmcore */
584 783e9b48 Wen Congyang
        if (write_elf64_note(s) < 0) {
585 783e9b48 Wen Congyang
            return -1;
586 783e9b48 Wen Congyang
        }
587 783e9b48 Wen Congyang
588 783e9b48 Wen Congyang
        /* write all PT_LOAD to vmcore */
589 783e9b48 Wen Congyang
        if (write_elf_loads(s) < 0) {
590 783e9b48 Wen Congyang
            return -1;
591 783e9b48 Wen Congyang
        }
592 783e9b48 Wen Congyang
593 783e9b48 Wen Congyang
        /* write section to vmcore */
594 783e9b48 Wen Congyang
        if (s->have_section) {
595 783e9b48 Wen Congyang
            if (write_elf_section(s, 1) < 0) {
596 783e9b48 Wen Congyang
                return -1;
597 783e9b48 Wen Congyang
            }
598 783e9b48 Wen Congyang
        }
599 783e9b48 Wen Congyang
600 783e9b48 Wen Congyang
        /* write notes to vmcore */
601 6a519918 qiaonuohan
        if (write_elf64_notes(fd_write_vmcore, s) < 0) {
602 783e9b48 Wen Congyang
            return -1;
603 783e9b48 Wen Congyang
        }
604 783e9b48 Wen Congyang
605 783e9b48 Wen Congyang
    } else {
606 783e9b48 Wen Congyang
        /* write PT_NOTE to vmcore */
607 783e9b48 Wen Congyang
        if (write_elf32_note(s) < 0) {
608 783e9b48 Wen Congyang
            return -1;
609 783e9b48 Wen Congyang
        }
610 783e9b48 Wen Congyang
611 783e9b48 Wen Congyang
        /* write all PT_LOAD to vmcore */
612 783e9b48 Wen Congyang
        if (write_elf_loads(s) < 0) {
613 783e9b48 Wen Congyang
            return -1;
614 783e9b48 Wen Congyang
        }
615 783e9b48 Wen Congyang
616 783e9b48 Wen Congyang
        /* write section to vmcore */
617 783e9b48 Wen Congyang
        if (s->have_section) {
618 783e9b48 Wen Congyang
            if (write_elf_section(s, 0) < 0) {
619 783e9b48 Wen Congyang
                return -1;
620 783e9b48 Wen Congyang
            }
621 783e9b48 Wen Congyang
        }
622 783e9b48 Wen Congyang
623 783e9b48 Wen Congyang
        /* write notes to vmcore */
624 6a519918 qiaonuohan
        if (write_elf32_notes(fd_write_vmcore, s) < 0) {
625 783e9b48 Wen Congyang
            return -1;
626 783e9b48 Wen Congyang
        }
627 783e9b48 Wen Congyang
    }
628 783e9b48 Wen Congyang
629 783e9b48 Wen Congyang
    return 0;
630 783e9b48 Wen Congyang
}
631 783e9b48 Wen Congyang
632 783e9b48 Wen Congyang
/* write PT_LOAD to vmcore */
633 783e9b48 Wen Congyang
static int dump_completed(DumpState *s)
634 783e9b48 Wen Congyang
{
635 783e9b48 Wen Congyang
    dump_cleanup(s);
636 783e9b48 Wen Congyang
    return 0;
637 783e9b48 Wen Congyang
}
638 783e9b48 Wen Congyang
639 56c4bfb3 Laszlo Ersek
static int get_next_block(DumpState *s, GuestPhysBlock *block)
640 783e9b48 Wen Congyang
{
641 783e9b48 Wen Congyang
    while (1) {
642 a3161038 Paolo Bonzini
        block = QTAILQ_NEXT(block, next);
643 783e9b48 Wen Congyang
        if (!block) {
644 783e9b48 Wen Congyang
            /* no more block */
645 783e9b48 Wen Congyang
            return 1;
646 783e9b48 Wen Congyang
        }
647 783e9b48 Wen Congyang
648 783e9b48 Wen Congyang
        s->start = 0;
649 56c4bfb3 Laszlo Ersek
        s->next_block = block;
650 783e9b48 Wen Congyang
        if (s->has_filter) {
651 56c4bfb3 Laszlo Ersek
            if (block->target_start >= s->begin + s->length ||
652 56c4bfb3 Laszlo Ersek
                block->target_end <= s->begin) {
653 783e9b48 Wen Congyang
                /* This block is out of the range */
654 783e9b48 Wen Congyang
                continue;
655 783e9b48 Wen Congyang
            }
656 783e9b48 Wen Congyang
657 56c4bfb3 Laszlo Ersek
            if (s->begin > block->target_start) {
658 56c4bfb3 Laszlo Ersek
                s->start = s->begin - block->target_start;
659 783e9b48 Wen Congyang
            }
660 783e9b48 Wen Congyang
        }
661 783e9b48 Wen Congyang
662 783e9b48 Wen Congyang
        return 0;
663 783e9b48 Wen Congyang
    }
664 783e9b48 Wen Congyang
}
665 783e9b48 Wen Congyang
666 783e9b48 Wen Congyang
/* write all memory to vmcore */
667 783e9b48 Wen Congyang
static int dump_iterate(DumpState *s)
668 783e9b48 Wen Congyang
{
669 56c4bfb3 Laszlo Ersek
    GuestPhysBlock *block;
670 783e9b48 Wen Congyang
    int64_t size;
671 783e9b48 Wen Congyang
    int ret;
672 783e9b48 Wen Congyang
673 783e9b48 Wen Congyang
    while (1) {
674 56c4bfb3 Laszlo Ersek
        block = s->next_block;
675 783e9b48 Wen Congyang
676 56c4bfb3 Laszlo Ersek
        size = block->target_end - block->target_start;
677 783e9b48 Wen Congyang
        if (s->has_filter) {
678 783e9b48 Wen Congyang
            size -= s->start;
679 56c4bfb3 Laszlo Ersek
            if (s->begin + s->length < block->target_end) {
680 56c4bfb3 Laszlo Ersek
                size -= block->target_end - (s->begin + s->length);
681 783e9b48 Wen Congyang
            }
682 783e9b48 Wen Congyang
        }
683 783e9b48 Wen Congyang
        ret = write_memory(s, block, s->start, size);
684 783e9b48 Wen Congyang
        if (ret == -1) {
685 783e9b48 Wen Congyang
            return ret;
686 783e9b48 Wen Congyang
        }
687 783e9b48 Wen Congyang
688 783e9b48 Wen Congyang
        ret = get_next_block(s, block);
689 783e9b48 Wen Congyang
        if (ret == 1) {
690 783e9b48 Wen Congyang
            dump_completed(s);
691 783e9b48 Wen Congyang
            return 0;
692 783e9b48 Wen Congyang
        }
693 783e9b48 Wen Congyang
    }
694 783e9b48 Wen Congyang
}
695 783e9b48 Wen Congyang
696 783e9b48 Wen Congyang
static int create_vmcore(DumpState *s)
697 783e9b48 Wen Congyang
{
698 783e9b48 Wen Congyang
    int ret;
699 783e9b48 Wen Congyang
700 783e9b48 Wen Congyang
    ret = dump_begin(s);
701 783e9b48 Wen Congyang
    if (ret < 0) {
702 783e9b48 Wen Congyang
        return -1;
703 783e9b48 Wen Congyang
    }
704 783e9b48 Wen Congyang
705 783e9b48 Wen Congyang
    ret = dump_iterate(s);
706 783e9b48 Wen Congyang
    if (ret < 0) {
707 783e9b48 Wen Congyang
        return -1;
708 783e9b48 Wen Congyang
    }
709 783e9b48 Wen Congyang
710 783e9b48 Wen Congyang
    return 0;
711 783e9b48 Wen Congyang
}
712 783e9b48 Wen Congyang
713 fda05387 qiaonuohan
static int write_start_flat_header(int fd)
714 fda05387 qiaonuohan
{
715 fda05387 qiaonuohan
    uint8_t *buf;
716 fda05387 qiaonuohan
    MakedumpfileHeader mh;
717 fda05387 qiaonuohan
    int ret = 0;
718 fda05387 qiaonuohan
719 fda05387 qiaonuohan
    memset(&mh, 0, sizeof(mh));
720 fda05387 qiaonuohan
    strncpy(mh.signature, MAKEDUMPFILE_SIGNATURE,
721 fda05387 qiaonuohan
            strlen(MAKEDUMPFILE_SIGNATURE));
722 fda05387 qiaonuohan
723 fda05387 qiaonuohan
    mh.type = cpu_to_be64(TYPE_FLAT_HEADER);
724 fda05387 qiaonuohan
    mh.version = cpu_to_be64(VERSION_FLAT_HEADER);
725 fda05387 qiaonuohan
726 fda05387 qiaonuohan
    buf = g_malloc0(MAX_SIZE_MDF_HEADER);
727 fda05387 qiaonuohan
    memcpy(buf, &mh, sizeof(mh));
728 fda05387 qiaonuohan
729 fda05387 qiaonuohan
    size_t written_size;
730 fda05387 qiaonuohan
    written_size = qemu_write_full(fd, buf, MAX_SIZE_MDF_HEADER);
731 fda05387 qiaonuohan
    if (written_size != MAX_SIZE_MDF_HEADER) {
732 fda05387 qiaonuohan
        ret = -1;
733 fda05387 qiaonuohan
    }
734 fda05387 qiaonuohan
735 fda05387 qiaonuohan
    g_free(buf);
736 fda05387 qiaonuohan
    return ret;
737 fda05387 qiaonuohan
}
738 fda05387 qiaonuohan
739 fda05387 qiaonuohan
static int write_end_flat_header(int fd)
740 fda05387 qiaonuohan
{
741 fda05387 qiaonuohan
    MakedumpfileDataHeader mdh;
742 fda05387 qiaonuohan
743 fda05387 qiaonuohan
    mdh.offset = END_FLAG_FLAT_HEADER;
744 fda05387 qiaonuohan
    mdh.buf_size = END_FLAG_FLAT_HEADER;
745 fda05387 qiaonuohan
746 fda05387 qiaonuohan
    size_t written_size;
747 fda05387 qiaonuohan
    written_size = qemu_write_full(fd, &mdh, sizeof(mdh));
748 fda05387 qiaonuohan
    if (written_size != sizeof(mdh)) {
749 fda05387 qiaonuohan
        return -1;
750 fda05387 qiaonuohan
    }
751 fda05387 qiaonuohan
752 fda05387 qiaonuohan
    return 0;
753 fda05387 qiaonuohan
}
754 fda05387 qiaonuohan
755 5d31babe qiaonuohan
static int write_buffer(int fd, off_t offset, const void *buf, size_t size)
756 5d31babe qiaonuohan
{
757 5d31babe qiaonuohan
    size_t written_size;
758 5d31babe qiaonuohan
    MakedumpfileDataHeader mdh;
759 5d31babe qiaonuohan
760 5d31babe qiaonuohan
    mdh.offset = cpu_to_be64(offset);
761 5d31babe qiaonuohan
    mdh.buf_size = cpu_to_be64(size);
762 5d31babe qiaonuohan
763 5d31babe qiaonuohan
    written_size = qemu_write_full(fd, &mdh, sizeof(mdh));
764 5d31babe qiaonuohan
    if (written_size != sizeof(mdh)) {
765 5d31babe qiaonuohan
        return -1;
766 5d31babe qiaonuohan
    }
767 5d31babe qiaonuohan
768 5d31babe qiaonuohan
    written_size = qemu_write_full(fd, buf, size);
769 5d31babe qiaonuohan
    if (written_size != size) {
770 5d31babe qiaonuohan
        return -1;
771 5d31babe qiaonuohan
    }
772 5d31babe qiaonuohan
773 5d31babe qiaonuohan
    return 0;
774 5d31babe qiaonuohan
}
775 5d31babe qiaonuohan
776 4835ef77 qiaonuohan
static int buf_write_note(const void *buf, size_t size, void *opaque)
777 4835ef77 qiaonuohan
{
778 4835ef77 qiaonuohan
    DumpState *s = opaque;
779 4835ef77 qiaonuohan
780 4835ef77 qiaonuohan
    /* note_buf is not enough */
781 4835ef77 qiaonuohan
    if (s->note_buf_offset + size > s->note_size) {
782 4835ef77 qiaonuohan
        return -1;
783 4835ef77 qiaonuohan
    }
784 4835ef77 qiaonuohan
785 4835ef77 qiaonuohan
    memcpy(s->note_buf + s->note_buf_offset, buf, size);
786 4835ef77 qiaonuohan
787 4835ef77 qiaonuohan
    s->note_buf_offset += size;
788 4835ef77 qiaonuohan
789 4835ef77 qiaonuohan
    return 0;
790 4835ef77 qiaonuohan
}
791 4835ef77 qiaonuohan
792 298f1168 qiaonuohan
/* write common header, sub header and elf note to vmcore */
793 298f1168 qiaonuohan
static int create_header32(DumpState *s)
794 298f1168 qiaonuohan
{
795 298f1168 qiaonuohan
    int ret = 0;
796 298f1168 qiaonuohan
    DiskDumpHeader32 *dh = NULL;
797 298f1168 qiaonuohan
    KdumpSubHeader32 *kh = NULL;
798 298f1168 qiaonuohan
    size_t size;
799 298f1168 qiaonuohan
    int endian = s->dump_info.d_endian;
800 298f1168 qiaonuohan
    uint32_t block_size;
801 298f1168 qiaonuohan
    uint32_t sub_hdr_size;
802 298f1168 qiaonuohan
    uint32_t bitmap_blocks;
803 298f1168 qiaonuohan
    uint32_t status = 0;
804 298f1168 qiaonuohan
    uint64_t offset_note;
805 298f1168 qiaonuohan
806 298f1168 qiaonuohan
    /* write common header, the version of kdump-compressed format is 6th */
807 298f1168 qiaonuohan
    size = sizeof(DiskDumpHeader32);
808 298f1168 qiaonuohan
    dh = g_malloc0(size);
809 298f1168 qiaonuohan
810 298f1168 qiaonuohan
    strncpy(dh->signature, KDUMP_SIGNATURE, strlen(KDUMP_SIGNATURE));
811 298f1168 qiaonuohan
    dh->header_version = cpu_convert_to_target32(6, endian);
812 298f1168 qiaonuohan
    block_size = s->page_size;
813 298f1168 qiaonuohan
    dh->block_size = cpu_convert_to_target32(block_size, endian);
814 298f1168 qiaonuohan
    sub_hdr_size = sizeof(struct KdumpSubHeader32) + s->note_size;
815 298f1168 qiaonuohan
    sub_hdr_size = DIV_ROUND_UP(sub_hdr_size, block_size);
816 298f1168 qiaonuohan
    dh->sub_hdr_size = cpu_convert_to_target32(sub_hdr_size, endian);
817 298f1168 qiaonuohan
    /* dh->max_mapnr may be truncated, full 64bit is in kh.max_mapnr_64 */
818 298f1168 qiaonuohan
    dh->max_mapnr = cpu_convert_to_target32(MIN(s->max_mapnr, UINT_MAX),
819 298f1168 qiaonuohan
                                            endian);
820 298f1168 qiaonuohan
    dh->nr_cpus = cpu_convert_to_target32(s->nr_cpus, endian);
821 298f1168 qiaonuohan
    bitmap_blocks = DIV_ROUND_UP(s->len_dump_bitmap, block_size) * 2;
822 298f1168 qiaonuohan
    dh->bitmap_blocks = cpu_convert_to_target32(bitmap_blocks, endian);
823 4ab23a91 qiaonuohan
    strncpy(dh->utsname.machine, ELF_MACHINE_UNAME, sizeof(dh->utsname.machine));
824 298f1168 qiaonuohan
825 298f1168 qiaonuohan
    if (s->flag_compress & DUMP_DH_COMPRESSED_ZLIB) {
826 298f1168 qiaonuohan
        status |= DUMP_DH_COMPRESSED_ZLIB;
827 298f1168 qiaonuohan
    }
828 298f1168 qiaonuohan
#ifdef CONFIG_LZO
829 298f1168 qiaonuohan
    if (s->flag_compress & DUMP_DH_COMPRESSED_LZO) {
830 298f1168 qiaonuohan
        status |= DUMP_DH_COMPRESSED_LZO;
831 298f1168 qiaonuohan
    }
832 298f1168 qiaonuohan
#endif
833 298f1168 qiaonuohan
#ifdef CONFIG_SNAPPY
834 298f1168 qiaonuohan
    if (s->flag_compress & DUMP_DH_COMPRESSED_SNAPPY) {
835 298f1168 qiaonuohan
        status |= DUMP_DH_COMPRESSED_SNAPPY;
836 298f1168 qiaonuohan
    }
837 298f1168 qiaonuohan
#endif
838 298f1168 qiaonuohan
    dh->status = cpu_convert_to_target32(status, endian);
839 298f1168 qiaonuohan
840 298f1168 qiaonuohan
    if (write_buffer(s->fd, 0, dh, size) < 0) {
841 298f1168 qiaonuohan
        dump_error(s, "dump: failed to write disk dump header.\n");
842 298f1168 qiaonuohan
        ret = -1;
843 298f1168 qiaonuohan
        goto out;
844 298f1168 qiaonuohan
    }
845 298f1168 qiaonuohan
846 298f1168 qiaonuohan
    /* write sub header */
847 298f1168 qiaonuohan
    size = sizeof(KdumpSubHeader32);
848 298f1168 qiaonuohan
    kh = g_malloc0(size);
849 298f1168 qiaonuohan
850 298f1168 qiaonuohan
    /* 64bit max_mapnr_64 */
851 298f1168 qiaonuohan
    kh->max_mapnr_64 = cpu_convert_to_target64(s->max_mapnr, endian);
852 298f1168 qiaonuohan
    kh->phys_base = cpu_convert_to_target32(PHYS_BASE, endian);
853 298f1168 qiaonuohan
    kh->dump_level = cpu_convert_to_target32(DUMP_LEVEL, endian);
854 298f1168 qiaonuohan
855 298f1168 qiaonuohan
    offset_note = DISKDUMP_HEADER_BLOCKS * block_size + size;
856 298f1168 qiaonuohan
    kh->offset_note = cpu_convert_to_target64(offset_note, endian);
857 298f1168 qiaonuohan
    kh->note_size = cpu_convert_to_target32(s->note_size, endian);
858 298f1168 qiaonuohan
859 298f1168 qiaonuohan
    if (write_buffer(s->fd, DISKDUMP_HEADER_BLOCKS *
860 298f1168 qiaonuohan
                     block_size, kh, size) < 0) {
861 298f1168 qiaonuohan
        dump_error(s, "dump: failed to write kdump sub header.\n");
862 298f1168 qiaonuohan
        ret = -1;
863 298f1168 qiaonuohan
        goto out;
864 298f1168 qiaonuohan
    }
865 298f1168 qiaonuohan
866 298f1168 qiaonuohan
    /* write note */
867 298f1168 qiaonuohan
    s->note_buf = g_malloc0(s->note_size);
868 298f1168 qiaonuohan
    s->note_buf_offset = 0;
869 298f1168 qiaonuohan
870 298f1168 qiaonuohan
    /* use s->note_buf to store notes temporarily */
871 298f1168 qiaonuohan
    if (write_elf32_notes(buf_write_note, s) < 0) {
872 298f1168 qiaonuohan
        ret = -1;
873 298f1168 qiaonuohan
        goto out;
874 298f1168 qiaonuohan
    }
875 298f1168 qiaonuohan
876 298f1168 qiaonuohan
    if (write_buffer(s->fd, offset_note, s->note_buf,
877 298f1168 qiaonuohan
                     s->note_size) < 0) {
878 298f1168 qiaonuohan
        dump_error(s, "dump: failed to write notes");
879 298f1168 qiaonuohan
        ret = -1;
880 298f1168 qiaonuohan
        goto out;
881 298f1168 qiaonuohan
    }
882 298f1168 qiaonuohan
883 298f1168 qiaonuohan
    /* get offset of dump_bitmap */
884 298f1168 qiaonuohan
    s->offset_dump_bitmap = (DISKDUMP_HEADER_BLOCKS + sub_hdr_size) *
885 298f1168 qiaonuohan
                             block_size;
886 298f1168 qiaonuohan
887 298f1168 qiaonuohan
    /* get offset of page */
888 298f1168 qiaonuohan
    s->offset_page = (DISKDUMP_HEADER_BLOCKS + sub_hdr_size + bitmap_blocks) *
889 298f1168 qiaonuohan
                     block_size;
890 298f1168 qiaonuohan
891 298f1168 qiaonuohan
out:
892 298f1168 qiaonuohan
    g_free(dh);
893 298f1168 qiaonuohan
    g_free(kh);
894 298f1168 qiaonuohan
    g_free(s->note_buf);
895 298f1168 qiaonuohan
896 298f1168 qiaonuohan
    return ret;
897 298f1168 qiaonuohan
}
898 298f1168 qiaonuohan
899 298f1168 qiaonuohan
/* write common header, sub header and elf note to vmcore */
900 298f1168 qiaonuohan
static int create_header64(DumpState *s)
901 298f1168 qiaonuohan
{
902 298f1168 qiaonuohan
    int ret = 0;
903 298f1168 qiaonuohan
    DiskDumpHeader64 *dh = NULL;
904 298f1168 qiaonuohan
    KdumpSubHeader64 *kh = NULL;
905 298f1168 qiaonuohan
    size_t size;
906 298f1168 qiaonuohan
    int endian = s->dump_info.d_endian;
907 298f1168 qiaonuohan
    uint32_t block_size;
908 298f1168 qiaonuohan
    uint32_t sub_hdr_size;
909 298f1168 qiaonuohan
    uint32_t bitmap_blocks;
910 298f1168 qiaonuohan
    uint32_t status = 0;
911 298f1168 qiaonuohan
    uint64_t offset_note;
912 298f1168 qiaonuohan
913 298f1168 qiaonuohan
    /* write common header, the version of kdump-compressed format is 6th */
914 298f1168 qiaonuohan
    size = sizeof(DiskDumpHeader64);
915 298f1168 qiaonuohan
    dh = g_malloc0(size);
916 298f1168 qiaonuohan
917 298f1168 qiaonuohan
    strncpy(dh->signature, KDUMP_SIGNATURE, strlen(KDUMP_SIGNATURE));
918 298f1168 qiaonuohan
    dh->header_version = cpu_convert_to_target32(6, endian);
919 298f1168 qiaonuohan
    block_size = s->page_size;
920 298f1168 qiaonuohan
    dh->block_size = cpu_convert_to_target32(block_size, endian);
921 298f1168 qiaonuohan
    sub_hdr_size = sizeof(struct KdumpSubHeader64) + s->note_size;
922 298f1168 qiaonuohan
    sub_hdr_size = DIV_ROUND_UP(sub_hdr_size, block_size);
923 298f1168 qiaonuohan
    dh->sub_hdr_size = cpu_convert_to_target32(sub_hdr_size, endian);
924 298f1168 qiaonuohan
    /* dh->max_mapnr may be truncated, full 64bit is in kh.max_mapnr_64 */
925 298f1168 qiaonuohan
    dh->max_mapnr = cpu_convert_to_target32(MIN(s->max_mapnr, UINT_MAX),
926 298f1168 qiaonuohan
                                            endian);
927 298f1168 qiaonuohan
    dh->nr_cpus = cpu_convert_to_target32(s->nr_cpus, endian);
928 298f1168 qiaonuohan
    bitmap_blocks = DIV_ROUND_UP(s->len_dump_bitmap, block_size) * 2;
929 298f1168 qiaonuohan
    dh->bitmap_blocks = cpu_convert_to_target32(bitmap_blocks, endian);
930 4ab23a91 qiaonuohan
    strncpy(dh->utsname.machine, ELF_MACHINE_UNAME, sizeof(dh->utsname.machine));
931 298f1168 qiaonuohan
932 298f1168 qiaonuohan
    if (s->flag_compress & DUMP_DH_COMPRESSED_ZLIB) {
933 298f1168 qiaonuohan
        status |= DUMP_DH_COMPRESSED_ZLIB;
934 298f1168 qiaonuohan
    }
935 298f1168 qiaonuohan
#ifdef CONFIG_LZO
936 298f1168 qiaonuohan
    if (s->flag_compress & DUMP_DH_COMPRESSED_LZO) {
937 298f1168 qiaonuohan
        status |= DUMP_DH_COMPRESSED_LZO;
938 298f1168 qiaonuohan
    }
939 298f1168 qiaonuohan
#endif
940 298f1168 qiaonuohan
#ifdef CONFIG_SNAPPY
941 298f1168 qiaonuohan
    if (s->flag_compress & DUMP_DH_COMPRESSED_SNAPPY) {
942 298f1168 qiaonuohan
        status |= DUMP_DH_COMPRESSED_SNAPPY;
943 298f1168 qiaonuohan
    }
944 298f1168 qiaonuohan
#endif
945 298f1168 qiaonuohan
    dh->status = cpu_convert_to_target32(status, endian);
946 298f1168 qiaonuohan
947 298f1168 qiaonuohan
    if (write_buffer(s->fd, 0, dh, size) < 0) {
948 298f1168 qiaonuohan
        dump_error(s, "dump: failed to write disk dump header.\n");
949 298f1168 qiaonuohan
        ret = -1;
950 298f1168 qiaonuohan
        goto out;
951 298f1168 qiaonuohan
    }
952 298f1168 qiaonuohan
953 298f1168 qiaonuohan
    /* write sub header */
954 298f1168 qiaonuohan
    size = sizeof(KdumpSubHeader64);
955 298f1168 qiaonuohan
    kh = g_malloc0(size);
956 298f1168 qiaonuohan
957 298f1168 qiaonuohan
    /* 64bit max_mapnr_64 */
958 298f1168 qiaonuohan
    kh->max_mapnr_64 = cpu_convert_to_target64(s->max_mapnr, endian);
959 298f1168 qiaonuohan
    kh->phys_base = cpu_convert_to_target64(PHYS_BASE, endian);
960 298f1168 qiaonuohan
    kh->dump_level = cpu_convert_to_target32(DUMP_LEVEL, endian);
961 298f1168 qiaonuohan
962 298f1168 qiaonuohan
    offset_note = DISKDUMP_HEADER_BLOCKS * block_size + size;
963 298f1168 qiaonuohan
    kh->offset_note = cpu_convert_to_target64(offset_note, endian);
964 298f1168 qiaonuohan
    kh->note_size = cpu_convert_to_target64(s->note_size, endian);
965 298f1168 qiaonuohan
966 298f1168 qiaonuohan
    if (write_buffer(s->fd, DISKDUMP_HEADER_BLOCKS *
967 298f1168 qiaonuohan
                     block_size, kh, size) < 0) {
968 298f1168 qiaonuohan
        dump_error(s, "dump: failed to write kdump sub header.\n");
969 298f1168 qiaonuohan
        ret = -1;
970 298f1168 qiaonuohan
        goto out;
971 298f1168 qiaonuohan
    }
972 298f1168 qiaonuohan
973 298f1168 qiaonuohan
    /* write note */
974 298f1168 qiaonuohan
    s->note_buf = g_malloc0(s->note_size);
975 298f1168 qiaonuohan
    s->note_buf_offset = 0;
976 298f1168 qiaonuohan
977 298f1168 qiaonuohan
    /* use s->note_buf to store notes temporarily */
978 298f1168 qiaonuohan
    if (write_elf64_notes(buf_write_note, s) < 0) {
979 298f1168 qiaonuohan
        ret = -1;
980 298f1168 qiaonuohan
        goto out;
981 298f1168 qiaonuohan
    }
982 298f1168 qiaonuohan
983 298f1168 qiaonuohan
    if (write_buffer(s->fd, offset_note, s->note_buf,
984 298f1168 qiaonuohan
                     s->note_size) < 0) {
985 298f1168 qiaonuohan
        dump_error(s, "dump: failed to write notes");
986 298f1168 qiaonuohan
        ret = -1;
987 298f1168 qiaonuohan
        goto out;
988 298f1168 qiaonuohan
    }
989 298f1168 qiaonuohan
990 298f1168 qiaonuohan
    /* get offset of dump_bitmap */
991 298f1168 qiaonuohan
    s->offset_dump_bitmap = (DISKDUMP_HEADER_BLOCKS + sub_hdr_size) *
992 298f1168 qiaonuohan
                             block_size;
993 298f1168 qiaonuohan
994 298f1168 qiaonuohan
    /* get offset of page */
995 298f1168 qiaonuohan
    s->offset_page = (DISKDUMP_HEADER_BLOCKS + sub_hdr_size + bitmap_blocks) *
996 298f1168 qiaonuohan
                     block_size;
997 298f1168 qiaonuohan
998 298f1168 qiaonuohan
out:
999 298f1168 qiaonuohan
    g_free(dh);
1000 298f1168 qiaonuohan
    g_free(kh);
1001 298f1168 qiaonuohan
    g_free(s->note_buf);
1002 298f1168 qiaonuohan
1003 298f1168 qiaonuohan
    return ret;
1004 298f1168 qiaonuohan
}
1005 298f1168 qiaonuohan
1006 298f1168 qiaonuohan
static int write_dump_header(DumpState *s)
1007 298f1168 qiaonuohan
{
1008 298f1168 qiaonuohan
    if (s->dump_info.d_machine == EM_386) {
1009 298f1168 qiaonuohan
        return create_header32(s);
1010 298f1168 qiaonuohan
    } else {
1011 298f1168 qiaonuohan
        return create_header64(s);
1012 298f1168 qiaonuohan
    }
1013 298f1168 qiaonuohan
}
1014 298f1168 qiaonuohan
1015 d0686c72 qiaonuohan
/*
1016 d0686c72 qiaonuohan
 * set dump_bitmap sequencely. the bit before last_pfn is not allowed to be
1017 d0686c72 qiaonuohan
 * rewritten, so if need to set the first bit, set last_pfn and pfn to 0.
1018 d0686c72 qiaonuohan
 * set_dump_bitmap will always leave the recently set bit un-sync. And setting
1019 d0686c72 qiaonuohan
 * (last bit + sizeof(buf) * 8) to 0 will do flushing the content in buf into
1020 d0686c72 qiaonuohan
 * vmcore, ie. synchronizing un-sync bit into vmcore.
1021 d0686c72 qiaonuohan
 */
1022 d0686c72 qiaonuohan
static int set_dump_bitmap(uint64_t last_pfn, uint64_t pfn, bool value,
1023 d0686c72 qiaonuohan
                           uint8_t *buf, DumpState *s)
1024 d0686c72 qiaonuohan
{
1025 d0686c72 qiaonuohan
    off_t old_offset, new_offset;
1026 d0686c72 qiaonuohan
    off_t offset_bitmap1, offset_bitmap2;
1027 d0686c72 qiaonuohan
    uint32_t byte, bit;
1028 d0686c72 qiaonuohan
1029 d0686c72 qiaonuohan
    /* should not set the previous place */
1030 d0686c72 qiaonuohan
    assert(last_pfn <= pfn);
1031 d0686c72 qiaonuohan
1032 d0686c72 qiaonuohan
    /*
1033 d0686c72 qiaonuohan
     * if the bit needed to be set is not cached in buf, flush the data in buf
1034 d0686c72 qiaonuohan
     * to vmcore firstly.
1035 d0686c72 qiaonuohan
     * making new_offset be bigger than old_offset can also sync remained data
1036 d0686c72 qiaonuohan
     * into vmcore.
1037 d0686c72 qiaonuohan
     */
1038 d0686c72 qiaonuohan
    old_offset = BUFSIZE_BITMAP * (last_pfn / PFN_BUFBITMAP);
1039 d0686c72 qiaonuohan
    new_offset = BUFSIZE_BITMAP * (pfn / PFN_BUFBITMAP);
1040 d0686c72 qiaonuohan
1041 d0686c72 qiaonuohan
    while (old_offset < new_offset) {
1042 d0686c72 qiaonuohan
        /* calculate the offset and write dump_bitmap */
1043 d0686c72 qiaonuohan
        offset_bitmap1 = s->offset_dump_bitmap + old_offset;
1044 d0686c72 qiaonuohan
        if (write_buffer(s->fd, offset_bitmap1, buf,
1045 d0686c72 qiaonuohan
                         BUFSIZE_BITMAP) < 0) {
1046 d0686c72 qiaonuohan
            return -1;
1047 d0686c72 qiaonuohan
        }
1048 d0686c72 qiaonuohan
1049 d0686c72 qiaonuohan
        /* dump level 1 is chosen, so 1st and 2nd bitmap are same */
1050 d0686c72 qiaonuohan
        offset_bitmap2 = s->offset_dump_bitmap + s->len_dump_bitmap +
1051 d0686c72 qiaonuohan
                         old_offset;
1052 d0686c72 qiaonuohan
        if (write_buffer(s->fd, offset_bitmap2, buf,
1053 d0686c72 qiaonuohan
                         BUFSIZE_BITMAP) < 0) {
1054 d0686c72 qiaonuohan
            return -1;
1055 d0686c72 qiaonuohan
        }
1056 d0686c72 qiaonuohan
1057 d0686c72 qiaonuohan
        memset(buf, 0, BUFSIZE_BITMAP);
1058 d0686c72 qiaonuohan
        old_offset += BUFSIZE_BITMAP;
1059 d0686c72 qiaonuohan
    }
1060 d0686c72 qiaonuohan
1061 d0686c72 qiaonuohan
    /* get the exact place of the bit in the buf, and set it */
1062 d0686c72 qiaonuohan
    byte = (pfn % PFN_BUFBITMAP) / CHAR_BIT;
1063 d0686c72 qiaonuohan
    bit = (pfn % PFN_BUFBITMAP) % CHAR_BIT;
1064 d0686c72 qiaonuohan
    if (value) {
1065 d0686c72 qiaonuohan
        buf[byte] |= 1u << bit;
1066 d0686c72 qiaonuohan
    } else {
1067 d0686c72 qiaonuohan
        buf[byte] &= ~(1u << bit);
1068 d0686c72 qiaonuohan
    }
1069 d0686c72 qiaonuohan
1070 d0686c72 qiaonuohan
    return 0;
1071 d0686c72 qiaonuohan
}
1072 d0686c72 qiaonuohan
1073 d0686c72 qiaonuohan
/*
1074 d0686c72 qiaonuohan
 * exam every page and return the page frame number and the address of the page.
1075 d0686c72 qiaonuohan
 * bufptr can be NULL. note: the blocks here is supposed to reflect guest-phys
1076 d0686c72 qiaonuohan
 * blocks, so block->target_start and block->target_end should be interal
1077 d0686c72 qiaonuohan
 * multiples of the target page size.
1078 d0686c72 qiaonuohan
 */
1079 d0686c72 qiaonuohan
static bool get_next_page(GuestPhysBlock **blockptr, uint64_t *pfnptr,
1080 d0686c72 qiaonuohan
                          uint8_t **bufptr, DumpState *s)
1081 d0686c72 qiaonuohan
{
1082 d0686c72 qiaonuohan
    GuestPhysBlock *block = *blockptr;
1083 d0686c72 qiaonuohan
    hwaddr addr;
1084 d0686c72 qiaonuohan
    uint8_t *buf;
1085 d0686c72 qiaonuohan
1086 d0686c72 qiaonuohan
    /* block == NULL means the start of the iteration */
1087 d0686c72 qiaonuohan
    if (!block) {
1088 d0686c72 qiaonuohan
        block = QTAILQ_FIRST(&s->guest_phys_blocks.head);
1089 d0686c72 qiaonuohan
        *blockptr = block;
1090 d0686c72 qiaonuohan
        assert(block->target_start % s->page_size == 0);
1091 d0686c72 qiaonuohan
        assert(block->target_end % s->page_size == 0);
1092 d0686c72 qiaonuohan
        *pfnptr = paddr_to_pfn(block->target_start, s->page_shift);
1093 d0686c72 qiaonuohan
        if (bufptr) {
1094 d0686c72 qiaonuohan
            *bufptr = block->host_addr;
1095 d0686c72 qiaonuohan
        }
1096 d0686c72 qiaonuohan
        return true;
1097 d0686c72 qiaonuohan
    }
1098 d0686c72 qiaonuohan
1099 d0686c72 qiaonuohan
    *pfnptr = *pfnptr + 1;
1100 d0686c72 qiaonuohan
    addr = pfn_to_paddr(*pfnptr, s->page_shift);
1101 d0686c72 qiaonuohan
1102 d0686c72 qiaonuohan
    if ((addr >= block->target_start) &&
1103 d0686c72 qiaonuohan
        (addr + s->page_size <= block->target_end)) {
1104 d0686c72 qiaonuohan
        buf = block->host_addr + (addr - block->target_start);
1105 d0686c72 qiaonuohan
    } else {
1106 d0686c72 qiaonuohan
        /* the next page is in the next block */
1107 d0686c72 qiaonuohan
        block = QTAILQ_NEXT(block, next);
1108 d0686c72 qiaonuohan
        *blockptr = block;
1109 d0686c72 qiaonuohan
        if (!block) {
1110 d0686c72 qiaonuohan
            return false;
1111 d0686c72 qiaonuohan
        }
1112 d0686c72 qiaonuohan
        assert(block->target_start % s->page_size == 0);
1113 d0686c72 qiaonuohan
        assert(block->target_end % s->page_size == 0);
1114 d0686c72 qiaonuohan
        *pfnptr = paddr_to_pfn(block->target_start, s->page_shift);
1115 d0686c72 qiaonuohan
        buf = block->host_addr;
1116 d0686c72 qiaonuohan
    }
1117 d0686c72 qiaonuohan
1118 d0686c72 qiaonuohan
    if (bufptr) {
1119 d0686c72 qiaonuohan
        *bufptr = buf;
1120 d0686c72 qiaonuohan
    }
1121 d0686c72 qiaonuohan
1122 d0686c72 qiaonuohan
    return true;
1123 d0686c72 qiaonuohan
}
1124 d0686c72 qiaonuohan
1125 d0686c72 qiaonuohan
static int write_dump_bitmap(DumpState *s)
1126 d0686c72 qiaonuohan
{
1127 d0686c72 qiaonuohan
    int ret = 0;
1128 d0686c72 qiaonuohan
    uint64_t last_pfn, pfn;
1129 d0686c72 qiaonuohan
    void *dump_bitmap_buf;
1130 d0686c72 qiaonuohan
    size_t num_dumpable;
1131 d0686c72 qiaonuohan
    GuestPhysBlock *block_iter = NULL;
1132 d0686c72 qiaonuohan
1133 d0686c72 qiaonuohan
    /* dump_bitmap_buf is used to store dump_bitmap temporarily */
1134 d0686c72 qiaonuohan
    dump_bitmap_buf = g_malloc0(BUFSIZE_BITMAP);
1135 d0686c72 qiaonuohan
1136 d0686c72 qiaonuohan
    num_dumpable = 0;
1137 d0686c72 qiaonuohan
    last_pfn = 0;
1138 d0686c72 qiaonuohan
1139 d0686c72 qiaonuohan
    /*
1140 d0686c72 qiaonuohan
     * exam memory page by page, and set the bit in dump_bitmap corresponded
1141 d0686c72 qiaonuohan
     * to the existing page.
1142 d0686c72 qiaonuohan
     */
1143 d0686c72 qiaonuohan
    while (get_next_page(&block_iter, &pfn, NULL, s)) {
1144 d0686c72 qiaonuohan
        ret = set_dump_bitmap(last_pfn, pfn, true, dump_bitmap_buf, s);
1145 d0686c72 qiaonuohan
        if (ret < 0) {
1146 d0686c72 qiaonuohan
            dump_error(s, "dump: failed to set dump_bitmap.\n");
1147 d0686c72 qiaonuohan
            ret = -1;
1148 d0686c72 qiaonuohan
            goto out;
1149 d0686c72 qiaonuohan
        }
1150 d0686c72 qiaonuohan
1151 d0686c72 qiaonuohan
        last_pfn = pfn;
1152 d0686c72 qiaonuohan
        num_dumpable++;
1153 d0686c72 qiaonuohan
    }
1154 d0686c72 qiaonuohan
1155 d0686c72 qiaonuohan
    /*
1156 d0686c72 qiaonuohan
     * set_dump_bitmap will always leave the recently set bit un-sync. Here we
1157 d0686c72 qiaonuohan
     * set last_pfn + PFN_BUFBITMAP to 0 and those set but un-sync bit will be
1158 d0686c72 qiaonuohan
     * synchronized into vmcore.
1159 d0686c72 qiaonuohan
     */
1160 d0686c72 qiaonuohan
    if (num_dumpable > 0) {
1161 d0686c72 qiaonuohan
        ret = set_dump_bitmap(last_pfn, last_pfn + PFN_BUFBITMAP, false,
1162 d0686c72 qiaonuohan
                              dump_bitmap_buf, s);
1163 d0686c72 qiaonuohan
        if (ret < 0) {
1164 d0686c72 qiaonuohan
            dump_error(s, "dump: failed to sync dump_bitmap.\n");
1165 d0686c72 qiaonuohan
            ret = -1;
1166 d0686c72 qiaonuohan
            goto out;
1167 d0686c72 qiaonuohan
        }
1168 d0686c72 qiaonuohan
    }
1169 d0686c72 qiaonuohan
1170 d0686c72 qiaonuohan
    /* number of dumpable pages that will be dumped later */
1171 d0686c72 qiaonuohan
    s->num_dumpable = num_dumpable;
1172 d0686c72 qiaonuohan
1173 d0686c72 qiaonuohan
out:
1174 d0686c72 qiaonuohan
    g_free(dump_bitmap_buf);
1175 d0686c72 qiaonuohan
1176 d0686c72 qiaonuohan
    return ret;
1177 d0686c72 qiaonuohan
}
1178 d0686c72 qiaonuohan
1179 64cfba6a qiaonuohan
static void prepare_data_cache(DataCache *data_cache, DumpState *s,
1180 64cfba6a qiaonuohan
                               off_t offset)
1181 64cfba6a qiaonuohan
{
1182 64cfba6a qiaonuohan
    data_cache->fd = s->fd;
1183 64cfba6a qiaonuohan
    data_cache->data_size = 0;
1184 64cfba6a qiaonuohan
    data_cache->buf_size = BUFSIZE_DATA_CACHE;
1185 64cfba6a qiaonuohan
    data_cache->buf = g_malloc0(BUFSIZE_DATA_CACHE);
1186 64cfba6a qiaonuohan
    data_cache->offset = offset;
1187 64cfba6a qiaonuohan
}
1188 64cfba6a qiaonuohan
1189 64cfba6a qiaonuohan
static int write_cache(DataCache *dc, const void *buf, size_t size,
1190 64cfba6a qiaonuohan
                       bool flag_sync)
1191 64cfba6a qiaonuohan
{
1192 64cfba6a qiaonuohan
    /*
1193 64cfba6a qiaonuohan
     * dc->buf_size should not be less than size, otherwise dc will never be
1194 64cfba6a qiaonuohan
     * enough
1195 64cfba6a qiaonuohan
     */
1196 64cfba6a qiaonuohan
    assert(size <= dc->buf_size);
1197 64cfba6a qiaonuohan
1198 64cfba6a qiaonuohan
    /*
1199 64cfba6a qiaonuohan
     * if flag_sync is set, synchronize data in dc->buf into vmcore.
1200 64cfba6a qiaonuohan
     * otherwise check if the space is enough for caching data in buf, if not,
1201 64cfba6a qiaonuohan
     * write the data in dc->buf to dc->fd and reset dc->buf
1202 64cfba6a qiaonuohan
     */
1203 64cfba6a qiaonuohan
    if ((!flag_sync && dc->data_size + size > dc->buf_size) ||
1204 64cfba6a qiaonuohan
        (flag_sync && dc->data_size > 0)) {
1205 64cfba6a qiaonuohan
        if (write_buffer(dc->fd, dc->offset, dc->buf, dc->data_size) < 0) {
1206 64cfba6a qiaonuohan
            return -1;
1207 64cfba6a qiaonuohan
        }
1208 64cfba6a qiaonuohan
1209 64cfba6a qiaonuohan
        dc->offset += dc->data_size;
1210 64cfba6a qiaonuohan
        dc->data_size = 0;
1211 64cfba6a qiaonuohan
    }
1212 64cfba6a qiaonuohan
1213 64cfba6a qiaonuohan
    if (!flag_sync) {
1214 64cfba6a qiaonuohan
        memcpy(dc->buf + dc->data_size, buf, size);
1215 64cfba6a qiaonuohan
        dc->data_size += size;
1216 64cfba6a qiaonuohan
    }
1217 64cfba6a qiaonuohan
1218 64cfba6a qiaonuohan
    return 0;
1219 64cfba6a qiaonuohan
}
1220 64cfba6a qiaonuohan
1221 64cfba6a qiaonuohan
static void free_data_cache(DataCache *data_cache)
1222 64cfba6a qiaonuohan
{
1223 64cfba6a qiaonuohan
    g_free(data_cache->buf);
1224 64cfba6a qiaonuohan
}
1225 64cfba6a qiaonuohan
1226 d12f57ec qiaonuohan
static size_t get_len_buf_out(size_t page_size, uint32_t flag_compress)
1227 d12f57ec qiaonuohan
{
1228 d12f57ec qiaonuohan
    size_t len_buf_out_zlib, len_buf_out_lzo, len_buf_out_snappy;
1229 d12f57ec qiaonuohan
    size_t len_buf_out;
1230 d12f57ec qiaonuohan
1231 d12f57ec qiaonuohan
    /* init buf_out */
1232 d12f57ec qiaonuohan
    len_buf_out_zlib = len_buf_out_lzo = len_buf_out_snappy = 0;
1233 d12f57ec qiaonuohan
1234 d12f57ec qiaonuohan
    /* buf size for zlib */
1235 d12f57ec qiaonuohan
    len_buf_out_zlib = compressBound(page_size);
1236 d12f57ec qiaonuohan
1237 d12f57ec qiaonuohan
    /* buf size for lzo */
1238 d12f57ec qiaonuohan
#ifdef CONFIG_LZO
1239 d12f57ec qiaonuohan
    if (flag_compress & DUMP_DH_COMPRESSED_LZO) {
1240 d12f57ec qiaonuohan
        if (lzo_init() != LZO_E_OK) {
1241 d12f57ec qiaonuohan
            /* return 0 to indicate lzo is unavailable */
1242 d12f57ec qiaonuohan
            return 0;
1243 d12f57ec qiaonuohan
        }
1244 d12f57ec qiaonuohan
    }
1245 d12f57ec qiaonuohan
1246 d12f57ec qiaonuohan
    /*
1247 d12f57ec qiaonuohan
     * LZO will expand incompressible data by a little amount. please check the
1248 d12f57ec qiaonuohan
     * following URL to see the expansion calculation:
1249 d12f57ec qiaonuohan
     * http://www.oberhumer.com/opensource/lzo/lzofaq.php
1250 d12f57ec qiaonuohan
     */
1251 d12f57ec qiaonuohan
    len_buf_out_lzo = page_size + page_size / 16 + 64 + 3;
1252 d12f57ec qiaonuohan
#endif
1253 d12f57ec qiaonuohan
1254 d12f57ec qiaonuohan
#ifdef CONFIG_SNAPPY
1255 d12f57ec qiaonuohan
    /* buf size for snappy */
1256 d12f57ec qiaonuohan
    len_buf_out_snappy = snappy_max_compressed_length(page_size);
1257 d12f57ec qiaonuohan
#endif
1258 d12f57ec qiaonuohan
1259 d12f57ec qiaonuohan
    /* get the biggest that can store all kinds of compressed page */
1260 d12f57ec qiaonuohan
    len_buf_out = MAX(len_buf_out_zlib,
1261 d12f57ec qiaonuohan
                      MAX(len_buf_out_lzo, len_buf_out_snappy));
1262 d12f57ec qiaonuohan
1263 d12f57ec qiaonuohan
    return len_buf_out;
1264 d12f57ec qiaonuohan
}
1265 d12f57ec qiaonuohan
1266 d12f57ec qiaonuohan
/*
1267 d12f57ec qiaonuohan
 * check if the page is all 0
1268 d12f57ec qiaonuohan
 */
1269 d12f57ec qiaonuohan
static inline bool is_zero_page(const uint8_t *buf, size_t page_size)
1270 d12f57ec qiaonuohan
{
1271 d12f57ec qiaonuohan
    return buffer_is_zero(buf, page_size);
1272 d12f57ec qiaonuohan
}
1273 d12f57ec qiaonuohan
1274 d12f57ec qiaonuohan
static int write_dump_pages(DumpState *s)
1275 d12f57ec qiaonuohan
{
1276 d12f57ec qiaonuohan
    int ret = 0;
1277 d12f57ec qiaonuohan
    DataCache page_desc, page_data;
1278 d12f57ec qiaonuohan
    size_t len_buf_out, size_out;
1279 d12f57ec qiaonuohan
#ifdef CONFIG_LZO
1280 d12f57ec qiaonuohan
    lzo_bytep wrkmem = NULL;
1281 d12f57ec qiaonuohan
#endif
1282 d12f57ec qiaonuohan
    uint8_t *buf_out = NULL;
1283 d12f57ec qiaonuohan
    off_t offset_desc, offset_data;
1284 d12f57ec qiaonuohan
    PageDescriptor pd, pd_zero;
1285 d12f57ec qiaonuohan
    uint8_t *buf;
1286 d12f57ec qiaonuohan
    int endian = s->dump_info.d_endian;
1287 d12f57ec qiaonuohan
    GuestPhysBlock *block_iter = NULL;
1288 d12f57ec qiaonuohan
    uint64_t pfn_iter;
1289 d12f57ec qiaonuohan
1290 d12f57ec qiaonuohan
    /* get offset of page_desc and page_data in dump file */
1291 d12f57ec qiaonuohan
    offset_desc = s->offset_page;
1292 d12f57ec qiaonuohan
    offset_data = offset_desc + sizeof(PageDescriptor) * s->num_dumpable;
1293 d12f57ec qiaonuohan
1294 d12f57ec qiaonuohan
    prepare_data_cache(&page_desc, s, offset_desc);
1295 d12f57ec qiaonuohan
    prepare_data_cache(&page_data, s, offset_data);
1296 d12f57ec qiaonuohan
1297 d12f57ec qiaonuohan
    /* prepare buffer to store compressed data */
1298 d12f57ec qiaonuohan
    len_buf_out = get_len_buf_out(s->page_size, s->flag_compress);
1299 d12f57ec qiaonuohan
    if (len_buf_out == 0) {
1300 d12f57ec qiaonuohan
        dump_error(s, "dump: failed to get length of output buffer.\n");
1301 d12f57ec qiaonuohan
        goto out;
1302 d12f57ec qiaonuohan
    }
1303 d12f57ec qiaonuohan
1304 d12f57ec qiaonuohan
#ifdef CONFIG_LZO
1305 d12f57ec qiaonuohan
    wrkmem = g_malloc(LZO1X_1_MEM_COMPRESS);
1306 d12f57ec qiaonuohan
#endif
1307 d12f57ec qiaonuohan
1308 d12f57ec qiaonuohan
    buf_out = g_malloc(len_buf_out);
1309 d12f57ec qiaonuohan
1310 d12f57ec qiaonuohan
    /*
1311 d12f57ec qiaonuohan
     * init zero page's page_desc and page_data, because every zero page
1312 d12f57ec qiaonuohan
     * uses the same page_data
1313 d12f57ec qiaonuohan
     */
1314 d12f57ec qiaonuohan
    pd_zero.size = cpu_convert_to_target32(s->page_size, endian);
1315 d12f57ec qiaonuohan
    pd_zero.flags = cpu_convert_to_target32(0, endian);
1316 d12f57ec qiaonuohan
    pd_zero.offset = cpu_convert_to_target64(offset_data, endian);
1317 d12f57ec qiaonuohan
    pd_zero.page_flags = cpu_convert_to_target64(0, endian);
1318 d12f57ec qiaonuohan
    buf = g_malloc0(s->page_size);
1319 d12f57ec qiaonuohan
    ret = write_cache(&page_data, buf, s->page_size, false);
1320 d12f57ec qiaonuohan
    g_free(buf);
1321 d12f57ec qiaonuohan
    if (ret < 0) {
1322 d12f57ec qiaonuohan
        dump_error(s, "dump: failed to write page data(zero page).\n");
1323 d12f57ec qiaonuohan
        goto out;
1324 d12f57ec qiaonuohan
    }
1325 d12f57ec qiaonuohan
1326 d12f57ec qiaonuohan
    offset_data += s->page_size;
1327 d12f57ec qiaonuohan
1328 d12f57ec qiaonuohan
    /*
1329 d12f57ec qiaonuohan
     * dump memory to vmcore page by page. zero page will all be resided in the
1330 d12f57ec qiaonuohan
     * first page of page section
1331 d12f57ec qiaonuohan
     */
1332 d12f57ec qiaonuohan
    while (get_next_page(&block_iter, &pfn_iter, &buf, s)) {
1333 d12f57ec qiaonuohan
        /* check zero page */
1334 d12f57ec qiaonuohan
        if (is_zero_page(buf, s->page_size)) {
1335 d12f57ec qiaonuohan
            ret = write_cache(&page_desc, &pd_zero, sizeof(PageDescriptor),
1336 d12f57ec qiaonuohan
                              false);
1337 d12f57ec qiaonuohan
            if (ret < 0) {
1338 d12f57ec qiaonuohan
                dump_error(s, "dump: failed to write page desc.\n");
1339 d12f57ec qiaonuohan
                goto out;
1340 d12f57ec qiaonuohan
            }
1341 d12f57ec qiaonuohan
        } else {
1342 d12f57ec qiaonuohan
            /*
1343 d12f57ec qiaonuohan
             * not zero page, then:
1344 d12f57ec qiaonuohan
             * 1. compress the page
1345 d12f57ec qiaonuohan
             * 2. write the compressed page into the cache of page_data
1346 d12f57ec qiaonuohan
             * 3. get page desc of the compressed page and write it into the
1347 d12f57ec qiaonuohan
             *    cache of page_desc
1348 d12f57ec qiaonuohan
             *
1349 d12f57ec qiaonuohan
             * only one compression format will be used here, for
1350 d12f57ec qiaonuohan
             * s->flag_compress is set. But when compression fails to work,
1351 d12f57ec qiaonuohan
             * we fall back to save in plaintext.
1352 d12f57ec qiaonuohan
             */
1353 d12f57ec qiaonuohan
             size_out = len_buf_out;
1354 d12f57ec qiaonuohan
             if ((s->flag_compress & DUMP_DH_COMPRESSED_ZLIB) &&
1355 d12f57ec qiaonuohan
                    (compress2(buf_out, (uLongf *)&size_out, buf, s->page_size,
1356 d12f57ec qiaonuohan
                    Z_BEST_SPEED) == Z_OK) && (size_out < s->page_size)) {
1357 d12f57ec qiaonuohan
                pd.flags = cpu_convert_to_target32(DUMP_DH_COMPRESSED_ZLIB,
1358 d12f57ec qiaonuohan
                                                   endian);
1359 d12f57ec qiaonuohan
                pd.size  = cpu_convert_to_target32(size_out, endian);
1360 d12f57ec qiaonuohan
1361 d12f57ec qiaonuohan
                ret = write_cache(&page_data, buf_out, size_out, false);
1362 d12f57ec qiaonuohan
                if (ret < 0) {
1363 d12f57ec qiaonuohan
                    dump_error(s, "dump: failed to write page data.\n");
1364 d12f57ec qiaonuohan
                    goto out;
1365 d12f57ec qiaonuohan
                }
1366 d12f57ec qiaonuohan
#ifdef CONFIG_LZO
1367 d12f57ec qiaonuohan
            } else if ((s->flag_compress & DUMP_DH_COMPRESSED_LZO) &&
1368 d12f57ec qiaonuohan
                    (lzo1x_1_compress(buf, s->page_size, buf_out,
1369 d12f57ec qiaonuohan
                    (lzo_uint *)&size_out, wrkmem) == LZO_E_OK) &&
1370 d12f57ec qiaonuohan
                    (size_out < s->page_size)) {
1371 d12f57ec qiaonuohan
                pd.flags = cpu_convert_to_target32(DUMP_DH_COMPRESSED_LZO,
1372 d12f57ec qiaonuohan
                                                   endian);
1373 d12f57ec qiaonuohan
                pd.size  = cpu_convert_to_target32(size_out, endian);
1374 d12f57ec qiaonuohan
1375 d12f57ec qiaonuohan
                ret = write_cache(&page_data, buf_out, size_out, false);
1376 d12f57ec qiaonuohan
                if (ret < 0) {
1377 d12f57ec qiaonuohan
                    dump_error(s, "dump: failed to write page data.\n");
1378 d12f57ec qiaonuohan
                    goto out;
1379 d12f57ec qiaonuohan
                }
1380 d12f57ec qiaonuohan
#endif
1381 d12f57ec qiaonuohan
#ifdef CONFIG_SNAPPY
1382 d12f57ec qiaonuohan
            } else if ((s->flag_compress & DUMP_DH_COMPRESSED_SNAPPY) &&
1383 d12f57ec qiaonuohan
                    (snappy_compress((char *)buf, s->page_size,
1384 d12f57ec qiaonuohan
                    (char *)buf_out, &size_out) == SNAPPY_OK) &&
1385 d12f57ec qiaonuohan
                    (size_out < s->page_size)) {
1386 d12f57ec qiaonuohan
                pd.flags = cpu_convert_to_target32(
1387 d12f57ec qiaonuohan
                                        DUMP_DH_COMPRESSED_SNAPPY, endian);
1388 d12f57ec qiaonuohan
                pd.size  = cpu_convert_to_target32(size_out, endian);
1389 d12f57ec qiaonuohan
1390 d12f57ec qiaonuohan
                ret = write_cache(&page_data, buf_out, size_out, false);
1391 d12f57ec qiaonuohan
                if (ret < 0) {
1392 d12f57ec qiaonuohan
                    dump_error(s, "dump: failed to write page data.\n");
1393 d12f57ec qiaonuohan
                    goto out;
1394 d12f57ec qiaonuohan
                }
1395 d12f57ec qiaonuohan
#endif
1396 d12f57ec qiaonuohan
            } else {
1397 d12f57ec qiaonuohan
                /*
1398 d12f57ec qiaonuohan
                 * fall back to save in plaintext, size_out should be
1399 d12f57ec qiaonuohan
                 * assigned to s->page_size
1400 d12f57ec qiaonuohan
                 */
1401 d12f57ec qiaonuohan
                pd.flags = cpu_convert_to_target32(0, endian);
1402 d12f57ec qiaonuohan
                size_out = s->page_size;
1403 d12f57ec qiaonuohan
                pd.size = cpu_convert_to_target32(size_out, endian);
1404 d12f57ec qiaonuohan
1405 d12f57ec qiaonuohan
                ret = write_cache(&page_data, buf, s->page_size, false);
1406 d12f57ec qiaonuohan
                if (ret < 0) {
1407 d12f57ec qiaonuohan
                    dump_error(s, "dump: failed to write page data.\n");
1408 d12f57ec qiaonuohan
                    goto out;
1409 d12f57ec qiaonuohan
                }
1410 d12f57ec qiaonuohan
            }
1411 d12f57ec qiaonuohan
1412 d12f57ec qiaonuohan
            /* get and write page desc here */
1413 d12f57ec qiaonuohan
            pd.page_flags = cpu_convert_to_target64(0, endian);
1414 d12f57ec qiaonuohan
            pd.offset = cpu_convert_to_target64(offset_data, endian);
1415 d12f57ec qiaonuohan
            offset_data += size_out;
1416 d12f57ec qiaonuohan
1417 d12f57ec qiaonuohan
            ret = write_cache(&page_desc, &pd, sizeof(PageDescriptor), false);
1418 d12f57ec qiaonuohan
            if (ret < 0) {
1419 d12f57ec qiaonuohan
                dump_error(s, "dump: failed to write page desc.\n");
1420 d12f57ec qiaonuohan
                goto out;
1421 d12f57ec qiaonuohan
            }
1422 d12f57ec qiaonuohan
        }
1423 d12f57ec qiaonuohan
    }
1424 d12f57ec qiaonuohan
1425 d12f57ec qiaonuohan
    ret = write_cache(&page_desc, NULL, 0, true);
1426 d12f57ec qiaonuohan
    if (ret < 0) {
1427 d12f57ec qiaonuohan
        dump_error(s, "dump: failed to sync cache for page_desc.\n");
1428 d12f57ec qiaonuohan
        goto out;
1429 d12f57ec qiaonuohan
    }
1430 d12f57ec qiaonuohan
    ret = write_cache(&page_data, NULL, 0, true);
1431 d12f57ec qiaonuohan
    if (ret < 0) {
1432 d12f57ec qiaonuohan
        dump_error(s, "dump: failed to sync cache for page_data.\n");
1433 d12f57ec qiaonuohan
        goto out;
1434 d12f57ec qiaonuohan
    }
1435 d12f57ec qiaonuohan
1436 d12f57ec qiaonuohan
out:
1437 d12f57ec qiaonuohan
    free_data_cache(&page_desc);
1438 d12f57ec qiaonuohan
    free_data_cache(&page_data);
1439 d12f57ec qiaonuohan
1440 d12f57ec qiaonuohan
#ifdef CONFIG_LZO
1441 d12f57ec qiaonuohan
    g_free(wrkmem);
1442 d12f57ec qiaonuohan
#endif
1443 d12f57ec qiaonuohan
1444 d12f57ec qiaonuohan
    g_free(buf_out);
1445 d12f57ec qiaonuohan
1446 d12f57ec qiaonuohan
    return ret;
1447 d12f57ec qiaonuohan
}
1448 d12f57ec qiaonuohan
1449 b53ccc30 qiaonuohan
static int create_kdump_vmcore(DumpState *s)
1450 b53ccc30 qiaonuohan
{
1451 b53ccc30 qiaonuohan
    int ret;
1452 b53ccc30 qiaonuohan
1453 b53ccc30 qiaonuohan
    /*
1454 b53ccc30 qiaonuohan
     * the kdump-compressed format is:
1455 b53ccc30 qiaonuohan
     *                                               File offset
1456 b53ccc30 qiaonuohan
     *  +------------------------------------------+ 0x0
1457 b53ccc30 qiaonuohan
     *  |    main header (struct disk_dump_header) |
1458 b53ccc30 qiaonuohan
     *  |------------------------------------------+ block 1
1459 b53ccc30 qiaonuohan
     *  |    sub header (struct kdump_sub_header)  |
1460 b53ccc30 qiaonuohan
     *  |------------------------------------------+ block 2
1461 b53ccc30 qiaonuohan
     *  |            1st-dump_bitmap               |
1462 b53ccc30 qiaonuohan
     *  |------------------------------------------+ block 2 + X blocks
1463 b53ccc30 qiaonuohan
     *  |            2nd-dump_bitmap               | (aligned by block)
1464 b53ccc30 qiaonuohan
     *  |------------------------------------------+ block 2 + 2 * X blocks
1465 b53ccc30 qiaonuohan
     *  |  page desc for pfn 0 (struct page_desc)  | (aligned by block)
1466 b53ccc30 qiaonuohan
     *  |  page desc for pfn 1 (struct page_desc)  |
1467 b53ccc30 qiaonuohan
     *  |                    :                     |
1468 b53ccc30 qiaonuohan
     *  |------------------------------------------| (not aligned by block)
1469 b53ccc30 qiaonuohan
     *  |         page data (pfn 0)                |
1470 b53ccc30 qiaonuohan
     *  |         page data (pfn 1)                |
1471 b53ccc30 qiaonuohan
     *  |                    :                     |
1472 b53ccc30 qiaonuohan
     *  +------------------------------------------+
1473 b53ccc30 qiaonuohan
     */
1474 b53ccc30 qiaonuohan
1475 b53ccc30 qiaonuohan
    ret = write_start_flat_header(s->fd);
1476 b53ccc30 qiaonuohan
    if (ret < 0) {
1477 b53ccc30 qiaonuohan
        dump_error(s, "dump: failed to write start flat header.\n");
1478 b53ccc30 qiaonuohan
        return -1;
1479 b53ccc30 qiaonuohan
    }
1480 b53ccc30 qiaonuohan
1481 b53ccc30 qiaonuohan
    ret = write_dump_header(s);
1482 b53ccc30 qiaonuohan
    if (ret < 0) {
1483 b53ccc30 qiaonuohan
        return -1;
1484 b53ccc30 qiaonuohan
    }
1485 b53ccc30 qiaonuohan
1486 b53ccc30 qiaonuohan
    ret = write_dump_bitmap(s);
1487 b53ccc30 qiaonuohan
    if (ret < 0) {
1488 b53ccc30 qiaonuohan
        return -1;
1489 b53ccc30 qiaonuohan
    }
1490 b53ccc30 qiaonuohan
1491 b53ccc30 qiaonuohan
    ret = write_dump_pages(s);
1492 b53ccc30 qiaonuohan
    if (ret < 0) {
1493 b53ccc30 qiaonuohan
        return -1;
1494 b53ccc30 qiaonuohan
    }
1495 b53ccc30 qiaonuohan
1496 b53ccc30 qiaonuohan
    ret = write_end_flat_header(s->fd);
1497 b53ccc30 qiaonuohan
    if (ret < 0) {
1498 b53ccc30 qiaonuohan
        dump_error(s, "dump: failed to write end flat header.\n");
1499 b53ccc30 qiaonuohan
        return -1;
1500 b53ccc30 qiaonuohan
    }
1501 b53ccc30 qiaonuohan
1502 b53ccc30 qiaonuohan
    dump_completed(s);
1503 b53ccc30 qiaonuohan
1504 b53ccc30 qiaonuohan
    return 0;
1505 b53ccc30 qiaonuohan
}
1506 b53ccc30 qiaonuohan
1507 783e9b48 Wen Congyang
static ram_addr_t get_start_block(DumpState *s)
1508 783e9b48 Wen Congyang
{
1509 56c4bfb3 Laszlo Ersek
    GuestPhysBlock *block;
1510 783e9b48 Wen Congyang
1511 783e9b48 Wen Congyang
    if (!s->has_filter) {
1512 56c4bfb3 Laszlo Ersek
        s->next_block = QTAILQ_FIRST(&s->guest_phys_blocks.head);
1513 783e9b48 Wen Congyang
        return 0;
1514 783e9b48 Wen Congyang
    }
1515 783e9b48 Wen Congyang
1516 56c4bfb3 Laszlo Ersek
    QTAILQ_FOREACH(block, &s->guest_phys_blocks.head, next) {
1517 56c4bfb3 Laszlo Ersek
        if (block->target_start >= s->begin + s->length ||
1518 56c4bfb3 Laszlo Ersek
            block->target_end <= s->begin) {
1519 783e9b48 Wen Congyang
            /* This block is out of the range */
1520 783e9b48 Wen Congyang
            continue;
1521 783e9b48 Wen Congyang
        }
1522 783e9b48 Wen Congyang
1523 56c4bfb3 Laszlo Ersek
        s->next_block = block;
1524 56c4bfb3 Laszlo Ersek
        if (s->begin > block->target_start) {
1525 56c4bfb3 Laszlo Ersek
            s->start = s->begin - block->target_start;
1526 783e9b48 Wen Congyang
        } else {
1527 783e9b48 Wen Congyang
            s->start = 0;
1528 783e9b48 Wen Congyang
        }
1529 783e9b48 Wen Congyang
        return s->start;
1530 783e9b48 Wen Congyang
    }
1531 783e9b48 Wen Congyang
1532 783e9b48 Wen Congyang
    return -1;
1533 783e9b48 Wen Congyang
}
1534 783e9b48 Wen Congyang
1535 7aad248d qiaonuohan
static void get_max_mapnr(DumpState *s)
1536 7aad248d qiaonuohan
{
1537 7aad248d qiaonuohan
    GuestPhysBlock *last_block;
1538 7aad248d qiaonuohan
1539 7aad248d qiaonuohan
    last_block = QTAILQ_LAST(&s->guest_phys_blocks.head, GuestPhysBlockHead);
1540 7aad248d qiaonuohan
    s->max_mapnr = paddr_to_pfn(last_block->target_end, s->page_shift);
1541 7aad248d qiaonuohan
}
1542 7aad248d qiaonuohan
1543 b53ccc30 qiaonuohan
static int dump_init(DumpState *s, int fd, bool has_format,
1544 b53ccc30 qiaonuohan
                     DumpGuestMemoryFormat format, bool paging, bool has_filter,
1545 783e9b48 Wen Congyang
                     int64_t begin, int64_t length, Error **errp)
1546 783e9b48 Wen Congyang
{
1547 182735ef Andreas Färber
    CPUState *cpu;
1548 783e9b48 Wen Congyang
    int nr_cpus;
1549 11ed09cf Andreas Färber
    Error *err = NULL;
1550 783e9b48 Wen Congyang
    int ret;
1551 783e9b48 Wen Congyang
1552 b53ccc30 qiaonuohan
    /* kdump-compressed is conflict with paging and filter */
1553 b53ccc30 qiaonuohan
    if (has_format && format != DUMP_GUEST_MEMORY_FORMAT_ELF) {
1554 b53ccc30 qiaonuohan
        assert(!paging && !has_filter);
1555 b53ccc30 qiaonuohan
    }
1556 b53ccc30 qiaonuohan
1557 783e9b48 Wen Congyang
    if (runstate_is_running()) {
1558 783e9b48 Wen Congyang
        vm_stop(RUN_STATE_SAVE_VM);
1559 783e9b48 Wen Congyang
        s->resume = true;
1560 783e9b48 Wen Congyang
    } else {
1561 783e9b48 Wen Congyang
        s->resume = false;
1562 783e9b48 Wen Congyang
    }
1563 783e9b48 Wen Congyang
1564 5ee163e8 Laszlo Ersek
    /* If we use KVM, we should synchronize the registers before we get dump
1565 5ee163e8 Laszlo Ersek
     * info or physmap info.
1566 5ee163e8 Laszlo Ersek
     */
1567 5ee163e8 Laszlo Ersek
    cpu_synchronize_all_states();
1568 5ee163e8 Laszlo Ersek
    nr_cpus = 0;
1569 bdc44640 Andreas Färber
    CPU_FOREACH(cpu) {
1570 5ee163e8 Laszlo Ersek
        nr_cpus++;
1571 5ee163e8 Laszlo Ersek
    }
1572 5ee163e8 Laszlo Ersek
1573 783e9b48 Wen Congyang
    s->errp = errp;
1574 783e9b48 Wen Congyang
    s->fd = fd;
1575 783e9b48 Wen Congyang
    s->has_filter = has_filter;
1576 783e9b48 Wen Congyang
    s->begin = begin;
1577 783e9b48 Wen Congyang
    s->length = length;
1578 5ee163e8 Laszlo Ersek
1579 5ee163e8 Laszlo Ersek
    guest_phys_blocks_init(&s->guest_phys_blocks);
1580 c5d7f60f Laszlo Ersek
    guest_phys_blocks_append(&s->guest_phys_blocks);
1581 5ee163e8 Laszlo Ersek
1582 783e9b48 Wen Congyang
    s->start = get_start_block(s);
1583 783e9b48 Wen Congyang
    if (s->start == -1) {
1584 783e9b48 Wen Congyang
        error_set(errp, QERR_INVALID_PARAMETER, "begin");
1585 783e9b48 Wen Congyang
        goto cleanup;
1586 783e9b48 Wen Congyang
    }
1587 783e9b48 Wen Congyang
1588 5ee163e8 Laszlo Ersek
    /* get dump info: endian, class and architecture.
1589 783e9b48 Wen Congyang
     * If the target architecture is not supported, cpu_get_dump_info() will
1590 783e9b48 Wen Congyang
     * return -1.
1591 783e9b48 Wen Congyang
     */
1592 56c4bfb3 Laszlo Ersek
    ret = cpu_get_dump_info(&s->dump_info, &s->guest_phys_blocks);
1593 783e9b48 Wen Congyang
    if (ret < 0) {
1594 783e9b48 Wen Congyang
        error_set(errp, QERR_UNSUPPORTED);
1595 783e9b48 Wen Congyang
        goto cleanup;
1596 783e9b48 Wen Congyang
    }
1597 783e9b48 Wen Congyang
1598 4720bd05 Paolo Bonzini
    s->note_size = cpu_get_note_size(s->dump_info.d_class,
1599 4720bd05 Paolo Bonzini
                                     s->dump_info.d_machine, nr_cpus);
1600 bb6b6843 Aneesh Kumar K.V
    if (s->note_size < 0) {
1601 4720bd05 Paolo Bonzini
        error_set(errp, QERR_UNSUPPORTED);
1602 4720bd05 Paolo Bonzini
        goto cleanup;
1603 4720bd05 Paolo Bonzini
    }
1604 4720bd05 Paolo Bonzini
1605 783e9b48 Wen Congyang
    /* get memory mapping */
1606 783e9b48 Wen Congyang
    memory_mapping_list_init(&s->list);
1607 783e9b48 Wen Congyang
    if (paging) {
1608 56c4bfb3 Laszlo Ersek
        qemu_get_guest_memory_mapping(&s->list, &s->guest_phys_blocks, &err);
1609 11ed09cf Andreas Färber
        if (err != NULL) {
1610 11ed09cf Andreas Färber
            error_propagate(errp, err);
1611 11ed09cf Andreas Färber
            goto cleanup;
1612 11ed09cf Andreas Färber
        }
1613 783e9b48 Wen Congyang
    } else {
1614 56c4bfb3 Laszlo Ersek
        qemu_get_guest_simple_memory_mapping(&s->list, &s->guest_phys_blocks);
1615 783e9b48 Wen Congyang
    }
1616 783e9b48 Wen Congyang
1617 7aad248d qiaonuohan
    s->nr_cpus = nr_cpus;
1618 7aad248d qiaonuohan
    s->page_size = TARGET_PAGE_SIZE;
1619 7aad248d qiaonuohan
    s->page_shift = ffs(s->page_size) - 1;
1620 7aad248d qiaonuohan
1621 7aad248d qiaonuohan
    get_max_mapnr(s);
1622 7aad248d qiaonuohan
1623 7aad248d qiaonuohan
    uint64_t tmp;
1624 7aad248d qiaonuohan
    tmp = DIV_ROUND_UP(DIV_ROUND_UP(s->max_mapnr, CHAR_BIT), s->page_size);
1625 7aad248d qiaonuohan
    s->len_dump_bitmap = tmp * s->page_size;
1626 7aad248d qiaonuohan
1627 b53ccc30 qiaonuohan
    /* init for kdump-compressed format */
1628 b53ccc30 qiaonuohan
    if (has_format && format != DUMP_GUEST_MEMORY_FORMAT_ELF) {
1629 b53ccc30 qiaonuohan
        switch (format) {
1630 b53ccc30 qiaonuohan
        case DUMP_GUEST_MEMORY_FORMAT_KDUMP_ZLIB:
1631 b53ccc30 qiaonuohan
            s->flag_compress = DUMP_DH_COMPRESSED_ZLIB;
1632 b53ccc30 qiaonuohan
            break;
1633 b53ccc30 qiaonuohan
1634 b53ccc30 qiaonuohan
        case DUMP_GUEST_MEMORY_FORMAT_KDUMP_LZO:
1635 b53ccc30 qiaonuohan
            s->flag_compress = DUMP_DH_COMPRESSED_LZO;
1636 b53ccc30 qiaonuohan
            break;
1637 b53ccc30 qiaonuohan
1638 b53ccc30 qiaonuohan
        case DUMP_GUEST_MEMORY_FORMAT_KDUMP_SNAPPY:
1639 b53ccc30 qiaonuohan
            s->flag_compress = DUMP_DH_COMPRESSED_SNAPPY;
1640 b53ccc30 qiaonuohan
            break;
1641 b53ccc30 qiaonuohan
1642 b53ccc30 qiaonuohan
        default:
1643 b53ccc30 qiaonuohan
            s->flag_compress = 0;
1644 b53ccc30 qiaonuohan
        }
1645 b53ccc30 qiaonuohan
1646 b53ccc30 qiaonuohan
        return 0;
1647 b53ccc30 qiaonuohan
    }
1648 b53ccc30 qiaonuohan
1649 783e9b48 Wen Congyang
    if (s->has_filter) {
1650 783e9b48 Wen Congyang
        memory_mapping_filter(&s->list, s->begin, s->length);
1651 783e9b48 Wen Congyang
    }
1652 783e9b48 Wen Congyang
1653 783e9b48 Wen Congyang
    /*
1654 783e9b48 Wen Congyang
     * calculate phdr_num
1655 783e9b48 Wen Congyang
     *
1656 783e9b48 Wen Congyang
     * the type of ehdr->e_phnum is uint16_t, so we should avoid overflow
1657 783e9b48 Wen Congyang
     */
1658 783e9b48 Wen Congyang
    s->phdr_num = 1; /* PT_NOTE */
1659 783e9b48 Wen Congyang
    if (s->list.num < UINT16_MAX - 2) {
1660 783e9b48 Wen Congyang
        s->phdr_num += s->list.num;
1661 783e9b48 Wen Congyang
        s->have_section = false;
1662 783e9b48 Wen Congyang
    } else {
1663 783e9b48 Wen Congyang
        s->have_section = true;
1664 783e9b48 Wen Congyang
        s->phdr_num = PN_XNUM;
1665 783e9b48 Wen Congyang
        s->sh_info = 1; /* PT_NOTE */
1666 783e9b48 Wen Congyang
1667 783e9b48 Wen Congyang
        /* the type of shdr->sh_info is uint32_t, so we should avoid overflow */
1668 783e9b48 Wen Congyang
        if (s->list.num <= UINT32_MAX - 1) {
1669 783e9b48 Wen Congyang
            s->sh_info += s->list.num;
1670 783e9b48 Wen Congyang
        } else {
1671 783e9b48 Wen Congyang
            s->sh_info = UINT32_MAX;
1672 783e9b48 Wen Congyang
        }
1673 783e9b48 Wen Congyang
    }
1674 783e9b48 Wen Congyang
1675 783e9b48 Wen Congyang
    if (s->dump_info.d_class == ELFCLASS64) {
1676 783e9b48 Wen Congyang
        if (s->have_section) {
1677 783e9b48 Wen Congyang
            s->memory_offset = sizeof(Elf64_Ehdr) +
1678 783e9b48 Wen Congyang
                               sizeof(Elf64_Phdr) * s->sh_info +
1679 783e9b48 Wen Congyang
                               sizeof(Elf64_Shdr) + s->note_size;
1680 783e9b48 Wen Congyang
        } else {
1681 783e9b48 Wen Congyang
            s->memory_offset = sizeof(Elf64_Ehdr) +
1682 783e9b48 Wen Congyang
                               sizeof(Elf64_Phdr) * s->phdr_num + s->note_size;
1683 783e9b48 Wen Congyang
        }
1684 783e9b48 Wen Congyang
    } else {
1685 783e9b48 Wen Congyang
        if (s->have_section) {
1686 783e9b48 Wen Congyang
            s->memory_offset = sizeof(Elf32_Ehdr) +
1687 783e9b48 Wen Congyang
                               sizeof(Elf32_Phdr) * s->sh_info +
1688 783e9b48 Wen Congyang
                               sizeof(Elf32_Shdr) + s->note_size;
1689 783e9b48 Wen Congyang
        } else {
1690 783e9b48 Wen Congyang
            s->memory_offset = sizeof(Elf32_Ehdr) +
1691 783e9b48 Wen Congyang
                               sizeof(Elf32_Phdr) * s->phdr_num + s->note_size;
1692 783e9b48 Wen Congyang
        }
1693 783e9b48 Wen Congyang
    }
1694 783e9b48 Wen Congyang
1695 783e9b48 Wen Congyang
    return 0;
1696 783e9b48 Wen Congyang
1697 783e9b48 Wen Congyang
cleanup:
1698 5ee163e8 Laszlo Ersek
    guest_phys_blocks_free(&s->guest_phys_blocks);
1699 5ee163e8 Laszlo Ersek
1700 783e9b48 Wen Congyang
    if (s->resume) {
1701 783e9b48 Wen Congyang
        vm_start();
1702 783e9b48 Wen Congyang
    }
1703 783e9b48 Wen Congyang
1704 783e9b48 Wen Congyang
    return -1;
1705 783e9b48 Wen Congyang
}
1706 783e9b48 Wen Congyang
1707 783e9b48 Wen Congyang
void qmp_dump_guest_memory(bool paging, const char *file, bool has_begin,
1708 b53ccc30 qiaonuohan
                           int64_t begin, bool has_length,
1709 b53ccc30 qiaonuohan
                           int64_t length, bool has_format,
1710 b53ccc30 qiaonuohan
                           DumpGuestMemoryFormat format, Error **errp)
1711 783e9b48 Wen Congyang
{
1712 783e9b48 Wen Congyang
    const char *p;
1713 783e9b48 Wen Congyang
    int fd = -1;
1714 783e9b48 Wen Congyang
    DumpState *s;
1715 783e9b48 Wen Congyang
    int ret;
1716 783e9b48 Wen Congyang
1717 b53ccc30 qiaonuohan
    /*
1718 b53ccc30 qiaonuohan
     * kdump-compressed format need the whole memory dumped, so paging or
1719 b53ccc30 qiaonuohan
     * filter is not supported here.
1720 b53ccc30 qiaonuohan
     */
1721 b53ccc30 qiaonuohan
    if ((has_format && format != DUMP_GUEST_MEMORY_FORMAT_ELF) &&
1722 b53ccc30 qiaonuohan
        (paging || has_begin || has_length)) {
1723 b53ccc30 qiaonuohan
        error_setg(errp, "kdump-compressed format doesn't support paging or "
1724 b53ccc30 qiaonuohan
                         "filter");
1725 b53ccc30 qiaonuohan
        return;
1726 b53ccc30 qiaonuohan
    }
1727 783e9b48 Wen Congyang
    if (has_begin && !has_length) {
1728 783e9b48 Wen Congyang
        error_set(errp, QERR_MISSING_PARAMETER, "length");
1729 783e9b48 Wen Congyang
        return;
1730 783e9b48 Wen Congyang
    }
1731 783e9b48 Wen Congyang
    if (!has_begin && has_length) {
1732 783e9b48 Wen Congyang
        error_set(errp, QERR_MISSING_PARAMETER, "begin");
1733 783e9b48 Wen Congyang
        return;
1734 783e9b48 Wen Congyang
    }
1735 783e9b48 Wen Congyang
1736 b53ccc30 qiaonuohan
    /* check whether lzo/snappy is supported */
1737 b53ccc30 qiaonuohan
#ifndef CONFIG_LZO
1738 b53ccc30 qiaonuohan
    if (has_format && format == DUMP_GUEST_MEMORY_FORMAT_KDUMP_LZO) {
1739 b53ccc30 qiaonuohan
        error_setg(errp, "kdump-lzo is not available now");
1740 b53ccc30 qiaonuohan
        return;
1741 b53ccc30 qiaonuohan
    }
1742 b53ccc30 qiaonuohan
#endif
1743 b53ccc30 qiaonuohan
1744 b53ccc30 qiaonuohan
#ifndef CONFIG_SNAPPY
1745 b53ccc30 qiaonuohan
    if (has_format && format == DUMP_GUEST_MEMORY_FORMAT_KDUMP_SNAPPY) {
1746 b53ccc30 qiaonuohan
        error_setg(errp, "kdump-snappy is not available now");
1747 b53ccc30 qiaonuohan
        return;
1748 b53ccc30 qiaonuohan
    }
1749 b53ccc30 qiaonuohan
#endif
1750 b53ccc30 qiaonuohan
1751 783e9b48 Wen Congyang
#if !defined(WIN32)
1752 783e9b48 Wen Congyang
    if (strstart(file, "fd:", &p)) {
1753 a9940fc4 Paolo Bonzini
        fd = monitor_get_fd(cur_mon, p, errp);
1754 783e9b48 Wen Congyang
        if (fd == -1) {
1755 783e9b48 Wen Congyang
            return;
1756 783e9b48 Wen Congyang
        }
1757 783e9b48 Wen Congyang
    }
1758 783e9b48 Wen Congyang
#endif
1759 783e9b48 Wen Congyang
1760 783e9b48 Wen Congyang
    if  (strstart(file, "file:", &p)) {
1761 783e9b48 Wen Congyang
        fd = qemu_open(p, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY, S_IRUSR);
1762 783e9b48 Wen Congyang
        if (fd < 0) {
1763 7581766b Luiz Capitulino
            error_setg_file_open(errp, errno, p);
1764 783e9b48 Wen Congyang
            return;
1765 783e9b48 Wen Congyang
        }
1766 783e9b48 Wen Congyang
    }
1767 783e9b48 Wen Congyang
1768 783e9b48 Wen Congyang
    if (fd == -1) {
1769 783e9b48 Wen Congyang
        error_set(errp, QERR_INVALID_PARAMETER, "protocol");
1770 783e9b48 Wen Congyang
        return;
1771 783e9b48 Wen Congyang
    }
1772 783e9b48 Wen Congyang
1773 5ee163e8 Laszlo Ersek
    s = g_malloc0(sizeof(DumpState));
1774 783e9b48 Wen Congyang
1775 b53ccc30 qiaonuohan
    ret = dump_init(s, fd, has_format, format, paging, has_begin,
1776 b53ccc30 qiaonuohan
                    begin, length, errp);
1777 783e9b48 Wen Congyang
    if (ret < 0) {
1778 783e9b48 Wen Congyang
        g_free(s);
1779 783e9b48 Wen Congyang
        return;
1780 783e9b48 Wen Congyang
    }
1781 783e9b48 Wen Congyang
1782 b53ccc30 qiaonuohan
    if (has_format && format != DUMP_GUEST_MEMORY_FORMAT_ELF) {
1783 b53ccc30 qiaonuohan
        if (create_kdump_vmcore(s) < 0 && !error_is_set(s->errp)) {
1784 b53ccc30 qiaonuohan
            error_set(errp, QERR_IO_ERROR);
1785 b53ccc30 qiaonuohan
        }
1786 b53ccc30 qiaonuohan
    } else {
1787 b53ccc30 qiaonuohan
        if (create_vmcore(s) < 0 && !error_is_set(s->errp)) {
1788 b53ccc30 qiaonuohan
            error_set(errp, QERR_IO_ERROR);
1789 b53ccc30 qiaonuohan
        }
1790 783e9b48 Wen Congyang
    }
1791 783e9b48 Wen Congyang
1792 783e9b48 Wen Congyang
    g_free(s);
1793 783e9b48 Wen Congyang
}
1794 7d6dc7f3 qiaonuohan
1795 7d6dc7f3 qiaonuohan
DumpGuestMemoryCapability *qmp_query_dump_guest_memory_capability(Error **errp)
1796 7d6dc7f3 qiaonuohan
{
1797 7d6dc7f3 qiaonuohan
    DumpGuestMemoryFormatList *item;
1798 7d6dc7f3 qiaonuohan
    DumpGuestMemoryCapability *cap =
1799 7d6dc7f3 qiaonuohan
                                  g_malloc0(sizeof(DumpGuestMemoryCapability));
1800 7d6dc7f3 qiaonuohan
1801 7d6dc7f3 qiaonuohan
    /* elf is always available */
1802 7d6dc7f3 qiaonuohan
    item = g_malloc0(sizeof(DumpGuestMemoryFormatList));
1803 7d6dc7f3 qiaonuohan
    cap->formats = item;
1804 7d6dc7f3 qiaonuohan
    item->value = DUMP_GUEST_MEMORY_FORMAT_ELF;
1805 7d6dc7f3 qiaonuohan
1806 7d6dc7f3 qiaonuohan
    /* kdump-zlib is always available */
1807 7d6dc7f3 qiaonuohan
    item->next = g_malloc0(sizeof(DumpGuestMemoryFormatList));
1808 7d6dc7f3 qiaonuohan
    item = item->next;
1809 7d6dc7f3 qiaonuohan
    item->value = DUMP_GUEST_MEMORY_FORMAT_KDUMP_ZLIB;
1810 7d6dc7f3 qiaonuohan
1811 7d6dc7f3 qiaonuohan
    /* add new item if kdump-lzo is available */
1812 7d6dc7f3 qiaonuohan
#ifdef CONFIG_LZO
1813 7d6dc7f3 qiaonuohan
    item->next = g_malloc0(sizeof(DumpGuestMemoryFormatList));
1814 7d6dc7f3 qiaonuohan
    item = item->next;
1815 7d6dc7f3 qiaonuohan
    item->value = DUMP_GUEST_MEMORY_FORMAT_KDUMP_LZO;
1816 7d6dc7f3 qiaonuohan
#endif
1817 7d6dc7f3 qiaonuohan
1818 7d6dc7f3 qiaonuohan
    /* add new item if kdump-snappy is available */
1819 7d6dc7f3 qiaonuohan
#ifdef CONFIG_SNAPPY
1820 7d6dc7f3 qiaonuohan
    item->next = g_malloc0(sizeof(DumpGuestMemoryFormatList));
1821 7d6dc7f3 qiaonuohan
    item = item->next;
1822 7d6dc7f3 qiaonuohan
    item->value = DUMP_GUEST_MEMORY_FORMAT_KDUMP_SNAPPY;
1823 7d6dc7f3 qiaonuohan
#endif
1824 7d6dc7f3 qiaonuohan
1825 7d6dc7f3 qiaonuohan
    return cap;
1826 7d6dc7f3 qiaonuohan
}