qemu-io.c 11.5 KB
Newer Older
1 2 3 4 5 6 7 8 9
/*
 * Command line utility to exercise the QEMU I/O path.
 *
 * Copyright (C) 2009 Red Hat, Inc.
 * Copyright (c) 2003-2005 Silicon Graphics, Inc.
 *
 * This work is licensed under the terms of the GNU GPL, version 2 or later.
 * See the COPYING file in the top-level directory.
 */
10
#include <sys/time.h>
11 12 13 14
#include <sys/types.h>
#include <stdarg.h>
#include <stdio.h>
#include <getopt.h>
15
#include <libgen.h>
16

Kevin Wolf's avatar
Kevin Wolf committed
17
#include "qemu-io.h"
18
#include "qemu/main-loop.h"
19 20
#include "qemu/option.h"
#include "qemu/config-file.h"
Stefan Hajnoczi's avatar
Stefan Hajnoczi committed
21
#include "qemu/readline.h"
22
#include "block/block_int.h"
23
#include "trace/control.h"
24

Devin Nakamura's avatar
Devin Nakamura committed
25
#define CMD_NOFILE_OK   0x01
26 27 28

char *progname;

29
BlockDriverState *qemuio_bs;
30
extern int qemuio_misalign;
Kevin Wolf's avatar
Kevin Wolf committed
31

32 33 34 35
/* qemu-io commands passed using -c */
static int ncmdline;
static char **cmdline;

Stefan Hajnoczi's avatar
Stefan Hajnoczi committed
36 37
static ReadLineState *readline_state;

38
static int close_f(BlockDriverState *bs, int argc, char **argv)
39
{
Fam Zheng's avatar
Fam Zheng committed
40
    bdrv_unref(bs);
41
    qemuio_bs = NULL;
Devin Nakamura's avatar
Devin Nakamura committed
42
    return 0;
43 44 45
}

static const cmdinfo_t close_cmd = {
Devin Nakamura's avatar
Devin Nakamura committed
46 47 48 49
    .name       = "close",
    .altname    = "c",
    .cfunc      = close_f,
    .oneline    = "close the current open file",
50 51
};

52
static int openfile(char *name, int flags, int growable, QDict *opts)
53
{
54 55
    Error *local_err = NULL;

56
    if (qemuio_bs) {
Devin Nakamura's avatar
Devin Nakamura committed
57 58 59 60 61
        fprintf(stderr, "file open already, try 'help close'\n");
        return 1;
    }

    if (growable) {
62
        if (bdrv_file_open(&qemuio_bs, name, NULL, opts, flags, &local_err)) {
63 64 65
            fprintf(stderr, "%s: can't open device %s: %s\n", progname, name,
                    error_get_pretty(local_err));
            error_free(local_err);
Devin Nakamura's avatar
Devin Nakamura committed
66 67 68
            return 1;
        }
    } else {
69
        qemuio_bs = bdrv_new("hda");
Devin Nakamura's avatar
Devin Nakamura committed
70

71
        if (bdrv_open(qemuio_bs, name, opts, flags, NULL, &local_err) < 0) {
72 73 74
            fprintf(stderr, "%s: can't open device %s: %s\n", progname, name,
                    error_get_pretty(local_err));
            error_free(local_err);
Fam Zheng's avatar
Fam Zheng committed
75
            bdrv_unref(qemuio_bs);
76
            qemuio_bs = NULL;
Devin Nakamura's avatar
Devin Nakamura committed
77 78 79 80 81
            return 1;
        }
    }

    return 0;
82 83
}

Devin Nakamura's avatar
Devin Nakamura committed
84
static void open_help(void)
85
{
Devin Nakamura's avatar
Devin Nakamura committed
86
    printf(
87 88 89 90 91 92 93 94 95 96
"\n"
" opens a new file in the requested mode\n"
"\n"
" Example:\n"
" 'open -Cn /tmp/data' - creates/opens data file read-write and uncached\n"
"\n"
" Opens a file for subsequent use by all of the other qemu-io commands.\n"
" -r, -- open file read-only\n"
" -s, -- use snapshot file\n"
" -n, -- disable host cache\n"
97 98
" -g, -- allow file to grow (only applies to protocols)\n"
" -o, -- options to be given to the block driver"
99 100 101
"\n");
}

102
static int open_f(BlockDriverState *bs, int argc, char **argv);
Blue Swirl's avatar
Blue Swirl committed
103 104

static const cmdinfo_t open_cmd = {
Devin Nakamura's avatar
Devin Nakamura committed
105 106 107 108 109 110
    .name       = "open",
    .altname    = "o",
    .cfunc      = open_f,
    .argmin     = 1,
    .argmax     = -1,
    .flags      = CMD_NOFILE_OK,
111
    .args       = "[-Crsn] [-o options] [path]",
Devin Nakamura's avatar
Devin Nakamura committed
112 113
    .oneline    = "open the file specified by path",
    .help       = open_help,
Blue Swirl's avatar
Blue Swirl committed
114
};
115

