[qemu.git] / tests / hd-geo-test.c

/*
 * Hard disk geometry test cases.
 *
 * Copyright (c) 2012 Red Hat Inc.
 *
 * Authors:
 *  Markus Armbruster <[email protected]>,
 *
 * This work is licensed under the terms of the GNU GPL, version 2 or later.
 * See the COPYING file in the top-level directory.
 */

/*
 * Covers only IDE and tests only CMOS contents.  Better than nothing.
 * Improvements welcome.
 */

#include "qemu/osdep.h"
#include "qemu-common.h"
#include "libqtest.h"

#define ARGV_SIZE 256

static char *create_test_img(int secs)
{
    char *template = strdup("/tmp/qtest.XXXXXX");
    int fd, ret;

    fd = mkstemp(template);
    g_assert(fd >= 0);
    ret = ftruncate(fd, (off_t)secs * 512);
    g_assert(ret == 0);
    close(fd);
    return template;
}

typedef struct {
    int cyls, heads, secs, trans;
} CHST;

typedef enum {
    mbr_blank, mbr_lba, mbr_chs,
    mbr_last
} MBRcontents;

typedef enum {
    /* order is relevant */
    backend_small, backend_large, backend_empty,
    backend_last
} Backend;

static const int img_secs[backend_last] = {
    [backend_small] = 61440,
    [backend_large] = 8388608,
    [backend_empty] = -1,
};

static const CHST hd_chst[backend_last][mbr_last] = {
    [backend_small] = {
        [mbr_blank] = { 60, 16, 63, 0 },
        [mbr_lba]   = { 60, 16, 63, 2 },
        [mbr_chs]   = { 60, 16, 63, 0 }
    },
    [backend_large] = {
        [mbr_blank] = { 8322, 16, 63, 1 },
        [mbr_lba]   = { 8322, 16, 63, 1 },
        [mbr_chs]   = { 8322, 16, 63, 0 }
    },
};

static char *img_file_name[backend_last];

static const CHST *cur_ide[4];

static bool is_hd(const CHST *expected_chst)
{
    return expected_chst && expected_chst->cyls;
}

static void test_cmos_byte(int reg, int expected)
{
    enum { cmos_base = 0x70 };
    int actual;

    outb(cmos_base + 0, reg);
    actual = inb(cmos_base + 1);
    g_assert(actual == expected);
}

static void test_cmos_bytes(int reg0, int n, uint8_t expected[])
{
    int i;

    for (i = 0; i < 9; i++) {
        test_cmos_byte(reg0 + i, expected[i]);
    }
}

static void test_cmos_disk_data(void)
{
    test_cmos_byte(0x12,
                   (is_hd(cur_ide[0]) ? 0xf0 : 0) |
                   (is_hd(cur_ide[1]) ? 0x0f : 0));
}

static void test_cmos_drive_cyl(int reg0, const CHST *expected_chst)
{
    if (is_hd(expected_chst)) {
        int c = expected_chst->cyls;
        int h = expected_chst->heads;
        int s = expected_chst->secs;
        uint8_t expected_bytes[9] = {
            c & 0xff, c >> 8, h, 0xff, 0xff, 0xc0 | ((h > 8) << 3),
            c & 0xff, c >> 8, s
        };
        test_cmos_bytes(reg0, 9, expected_bytes);
    } else {
        int i;

        for (i = 0; i < 9; i++) {
            test_cmos_byte(reg0 + i, 0);
        }
    }
}

static void test_cmos_drive1(void)
{
    test_cmos_byte(0x19, is_hd(cur_ide[0]) ? 47 : 0);
    test_cmos_drive_cyl(0x1b, cur_ide[0]);
}

static void test_cmos_drive2(void)
{
    test_cmos_byte(0x1a, is_hd(cur_ide[1]) ? 47 : 0);
    test_cmos_drive_cyl(0x24, cur_ide[1]);
}

