1 /* libguestfs-test-tool
2 * Copyright (C) 2009 Red Hat Inc.
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
29 #include <sys/types.h>
37 #define _(str) dgettext(PACKAGE, (str))
38 #define N_(str) dgettext(PACKAGE, (str))
44 #define STREQ(a,b) (strcmp((a),(b)) == 0)
45 #define STRCASEEQ(a,b) (strcasecmp((a),(b)) == 0)
46 #define STRNEQ(a,b) (strcmp((a),(b)) != 0)
47 #define STRCASENEQ(a,b) (strcasecmp((a),(b)) != 0)
48 #define STREQLEN(a,b,n) (strncmp((a),(b),(n)) == 0)
49 #define STRCASEEQLEN(a,b,n) (strncasecmp((a),(b),(n)) == 0)
50 #define STRNEQLEN(a,b,n) (strncmp((a),(b),(n)) != 0)
51 #define STRCASENEQLEN(a,b,n) (strncasecmp((a),(b),(n)) != 0)
52 #define STRPREFIX(a,b) (strncmp((a),(b),strlen((b))) == 0)
54 #define DEFAULT_TIMEOUT 120
56 static const char *helper = DEFAULT_HELPER;
57 static int timeout = DEFAULT_TIMEOUT;
58 static char tmpf[] = "/tmp/libguestfs-test-tool-sda-XXXXXX";
59 static char isof[] = "/tmp/libguestfs-test-tool-iso-XXXXXX";
62 static void preruncheck (void);
63 static void make_files (void);
64 static void set_qemu (const char *path, int use_wrapper);
69 printf (_("libguestfs-test-tool: interactive test tool\n"
70 "Copyright (C) 2009 Red Hat Inc.\n"
72 " libguestfs-test-tool [--options]\n"
74 " --help Display usage\n"
75 " --helper libguestfs-test-tool-helper\n"
76 " Helper program (default: %s)\n"
77 " --qemudir dir Specify QEMU source directory\n"
78 " --qemu qemu Specify QEMU binary\n"
80 " -t n Set launch timeout (default: %d seconds)\n"
82 DEFAULT_HELPER, DEFAULT_TIMEOUT);
86 main (int argc, char *argv[])
88 setlocale (LC_ALL, "");
89 bindtextdomain (PACKAGE, LOCALEBASEDIR);
92 static const char *options = "t:?";
93 static const struct option long_options[] = {
94 { "help", 0, 0, '?' },
95 { "helper", 1, 0, 0 },
97 { "qemudir", 1, 0, 0 },
98 { "timeout", 1, 0, 't' },
103 extern char **environ;
105 struct guestfs_version *vers;
106 char *sfdisk_lines[] = { ",", NULL };
108 /* XXX This is wrong if the user renames the helper. */
109 char *helper_args[] = { "/iso/libguestfs-test-tool-helper", NULL };
112 c = getopt_long (argc, argv, options, long_options, &option_index);
116 case 0: /* options which are long only */
117 if (STREQ (long_options[option_index].name, "helper"))
119 else if (STREQ (long_options[option_index].name, "qemu"))
120 set_qemu (optarg, 0);
121 else if (STREQ (long_options[option_index].name, "qemudir"))
122 set_qemu (optarg, 1);
125 _("libguestfs-test-tool: unknown long option: %s (%d)\n"),
126 long_options[option_index].name, option_index);
132 if (sscanf (optarg, "%d", &timeout) != 1 || timeout < 0) {
134 _("libguestfs-test-tool: invalid timeout: %s\n"),
146 _("libguestfs-test-tool: unexpected command line option 0x%x\n"),
155 printf ("===== Test starts here =====\n");
157 /* Must set LIBGUESTFS_DEBUG=1 */
158 setenv ("LIBGUESTFS_DEBUG", "1", 1);
160 /* Print out any environment variables which may relate to this test. */
161 for (i = 0; environ[i] != NULL; ++i)
162 if (STREQLEN (environ[i], "LIBGUESTFS_", 11))
163 printf ("%s\n", environ[i]);
165 /* Create the handle and configure it. */
166 g = guestfs_create ();
169 _("libguestfs-test-tool: failed to create libguestfs handle\n"));
172 if (guestfs_add_drive (g, tmpf) == -1) {
174 _("libguestfs-test-tool: failed to add drive '%s'\n"),
178 if (guestfs_add_drive (g, isof) == -1) {
180 _("libguestfs-test-tool: failed to add drive '%s'\n"),
185 /* Print any version info etc. */
186 vers = guestfs_version (g);
188 fprintf (stderr, _("libguestfs-test-tool: guestfs_version failed\n"));
191 printf ("library version: %"PRIi64".%"PRIi64".%"PRIi64"%s\n",
192 vers->major, vers->minor, vers->release, vers->extra);
193 guestfs_free_version (vers);
195 printf ("guestfs_get_append: %s\n", guestfs_get_append (g) ? : "(null)");
196 printf ("guestfs_get_autosync: %d\n", guestfs_get_autosync (g));
197 printf ("guestfs_get_memsize: %d\n", guestfs_get_memsize (g));
198 printf ("guestfs_get_path: %s\n", guestfs_get_path (g));
199 printf ("guestfs_get_qemu: %s\n", guestfs_get_qemu (g));
200 printf ("guestfs_get_verbose: %d\n", guestfs_get_verbose (g));
202 /* Launch the guest handle. */
203 printf ("Launching appliance, timeout set to %d seconds.\n", timeout);
208 if (guestfs_launch (g) == -1) {
210 _("libguestfs-test-tool: failed to launch appliance\n"));
216 printf ("Guest launched OK.\n");
219 /* Create the filesystem and mount everything. */
220 if (guestfs_sfdiskM (g, "/dev/sda", sfdisk_lines) == -1) {
222 _("libguestfs-test-tool: failed to run sfdisk\n"));
226 if (guestfs_mkfs (g, "ext2", "/dev/sda1") == -1) {
228 _("libguestfs-test-tool: failed to mkfs.ext2\n"));
232 if (guestfs_mount_options (g, "", "/dev/sda1", "/") == -1) {
234 _("libguestfs-test-tool: failed to mount /dev/sda1 on /\n"));
238 if (guestfs_mkdir (g, "/iso") == -1) {
240 _("libguestfs-test-tool: failed to mkdir /iso\n"));
244 if (guestfs_mount (g, "/dev/sdb", "/iso") == -1) {
246 _("libguestfs-test-tool: failed to mount /dev/sdb on /iso\n"));
250 /* Let's now run some simple tests using the helper program. */
251 str = guestfs_command (g, helper_args);
254 _("libguestfs-test-tool: could not run helper program, or helper failed\n"));
259 printf ("===== TEST FINISHED OK =====\n");
263 static char qemuwrapper[] = "/tmp/libguestfs-test-tool-wrapper-XXXXXX";
266 cleanup_wrapper (void)
268 unlink (qemuwrapper);
271 /* Handle the --qemu and --qemudir parameters. use_wrapper is true
272 * in the --qemudir (source directory) case, where we have to create
273 * a wrapper shell script.
276 set_qemu (const char *path, int use_wrapper)
278 char buffer[PATH_MAX];
283 if (getenv ("LIBGUESTFS_QEMU")) {
285 _("LIBGUESTFS_QEMU environment variable is already set, so\n"
286 "--qemu/--qemudir options cannot be used.\n"));
291 if (access (path, X_OK) == -1) {
293 _("Binary '%s' does not exist or is not executable\n"),
298 setenv ("LIBGUESTFS_QEMU", path, 1);
302 /* This should be a source directory, so check it. */
303 snprintf (buffer, sizeof buffer, "%s/pc-bios", path);
304 if (stat (buffer, &statbuf) == -1 ||
305 !S_ISDIR (statbuf.st_mode)) {
307 _("%s: does not look like a qemu source directory\n"),
312 /* Make a wrapper script. */
313 fd = mkstemp (qemuwrapper);
315 perror (qemuwrapper);
321 fp = fdopen (fd, "w");
328 /* Select the right qemu binary for the wrapper script. */
330 fprintf (fp, "i386-softmmu/qemu");
332 fprintf (fp, host_cpu "-softmmu/qemu-system-" host_cpu);
335 fprintf (fp, " -L \"$qemudir\"/pc-bios \"$@\"\n");
339 setenv ("LIBGUESTFS_QEMU", qemuwrapper, 1);
340 atexit (cleanup_wrapper);
343 /* After getting the command line args, but before running
344 * anything, we check everything is in place to do the tests.
354 if (access (helper, R_OK) == -1) {
356 _("Test tool helper program 'libguestfs-test-tool-helper' is not\n"
357 "available. Expected to find it in '%s'\n"
359 "Use the --helper option to specify the location of this program.\n"),
364 snprintf (cmd, sizeof cmd, "file '%s'", helper);
365 fp = popen (cmd, "r");
370 r = fread (buffer, 1, sizeof buffer - 1, fp);
372 fprintf (stderr, _("command failed: %s"), cmd);
378 if (strstr (buffer, "statically linked") == NULL) {
380 _("Test tool helper program %s\n"
381 "is not statically linked. This is a build error when this test tool\n"
389 cleanup_tmpfiles (void)
401 /* Make the ISO which will contain the helper program. */
409 snprintf (cmd, sizeof cmd, "mkisofs -quiet -rJT -o '%s' '%s'",
412 if (r == -1 || WEXITSTATUS(r) != 0) {
414 _("mkisofs command failed: %s\n"), cmd);
418 /* Allocate the sparse file for /dev/sda. */
426 if (lseek (fd, 100 * 1024 * 1024 - 1, SEEK_SET) == -1) {
434 if (write (fd, "\0", 1) == -1) {
444 atexit (cleanup_tmpfiles); /* Removes tmpf and isof. */