116 117 118 119 120 121 122 123 124
static QemuOptsList empty_opts = {
    .name = "drive",
    .head = QTAILQ_HEAD_INITIALIZER(empty_opts.head),
    .desc = {
        /* no elements => accept any params */
        { /* end of list */ }
    },
};

125
static int open_f(BlockDriverState *bs, int argc, char **argv)
126
{
Devin Nakamura's avatar
Devin Nakamura committed
127 128 129 130
    int flags = 0;
    int readonly = 0;
    int growable = 0;
    int c;
131 132
    QemuOpts *qopts;
    QDict *opts = NULL;
Devin Nakamura's avatar
Devin Nakamura committed
133

134
    while ((c = getopt(argc, argv, "snrgo:")) != EOF) {
Devin Nakamura's avatar
Devin Nakamura committed
135 136 137 138 139 140 141 142 143 144 145 146 147
        switch (c) {
        case 's':
            flags |= BDRV_O_SNAPSHOT;
            break;
        case 'n':
            flags |= BDRV_O_NOCACHE | BDRV_O_CACHE_WB;
            break;
        case 'r':
            readonly = 1;
            break;
        case 'g':
            growable = 1;
            break;
148 149 150 151 152 153 154 155 156
        case 'o':
            qopts = qemu_opts_parse(&empty_opts, optarg, 0);
            if (qopts == NULL) {
                printf("could not parse option list -- %s\n", optarg);
                return 0;
            }
            opts = qemu_opts_to_qdict(qopts, opts);
            qemu_opts_del(qopts);
            break;
Devin Nakamura's avatar
Devin Nakamura committed
157
        default:
158
            return qemuio_command_usage(&open_cmd);
159
        }
Devin Nakamura's avatar
Devin Nakamura committed
160 161 162 163 164
    }

    if (!readonly) {
        flags |= BDRV_O_RDWR;
    }
165

Max Reitz's avatar
Max Reitz committed
166 167 168 169 170
    if (optind == argc - 1) {
        return openfile(argv[optind], flags, growable, opts);
    } else if (optind == argc) {
        return openfile(NULL, flags, growable, opts);
    } else {
171
        return qemuio_command_usage(&open_cmd);
Devin Nakamura's avatar
Devin Nakamura committed
172
    }
173 174
}

Kevin Wolf's avatar
Kevin Wolf committed
175 176 177 178 179 180 181 182 183 184 185 186 187 188 189
static int quit_f(BlockDriverState *bs, int argc, char **argv)
{
    return 1;
}

static const cmdinfo_t quit_cmd = {
    .name       = "quit",
    .altname    = "q",
    .cfunc      = quit_f,
    .argmin     = -1,
    .argmax     = -1,
    .flags      = CMD_FLAG_GLOBAL,
    .oneline    = "exit the program",
};

190 191
static void usage(const char *name)
{
Devin Nakamura's avatar
Devin Nakamura committed
192
    printf(
Christoph Hellwig's avatar
Christoph Hellwig committed
193
"Usage: %s [-h] [-V] [-rsnm] [-c cmd] ... [file]\n"
194
"QEMU Disk exerciser\n"
195 196 197 198 199
"\n"
"  -c, --cmd            command to execute\n"
"  -r, --read-only      export read-only\n"
"  -s, --snapshot       use snapshot file\n"
"  -n, --nocache        disable host cache\n"
200
"  -g, --growable       allow file to grow (only applies to protocols)\n"
201
"  -m, --misalign       misalign allocations for O_DIRECT\n"
202
"  -k, --native-aio     use kernel AIO implementation (on Linux only)\n"
203
"  -t, --cache=MODE     use the given cache mode for the image\n"
204
"  -T, --trace FILE     enable trace events listed in the given file\n"
205 206 207
"  -h, --help           display this help and exit\n"
"  -V, --version        output version information and exit\n"
"\n",
Devin Nakamura's avatar
Devin Nakamura committed
208
    name);
209 210
}

211 212 213 214 215 216 217 218 219 220 221
static char *get_prompt(void)
{
    static char prompt[FILENAME_MAX + 2 /*"> "*/ + 1 /*"\0"*/ ];

    if (!prompt[0]) {
        snprintf(prompt, sizeof(prompt), "%s> ", progname);
    }

    return prompt;
}

Stefan Hajnoczi's avatar
Stefan Hajnoczi committed
222
static void readline_printf_func(void *opaque, const char *fmt, ...)
223
{
Stefan Hajnoczi's avatar
Stefan Hajnoczi committed
224 225 226 227
    va_list ap;
    va_start(ap, fmt);
    vprintf(fmt, ap);
    va_end(ap);
228
}
Stefan Hajnoczi's avatar
Stefan Hajnoczi committed
229 230