static void test_cmos_disktransflag(void)
{
    int val, i;

    val = 0;
    for (i = 0; i < ARRAY_SIZE(cur_ide); i++) {
        if (is_hd(cur_ide[i])) {
            val |= cur_ide[i]->trans << (2 * i);
        }
    }
    test_cmos_byte(0x39, val);
}

static void test_cmos(void)
{
    test_cmos_disk_data();
    test_cmos_drive1();
    test_cmos_drive2();
    test_cmos_disktransflag();
}

static int append_arg(int argc, char *argv[], int argv_sz, char *arg)
{
    g_assert(argc + 1 < argv_sz);
    argv[argc++] = arg;
    argv[argc] = NULL;
    return argc;
}

static int setup_common(char *argv[], int argv_sz)
{
    memset(cur_ide, 0, sizeof(cur_ide));
    return append_arg(0, argv, argv_sz,
                      g_strdup("-nodefaults"));
}

static void setup_mbr(int img_idx, MBRcontents mbr)
{
    static const uint8_t part_lba[16] = {
        /* chs 0,1,1 (lba 63) to chs 0,127,63 (8001 sectors) */
        0x80, 1, 1, 0, 6, 127, 63, 0, 63, 0, 0, 0, 0x41, 0x1F, 0, 0,
    };
    static const uint8_t part_chs[16] = {
        /* chs 0,1,1 (lba 63) to chs 7,15,63 (8001 sectors) */
        0x80, 1, 1, 0, 6,  15, 63, 7, 63, 0, 0, 0, 0x41, 0x1F, 0, 0,
    };
    uint8_t buf[512];
    int fd, ret;

    memset(buf, 0, sizeof(buf));

    if (mbr != mbr_blank) {
        buf[0x1fe] = 0x55;
        buf[0x1ff] = 0xAA;
        memcpy(buf + 0x1BE, mbr == mbr_lba ? part_lba : part_chs, 16);
    }

    fd = open(img_file_name[img_idx], O_WRONLY);
    g_assert(fd >= 0);
    ret = write(fd, buf, sizeof(buf));
    g_assert(ret == sizeof(buf));
    close(fd);
}

static int setup_ide(int argc, char *argv[], int argv_sz,
                     int ide_idx, const char *dev, int img_idx,
                     MBRcontents mbr, const char *opts)
{
    char *s1, *s2, *s3;

    s1 = g_strdup_printf("-drive id=drive%d,if=%s",
                         ide_idx, dev ? "none" : "ide");
    s2 = dev ? g_strdup("") : g_strdup_printf(",index=%d", ide_idx);

    if (img_secs[img_idx] >= 0) {
        setup_mbr(img_idx, mbr);
        s3 = g_strdup_printf(",format=raw,file=%s", img_file_name[img_idx]);
    } else {
        s3 = g_strdup(",media=cdrom");
    }
    argc = append_arg(argc, argv, argv_sz,
                      g_strdup_printf("%s%s%s%s", s1, s2, s3, opts));
    g_free(s1);
    g_free(s2);
    g_free(s3);

    if (dev) {
        argc = append_arg(argc, argv, argv_sz,
                          g_strdup_printf("-device %s,drive=drive%d,"
                                          "bus=ide.%d,unit=%d",
                                          dev, ide_idx,
                                          ide_idx / 2, ide_idx % 2));
    }
    return argc;
}

/*
 * Test case: no IDE devices
 */
static void test_ide_none(void)
{
    char **argv = g_new0(char *, ARGV_SIZE);
    char *args;

    setup_common(argv, ARGV_SIZE);
    args = g_strjoinv(" ", argv);
    qtest_start(args);
    g_strfreev(argv);
    g_free(args);
    test_cmos();
    qtest_end();
}