static void readline_flush_func(void *opaque)
231
{
Stefan Hajnoczi's avatar
Stefan Hajnoczi committed
232
    fflush(stdout);
233 234
}

Stefan Hajnoczi's avatar
Stefan Hajnoczi committed
235
static void readline_func(void *opaque, const char *str, void *readline_opaque)
236
{
Stefan Hajnoczi's avatar
Stefan Hajnoczi committed
237 238 239 240
    char **line = readline_opaque;
    *line = g_strdup(str);
}

241 242 243 244 245
static void completion_match(const char *cmd, void *opaque)
{
    readline_add_completion(readline_state, cmd);
}

Stefan Hajnoczi's avatar
Stefan Hajnoczi committed
246 247
static void readline_completion_func(void *opaque, const char *str)
{
248 249
    readline_set_completion_index(readline_state, strlen(str));
    qemuio_complete_command(str, completion_match, NULL);
Stefan Hajnoczi's avatar
Stefan Hajnoczi committed
250 251 252 253 254 255 256 257 258 259 260
}

static char *fetchline_readline(void)
{
    char *line = NULL;

    readline_start(readline_state, get_prompt(), 0, readline_func, &line);
    while (!line) {
        int ch = getchar();
        if (ch == EOF) {
            break;
261
        }
Stefan Hajnoczi's avatar
Stefan Hajnoczi committed
262
        readline_handle_byte(readline_state, ch);
263 264 265
    }
    return line;
}
Stefan Hajnoczi's avatar
Stefan Hajnoczi committed
266 267 268

#define MAXREADLINESZ 1024
static char *fetchline_fgets(void)
269 270 271 272 273 274 275 276 277 278 279 280 281 282 283
{
    char *p, *line = g_malloc(MAXREADLINESZ);

    if (!fgets(line, MAXREADLINESZ, stdin)) {
        g_free(line);
        return NULL;
    }

    p = line + strlen(line);
    if (p != line && p[-1] == '\n') {
        p[-1] = '\0';
    }

    return line;
}
Stefan Hajnoczi's avatar
Stefan Hajnoczi committed
284 285 286 287 288 289 290 291 292

static char *fetchline(void)
{
    if (readline_state) {
        return fetchline_readline();
    } else {
        return fetchline_fgets();
    }
}
293 294 295 296 297 298 299 300 301 302 303 304 305 306 307

static void prep_fetchline(void *opaque)
{
    int *fetchable = opaque;

    qemu_set_fd_handler(STDIN_FILENO, NULL, NULL, NULL);
    *fetchable= 1;
}

static void command_loop(void)
{
    int i, done = 0, fetchable = 0, prompted = 0;
    char *input;

    for (i = 0; !done && i < ncmdline; i++) {
Kevin Wolf's avatar
Kevin Wolf committed
308
        done = qemuio_command(qemuio_bs, cmdline[i]);
309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332
    }
    if (cmdline) {
        g_free(cmdline);
        return;
    }

    while (!done) {
        if (!prompted) {
            printf("%s", get_prompt());
            fflush(stdout);
            qemu_set_fd_handler(STDIN_FILENO, prep_fetchline, NULL, &fetchable);
            prompted = 1;
        }

        main_loop_wait(false);

        if (!fetchable) {
            continue;
        }

        input = fetchline();
        if (input == NULL) {
            break;
        }
Kevin Wolf's avatar
Kevin Wolf committed
333
        done = qemuio_command(qemuio_bs, input);
334 335 336 337 338 339 340 341 342 343 344 345 346 347
        g_free(input);

        prompted = 0;
        fetchable = 0;
    }
    qemu_set_fd_handler(STDIN_FILENO, NULL, NULL, NULL);
}

static void add_user_command(char *optarg)
{
    cmdline = g_realloc(cmdline, ++ncmdline * sizeof(char *));
    cmdline[ncmdline-1] = optarg;
}

Stefan Hajnoczi's avatar
Stefan Hajnoczi committed
348 349 350 351 352
static void reenable_tty_echo(void)
{
    qemu_set_tty_echo(STDIN_FILENO, true);
}