static void test_ide_mbr(bool use_device, MBRcontents mbr)
{
    char **argv = g_new0(char *, ARGV_SIZE);
    char *args;
    int argc;
    Backend i;
    const char *dev;

    argc = setup_common(argv, ARGV_SIZE);
    for (i = 0; i < backend_last; i++) {
        cur_ide[i] = &hd_chst[i][mbr];
        dev = use_device ? (is_hd(cur_ide[i]) ? "ide-hd" : "ide-cd") : NULL;
        argc = setup_ide(argc, argv, ARGV_SIZE, i, dev, i, mbr, "");
    }
    args = g_strjoinv(" ", argv);
    qtest_start(args);
    g_strfreev(argv);
    g_free(args);
    test_cmos();
    qtest_end();
}

/*
 * Test case: IDE devices (if=ide) with blank MBRs
 */
static void test_ide_drive_mbr_blank(void)
{
    test_ide_mbr(false, mbr_blank);
}

/*
 * Test case: IDE devices (if=ide) with MBRs indicating LBA is in use
 */
static void test_ide_drive_mbr_lba(void)
{
    test_ide_mbr(false, mbr_lba);
}

/*
 * Test case: IDE devices (if=ide) with MBRs indicating CHS is in use
 */
static void test_ide_drive_mbr_chs(void)
{
    test_ide_mbr(false, mbr_chs);
}

/*
 * Test case: IDE devices (if=none) with blank MBRs
 */
static void test_ide_device_mbr_blank(void)
{
    test_ide_mbr(true, mbr_blank);
}

/*
 * Test case: IDE devices (if=none) with MBRs indicating LBA is in use
 */
static void test_ide_device_mbr_lba(void)
{
    test_ide_mbr(true, mbr_lba);
}

/*
 * Test case: IDE devices (if=none) with MBRs indicating CHS is in use
 */
static void test_ide_device_mbr_chs(void)
{
    test_ide_mbr(true, mbr_chs);
}

static void test_ide_drive_user(const char *dev, bool trans)
{
    char **argv = g_new0(char *, ARGV_SIZE);
    char *args, *opts;
    int argc;
    int secs = img_secs[backend_small];
    const CHST expected_chst = { secs / (4 * 32) , 4, 32, trans };

    argc = setup_common(argv, ARGV_SIZE);
    opts = g_strdup_printf("%s,%s%scyls=%d,heads=%d,secs=%d",
                           dev ?: "",
                           trans && dev ? "bios-chs-" : "",
                           trans ? "trans=lba," : "",
                           expected_chst.cyls, expected_chst.heads,
                           expected_chst.secs);
    cur_ide[0] = &expected_chst;
    argc = setup_ide(argc, argv, ARGV_SIZE,
                     0, dev ? opts : NULL, backend_small, mbr_chs,
                     dev ? "" : opts);
    g_free(opts);
    args = g_strjoinv(" ", argv);
    qtest_start(args);
    g_strfreev(argv);
    g_free(args);
    test_cmos();
    qtest_end();
}

/*
 * Test case: IDE device (if=ide) with explicit CHS
 */
static void test_ide_drive_user_chs(void)
{
    test_ide_drive_user(NULL, false);
}

/*
 * Test case: IDE device (if=ide) with explicit CHS and translation
 */
static void test_ide_drive_user_chst(void)
{
    test_ide_drive_user(NULL, true);
}

/*
 * Test case: IDE device (if=none) with explicit CHS
 */
static void test_ide_device_user_chs(void)
{
    test_ide_drive_user("ide-hd", false);
}

/*
 * Test case: IDE device (if=none) with explicit CHS and translation
 */
static void test_ide_device_user_chst(void)
{
    test_ide_drive_user("ide-hd", true);
}

/*
 * Test case: IDE devices (if=ide), but use index=0 for CD-ROM
 */
static void test_ide_drive_cd_0(void)
{
    char **argv = g_new0(char *, ARGV_SIZE);
    char *args;
    int argc, ide_idx;
    Backend i;

    argc = setup_common(argv, ARGV_SIZE);
    for (i = 0; i <= backend_empty; i++) {
        ide_idx = backend_empty - i;
        cur_ide[ide_idx] = &hd_chst[i][mbr_blank];
        argc = setup_ide(argc, argv, ARGV_SIZE,
                         ide_idx, NULL, i, mbr_blank, "");
    }
    args = g_strjoinv(" ", argv);
    qtest_start(args);
    g_strfreev(argv);
    g_free(args);
    test_cmos();
    qtest_end();
}