353 354
int main(int argc, char **argv)
{
Devin Nakamura's avatar
Devin Nakamura committed
355 356
    int readonly = 0;
    int growable = 0;
357
    const char *sopt = "hVc:d:rsnmgkt:T:";
Devin Nakamura's avatar
Devin Nakamura committed
358 359 360 361 362 363 364 365 366 367 368
    const struct option lopt[] = {
        { "help", 0, NULL, 'h' },
        { "version", 0, NULL, 'V' },
        { "offset", 1, NULL, 'o' },
        { "cmd", 1, NULL, 'c' },
        { "read-only", 0, NULL, 'r' },
        { "snapshot", 0, NULL, 's' },
        { "nocache", 0, NULL, 'n' },
        { "misalign", 0, NULL, 'm' },
        { "growable", 0, NULL, 'g' },
        { "native-aio", 0, NULL, 'k' },
369
        { "discard", 1, NULL, 'd' },
370
        { "cache", 1, NULL, 't' },
371
        { "trace", 1, NULL, 'T' },
Devin Nakamura's avatar
Devin Nakamura committed
372 373 374 375
        { NULL, 0, NULL, 0 }
    };
    int c;
    int opt_index = 0;
376
    int flags = BDRV_O_UNMAP;
Devin Nakamura's avatar
Devin Nakamura committed
377

378 379 380 381
#ifdef CONFIG_POSIX
    signal(SIGPIPE, SIG_IGN);
#endif

Devin Nakamura's avatar
Devin Nakamura committed
382 383 384 385 386 387 388 389 390 391
    progname = basename(argv[0]);

    while ((c = getopt_long(argc, argv, sopt, lopt, &opt_index)) != -1) {
        switch (c) {
        case 's':
            flags |= BDRV_O_SNAPSHOT;
            break;
        case 'n':
            flags |= BDRV_O_NOCACHE | BDRV_O_CACHE_WB;
            break;
392 393 394 395 396 397
        case 'd':
            if (bdrv_parse_discard_flags(optarg, &flags) < 0) {
                error_report("Invalid discard option: %s", optarg);
                exit(1);
            }
            break;
Devin Nakamura's avatar
Devin Nakamura committed
398 399 400 401 402 403 404
        case 'c':
            add_user_command(optarg);
            break;
        case 'r':
            readonly = 1;
            break;
        case 'm':
405
            qemuio_misalign = 1;
Devin Nakamura's avatar
Devin Nakamura committed
406 407 408 409 410 411 412
            break;
        case 'g':
            growable = 1;
            break;
        case 'k':
            flags |= BDRV_O_NATIVE_AIO;
            break;
413 414 415 416 417 418
        case 't':
            if (bdrv_parse_cache_flags(optarg, &flags) < 0) {
                error_report("Invalid cache option: %s", optarg);
                exit(1);
            }
            break;
419 420 421 422 423
        case 'T':
            if (!trace_backend_init(optarg, NULL)) {
                exit(1); /* error message will have been printed */
            }
            break;
Devin Nakamura's avatar
Devin Nakamura committed
424
        case 'V':
425
            printf("%s version %s\n", progname, QEMU_VERSION);
Devin Nakamura's avatar
Devin Nakamura committed
426 427 428 429 430 431 432
            exit(0);
        case 'h':
            usage(progname);
            exit(0);
        default:
            usage(progname);
            exit(1);
433
        }
Devin Nakamura's avatar
Devin Nakamura committed
434 435 436 437 438 439
    }

    if ((argc - optind) > 1) {
        usage(progname);
        exit(1);
    }
440

441
    qemu_init_main_loop();
442
    bdrv_init();
443

Devin Nakamura's avatar
Devin Nakamura committed
444
    /* initialize commands */
445 446 447
    qemuio_add_command(&quit_cmd);
    qemuio_add_command(&open_cmd);
    qemuio_add_command(&close_cmd);
Devin Nakamura's avatar
Devin Nakamura committed
448

Stefan Hajnoczi's avatar
Stefan Hajnoczi committed
449 450 451 452 453 454 455 456 457
    if (isatty(STDIN_FILENO)) {
        readline_state = readline_init(readline_printf_func,
                                       readline_flush_func,
                                       NULL,
                                       readline_completion_func);
        qemu_set_tty_echo(STDIN_FILENO, false);
        atexit(reenable_tty_echo);
    }

Devin Nakamura's avatar
Devin Nakamura committed
458 459 460 461 462 463
    /* open the device */
    if (!readonly) {
        flags |= BDRV_O_RDWR;
    }

    if ((argc - optind) == 1) {
464
        openfile(argv[optind], flags, growable, NULL);
Devin Nakamura's avatar
Devin Nakamura committed
465 466
    }
    command_loop();
467

Devin Nakamura's avatar
Devin Nakamura committed
468
    /*
469
     * Make sure all outstanding requests complete before the program exits.
Devin Nakamura's avatar
Devin Nakamura committed
470
     */
471
    bdrv_drain_all();
472

473
    if (qemuio_bs) {
Fam Zheng's avatar
Fam Zheng committed
474
        bdrv_unref(qemuio_bs);
Devin Nakamura's avatar
Devin Nakamura committed
475
    }
Stefan Hajnoczi's avatar
Stefan Hajnoczi committed
476
    g_free(readline_state);
Devin Nakamura's avatar
Devin Nakamura committed
477
    return 0;
478
}