int main(int argc, char **argv)
{
    Backend i;
    int ret;

    g_test_init(&argc, &argv, NULL);

    for (i = 0; i < backend_last; i++) {
        if (img_secs[i] >= 0) {
            img_file_name[i] = create_test_img(img_secs[i]);
        } else {
            img_file_name[i] = NULL;
        }
    }

    qtest_add_func("hd-geo/ide/none", test_ide_none);
    qtest_add_func("hd-geo/ide/drive/mbr/blank", test_ide_drive_mbr_blank);
    qtest_add_func("hd-geo/ide/drive/mbr/lba", test_ide_drive_mbr_lba);
    qtest_add_func("hd-geo/ide/drive/mbr/chs", test_ide_drive_mbr_chs);
    qtest_add_func("hd-geo/ide/drive/user/chs", test_ide_drive_user_chs);
    qtest_add_func("hd-geo/ide/drive/user/chst", test_ide_drive_user_chst);
    qtest_add_func("hd-geo/ide/drive/cd_0", test_ide_drive_cd_0);
    qtest_add_func("hd-geo/ide/device/mbr/blank", test_ide_device_mbr_blank);
    qtest_add_func("hd-geo/ide/device/mbr/lba", test_ide_device_mbr_lba);
    qtest_add_func("hd-geo/ide/device/mbr/chs", test_ide_device_mbr_chs);
    qtest_add_func("hd-geo/ide/device/user/chs", test_ide_device_user_chs);
    qtest_add_func("hd-geo/ide/device/user/chst", test_ide_device_user_chst);

    ret = g_test_run();

    for (i = 0; i < backend_last; i++) {
        if (img_file_name[i]) {
            unlink(img_file_name[i]);
            free(img_file_name[i]);
        }
    }

    return ret;
}
Commit	Line	Data
0e8a8c8f MA	1	/*
	2	* Hard disk geometry test cases.
	3	*
	4	* Copyright (c) 2012 Red Hat Inc.
	5	*
	6	* Authors:
	7	* Markus Armbruster <[email protected]>,
	8	*
	9	* This work is licensed under the terms of the GNU GPL, version 2 or later.
	10	* See the COPYING file in the top-level directory.
	11	*/
	12
	13	/*
	14	* Covers only IDE and tests only CMOS contents. Better than nothing.
	15	* Improvements welcome.
	16	*/
	17
681c28a3	18	#include "qemu/osdep.h"
0e8a8c8f MA	19	#include "qemu-common.h"
	20	#include "libqtest.h"
	21
2c8f8696 MAL	22	#define ARGV_SIZE 256
2c8f8696 MAL	23
0e8a8c8f MA	24	static char *create_test_img(int secs)
	25	{
	26	char *template = strdup("/tmp/qtest.XXXXXX");
	27	int fd, ret;
	28
	29	fd = mkstemp(template);
	30	g_assert(fd >= 0);
	31	ret = ftruncate(fd, (off_t)secs * 512);
	32	g_assert(ret == 0);
	33	close(fd);
	34	return template;
	35	}
	36
	37	typedef struct {
	38	int cyls, heads, secs, trans;
	39	} CHST;
	40
	41	typedef enum {
	42	mbr_blank, mbr_lba, mbr_chs,
	43	mbr_last
	44	} MBRcontents;
	45
	46	typedef enum {
	47	/* order is relevant */
	48	backend_small, backend_large, backend_empty,
	49	backend_last
	50	} Backend;
	51
	52	static const int img_secs[backend_last] = {
	53	[backend_small] = 61440,
	54	[backend_large] = 8388608,
	55	[backend_empty] = -1,
	56	};
	57
	58	static const CHST hd_chst[backend_last][mbr_last] = {
	59	[backend_small] = {
	60	[mbr_blank] = { 60, 16, 63, 0 },
	61	[mbr_lba] = { 60, 16, 63, 2 },
	62	[mbr_chs] = { 60, 16, 63, 0 }
	63	},
	64	[backend_large] = {
	65	[mbr_blank] = { 8322, 16, 63, 1 },
	66	[mbr_lba] = { 8322, 16, 63, 1 },
	67	[mbr_chs] = { 8322, 16, 63, 0 }
	68	},
	69	};
	70
2c8f8696	71	static char *img_file_name[backend_last];
0e8a8c8f MA	72
	73	static const CHST *cur_ide[4];
	74
	75	static bool is_hd(const CHST *expected_chst)
	76	{
	77	return expected_chst && expected_chst->cyls;
	78	}
	79
	80	static void test_cmos_byte(int reg, int expected)
	81	{
	82	enum { cmos_base = 0x70 };
	83	int actual;
	84
	85	outb(cmos_base + 0, reg);
	86	actual = inb(cmos_base + 1);
	87	g_assert(actual == expected);
	88	}
	89
	90	static void test_cmos_bytes(int reg0, int n, uint8_t expected[])
	91	{
	92	int i;
	93
	94	for (i = 0; i < 9; i++) {
	95	test_cmos_byte(reg0 + i, expected[i]);
	96	}
	97	}
	98
	99	static void test_cmos_disk_data(void)
	100	{
	101	test_cmos_byte(0x12,
	102	(is_hd(cur_ide[0]) ? 0xf0 : 0) \|
	103	(is_hd(cur_ide[1]) ? 0x0f : 0));
	104	}
	105
	106	static void test_cmos_drive_cyl(int reg0, const CHST *expected_chst)
	107	{
	108	if (is_hd(expected_chst)) {
	109	int c = expected_chst->cyls;
	110	int h = expected_chst->heads;
	111	int s = expected_chst->secs;
	112	uint8_t expected_bytes[9] = {
	113	c & 0xff, c >> 8, h, 0xff, 0xff, 0xc0 \| ((h > 8) << 3),
	114	c & 0xff, c >> 8, s
	115	};
	116	test_cmos_bytes(reg0, 9, expected_bytes);
	117	} else {
	118	int i;
	119
	120	for (i = 0; i < 9; i++) {
	121	test_cmos_byte(reg0 + i, 0);
	122	}
	123	}
	124	}
	125
	126	static void test_cmos_drive1(void)
	127	{
	128	test_cmos_byte(0x19, is_hd(cur_ide[0]) ? 47 : 0);
	129	test_cmos_drive_cyl(0x1b, cur_ide[0]);
	130	}
	131
	132	static void test_cmos_drive2(void)
	133	{
	134	test_cmos_byte(0x1a, is_hd(cur_ide[1]) ? 47 : 0);
	135	test_cmos_drive_cyl(0x24, cur_ide[1]);
136	}
137
138	static void test_cmos_disktransflag(void)
139	{
140	int val, i;
141
142	val = 0;
143	for (i = 0; i < ARRAY_SIZE(cur_ide); i++) {
144	if (is_hd(cur_ide[i])) {
145	val \|= cur_ide[i]->trans << (2 * i);
146	}
147	}
148	test_cmos_byte(0x39, val);
149	}
150
151	static void test_cmos(void)
152	{
153	test_cmos_disk_data();
154	test_cmos_drive1();
155	test_cmos_drive2();
156	test_cmos_disktransflag();
157	}
158
159	static int append_arg(int argc, char argv[], int argv_sz, char arg)
160	{
161	g_assert(argc + 1 < argv_sz);
162	argv[argc++] = arg;
163	argv[argc] = NULL;
164	return argc;
165	}
166
167	static int setup_common(char *argv[], int argv_sz)
168	{
169	memset(cur_ide, 0, sizeof(cur_ide));
170	return append_arg(0, argv, argv_sz,
2ad645d2	171	g_strdup("-nodefaults"));
0e8a8c8f MA	172	}
	173
	174	static void setup_mbr(int img_idx, MBRcontents mbr)
	175	{
	176	static const uint8_t part_lba[16] = {
	177	/* chs 0,1,1 (lba 63) to chs 0,127,63 (8001 sectors) */
	178	0x80, 1, 1, 0, 6, 127, 63, 0, 63, 0, 0, 0, 0x41, 0x1F, 0, 0,
	179	};
	180	static const uint8_t part_chs[16] = {
	181	/* chs 0,1,1 (lba 63) to chs 7,15,63 (8001 sectors) */
	182	0x80, 1, 1, 0, 6, 15, 63, 7, 63, 0, 0, 0, 0x41, 0x1F, 0, 0,
	183	};
	184	uint8_t buf[512];
	185	int fd, ret;
	186
	187	memset(buf, 0, sizeof(buf));
	188
	189	if (mbr != mbr_blank) {
	190	buf[0x1fe] = 0x55;
	191	buf[0x1ff] = 0xAA;
	192	memcpy(buf + 0x1BE, mbr == mbr_lba ? part_lba : part_chs, 16);
	193	}
	194
	195	fd = open(img_file_name[img_idx], O_WRONLY);
	196	g_assert(fd >= 0);
	197	ret = write(fd, buf, sizeof(buf));
	198	g_assert(ret == sizeof(buf));
	199	close(fd);
	200	}
	201
	202	static int setup_ide(int argc, char *argv[], int argv_sz,
	203	int ide_idx, const char *dev, int img_idx,
	204	MBRcontents mbr, const char *opts)
	205	{
	206	char s1, s2, *s3;
	207
39c4ae94	208	s1 = g_strdup_printf("-drive id=drive%d,if=%s",
0e8a8c8f MA	209	ide_idx, dev ? "none" : "ide");
	210	s2 = dev ? g_strdup("") : g_strdup_printf(",index=%d", ide_idx);
	211
	212	if (img_secs[img_idx] >= 0) {
	213	setup_mbr(img_idx, mbr);
39c4ae94	214	s3 = g_strdup_printf(",format=raw,file=%s", img_file_name[img_idx]);
0e8a8c8f MA	215	} else {
	216	s3 = g_strdup(",media=cdrom");
	217	}
	218	argc = append_arg(argc, argv, argv_sz,
	219	g_strdup_printf("%s%s%s%s", s1, s2, s3, opts));
	220	g_free(s1);
	221	g_free(s2);
	222	g_free(s3);
	223
	224	if (dev) {
	225	argc = append_arg(argc, argv, argv_sz,
	226	g_strdup_printf("-device %s,drive=drive%d,"
	227	"bus=ide.%d,unit=%d",
	228	dev, ide_idx,
	229	ide_idx / 2, ide_idx % 2));
	230	}
	231	return argc;
	232	}
	233
	234	/*
	235	* Test case: no IDE devices
	236	*/
	237	static void test_ide_none(void)
	238	{
2c8f8696 MAL	239	char *argv = g_new0(char , ARGV_SIZE);
	240	char *args;
	241
	242	setup_common(argv, ARGV_SIZE);
	243	args = g_strjoinv(" ", argv);
	244	qtest_start(args);
	245	g_strfreev(argv);
	246	g_free(args);
0e8a8c8f	247	test_cmos();
1d9358e6	248	qtest_end();
0e8a8c8f MA	249	}
	250
	251	static void test_ide_mbr(bool use_device, MBRcontents mbr)
	252	{
2c8f8696 MAL	253	char *argv = g_new0(char , ARGV_SIZE);
2c8f8696 MAL	254	char *args;
0e8a8c8f MA	255	int argc;
	256	Backend i;
	257	const char *dev;
	258
2c8f8696	259	argc = setup_common(argv, ARGV_SIZE);
0e8a8c8f MA	260	for (i = 0; i < backend_last; i++) {
	261	cur_ide[i] = &hd_chst[i][mbr];
	262	dev = use_device ? (is_hd(cur_ide[i]) ? "ide-hd" : "ide-cd") : NULL;
2c8f8696	263	argc = setup_ide(argc, argv, ARGV_SIZE, i, dev, i, mbr, "");
0e8a8c8f	264	}
2c8f8696 MAL	265	args = g_strjoinv(" ", argv);
	266	qtest_start(args);
	267	g_strfreev(argv);
	268	g_free(args);
0e8a8c8f	269	test_cmos();
1d9358e6	270	qtest_end();
0e8a8c8f MA	271	}
	272
	273	/*
	274	* Test case: IDE devices (if=ide) with blank MBRs
	275	*/
	276	static void test_ide_drive_mbr_blank(void)
	277	{
	278	test_ide_mbr(false, mbr_blank);
	279	}
	280
	281	/*
	282	* Test case: IDE devices (if=ide) with MBRs indicating LBA is in use
	283	*/
	284	static void test_ide_drive_mbr_lba(void)
	285	{
	286	test_ide_mbr(false, mbr_lba);
	287	}
	288
	289	/*
	290	* Test case: IDE devices (if=ide) with MBRs indicating CHS is in use
	291	*/
	292	static void test_ide_drive_mbr_chs(void)
	293	{
	294	test_ide_mbr(false, mbr_chs);
	295	}
	296
	297	/*
	298	* Test case: IDE devices (if=none) with blank MBRs
	299	*/
	300	static void test_ide_device_mbr_blank(void)
	301	{
	302	test_ide_mbr(true, mbr_blank);
	303	}
	304
	305	/*
	306	* Test case: IDE devices (if=none) with MBRs indicating LBA is in use
	307	*/
	308	static void test_ide_device_mbr_lba(void)
	309	{
	310	test_ide_mbr(true, mbr_lba);
	311	}
	312
	313	/*
	314	* Test case: IDE devices (if=none) with MBRs indicating CHS is in use
	315	*/
	316	static void test_ide_device_mbr_chs(void)
	317	{
	318	test_ide_mbr(true, mbr_chs);
	319	}
	320
	321	static void test_ide_drive_user(const char *dev, bool trans)
	322	{
2c8f8696 MAL	323	char *argv = g_new0(char , ARGV_SIZE);
2c8f8696 MAL	324	char args, opts;
0e8a8c8f MA	325	int argc;
	326	int secs = img_secs[backend_small];
	327	const CHST expected_chst = { secs / (4 * 32) , 4, 32, trans };
	328
2c8f8696	329	argc = setup_common(argv, ARGV_SIZE);
856dcba2 MA	330	opts = g_strdup_printf("%s,%s%scyls=%d,heads=%d,secs=%d",
	331	dev ?: "",
	332	trans && dev ? "bios-chs-" : "",
	333	trans ? "trans=lba," : "",
0e8a8c8f	334	expected_chst.cyls, expected_chst.heads,
856dcba2	335	expected_chst.secs);
0e8a8c8f	336	cur_ide[0] = &expected_chst;
2c8f8696	337	argc = setup_ide(argc, argv, ARGV_SIZE,
856dcba2 MA	338	0, dev ? opts : NULL, backend_small, mbr_chs,
856dcba2 MA	339	dev ? "" : opts);
0e8a8c8f	340	g_free(opts);
2c8f8696 MAL	341	args = g_strjoinv(" ", argv);
	342	qtest_start(args);
	343	g_strfreev(argv);
	344	g_free(args);
0e8a8c8f	345	test_cmos();
1d9358e6	346	qtest_end();
0e8a8c8f MA	347	}
	348
	349	/*
	350	* Test case: IDE device (if=ide) with explicit CHS
	351	*/
	352	static void test_ide_drive_user_chs(void)
	353	{
	354	test_ide_drive_user(NULL, false);
	355	}
	356
	357	/*
	358	* Test case: IDE device (if=ide) with explicit CHS and translation
	359	*/
	360	static void test_ide_drive_user_chst(void)
	361	{
	362	test_ide_drive_user(NULL, true);
	363	}
	364
	365	/*
	366	* Test case: IDE device (if=none) with explicit CHS
	367	*/
	368	static void test_ide_device_user_chs(void)
	369	{
	370	test_ide_drive_user("ide-hd", false);
	371	}
	372
	373	/*
	374	* Test case: IDE device (if=none) with explicit CHS and translation
	375	*/
	376	static void test_ide_device_user_chst(void)
	377	{
	378	test_ide_drive_user("ide-hd", true);
	379	}
	380
4e4e6e31 MA	381	/*
	382	* Test case: IDE devices (if=ide), but use index=0 for CD-ROM
	383	*/
	384	static void test_ide_drive_cd_0(void)
	385	{
2c8f8696 MAL	386	char *argv = g_new0(char , ARGV_SIZE);
2c8f8696 MAL	387	char *args;
4e4e6e31 MA	388	int argc, ide_idx;
	389	Backend i;
	390
2c8f8696	391	argc = setup_common(argv, ARGV_SIZE);
4e4e6e31 MA	392	for (i = 0; i <= backend_empty; i++) {
	393	ide_idx = backend_empty - i;
	394	cur_ide[ide_idx] = &hd_chst[i][mbr_blank];
2c8f8696	395	argc = setup_ide(argc, argv, ARGV_SIZE,
4e4e6e31 MA	396	ide_idx, NULL, i, mbr_blank, "");
4e4e6e31 MA	397	}
2c8f8696 MAL	398	args = g_strjoinv(" ", argv);
	399	qtest_start(args);
	400	g_strfreev(argv);
	401	g_free(args);
4e4e6e31	402	test_cmos();
1d9358e6	403	qtest_end();
4e4e6e31 MA	404	}
4e4e6e31 MA	405
0e8a8c8f MA	406	int main(int argc, char **argv)
	407	{
	408	Backend i;
	409	int ret;
	410
	411	g_test_init(&argc, &argv, NULL);
	412
	413	for (i = 0; i < backend_last; i++) {
	414	if (img_secs[i] >= 0) {
	415	img_file_name[i] = create_test_img(img_secs[i]);
	416	} else {
	417	img_file_name[i] = NULL;
	418	}
	419	}
	420
	421	qtest_add_func("hd-geo/ide/none", test_ide_none);
	422	qtest_add_func("hd-geo/ide/drive/mbr/blank", test_ide_drive_mbr_blank);
	423	qtest_add_func("hd-geo/ide/drive/mbr/lba", test_ide_drive_mbr_lba);
	424	qtest_add_func("hd-geo/ide/drive/mbr/chs", test_ide_drive_mbr_chs);
	425	qtest_add_func("hd-geo/ide/drive/user/chs", test_ide_drive_user_chs);
	426	qtest_add_func("hd-geo/ide/drive/user/chst", test_ide_drive_user_chst);
4e4e6e31	427	qtest_add_func("hd-geo/ide/drive/cd_0", test_ide_drive_cd_0);
0e8a8c8f MA	428	qtest_add_func("hd-geo/ide/device/mbr/blank", test_ide_device_mbr_blank);
	429	qtest_add_func("hd-geo/ide/device/mbr/lba", test_ide_device_mbr_lba);
	430	qtest_add_func("hd-geo/ide/device/mbr/chs", test_ide_device_mbr_chs);
	431	qtest_add_func("hd-geo/ide/device/user/chs", test_ide_device_user_chs);
	432	qtest_add_func("hd-geo/ide/device/user/chst", test_ide_device_user_chst);
	433
	434	ret = g_test_run();
	435
	436	for (i = 0; i < backend_last; i++) {
0813cbf9 PM	437	if (img_file_name[i]) {
0813cbf9 PM	438	unlink(img_file_name[i]);
2c8f8696	439	free(img_file_name[i]);
0813cbf9	440	}
0e8a8c8f MA	441	}
	442
	443	return ret;
	444	}