Improve error message when appliance doesn't match library.
[libguestfs.git] / src / generator.ml
1 #!/usr/bin/env ocaml
2 (* libguestfs
3  * Copyright (C) 2009 Red Hat Inc.
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License as published by
7  * the Free Software Foundation; either version 2 of the License, or
8  * (at your option) any later version.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, write to the Free Software
17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18  *)
19
20 (* This script generates a large amount of code and documentation for
21  * all the daemon actions.
22  *
23  * To add a new action there are only two files you need to change,
24  * this one to describe the interface (see the big table below), and
25  * daemon/<somefile>.c to write the implementation.
26  *
27  * After editing this file, run it (./src/generator.ml) to regenerate
28  * all the output files.
29  *
30  * IMPORTANT: This script should NOT print any warnings.  If it prints
31  * warnings, you should treat them as errors.
32  * [Need to add -warn-error to ocaml command line]
33  *)
34
35 #load "unix.cma";;
36 #load "str.cma";;
37
38 open Printf
39
40 type style = ret * args
41 and ret =
42     (* "RErr" as a return value means an int used as a simple error
43      * indication, ie. 0 or -1.
44      *)
45   | RErr
46     (* "RInt" as a return value means an int which is -1 for error
47      * or any value >= 0 on success.  Only use this for smallish
48      * positive ints (0 <= i < 2^30).
49      *)
50   | RInt of string
51     (* "RInt64" is the same as RInt, but is guaranteed to be able
52      * to return a full 64 bit value, _except_ that -1 means error
53      * (so -1 cannot be a valid, non-error return value).
54      *)
55   | RInt64 of string
56     (* "RBool" is a bool return value which can be true/false or
57      * -1 for error.
58      *)
59   | RBool of string
60     (* "RConstString" is a string that refers to a constant value.
61      * Try to avoid using this.  In particular you cannot use this
62      * for values returned from the daemon, because there is no
63      * thread-safe way to return them in the C API.
64      *)
65   | RConstString of string
66     (* "RString" and "RStringList" are caller-frees. *)
67   | RString of string
68   | RStringList of string
69     (* Some limited tuples are possible: *)
70   | RIntBool of string * string
71     (* LVM PVs, VGs and LVs. *)
72   | RPVList of string
73   | RVGList of string
74   | RLVList of string
75     (* Stat buffers. *)
76   | RStat of string
77   | RStatVFS of string
78     (* Key-value pairs of untyped strings.  Turns into a hashtable or
79      * dictionary in languages which support it.  DON'T use this as a
80      * general "bucket" for results.  Prefer a stronger typed return
81      * value if one is available, or write a custom struct.  Don't use
82      * this if the list could potentially be very long, since it is
83      * inefficient.  Keys should be unique.  NULLs are not permitted.
84      *)
85   | RHashtable of string
86
87 and args = argt list    (* Function parameters, guestfs handle is implicit. *)
88
89     (* Note in future we should allow a "variable args" parameter as
90      * the final parameter, to allow commands like
91      *   chmod mode file [file(s)...]
92      * This is not implemented yet, but many commands (such as chmod)
93      * are currently defined with the argument order keeping this future
94      * possibility in mind.
95      *)
96 and argt =
97   | String of string    (* const char *name, cannot be NULL *)
98   | OptString of string (* const char *name, may be NULL *)
99   | StringList of string(* list of strings (each string cannot be NULL) *)
100   | Bool of string      (* boolean *)
101   | Int of string       (* int (smallish ints, signed, <= 31 bits) *)
102     (* These are treated as filenames (simple string parameters) in
103      * the C API and bindings.  But in the RPC protocol, we transfer
104      * the actual file content up to or down from the daemon.
105      * FileIn: local machine -> daemon (in request)
106      * FileOut: daemon -> local machine (in reply)
107      * In guestfish (only), the special name "-" means read from
108      * stdin or write to stdout.
109      *)
110   | FileIn of string
111   | FileOut of string
112
113 type flags =
114   | ProtocolLimitWarning  (* display warning about protocol size limits *)
115   | DangerWillRobinson    (* flags particularly dangerous commands *)
116   | FishAlias of string   (* provide an alias for this cmd in guestfish *)
117   | FishAction of string  (* call this function in guestfish *)
118   | NotInFish             (* do not export via guestfish *)
119   | NotInDocs             (* do not add this function to documentation *)
120
121 let protocol_limit_warning =
122   "Because of the message protocol, there is a transfer limit 
123 of somewhere between 2MB and 4MB.  To transfer large files you should use
124 FTP."
125
126 let danger_will_robinson =
127   "B<This command is dangerous.  Without careful use you
128 can easily destroy all your data>."
129
130 (* You can supply zero or as many tests as you want per API call.
131  *
132  * Note that the test environment has 3 block devices, of size 500MB,
133  * 50MB and 10MB (respectively /dev/sda, /dev/sdb, /dev/sdc), and
134  * a fourth squashfs block device with some known files on it (/dev/sdd).
135  *
136  * Note for partitioning purposes, the 500MB device has 63 cylinders.
137  *
138  * The squashfs block device (/dev/sdd) comes from images/test.sqsh.
139  *
140  * To be able to run the tests in a reasonable amount of time,
141  * the virtual machine and block devices are reused between tests.
142  * So don't try testing kill_subprocess :-x
143  *
144  * Between each test we blockdev-setrw, umount-all, lvm-remove-all.
145  *
146  * Don't assume anything about the previous contents of the block
147  * devices.  Use 'Init*' to create some initial scenarios.
148  *
149  * You can add a prerequisite clause to any individual test.  This
150  * is a run-time check, which, if it fails, causes the test to be
151  * skipped.  Useful if testing a command which might not work on
152  * all variations of libguestfs builds.  A test that has prerequisite
153  * of 'Always' is run unconditionally.
154  *
155  * In addition, packagers can skip individual tests by setting the
156  * environment variables:     eg:
157  *   SKIP_TEST_<CMD>_<NUM>=1  SKIP_TEST_COMMAND_3=1  (skips test #3 of command)
158  *   SKIP_TEST_<CMD>=1        SKIP_TEST_ZEROFREE=1   (skips all zerofree tests)
159  *)
160 type tests = (test_init * test_prereq * test) list
161 and test =
162     (* Run the command sequence and just expect nothing to fail. *)
163   | TestRun of seq
164     (* Run the command sequence and expect the output of the final
165      * command to be the string.
166      *)
167   | TestOutput of seq * string
168     (* Run the command sequence and expect the output of the final
169      * command to be the list of strings.
170      *)
171   | TestOutputList of seq * string list
172     (* Run the command sequence and expect the output of the final
173      * command to be the list of block devices (could be either
174      * "/dev/sd.." or "/dev/hd.." form - we don't check the 5th
175      * character of each string).
176      *)
177   | TestOutputListOfDevices of seq * string list
178     (* Run the command sequence and expect the output of the final
179      * command to be the integer.
180      *)
181   | TestOutputInt of seq * int
182     (* Run the command sequence and expect the output of the final
183      * command to be a true value (!= 0 or != NULL).
184      *)
185   | TestOutputTrue of seq
186     (* Run the command sequence and expect the output of the final
187      * command to be a false value (== 0 or == NULL, but not an error).
188      *)
189   | TestOutputFalse of seq
190     (* Run the command sequence and expect the output of the final
191      * command to be a list of the given length (but don't care about
192      * content).
193      *)
194   | TestOutputLength of seq * int
195     (* Run the command sequence and expect the output of the final
196      * command to be a structure.
197      *)
198   | TestOutputStruct of seq * test_field_compare list
199     (* Run the command sequence and expect the final command (only)
200      * to fail.
201      *)
202   | TestLastFail of seq
203
204 and test_field_compare =
205   | CompareWithInt of string * int
206   | CompareWithString of string * string
207   | CompareFieldsIntEq of string * string
208   | CompareFieldsStrEq of string * string
209
210 (* Test prerequisites. *)
211 and test_prereq =
212     (* Test always runs. *)
213   | Always
214     (* Test is currently disabled - eg. it fails, or it tests some
215      * unimplemented feature.
216      *)
217   | Disabled
218     (* 'string' is some C code (a function body) that should return
219      * true or false.  The test will run if the code returns true.
220      *)
221   | If of string
222     (* As for 'If' but the test runs _unless_ the code returns true. *)
223   | Unless of string
224
225 (* Some initial scenarios for testing. *)
226 and test_init =
227     (* Do nothing, block devices could contain random stuff including
228      * LVM PVs, and some filesystems might be mounted.  This is usually
229      * a bad idea.
230      *)
231   | InitNone
232     (* Block devices are empty and no filesystems are mounted. *)
233   | InitEmpty
234     (* /dev/sda contains a single partition /dev/sda1, which is formatted
235      * as ext2, empty [except for lost+found] and mounted on /.
236      * /dev/sdb and /dev/sdc may have random content.
237      * No LVM.
238      *)
239   | InitBasicFS
240     (* /dev/sda:
241      *   /dev/sda1 (is a PV):
242      *     /dev/VG/LV (size 8MB):
243      *       formatted as ext2, empty [except for lost+found], mounted on /
244      * /dev/sdb and /dev/sdc may have random content.
245      *)
246   | InitBasicFSonLVM
247
248 (* Sequence of commands for testing. *)
249 and seq = cmd list
250 and cmd = string list
251
252 (* Note about long descriptions: When referring to another
253  * action, use the format C<guestfs_other> (ie. the full name of
254  * the C function).  This will be replaced as appropriate in other
255  * language bindings.
256  *
257  * Apart from that, long descriptions are just perldoc paragraphs.
258  *)
259
260 (* These test functions are used in the language binding tests. *)
261
262 let test_all_args = [
263   String "str";
264   OptString "optstr";
265   StringList "strlist";
266   Bool "b";
267   Int "integer";
268   FileIn "filein";
269   FileOut "fileout";
270 ]
271
272 let test_all_rets = [
273   (* except for RErr, which is tested thoroughly elsewhere *)
274   "test0rint",         RInt "valout";
275   "test0rint64",       RInt64 "valout";
276   "test0rbool",        RBool "valout";
277   "test0rconststring", RConstString "valout";
278   "test0rstring",      RString "valout";
279   "test0rstringlist",  RStringList "valout";
280   "test0rintbool",     RIntBool ("valout", "valout");
281   "test0rpvlist",      RPVList "valout";
282   "test0rvglist",      RVGList "valout";
283   "test0rlvlist",      RLVList "valout";
284   "test0rstat",        RStat "valout";
285   "test0rstatvfs",     RStatVFS "valout";
286   "test0rhashtable",   RHashtable "valout";
287 ]
288
289 let test_functions = [
290   ("test0", (RErr, test_all_args), -1, [NotInFish; NotInDocs],
291    [],
292    "internal test function - do not use",
293    "\
294 This is an internal test function which is used to test whether
295 the automatically generated bindings can handle every possible
296 parameter type correctly.
297
298 It echos the contents of each parameter to stdout.
299
300 You probably don't want to call this function.");
301 ] @ List.flatten (
302   List.map (
303     fun (name, ret) ->
304       [(name, (ret, [String "val"]), -1, [NotInFish; NotInDocs],
305         [],
306         "internal test function - do not use",
307         "\
308 This is an internal test function which is used to test whether
309 the automatically generated bindings can handle every possible
310 return type correctly.
311
312 It converts string C<val> to the return type.
313
314 You probably don't want to call this function.");
315        (name ^ "err", (ret, []), -1, [NotInFish; NotInDocs],
316         [],
317         "internal test function - do not use",
318         "\
319 This is an internal test function which is used to test whether
320 the automatically generated bindings can handle every possible
321 return type correctly.
322
323 This function always returns an error.
324
325 You probably don't want to call this function.")]
326   ) test_all_rets
327 )
328
329 (* non_daemon_functions are any functions which don't get processed
330  * in the daemon, eg. functions for setting and getting local
331  * configuration values.
332  *)
333
334 let non_daemon_functions = test_functions @ [
335   ("launch", (RErr, []), -1, [FishAlias "run"; FishAction "launch"],
336    [],
337    "launch the qemu subprocess",
338    "\
339 Internally libguestfs is implemented by running a virtual machine
340 using L<qemu(1)>.
341
342 You should call this after configuring the handle
343 (eg. adding drives) but before performing any actions.");
344
345   ("wait_ready", (RErr, []), -1, [NotInFish],
346    [],
347    "wait until the qemu subprocess launches",
348    "\
349 Internally libguestfs is implemented by running a virtual machine
350 using L<qemu(1)>.
351
352 You should call this after C<guestfs_launch> to wait for the launch
353 to complete.");
354
355   ("kill_subprocess", (RErr, []), -1, [],
356    [],
357    "kill the qemu subprocess",
358    "\
359 This kills the qemu subprocess.  You should never need to call this.");
360
361   ("add_drive", (RErr, [String "filename"]), -1, [FishAlias "add"],
362    [],
363    "add an image to examine or modify",
364    "\
365 This function adds a virtual machine disk image C<filename> to the
366 guest.  The first time you call this function, the disk appears as IDE
367 disk 0 (C</dev/sda>) in the guest, the second time as C</dev/sdb>, and
368 so on.
369
370 You don't necessarily need to be root when using libguestfs.  However
371 you obviously do need sufficient permissions to access the filename
372 for whatever operations you want to perform (ie. read access if you
373 just want to read the image or write access if you want to modify the
374 image).
375
376 This is equivalent to the qemu parameter C<-drive file=filename>.
377
378 Note that this call checks for the existence of C<filename>.  This
379 stops you from specifying other types of drive which are supported
380 by qemu such as C<nbd:> and C<http:> URLs.  To specify those, use
381 the general C<guestfs_config> call instead.");
382
383   ("add_cdrom", (RErr, [String "filename"]), -1, [FishAlias "cdrom"],
384    [],
385    "add a CD-ROM disk image to examine",
386    "\
387 This function adds a virtual CD-ROM disk image to the guest.
388
389 This is equivalent to the qemu parameter C<-cdrom filename>.
390
391 Note that this call checks for the existence of C<filename>.  This
392 stops you from specifying other types of drive which are supported
393 by qemu such as C<nbd:> and C<http:> URLs.  To specify those, use
394 the general C<guestfs_config> call instead.");
395
396   ("add_drive_ro", (RErr, [String "filename"]), -1, [FishAlias "add-ro"],
397    [],
398    "add a drive in snapshot mode (read-only)",
399    "\
400 This adds a drive in snapshot mode, making it effectively
401 read-only.
402
403 Note that writes to the device are allowed, and will be seen for
404 the duration of the guestfs handle, but they are written
405 to a temporary file which is discarded as soon as the guestfs
406 handle is closed.  We don't currently have any method to enable
407 changes to be committed, although qemu can support this.
408
409 This is equivalent to the qemu parameter
410 C<-drive file=filename,snapshot=on>.
411
412 Note that this call checks for the existence of C<filename>.  This
413 stops you from specifying other types of drive which are supported
414 by qemu such as C<nbd:> and C<http:> URLs.  To specify those, use
415 the general C<guestfs_config> call instead.");
416
417   ("config", (RErr, [String "qemuparam"; OptString "qemuvalue"]), -1, [],
418    [],
419    "add qemu parameters",
420    "\
421 This can be used to add arbitrary qemu command line parameters
422 of the form C<-param value>.  Actually it's not quite arbitrary - we
423 prevent you from setting some parameters which would interfere with
424 parameters that we use.
425
426 The first character of C<param> string must be a C<-> (dash).
427
428 C<value> can be NULL.");
429
430   ("set_qemu", (RErr, [String "qemu"]), -1, [FishAlias "qemu"],
431    [],
432    "set the qemu binary",
433    "\
434 Set the qemu binary that we will use.
435
436 The default is chosen when the library was compiled by the
437 configure script.
438
439 You can also override this by setting the C<LIBGUESTFS_QEMU>
440 environment variable.
441
442 Setting C<qemu> to C<NULL> restores the default qemu binary.");
443
444   ("get_qemu", (RConstString "qemu", []), -1, [],
445    [],
446    "get the qemu binary",
447    "\
448 Return the current qemu binary.
449
450 This is always non-NULL.  If it wasn't set already, then this will
451 return the default qemu binary name.");
452
453   ("set_path", (RErr, [String "path"]), -1, [FishAlias "path"],
454    [],
455    "set the search path",
456    "\
457 Set the path that libguestfs searches for kernel and initrd.img.
458
459 The default is C<$libdir/guestfs> unless overridden by setting
460 C<LIBGUESTFS_PATH> environment variable.
461
462 Setting C<path> to C<NULL> restores the default path.");
463
464   ("get_path", (RConstString "path", []), -1, [],
465    [],
466    "get the search path",
467    "\
468 Return the current search path.
469
470 This is always non-NULL.  If it wasn't set already, then this will
471 return the default path.");
472
473   ("set_append", (RErr, [String "append"]), -1, [FishAlias "append"],
474    [],
475    "add options to kernel command line",
476    "\
477 This function is used to add additional options to the
478 guest kernel command line.
479
480 The default is C<NULL> unless overridden by setting
481 C<LIBGUESTFS_APPEND> environment variable.
482
483 Setting C<append> to C<NULL> means I<no> additional options
484 are passed (libguestfs always adds a few of its own).");
485
486   ("get_append", (RConstString "append", []), -1, [],
487    [],
488    "get the additional kernel options",
489    "\
490 Return the additional kernel options which are added to the
491 guest kernel command line.
492
493 If C<NULL> then no options are added.");
494
495   ("set_autosync", (RErr, [Bool "autosync"]), -1, [FishAlias "autosync"],
496    [],
497    "set autosync mode",
498    "\
499 If C<autosync> is true, this enables autosync.  Libguestfs will make a
500 best effort attempt to run C<guestfs_umount_all> followed by
501 C<guestfs_sync> when the handle is closed
502 (also if the program exits without closing handles).
503
504 This is disabled by default (except in guestfish where it is
505 enabled by default).");
506
507   ("get_autosync", (RBool "autosync", []), -1, [],
508    [],
509    "get autosync mode",
510    "\
511 Get the autosync flag.");
512
513   ("set_verbose", (RErr, [Bool "verbose"]), -1, [FishAlias "verbose"],
514    [],
515    "set verbose mode",
516    "\
517 If C<verbose> is true, this turns on verbose messages (to C<stderr>).
518
519 Verbose messages are disabled unless the environment variable
520 C<LIBGUESTFS_DEBUG> is defined and set to C<1>.");
521
522   ("get_verbose", (RBool "verbose", []), -1, [],
523    [],
524    "get verbose mode",
525    "\
526 This returns the verbose messages flag.");
527
528   ("is_ready", (RBool "ready", []), -1, [],
529    [],
530    "is ready to accept commands",
531    "\
532 This returns true iff this handle is ready to accept commands
533 (in the C<READY> state).
534
535 For more information on states, see L<guestfs(3)>.");
536
537   ("is_config", (RBool "config", []), -1, [],
538    [],
539    "is in configuration state",
540    "\
541 This returns true iff this handle is being configured
542 (in the C<CONFIG> state).
543
544 For more information on states, see L<guestfs(3)>.");
545
546   ("is_launching", (RBool "launching", []), -1, [],
547    [],
548    "is launching subprocess",
549    "\
550 This returns true iff this handle is launching the subprocess
551 (in the C<LAUNCHING> state).
552
553 For more information on states, see L<guestfs(3)>.");
554
555   ("is_busy", (RBool "busy", []), -1, [],
556    [],
557    "is busy processing a command",
558    "\
559 This returns true iff this handle is busy processing a command
560 (in the C<BUSY> state).
561
562 For more information on states, see L<guestfs(3)>.");
563
564   ("get_state", (RInt "state", []), -1, [],
565    [],
566    "get the current state",
567    "\
568 This returns the current state as an opaque integer.  This is
569 only useful for printing debug and internal error messages.
570
571 For more information on states, see L<guestfs(3)>.");
572
573   ("set_busy", (RErr, []), -1, [NotInFish],
574    [],
575    "set state to busy",
576    "\
577 This sets the state to C<BUSY>.  This is only used when implementing
578 actions using the low-level API.
579
580 For more information on states, see L<guestfs(3)>.");
581
582   ("set_ready", (RErr, []), -1, [NotInFish],
583    [],
584    "set state to ready",
585    "\
586 This sets the state to C<READY>.  This is only used when implementing
587 actions using the low-level API.
588
589 For more information on states, see L<guestfs(3)>.");
590
591   ("end_busy", (RErr, []), -1, [NotInFish],
592    [],
593    "leave the busy state",
594    "\
595 This sets the state to C<READY>, or if in C<CONFIG> then it leaves the
596 state as is.  This is only used when implementing
597 actions using the low-level API.
598
599 For more information on states, see L<guestfs(3)>.");
600
601 ]
602
603 (* daemon_functions are any functions which cause some action
604  * to take place in the daemon.
605  *)
606
607 let daemon_functions = [
608   ("mount", (RErr, [String "device"; String "mountpoint"]), 1, [],
609    [InitEmpty, Always, TestOutput (
610       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
611        ["mkfs"; "ext2"; "/dev/sda1"];
612        ["mount"; "/dev/sda1"; "/"];
613        ["write_file"; "/new"; "new file contents"; "0"];
614        ["cat"; "/new"]], "new file contents")],
615    "mount a guest disk at a position in the filesystem",
616    "\
617 Mount a guest disk at a position in the filesystem.  Block devices
618 are named C</dev/sda>, C</dev/sdb> and so on, as they were added to
619 the guest.  If those block devices contain partitions, they will have
620 the usual names (eg. C</dev/sda1>).  Also LVM C</dev/VG/LV>-style
621 names can be used.
622
623 The rules are the same as for L<mount(2)>:  A filesystem must
624 first be mounted on C</> before others can be mounted.  Other
625 filesystems can only be mounted on directories which already
626 exist.
627
628 The mounted filesystem is writable, if we have sufficient permissions
629 on the underlying device.
630
631 The filesystem options C<sync> and C<noatime> are set with this
632 call, in order to improve reliability.");
633
634   ("sync", (RErr, []), 2, [],
635    [ InitEmpty, Always, TestRun [["sync"]]],
636    "sync disks, writes are flushed through to the disk image",
637    "\
638 This syncs the disk, so that any writes are flushed through to the
639 underlying disk image.
640
641 You should always call this if you have modified a disk image, before
642 closing the handle.");
643
644   ("touch", (RErr, [String "path"]), 3, [],
645    [InitBasicFS, Always, TestOutputTrue (
646       [["touch"; "/new"];
647        ["exists"; "/new"]])],
648    "update file timestamps or create a new file",
649    "\
650 Touch acts like the L<touch(1)> command.  It can be used to
651 update the timestamps on a file, or, if the file does not exist,
652 to create a new zero-length file.");
653
654   ("cat", (RString "content", [String "path"]), 4, [ProtocolLimitWarning],
655    [InitBasicFS, Always, TestOutput (
656       [["write_file"; "/new"; "new file contents"; "0"];
657        ["cat"; "/new"]], "new file contents")],
658    "list the contents of a file",
659    "\
660 Return the contents of the file named C<path>.
661
662 Note that this function cannot correctly handle binary files
663 (specifically, files containing C<\\0> character which is treated
664 as end of string).  For those you need to use the C<guestfs_download>
665 function which has a more complex interface.");
666
667   ("ll", (RString "listing", [String "directory"]), 5, [],
668    [], (* XXX Tricky to test because it depends on the exact format
669         * of the 'ls -l' command, which changes between F10 and F11.
670         *)
671    "list the files in a directory (long format)",
672    "\
673 List the files in C<directory> (relative to the root directory,
674 there is no cwd) in the format of 'ls -la'.
675
676 This command is mostly useful for interactive sessions.  It
677 is I<not> intended that you try to parse the output string.");
678
679   ("ls", (RStringList "listing", [String "directory"]), 6, [],
680    [InitBasicFS, Always, TestOutputList (
681       [["touch"; "/new"];
682        ["touch"; "/newer"];
683        ["touch"; "/newest"];
684        ["ls"; "/"]], ["lost+found"; "new"; "newer"; "newest"])],
685    "list the files in a directory",
686    "\
687 List the files in C<directory> (relative to the root directory,
688 there is no cwd).  The '.' and '..' entries are not returned, but
689 hidden files are shown.
690
691 This command is mostly useful for interactive sessions.  Programs
692 should probably use C<guestfs_readdir> instead.");
693
694   ("list_devices", (RStringList "devices", []), 7, [],
695    [InitEmpty, Always, TestOutputListOfDevices (
696       [["list_devices"]], ["/dev/sda"; "/dev/sdb"; "/dev/sdc"; "/dev/sdd"])],
697    "list the block devices",
698    "\
699 List all the block devices.
700
701 The full block device names are returned, eg. C</dev/sda>");
702
703   ("list_partitions", (RStringList "partitions", []), 8, [],
704    [InitBasicFS, Always, TestOutputListOfDevices (
705       [["list_partitions"]], ["/dev/sda1"]);
706     InitEmpty, Always, TestOutputListOfDevices (
707       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ",10 ,20 ,"];
708        ["list_partitions"]], ["/dev/sda1"; "/dev/sda2"; "/dev/sda3"])],
709    "list the partitions",
710    "\
711 List all the partitions detected on all block devices.
712
713 The full partition device names are returned, eg. C</dev/sda1>
714
715 This does not return logical volumes.  For that you will need to
716 call C<guestfs_lvs>.");
717
718   ("pvs", (RStringList "physvols", []), 9, [],
719    [InitBasicFSonLVM, Always, TestOutputListOfDevices (
720       [["pvs"]], ["/dev/sda1"]);
721     InitEmpty, Always, TestOutputListOfDevices (
722       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ",10 ,20 ,"];
723        ["pvcreate"; "/dev/sda1"];
724        ["pvcreate"; "/dev/sda2"];
725        ["pvcreate"; "/dev/sda3"];
726        ["pvs"]], ["/dev/sda1"; "/dev/sda2"; "/dev/sda3"])],
727    "list the LVM physical volumes (PVs)",
728    "\
729 List all the physical volumes detected.  This is the equivalent
730 of the L<pvs(8)> command.
731
732 This returns a list of just the device names that contain
733 PVs (eg. C</dev/sda2>).
734
735 See also C<guestfs_pvs_full>.");
736
737   ("vgs", (RStringList "volgroups", []), 10, [],
738    [InitBasicFSonLVM, Always, TestOutputList (
739       [["vgs"]], ["VG"]);
740     InitEmpty, Always, TestOutputList (
741       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ",10 ,20 ,"];
742        ["pvcreate"; "/dev/sda1"];
743        ["pvcreate"; "/dev/sda2"];
744        ["pvcreate"; "/dev/sda3"];
745        ["vgcreate"; "VG1"; "/dev/sda1 /dev/sda2"];
746        ["vgcreate"; "VG2"; "/dev/sda3"];
747        ["vgs"]], ["VG1"; "VG2"])],
748    "list the LVM volume groups (VGs)",
749    "\
750 List all the volumes groups detected.  This is the equivalent
751 of the L<vgs(8)> command.
752
753 This returns a list of just the volume group names that were
754 detected (eg. C<VolGroup00>).
755
756 See also C<guestfs_vgs_full>.");
757
758   ("lvs", (RStringList "logvols", []), 11, [],
759    [InitBasicFSonLVM, Always, TestOutputList (
760       [["lvs"]], ["/dev/VG/LV"]);
761     InitEmpty, Always, TestOutputList (
762       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ",10 ,20 ,"];
763        ["pvcreate"; "/dev/sda1"];
764        ["pvcreate"; "/dev/sda2"];
765        ["pvcreate"; "/dev/sda3"];
766        ["vgcreate"; "VG1"; "/dev/sda1 /dev/sda2"];
767        ["vgcreate"; "VG2"; "/dev/sda3"];
768        ["lvcreate"; "LV1"; "VG1"; "50"];
769        ["lvcreate"; "LV2"; "VG1"; "50"];
770        ["lvcreate"; "LV3"; "VG2"; "50"];
771        ["lvs"]], ["/dev/VG1/LV1"; "/dev/VG1/LV2"; "/dev/VG2/LV3"])],
772    "list the LVM logical volumes (LVs)",
773    "\
774 List all the logical volumes detected.  This is the equivalent
775 of the L<lvs(8)> command.
776
777 This returns a list of the logical volume device names
778 (eg. C</dev/VolGroup00/LogVol00>).
779
780 See also C<guestfs_lvs_full>.");
781
782   ("pvs_full", (RPVList "physvols", []), 12, [],
783    [], (* XXX how to test? *)
784    "list the LVM physical volumes (PVs)",
785    "\
786 List all the physical volumes detected.  This is the equivalent
787 of the L<pvs(8)> command.  The \"full\" version includes all fields.");
788
789   ("vgs_full", (RVGList "volgroups", []), 13, [],
790    [], (* XXX how to test? *)
791    "list the LVM volume groups (VGs)",
792    "\
793 List all the volumes groups detected.  This is the equivalent
794 of the L<vgs(8)> command.  The \"full\" version includes all fields.");
795
796   ("lvs_full", (RLVList "logvols", []), 14, [],
797    [], (* XXX how to test? *)
798    "list the LVM logical volumes (LVs)",
799    "\
800 List all the logical volumes detected.  This is the equivalent
801 of the L<lvs(8)> command.  The \"full\" version includes all fields.");
802
803   ("read_lines", (RStringList "lines", [String "path"]), 15, [],
804    [InitBasicFS, Always, TestOutputList (
805       [["write_file"; "/new"; "line1\r\nline2\nline3"; "0"];
806        ["read_lines"; "/new"]], ["line1"; "line2"; "line3"]);
807     InitBasicFS, Always, TestOutputList (
808       [["write_file"; "/new"; ""; "0"];
809        ["read_lines"; "/new"]], [])],
810    "read file as lines",
811    "\
812 Return the contents of the file named C<path>.
813
814 The file contents are returned as a list of lines.  Trailing
815 C<LF> and C<CRLF> character sequences are I<not> returned.
816
817 Note that this function cannot correctly handle binary files
818 (specifically, files containing C<\\0> character which is treated
819 as end of line).  For those you need to use the C<guestfs_read_file>
820 function which has a more complex interface.");
821
822   ("aug_init", (RErr, [String "root"; Int "flags"]), 16, [],
823    [], (* XXX Augeas code needs tests. *)
824    "create a new Augeas handle",
825    "\
826 Create a new Augeas handle for editing configuration files.
827 If there was any previous Augeas handle associated with this
828 guestfs session, then it is closed.
829
830 You must call this before using any other C<guestfs_aug_*>
831 commands.
832
833 C<root> is the filesystem root.  C<root> must not be NULL,
834 use C</> instead.
835
836 The flags are the same as the flags defined in
837 E<lt>augeas.hE<gt>, the logical I<or> of the following
838 integers:
839
840 =over 4
841
842 =item C<AUG_SAVE_BACKUP> = 1
843
844 Keep the original file with a C<.augsave> extension.
845
846 =item C<AUG_SAVE_NEWFILE> = 2
847
848 Save changes into a file with extension C<.augnew>, and
849 do not overwrite original.  Overrides C<AUG_SAVE_BACKUP>.
850
851 =item C<AUG_TYPE_CHECK> = 4
852
853 Typecheck lenses (can be expensive).
854
855 =item C<AUG_NO_STDINC> = 8
856
857 Do not use standard load path for modules.
858
859 =item C<AUG_SAVE_NOOP> = 16
860
861 Make save a no-op, just record what would have been changed.
862
863 =item C<AUG_NO_LOAD> = 32
864
865 Do not load the tree in C<guestfs_aug_init>.
866
867 =back
868
869 To close the handle, you can call C<guestfs_aug_close>.
870
871 To find out more about Augeas, see L<http://augeas.net/>.");
872
873   ("aug_close", (RErr, []), 26, [],
874    [], (* XXX Augeas code needs tests. *)
875    "close the current Augeas handle",
876    "\
877 Close the current Augeas handle and free up any resources
878 used by it.  After calling this, you have to call
879 C<guestfs_aug_init> again before you can use any other
880 Augeas functions.");
881
882   ("aug_defvar", (RInt "nrnodes", [String "name"; OptString "expr"]), 17, [],
883    [], (* XXX Augeas code needs tests. *)
884    "define an Augeas variable",
885    "\
886 Defines an Augeas variable C<name> whose value is the result
887 of evaluating C<expr>.  If C<expr> is NULL, then C<name> is
888 undefined.
889
890 On success this returns the number of nodes in C<expr>, or
891 C<0> if C<expr> evaluates to something which is not a nodeset.");
892
893   ("aug_defnode", (RIntBool ("nrnodes", "created"), [String "name"; String "expr"; String "val"]), 18, [],
894    [], (* XXX Augeas code needs tests. *)
895    "define an Augeas node",
896    "\
897 Defines a variable C<name> whose value is the result of
898 evaluating C<expr>.
899
900 If C<expr> evaluates to an empty nodeset, a node is created,
901 equivalent to calling C<guestfs_aug_set> C<expr>, C<value>.
902 C<name> will be the nodeset containing that single node.
903
904 On success this returns a pair containing the
905 number of nodes in the nodeset, and a boolean flag
906 if a node was created.");
907
908   ("aug_get", (RString "val", [String "path"]), 19, [],
909    [], (* XXX Augeas code needs tests. *)
910    "look up the value of an Augeas path",
911    "\
912 Look up the value associated with C<path>.  If C<path>
913 matches exactly one node, the C<value> is returned.");
914
915   ("aug_set", (RErr, [String "path"; String "val"]), 20, [],
916    [], (* XXX Augeas code needs tests. *)
917    "set Augeas path to value",
918    "\
919 Set the value associated with C<path> to C<value>.");
920
921   ("aug_insert", (RErr, [String "path"; String "label"; Bool "before"]), 21, [],
922    [], (* XXX Augeas code needs tests. *)
923    "insert a sibling Augeas node",
924    "\
925 Create a new sibling C<label> for C<path>, inserting it into
926 the tree before or after C<path> (depending on the boolean
927 flag C<before>).
928
929 C<path> must match exactly one existing node in the tree, and
930 C<label> must be a label, ie. not contain C</>, C<*> or end
931 with a bracketed index C<[N]>.");
932
933   ("aug_rm", (RInt "nrnodes", [String "path"]), 22, [],
934    [], (* XXX Augeas code needs tests. *)
935    "remove an Augeas path",
936    "\
937 Remove C<path> and all of its children.
938
939 On success this returns the number of entries which were removed.");
940
941   ("aug_mv", (RErr, [String "src"; String "dest"]), 23, [],
942    [], (* XXX Augeas code needs tests. *)
943    "move Augeas node",
944    "\
945 Move the node C<src> to C<dest>.  C<src> must match exactly
946 one node.  C<dest> is overwritten if it exists.");
947
948   ("aug_match", (RStringList "matches", [String "path"]), 24, [],
949    [], (* XXX Augeas code needs tests. *)
950    "return Augeas nodes which match path",
951    "\
952 Returns a list of paths which match the path expression C<path>.
953 The returned paths are sufficiently qualified so that they match
954 exactly one node in the current tree.");
955
956   ("aug_save", (RErr, []), 25, [],
957    [], (* XXX Augeas code needs tests. *)
958    "write all pending Augeas changes to disk",
959    "\
960 This writes all pending changes to disk.
961
962 The flags which were passed to C<guestfs_aug_init> affect exactly
963 how files are saved.");
964
965   ("aug_load", (RErr, []), 27, [],
966    [], (* XXX Augeas code needs tests. *)
967    "load files into the tree",
968    "\
969 Load files into the tree.
970
971 See C<aug_load> in the Augeas documentation for the full gory
972 details.");
973
974   ("aug_ls", (RStringList "matches", [String "path"]), 28, [],
975    [], (* XXX Augeas code needs tests. *)
976    "list Augeas nodes under a path",
977    "\
978 This is just a shortcut for listing C<guestfs_aug_match>
979 C<path/*> and sorting the resulting nodes into alphabetical order.");
980
981   ("rm", (RErr, [String "path"]), 29, [],
982    [InitBasicFS, Always, TestRun
983       [["touch"; "/new"];
984        ["rm"; "/new"]];
985     InitBasicFS, Always, TestLastFail
986       [["rm"; "/new"]];
987     InitBasicFS, Always, TestLastFail
988       [["mkdir"; "/new"];
989        ["rm"; "/new"]]],
990    "remove a file",
991    "\
992 Remove the single file C<path>.");
993
994   ("rmdir", (RErr, [String "path"]), 30, [],
995    [InitBasicFS, Always, TestRun
996       [["mkdir"; "/new"];
997        ["rmdir"; "/new"]];
998     InitBasicFS, Always, TestLastFail
999       [["rmdir"; "/new"]];
1000     InitBasicFS, Always, TestLastFail
1001       [["touch"; "/new"];
1002        ["rmdir"; "/new"]]],
1003    "remove a directory",
1004    "\
1005 Remove the single directory C<path>.");
1006
1007   ("rm_rf", (RErr, [String "path"]), 31, [],
1008    [InitBasicFS, Always, TestOutputFalse
1009       [["mkdir"; "/new"];
1010        ["mkdir"; "/new/foo"];
1011        ["touch"; "/new/foo/bar"];
1012        ["rm_rf"; "/new"];
1013        ["exists"; "/new"]]],
1014    "remove a file or directory recursively",
1015    "\
1016 Remove the file or directory C<path>, recursively removing the
1017 contents if its a directory.  This is like the C<rm -rf> shell
1018 command.");
1019
1020   ("mkdir", (RErr, [String "path"]), 32, [],
1021    [InitBasicFS, Always, TestOutputTrue
1022       [["mkdir"; "/new"];
1023        ["is_dir"; "/new"]];
1024     InitBasicFS, Always, TestLastFail
1025       [["mkdir"; "/new/foo/bar"]]],
1026    "create a directory",
1027    "\
1028 Create a directory named C<path>.");
1029
1030   ("mkdir_p", (RErr, [String "path"]), 33, [],
1031    [InitBasicFS, Always, TestOutputTrue
1032       [["mkdir_p"; "/new/foo/bar"];
1033        ["is_dir"; "/new/foo/bar"]];
1034     InitBasicFS, Always, TestOutputTrue
1035       [["mkdir_p"; "/new/foo/bar"];
1036        ["is_dir"; "/new/foo"]];
1037     InitBasicFS, Always, TestOutputTrue
1038       [["mkdir_p"; "/new/foo/bar"];
1039        ["is_dir"; "/new"]];
1040     (* Regression tests for RHBZ#503133: *)
1041     InitBasicFS, Always, TestRun
1042       [["mkdir"; "/new"];
1043        ["mkdir_p"; "/new"]];
1044     InitBasicFS, Always, TestLastFail
1045       [["touch"; "/new"];
1046        ["mkdir_p"; "/new"]]],
1047    "create a directory and parents",
1048    "\
1049 Create a directory named C<path>, creating any parent directories
1050 as necessary.  This is like the C<mkdir -p> shell command.");
1051
1052   ("chmod", (RErr, [Int "mode"; String "path"]), 34, [],
1053    [], (* XXX Need stat command to test *)
1054    "change file mode",
1055    "\
1056 Change the mode (permissions) of C<path> to C<mode>.  Only
1057 numeric modes are supported.");
1058
1059   ("chown", (RErr, [Int "owner"; Int "group"; String "path"]), 35, [],
1060    [], (* XXX Need stat command to test *)
1061    "change file owner and group",
1062    "\
1063 Change the file owner to C<owner> and group to C<group>.
1064
1065 Only numeric uid and gid are supported.  If you want to use
1066 names, you will need to locate and parse the password file
1067 yourself (Augeas support makes this relatively easy).");
1068
1069   ("exists", (RBool "existsflag", [String "path"]), 36, [],
1070    [InitBasicFS, Always, TestOutputTrue (
1071       [["touch"; "/new"];
1072        ["exists"; "/new"]]);
1073     InitBasicFS, Always, TestOutputTrue (
1074       [["mkdir"; "/new"];
1075        ["exists"; "/new"]])],
1076    "test if file or directory exists",
1077    "\
1078 This returns C<true> if and only if there is a file, directory
1079 (or anything) with the given C<path> name.
1080
1081 See also C<guestfs_is_file>, C<guestfs_is_dir>, C<guestfs_stat>.");
1082
1083   ("is_file", (RBool "fileflag", [String "path"]), 37, [],
1084    [InitBasicFS, Always, TestOutputTrue (
1085       [["touch"; "/new"];
1086        ["is_file"; "/new"]]);
1087     InitBasicFS, Always, TestOutputFalse (
1088       [["mkdir"; "/new"];
1089        ["is_file"; "/new"]])],
1090    "test if file exists",
1091    "\
1092 This returns C<true> if and only if there is a file
1093 with the given C<path> name.  Note that it returns false for
1094 other objects like directories.
1095
1096 See also C<guestfs_stat>.");
1097
1098   ("is_dir", (RBool "dirflag", [String "path"]), 38, [],
1099    [InitBasicFS, Always, TestOutputFalse (
1100       [["touch"; "/new"];
1101        ["is_dir"; "/new"]]);
1102     InitBasicFS, Always, TestOutputTrue (
1103       [["mkdir"; "/new"];
1104        ["is_dir"; "/new"]])],
1105    "test if file exists",
1106    "\
1107 This returns C<true> if and only if there is a directory
1108 with the given C<path> name.  Note that it returns false for
1109 other objects like files.
1110
1111 See also C<guestfs_stat>.");
1112
1113   ("pvcreate", (RErr, [String "device"]), 39, [],
1114    [InitEmpty, Always, TestOutputListOfDevices (
1115       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ",10 ,20 ,"];
1116        ["pvcreate"; "/dev/sda1"];
1117        ["pvcreate"; "/dev/sda2"];
1118        ["pvcreate"; "/dev/sda3"];
1119        ["pvs"]], ["/dev/sda1"; "/dev/sda2"; "/dev/sda3"])],
1120    "create an LVM physical volume",
1121    "\
1122 This creates an LVM physical volume on the named C<device>,
1123 where C<device> should usually be a partition name such
1124 as C</dev/sda1>.");
1125
1126   ("vgcreate", (RErr, [String "volgroup"; StringList "physvols"]), 40, [],
1127    [InitEmpty, Always, TestOutputList (
1128       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ",10 ,20 ,"];
1129        ["pvcreate"; "/dev/sda1"];
1130        ["pvcreate"; "/dev/sda2"];
1131        ["pvcreate"; "/dev/sda3"];
1132        ["vgcreate"; "VG1"; "/dev/sda1 /dev/sda2"];
1133        ["vgcreate"; "VG2"; "/dev/sda3"];
1134        ["vgs"]], ["VG1"; "VG2"])],
1135    "create an LVM volume group",
1136    "\
1137 This creates an LVM volume group called C<volgroup>
1138 from the non-empty list of physical volumes C<physvols>.");
1139
1140   ("lvcreate", (RErr, [String "logvol"; String "volgroup"; Int "mbytes"]), 41, [],
1141    [InitEmpty, Always, TestOutputList (
1142       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ",10 ,20 ,"];
1143        ["pvcreate"; "/dev/sda1"];
1144        ["pvcreate"; "/dev/sda2"];
1145        ["pvcreate"; "/dev/sda3"];
1146        ["vgcreate"; "VG1"; "/dev/sda1 /dev/sda2"];
1147        ["vgcreate"; "VG2"; "/dev/sda3"];
1148        ["lvcreate"; "LV1"; "VG1"; "50"];
1149        ["lvcreate"; "LV2"; "VG1"; "50"];
1150        ["lvcreate"; "LV3"; "VG2"; "50"];
1151        ["lvcreate"; "LV4"; "VG2"; "50"];
1152        ["lvcreate"; "LV5"; "VG2"; "50"];
1153        ["lvs"]],
1154       ["/dev/VG1/LV1"; "/dev/VG1/LV2";
1155        "/dev/VG2/LV3"; "/dev/VG2/LV4"; "/dev/VG2/LV5"])],
1156    "create an LVM volume group",
1157    "\
1158 This creates an LVM volume group called C<logvol>
1159 on the volume group C<volgroup>, with C<size> megabytes.");
1160
1161   ("mkfs", (RErr, [String "fstype"; String "device"]), 42, [],
1162    [InitEmpty, Always, TestOutput (
1163       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
1164        ["mkfs"; "ext2"; "/dev/sda1"];
1165        ["mount"; "/dev/sda1"; "/"];
1166        ["write_file"; "/new"; "new file contents"; "0"];
1167        ["cat"; "/new"]], "new file contents")],
1168    "make a filesystem",
1169    "\
1170 This creates a filesystem on C<device> (usually a partition
1171 or LVM logical volume).  The filesystem type is C<fstype>, for
1172 example C<ext3>.");
1173
1174   ("sfdisk", (RErr, [String "device";
1175                      Int "cyls"; Int "heads"; Int "sectors";
1176                      StringList "lines"]), 43, [DangerWillRobinson],
1177    [],
1178    "create partitions on a block device",
1179    "\
1180 This is a direct interface to the L<sfdisk(8)> program for creating
1181 partitions on block devices.
1182
1183 C<device> should be a block device, for example C</dev/sda>.
1184
1185 C<cyls>, C<heads> and C<sectors> are the number of cylinders, heads
1186 and sectors on the device, which are passed directly to sfdisk as
1187 the I<-C>, I<-H> and I<-S> parameters.  If you pass C<0> for any
1188 of these, then the corresponding parameter is omitted.  Usually for
1189 'large' disks, you can just pass C<0> for these, but for small
1190 (floppy-sized) disks, sfdisk (or rather, the kernel) cannot work
1191 out the right geometry and you will need to tell it.
1192
1193 C<lines> is a list of lines that we feed to C<sfdisk>.  For more
1194 information refer to the L<sfdisk(8)> manpage.
1195
1196 To create a single partition occupying the whole disk, you would
1197 pass C<lines> as a single element list, when the single element being
1198 the string C<,> (comma).
1199
1200 See also: C<guestfs_sfdisk_l>, C<guestfs_sfdisk_N>");
1201
1202   ("write_file", (RErr, [String "path"; String "content"; Int "size"]), 44, [ProtocolLimitWarning],
1203    [InitBasicFS, Always, TestOutput (
1204       [["write_file"; "/new"; "new file contents"; "0"];
1205        ["cat"; "/new"]], "new file contents");
1206     InitBasicFS, Always, TestOutput (
1207       [["write_file"; "/new"; "\nnew file contents\n"; "0"];
1208        ["cat"; "/new"]], "\nnew file contents\n");
1209     InitBasicFS, Always, TestOutput (
1210       [["write_file"; "/new"; "\n\n"; "0"];
1211        ["cat"; "/new"]], "\n\n");
1212     InitBasicFS, Always, TestOutput (
1213       [["write_file"; "/new"; ""; "0"];
1214        ["cat"; "/new"]], "");
1215     InitBasicFS, Always, TestOutput (
1216       [["write_file"; "/new"; "\n\n\n"; "0"];
1217        ["cat"; "/new"]], "\n\n\n");
1218     InitBasicFS, Always, TestOutput (
1219       [["write_file"; "/new"; "\n"; "0"];
1220        ["cat"; "/new"]], "\n")],
1221    "create a file",
1222    "\
1223 This call creates a file called C<path>.  The contents of the
1224 file is the string C<content> (which can contain any 8 bit data),
1225 with length C<size>.
1226
1227 As a special case, if C<size> is C<0>
1228 then the length is calculated using C<strlen> (so in this case
1229 the content cannot contain embedded ASCII NULs).
1230
1231 I<NB.> Owing to a bug, writing content containing ASCII NUL
1232 characters does I<not> work, even if the length is specified.
1233 We hope to resolve this bug in a future version.  In the meantime
1234 use C<guestfs_upload>.");
1235
1236   ("umount", (RErr, [String "pathordevice"]), 45, [FishAlias "unmount"],
1237    [InitEmpty, Always, TestOutputListOfDevices (
1238       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
1239        ["mkfs"; "ext2"; "/dev/sda1"];
1240        ["mount"; "/dev/sda1"; "/"];
1241        ["mounts"]], ["/dev/sda1"]);
1242     InitEmpty, Always, TestOutputList (
1243       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
1244        ["mkfs"; "ext2"; "/dev/sda1"];
1245        ["mount"; "/dev/sda1"; "/"];
1246        ["umount"; "/"];
1247        ["mounts"]], [])],
1248    "unmount a filesystem",
1249    "\
1250 This unmounts the given filesystem.  The filesystem may be
1251 specified either by its mountpoint (path) or the device which
1252 contains the filesystem.");
1253
1254   ("mounts", (RStringList "devices", []), 46, [],
1255    [InitBasicFS, Always, TestOutputListOfDevices (
1256       [["mounts"]], ["/dev/sda1"])],
1257    "show mounted filesystems",
1258    "\
1259 This returns the list of currently mounted filesystems.  It returns
1260 the list of devices (eg. C</dev/sda1>, C</dev/VG/LV>).
1261
1262 Some internal mounts are not shown.");
1263
1264   ("umount_all", (RErr, []), 47, [FishAlias "unmount-all"],
1265    [InitBasicFS, Always, TestOutputList (
1266       [["umount_all"];
1267        ["mounts"]], []);
1268     (* check that umount_all can unmount nested mounts correctly: *)
1269     InitEmpty, Always, TestOutputList (
1270       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ",10 ,20 ,"];
1271        ["mkfs"; "ext2"; "/dev/sda1"];
1272        ["mkfs"; "ext2"; "/dev/sda2"];
1273        ["mkfs"; "ext2"; "/dev/sda3"];
1274        ["mount"; "/dev/sda1"; "/"];
1275        ["mkdir"; "/mp1"];
1276        ["mount"; "/dev/sda2"; "/mp1"];
1277        ["mkdir"; "/mp1/mp2"];
1278        ["mount"; "/dev/sda3"; "/mp1/mp2"];
1279        ["mkdir"; "/mp1/mp2/mp3"];
1280        ["umount_all"];
1281        ["mounts"]], [])],
1282    "unmount all filesystems",
1283    "\
1284 This unmounts all mounted filesystems.
1285
1286 Some internal mounts are not unmounted by this call.");
1287
1288   ("lvm_remove_all", (RErr, []), 48, [DangerWillRobinson],
1289    [],
1290    "remove all LVM LVs, VGs and PVs",
1291    "\
1292 This command removes all LVM logical volumes, volume groups
1293 and physical volumes.");
1294
1295   ("file", (RString "description", [String "path"]), 49, [],
1296    [InitBasicFS, Always, TestOutput (
1297       [["touch"; "/new"];
1298        ["file"; "/new"]], "empty");
1299     InitBasicFS, Always, TestOutput (
1300       [["write_file"; "/new"; "some content\n"; "0"];
1301        ["file"; "/new"]], "ASCII text");
1302     InitBasicFS, Always, TestLastFail (
1303       [["file"; "/nofile"]])],
1304    "determine file type",
1305    "\
1306 This call uses the standard L<file(1)> command to determine
1307 the type or contents of the file.  This also works on devices,
1308 for example to find out whether a partition contains a filesystem.
1309
1310 The exact command which runs is C<file -bsL path>.  Note in
1311 particular that the filename is not prepended to the output
1312 (the C<-b> option).");
1313
1314   ("command", (RString "output", [StringList "arguments"]), 50, [ProtocolLimitWarning],
1315    [InitBasicFS, Always, TestOutput (
1316       [["upload"; "test-command"; "/test-command"];
1317        ["chmod"; "493"; "/test-command"];
1318        ["command"; "/test-command 1"]], "Result1");
1319     InitBasicFS, Always, TestOutput (
1320       [["upload"; "test-command"; "/test-command"];
1321        ["chmod"; "493"; "/test-command"];
1322        ["command"; "/test-command 2"]], "Result2\n");
1323     InitBasicFS, Always, TestOutput (
1324       [["upload"; "test-command"; "/test-command"];
1325        ["chmod"; "493"; "/test-command"];
1326        ["command"; "/test-command 3"]], "\nResult3");
1327     InitBasicFS, Always, TestOutput (
1328       [["upload"; "test-command"; "/test-command"];
1329        ["chmod"; "493"; "/test-command"];
1330        ["command"; "/test-command 4"]], "\nResult4\n");
1331     InitBasicFS, Always, TestOutput (
1332       [["upload"; "test-command"; "/test-command"];
1333        ["chmod"; "493"; "/test-command"];
1334        ["command"; "/test-command 5"]], "\nResult5\n\n");
1335     InitBasicFS, Always, TestOutput (
1336       [["upload"; "test-command"; "/test-command"];
1337        ["chmod"; "493"; "/test-command"];
1338        ["command"; "/test-command 6"]], "\n\nResult6\n\n");
1339     InitBasicFS, Always, TestOutput (
1340       [["upload"; "test-command"; "/test-command"];
1341        ["chmod"; "493"; "/test-command"];
1342        ["command"; "/test-command 7"]], "");
1343     InitBasicFS, Always, TestOutput (
1344       [["upload"; "test-command"; "/test-command"];
1345        ["chmod"; "493"; "/test-command"];
1346        ["command"; "/test-command 8"]], "\n");
1347     InitBasicFS, Always, TestOutput (
1348       [["upload"; "test-command"; "/test-command"];
1349        ["chmod"; "493"; "/test-command"];
1350        ["command"; "/test-command 9"]], "\n\n");
1351     InitBasicFS, Always, TestOutput (
1352       [["upload"; "test-command"; "/test-command"];
1353        ["chmod"; "493"; "/test-command"];
1354        ["command"; "/test-command 10"]], "Result10-1\nResult10-2\n");
1355     InitBasicFS, Always, TestOutput (
1356       [["upload"; "test-command"; "/test-command"];
1357        ["chmod"; "493"; "/test-command"];
1358        ["command"; "/test-command 11"]], "Result11-1\nResult11-2");
1359     InitBasicFS, Always, TestLastFail (
1360       [["upload"; "test-command"; "/test-command"];
1361        ["chmod"; "493"; "/test-command"];
1362        ["command"; "/test-command"]])],
1363    "run a command from the guest filesystem",
1364    "\
1365 This call runs a command from the guest filesystem.  The
1366 filesystem must be mounted, and must contain a compatible
1367 operating system (ie. something Linux, with the same
1368 or compatible processor architecture).
1369
1370 The single parameter is an argv-style list of arguments.
1371 The first element is the name of the program to run.
1372 Subsequent elements are parameters.  The list must be
1373 non-empty (ie. must contain a program name).  Note that
1374 the command runs directly, and is I<not> invoked via
1375 the shell (see C<guestfs_sh>).
1376
1377 The return value is anything printed to I<stdout> by
1378 the command.
1379
1380 If the command returns a non-zero exit status, then
1381 this function returns an error message.  The error message
1382 string is the content of I<stderr> from the command.
1383
1384 The C<$PATH> environment variable will contain at least
1385 C</usr/bin> and C</bin>.  If you require a program from
1386 another location, you should provide the full path in the
1387 first parameter.
1388
1389 Shared libraries and data files required by the program
1390 must be available on filesystems which are mounted in the
1391 correct places.  It is the caller's responsibility to ensure
1392 all filesystems that are needed are mounted at the right
1393 locations.");
1394
1395   ("command_lines", (RStringList "lines", [StringList "arguments"]), 51, [ProtocolLimitWarning],
1396    [InitBasicFS, Always, TestOutputList (
1397       [["upload"; "test-command"; "/test-command"];
1398        ["chmod"; "493"; "/test-command"];
1399        ["command_lines"; "/test-command 1"]], ["Result1"]);
1400     InitBasicFS, Always, TestOutputList (
1401       [["upload"; "test-command"; "/test-command"];
1402        ["chmod"; "493"; "/test-command"];
1403        ["command_lines"; "/test-command 2"]], ["Result2"]);
1404     InitBasicFS, Always, TestOutputList (
1405       [["upload"; "test-command"; "/test-command"];
1406        ["chmod"; "493"; "/test-command"];
1407        ["command_lines"; "/test-command 3"]], ["";"Result3"]);
1408     InitBasicFS, Always, TestOutputList (
1409       [["upload"; "test-command"; "/test-command"];
1410        ["chmod"; "493"; "/test-command"];
1411        ["command_lines"; "/test-command 4"]], ["";"Result4"]);
1412     InitBasicFS, Always, TestOutputList (
1413       [["upload"; "test-command"; "/test-command"];
1414        ["chmod"; "493"; "/test-command"];
1415        ["command_lines"; "/test-command 5"]], ["";"Result5";""]);
1416     InitBasicFS, Always, TestOutputList (
1417       [["upload"; "test-command"; "/test-command"];
1418        ["chmod"; "493"; "/test-command"];
1419        ["command_lines"; "/test-command 6"]], ["";"";"Result6";""]);
1420     InitBasicFS, Always, TestOutputList (
1421       [["upload"; "test-command"; "/test-command"];
1422        ["chmod"; "493"; "/test-command"];
1423        ["command_lines"; "/test-command 7"]], []);
1424     InitBasicFS, Always, TestOutputList (
1425       [["upload"; "test-command"; "/test-command"];
1426        ["chmod"; "493"; "/test-command"];
1427        ["command_lines"; "/test-command 8"]], [""]);
1428     InitBasicFS, Always, TestOutputList (
1429       [["upload"; "test-command"; "/test-command"];
1430        ["chmod"; "493"; "/test-command"];
1431        ["command_lines"; "/test-command 9"]], ["";""]);
1432     InitBasicFS, Always, TestOutputList (
1433       [["upload"; "test-command"; "/test-command"];
1434        ["chmod"; "493"; "/test-command"];
1435        ["command_lines"; "/test-command 10"]], ["Result10-1";"Result10-2"]);
1436     InitBasicFS, Always, TestOutputList (
1437       [["upload"; "test-command"; "/test-command"];
1438        ["chmod"; "493"; "/test-command"];
1439        ["command_lines"; "/test-command 11"]], ["Result11-1";"Result11-2"])],
1440    "run a command, returning lines",
1441    "\
1442 This is the same as C<guestfs_command>, but splits the
1443 result into a list of lines.
1444
1445 See also: C<guestfs_sh_lines>");
1446
1447   ("stat", (RStat "statbuf", [String "path"]), 52, [],
1448    [InitBasicFS, Always, TestOutputStruct (
1449       [["touch"; "/new"];
1450        ["stat"; "/new"]], [CompareWithInt ("size", 0)])],
1451    "get file information",
1452    "\
1453 Returns file information for the given C<path>.
1454
1455 This is the same as the C<stat(2)> system call.");
1456
1457   ("lstat", (RStat "statbuf", [String "path"]), 53, [],
1458    [InitBasicFS, Always, TestOutputStruct (
1459       [["touch"; "/new"];
1460        ["lstat"; "/new"]], [CompareWithInt ("size", 0)])],
1461    "get file information for a symbolic link",
1462    "\
1463 Returns file information for the given C<path>.
1464
1465 This is the same as C<guestfs_stat> except that if C<path>
1466 is a symbolic link, then the link is stat-ed, not the file it
1467 refers to.
1468
1469 This is the same as the C<lstat(2)> system call.");
1470
1471   ("statvfs", (RStatVFS "statbuf", [String "path"]), 54, [],
1472    [InitBasicFS, Always, TestOutputStruct (
1473       [["statvfs"; "/"]], [CompareWithInt ("bfree", 487702);
1474                            CompareWithInt ("blocks", 490020);
1475                            CompareWithInt ("bsize", 1024)])],
1476    "get file system statistics",
1477    "\
1478 Returns file system statistics for any mounted file system.
1479 C<path> should be a file or directory in the mounted file system
1480 (typically it is the mount point itself, but it doesn't need to be).
1481
1482 This is the same as the C<statvfs(2)> system call.");
1483
1484   ("tune2fs_l", (RHashtable "superblock", [String "device"]), 55, [],
1485    [], (* XXX test *)
1486    "get ext2/ext3/ext4 superblock details",
1487    "\
1488 This returns the contents of the ext2, ext3 or ext4 filesystem
1489 superblock on C<device>.
1490
1491 It is the same as running C<tune2fs -l device>.  See L<tune2fs(8)>
1492 manpage for more details.  The list of fields returned isn't
1493 clearly defined, and depends on both the version of C<tune2fs>
1494 that libguestfs was built against, and the filesystem itself.");
1495
1496   ("blockdev_setro", (RErr, [String "device"]), 56, [],
1497    [InitEmpty, Always, TestOutputTrue (
1498       [["blockdev_setro"; "/dev/sda"];
1499        ["blockdev_getro"; "/dev/sda"]])],
1500    "set block device to read-only",
1501    "\
1502 Sets the block device named C<device> to read-only.
1503
1504 This uses the L<blockdev(8)> command.");
1505
1506   ("blockdev_setrw", (RErr, [String "device"]), 57, [],
1507    [InitEmpty, Always, TestOutputFalse (
1508       [["blockdev_setrw"; "/dev/sda"];
1509        ["blockdev_getro"; "/dev/sda"]])],
1510    "set block device to read-write",
1511    "\
1512 Sets the block device named C<device> to read-write.
1513
1514 This uses the L<blockdev(8)> command.");
1515
1516   ("blockdev_getro", (RBool "ro", [String "device"]), 58, [],
1517    [InitEmpty, Always, TestOutputTrue (
1518       [["blockdev_setro"; "/dev/sda"];
1519        ["blockdev_getro"; "/dev/sda"]])],
1520    "is block device set to read-only",
1521    "\
1522 Returns a boolean indicating if the block device is read-only
1523 (true if read-only, false if not).
1524
1525 This uses the L<blockdev(8)> command.");
1526
1527   ("blockdev_getss", (RInt "sectorsize", [String "device"]), 59, [],
1528    [InitEmpty, Always, TestOutputInt (
1529       [["blockdev_getss"; "/dev/sda"]], 512)],
1530    "get sectorsize of block device",
1531    "\
1532 This returns the size of sectors on a block device.
1533 Usually 512, but can be larger for modern devices.
1534
1535 (Note, this is not the size in sectors, use C<guestfs_blockdev_getsz>
1536 for that).
1537
1538 This uses the L<blockdev(8)> command.");
1539
1540   ("blockdev_getbsz", (RInt "blocksize", [String "device"]), 60, [],
1541    [InitEmpty, Always, TestOutputInt (
1542       [["blockdev_getbsz"; "/dev/sda"]], 4096)],
1543    "get blocksize of block device",
1544    "\
1545 This returns the block size of a device.
1546
1547 (Note this is different from both I<size in blocks> and
1548 I<filesystem block size>).
1549
1550 This uses the L<blockdev(8)> command.");
1551
1552   ("blockdev_setbsz", (RErr, [String "device"; Int "blocksize"]), 61, [],
1553    [], (* XXX test *)
1554    "set blocksize of block device",
1555    "\
1556 This sets the block size of a device.
1557
1558 (Note this is different from both I<size in blocks> and
1559 I<filesystem block size>).
1560
1561 This uses the L<blockdev(8)> command.");
1562
1563   ("blockdev_getsz", (RInt64 "sizeinsectors", [String "device"]), 62, [],
1564    [InitEmpty, Always, TestOutputInt (
1565       [["blockdev_getsz"; "/dev/sda"]], 1024000)],
1566    "get total size of device in 512-byte sectors",
1567    "\
1568 This returns the size of the device in units of 512-byte sectors
1569 (even if the sectorsize isn't 512 bytes ... weird).
1570
1571 See also C<guestfs_blockdev_getss> for the real sector size of
1572 the device, and C<guestfs_blockdev_getsize64> for the more
1573 useful I<size in bytes>.
1574
1575 This uses the L<blockdev(8)> command.");
1576
1577   ("blockdev_getsize64", (RInt64 "sizeinbytes", [String "device"]), 63, [],
1578    [InitEmpty, Always, TestOutputInt (
1579       [["blockdev_getsize64"; "/dev/sda"]], 524288000)],
1580    "get total size of device in bytes",
1581    "\
1582 This returns the size of the device in bytes.
1583
1584 See also C<guestfs_blockdev_getsz>.
1585
1586 This uses the L<blockdev(8)> command.");
1587
1588   ("blockdev_flushbufs", (RErr, [String "device"]), 64, [],
1589    [InitEmpty, Always, TestRun
1590       [["blockdev_flushbufs"; "/dev/sda"]]],
1591    "flush device buffers",
1592    "\
1593 This tells the kernel to flush internal buffers associated
1594 with C<device>.
1595
1596 This uses the L<blockdev(8)> command.");
1597
1598   ("blockdev_rereadpt", (RErr, [String "device"]), 65, [],
1599    [InitEmpty, Always, TestRun
1600       [["blockdev_rereadpt"; "/dev/sda"]]],
1601    "reread partition table",
1602    "\
1603 Reread the partition table on C<device>.
1604
1605 This uses the L<blockdev(8)> command.");
1606
1607   ("upload", (RErr, [FileIn "filename"; String "remotefilename"]), 66, [],
1608    [InitBasicFS, Always, TestOutput (
1609       (* Pick a file from cwd which isn't likely to change. *)
1610     [["upload"; "../COPYING.LIB"; "/COPYING.LIB"];
1611      ["checksum"; "md5"; "/COPYING.LIB"]], "e3eda01d9815f8d24aae2dbd89b68b06")],
1612    "upload a file from the local machine",
1613    "\
1614 Upload local file C<filename> to C<remotefilename> on the
1615 filesystem.
1616
1617 C<filename> can also be a named pipe.
1618
1619 See also C<guestfs_download>.");
1620
1621   ("download", (RErr, [String "remotefilename"; FileOut "filename"]), 67, [],
1622    [InitBasicFS, Always, TestOutput (
1623       (* Pick a file from cwd which isn't likely to change. *)
1624     [["upload"; "../COPYING.LIB"; "/COPYING.LIB"];
1625      ["download"; "/COPYING.LIB"; "testdownload.tmp"];
1626      ["upload"; "testdownload.tmp"; "/upload"];
1627      ["checksum"; "md5"; "/upload"]], "e3eda01d9815f8d24aae2dbd89b68b06")],
1628    "download a file to the local machine",
1629    "\
1630 Download file C<remotefilename> and save it as C<filename>
1631 on the local machine.
1632
1633 C<filename> can also be a named pipe.
1634
1635 See also C<guestfs_upload>, C<guestfs_cat>.");
1636
1637   ("checksum", (RString "checksum", [String "csumtype"; String "path"]), 68, [],
1638    [InitBasicFS, Always, TestOutput (
1639       [["write_file"; "/new"; "test\n"; "0"];
1640        ["checksum"; "crc"; "/new"]], "935282863");
1641     InitBasicFS, Always, TestLastFail (
1642       [["checksum"; "crc"; "/new"]]);
1643     InitBasicFS, Always, TestOutput (
1644       [["write_file"; "/new"; "test\n"; "0"];
1645        ["checksum"; "md5"; "/new"]], "d8e8fca2dc0f896fd7cb4cb0031ba249");
1646     InitBasicFS, Always, TestOutput (
1647       [["write_file"; "/new"; "test\n"; "0"];
1648        ["checksum"; "sha1"; "/new"]], "4e1243bd22c66e76c2ba9eddc1f91394e57f9f83");
1649     InitBasicFS, Always, TestOutput (
1650       [["write_file"; "/new"; "test\n"; "0"];
1651        ["checksum"; "sha224"; "/new"]], "52f1bf093f4b7588726035c176c0cdb4376cfea53819f1395ac9e6ec");
1652     InitBasicFS, Always, TestOutput (
1653       [["write_file"; "/new"; "test\n"; "0"];
1654        ["checksum"; "sha256"; "/new"]], "f2ca1bb6c7e907d06dafe4687e579fce76b37e4e93b7605022da52e6ccc26fd2");
1655     InitBasicFS, Always, TestOutput (
1656       [["write_file"; "/new"; "test\n"; "0"];
1657        ["checksum"; "sha384"; "/new"]], "109bb6b5b6d5547c1ce03c7a8bd7d8f80c1cb0957f50c4f7fda04692079917e4f9cad52b878f3d8234e1a170b154b72d");
1658     InitBasicFS, Always, TestOutput (
1659       [["write_file"; "/new"; "test\n"; "0"];
1660        ["checksum"; "sha512"; "/new"]], "0e3e75234abc68f4378a86b3f4b32a198ba301845b0cd6e50106e874345700cc6663a86c1ea125dc5e92be17c98f9a0f85ca9d5f595db2012f7cc3571945c123");
1661     InitBasicFS, Always, TestOutput (
1662       (* RHEL 5 thinks this is an HFS+ filesystem unless we give
1663        * the type explicitly.
1664        *)
1665       [["mount_vfs"; "ro"; "squashfs"; "/dev/sdd"; "/"];
1666        ["checksum"; "md5"; "/known-3"]], "46d6ca27ee07cdc6fa99c2e138cc522c")],
1667    "compute MD5, SHAx or CRC checksum of file",
1668    "\
1669 This call computes the MD5, SHAx or CRC checksum of the
1670 file named C<path>.
1671
1672 The type of checksum to compute is given by the C<csumtype>
1673 parameter which must have one of the following values:
1674
1675 =over 4
1676
1677 =item C<crc>
1678
1679 Compute the cyclic redundancy check (CRC) specified by POSIX
1680 for the C<cksum> command.
1681
1682 =item C<md5>
1683
1684 Compute the MD5 hash (using the C<md5sum> program).
1685
1686 =item C<sha1>
1687
1688 Compute the SHA1 hash (using the C<sha1sum> program).
1689
1690 =item C<sha224>
1691
1692 Compute the SHA224 hash (using the C<sha224sum> program).
1693
1694 =item C<sha256>
1695
1696 Compute the SHA256 hash (using the C<sha256sum> program).
1697
1698 =item C<sha384>
1699
1700 Compute the SHA384 hash (using the C<sha384sum> program).
1701
1702 =item C<sha512>
1703
1704 Compute the SHA512 hash (using the C<sha512sum> program).
1705
1706 =back
1707
1708 The checksum is returned as a printable string.");
1709
1710   ("tar_in", (RErr, [FileIn "tarfile"; String "directory"]), 69, [],
1711    [InitBasicFS, Always, TestOutput (
1712       [["tar_in"; "../images/helloworld.tar"; "/"];
1713        ["cat"; "/hello"]], "hello\n")],
1714    "unpack tarfile to directory",
1715    "\
1716 This command uploads and unpacks local file C<tarfile> (an
1717 I<uncompressed> tar file) into C<directory>.
1718
1719 To upload a compressed tarball, use C<guestfs_tgz_in>.");
1720
1721   ("tar_out", (RErr, [String "directory"; FileOut "tarfile"]), 70, [],
1722    [],
1723    "pack directory into tarfile",
1724    "\
1725 This command packs the contents of C<directory> and downloads
1726 it to local file C<tarfile>.
1727
1728 To download a compressed tarball, use C<guestfs_tgz_out>.");
1729
1730   ("tgz_in", (RErr, [FileIn "tarball"; String "directory"]), 71, [],
1731    [InitBasicFS, Always, TestOutput (
1732       [["tgz_in"; "../images/helloworld.tar.gz"; "/"];
1733        ["cat"; "/hello"]], "hello\n")],
1734    "unpack compressed tarball to directory",
1735    "\
1736 This command uploads and unpacks local file C<tarball> (a
1737 I<gzip compressed> tar file) into C<directory>.
1738
1739 To upload an uncompressed tarball, use C<guestfs_tar_in>.");
1740
1741   ("tgz_out", (RErr, [String "directory"; FileOut "tarball"]), 72, [],
1742    [],
1743    "pack directory into compressed tarball",
1744    "\
1745 This command packs the contents of C<directory> and downloads
1746 it to local file C<tarball>.
1747
1748 To download an uncompressed tarball, use C<guestfs_tar_out>.");
1749
1750   ("mount_ro", (RErr, [String "device"; String "mountpoint"]), 73, [],
1751    [InitBasicFS, Always, TestLastFail (
1752       [["umount"; "/"];
1753        ["mount_ro"; "/dev/sda1"; "/"];
1754        ["touch"; "/new"]]);
1755     InitBasicFS, Always, TestOutput (
1756       [["write_file"; "/new"; "data"; "0"];
1757        ["umount"; "/"];
1758        ["mount_ro"; "/dev/sda1"; "/"];
1759        ["cat"; "/new"]], "data")],
1760    "mount a guest disk, read-only",
1761    "\
1762 This is the same as the C<guestfs_mount> command, but it
1763 mounts the filesystem with the read-only (I<-o ro>) flag.");
1764
1765   ("mount_options", (RErr, [String "options"; String "device"; String "mountpoint"]), 74, [],
1766    [],
1767    "mount a guest disk with mount options",
1768    "\
1769 This is the same as the C<guestfs_mount> command, but it
1770 allows you to set the mount options as for the
1771 L<mount(8)> I<-o> flag.");
1772
1773   ("mount_vfs", (RErr, [String "options"; String "vfstype"; String "device"; String "mountpoint"]), 75, [],
1774    [],
1775    "mount a guest disk with mount options and vfstype",
1776    "\
1777 This is the same as the C<guestfs_mount> command, but it
1778 allows you to set both the mount options and the vfstype
1779 as for the L<mount(8)> I<-o> and I<-t> flags.");
1780
1781   ("debug", (RString "result", [String "subcmd"; StringList "extraargs"]), 76, [],
1782    [],
1783    "debugging and internals",
1784    "\
1785 The C<guestfs_debug> command exposes some internals of
1786 C<guestfsd> (the guestfs daemon) that runs inside the
1787 qemu subprocess.
1788
1789 There is no comprehensive help for this command.  You have
1790 to look at the file C<daemon/debug.c> in the libguestfs source
1791 to find out what you can do.");
1792
1793   ("lvremove", (RErr, [String "device"]), 77, [],
1794    [InitEmpty, Always, TestOutputList (
1795       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
1796        ["pvcreate"; "/dev/sda1"];
1797        ["vgcreate"; "VG"; "/dev/sda1"];
1798        ["lvcreate"; "LV1"; "VG"; "50"];
1799        ["lvcreate"; "LV2"; "VG"; "50"];
1800        ["lvremove"; "/dev/VG/LV1"];
1801        ["lvs"]], ["/dev/VG/LV2"]);
1802     InitEmpty, Always, TestOutputList (
1803       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
1804        ["pvcreate"; "/dev/sda1"];
1805        ["vgcreate"; "VG"; "/dev/sda1"];
1806        ["lvcreate"; "LV1"; "VG"; "50"];
1807        ["lvcreate"; "LV2"; "VG"; "50"];
1808        ["lvremove"; "/dev/VG"];
1809        ["lvs"]], []);
1810     InitEmpty, Always, TestOutputList (
1811       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
1812        ["pvcreate"; "/dev/sda1"];
1813        ["vgcreate"; "VG"; "/dev/sda1"];
1814        ["lvcreate"; "LV1"; "VG"; "50"];
1815        ["lvcreate"; "LV2"; "VG"; "50"];
1816        ["lvremove"; "/dev/VG"];
1817        ["vgs"]], ["VG"])],
1818    "remove an LVM logical volume",
1819    "\
1820 Remove an LVM logical volume C<device>, where C<device> is
1821 the path to the LV, such as C</dev/VG/LV>.
1822
1823 You can also remove all LVs in a volume group by specifying
1824 the VG name, C</dev/VG>.");
1825
1826   ("vgremove", (RErr, [String "vgname"]), 78, [],
1827    [InitEmpty, Always, TestOutputList (
1828       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
1829        ["pvcreate"; "/dev/sda1"];
1830        ["vgcreate"; "VG"; "/dev/sda1"];
1831        ["lvcreate"; "LV1"; "VG"; "50"];
1832        ["lvcreate"; "LV2"; "VG"; "50"];
1833        ["vgremove"; "VG"];
1834        ["lvs"]], []);
1835     InitEmpty, Always, TestOutputList (
1836       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
1837        ["pvcreate"; "/dev/sda1"];
1838        ["vgcreate"; "VG"; "/dev/sda1"];
1839        ["lvcreate"; "LV1"; "VG"; "50"];
1840        ["lvcreate"; "LV2"; "VG"; "50"];
1841        ["vgremove"; "VG"];
1842        ["vgs"]], [])],
1843    "remove an LVM volume group",
1844    "\
1845 Remove an LVM volume group C<vgname>, (for example C<VG>).
1846
1847 This also forcibly removes all logical volumes in the volume
1848 group (if any).");
1849
1850   ("pvremove", (RErr, [String "device"]), 79, [],
1851    [InitEmpty, Always, TestOutputListOfDevices (
1852       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
1853        ["pvcreate"; "/dev/sda1"];
1854        ["vgcreate"; "VG"; "/dev/sda1"];
1855        ["lvcreate"; "LV1"; "VG"; "50"];
1856        ["lvcreate"; "LV2"; "VG"; "50"];
1857        ["vgremove"; "VG"];
1858        ["pvremove"; "/dev/sda1"];
1859        ["lvs"]], []);
1860     InitEmpty, Always, TestOutputListOfDevices (
1861       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
1862        ["pvcreate"; "/dev/sda1"];
1863        ["vgcreate"; "VG"; "/dev/sda1"];
1864        ["lvcreate"; "LV1"; "VG"; "50"];
1865        ["lvcreate"; "LV2"; "VG"; "50"];
1866        ["vgremove"; "VG"];
1867        ["pvremove"; "/dev/sda1"];
1868        ["vgs"]], []);
1869     InitEmpty, Always, TestOutputListOfDevices (
1870       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
1871        ["pvcreate"; "/dev/sda1"];
1872        ["vgcreate"; "VG"; "/dev/sda1"];
1873        ["lvcreate"; "LV1"; "VG"; "50"];
1874        ["lvcreate"; "LV2"; "VG"; "50"];
1875        ["vgremove"; "VG"];
1876        ["pvremove"; "/dev/sda1"];
1877        ["pvs"]], [])],
1878    "remove an LVM physical volume",
1879    "\
1880 This wipes a physical volume C<device> so that LVM will no longer
1881 recognise it.
1882
1883 The implementation uses the C<pvremove> command which refuses to
1884 wipe physical volumes that contain any volume groups, so you have
1885 to remove those first.");
1886
1887   ("set_e2label", (RErr, [String "device"; String "label"]), 80, [],
1888    [InitBasicFS, Always, TestOutput (
1889       [["set_e2label"; "/dev/sda1"; "testlabel"];
1890        ["get_e2label"; "/dev/sda1"]], "testlabel")],
1891    "set the ext2/3/4 filesystem label",
1892    "\
1893 This sets the ext2/3/4 filesystem label of the filesystem on
1894 C<device> to C<label>.  Filesystem labels are limited to
1895 16 characters.
1896
1897 You can use either C<guestfs_tune2fs_l> or C<guestfs_get_e2label>
1898 to return the existing label on a filesystem.");
1899
1900   ("get_e2label", (RString "label", [String "device"]), 81, [],
1901    [],
1902    "get the ext2/3/4 filesystem label",
1903    "\
1904 This returns the ext2/3/4 filesystem label of the filesystem on
1905 C<device>.");
1906
1907   ("set_e2uuid", (RErr, [String "device"; String "uuid"]), 82, [],
1908    [InitBasicFS, Always, TestOutput (
1909       [["set_e2uuid"; "/dev/sda1"; "a3a61220-882b-4f61-89f4-cf24dcc7297d"];
1910        ["get_e2uuid"; "/dev/sda1"]], "a3a61220-882b-4f61-89f4-cf24dcc7297d");
1911     InitBasicFS, Always, TestOutput (
1912       [["set_e2uuid"; "/dev/sda1"; "clear"];
1913        ["get_e2uuid"; "/dev/sda1"]], "");
1914     (* We can't predict what UUIDs will be, so just check the commands run. *)
1915     InitBasicFS, Always, TestRun (
1916       [["set_e2uuid"; "/dev/sda1"; "random"]]);
1917     InitBasicFS, Always, TestRun (
1918       [["set_e2uuid"; "/dev/sda1"; "time"]])],
1919    "set the ext2/3/4 filesystem UUID",
1920    "\
1921 This sets the ext2/3/4 filesystem UUID of the filesystem on
1922 C<device> to C<uuid>.  The format of the UUID and alternatives
1923 such as C<clear>, C<random> and C<time> are described in the
1924 L<tune2fs(8)> manpage.
1925
1926 You can use either C<guestfs_tune2fs_l> or C<guestfs_get_e2uuid>
1927 to return the existing UUID of a filesystem.");
1928
1929   ("get_e2uuid", (RString "uuid", [String "device"]), 83, [],
1930    [],
1931    "get the ext2/3/4 filesystem UUID",
1932    "\
1933 This returns the ext2/3/4 filesystem UUID of the filesystem on
1934 C<device>.");
1935
1936   ("fsck", (RInt "status", [String "fstype"; String "device"]), 84, [],
1937    [InitBasicFS, Always, TestOutputInt (
1938       [["umount"; "/dev/sda1"];
1939        ["fsck"; "ext2"; "/dev/sda1"]], 0);
1940     InitBasicFS, Always, TestOutputInt (
1941       [["umount"; "/dev/sda1"];
1942        ["zero"; "/dev/sda1"];
1943        ["fsck"; "ext2"; "/dev/sda1"]], 8)],
1944    "run the filesystem checker",
1945    "\
1946 This runs the filesystem checker (fsck) on C<device> which
1947 should have filesystem type C<fstype>.
1948
1949 The returned integer is the status.  See L<fsck(8)> for the
1950 list of status codes from C<fsck>.
1951
1952 Notes:
1953
1954 =over 4
1955
1956 =item *
1957
1958 Multiple status codes can be summed together.
1959
1960 =item *
1961
1962 A non-zero return code can mean \"success\", for example if
1963 errors have been corrected on the filesystem.
1964
1965 =item *
1966
1967 Checking or repairing NTFS volumes is not supported
1968 (by linux-ntfs).
1969
1970 =back
1971
1972 This command is entirely equivalent to running C<fsck -a -t fstype device>.");
1973
1974   ("zero", (RErr, [String "device"]), 85, [],
1975    [InitBasicFS, Always, TestOutput (
1976       [["umount"; "/dev/sda1"];
1977        ["zero"; "/dev/sda1"];
1978        ["file"; "/dev/sda1"]], "data")],
1979    "write zeroes to the device",
1980    "\
1981 This command writes zeroes over the first few blocks of C<device>.
1982
1983 How many blocks are zeroed isn't specified (but it's I<not> enough
1984 to securely wipe the device).  It should be sufficient to remove
1985 any partition tables, filesystem superblocks and so on.");
1986
1987   ("grub_install", (RErr, [String "root"; String "device"]), 86, [],
1988    [InitBasicFS, Always, TestOutputTrue (
1989       [["grub_install"; "/"; "/dev/sda1"];
1990        ["is_dir"; "/boot"]])],
1991    "install GRUB",
1992    "\
1993 This command installs GRUB (the Grand Unified Bootloader) on
1994 C<device>, with the root directory being C<root>.");
1995
1996   ("cp", (RErr, [String "src"; String "dest"]), 87, [],
1997    [InitBasicFS, Always, TestOutput (
1998       [["write_file"; "/old"; "file content"; "0"];
1999        ["cp"; "/old"; "/new"];
2000        ["cat"; "/new"]], "file content");
2001     InitBasicFS, Always, TestOutputTrue (
2002       [["write_file"; "/old"; "file content"; "0"];
2003        ["cp"; "/old"; "/new"];
2004        ["is_file"; "/old"]]);
2005     InitBasicFS, Always, TestOutput (
2006       [["write_file"; "/old"; "file content"; "0"];
2007        ["mkdir"; "/dir"];
2008        ["cp"; "/old"; "/dir/new"];
2009        ["cat"; "/dir/new"]], "file content")],
2010    "copy a file",
2011    "\
2012 This copies a file from C<src> to C<dest> where C<dest> is
2013 either a destination filename or destination directory.");
2014
2015   ("cp_a", (RErr, [String "src"; String "dest"]), 88, [],
2016    [InitBasicFS, Always, TestOutput (
2017       [["mkdir"; "/olddir"];
2018        ["mkdir"; "/newdir"];
2019        ["write_file"; "/olddir/file"; "file content"; "0"];
2020        ["cp_a"; "/olddir"; "/newdir"];
2021        ["cat"; "/newdir/olddir/file"]], "file content")],
2022    "copy a file or directory recursively",
2023    "\
2024 This copies a file or directory from C<src> to C<dest>
2025 recursively using the C<cp -a> command.");
2026
2027   ("mv", (RErr, [String "src"; String "dest"]), 89, [],
2028    [InitBasicFS, Always, TestOutput (
2029       [["write_file"; "/old"; "file content"; "0"];
2030        ["mv"; "/old"; "/new"];
2031        ["cat"; "/new"]], "file content");
2032     InitBasicFS, Always, TestOutputFalse (
2033       [["write_file"; "/old"; "file content"; "0"];
2034        ["mv"; "/old"; "/new"];
2035        ["is_file"; "/old"]])],
2036    "move a file",
2037    "\
2038 This moves a file from C<src> to C<dest> where C<dest> is
2039 either a destination filename or destination directory.");
2040
2041   ("drop_caches", (RErr, [Int "whattodrop"]), 90, [],
2042    [InitEmpty, Always, TestRun (
2043       [["drop_caches"; "3"]])],
2044    "drop kernel page cache, dentries and inodes",
2045    "\
2046 This instructs the guest kernel to drop its page cache,
2047 and/or dentries and inode caches.  The parameter C<whattodrop>
2048 tells the kernel what precisely to drop, see
2049 L<http://linux-mm.org/Drop_Caches>
2050
2051 Setting C<whattodrop> to 3 should drop everything.
2052
2053 This automatically calls L<sync(2)> before the operation,
2054 so that the maximum guest memory is freed.");
2055
2056   ("dmesg", (RString "kmsgs", []), 91, [],
2057    [InitEmpty, Always, TestRun (
2058       [["dmesg"]])],
2059    "return kernel messages",
2060    "\
2061 This returns the kernel messages (C<dmesg> output) from
2062 the guest kernel.  This is sometimes useful for extended
2063 debugging of problems.
2064
2065 Another way to get the same information is to enable
2066 verbose messages with C<guestfs_set_verbose> or by setting
2067 the environment variable C<LIBGUESTFS_DEBUG=1> before
2068 running the program.");
2069
2070   ("ping_daemon", (RErr, []), 92, [],
2071    [InitEmpty, Always, TestRun (
2072       [["ping_daemon"]])],
2073    "ping the guest daemon",
2074    "\
2075 This is a test probe into the guestfs daemon running inside
2076 the qemu subprocess.  Calling this function checks that the
2077 daemon responds to the ping message, without affecting the daemon
2078 or attached block device(s) in any other way.");
2079
2080   ("equal", (RBool "equality", [String "file1"; String "file2"]), 93, [],
2081    [InitBasicFS, Always, TestOutputTrue (
2082       [["write_file"; "/file1"; "contents of a file"; "0"];
2083        ["cp"; "/file1"; "/file2"];
2084        ["equal"; "/file1"; "/file2"]]);
2085     InitBasicFS, Always, TestOutputFalse (
2086       [["write_file"; "/file1"; "contents of a file"; "0"];
2087        ["write_file"; "/file2"; "contents of another file"; "0"];
2088        ["equal"; "/file1"; "/file2"]]);
2089     InitBasicFS, Always, TestLastFail (
2090       [["equal"; "/file1"; "/file2"]])],
2091    "test if two files have equal contents",
2092    "\
2093 This compares the two files C<file1> and C<file2> and returns
2094 true if their content is exactly equal, or false otherwise.
2095
2096 The external L<cmp(1)> program is used for the comparison.");
2097
2098   ("strings", (RStringList "stringsout", [String "path"]), 94, [ProtocolLimitWarning],
2099    [InitBasicFS, Always, TestOutputList (
2100       [["write_file"; "/new"; "hello\nworld\n"; "0"];
2101        ["strings"; "/new"]], ["hello"; "world"]);
2102     InitBasicFS, Always, TestOutputList (
2103       [["touch"; "/new"];
2104        ["strings"; "/new"]], [])],
2105    "print the printable strings in a file",
2106    "\
2107 This runs the L<strings(1)> command on a file and returns
2108 the list of printable strings found.");
2109
2110   ("strings_e", (RStringList "stringsout", [String "encoding"; String "path"]), 95, [ProtocolLimitWarning],
2111    [InitBasicFS, Always, TestOutputList (
2112       [["write_file"; "/new"; "hello\nworld\n"; "0"];
2113        ["strings_e"; "b"; "/new"]], []);
2114     InitBasicFS, Disabled, TestOutputList (
2115       [["write_file"; "/new"; "\000h\000e\000l\000l\000o\000\n\000w\000o\000r\000l\000d\000\n"; "24"];
2116        ["strings_e"; "b"; "/new"]], ["hello"; "world"])],
2117    "print the printable strings in a file",
2118    "\
2119 This is like the C<guestfs_strings> command, but allows you to
2120 specify the encoding.
2121
2122 See the L<strings(1)> manpage for the full list of encodings.
2123
2124 Commonly useful encodings are C<l> (lower case L) which will
2125 show strings inside Windows/x86 files.
2126
2127 The returned strings are transcoded to UTF-8.");
2128
2129   ("hexdump", (RString "dump", [String "path"]), 96, [ProtocolLimitWarning],
2130    [InitBasicFS, Always, TestOutput (
2131       [["write_file"; "/new"; "hello\nworld\n"; "12"];
2132        ["hexdump"; "/new"]], "00000000  68 65 6c 6c 6f 0a 77 6f  72 6c 64 0a              |hello.world.|\n0000000c\n")],
2133    "dump a file in hexadecimal",
2134    "\
2135 This runs C<hexdump -C> on the given C<path>.  The result is
2136 the human-readable, canonical hex dump of the file.");
2137
2138   ("zerofree", (RErr, [String "device"]), 97, [],
2139    [InitNone, Always, TestOutput (
2140       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
2141        ["mkfs"; "ext3"; "/dev/sda1"];
2142        ["mount"; "/dev/sda1"; "/"];
2143        ["write_file"; "/new"; "test file"; "0"];
2144        ["umount"; "/dev/sda1"];
2145        ["zerofree"; "/dev/sda1"];
2146        ["mount"; "/dev/sda1"; "/"];
2147        ["cat"; "/new"]], "test file")],
2148    "zero unused inodes and disk blocks on ext2/3 filesystem",
2149    "\
2150 This runs the I<zerofree> program on C<device>.  This program
2151 claims to zero unused inodes and disk blocks on an ext2/3
2152 filesystem, thus making it possible to compress the filesystem
2153 more effectively.
2154
2155 You should B<not> run this program if the filesystem is
2156 mounted.
2157
2158 It is possible that using this program can damage the filesystem
2159 or data on the filesystem.");
2160
2161   ("pvresize", (RErr, [String "device"]), 98, [],
2162    [],
2163    "resize an LVM physical volume",
2164    "\
2165 This resizes (expands or shrinks) an existing LVM physical
2166 volume to match the new size of the underlying device.");
2167
2168   ("sfdisk_N", (RErr, [String "device"; Int "n";
2169                        Int "cyls"; Int "heads"; Int "sectors";
2170                        String "line"]), 99, [DangerWillRobinson],
2171    [],
2172    "modify a single partition on a block device",
2173    "\
2174 This runs L<sfdisk(8)> option to modify just the single
2175 partition C<n> (note: C<n> counts from 1).
2176
2177 For other parameters, see C<guestfs_sfdisk>.  You should usually
2178 pass C<0> for the cyls/heads/sectors parameters.");
2179
2180   ("sfdisk_l", (RString "partitions", [String "device"]), 100, [],
2181    [],
2182    "display the partition table",
2183    "\
2184 This displays the partition table on C<device>, in the
2185 human-readable output of the L<sfdisk(8)> command.  It is
2186 not intended to be parsed.");
2187
2188   ("sfdisk_kernel_geometry", (RString "partitions", [String "device"]), 101, [],
2189    [],
2190    "display the kernel geometry",
2191    "\
2192 This displays the kernel's idea of the geometry of C<device>.
2193
2194 The result is in human-readable format, and not designed to
2195 be parsed.");
2196
2197   ("sfdisk_disk_geometry", (RString "partitions", [String "device"]), 102, [],
2198    [],
2199    "display the disk geometry from the partition table",
2200    "\
2201 This displays the disk geometry of C<device> read from the
2202 partition table.  Especially in the case where the underlying
2203 block device has been resized, this can be different from the
2204 kernel's idea of the geometry (see C<guestfs_sfdisk_kernel_geometry>).
2205
2206 The result is in human-readable format, and not designed to
2207 be parsed.");
2208
2209   ("vg_activate_all", (RErr, [Bool "activate"]), 103, [],
2210    [],
2211    "activate or deactivate all volume groups",
2212    "\
2213 This command activates or (if C<activate> is false) deactivates
2214 all logical volumes in all volume groups.
2215 If activated, then they are made known to the
2216 kernel, ie. they appear as C</dev/mapper> devices.  If deactivated,
2217 then those devices disappear.
2218
2219 This command is the same as running C<vgchange -a y|n>");
2220
2221   ("vg_activate", (RErr, [Bool "activate"; StringList "volgroups"]), 104, [],
2222    [],
2223    "activate or deactivate some volume groups",
2224    "\
2225 This command activates or (if C<activate> is false) deactivates
2226 all logical volumes in the listed volume groups C<volgroups>.
2227 If activated, then they are made known to the
2228 kernel, ie. they appear as C</dev/mapper> devices.  If deactivated,
2229 then those devices disappear.
2230
2231 This command is the same as running C<vgchange -a y|n volgroups...>
2232
2233 Note that if C<volgroups> is an empty list then B<all> volume groups
2234 are activated or deactivated.");
2235
2236   ("lvresize", (RErr, [String "device"; Int "mbytes"]), 105, [],
2237    [InitNone, Always, TestOutput (
2238     [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
2239      ["pvcreate"; "/dev/sda1"];
2240      ["vgcreate"; "VG"; "/dev/sda1"];
2241      ["lvcreate"; "LV"; "VG"; "10"];
2242      ["mkfs"; "ext2"; "/dev/VG/LV"];
2243      ["mount"; "/dev/VG/LV"; "/"];
2244      ["write_file"; "/new"; "test content"; "0"];
2245      ["umount"; "/"];
2246      ["lvresize"; "/dev/VG/LV"; "20"];
2247      ["e2fsck_f"; "/dev/VG/LV"];
2248      ["resize2fs"; "/dev/VG/LV"];
2249      ["mount"; "/dev/VG/LV"; "/"];
2250      ["cat"; "/new"]], "test content")],
2251    "resize an LVM logical volume",
2252    "\
2253 This resizes (expands or shrinks) an existing LVM logical
2254 volume to C<mbytes>.  When reducing, data in the reduced part
2255 is lost.");
2256
2257   ("resize2fs", (RErr, [String "device"]), 106, [],
2258    [], (* lvresize tests this *)
2259    "resize an ext2/ext3 filesystem",
2260    "\
2261 This resizes an ext2 or ext3 filesystem to match the size of
2262 the underlying device.
2263
2264 I<Note:> It is sometimes required that you run C<guestfs_e2fsck_f>
2265 on the C<device> before calling this command.  For unknown reasons
2266 C<resize2fs> sometimes gives an error about this and sometimes not.
2267 In any case, it is always safe to call C<guestfs_e2fsck_f> before
2268 calling this function.");
2269
2270   ("find", (RStringList "names", [String "directory"]), 107, [],
2271    [InitBasicFS, Always, TestOutputList (
2272       [["find"; "/"]], ["lost+found"]);
2273     InitBasicFS, Always, TestOutputList (
2274       [["touch"; "/a"];
2275        ["mkdir"; "/b"];
2276        ["touch"; "/b/c"];
2277        ["find"; "/"]], ["a"; "b"; "b/c"; "lost+found"]);
2278     InitBasicFS, Always, TestOutputList (
2279       [["mkdir_p"; "/a/b/c"];
2280        ["touch"; "/a/b/c/d"];
2281        ["find"; "/a/b/"]], ["c"; "c/d"])],
2282    "find all files and directories",
2283    "\
2284 This command lists out all files and directories, recursively,
2285 starting at C<directory>.  It is essentially equivalent to
2286 running the shell command C<find directory -print> but some
2287 post-processing happens on the output, described below.
2288
2289 This returns a list of strings I<without any prefix>.  Thus
2290 if the directory structure was:
2291
2292  /tmp/a
2293  /tmp/b
2294  /tmp/c/d
2295
2296 then the returned list from C<guestfs_find> C</tmp> would be
2297 4 elements:
2298
2299  a
2300  b
2301  c
2302  c/d
2303
2304 If C<directory> is not a directory, then this command returns
2305 an error.
2306
2307 The returned list is sorted.");
2308
2309   ("e2fsck_f", (RErr, [String "device"]), 108, [],
2310    [], (* lvresize tests this *)
2311    "check an ext2/ext3 filesystem",
2312    "\
2313 This runs C<e2fsck -p -f device>, ie. runs the ext2/ext3
2314 filesystem checker on C<device>, noninteractively (C<-p>),
2315 even if the filesystem appears to be clean (C<-f>).
2316
2317 This command is only needed because of C<guestfs_resize2fs>
2318 (q.v.).  Normally you should use C<guestfs_fsck>.");
2319
2320   ("sleep", (RErr, [Int "secs"]), 109, [],
2321    [InitNone, Always, TestRun (
2322     [["sleep"; "1"]])],
2323    "sleep for some seconds",
2324    "\
2325 Sleep for C<secs> seconds.");
2326
2327   ("ntfs_3g_probe", (RInt "status", [Bool "rw"; String "device"]), 110, [],
2328    [InitNone, Always, TestOutputInt (
2329       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
2330        ["mkfs"; "ntfs"; "/dev/sda1"];
2331        ["ntfs_3g_probe"; "true"; "/dev/sda1"]], 0);
2332     InitNone, Always, TestOutputInt (
2333       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
2334        ["mkfs"; "ext2"; "/dev/sda1"];
2335        ["ntfs_3g_probe"; "true"; "/dev/sda1"]], 12)],
2336    "probe NTFS volume",
2337    "\
2338 This command runs the L<ntfs-3g.probe(8)> command which probes
2339 an NTFS C<device> for mountability.  (Not all NTFS volumes can
2340 be mounted read-write, and some cannot be mounted at all).
2341
2342 C<rw> is a boolean flag.  Set it to true if you want to test
2343 if the volume can be mounted read-write.  Set it to false if
2344 you want to test if the volume can be mounted read-only.
2345
2346 The return value is an integer which C<0> if the operation
2347 would succeed, or some non-zero value documented in the
2348 L<ntfs-3g.probe(8)> manual page.");
2349
2350   ("sh", (RString "output", [String "command"]), 111, [],
2351    [], (* XXX needs tests *)
2352    "run a command via the shell",
2353    "\
2354 This call runs a command from the guest filesystem via the
2355 guest's C</bin/sh>.
2356
2357 This is like C<guestfs_command>, but passes the command to:
2358
2359  /bin/sh -c \"command\"
2360
2361 Depending on the guest's shell, this usually results in
2362 wildcards being expanded, shell expressions being interpolated
2363 and so on.
2364
2365 All the provisos about C<guestfs_command> apply to this call.");
2366
2367   ("sh_lines", (RStringList "lines", [String "command"]), 112, [],
2368    [], (* XXX needs tests *)
2369    "run a command via the shell returning lines",
2370    "\
2371 This is the same as C<guestfs_sh>, but splits the result
2372 into a list of lines.
2373
2374 See also: C<guestfs_command_lines>");
2375
2376   ("glob_expand", (RStringList "paths", [String "pattern"]), 113, [],
2377    [InitBasicFS, Always, TestOutputList (
2378       [["mkdir_p"; "/a/b/c"];
2379        ["touch"; "/a/b/c/d"];
2380        ["touch"; "/a/b/c/e"];
2381        ["glob_expand"; "/a/b/c/*"]], ["/a/b/c/d"; "/a/b/c/e"]);
2382     InitBasicFS, Always, TestOutputList (
2383       [["mkdir_p"; "/a/b/c"];
2384        ["touch"; "/a/b/c/d"];
2385        ["touch"; "/a/b/c/e"];
2386        ["glob_expand"; "/a/*/c/*"]], ["/a/b/c/d"; "/a/b/c/e"]);
2387     InitBasicFS, Always, TestOutputList (
2388       [["mkdir_p"; "/a/b/c"];
2389        ["touch"; "/a/b/c/d"];
2390        ["touch"; "/a/b/c/e"];
2391        ["glob_expand"; "/a/*/x/*"]], [])],
2392    "expand a wildcard path",
2393    "\
2394 This command searches for all the pathnames matching
2395 C<pattern> according to the wildcard expansion rules
2396 used by the shell.
2397
2398 If no paths match, then this returns an empty list
2399 (note: not an error).
2400
2401 It is just a wrapper around the C L<glob(3)> function
2402 with flags C<GLOB_MARK|GLOB_BRACE>.
2403 See that manual page for more details.");
2404
2405 ]
2406
2407 let all_functions = non_daemon_functions @ daemon_functions
2408
2409 (* In some places we want the functions to be displayed sorted
2410  * alphabetically, so this is useful:
2411  *)
2412 let all_functions_sorted =
2413   List.sort (fun (n1,_,_,_,_,_,_) (n2,_,_,_,_,_,_) ->
2414                compare n1 n2) all_functions
2415
2416 (* Column names and types from LVM PVs/VGs/LVs. *)
2417 let pv_cols = [
2418   "pv_name", `String;
2419   "pv_uuid", `UUID;
2420   "pv_fmt", `String;
2421   "pv_size", `Bytes;
2422   "dev_size", `Bytes;
2423   "pv_free", `Bytes;
2424   "pv_used", `Bytes;
2425   "pv_attr", `String (* XXX *);
2426   "pv_pe_count", `Int;
2427   "pv_pe_alloc_count", `Int;
2428   "pv_tags", `String;
2429   "pe_start", `Bytes;
2430   "pv_mda_count", `Int;
2431   "pv_mda_free", `Bytes;
2432 (* Not in Fedora 10:
2433   "pv_mda_size", `Bytes;
2434 *)
2435 ]
2436 let vg_cols = [
2437   "vg_name", `String;
2438   "vg_uuid", `UUID;
2439   "vg_fmt", `String;
2440   "vg_attr", `String (* XXX *);
2441   "vg_size", `Bytes;
2442   "vg_free", `Bytes;
2443   "vg_sysid", `String;
2444   "vg_extent_size", `Bytes;
2445   "vg_extent_count", `Int;
2446   "vg_free_count", `Int;
2447   "max_lv", `Int;
2448   "max_pv", `Int;
2449   "pv_count", `Int;
2450   "lv_count", `Int;
2451   "snap_count", `Int;
2452   "vg_seqno", `Int;
2453   "vg_tags", `String;
2454   "vg_mda_count", `Int;
2455   "vg_mda_free", `Bytes;
2456 (* Not in Fedora 10:
2457   "vg_mda_size", `Bytes;
2458 *)
2459 ]
2460 let lv_cols = [
2461   "lv_name", `String;
2462   "lv_uuid", `UUID;
2463   "lv_attr", `String (* XXX *);
2464   "lv_major", `Int;
2465   "lv_minor", `Int;
2466   "lv_kernel_major", `Int;
2467   "lv_kernel_minor", `Int;
2468   "lv_size", `Bytes;
2469   "seg_count", `Int;
2470   "origin", `String;
2471   "snap_percent", `OptPercent;
2472   "copy_percent", `OptPercent;
2473   "move_pv", `String;
2474   "lv_tags", `String;
2475   "mirror_log", `String;
2476   "modules", `String;
2477 ]
2478
2479 (* Column names and types from stat structures.
2480  * NB. Can't use things like 'st_atime' because glibc header files
2481  * define some of these as macros.  Ugh.
2482  *)
2483 let stat_cols = [
2484   "dev", `Int;
2485   "ino", `Int;
2486   "mode", `Int;
2487   "nlink", `Int;
2488   "uid", `Int;
2489   "gid", `Int;
2490   "rdev", `Int;
2491   "size", `Int;
2492   "blksize", `Int;
2493   "blocks", `Int;
2494   "atime", `Int;
2495   "mtime", `Int;
2496   "ctime", `Int;
2497 ]
2498 let statvfs_cols = [
2499   "bsize", `Int;
2500   "frsize", `Int;
2501   "blocks", `Int;
2502   "bfree", `Int;
2503   "bavail", `Int;
2504   "files", `Int;
2505   "ffree", `Int;
2506   "favail", `Int;
2507   "fsid", `Int;
2508   "flag", `Int;
2509   "namemax", `Int;
2510 ]
2511
2512 (* Used for testing language bindings. *)
2513 type callt =
2514   | CallString of string
2515   | CallOptString of string option
2516   | CallStringList of string list
2517   | CallInt of int
2518   | CallBool of bool
2519
2520 (* Useful functions.
2521  * Note we don't want to use any external OCaml libraries which
2522  * makes this a bit harder than it should be.
2523  *)
2524 let failwithf fs = ksprintf failwith fs
2525
2526 let replace_char s c1 c2 =
2527   let s2 = String.copy s in
2528   let r = ref false in
2529   for i = 0 to String.length s2 - 1 do
2530     if String.unsafe_get s2 i = c1 then (
2531       String.unsafe_set s2 i c2;
2532       r := true
2533     )
2534   done;
2535   if not !r then s else s2
2536
2537 let isspace c =
2538   c = ' '
2539   (* || c = '\f' *) || c = '\n' || c = '\r' || c = '\t' (* || c = '\v' *)
2540
2541 let triml ?(test = isspace) str =
2542   let i = ref 0 in
2543   let n = ref (String.length str) in
2544   while !n > 0 && test str.[!i]; do
2545     decr n;
2546     incr i
2547   done;
2548   if !i = 0 then str
2549   else String.sub str !i !n
2550
2551 let trimr ?(test = isspace) str =
2552   let n = ref (String.length str) in
2553   while !n > 0 && test str.[!n-1]; do
2554     decr n
2555   done;
2556   if !n = String.length str then str
2557   else String.sub str 0 !n
2558
2559 let trim ?(test = isspace) str =
2560   trimr ~test (triml ~test str)
2561
2562 let rec find s sub =
2563   let len = String.length s in
2564   let sublen = String.length sub in
2565   let rec loop i =
2566     if i <= len-sublen then (
2567       let rec loop2 j =
2568         if j < sublen then (
2569           if s.[i+j] = sub.[j] then loop2 (j+1)
2570           else -1
2571         ) else
2572           i (* found *)
2573       in
2574       let r = loop2 0 in
2575       if r = -1 then loop (i+1) else r
2576     ) else
2577       -1 (* not found *)
2578   in
2579   loop 0
2580
2581 let rec replace_str s s1 s2 =
2582   let len = String.length s in
2583   let sublen = String.length s1 in
2584   let i = find s s1 in
2585   if i = -1 then s
2586   else (
2587     let s' = String.sub s 0 i in
2588     let s'' = String.sub s (i+sublen) (len-i-sublen) in
2589     s' ^ s2 ^ replace_str s'' s1 s2
2590   )
2591
2592 let rec string_split sep str =
2593   let len = String.length str in
2594   let seplen = String.length sep in
2595   let i = find str sep in
2596   if i = -1 then [str]
2597   else (
2598     let s' = String.sub str 0 i in
2599     let s'' = String.sub str (i+seplen) (len-i-seplen) in
2600     s' :: string_split sep s''
2601   )
2602
2603 let files_equal n1 n2 =
2604   let cmd = sprintf "cmp -s %s %s" (Filename.quote n1) (Filename.quote n2) in
2605   match Sys.command cmd with
2606   | 0 -> true
2607   | 1 -> false
2608   | i -> failwithf "%s: failed with error code %d" cmd i
2609
2610 let rec find_map f = function
2611   | [] -> raise Not_found
2612   | x :: xs ->
2613       match f x with
2614       | Some y -> y
2615       | None -> find_map f xs
2616
2617 let iteri f xs =
2618   let rec loop i = function
2619     | [] -> ()
2620     | x :: xs -> f i x; loop (i+1) xs
2621   in
2622   loop 0 xs
2623
2624 let mapi f xs =
2625   let rec loop i = function
2626     | [] -> []
2627     | x :: xs -> let r = f i x in r :: loop (i+1) xs
2628   in
2629   loop 0 xs
2630
2631 let name_of_argt = function
2632   | String n | OptString n | StringList n | Bool n | Int n
2633   | FileIn n | FileOut n -> n
2634
2635 let seq_of_test = function
2636   | TestRun s | TestOutput (s, _) | TestOutputList (s, _)
2637   | TestOutputListOfDevices (s, _)
2638   | TestOutputInt (s, _) | TestOutputTrue s | TestOutputFalse s
2639   | TestOutputLength (s, _) | TestOutputStruct (s, _)
2640   | TestLastFail s -> s
2641
2642 (* Check function names etc. for consistency. *)
2643 let check_functions () =
2644   let contains_uppercase str =
2645     let len = String.length str in
2646     let rec loop i =
2647       if i >= len then false
2648       else (
2649         let c = str.[i] in
2650         if c >= 'A' && c <= 'Z' then true
2651         else loop (i+1)
2652       )
2653     in
2654     loop 0
2655   in
2656
2657   (* Check function names. *)
2658   List.iter (
2659     fun (name, _, _, _, _, _, _) ->
2660       if String.length name >= 7 && String.sub name 0 7 = "guestfs" then
2661         failwithf "function name %s does not need 'guestfs' prefix" name;
2662       if name = "" then
2663         failwithf "function name is empty";
2664       if name.[0] < 'a' || name.[0] > 'z' then
2665         failwithf "function name %s must start with lowercase a-z" name;
2666       if String.contains name '-' then
2667         failwithf "function name %s should not contain '-', use '_' instead."
2668           name
2669   ) all_functions;
2670
2671   (* Check function parameter/return names. *)
2672   List.iter (
2673     fun (name, style, _, _, _, _, _) ->
2674       let check_arg_ret_name n =
2675         if contains_uppercase n then
2676           failwithf "%s param/ret %s should not contain uppercase chars"
2677             name n;
2678         if String.contains n '-' || String.contains n '_' then
2679           failwithf "%s param/ret %s should not contain '-' or '_'"
2680             name n;
2681         if n = "value" then
2682           failwithf "%s has a param/ret called 'value', which causes conflicts in the OCaml bindings, use something like 'val' or a more descriptive name" name;
2683         if n = "int" || n = "char" || n = "short" || n = "long" then
2684           failwithf "%s has a param/ret which conflicts with a C type (eg. 'int', 'char' etc.)" name;
2685         if n = "i" then
2686           failwithf "%s has a param/ret called 'i', which will cause some conflicts in the generated code" name;
2687         if n = "argv" || n = "args" then
2688           failwithf "%s has a param/ret called 'argv' or 'args', which will cause some conflicts in the generated code" name
2689       in
2690
2691       (match fst style with
2692        | RErr -> ()
2693        | RInt n | RInt64 n | RBool n | RConstString n | RString n
2694        | RStringList n | RPVList n | RVGList n | RLVList n
2695        | RStat n | RStatVFS n
2696        | RHashtable n ->
2697            check_arg_ret_name n
2698        | RIntBool (n,m) ->
2699            check_arg_ret_name n;
2700            check_arg_ret_name m
2701       );
2702       List.iter (fun arg -> check_arg_ret_name (name_of_argt arg)) (snd style)
2703   ) all_functions;
2704
2705   (* Check short descriptions. *)
2706   List.iter (
2707     fun (name, _, _, _, _, shortdesc, _) ->
2708       if shortdesc.[0] <> Char.lowercase shortdesc.[0] then
2709         failwithf "short description of %s should begin with lowercase." name;
2710       let c = shortdesc.[String.length shortdesc-1] in
2711       if c = '\n' || c = '.' then
2712         failwithf "short description of %s should not end with . or \\n." name
2713   ) all_functions;
2714
2715   (* Check long dscriptions. *)
2716   List.iter (
2717     fun (name, _, _, _, _, _, longdesc) ->
2718       if longdesc.[String.length longdesc-1] = '\n' then
2719         failwithf "long description of %s should not end with \\n." name
2720   ) all_functions;
2721
2722   (* Check proc_nrs. *)
2723   List.iter (
2724     fun (name, _, proc_nr, _, _, _, _) ->
2725       if proc_nr <= 0 then
2726         failwithf "daemon function %s should have proc_nr > 0" name
2727   ) daemon_functions;
2728
2729   List.iter (
2730     fun (name, _, proc_nr, _, _, _, _) ->
2731       if proc_nr <> -1 then
2732         failwithf "non-daemon function %s should have proc_nr -1" name
2733   ) non_daemon_functions;
2734
2735   let proc_nrs =
2736     List.map (fun (name, _, proc_nr, _, _, _, _) -> name, proc_nr)
2737       daemon_functions in
2738   let proc_nrs =
2739     List.sort (fun (_,nr1) (_,nr2) -> compare nr1 nr2) proc_nrs in
2740   let rec loop = function
2741     | [] -> ()
2742     | [_] -> ()
2743     | (name1,nr1) :: ((name2,nr2) :: _ as rest) when nr1 < nr2 ->
2744         loop rest
2745     | (name1,nr1) :: (name2,nr2) :: _ ->
2746         failwithf "%s and %s have conflicting procedure numbers (%d, %d)"
2747           name1 name2 nr1 nr2
2748   in
2749   loop proc_nrs;
2750
2751   (* Check tests. *)
2752   List.iter (
2753     function
2754       (* Ignore functions that have no tests.  We generate a
2755        * warning when the user does 'make check' instead.
2756        *)
2757     | name, _, _, _, [], _, _ -> ()
2758     | name, _, _, _, tests, _, _ ->
2759         let funcs =
2760           List.map (
2761             fun (_, _, test) ->
2762               match seq_of_test test with
2763               | [] ->
2764                   failwithf "%s has a test containing an empty sequence" name
2765               | cmds -> List.map List.hd cmds
2766           ) tests in
2767         let funcs = List.flatten funcs in
2768
2769         let tested = List.mem name funcs in
2770
2771         if not tested then
2772           failwithf "function %s has tests but does not test itself" name
2773   ) all_functions
2774
2775 (* 'pr' prints to the current output file. *)
2776 let chan = ref stdout
2777 let pr fs = ksprintf (output_string !chan) fs
2778
2779 (* Generate a header block in a number of standard styles. *)
2780 type comment_style = CStyle | HashStyle | OCamlStyle | HaskellStyle
2781 type license = GPLv2 | LGPLv2
2782
2783 let generate_header comment license =
2784   let c = match comment with
2785     | CStyle ->     pr "/* "; " *"
2786     | HashStyle ->  pr "# ";  "#"
2787     | OCamlStyle -> pr "(* "; " *"
2788     | HaskellStyle -> pr "{- "; "  " in
2789   pr "libguestfs generated file\n";
2790   pr "%s WARNING: THIS FILE IS GENERATED BY 'src/generator.ml'.\n" c;
2791   pr "%s ANY CHANGES YOU MAKE TO THIS FILE WILL BE LOST.\n" c;
2792   pr "%s\n" c;
2793   pr "%s Copyright (C) 2009 Red Hat Inc.\n" c;
2794   pr "%s\n" c;
2795   (match license with
2796    | GPLv2 ->
2797        pr "%s This program is free software; you can redistribute it and/or modify\n" c;
2798        pr "%s it under the terms of the GNU General Public License as published by\n" c;
2799        pr "%s the Free Software Foundation; either version 2 of the License, or\n" c;
2800        pr "%s (at your option) any later version.\n" c;
2801        pr "%s\n" c;
2802        pr "%s This program is distributed in the hope that it will be useful,\n" c;
2803        pr "%s but WITHOUT ANY WARRANTY; without even the implied warranty of\n" c;
2804        pr "%s MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\n" c;
2805        pr "%s GNU General Public License for more details.\n" c;
2806        pr "%s\n" c;
2807        pr "%s You should have received a copy of the GNU General Public License along\n" c;
2808        pr "%s with this program; if not, write to the Free Software Foundation, Inc.,\n" c;
2809        pr "%s 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.\n" c;
2810
2811    | LGPLv2 ->
2812        pr "%s This library is free software; you can redistribute it and/or\n" c;
2813        pr "%s modify it under the terms of the GNU Lesser General Public\n" c;
2814        pr "%s License as published by the Free Software Foundation; either\n" c;
2815        pr "%s version 2 of the License, or (at your option) any later version.\n" c;
2816        pr "%s\n" c;
2817        pr "%s This library is distributed in the hope that it will be useful,\n" c;
2818        pr "%s but WITHOUT ANY WARRANTY; without even the implied warranty of\n" c;
2819        pr "%s MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU\n" c;
2820        pr "%s Lesser General Public License for more details.\n" c;
2821        pr "%s\n" c;
2822        pr "%s You should have received a copy of the GNU Lesser General Public\n" c;
2823        pr "%s License along with this library; if not, write to the Free Software\n" c;
2824        pr "%s Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA\n" c;
2825   );
2826   (match comment with
2827    | CStyle -> pr " */\n"
2828    | HashStyle -> ()
2829    | OCamlStyle -> pr " *)\n"
2830    | HaskellStyle -> pr "-}\n"
2831   );
2832   pr "\n"
2833
2834 (* Start of main code generation functions below this line. *)
2835
2836 (* Generate the pod documentation for the C API. *)
2837 let rec generate_actions_pod () =
2838   List.iter (
2839     fun (shortname, style, _, flags, _, _, longdesc) ->
2840       if not (List.mem NotInDocs flags) then (
2841         let name = "guestfs_" ^ shortname in
2842         pr "=head2 %s\n\n" name;
2843         pr " ";
2844         generate_prototype ~extern:false ~handle:"handle" name style;
2845         pr "\n\n";
2846         pr "%s\n\n" longdesc;
2847         (match fst style with
2848          | RErr ->
2849              pr "This function returns 0 on success or -1 on error.\n\n"
2850          | RInt _ ->
2851              pr "On error this function returns -1.\n\n"
2852          | RInt64 _ ->
2853              pr "On error this function returns -1.\n\n"
2854          | RBool _ ->
2855              pr "This function returns a C truth value on success or -1 on error.\n\n"
2856          | RConstString _ ->
2857              pr "This function returns a string, or NULL on error.
2858 The string is owned by the guest handle and must I<not> be freed.\n\n"
2859          | RString _ ->
2860              pr "This function returns a string, or NULL on error.
2861 I<The caller must free the returned string after use>.\n\n"
2862          | RStringList _ ->
2863              pr "This function returns a NULL-terminated array of strings
2864 (like L<environ(3)>), or NULL if there was an error.
2865 I<The caller must free the strings and the array after use>.\n\n"
2866          | RIntBool _ ->
2867              pr "This function returns a C<struct guestfs_int_bool *>,
2868 or NULL if there was an error.
2869 I<The caller must call C<guestfs_free_int_bool> after use>.\n\n"
2870          | RPVList _ ->
2871              pr "This function returns a C<struct guestfs_lvm_pv_list *>
2872 (see E<lt>guestfs-structs.hE<gt>),
2873 or NULL if there was an error.
2874 I<The caller must call C<guestfs_free_lvm_pv_list> after use>.\n\n"
2875          | RVGList _ ->
2876              pr "This function returns a C<struct guestfs_lvm_vg_list *>
2877 (see E<lt>guestfs-structs.hE<gt>),
2878 or NULL if there was an error.
2879 I<The caller must call C<guestfs_free_lvm_vg_list> after use>.\n\n"
2880          | RLVList _ ->
2881              pr "This function returns a C<struct guestfs_lvm_lv_list *>
2882 (see E<lt>guestfs-structs.hE<gt>),
2883 or NULL if there was an error.
2884 I<The caller must call C<guestfs_free_lvm_lv_list> after use>.\n\n"
2885          | RStat _ ->
2886              pr "This function returns a C<struct guestfs_stat *>
2887 (see L<stat(2)> and E<lt>guestfs-structs.hE<gt>),
2888 or NULL if there was an error.
2889 I<The caller must call C<free> after use>.\n\n"
2890          | RStatVFS _ ->
2891              pr "This function returns a C<struct guestfs_statvfs *>
2892 (see L<statvfs(2)> and E<lt>guestfs-structs.hE<gt>),
2893 or NULL if there was an error.
2894 I<The caller must call C<free> after use>.\n\n"
2895          | RHashtable _ ->
2896              pr "This function returns a NULL-terminated array of
2897 strings, or NULL if there was an error.
2898 The array of strings will always have length C<2n+1>, where
2899 C<n> keys and values alternate, followed by the trailing NULL entry.
2900 I<The caller must free the strings and the array after use>.\n\n"
2901         );
2902         if List.mem ProtocolLimitWarning flags then
2903           pr "%s\n\n" protocol_limit_warning;
2904         if List.mem DangerWillRobinson flags then
2905           pr "%s\n\n" danger_will_robinson
2906       )
2907   ) all_functions_sorted
2908
2909 and generate_structs_pod () =
2910   (* LVM structs documentation. *)
2911   List.iter (
2912     fun (typ, cols) ->
2913       pr "=head2 guestfs_lvm_%s\n" typ;
2914       pr "\n";
2915       pr " struct guestfs_lvm_%s {\n" typ;
2916       List.iter (
2917         function
2918         | name, `String -> pr "  char *%s;\n" name
2919         | name, `UUID ->
2920             pr "  /* The next field is NOT nul-terminated, be careful when printing it: */\n";
2921             pr "  char %s[32];\n" name
2922         | name, `Bytes -> pr "  uint64_t %s;\n" name
2923         | name, `Int -> pr "  int64_t %s;\n" name
2924         | name, `OptPercent ->
2925             pr "  /* The next field is [0..100] or -1 meaning 'not present': */\n";
2926             pr "  float %s;\n" name
2927       ) cols;
2928       pr " \n";
2929       pr " struct guestfs_lvm_%s_list {\n" typ;
2930       pr "   uint32_t len; /* Number of elements in list. */\n";
2931       pr "   struct guestfs_lvm_%s *val; /* Elements. */\n" typ;
2932       pr " };\n";
2933       pr " \n";
2934       pr " void guestfs_free_lvm_%s_list (struct guestfs_free_lvm_%s_list *);\n"
2935         typ typ;
2936       pr "\n"
2937   ) ["pv", pv_cols; "vg", vg_cols; "lv", lv_cols]
2938
2939 (* Generate the protocol (XDR) file, 'guestfs_protocol.x' and
2940  * indirectly 'guestfs_protocol.h' and 'guestfs_protocol.c'.
2941  *
2942  * We have to use an underscore instead of a dash because otherwise
2943  * rpcgen generates incorrect code.
2944  *
2945  * This header is NOT exported to clients, but see also generate_structs_h.
2946  *)
2947 and generate_xdr () =
2948   generate_header CStyle LGPLv2;
2949
2950   (* This has to be defined to get around a limitation in Sun's rpcgen. *)
2951   pr "typedef string str<>;\n";
2952   pr "\n";
2953
2954   (* LVM internal structures. *)
2955   List.iter (
2956     function
2957     | typ, cols ->
2958         pr "struct guestfs_lvm_int_%s {\n" typ;
2959         List.iter (function
2960                    | name, `String -> pr "  string %s<>;\n" name
2961                    | name, `UUID -> pr "  opaque %s[32];\n" name
2962                    | name, `Bytes -> pr "  hyper %s;\n" name
2963                    | name, `Int -> pr "  hyper %s;\n" name
2964                    | name, `OptPercent -> pr "  float %s;\n" name
2965                   ) cols;
2966         pr "};\n";
2967         pr "\n";
2968         pr "typedef struct guestfs_lvm_int_%s guestfs_lvm_int_%s_list<>;\n" typ typ;
2969         pr "\n";
2970   ) ["pv", pv_cols; "vg", vg_cols; "lv", lv_cols];
2971
2972   (* Stat internal structures. *)
2973   List.iter (
2974     function
2975     | typ, cols ->
2976         pr "struct guestfs_int_%s {\n" typ;
2977         List.iter (function
2978                    | name, `Int -> pr "  hyper %s;\n" name
2979                   ) cols;
2980         pr "};\n";
2981         pr "\n";
2982   ) ["stat", stat_cols; "statvfs", statvfs_cols];
2983
2984   List.iter (
2985     fun (shortname, style, _, _, _, _, _) ->
2986       let name = "guestfs_" ^ shortname in
2987
2988       (match snd style with
2989        | [] -> ()
2990        | args ->
2991            pr "struct %s_args {\n" name;
2992            List.iter (
2993              function
2994              | String n -> pr "  string %s<>;\n" n
2995              | OptString n -> pr "  str *%s;\n" n
2996              | StringList n -> pr "  str %s<>;\n" n
2997              | Bool n -> pr "  bool %s;\n" n
2998              | Int n -> pr "  int %s;\n" n
2999              | FileIn _ | FileOut _ -> ()
3000            ) args;
3001            pr "};\n\n"
3002       );
3003       (match fst style with
3004        | RErr -> ()
3005        | RInt n ->
3006            pr "struct %s_ret {\n" name;
3007            pr "  int %s;\n" n;
3008            pr "};\n\n"
3009        | RInt64 n ->
3010            pr "struct %s_ret {\n" name;
3011            pr "  hyper %s;\n" n;
3012            pr "};\n\n"
3013        | RBool n ->
3014            pr "struct %s_ret {\n" name;
3015            pr "  bool %s;\n" n;
3016            pr "};\n\n"
3017        | RConstString _ ->
3018            failwithf "RConstString cannot be returned from a daemon function"
3019        | RString n ->
3020            pr "struct %s_ret {\n" name;
3021            pr "  string %s<>;\n" n;
3022            pr "};\n\n"
3023        | RStringList n ->
3024            pr "struct %s_ret {\n" name;
3025            pr "  str %s<>;\n" n;
3026            pr "};\n\n"
3027        | RIntBool (n,m) ->
3028            pr "struct %s_ret {\n" name;
3029            pr "  int %s;\n" n;
3030            pr "  bool %s;\n" m;
3031            pr "};\n\n"
3032        | RPVList n ->
3033            pr "struct %s_ret {\n" name;
3034            pr "  guestfs_lvm_int_pv_list %s;\n" n;
3035            pr "};\n\n"
3036        | RVGList n ->
3037            pr "struct %s_ret {\n" name;
3038            pr "  guestfs_lvm_int_vg_list %s;\n" n;
3039            pr "};\n\n"
3040        | RLVList n ->
3041            pr "struct %s_ret {\n" name;
3042            pr "  guestfs_lvm_int_lv_list %s;\n" n;
3043            pr "};\n\n"
3044        | RStat n ->
3045            pr "struct %s_ret {\n" name;
3046            pr "  guestfs_int_stat %s;\n" n;
3047            pr "};\n\n"
3048        | RStatVFS n ->
3049            pr "struct %s_ret {\n" name;
3050            pr "  guestfs_int_statvfs %s;\n" n;
3051            pr "};\n\n"
3052        | RHashtable n ->
3053            pr "struct %s_ret {\n" name;
3054            pr "  str %s<>;\n" n;
3055            pr "};\n\n"
3056       );
3057   ) daemon_functions;
3058
3059   (* Table of procedure numbers. *)
3060   pr "enum guestfs_procedure {\n";
3061   List.iter (
3062     fun (shortname, _, proc_nr, _, _, _, _) ->
3063       pr "  GUESTFS_PROC_%s = %d,\n" (String.uppercase shortname) proc_nr
3064   ) daemon_functions;
3065   pr "  GUESTFS_PROC_NR_PROCS\n";
3066   pr "};\n";
3067   pr "\n";
3068
3069   (* Having to choose a maximum message size is annoying for several
3070    * reasons (it limits what we can do in the API), but it (a) makes
3071    * the protocol a lot simpler, and (b) provides a bound on the size
3072    * of the daemon which operates in limited memory space.  For large
3073    * file transfers you should use FTP.
3074    *)
3075   pr "const GUESTFS_MESSAGE_MAX = %d;\n" (4 * 1024 * 1024);
3076   pr "\n";
3077
3078   (* Message header, etc. *)
3079   pr "\
3080 /* The communication protocol is now documented in the guestfs(3)
3081  * manpage.
3082  */
3083
3084 const GUESTFS_PROGRAM = 0x2000F5F5;
3085 const GUESTFS_PROTOCOL_VERSION = 1;
3086
3087 /* These constants must be larger than any possible message length. */
3088 const GUESTFS_LAUNCH_FLAG = 0xf5f55ff5;
3089 const GUESTFS_CANCEL_FLAG = 0xffffeeee;
3090
3091 enum guestfs_message_direction {
3092   GUESTFS_DIRECTION_CALL = 0,        /* client -> daemon */
3093   GUESTFS_DIRECTION_REPLY = 1        /* daemon -> client */
3094 };
3095
3096 enum guestfs_message_status {
3097   GUESTFS_STATUS_OK = 0,
3098   GUESTFS_STATUS_ERROR = 1
3099 };
3100
3101 const GUESTFS_ERROR_LEN = 256;
3102
3103 struct guestfs_message_error {
3104   string error_message<GUESTFS_ERROR_LEN>;
3105 };
3106
3107 struct guestfs_message_header {
3108   unsigned prog;                     /* GUESTFS_PROGRAM */
3109   unsigned vers;                     /* GUESTFS_PROTOCOL_VERSION */
3110   guestfs_procedure proc;            /* GUESTFS_PROC_x */
3111   guestfs_message_direction direction;
3112   unsigned serial;                   /* message serial number */
3113   guestfs_message_status status;
3114 };
3115
3116 const GUESTFS_MAX_CHUNK_SIZE = 8192;
3117
3118 struct guestfs_chunk {
3119   int cancel;                        /* if non-zero, transfer is cancelled */
3120   /* data size is 0 bytes if the transfer has finished successfully */
3121   opaque data<GUESTFS_MAX_CHUNK_SIZE>;
3122 };
3123 "
3124
3125 (* Generate the guestfs-structs.h file. *)
3126 and generate_structs_h () =
3127   generate_header CStyle LGPLv2;
3128
3129   (* This is a public exported header file containing various
3130    * structures.  The structures are carefully written to have
3131    * exactly the same in-memory format as the XDR structures that
3132    * we use on the wire to the daemon.  The reason for creating
3133    * copies of these structures here is just so we don't have to
3134    * export the whole of guestfs_protocol.h (which includes much
3135    * unrelated and XDR-dependent stuff that we don't want to be
3136    * public, or required by clients).
3137    *
3138    * To reiterate, we will pass these structures to and from the
3139    * client with a simple assignment or memcpy, so the format
3140    * must be identical to what rpcgen / the RFC defines.
3141    *)
3142
3143   (* guestfs_int_bool structure. *)
3144   pr "struct guestfs_int_bool {\n";
3145   pr "  int32_t i;\n";
3146   pr "  int32_t b;\n";
3147   pr "};\n";
3148   pr "\n";
3149
3150   (* LVM public structures. *)
3151   List.iter (
3152     function
3153     | typ, cols ->
3154         pr "struct guestfs_lvm_%s {\n" typ;
3155         List.iter (
3156           function
3157           | name, `String -> pr "  char *%s;\n" name
3158           | name, `UUID -> pr "  char %s[32]; /* this is NOT nul-terminated, be careful when printing */\n" name
3159           | name, `Bytes -> pr "  uint64_t %s;\n" name
3160           | name, `Int -> pr "  int64_t %s;\n" name
3161           | name, `OptPercent -> pr "  float %s; /* [0..100] or -1 */\n" name
3162         ) cols;
3163         pr "};\n";
3164         pr "\n";
3165         pr "struct guestfs_lvm_%s_list {\n" typ;
3166         pr "  uint32_t len;\n";
3167         pr "  struct guestfs_lvm_%s *val;\n" typ;
3168         pr "};\n";
3169         pr "\n"
3170   ) ["pv", pv_cols; "vg", vg_cols; "lv", lv_cols];
3171
3172   (* Stat structures. *)
3173   List.iter (
3174     function
3175     | typ, cols ->
3176         pr "struct guestfs_%s {\n" typ;
3177         List.iter (
3178           function
3179           | name, `Int -> pr "  int64_t %s;\n" name
3180         ) cols;
3181         pr "};\n";
3182         pr "\n"
3183   ) ["stat", stat_cols; "statvfs", statvfs_cols]
3184
3185 (* Generate the guestfs-actions.h file. *)
3186 and generate_actions_h () =
3187   generate_header CStyle LGPLv2;
3188   List.iter (
3189     fun (shortname, style, _, _, _, _, _) ->
3190       let name = "guestfs_" ^ shortname in
3191       generate_prototype ~single_line:true ~newline:true ~handle:"handle"
3192         name style
3193   ) all_functions
3194
3195 (* Generate the client-side dispatch stubs. *)
3196 and generate_client_actions () =
3197   generate_header CStyle LGPLv2;
3198
3199   pr "\
3200 #include <stdio.h>
3201 #include <stdlib.h>
3202
3203 #include \"guestfs.h\"
3204 #include \"guestfs_protocol.h\"
3205
3206 #define error guestfs_error
3207 #define perrorf guestfs_perrorf
3208 #define safe_malloc guestfs_safe_malloc
3209 #define safe_realloc guestfs_safe_realloc
3210 #define safe_strdup guestfs_safe_strdup
3211 #define safe_memdup guestfs_safe_memdup
3212
3213 /* Check the return message from a call for validity. */
3214 static int
3215 check_reply_header (guestfs_h *g,
3216                     const struct guestfs_message_header *hdr,
3217                     int proc_nr, int serial)
3218 {
3219   if (hdr->prog != GUESTFS_PROGRAM) {
3220     error (g, \"wrong program (%%d/%%d)\", hdr->prog, GUESTFS_PROGRAM);
3221     return -1;
3222   }
3223   if (hdr->vers != GUESTFS_PROTOCOL_VERSION) {
3224     error (g, \"wrong protocol version (%%d/%%d)\",
3225            hdr->vers, GUESTFS_PROTOCOL_VERSION);
3226     return -1;
3227   }
3228   if (hdr->direction != GUESTFS_DIRECTION_REPLY) {
3229     error (g, \"unexpected message direction (%%d/%%d)\",
3230            hdr->direction, GUESTFS_DIRECTION_REPLY);
3231     return -1;
3232   }
3233   if (hdr->proc != proc_nr) {
3234     error (g, \"unexpected procedure number (%%d/%%d)\", hdr->proc, proc_nr);
3235     return -1;
3236   }
3237   if (hdr->serial != serial) {
3238     error (g, \"unexpected serial (%%d/%%d)\", hdr->serial, serial);
3239     return -1;
3240   }
3241
3242   return 0;
3243 }
3244
3245 /* Check we are in the right state to run a high-level action. */
3246 static int
3247 check_state (guestfs_h *g, const char *caller)
3248 {
3249   if (!guestfs_is_ready (g)) {
3250     if (guestfs_is_config (g))
3251       error (g, \"%%s: call launch() before using this function\",
3252         caller);
3253     else if (guestfs_is_launching (g))
3254       error (g, \"%%s: call wait_ready() before using this function\",
3255         caller);
3256     else
3257       error (g, \"%%s called from the wrong state, %%d != READY\",
3258         caller, guestfs_get_state (g));
3259     return -1;
3260   }
3261   return 0;
3262 }
3263
3264 ";
3265
3266   (* Client-side stubs for each function. *)
3267   List.iter (
3268     fun (shortname, style, _, _, _, _, _) ->
3269       let name = "guestfs_" ^ shortname in
3270
3271       (* Generate the context struct which stores the high-level
3272        * state between callback functions.
3273        *)
3274       pr "struct %s_ctx {\n" shortname;
3275       pr "  /* This flag is set by the callbacks, so we know we've done\n";
3276       pr "   * the callbacks as expected, and in the right sequence.\n";
3277       pr "   * 0 = not called, 1 = reply_cb called.\n";
3278       pr "   */\n";
3279       pr "  int cb_sequence;\n";
3280       pr "  struct guestfs_message_header hdr;\n";
3281       pr "  struct guestfs_message_error err;\n";
3282       (match fst style with
3283        | RErr -> ()
3284        | RConstString _ ->
3285            failwithf "RConstString cannot be returned from a daemon function"
3286        | RInt _ | RInt64 _
3287        | RBool _ | RString _ | RStringList _
3288        | RIntBool _
3289        | RPVList _ | RVGList _ | RLVList _
3290        | RStat _ | RStatVFS _
3291        | RHashtable _ ->
3292            pr "  struct %s_ret ret;\n" name
3293       );
3294       pr "};\n";
3295       pr "\n";
3296
3297       (* Generate the reply callback function. *)
3298       pr "static void %s_reply_cb (guestfs_h *g, void *data, XDR *xdr)\n" shortname;
3299       pr "{\n";
3300       pr "  guestfs_main_loop *ml = guestfs_get_main_loop (g);\n";
3301       pr "  struct %s_ctx *ctx = (struct %s_ctx *) data;\n" shortname shortname;
3302       pr "\n";
3303       pr "  /* This should definitely not happen. */\n";
3304       pr "  if (ctx->cb_sequence != 0) {\n";
3305       pr "    ctx->cb_sequence = 9999;\n";
3306       pr "    error (g, \"%%s: internal error: reply callback called twice\", \"%s\");\n" name;
3307       pr "    return;\n";
3308       pr "  }\n";
3309       pr "\n";
3310       pr "  ml->main_loop_quit (ml, g);\n";
3311       pr "\n";
3312       pr "  if (!xdr_guestfs_message_header (xdr, &ctx->hdr)) {\n";
3313       pr "    error (g, \"%%s: failed to parse reply header\", \"%s\");\n" name;
3314       pr "    return;\n";
3315       pr "  }\n";
3316       pr "  if (ctx->hdr.status == GUESTFS_STATUS_ERROR) {\n";
3317       pr "    if (!xdr_guestfs_message_error (xdr, &ctx->err)) {\n";
3318       pr "      error (g, \"%%s: failed to parse reply error\", \"%s\");\n"
3319         name;
3320       pr "      return;\n";
3321       pr "    }\n";
3322       pr "    goto done;\n";
3323       pr "  }\n";
3324
3325       (match fst style with
3326        | RErr -> ()
3327        | RConstString _ ->
3328            failwithf "RConstString cannot be returned from a daemon function"
3329        | RInt _ | RInt64 _
3330        | RBool _ | RString _ | RStringList _
3331        | RIntBool _
3332        | RPVList _ | RVGList _ | RLVList _
3333        | RStat _ | RStatVFS _
3334        | RHashtable _ ->
3335             pr "  if (!xdr_%s_ret (xdr, &ctx->ret)) {\n" name;
3336             pr "    error (g, \"%%s: failed to parse reply\", \"%s\");\n" name;
3337             pr "    return;\n";
3338             pr "  }\n";
3339       );
3340
3341       pr " done:\n";
3342       pr "  ctx->cb_sequence = 1;\n";
3343       pr "}\n\n";
3344
3345       (* Generate the action stub. *)
3346       generate_prototype ~extern:false ~semicolon:false ~newline:true
3347         ~handle:"g" name style;
3348
3349       let error_code =
3350         match fst style with
3351         | RErr | RInt _ | RInt64 _ | RBool _ -> "-1"
3352         | RConstString _ ->
3353             failwithf "RConstString cannot be returned from a daemon function"
3354         | RString _ | RStringList _ | RIntBool _
3355         | RPVList _ | RVGList _ | RLVList _
3356         | RStat _ | RStatVFS _
3357         | RHashtable _ ->
3358             "NULL" in
3359
3360       pr "{\n";
3361
3362       (match snd style with
3363        | [] -> ()
3364        | _ -> pr "  struct %s_args args;\n" name
3365       );
3366
3367       pr "  struct %s_ctx ctx;\n" shortname;
3368       pr "  guestfs_main_loop *ml = guestfs_get_main_loop (g);\n";
3369       pr "  int serial;\n";
3370       pr "\n";
3371       pr "  if (check_state (g, \"%s\") == -1) return %s;\n" name error_code;
3372       pr "  guestfs_set_busy (g);\n";
3373       pr "\n";
3374       pr "  memset (&ctx, 0, sizeof ctx);\n";
3375       pr "\n";
3376
3377       (* Send the main header and arguments. *)
3378       (match snd style with
3379        | [] ->
3380            pr "  serial = guestfs__send_sync (g, GUESTFS_PROC_%s, NULL, NULL);\n"
3381              (String.uppercase shortname)
3382        | args ->
3383            List.iter (
3384              function
3385              | String n ->
3386                  pr "  args.%s = (char *) %s;\n" n n
3387              | OptString n ->
3388                  pr "  args.%s = %s ? (char **) &%s : NULL;\n" n n n
3389              | StringList n ->
3390                  pr "  args.%s.%s_val = (char **) %s;\n" n n n;
3391                  pr "  for (args.%s.%s_len = 0; %s[args.%s.%s_len]; args.%s.%s_len++) ;\n" n n n n n n n;
3392              | Bool n ->
3393                  pr "  args.%s = %s;\n" n n
3394              | Int n ->
3395                  pr "  args.%s = %s;\n" n n
3396              | FileIn _ | FileOut _ -> ()
3397            ) args;
3398            pr "  serial = guestfs__send_sync (g, GUESTFS_PROC_%s,\n"
3399              (String.uppercase shortname);
3400            pr "        (xdrproc_t) xdr_%s_args, (char *) &args);\n"
3401              name;
3402       );
3403       pr "  if (serial == -1) {\n";
3404       pr "    guestfs_end_busy (g);\n";
3405       pr "    return %s;\n" error_code;
3406       pr "  }\n";
3407       pr "\n";
3408
3409       (* Send any additional files (FileIn) requested. *)
3410       let need_read_reply_label = ref false in
3411       List.iter (
3412         function
3413         | FileIn n ->
3414             pr "  {\n";
3415             pr "    int r;\n";
3416             pr "\n";
3417             pr "    r = guestfs__send_file_sync (g, %s);\n" n;
3418             pr "    if (r == -1) {\n";
3419             pr "      guestfs_end_busy (g);\n";
3420             pr "      return %s;\n" error_code;
3421             pr "    }\n";
3422             pr "    if (r == -2) /* daemon cancelled */\n";
3423             pr "      goto read_reply;\n";
3424             need_read_reply_label := true;
3425             pr "  }\n";
3426             pr "\n";
3427         | _ -> ()
3428       ) (snd style);
3429
3430       (* Wait for the reply from the remote end. *)
3431       if !need_read_reply_label then pr " read_reply:\n";
3432       pr "  guestfs__switch_to_receiving (g);\n";
3433       pr "  ctx.cb_sequence = 0;\n";
3434       pr "  guestfs_set_reply_callback (g, %s_reply_cb, &ctx);\n" shortname;
3435       pr "  (void) ml->main_loop_run (ml, g);\n";
3436       pr "  guestfs_set_reply_callback (g, NULL, NULL);\n";
3437       pr "  if (ctx.cb_sequence != 1) {\n";
3438       pr "    error (g, \"%%s reply failed, see earlier error messages\", \"%s\");\n" name;
3439       pr "    guestfs_end_busy (g);\n";
3440       pr "    return %s;\n" error_code;
3441       pr "  }\n";
3442       pr "\n";
3443
3444       pr "  if (check_reply_header (g, &ctx.hdr, GUESTFS_PROC_%s, serial) == -1) {\n"
3445         (String.uppercase shortname);
3446       pr "    guestfs_end_busy (g);\n";
3447       pr "    return %s;\n" error_code;
3448       pr "  }\n";
3449       pr "\n";
3450
3451       pr "  if (ctx.hdr.status == GUESTFS_STATUS_ERROR) {\n";
3452       pr "    error (g, \"%%s\", ctx.err.error_message);\n";
3453       pr "    free (ctx.err.error_message);\n";
3454       pr "    guestfs_end_busy (g);\n";
3455       pr "    return %s;\n" error_code;
3456       pr "  }\n";
3457       pr "\n";
3458
3459       (* Expecting to receive further files (FileOut)? *)
3460       List.iter (
3461         function
3462         | FileOut n ->
3463             pr "  if (guestfs__receive_file_sync (g, %s) == -1) {\n" n;
3464             pr "    guestfs_end_busy (g);\n";
3465             pr "    return %s;\n" error_code;
3466             pr "  }\n";
3467             pr "\n";
3468         | _ -> ()
3469       ) (snd style);
3470
3471       pr "  guestfs_end_busy (g);\n";
3472
3473       (match fst style with
3474        | RErr -> pr "  return 0;\n"
3475        | RInt n | RInt64 n | RBool n ->
3476            pr "  return ctx.ret.%s;\n" n
3477        | RConstString _ ->
3478            failwithf "RConstString cannot be returned from a daemon function"
3479        | RString n ->
3480            pr "  return ctx.ret.%s; /* caller will free */\n" n
3481        | RStringList n | RHashtable n ->
3482            pr "  /* caller will free this, but we need to add a NULL entry */\n";
3483            pr "  ctx.ret.%s.%s_val =\n" n n;
3484            pr "    safe_realloc (g, ctx.ret.%s.%s_val,\n" n n;
3485            pr "                  sizeof (char *) * (ctx.ret.%s.%s_len + 1));\n"
3486              n n;
3487            pr "  ctx.ret.%s.%s_val[ctx.ret.%s.%s_len] = NULL;\n" n n n n;
3488            pr "  return ctx.ret.%s.%s_val;\n" n n
3489        | RIntBool _ ->
3490            pr "  /* caller with free this */\n";
3491            pr "  return safe_memdup (g, &ctx.ret, sizeof (ctx.ret));\n"
3492        | RPVList n | RVGList n | RLVList n
3493        | RStat n | RStatVFS n ->
3494            pr "  /* caller will free this */\n";
3495            pr "  return safe_memdup (g, &ctx.ret.%s, sizeof (ctx.ret.%s));\n" n n
3496       );
3497
3498       pr "}\n\n"
3499   ) daemon_functions
3500
3501 (* Generate daemon/actions.h. *)
3502 and generate_daemon_actions_h () =
3503   generate_header CStyle GPLv2;
3504
3505   pr "#include \"../src/guestfs_protocol.h\"\n";
3506   pr "\n";
3507
3508   List.iter (
3509     fun (name, style, _, _, _, _, _) ->
3510         generate_prototype
3511           ~single_line:true ~newline:true ~in_daemon:true ~prefix:"do_"
3512           name style;
3513   ) daemon_functions
3514
3515 (* Generate the server-side stubs. *)
3516 and generate_daemon_actions () =
3517   generate_header CStyle GPLv2;
3518
3519   pr "#include <config.h>\n";
3520   pr "\n";
3521   pr "#include <stdio.h>\n";
3522   pr "#include <stdlib.h>\n";
3523   pr "#include <string.h>\n";
3524   pr "#include <inttypes.h>\n";
3525   pr "#include <ctype.h>\n";
3526   pr "#include <rpc/types.h>\n";
3527   pr "#include <rpc/xdr.h>\n";
3528   pr "\n";
3529   pr "#include \"daemon.h\"\n";
3530   pr "#include \"../src/guestfs_protocol.h\"\n";
3531   pr "#include \"actions.h\"\n";
3532   pr "\n";
3533
3534   List.iter (
3535     fun (name, style, _, _, _, _, _) ->
3536       (* Generate server-side stubs. *)
3537       pr "static void %s_stub (XDR *xdr_in)\n" name;
3538       pr "{\n";
3539       let error_code =
3540         match fst style with
3541         | RErr | RInt _ -> pr "  int r;\n"; "-1"
3542         | RInt64 _ -> pr "  int64_t r;\n"; "-1"
3543         | RBool _ -> pr "  int r;\n"; "-1"
3544         | RConstString _ ->
3545             failwithf "RConstString cannot be returned from a daemon function"
3546         | RString _ -> pr "  char *r;\n"; "NULL"
3547         | RStringList _ | RHashtable _ -> pr "  char **r;\n"; "NULL"
3548         | RIntBool _ -> pr "  guestfs_%s_ret *r;\n" name; "NULL"
3549         | RPVList _ -> pr "  guestfs_lvm_int_pv_list *r;\n"; "NULL"
3550         | RVGList _ -> pr "  guestfs_lvm_int_vg_list *r;\n"; "NULL"
3551         | RLVList _ -> pr "  guestfs_lvm_int_lv_list *r;\n"; "NULL"
3552         | RStat _ -> pr "  guestfs_int_stat *r;\n"; "NULL"
3553         | RStatVFS _ -> pr "  guestfs_int_statvfs *r;\n"; "NULL" in
3554
3555       (match snd style with
3556        | [] -> ()
3557        | args ->
3558            pr "  struct guestfs_%s_args args;\n" name;
3559            List.iter (
3560              function
3561                (* Note we allow the string to be writable, in order to
3562                 * allow device name translation.  This is safe because
3563                 * we can modify the string (passed from RPC).
3564                 *)
3565              | String n
3566              | OptString n -> pr "  char *%s;\n" n
3567              | StringList n -> pr "  char **%s;\n" n
3568              | Bool n -> pr "  int %s;\n" n
3569              | Int n -> pr "  int %s;\n" n
3570              | FileIn _ | FileOut _ -> ()
3571            ) args
3572       );
3573       pr "\n";
3574
3575       (match snd style with
3576        | [] -> ()
3577        | args ->
3578            pr "  memset (&args, 0, sizeof args);\n";
3579            pr "\n";
3580            pr "  if (!xdr_guestfs_%s_args (xdr_in, &args)) {\n" name;
3581            pr "    reply_with_error (\"%%s: daemon failed to decode procedure arguments\", \"%s\");\n" name;
3582            pr "    return;\n";
3583            pr "  }\n";
3584            List.iter (
3585              function
3586              | String n -> pr "  %s = args.%s;\n" n n
3587              | OptString n -> pr "  %s = args.%s ? *args.%s : NULL;\n" n n n
3588              | StringList n ->
3589                  pr "  %s = realloc (args.%s.%s_val,\n" n n n;
3590                  pr "                sizeof (char *) * (args.%s.%s_len+1));\n" n n;
3591                  pr "  if (%s == NULL) {\n" n;
3592                  pr "    reply_with_perror (\"realloc\");\n";
3593                  pr "    goto done;\n";
3594                  pr "  }\n";
3595                  pr "  %s[args.%s.%s_len] = NULL;\n" n n n;
3596                  pr "  args.%s.%s_val = %s;\n" n n n;
3597              | Bool n -> pr "  %s = args.%s;\n" n n
3598              | Int n -> pr "  %s = args.%s;\n" n n
3599              | FileIn _ | FileOut _ -> ()
3600            ) args;
3601            pr "\n"
3602       );
3603
3604       (* Don't want to call the impl with any FileIn or FileOut
3605        * parameters, since these go "outside" the RPC protocol.
3606        *)
3607       let argsnofile =
3608         List.filter (function FileIn _ | FileOut _ -> false | _ -> true)
3609           (snd style) in
3610       pr "  r = do_%s " name;
3611       generate_call_args argsnofile;
3612       pr ";\n";
3613
3614       pr "  if (r == %s)\n" error_code;
3615       pr "    /* do_%s has already called reply_with_error */\n" name;
3616       pr "    goto done;\n";
3617       pr "\n";
3618
3619       (* If there are any FileOut parameters, then the impl must
3620        * send its own reply.
3621        *)
3622       let no_reply =
3623         List.exists (function FileOut _ -> true | _ -> false) (snd style) in
3624       if no_reply then
3625         pr "  /* do_%s has already sent a reply */\n" name
3626       else (
3627         match fst style with
3628         | RErr -> pr "  reply (NULL, NULL);\n"
3629         | RInt n | RInt64 n | RBool n ->
3630             pr "  struct guestfs_%s_ret ret;\n" name;
3631             pr "  ret.%s = r;\n" n;
3632             pr "  reply ((xdrproc_t) &xdr_guestfs_%s_ret, (char *) &ret);\n"
3633               name
3634         | RConstString _ ->
3635             failwithf "RConstString cannot be returned from a daemon function"
3636         | RString n ->
3637             pr "  struct guestfs_%s_ret ret;\n" name;
3638             pr "  ret.%s = r;\n" n;
3639             pr "  reply ((xdrproc_t) &xdr_guestfs_%s_ret, (char *) &ret);\n"
3640               name;
3641             pr "  free (r);\n"
3642         | RStringList n | RHashtable n ->
3643             pr "  struct guestfs_%s_ret ret;\n" name;
3644             pr "  ret.%s.%s_len = count_strings (r);\n" n n;
3645             pr "  ret.%s.%s_val = r;\n" n n;
3646             pr "  reply ((xdrproc_t) &xdr_guestfs_%s_ret, (char *) &ret);\n"
3647               name;
3648             pr "  free_strings (r);\n"
3649         | RIntBool _ ->
3650             pr "  reply ((xdrproc_t) xdr_guestfs_%s_ret, (char *) r);\n"
3651               name;
3652             pr "  xdr_free ((xdrproc_t) xdr_guestfs_%s_ret, (char *) r);\n" name
3653         | RPVList n | RVGList n | RLVList n
3654         | RStat n | RStatVFS n ->
3655             pr "  struct guestfs_%s_ret ret;\n" name;
3656             pr "  ret.%s = *r;\n" n;
3657             pr "  reply ((xdrproc_t) xdr_guestfs_%s_ret, (char *) &ret);\n"
3658               name;
3659             pr "  xdr_free ((xdrproc_t) xdr_guestfs_%s_ret, (char *) &ret);\n"
3660               name
3661       );
3662
3663       (* Free the args. *)
3664       (match snd style with
3665        | [] ->
3666            pr "done: ;\n";
3667        | _ ->
3668            pr "done:\n";
3669            pr "  xdr_free ((xdrproc_t) xdr_guestfs_%s_args, (char *) &args);\n"
3670              name
3671       );
3672
3673       pr "}\n\n";
3674   ) daemon_functions;
3675
3676   (* Dispatch function. *)
3677   pr "void dispatch_incoming_message (XDR *xdr_in)\n";
3678   pr "{\n";
3679   pr "  switch (proc_nr) {\n";
3680
3681   List.iter (
3682     fun (name, style, _, _, _, _, _) ->
3683         pr "    case GUESTFS_PROC_%s:\n" (String.uppercase name);
3684         pr "      %s_stub (xdr_in);\n" name;
3685         pr "      break;\n"
3686   ) daemon_functions;
3687
3688   pr "    default:\n";
3689   pr "      reply_with_error (\"dispatch_incoming_message: unknown procedure number %%d, set LIBGUESTFS_PATH to point to the matching libguestfs appliance directory\", proc_nr);\n";
3690   pr "  }\n";
3691   pr "}\n";
3692   pr "\n";
3693
3694   (* LVM columns and tokenization functions. *)
3695   (* XXX This generates crap code.  We should rethink how we
3696    * do this parsing.
3697    *)
3698   List.iter (
3699     function
3700     | typ, cols ->
3701         pr "static const char *lvm_%s_cols = \"%s\";\n"
3702           typ (String.concat "," (List.map fst cols));
3703         pr "\n";
3704
3705         pr "static int lvm_tokenize_%s (char *str, struct guestfs_lvm_int_%s *r)\n" typ typ;
3706         pr "{\n";
3707         pr "  char *tok, *p, *next;\n";
3708         pr "  int i, j;\n";
3709         pr "\n";
3710         (*
3711         pr "  fprintf (stderr, \"%%s: <<%%s>>\\n\", __func__, str);\n";
3712         pr "\n";
3713         *)
3714         pr "  if (!str) {\n";
3715         pr "    fprintf (stderr, \"%%s: failed: passed a NULL string\\n\", __func__);\n";
3716         pr "    return -1;\n";
3717         pr "  }\n";
3718         pr "  if (!*str || isspace (*str)) {\n";
3719         pr "    fprintf (stderr, \"%%s: failed: passed a empty string or one beginning with whitespace\\n\", __func__);\n";
3720         pr "    return -1;\n";
3721         pr "  }\n";
3722         pr "  tok = str;\n";
3723         List.iter (
3724           fun (name, coltype) ->
3725             pr "  if (!tok) {\n";
3726             pr "    fprintf (stderr, \"%%s: failed: string finished early, around token %%s\\n\", __func__, \"%s\");\n" name;
3727             pr "    return -1;\n";
3728             pr "  }\n";
3729             pr "  p = strchrnul (tok, ',');\n";
3730             pr "  if (*p) next = p+1; else next = NULL;\n";
3731             pr "  *p = '\\0';\n";
3732             (match coltype with
3733              | `String ->
3734                  pr "  r->%s = strdup (tok);\n" name;
3735                  pr "  if (r->%s == NULL) {\n" name;
3736                  pr "    perror (\"strdup\");\n";
3737                  pr "    return -1;\n";
3738                  pr "  }\n"
3739              | `UUID ->
3740                  pr "  for (i = j = 0; i < 32; ++j) {\n";
3741                  pr "    if (tok[j] == '\\0') {\n";
3742                  pr "      fprintf (stderr, \"%%s: failed to parse UUID from '%%s'\\n\", __func__, tok);\n";
3743                  pr "      return -1;\n";
3744                  pr "    } else if (tok[j] != '-')\n";
3745                  pr "      r->%s[i++] = tok[j];\n" name;
3746                  pr "  }\n";
3747              | `Bytes ->
3748                  pr "  if (sscanf (tok, \"%%\"SCNu64, &r->%s) != 1) {\n" name;
3749                  pr "    fprintf (stderr, \"%%s: failed to parse size '%%s' from token %%s\\n\", __func__, tok, \"%s\");\n" name;
3750                  pr "    return -1;\n";
3751                  pr "  }\n";
3752              | `Int ->
3753                  pr "  if (sscanf (tok, \"%%\"SCNi64, &r->%s) != 1) {\n" name;
3754                  pr "    fprintf (stderr, \"%%s: failed to parse int '%%s' from token %%s\\n\", __func__, tok, \"%s\");\n" name;
3755                  pr "    return -1;\n";
3756                  pr "  }\n";
3757              | `OptPercent ->
3758                  pr "  if (tok[0] == '\\0')\n";
3759                  pr "    r->%s = -1;\n" name;
3760                  pr "  else if (sscanf (tok, \"%%f\", &r->%s) != 1) {\n" name;
3761                  pr "    fprintf (stderr, \"%%s: failed to parse float '%%s' from token %%s\\n\", __func__, tok, \"%s\");\n" name;
3762                  pr "    return -1;\n";
3763                  pr "  }\n";
3764             );
3765             pr "  tok = next;\n";
3766         ) cols;
3767
3768         pr "  if (tok != NULL) {\n";
3769         pr "    fprintf (stderr, \"%%s: failed: extra tokens at end of string\\n\", __func__);\n";
3770         pr "    return -1;\n";
3771         pr "  }\n";
3772         pr "  return 0;\n";
3773         pr "}\n";
3774         pr "\n";
3775
3776         pr "guestfs_lvm_int_%s_list *\n" typ;
3777         pr "parse_command_line_%ss (void)\n" typ;
3778         pr "{\n";
3779         pr "  char *out, *err;\n";
3780         pr "  char *p, *pend;\n";
3781         pr "  int r, i;\n";
3782         pr "  guestfs_lvm_int_%s_list *ret;\n" typ;
3783         pr "  void *newp;\n";
3784         pr "\n";
3785         pr "  ret = malloc (sizeof *ret);\n";
3786         pr "  if (!ret) {\n";
3787         pr "    reply_with_perror (\"malloc\");\n";
3788         pr "    return NULL;\n";
3789         pr "  }\n";
3790         pr "\n";
3791         pr "  ret->guestfs_lvm_int_%s_list_len = 0;\n" typ;
3792         pr "  ret->guestfs_lvm_int_%s_list_val = NULL;\n" typ;
3793         pr "\n";
3794         pr "  r = command (&out, &err,\n";
3795         pr "           \"/sbin/lvm\", \"%ss\",\n" typ;
3796         pr "           \"-o\", lvm_%s_cols, \"--unbuffered\", \"--noheadings\",\n" typ;
3797         pr "           \"--nosuffix\", \"--separator\", \",\", \"--units\", \"b\", NULL);\n";
3798         pr "  if (r == -1) {\n";
3799         pr "    reply_with_error (\"%%s\", err);\n";
3800         pr "    free (out);\n";
3801         pr "    free (err);\n";
3802         pr "    free (ret);\n";
3803         pr "    return NULL;\n";
3804         pr "  }\n";
3805         pr "\n";
3806         pr "  free (err);\n";
3807         pr "\n";
3808         pr "  /* Tokenize each line of the output. */\n";
3809         pr "  p = out;\n";
3810         pr "  i = 0;\n";
3811         pr "  while (p) {\n";
3812         pr "    pend = strchr (p, '\\n');       /* Get the next line of output. */\n";
3813         pr "    if (pend) {\n";
3814         pr "      *pend = '\\0';\n";
3815         pr "      pend++;\n";
3816         pr "    }\n";
3817         pr "\n";
3818         pr "    while (*p && isspace (*p))      /* Skip any leading whitespace. */\n";
3819         pr "      p++;\n";
3820         pr "\n";
3821         pr "    if (!*p) {                      /* Empty line?  Skip it. */\n";
3822         pr "      p = pend;\n";
3823         pr "      continue;\n";
3824         pr "    }\n";
3825         pr "\n";
3826         pr "    /* Allocate some space to store this next entry. */\n";
3827         pr "    newp = realloc (ret->guestfs_lvm_int_%s_list_val,\n" typ;
3828         pr "                sizeof (guestfs_lvm_int_%s) * (i+1));\n" typ;
3829         pr "    if (newp == NULL) {\n";
3830         pr "      reply_with_perror (\"realloc\");\n";
3831         pr "      free (ret->guestfs_lvm_int_%s_list_val);\n" typ;
3832         pr "      free (ret);\n";
3833         pr "      free (out);\n";
3834         pr "      return NULL;\n";
3835         pr "    }\n";
3836         pr "    ret->guestfs_lvm_int_%s_list_val = newp;\n" typ;
3837         pr "\n";
3838         pr "    /* Tokenize the next entry. */\n";
3839         pr "    r = lvm_tokenize_%s (p, &ret->guestfs_lvm_int_%s_list_val[i]);\n" typ typ;
3840         pr "    if (r == -1) {\n";
3841         pr "      reply_with_error (\"failed to parse output of '%ss' command\");\n" typ;
3842         pr "      free (ret->guestfs_lvm_int_%s_list_val);\n" typ;
3843         pr "      free (ret);\n";
3844         pr "      free (out);\n";
3845         pr "      return NULL;\n";
3846         pr "    }\n";
3847         pr "\n";
3848         pr "    ++i;\n";
3849         pr "    p = pend;\n";
3850         pr "  }\n";
3851         pr "\n";
3852         pr "  ret->guestfs_lvm_int_%s_list_len = i;\n" typ;
3853         pr "\n";
3854         pr "  free (out);\n";
3855         pr "  return ret;\n";
3856         pr "}\n"
3857
3858   ) ["pv", pv_cols; "vg", vg_cols; "lv", lv_cols]
3859
3860 (* Generate the tests. *)
3861 and generate_tests () =
3862   generate_header CStyle GPLv2;
3863
3864   pr "\
3865 #include <stdio.h>
3866 #include <stdlib.h>
3867 #include <string.h>
3868 #include <unistd.h>
3869 #include <sys/types.h>
3870 #include <fcntl.h>
3871
3872 #include \"guestfs.h\"
3873
3874 static guestfs_h *g;
3875 static int suppress_error = 0;
3876
3877 static void print_error (guestfs_h *g, void *data, const char *msg)
3878 {
3879   if (!suppress_error)
3880     fprintf (stderr, \"%%s\\n\", msg);
3881 }
3882
3883 static void print_strings (char * const * const argv)
3884 {
3885   int argc;
3886
3887   for (argc = 0; argv[argc] != NULL; ++argc)
3888     printf (\"\\t%%s\\n\", argv[argc]);
3889 }
3890
3891 /*
3892 static void print_table (char * const * const argv)
3893 {
3894   int i;
3895
3896   for (i = 0; argv[i] != NULL; i += 2)
3897     printf (\"%%s: %%s\\n\", argv[i], argv[i+1]);
3898 }
3899 */
3900
3901 static void no_test_warnings (void)
3902 {
3903 ";
3904
3905   List.iter (
3906     function
3907     | name, _, _, _, [], _, _ ->
3908         pr "  fprintf (stderr, \"warning: \\\"guestfs_%s\\\" has no tests\\n\");\n" name
3909     | name, _, _, _, tests, _, _ -> ()
3910   ) all_functions;
3911
3912   pr "}\n";
3913   pr "\n";
3914
3915   (* Generate the actual tests.  Note that we generate the tests
3916    * in reverse order, deliberately, so that (in general) the
3917    * newest tests run first.  This makes it quicker and easier to
3918    * debug them.
3919    *)
3920   let test_names =
3921     List.map (
3922       fun (name, _, _, _, tests, _, _) ->
3923         mapi (generate_one_test name) tests
3924     ) (List.rev all_functions) in
3925   let test_names = List.concat test_names in
3926   let nr_tests = List.length test_names in
3927
3928   pr "\
3929 int main (int argc, char *argv[])
3930 {
3931   char c = 0;
3932   int failed = 0;
3933   const char *filename;
3934   int fd;
3935   int nr_tests, test_num = 0;
3936
3937   no_test_warnings ();
3938
3939   g = guestfs_create ();
3940   if (g == NULL) {
3941     printf (\"guestfs_create FAILED\\n\");
3942     exit (1);
3943   }
3944
3945   guestfs_set_error_handler (g, print_error, NULL);
3946
3947   guestfs_set_path (g, \"../appliance\");
3948
3949   filename = \"test1.img\";
3950   fd = open (filename, O_WRONLY|O_CREAT|O_NOCTTY|O_NONBLOCK|O_TRUNC, 0666);
3951   if (fd == -1) {
3952     perror (filename);
3953     exit (1);
3954   }
3955   if (lseek (fd, %d, SEEK_SET) == -1) {
3956     perror (\"lseek\");
3957     close (fd);
3958     unlink (filename);
3959     exit (1);
3960   }
3961   if (write (fd, &c, 1) == -1) {
3962     perror (\"write\");
3963     close (fd);
3964     unlink (filename);
3965     exit (1);
3966   }
3967   if (close (fd) == -1) {
3968     perror (filename);
3969     unlink (filename);
3970     exit (1);
3971   }
3972   if (guestfs_add_drive (g, filename) == -1) {
3973     printf (\"guestfs_add_drive %%s FAILED\\n\", filename);
3974     exit (1);
3975   }
3976
3977   filename = \"test2.img\";
3978   fd = open (filename, O_WRONLY|O_CREAT|O_NOCTTY|O_NONBLOCK|O_TRUNC, 0666);
3979   if (fd == -1) {
3980     perror (filename);
3981     exit (1);
3982   }
3983   if (lseek (fd, %d, SEEK_SET) == -1) {
3984     perror (\"lseek\");
3985     close (fd);
3986     unlink (filename);
3987     exit (1);
3988   }
3989   if (write (fd, &c, 1) == -1) {
3990     perror (\"write\");
3991     close (fd);
3992     unlink (filename);
3993     exit (1);
3994   }
3995   if (close (fd) == -1) {
3996     perror (filename);
3997     unlink (filename);
3998     exit (1);
3999   }
4000   if (guestfs_add_drive (g, filename) == -1) {
4001     printf (\"guestfs_add_drive %%s FAILED\\n\", filename);
4002     exit (1);
4003   }
4004
4005   filename = \"test3.img\";
4006   fd = open (filename, O_WRONLY|O_CREAT|O_NOCTTY|O_NONBLOCK|O_TRUNC, 0666);
4007   if (fd == -1) {
4008     perror (filename);
4009     exit (1);
4010   }
4011   if (lseek (fd, %d, SEEK_SET) == -1) {
4012     perror (\"lseek\");
4013     close (fd);
4014     unlink (filename);
4015     exit (1);
4016   }
4017   if (write (fd, &c, 1) == -1) {
4018     perror (\"write\");
4019     close (fd);
4020     unlink (filename);
4021     exit (1);
4022   }
4023   if (close (fd) == -1) {
4024     perror (filename);
4025     unlink (filename);
4026     exit (1);
4027   }
4028   if (guestfs_add_drive (g, filename) == -1) {
4029     printf (\"guestfs_add_drive %%s FAILED\\n\", filename);
4030     exit (1);
4031   }
4032
4033   if (guestfs_add_drive_ro (g, \"../images/test.sqsh\") == -1) {
4034     printf (\"guestfs_add_drive_ro ../images/test.sqsh FAILED\\n\");
4035     exit (1);
4036   }
4037
4038   if (guestfs_launch (g) == -1) {
4039     printf (\"guestfs_launch FAILED\\n\");
4040     exit (1);
4041   }
4042
4043   /* Set a timeout in case qemu hangs during launch (RHBZ#505329). */
4044   alarm (600);
4045
4046   if (guestfs_wait_ready (g) == -1) {
4047     printf (\"guestfs_wait_ready FAILED\\n\");
4048     exit (1);
4049   }
4050
4051   /* Cancel previous alarm. */
4052   alarm (0);
4053
4054   nr_tests = %d;
4055
4056 " (500 * 1024 * 1024) (50 * 1024 * 1024) (10 * 1024 * 1024) nr_tests;
4057
4058   iteri (
4059     fun i test_name ->
4060       pr "  test_num++;\n";
4061       pr "  printf (\"%%3d/%%3d %s\\n\", test_num, nr_tests);\n" test_name;
4062       pr "  if (%s () == -1) {\n" test_name;
4063       pr "    printf (\"%s FAILED\\n\");\n" test_name;
4064       pr "    failed++;\n";
4065       pr "  }\n";
4066   ) test_names;
4067   pr "\n";
4068
4069   pr "  guestfs_close (g);\n";
4070   pr "  unlink (\"test1.img\");\n";
4071   pr "  unlink (\"test2.img\");\n";
4072   pr "  unlink (\"test3.img\");\n";
4073   pr "\n";
4074
4075   pr "  if (failed > 0) {\n";
4076   pr "    printf (\"***** %%d / %%d tests FAILED *****\\n\", failed, nr_tests);\n";
4077   pr "    exit (1);\n";
4078   pr "  }\n";
4079   pr "\n";
4080
4081   pr "  exit (0);\n";
4082   pr "}\n"
4083
4084 and generate_one_test name i (init, prereq, test) =
4085   let test_name = sprintf "test_%s_%d" name i in
4086
4087   pr "\
4088 static int %s_skip (void)
4089 {
4090   const char *str;
4091
4092   str = getenv (\"SKIP_%s\");
4093   if (str && strcmp (str, \"1\") == 0) return 1;
4094   str = getenv (\"SKIP_TEST_%s\");
4095   if (str && strcmp (str, \"1\") == 0) return 1;
4096   return 0;
4097 }
4098
4099 " test_name (String.uppercase test_name) (String.uppercase name);
4100
4101   (match prereq with
4102    | Disabled | Always -> ()
4103    | If code | Unless code ->
4104        pr "static int %s_prereq (void)\n" test_name;
4105        pr "{\n";
4106        pr "  %s\n" code;
4107        pr "}\n";
4108        pr "\n";
4109   );
4110
4111   pr "\
4112 static int %s (void)
4113 {
4114   if (%s_skip ()) {
4115     printf (\"%%s skipped (reason: SKIP_TEST_* variable set)\\n\", \"%s\");
4116     return 0;
4117   }
4118
4119 " test_name test_name test_name;
4120
4121   (match prereq with
4122    | Disabled ->
4123        pr "  printf (\"%%s skipped (reason: test disabled in generator)\\n\", \"%s\");\n" test_name
4124    | If _ ->
4125        pr "  if (! %s_prereq ()) {\n" test_name;
4126        pr "    printf (\"%%s skipped (reason: test prerequisite)\\n\", \"%s\");\n" test_name;
4127        pr "    return 0;\n";
4128        pr "  }\n";
4129        pr "\n";
4130        generate_one_test_body name i test_name init test;
4131    | Unless _ ->
4132        pr "  if (%s_prereq ()) {\n" test_name;
4133        pr "    printf (\"%%s skipped (reason: test prerequisite)\\n\", \"%s\");\n" test_name;
4134        pr "    return 0;\n";
4135        pr "  }\n";
4136        pr "\n";
4137        generate_one_test_body name i test_name init test;
4138    | Always ->
4139        generate_one_test_body name i test_name init test
4140   );
4141
4142   pr "  return 0;\n";
4143   pr "}\n";
4144   pr "\n";
4145   test_name
4146
4147 and generate_one_test_body name i test_name init test =
4148   (match init with
4149    | InitNone
4150    | InitEmpty ->
4151        pr "  /* InitNone|InitEmpty for %s */\n" test_name;
4152        List.iter (generate_test_command_call test_name)
4153          [["blockdev_setrw"; "/dev/sda"];
4154           ["umount_all"];
4155           ["lvm_remove_all"]]
4156    | InitBasicFS ->
4157        pr "  /* InitBasicFS for %s: create ext2 on /dev/sda1 */\n" test_name;
4158        List.iter (generate_test_command_call test_name)
4159          [["blockdev_setrw"; "/dev/sda"];
4160           ["umount_all"];
4161           ["lvm_remove_all"];
4162           ["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
4163           ["mkfs"; "ext2"; "/dev/sda1"];
4164           ["mount"; "/dev/sda1"; "/"]]
4165    | InitBasicFSonLVM ->
4166        pr "  /* InitBasicFSonLVM for %s: create ext2 on /dev/VG/LV */\n"
4167          test_name;
4168        List.iter (generate_test_command_call test_name)
4169          [["blockdev_setrw"; "/dev/sda"];
4170           ["umount_all"];
4171           ["lvm_remove_all"];
4172           ["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
4173           ["pvcreate"; "/dev/sda1"];
4174           ["vgcreate"; "VG"; "/dev/sda1"];
4175           ["lvcreate"; "LV"; "VG"; "8"];
4176           ["mkfs"; "ext2"; "/dev/VG/LV"];
4177           ["mount"; "/dev/VG/LV"; "/"]]
4178   );
4179
4180   let get_seq_last = function
4181     | [] ->
4182         failwithf "%s: you cannot use [] (empty list) when expecting a command"
4183           test_name
4184     | seq ->
4185         let seq = List.rev seq in
4186         List.rev (List.tl seq), List.hd seq
4187   in
4188
4189   match test with
4190   | TestRun seq ->
4191       pr "  /* TestRun for %s (%d) */\n" name i;
4192       List.iter (generate_test_command_call test_name) seq
4193   | TestOutput (seq, expected) ->
4194       pr "  /* TestOutput for %s (%d) */\n" name i;
4195       pr "  char expected[] = \"%s\";\n" (c_quote expected);
4196       let seq, last = get_seq_last seq in
4197       let test () =
4198         pr "    if (strcmp (r, expected) != 0) {\n";
4199         pr "      fprintf (stderr, \"%s: expected \\\"%%s\\\" but got \\\"%%s\\\"\\n\", expected, r);\n" test_name;
4200         pr "      return -1;\n";
4201         pr "    }\n"
4202       in
4203       List.iter (generate_test_command_call test_name) seq;
4204       generate_test_command_call ~test test_name last
4205   | TestOutputList (seq, expected) ->
4206       pr "  /* TestOutputList for %s (%d) */\n" name i;
4207       let seq, last = get_seq_last seq in
4208       let test () =
4209         iteri (
4210           fun i str ->
4211             pr "    if (!r[%d]) {\n" i;
4212             pr "      fprintf (stderr, \"%s: short list returned from command\\n\");\n" test_name;
4213             pr "      print_strings (r);\n";
4214             pr "      return -1;\n";
4215             pr "    }\n";
4216             pr "    {\n";
4217             pr "      char expected[] = \"%s\";\n" (c_quote str);
4218             pr "      if (strcmp (r[%d], expected) != 0) {\n" i;
4219             pr "        fprintf (stderr, \"%s: expected \\\"%%s\\\" but got \\\"%%s\\\"\\n\", expected, r[%d]);\n" test_name i;
4220             pr "        return -1;\n";
4221             pr "      }\n";
4222             pr "    }\n"
4223         ) expected;
4224         pr "    if (r[%d] != NULL) {\n" (List.length expected);
4225         pr "      fprintf (stderr, \"%s: extra elements returned from command\\n\");\n"
4226           test_name;
4227         pr "      print_strings (r);\n";
4228         pr "      return -1;\n";
4229         pr "    }\n"
4230       in
4231       List.iter (generate_test_command_call test_name) seq;
4232       generate_test_command_call ~test test_name last
4233   | TestOutputListOfDevices (seq, expected) ->
4234       pr "  /* TestOutputListOfDevices for %s (%d) */\n" name i;
4235       let seq, last = get_seq_last seq in
4236       let test () =
4237         iteri (
4238           fun i str ->
4239             pr "    if (!r[%d]) {\n" i;
4240             pr "      fprintf (stderr, \"%s: short list returned from command\\n\");\n" test_name;
4241             pr "      print_strings (r);\n";
4242             pr "      return -1;\n";
4243             pr "    }\n";
4244             pr "    {\n";
4245             pr "      char expected[] = \"%s\";\n" (c_quote str);
4246             pr "      r[%d][5] = 's';\n" i;
4247             pr "      if (strcmp (r[%d], expected) != 0) {\n" i;
4248             pr "        fprintf (stderr, \"%s: expected \\\"%%s\\\" but got \\\"%%s\\\"\\n\", expected, r[%d]);\n" test_name i;
4249             pr "        return -1;\n";
4250             pr "      }\n";
4251             pr "    }\n"
4252         ) expected;
4253         pr "    if (r[%d] != NULL) {\n" (List.length expected);
4254         pr "      fprintf (stderr, \"%s: extra elements returned from command\\n\");\n"
4255           test_name;
4256         pr "      print_strings (r);\n";
4257         pr "      return -1;\n";
4258         pr "    }\n"
4259       in
4260       List.iter (generate_test_command_call test_name) seq;
4261       generate_test_command_call ~test test_name last
4262   | TestOutputInt (seq, expected) ->
4263       pr "  /* TestOutputInt for %s (%d) */\n" name i;
4264       let seq, last = get_seq_last seq in
4265       let test () =
4266         pr "    if (r != %d) {\n" expected;
4267         pr "      fprintf (stderr, \"%s: expected %d but got %%d\\n\","
4268           test_name expected;
4269         pr "               (int) r);\n";
4270         pr "      return -1;\n";
4271         pr "    }\n"
4272       in
4273       List.iter (generate_test_command_call test_name) seq;
4274       generate_test_command_call ~test test_name last
4275   | TestOutputTrue seq ->
4276       pr "  /* TestOutputTrue for %s (%d) */\n" name i;
4277       let seq, last = get_seq_last seq in
4278       let test () =
4279         pr "    if (!r) {\n";
4280         pr "      fprintf (stderr, \"%s: expected true, got false\\n\");\n"
4281           test_name;
4282         pr "      return -1;\n";
4283         pr "    }\n"
4284       in
4285       List.iter (generate_test_command_call test_name) seq;
4286       generate_test_command_call ~test test_name last
4287   | TestOutputFalse seq ->
4288       pr "  /* TestOutputFalse for %s (%d) */\n" name i;
4289       let seq, last = get_seq_last seq in
4290       let test () =
4291         pr "    if (r) {\n";
4292         pr "      fprintf (stderr, \"%s: expected false, got true\\n\");\n"
4293           test_name;
4294         pr "      return -1;\n";
4295         pr "    }\n"
4296       in
4297       List.iter (generate_test_command_call test_name) seq;
4298       generate_test_command_call ~test test_name last
4299   | TestOutputLength (seq, expected) ->
4300       pr "  /* TestOutputLength for %s (%d) */\n" name i;
4301       let seq, last = get_seq_last seq in
4302       let test () =
4303         pr "    int j;\n";
4304         pr "    for (j = 0; j < %d; ++j)\n" expected;
4305         pr "      if (r[j] == NULL) {\n";
4306         pr "        fprintf (stderr, \"%s: short list returned\\n\");\n"
4307           test_name;
4308         pr "        print_strings (r);\n";
4309         pr "        return -1;\n";
4310         pr "      }\n";
4311         pr "    if (r[j] != NULL) {\n";
4312         pr "      fprintf (stderr, \"%s: long list returned\\n\");\n"
4313           test_name;
4314         pr "      print_strings (r);\n";
4315         pr "      return -1;\n";
4316         pr "    }\n"
4317       in
4318       List.iter (generate_test_command_call test_name) seq;
4319       generate_test_command_call ~test test_name last
4320   | TestOutputStruct (seq, checks) ->
4321       pr "  /* TestOutputStruct for %s (%d) */\n" name i;
4322       let seq, last = get_seq_last seq in
4323       let test () =
4324         List.iter (
4325           function
4326           | CompareWithInt (field, expected) ->
4327               pr "    if (r->%s != %d) {\n" field expected;
4328               pr "      fprintf (stderr, \"%s: %s was %%d, expected %d\\n\",\n"
4329                 test_name field expected;
4330               pr "               (int) r->%s);\n" field;
4331               pr "      return -1;\n";
4332               pr "    }\n"
4333           | CompareWithString (field, expected) ->
4334               pr "    if (strcmp (r->%s, \"%s\") != 0) {\n" field expected;
4335               pr "      fprintf (stderr, \"%s: %s was \"%%s\", expected \"%s\"\\n\",\n"
4336                 test_name field expected;
4337               pr "               r->%s);\n" field;
4338               pr "      return -1;\n";
4339               pr "    }\n"
4340           | CompareFieldsIntEq (field1, field2) ->
4341               pr "    if (r->%s != r->%s) {\n" field1 field2;
4342               pr "      fprintf (stderr, \"%s: %s (%%d) <> %s (%%d)\\n\",\n"
4343                 test_name field1 field2;
4344               pr "               (int) r->%s, (int) r->%s);\n" field1 field2;
4345               pr "      return -1;\n";
4346               pr "    }\n"
4347           | CompareFieldsStrEq (field1, field2) ->
4348               pr "    if (strcmp (r->%s, r->%s) != 0) {\n" field1 field2;
4349               pr "      fprintf (stderr, \"%s: %s (\"%%s\") <> %s (\"%%s\")\\n\",\n"
4350                 test_name field1 field2;
4351               pr "               r->%s, r->%s);\n" field1 field2;
4352               pr "      return -1;\n";
4353               pr "    }\n"
4354         ) checks
4355       in
4356       List.iter (generate_test_command_call test_name) seq;
4357       generate_test_command_call ~test test_name last
4358   | TestLastFail seq ->
4359       pr "  /* TestLastFail for %s (%d) */\n" name i;
4360       let seq, last = get_seq_last seq in
4361       List.iter (generate_test_command_call test_name) seq;
4362       generate_test_command_call test_name ~expect_error:true last
4363
4364 (* Generate the code to run a command, leaving the result in 'r'.
4365  * If you expect to get an error then you should set expect_error:true.
4366  *)
4367 and generate_test_command_call ?(expect_error = false) ?test test_name cmd =
4368   match cmd with
4369   | [] -> assert false
4370   | name :: args ->
4371       (* Look up the command to find out what args/ret it has. *)
4372       let style =
4373         try
4374           let _, style, _, _, _, _, _ =
4375             List.find (fun (n, _, _, _, _, _, _) -> n = name) all_functions in
4376           style
4377         with Not_found ->
4378           failwithf "%s: in test, command %s was not found" test_name name in
4379
4380       if List.length (snd style) <> List.length args then
4381         failwithf "%s: in test, wrong number of args given to %s"
4382           test_name name;
4383
4384       pr "  {\n";
4385
4386       List.iter (
4387         function
4388         | OptString n, "NULL" -> ()
4389         | String n, arg
4390         | OptString n, arg ->
4391             pr "    char %s[] = \"%s\";\n" n (c_quote arg);
4392         | Int _, _
4393         | Bool _, _
4394         | FileIn _, _ | FileOut _, _ -> ()
4395         | StringList n, arg ->
4396             let strs = string_split " " arg in
4397             iteri (
4398               fun i str ->
4399                 pr "    char %s_%d[] = \"%s\";\n" n i (c_quote str);
4400             ) strs;
4401             pr "    char *%s[] = {\n" n;
4402             iteri (
4403               fun i _ -> pr "      %s_%d,\n" n i
4404             ) strs;
4405             pr "      NULL\n";
4406             pr "    };\n";
4407       ) (List.combine (snd style) args);
4408
4409       let error_code =
4410         match fst style with
4411         | RErr | RInt _ | RBool _ -> pr "    int r;\n"; "-1"
4412         | RInt64 _ -> pr "    int64_t r;\n"; "-1"
4413         | RConstString _ -> pr "    const char *r;\n"; "NULL"
4414         | RString _ -> pr "    char *r;\n"; "NULL"
4415         | RStringList _ | RHashtable _ ->
4416             pr "    char **r;\n";
4417             pr "    int i;\n";
4418             "NULL"
4419         | RIntBool _ ->
4420             pr "    struct guestfs_int_bool *r;\n"; "NULL"
4421         | RPVList _ ->
4422             pr "    struct guestfs_lvm_pv_list *r;\n"; "NULL"
4423         | RVGList _ ->
4424             pr "    struct guestfs_lvm_vg_list *r;\n"; "NULL"
4425         | RLVList _ ->
4426             pr "    struct guestfs_lvm_lv_list *r;\n"; "NULL"
4427         | RStat _ ->
4428             pr "    struct guestfs_stat *r;\n"; "NULL"
4429         | RStatVFS _ ->
4430             pr "    struct guestfs_statvfs *r;\n"; "NULL" in
4431
4432       pr "    suppress_error = %d;\n" (if expect_error then 1 else 0);
4433       pr "    r = guestfs_%s (g" name;
4434
4435       (* Generate the parameters. *)
4436       List.iter (
4437         function
4438         | OptString _, "NULL" -> pr ", NULL"
4439         | String n, _
4440         | OptString n, _ ->
4441             pr ", %s" n
4442         | FileIn _, arg | FileOut _, arg ->
4443             pr ", \"%s\"" (c_quote arg)
4444         | StringList n, _ ->
4445             pr ", %s" n
4446         | Int _, arg ->
4447             let i =
4448               try int_of_string arg
4449               with Failure "int_of_string" ->
4450                 failwithf "%s: expecting an int, but got '%s'" test_name arg in
4451             pr ", %d" i
4452         | Bool _, arg ->
4453             let b = bool_of_string arg in pr ", %d" (if b then 1 else 0)
4454       ) (List.combine (snd style) args);
4455
4456       pr ");\n";
4457       if not expect_error then
4458         pr "    if (r == %s)\n" error_code
4459       else
4460         pr "    if (r != %s)\n" error_code;
4461       pr "      return -1;\n";
4462
4463       (* Insert the test code. *)
4464       (match test with
4465        | None -> ()
4466        | Some f -> f ()
4467       );
4468
4469       (match fst style with
4470        | RErr | RInt _ | RInt64 _ | RBool _ | RConstString _ -> ()
4471        | RString _ -> pr "    free (r);\n"
4472        | RStringList _ | RHashtable _ ->
4473            pr "    for (i = 0; r[i] != NULL; ++i)\n";
4474            pr "      free (r[i]);\n";
4475            pr "    free (r);\n"
4476        | RIntBool _ ->
4477            pr "    guestfs_free_int_bool (r);\n"
4478        | RPVList _ ->
4479            pr "    guestfs_free_lvm_pv_list (r);\n"
4480        | RVGList _ ->
4481            pr "    guestfs_free_lvm_vg_list (r);\n"
4482        | RLVList _ ->
4483            pr "    guestfs_free_lvm_lv_list (r);\n"
4484        | RStat _ | RStatVFS _ ->
4485            pr "    free (r);\n"
4486       );
4487
4488       pr "  }\n"
4489
4490 and c_quote str =
4491   let str = replace_str str "\r" "\\r" in
4492   let str = replace_str str "\n" "\\n" in
4493   let str = replace_str str "\t" "\\t" in
4494   let str = replace_str str "\000" "\\0" in
4495   str
4496
4497 (* Generate a lot of different functions for guestfish. *)
4498 and generate_fish_cmds () =
4499   generate_header CStyle GPLv2;
4500
4501   let all_functions =
4502     List.filter (
4503       fun (_, _, _, flags, _, _, _) -> not (List.mem NotInFish flags)
4504     ) all_functions in
4505   let all_functions_sorted =
4506     List.filter (
4507       fun (_, _, _, flags, _, _, _) -> not (List.mem NotInFish flags)
4508     ) all_functions_sorted in
4509
4510   pr "#include <stdio.h>\n";
4511   pr "#include <stdlib.h>\n";
4512   pr "#include <string.h>\n";
4513   pr "#include <inttypes.h>\n";
4514   pr "\n";
4515   pr "#include <guestfs.h>\n";
4516   pr "#include \"fish.h\"\n";
4517   pr "\n";
4518
4519   (* list_commands function, which implements guestfish -h *)
4520   pr "void list_commands (void)\n";
4521   pr "{\n";
4522   pr "  printf (\"    %%-16s     %%s\\n\", \"Command\", \"Description\");\n";
4523   pr "  list_builtin_commands ();\n";
4524   List.iter (
4525     fun (name, _, _, flags, _, shortdesc, _) ->
4526       let name = replace_char name '_' '-' in
4527       pr "  printf (\"%%-20s %%s\\n\", \"%s\", \"%s\");\n"
4528         name shortdesc
4529   ) all_functions_sorted;
4530   pr "  printf (\"    Use -h <cmd> / help <cmd> to show detailed help for a command.\\n\");\n";
4531   pr "}\n";
4532   pr "\n";
4533
4534   (* display_command function, which implements guestfish -h cmd *)
4535   pr "void display_command (const char *cmd)\n";
4536   pr "{\n";
4537   List.iter (
4538     fun (name, style, _, flags, _, shortdesc, longdesc) ->
4539       let name2 = replace_char name '_' '-' in
4540       let alias =
4541         try find_map (function FishAlias n -> Some n | _ -> None) flags
4542         with Not_found -> name in
4543       let longdesc = replace_str longdesc "C<guestfs_" "C<" in
4544       let synopsis =
4545         match snd style with
4546         | [] -> name2
4547         | args ->
4548             sprintf "%s <%s>"
4549               name2 (String.concat "> <" (List.map name_of_argt args)) in
4550
4551       let warnings =
4552         if List.mem ProtocolLimitWarning flags then
4553           ("\n\n" ^ protocol_limit_warning)
4554         else "" in
4555
4556       (* For DangerWillRobinson commands, we should probably have
4557        * guestfish prompt before allowing you to use them (especially
4558        * in interactive mode). XXX
4559        *)
4560       let warnings =
4561         warnings ^
4562           if List.mem DangerWillRobinson flags then
4563             ("\n\n" ^ danger_will_robinson)
4564           else "" in
4565
4566       let describe_alias =
4567         if name <> alias then
4568           sprintf "\n\nYou can use '%s' as an alias for this command." alias
4569         else "" in
4570
4571       pr "  if (";
4572       pr "strcasecmp (cmd, \"%s\") == 0" name;
4573       if name <> name2 then
4574         pr " || strcasecmp (cmd, \"%s\") == 0" name2;
4575       if name <> alias then
4576         pr " || strcasecmp (cmd, \"%s\") == 0" alias;
4577       pr ")\n";
4578       pr "    pod2text (\"%s - %s\", %S);\n"
4579         name2 shortdesc
4580         (" " ^ synopsis ^ "\n\n" ^ longdesc ^ warnings ^ describe_alias);
4581       pr "  else\n"
4582   ) all_functions;
4583   pr "    display_builtin_command (cmd);\n";
4584   pr "}\n";
4585   pr "\n";
4586
4587   (* print_{pv,vg,lv}_list functions *)
4588   List.iter (
4589     function
4590     | typ, cols ->
4591         pr "static void print_%s (struct guestfs_lvm_%s *%s)\n" typ typ typ;
4592         pr "{\n";
4593         pr "  int i;\n";
4594         pr "\n";
4595         List.iter (
4596           function
4597           | name, `String ->
4598               pr "  printf (\"%s: %%s\\n\", %s->%s);\n" name typ name
4599           | name, `UUID ->
4600               pr "  printf (\"%s: \");\n" name;
4601               pr "  for (i = 0; i < 32; ++i)\n";
4602               pr "    printf (\"%%c\", %s->%s[i]);\n" typ name;
4603               pr "  printf (\"\\n\");\n"
4604           | name, `Bytes ->
4605               pr "  printf (\"%s: %%\" PRIu64 \"\\n\", %s->%s);\n" name typ name
4606           | name, `Int ->
4607               pr "  printf (\"%s: %%\" PRIi64 \"\\n\", %s->%s);\n" name typ name
4608           | name, `OptPercent ->
4609               pr "  if (%s->%s >= 0) printf (\"%s: %%g %%%%\\n\", %s->%s);\n"
4610                 typ name name typ name;
4611               pr "  else printf (\"%s: \\n\");\n" name
4612         ) cols;
4613         pr "}\n";
4614         pr "\n";
4615         pr "static void print_%s_list (struct guestfs_lvm_%s_list *%ss)\n"
4616           typ typ typ;
4617         pr "{\n";
4618         pr "  int i;\n";
4619         pr "\n";
4620         pr "  for (i = 0; i < %ss->len; ++i)\n" typ;
4621         pr "    print_%s (&%ss->val[i]);\n" typ typ;
4622         pr "}\n";
4623         pr "\n";
4624   ) ["pv", pv_cols; "vg", vg_cols; "lv", lv_cols];
4625
4626   (* print_{stat,statvfs} functions *)
4627   List.iter (
4628     function
4629     | typ, cols ->
4630         pr "static void print_%s (struct guestfs_%s *%s)\n" typ typ typ;
4631         pr "{\n";
4632         List.iter (
4633           function
4634           | name, `Int ->
4635               pr "  printf (\"%s: %%\" PRIi64 \"\\n\", %s->%s);\n" name typ name
4636         ) cols;
4637         pr "}\n";
4638         pr "\n";
4639   ) ["stat", stat_cols; "statvfs", statvfs_cols];
4640
4641   (* run_<action> actions *)
4642   List.iter (
4643     fun (name, style, _, flags, _, _, _) ->
4644       pr "static int run_%s (const char *cmd, int argc, char *argv[])\n" name;
4645       pr "{\n";
4646       (match fst style with
4647        | RErr
4648        | RInt _
4649        | RBool _ -> pr "  int r;\n"
4650        | RInt64 _ -> pr "  int64_t r;\n"
4651        | RConstString _ -> pr "  const char *r;\n"
4652        | RString _ -> pr "  char *r;\n"
4653        | RStringList _ | RHashtable _ -> pr "  char **r;\n"
4654        | RIntBool _ -> pr "  struct guestfs_int_bool *r;\n"
4655        | RPVList _ -> pr "  struct guestfs_lvm_pv_list *r;\n"
4656        | RVGList _ -> pr "  struct guestfs_lvm_vg_list *r;\n"
4657        | RLVList _ -> pr "  struct guestfs_lvm_lv_list *r;\n"
4658        | RStat _ -> pr "  struct guestfs_stat *r;\n"
4659        | RStatVFS _ -> pr "  struct guestfs_statvfs *r;\n"
4660       );
4661       List.iter (
4662         function
4663         | String n
4664         | OptString n
4665         | FileIn n
4666         | FileOut n -> pr "  const char *%s;\n" n
4667         | StringList n -> pr "  char **%s;\n" n
4668         | Bool n -> pr "  int %s;\n" n
4669         | Int n -> pr "  int %s;\n" n
4670       ) (snd style);
4671
4672       (* Check and convert parameters. *)
4673       let argc_expected = List.length (snd style) in
4674       pr "  if (argc != %d) {\n" argc_expected;
4675       pr "    fprintf (stderr, \"%%s should have %d parameter(s)\\n\", cmd);\n"
4676         argc_expected;
4677       pr "    fprintf (stderr, \"type 'help %%s' for help on %%s\\n\", cmd, cmd);\n";
4678       pr "    return -1;\n";
4679       pr "  }\n";
4680       iteri (
4681         fun i ->
4682           function
4683           | String name -> pr "  %s = argv[%d];\n" name i
4684           | OptString name ->
4685               pr "  %s = strcmp (argv[%d], \"\") != 0 ? argv[%d] : NULL;\n"
4686                 name i i
4687           | FileIn name ->
4688               pr "  %s = strcmp (argv[%d], \"-\") != 0 ? argv[%d] : \"/dev/stdin\";\n"
4689                 name i i
4690           | FileOut name ->
4691               pr "  %s = strcmp (argv[%d], \"-\") != 0 ? argv[%d] : \"/dev/stdout\";\n"
4692                 name i i
4693           | StringList name ->
4694               pr "  %s = parse_string_list (argv[%d]);\n" name i
4695           | Bool name ->
4696               pr "  %s = is_true (argv[%d]) ? 1 : 0;\n" name i
4697           | Int name ->
4698               pr "  %s = atoi (argv[%d]);\n" name i
4699       ) (snd style);
4700
4701       (* Call C API function. *)
4702       let fn =
4703         try find_map (function FishAction n -> Some n | _ -> None) flags
4704         with Not_found -> sprintf "guestfs_%s" name in
4705       pr "  r = %s " fn;
4706       generate_call_args ~handle:"g" (snd style);
4707       pr ";\n";
4708
4709       (* Check return value for errors and display command results. *)
4710       (match fst style with
4711        | RErr -> pr "  return r;\n"
4712        | RInt _ ->
4713            pr "  if (r == -1) return -1;\n";
4714            pr "  printf (\"%%d\\n\", r);\n";
4715            pr "  return 0;\n"
4716        | RInt64 _ ->
4717            pr "  if (r == -1) return -1;\n";
4718            pr "  printf (\"%%\" PRIi64 \"\\n\", r);\n";
4719            pr "  return 0;\n"
4720        | RBool _ ->
4721            pr "  if (r == -1) return -1;\n";
4722            pr "  if (r) printf (\"true\\n\"); else printf (\"false\\n\");\n";
4723            pr "  return 0;\n"
4724        | RConstString _ ->
4725            pr "  if (r == NULL) return -1;\n";
4726            pr "  printf (\"%%s\\n\", r);\n";
4727            pr "  return 0;\n"
4728        | RString _ ->
4729            pr "  if (r == NULL) return -1;\n";
4730            pr "  printf (\"%%s\\n\", r);\n";
4731            pr "  free (r);\n";
4732            pr "  return 0;\n"
4733        | RStringList _ ->
4734            pr "  if (r == NULL) return -1;\n";
4735            pr "  print_strings (r);\n";
4736            pr "  free_strings (r);\n";
4737            pr "  return 0;\n"
4738        | RIntBool _ ->
4739            pr "  if (r == NULL) return -1;\n";
4740            pr "  printf (\"%%d, %%s\\n\", r->i,\n";
4741            pr "    r->b ? \"true\" : \"false\");\n";
4742            pr "  guestfs_free_int_bool (r);\n";
4743            pr "  return 0;\n"
4744        | RPVList _ ->
4745            pr "  if (r == NULL) return -1;\n";
4746            pr "  print_pv_list (r);\n";
4747            pr "  guestfs_free_lvm_pv_list (r);\n";
4748            pr "  return 0;\n"
4749        | RVGList _ ->
4750            pr "  if (r == NULL) return -1;\n";
4751            pr "  print_vg_list (r);\n";
4752            pr "  guestfs_free_lvm_vg_list (r);\n";
4753            pr "  return 0;\n"
4754        | RLVList _ ->
4755            pr "  if (r == NULL) return -1;\n";
4756            pr "  print_lv_list (r);\n";
4757            pr "  guestfs_free_lvm_lv_list (r);\n";
4758            pr "  return 0;\n"
4759        | RStat _ ->
4760            pr "  if (r == NULL) return -1;\n";
4761            pr "  print_stat (r);\n";
4762            pr "  free (r);\n";
4763            pr "  return 0;\n"
4764        | RStatVFS _ ->
4765            pr "  if (r == NULL) return -1;\n";
4766            pr "  print_statvfs (r);\n";
4767            pr "  free (r);\n";
4768            pr "  return 0;\n"
4769        | RHashtable _ ->
4770            pr "  if (r == NULL) return -1;\n";
4771            pr "  print_table (r);\n";
4772            pr "  free_strings (r);\n";
4773            pr "  return 0;\n"
4774       );
4775       pr "}\n";
4776       pr "\n"
4777   ) all_functions;
4778
4779   (* run_action function *)
4780   pr "int run_action (const char *cmd, int argc, char *argv[])\n";
4781   pr "{\n";
4782   List.iter (
4783     fun (name, _, _, flags, _, _, _) ->
4784       let name2 = replace_char name '_' '-' in
4785       let alias =
4786         try find_map (function FishAlias n -> Some n | _ -> None) flags
4787         with Not_found -> name in
4788       pr "  if (";
4789       pr "strcasecmp (cmd, \"%s\") == 0" name;
4790       if name <> name2 then
4791         pr " || strcasecmp (cmd, \"%s\") == 0" name2;
4792       if name <> alias then
4793         pr " || strcasecmp (cmd, \"%s\") == 0" alias;
4794       pr ")\n";
4795       pr "    return run_%s (cmd, argc, argv);\n" name;
4796       pr "  else\n";
4797   ) all_functions;
4798   pr "    {\n";
4799   pr "      fprintf (stderr, \"%%s: unknown command\\n\", cmd);\n";
4800   pr "      return -1;\n";
4801   pr "    }\n";
4802   pr "  return 0;\n";
4803   pr "}\n";
4804   pr "\n"
4805
4806 (* Readline completion for guestfish. *)
4807 and generate_fish_completion () =
4808   generate_header CStyle GPLv2;
4809
4810   let all_functions =
4811     List.filter (
4812       fun (_, _, _, flags, _, _, _) -> not (List.mem NotInFish flags)
4813     ) all_functions in
4814
4815   pr "\
4816 #include <config.h>
4817
4818 #include <stdio.h>
4819 #include <stdlib.h>
4820 #include <string.h>
4821
4822 #ifdef HAVE_LIBREADLINE
4823 #include <readline/readline.h>
4824 #endif
4825
4826 #include \"fish.h\"
4827
4828 #ifdef HAVE_LIBREADLINE
4829
4830 static const char *const commands[] = {
4831   BUILTIN_COMMANDS_FOR_COMPLETION,
4832 ";
4833
4834   (* Get the commands, including the aliases.  They don't need to be
4835    * sorted - the generator() function just does a dumb linear search.
4836    *)
4837   let commands =
4838     List.map (
4839       fun (name, _, _, flags, _, _, _) ->
4840         let name2 = replace_char name '_' '-' in
4841         let alias =
4842           try find_map (function FishAlias n -> Some n | _ -> None) flags
4843           with Not_found -> name in
4844
4845         if name <> alias then [name2; alias] else [name2]
4846     ) all_functions in
4847   let commands = List.flatten commands in
4848
4849   List.iter (pr "  \"%s\",\n") commands;
4850
4851   pr "  NULL
4852 };
4853
4854 static char *
4855 generator (const char *text, int state)
4856 {
4857   static int index, len;
4858   const char *name;
4859
4860   if (!state) {
4861     index = 0;
4862     len = strlen (text);
4863   }
4864
4865   rl_attempted_completion_over = 1;
4866
4867   while ((name = commands[index]) != NULL) {
4868     index++;
4869     if (strncasecmp (name, text, len) == 0)
4870       return strdup (name);
4871   }
4872
4873   return NULL;
4874 }
4875
4876 #endif /* HAVE_LIBREADLINE */
4877
4878 char **do_completion (const char *text, int start, int end)
4879 {
4880   char **matches = NULL;
4881
4882 #ifdef HAVE_LIBREADLINE
4883   rl_completion_append_character = ' ';
4884
4885   if (start == 0)
4886     matches = rl_completion_matches (text, generator);
4887   else if (complete_dest_paths)
4888     matches = rl_completion_matches (text, complete_dest_paths_generator);
4889 #endif
4890
4891   return matches;
4892 }
4893 ";
4894
4895 (* Generate the POD documentation for guestfish. *)
4896 and generate_fish_actions_pod () =
4897   let all_functions_sorted =
4898     List.filter (
4899       fun (_, _, _, flags, _, _, _) ->
4900         not (List.mem NotInFish flags || List.mem NotInDocs flags)
4901     ) all_functions_sorted in
4902
4903   let rex = Str.regexp "C<guestfs_\\([^>]+\\)>" in
4904
4905   List.iter (
4906     fun (name, style, _, flags, _, _, longdesc) ->
4907       let longdesc =
4908         Str.global_substitute rex (
4909           fun s ->
4910             let sub =
4911               try Str.matched_group 1 s
4912               with Not_found ->
4913                 failwithf "error substituting C<guestfs_...> in longdesc of function %s" name in
4914             "C<" ^ replace_char sub '_' '-' ^ ">"
4915         ) longdesc in
4916       let name = replace_char name '_' '-' in
4917       let alias =
4918         try find_map (function FishAlias n -> Some n | _ -> None) flags
4919         with Not_found -> name in
4920
4921       pr "=head2 %s" name;
4922       if name <> alias then
4923         pr " | %s" alias;
4924       pr "\n";
4925       pr "\n";
4926       pr " %s" name;
4927       List.iter (
4928         function
4929         | String n -> pr " %s" n
4930         | OptString n -> pr " %s" n
4931         | StringList n -> pr " '%s ...'" n
4932         | Bool _ -> pr " true|false"
4933         | Int n -> pr " %s" n
4934         | FileIn n | FileOut n -> pr " (%s|-)" n
4935       ) (snd style);
4936       pr "\n";
4937       pr "\n";
4938       pr "%s\n\n" longdesc;
4939
4940       if List.exists (function FileIn _ | FileOut _ -> true
4941                       | _ -> false) (snd style) then
4942         pr "Use C<-> instead of a filename to read/write from stdin/stdout.\n\n";
4943
4944       if List.mem ProtocolLimitWarning flags then
4945         pr "%s\n\n" protocol_limit_warning;
4946
4947       if List.mem DangerWillRobinson flags then
4948         pr "%s\n\n" danger_will_robinson
4949   ) all_functions_sorted
4950
4951 (* Generate a C function prototype. *)
4952 and generate_prototype ?(extern = true) ?(static = false) ?(semicolon = true)
4953     ?(single_line = false) ?(newline = false) ?(in_daemon = false)
4954     ?(prefix = "")
4955     ?handle name style =
4956   if extern then pr "extern ";
4957   if static then pr "static ";
4958   (match fst style with
4959    | RErr -> pr "int "
4960    | RInt _ -> pr "int "
4961    | RInt64 _ -> pr "int64_t "
4962    | RBool _ -> pr "int "
4963    | RConstString _ -> pr "const char *"
4964    | RString _ -> pr "char *"
4965    | RStringList _ | RHashtable _ -> pr "char **"
4966    | RIntBool _ ->
4967        if not in_daemon then pr "struct guestfs_int_bool *"
4968        else pr "guestfs_%s_ret *" name
4969    | RPVList _ ->
4970        if not in_daemon then pr "struct guestfs_lvm_pv_list *"
4971        else pr "guestfs_lvm_int_pv_list *"
4972    | RVGList _ ->
4973        if not in_daemon then pr "struct guestfs_lvm_vg_list *"
4974        else pr "guestfs_lvm_int_vg_list *"
4975    | RLVList _ ->
4976        if not in_daemon then pr "struct guestfs_lvm_lv_list *"
4977        else pr "guestfs_lvm_int_lv_list *"
4978    | RStat _ ->
4979        if not in_daemon then pr "struct guestfs_stat *"
4980        else pr "guestfs_int_stat *"
4981    | RStatVFS _ ->
4982        if not in_daemon then pr "struct guestfs_statvfs *"
4983        else pr "guestfs_int_statvfs *"
4984   );
4985   pr "%s%s (" prefix name;
4986   if handle = None && List.length (snd style) = 0 then
4987     pr "void"
4988   else (
4989     let comma = ref false in
4990     (match handle with
4991      | None -> ()
4992      | Some handle -> pr "guestfs_h *%s" handle; comma := true
4993     );
4994     let next () =
4995       if !comma then (
4996         if single_line then pr ", " else pr ",\n\t\t"
4997       );
4998       comma := true
4999     in
5000     List.iter (
5001       function
5002       | String n
5003       | OptString n ->
5004           next ();
5005           if not in_daemon then pr "const char *%s" n
5006           else pr "char *%s" n
5007       | StringList n ->
5008           next ();
5009           if not in_daemon then pr "char * const* const %s" n
5010           else pr "char **%s" n
5011       | Bool n -> next (); pr "int %s" n
5012       | Int n -> next (); pr "int %s" n
5013       | FileIn n
5014       | FileOut n ->
5015           if not in_daemon then (next (); pr "const char *%s" n)
5016     ) (snd style);
5017   );
5018   pr ")";
5019   if semicolon then pr ";";
5020   if newline then pr "\n"
5021
5022 (* Generate C call arguments, eg "(handle, foo, bar)" *)
5023 and generate_call_args ?handle args =
5024   pr "(";
5025   let comma = ref false in
5026   (match handle with
5027    | None -> ()
5028    | Some handle -> pr "%s" handle; comma := true
5029   );
5030   List.iter (
5031     fun arg ->
5032       if !comma then pr ", ";
5033       comma := true;
5034       pr "%s" (name_of_argt arg)
5035   ) args;
5036   pr ")"
5037
5038 (* Generate the OCaml bindings interface. *)
5039 and generate_ocaml_mli () =
5040   generate_header OCamlStyle LGPLv2;
5041
5042   pr "\
5043 (** For API documentation you should refer to the C API
5044     in the guestfs(3) manual page.  The OCaml API uses almost
5045     exactly the same calls. *)
5046
5047 type t
5048 (** A [guestfs_h] handle. *)
5049
5050 exception Error of string
5051 (** This exception is raised when there is an error. *)
5052
5053 val create : unit -> t
5054
5055 val close : t -> unit
5056 (** Handles are closed by the garbage collector when they become
5057     unreferenced, but callers can also call this in order to
5058     provide predictable cleanup. *)
5059
5060 ";
5061   generate_ocaml_lvm_structure_decls ();
5062
5063   generate_ocaml_stat_structure_decls ();
5064
5065   (* The actions. *)
5066   List.iter (
5067     fun (name, style, _, _, _, shortdesc, _) ->
5068       generate_ocaml_prototype name style;
5069       pr "(** %s *)\n" shortdesc;
5070       pr "\n"
5071   ) all_functions
5072
5073 (* Generate the OCaml bindings implementation. *)
5074 and generate_ocaml_ml () =
5075   generate_header OCamlStyle LGPLv2;
5076
5077   pr "\
5078 type t
5079 exception Error of string
5080 external create : unit -> t = \"ocaml_guestfs_create\"
5081 external close : t -> unit = \"ocaml_guestfs_close\"
5082
5083 let () =
5084   Callback.register_exception \"ocaml_guestfs_error\" (Error \"\")
5085
5086 ";
5087
5088   generate_ocaml_lvm_structure_decls ();
5089
5090   generate_ocaml_stat_structure_decls ();
5091
5092   (* The actions. *)
5093   List.iter (
5094     fun (name, style, _, _, _, shortdesc, _) ->
5095       generate_ocaml_prototype ~is_external:true name style;
5096   ) all_functions
5097
5098 (* Generate the OCaml bindings C implementation. *)
5099 and generate_ocaml_c () =
5100   generate_header CStyle LGPLv2;
5101
5102   pr "\
5103 #include <stdio.h>
5104 #include <stdlib.h>
5105 #include <string.h>
5106
5107 #include <caml/config.h>
5108 #include <caml/alloc.h>
5109 #include <caml/callback.h>
5110 #include <caml/fail.h>
5111 #include <caml/memory.h>
5112 #include <caml/mlvalues.h>
5113 #include <caml/signals.h>
5114
5115 #include <guestfs.h>
5116
5117 #include \"guestfs_c.h\"
5118
5119 /* Copy a hashtable of string pairs into an assoc-list.  We return
5120  * the list in reverse order, but hashtables aren't supposed to be
5121  * ordered anyway.
5122  */
5123 static CAMLprim value
5124 copy_table (char * const * argv)
5125 {
5126   CAMLparam0 ();
5127   CAMLlocal5 (rv, pairv, kv, vv, cons);
5128   int i;
5129
5130   rv = Val_int (0);
5131   for (i = 0; argv[i] != NULL; i += 2) {
5132     kv = caml_copy_string (argv[i]);
5133     vv = caml_copy_string (argv[i+1]);
5134     pairv = caml_alloc (2, 0);
5135     Store_field (pairv, 0, kv);
5136     Store_field (pairv, 1, vv);
5137     cons = caml_alloc (2, 0);
5138     Store_field (cons, 1, rv);
5139     rv = cons;
5140     Store_field (cons, 0, pairv);
5141   }
5142
5143   CAMLreturn (rv);
5144 }
5145
5146 ";
5147
5148   (* LVM struct copy functions. *)
5149   List.iter (
5150     fun (typ, cols) ->
5151       let has_optpercent_col =
5152         List.exists (function (_, `OptPercent) -> true | _ -> false) cols in
5153
5154       pr "static CAMLprim value\n";
5155       pr "copy_lvm_%s (const struct guestfs_lvm_%s *%s)\n" typ typ typ;
5156       pr "{\n";
5157       pr "  CAMLparam0 ();\n";
5158       if has_optpercent_col then
5159         pr "  CAMLlocal3 (rv, v, v2);\n"
5160       else
5161         pr "  CAMLlocal2 (rv, v);\n";
5162       pr "\n";
5163       pr "  rv = caml_alloc (%d, 0);\n" (List.length cols);
5164       iteri (
5165         fun i col ->
5166           (match col with
5167            | name, `String ->
5168                pr "  v = caml_copy_string (%s->%s);\n" typ name
5169            | name, `UUID ->
5170                pr "  v = caml_alloc_string (32);\n";
5171                pr "  memcpy (String_val (v), %s->%s, 32);\n" typ name
5172            | name, `Bytes
5173            | name, `Int ->
5174                pr "  v = caml_copy_int64 (%s->%s);\n" typ name
5175            | name, `OptPercent ->
5176                pr "  if (%s->%s >= 0) { /* Some %s */\n" typ name name;
5177                pr "    v2 = caml_copy_double (%s->%s);\n" typ name;
5178                pr "    v = caml_alloc (1, 0);\n";
5179                pr "    Store_field (v, 0, v2);\n";
5180                pr "  } else /* None */\n";
5181                pr "    v = Val_int (0);\n";
5182           );
5183           pr "  Store_field (rv, %d, v);\n" i
5184       ) cols;
5185       pr "  CAMLreturn (rv);\n";
5186       pr "}\n";
5187       pr "\n";
5188
5189       pr "static CAMLprim value\n";
5190       pr "copy_lvm_%s_list (const struct guestfs_lvm_%s_list *%ss)\n"
5191         typ typ typ;
5192       pr "{\n";
5193       pr "  CAMLparam0 ();\n";
5194       pr "  CAMLlocal2 (rv, v);\n";
5195       pr "  int i;\n";
5196       pr "\n";
5197       pr "  if (%ss->len == 0)\n" typ;
5198       pr "    CAMLreturn (Atom (0));\n";
5199       pr "  else {\n";
5200       pr "    rv = caml_alloc (%ss->len, 0);\n" typ;
5201       pr "    for (i = 0; i < %ss->len; ++i) {\n" typ;
5202       pr "      v = copy_lvm_%s (&%ss->val[i]);\n" typ typ;
5203       pr "      caml_modify (&Field (rv, i), v);\n";
5204       pr "    }\n";
5205       pr "    CAMLreturn (rv);\n";
5206       pr "  }\n";
5207       pr "}\n";
5208       pr "\n";
5209   ) ["pv", pv_cols; "vg", vg_cols; "lv", lv_cols];
5210
5211   (* Stat copy functions. *)
5212   List.iter (
5213     fun (typ, cols) ->
5214       pr "static CAMLprim value\n";
5215       pr "copy_%s (const struct guestfs_%s *%s)\n" typ typ typ;
5216       pr "{\n";
5217       pr "  CAMLparam0 ();\n";
5218       pr "  CAMLlocal2 (rv, v);\n";
5219       pr "\n";
5220       pr "  rv = caml_alloc (%d, 0);\n" (List.length cols);
5221       iteri (
5222         fun i col ->
5223           (match col with
5224            | name, `Int ->
5225                pr "  v = caml_copy_int64 (%s->%s);\n" typ name
5226           );
5227           pr "  Store_field (rv, %d, v);\n" i
5228       ) cols;
5229       pr "  CAMLreturn (rv);\n";
5230       pr "}\n";
5231       pr "\n";
5232   ) ["stat", stat_cols; "statvfs", statvfs_cols];
5233
5234   (* The wrappers. *)
5235   List.iter (
5236     fun (name, style, _, _, _, _, _) ->
5237       let params =
5238         "gv" :: List.map (fun arg -> name_of_argt arg ^ "v") (snd style) in
5239
5240       pr "CAMLprim value\n";
5241       pr "ocaml_guestfs_%s (value %s" name (List.hd params);
5242       List.iter (pr ", value %s") (List.tl params);
5243       pr ")\n";
5244       pr "{\n";
5245
5246       (match params with
5247        | [p1; p2; p3; p4; p5] ->
5248            pr "  CAMLparam5 (%s);\n" (String.concat ", " params)
5249        | p1 :: p2 :: p3 :: p4 :: p5 :: rest ->
5250            pr "  CAMLparam5 (%s);\n" (String.concat ", " [p1; p2; p3; p4; p5]);
5251            pr "  CAMLxparam%d (%s);\n"
5252              (List.length rest) (String.concat ", " rest)
5253        | ps ->
5254            pr "  CAMLparam%d (%s);\n" (List.length ps) (String.concat ", " ps)
5255       );
5256       pr "  CAMLlocal1 (rv);\n";
5257       pr "\n";
5258
5259       pr "  guestfs_h *g = Guestfs_val (gv);\n";
5260       pr "  if (g == NULL)\n";
5261       pr "    caml_failwith (\"%s: used handle after closing it\");\n" name;
5262       pr "\n";
5263
5264       List.iter (
5265         function
5266         | String n
5267         | FileIn n
5268         | FileOut n ->
5269             pr "  const char *%s = String_val (%sv);\n" n n
5270         | OptString n ->
5271             pr "  const char *%s =\n" n;
5272             pr "    %sv != Val_int (0) ? String_val (Field (%sv, 0)) : NULL;\n"
5273               n n
5274         | StringList n ->
5275             pr "  char **%s = ocaml_guestfs_strings_val (g, %sv);\n" n n
5276         | Bool n ->
5277             pr "  int %s = Bool_val (%sv);\n" n n
5278         | Int n ->
5279             pr "  int %s = Int_val (%sv);\n" n n
5280       ) (snd style);
5281       let error_code =
5282         match fst style with
5283         | RErr -> pr "  int r;\n"; "-1"
5284         | RInt _ -> pr "  int r;\n"; "-1"
5285         | RInt64 _ -> pr "  int64_t r;\n"; "-1"
5286         | RBool _ -> pr "  int r;\n"; "-1"
5287         | RConstString _ -> pr "  const char *r;\n"; "NULL"
5288         | RString _ -> pr "  char *r;\n"; "NULL"
5289         | RStringList _ ->
5290             pr "  int i;\n";
5291             pr "  char **r;\n";
5292             "NULL"
5293         | RIntBool _ ->
5294             pr "  struct guestfs_int_bool *r;\n"; "NULL"
5295         | RPVList _ ->
5296             pr "  struct guestfs_lvm_pv_list *r;\n"; "NULL"
5297         | RVGList _ ->
5298             pr "  struct guestfs_lvm_vg_list *r;\n"; "NULL"
5299         | RLVList _ ->
5300             pr "  struct guestfs_lvm_lv_list *r;\n"; "NULL"
5301         | RStat _ ->
5302             pr "  struct guestfs_stat *r;\n"; "NULL"
5303         | RStatVFS _ ->
5304             pr "  struct guestfs_statvfs *r;\n"; "NULL"
5305         | RHashtable _ ->
5306             pr "  int i;\n";
5307             pr "  char **r;\n";
5308             "NULL" in
5309       pr "\n";
5310
5311       pr "  caml_enter_blocking_section ();\n";
5312       pr "  r = guestfs_%s " name;
5313       generate_call_args ~handle:"g" (snd style);
5314       pr ";\n";
5315       pr "  caml_leave_blocking_section ();\n";
5316
5317       List.iter (
5318         function
5319         | StringList n ->
5320             pr "  ocaml_guestfs_free_strings (%s);\n" n;
5321         | String _ | OptString _ | Bool _ | Int _ | FileIn _ | FileOut _ -> ()
5322       ) (snd style);
5323
5324       pr "  if (r == %s)\n" error_code;
5325       pr "    ocaml_guestfs_raise_error (g, \"%s\");\n" name;
5326       pr "\n";
5327
5328       (match fst style with
5329        | RErr -> pr "  rv = Val_unit;\n"
5330        | RInt _ -> pr "  rv = Val_int (r);\n"
5331        | RInt64 _ ->
5332            pr "  rv = caml_copy_int64 (r);\n"
5333        | RBool _ -> pr "  rv = Val_bool (r);\n"
5334        | RConstString _ -> pr "  rv = caml_copy_string (r);\n"
5335        | RString _ ->
5336            pr "  rv = caml_copy_string (r);\n";
5337            pr "  free (r);\n"
5338        | RStringList _ ->
5339            pr "  rv = caml_copy_string_array ((const char **) r);\n";
5340            pr "  for (i = 0; r[i] != NULL; ++i) free (r[i]);\n";
5341            pr "  free (r);\n"
5342        | RIntBool _ ->
5343            pr "  rv = caml_alloc (2, 0);\n";
5344            pr "  Store_field (rv, 0, Val_int (r->i));\n";
5345            pr "  Store_field (rv, 1, Val_bool (r->b));\n";
5346            pr "  guestfs_free_int_bool (r);\n";
5347        | RPVList _ ->
5348            pr "  rv = copy_lvm_pv_list (r);\n";
5349            pr "  guestfs_free_lvm_pv_list (r);\n";
5350        | RVGList _ ->
5351            pr "  rv = copy_lvm_vg_list (r);\n";
5352            pr "  guestfs_free_lvm_vg_list (r);\n";
5353        | RLVList _ ->
5354            pr "  rv = copy_lvm_lv_list (r);\n";
5355            pr "  guestfs_free_lvm_lv_list (r);\n";
5356        | RStat _ ->
5357            pr "  rv = copy_stat (r);\n";
5358            pr "  free (r);\n";
5359        | RStatVFS _ ->
5360            pr "  rv = copy_statvfs (r);\n";
5361            pr "  free (r);\n";
5362        | RHashtable _ ->
5363            pr "  rv = copy_table (r);\n";
5364            pr "  for (i = 0; r[i] != NULL; ++i) free (r[i]);\n";
5365            pr "  free (r);\n";
5366       );
5367
5368       pr "  CAMLreturn (rv);\n";
5369       pr "}\n";
5370       pr "\n";
5371
5372       if List.length params > 5 then (
5373         pr "CAMLprim value\n";
5374         pr "ocaml_guestfs_%s_byte (value *argv, int argn)\n" name;
5375         pr "{\n";
5376         pr "  return ocaml_guestfs_%s (argv[0]" name;
5377         iteri (fun i _ -> pr ", argv[%d]" i) (List.tl params);
5378         pr ");\n";
5379         pr "}\n";
5380         pr "\n"
5381       )
5382   ) all_functions
5383
5384 and generate_ocaml_lvm_structure_decls () =
5385   List.iter (
5386     fun (typ, cols) ->
5387       pr "type lvm_%s = {\n" typ;
5388       List.iter (
5389         function
5390         | name, `String -> pr "  %s : string;\n" name
5391         | name, `UUID -> pr "  %s : string;\n" name
5392         | name, `Bytes -> pr "  %s : int64;\n" name
5393         | name, `Int -> pr "  %s : int64;\n" name
5394         | name, `OptPercent -> pr "  %s : float option;\n" name
5395       ) cols;
5396       pr "}\n";
5397       pr "\n"
5398   ) ["pv", pv_cols; "vg", vg_cols; "lv", lv_cols]
5399
5400 and generate_ocaml_stat_structure_decls () =
5401   List.iter (
5402     fun (typ, cols) ->
5403       pr "type %s = {\n" typ;
5404       List.iter (
5405         function
5406         | name, `Int -> pr "  %s : int64;\n" name
5407       ) cols;
5408       pr "}\n";
5409       pr "\n"
5410   ) ["stat", stat_cols; "statvfs", statvfs_cols]
5411
5412 and generate_ocaml_prototype ?(is_external = false) name style =
5413   if is_external then pr "external " else pr "val ";
5414   pr "%s : t -> " name;
5415   List.iter (
5416     function
5417     | String _ | FileIn _ | FileOut _ -> pr "string -> "
5418     | OptString _ -> pr "string option -> "
5419     | StringList _ -> pr "string array -> "
5420     | Bool _ -> pr "bool -> "
5421     | Int _ -> pr "int -> "
5422   ) (snd style);
5423   (match fst style with
5424    | RErr -> pr "unit" (* all errors are turned into exceptions *)
5425    | RInt _ -> pr "int"
5426    | RInt64 _ -> pr "int64"
5427    | RBool _ -> pr "bool"
5428    | RConstString _ -> pr "string"
5429    | RString _ -> pr "string"
5430    | RStringList _ -> pr "string array"
5431    | RIntBool _ -> pr "int * bool"
5432    | RPVList _ -> pr "lvm_pv array"
5433    | RVGList _ -> pr "lvm_vg array"
5434    | RLVList _ -> pr "lvm_lv array"
5435    | RStat _ -> pr "stat"
5436    | RStatVFS _ -> pr "statvfs"
5437    | RHashtable _ -> pr "(string * string) list"
5438   );
5439   if is_external then (
5440     pr " = ";
5441     if List.length (snd style) + 1 > 5 then
5442       pr "\"ocaml_guestfs_%s_byte\" " name;
5443     pr "\"ocaml_guestfs_%s\"" name
5444   );
5445   pr "\n"
5446
5447 (* Generate Perl xs code, a sort of crazy variation of C with macros. *)
5448 and generate_perl_xs () =
5449   generate_header CStyle LGPLv2;
5450
5451   pr "\
5452 #include \"EXTERN.h\"
5453 #include \"perl.h\"
5454 #include \"XSUB.h\"
5455
5456 #include <guestfs.h>
5457
5458 #ifndef PRId64
5459 #define PRId64 \"lld\"
5460 #endif
5461
5462 static SV *
5463 my_newSVll(long long val) {
5464 #ifdef USE_64_BIT_ALL
5465   return newSViv(val);
5466 #else
5467   char buf[100];
5468   int len;
5469   len = snprintf(buf, 100, \"%%\" PRId64, val);
5470   return newSVpv(buf, len);
5471 #endif
5472 }
5473
5474 #ifndef PRIu64
5475 #define PRIu64 \"llu\"
5476 #endif
5477
5478 static SV *
5479 my_newSVull(unsigned long long val) {
5480 #ifdef USE_64_BIT_ALL
5481   return newSVuv(val);
5482 #else
5483   char buf[100];
5484   int len;
5485   len = snprintf(buf, 100, \"%%\" PRIu64, val);
5486   return newSVpv(buf, len);
5487 #endif
5488 }
5489
5490 /* http://www.perlmonks.org/?node_id=680842 */
5491 static char **
5492 XS_unpack_charPtrPtr (SV *arg) {
5493   char **ret;
5494   AV *av;
5495   I32 i;
5496
5497   if (!arg || !SvOK (arg) || !SvROK (arg) || SvTYPE (SvRV (arg)) != SVt_PVAV)
5498     croak (\"array reference expected\");
5499
5500   av = (AV *)SvRV (arg);
5501   ret = malloc ((av_len (av) + 1 + 1) * sizeof (char *));
5502   if (!ret)
5503     croak (\"malloc failed\");
5504
5505   for (i = 0; i <= av_len (av); i++) {
5506     SV **elem = av_fetch (av, i, 0);
5507
5508     if (!elem || !*elem)
5509       croak (\"missing element in list\");
5510
5511     ret[i] = SvPV_nolen (*elem);
5512   }
5513
5514   ret[i] = NULL;
5515
5516   return ret;
5517 }
5518
5519 MODULE = Sys::Guestfs  PACKAGE = Sys::Guestfs
5520
5521 PROTOTYPES: ENABLE
5522
5523 guestfs_h *
5524 _create ()
5525    CODE:
5526       RETVAL = guestfs_create ();
5527       if (!RETVAL)
5528         croak (\"could not create guestfs handle\");
5529       guestfs_set_error_handler (RETVAL, NULL, NULL);
5530  OUTPUT:
5531       RETVAL
5532
5533 void
5534 DESTROY (g)
5535       guestfs_h *g;
5536  PPCODE:
5537       guestfs_close (g);
5538
5539 ";
5540
5541   List.iter (
5542     fun (name, style, _, _, _, _, _) ->
5543       (match fst style with
5544        | RErr -> pr "void\n"
5545        | RInt _ -> pr "SV *\n"
5546        | RInt64 _ -> pr "SV *\n"
5547        | RBool _ -> pr "SV *\n"
5548        | RConstString _ -> pr "SV *\n"
5549        | RString _ -> pr "SV *\n"
5550        | RStringList _
5551        | RIntBool _
5552        | RPVList _ | RVGList _ | RLVList _
5553        | RStat _ | RStatVFS _
5554        | RHashtable _ ->
5555            pr "void\n" (* all lists returned implictly on the stack *)
5556       );
5557       (* Call and arguments. *)
5558       pr "%s " name;
5559       generate_call_args ~handle:"g" (snd style);
5560       pr "\n";
5561       pr "      guestfs_h *g;\n";
5562       iteri (
5563         fun i ->
5564           function
5565           | String n | FileIn n | FileOut n -> pr "      char *%s;\n" n
5566           | OptString n ->
5567               (* http://www.perlmonks.org/?node_id=554277
5568                * Note that the implicit handle argument means we have
5569                * to add 1 to the ST(x) operator.
5570                *)
5571               pr "      char *%s = SvOK(ST(%d)) ? SvPV_nolen(ST(%d)) : NULL;\n" n (i+1) (i+1)
5572           | StringList n -> pr "      char **%s;\n" n
5573           | Bool n -> pr "      int %s;\n" n
5574           | Int n -> pr "      int %s;\n" n
5575       ) (snd style);
5576
5577       let do_cleanups () =
5578         List.iter (
5579           function
5580           | String _ | OptString _ | Bool _ | Int _
5581           | FileIn _ | FileOut _ -> ()
5582           | StringList n -> pr "      free (%s);\n" n
5583         ) (snd style)
5584       in
5585
5586       (* Code. *)
5587       (match fst style with
5588        | RErr ->
5589            pr "PREINIT:\n";
5590            pr "      int r;\n";
5591            pr " PPCODE:\n";
5592            pr "      r = guestfs_%s " name;
5593            generate_call_args ~handle:"g" (snd style);
5594            pr ";\n";
5595            do_cleanups ();
5596            pr "      if (r == -1)\n";
5597            pr "        croak (\"%s: %%s\", guestfs_last_error (g));\n" name;
5598        | RInt n
5599        | RBool n ->
5600            pr "PREINIT:\n";
5601            pr "      int %s;\n" n;
5602            pr "   CODE:\n";
5603            pr "      %s = guestfs_%s " n name;
5604            generate_call_args ~handle:"g" (snd style);
5605            pr ";\n";
5606            do_cleanups ();
5607            pr "      if (%s == -1)\n" n;
5608            pr "        croak (\"%s: %%s\", guestfs_last_error (g));\n" name;
5609            pr "      RETVAL = newSViv (%s);\n" n;
5610            pr " OUTPUT:\n";
5611            pr "      RETVAL\n"
5612        | RInt64 n ->
5613            pr "PREINIT:\n";
5614            pr "      int64_t %s;\n" n;
5615            pr "   CODE:\n";
5616            pr "      %s = guestfs_%s " n name;
5617            generate_call_args ~handle:"g" (snd style);
5618            pr ";\n";
5619            do_cleanups ();
5620            pr "      if (%s == -1)\n" n;
5621            pr "        croak (\"%s: %%s\", guestfs_last_error (g));\n" name;
5622            pr "      RETVAL = my_newSVll (%s);\n" n;
5623            pr " OUTPUT:\n";
5624            pr "      RETVAL\n"
5625        | RConstString n ->
5626            pr "PREINIT:\n";
5627            pr "      const char *%s;\n" n;
5628            pr "   CODE:\n";
5629            pr "      %s = guestfs_%s " n name;
5630            generate_call_args ~handle:"g" (snd style);
5631            pr ";\n";
5632            do_cleanups ();
5633            pr "      if (%s == NULL)\n" n;
5634            pr "        croak (\"%s: %%s\", guestfs_last_error (g));\n" name;
5635            pr "      RETVAL = newSVpv (%s, 0);\n" n;
5636            pr " OUTPUT:\n";
5637            pr "      RETVAL\n"
5638        | RString n ->
5639            pr "PREINIT:\n";
5640            pr "      char *%s;\n" n;
5641            pr "   CODE:\n";
5642            pr "      %s = guestfs_%s " n name;
5643            generate_call_args ~handle:"g" (snd style);
5644            pr ";\n";
5645            do_cleanups ();
5646            pr "      if (%s == NULL)\n" n;
5647            pr "        croak (\"%s: %%s\", guestfs_last_error (g));\n" name;
5648            pr "      RETVAL = newSVpv (%s, 0);\n" n;
5649            pr "      free (%s);\n" n;
5650            pr " OUTPUT:\n";
5651            pr "      RETVAL\n"
5652        | RStringList n | RHashtable n ->
5653            pr "PREINIT:\n";
5654            pr "      char **%s;\n" n;
5655            pr "      int i, n;\n";
5656            pr " PPCODE:\n";
5657            pr "      %s = guestfs_%s " n name;
5658            generate_call_args ~handle:"g" (snd style);
5659            pr ";\n";
5660            do_cleanups ();
5661            pr "      if (%s == NULL)\n" n;
5662            pr "        croak (\"%s: %%s\", guestfs_last_error (g));\n" name;
5663            pr "      for (n = 0; %s[n] != NULL; ++n) /**/;\n" n;
5664            pr "      EXTEND (SP, n);\n";
5665            pr "      for (i = 0; i < n; ++i) {\n";
5666            pr "        PUSHs (sv_2mortal (newSVpv (%s[i], 0)));\n" n;
5667            pr "        free (%s[i]);\n" n;
5668            pr "      }\n";
5669            pr "      free (%s);\n" n;
5670        | RIntBool _ ->
5671            pr "PREINIT:\n";
5672            pr "      struct guestfs_int_bool *r;\n";
5673            pr " PPCODE:\n";
5674            pr "      r = guestfs_%s " name;
5675            generate_call_args ~handle:"g" (snd style);
5676            pr ";\n";
5677            do_cleanups ();
5678            pr "      if (r == NULL)\n";
5679            pr "        croak (\"%s: %%s\", guestfs_last_error (g));\n" name;
5680            pr "      EXTEND (SP, 2);\n";
5681            pr "      PUSHs (sv_2mortal (newSViv (r->i)));\n";
5682            pr "      PUSHs (sv_2mortal (newSViv (r->b)));\n";
5683            pr "      guestfs_free_int_bool (r);\n";
5684        | RPVList n ->
5685            generate_perl_lvm_code "pv" pv_cols name style n do_cleanups
5686        | RVGList n ->
5687            generate_perl_lvm_code "vg" vg_cols name style n do_cleanups
5688        | RLVList n ->
5689            generate_perl_lvm_code "lv" lv_cols name style n do_cleanups
5690        | RStat n ->
5691            generate_perl_stat_code "stat" stat_cols name style n do_cleanups
5692        | RStatVFS n ->
5693            generate_perl_stat_code
5694              "statvfs" statvfs_cols name style n do_cleanups
5695       );
5696
5697       pr "\n"
5698   ) all_functions
5699
5700 and generate_perl_lvm_code typ cols name style n do_cleanups =
5701   pr "PREINIT:\n";
5702   pr "      struct guestfs_lvm_%s_list *%s;\n" typ n;
5703   pr "      int i;\n";
5704   pr "      HV *hv;\n";
5705   pr " PPCODE:\n";
5706   pr "      %s = guestfs_%s " n name;
5707   generate_call_args ~handle:"g" (snd style);
5708   pr ";\n";
5709   do_cleanups ();
5710   pr "      if (%s == NULL)\n" n;
5711   pr "        croak (\"%s: %%s\", guestfs_last_error (g));\n" name;
5712   pr "      EXTEND (SP, %s->len);\n" n;
5713   pr "      for (i = 0; i < %s->len; ++i) {\n" n;
5714   pr "        hv = newHV ();\n";
5715   List.iter (
5716     function
5717     | name, `String ->
5718         pr "        (void) hv_store (hv, \"%s\", %d, newSVpv (%s->val[i].%s, 0), 0);\n"
5719           name (String.length name) n name
5720     | name, `UUID ->
5721         pr "        (void) hv_store (hv, \"%s\", %d, newSVpv (%s->val[i].%s, 32), 0);\n"
5722           name (String.length name) n name
5723     | name, `Bytes ->
5724         pr "        (void) hv_store (hv, \"%s\", %d, my_newSVull (%s->val[i].%s), 0);\n"
5725           name (String.length name) n name
5726     | name, `Int ->
5727         pr "        (void) hv_store (hv, \"%s\", %d, my_newSVll (%s->val[i].%s), 0);\n"
5728           name (String.length name) n name
5729     | name, `OptPercent ->
5730         pr "        (void) hv_store (hv, \"%s\", %d, newSVnv (%s->val[i].%s), 0);\n"
5731           name (String.length name) n name
5732   ) cols;
5733   pr "        PUSHs (sv_2mortal ((SV *) hv));\n";
5734   pr "      }\n";
5735   pr "      guestfs_free_lvm_%s_list (%s);\n" typ n
5736
5737 and generate_perl_stat_code typ cols name style n do_cleanups =
5738   pr "PREINIT:\n";
5739   pr "      struct guestfs_%s *%s;\n" typ n;
5740   pr " PPCODE:\n";
5741   pr "      %s = guestfs_%s " n name;
5742   generate_call_args ~handle:"g" (snd style);
5743   pr ";\n";
5744   do_cleanups ();
5745   pr "      if (%s == NULL)\n" n;
5746   pr "        croak (\"%s: %%s\", guestfs_last_error (g));\n" name;
5747   pr "      EXTEND (SP, %d);\n" (List.length cols);
5748   List.iter (
5749     function
5750     | name, `Int ->
5751         pr "      PUSHs (sv_2mortal (my_newSVll (%s->%s)));\n" n name
5752   ) cols;
5753   pr "      free (%s);\n" n
5754
5755 (* Generate Sys/Guestfs.pm. *)
5756 and generate_perl_pm () =
5757   generate_header HashStyle LGPLv2;
5758
5759   pr "\
5760 =pod
5761
5762 =head1 NAME
5763
5764 Sys::Guestfs - Perl bindings for libguestfs
5765
5766 =head1 SYNOPSIS
5767
5768  use Sys::Guestfs;
5769  
5770  my $h = Sys::Guestfs->new ();
5771  $h->add_drive ('guest.img');
5772  $h->launch ();
5773  $h->wait_ready ();
5774  $h->mount ('/dev/sda1', '/');
5775  $h->touch ('/hello');
5776  $h->sync ();
5777
5778 =head1 DESCRIPTION
5779
5780 The C<Sys::Guestfs> module provides a Perl XS binding to the
5781 libguestfs API for examining and modifying virtual machine
5782 disk images.
5783
5784 Amongst the things this is good for: making batch configuration
5785 changes to guests, getting disk used/free statistics (see also:
5786 virt-df), migrating between virtualization systems (see also:
5787 virt-p2v), performing partial backups, performing partial guest
5788 clones, cloning guests and changing registry/UUID/hostname info, and
5789 much else besides.
5790
5791 Libguestfs uses Linux kernel and qemu code, and can access any type of
5792 guest filesystem that Linux and qemu can, including but not limited
5793 to: ext2/3/4, btrfs, FAT and NTFS, LVM, many different disk partition
5794 schemes, qcow, qcow2, vmdk.
5795
5796 Libguestfs provides ways to enumerate guest storage (eg. partitions,
5797 LVs, what filesystem is in each LV, etc.).  It can also run commands
5798 in the context of the guest.  Also you can access filesystems over FTP.
5799
5800 =head1 ERRORS
5801
5802 All errors turn into calls to C<croak> (see L<Carp(3)>).
5803
5804 =head1 METHODS
5805
5806 =over 4
5807
5808 =cut
5809
5810 package Sys::Guestfs;
5811
5812 use strict;
5813 use warnings;
5814
5815 require XSLoader;
5816 XSLoader::load ('Sys::Guestfs');
5817
5818 =item $h = Sys::Guestfs->new ();
5819
5820 Create a new guestfs handle.
5821
5822 =cut
5823
5824 sub new {
5825   my $proto = shift;
5826   my $class = ref ($proto) || $proto;
5827
5828   my $self = Sys::Guestfs::_create ();
5829   bless $self, $class;
5830   return $self;
5831 }
5832
5833 ";
5834
5835   (* Actions.  We only need to print documentation for these as
5836    * they are pulled in from the XS code automatically.
5837    *)
5838   List.iter (
5839     fun (name, style, _, flags, _, _, longdesc) ->
5840       if not (List.mem NotInDocs flags) then (
5841         let longdesc = replace_str longdesc "C<guestfs_" "C<$h-E<gt>" in
5842         pr "=item ";
5843         generate_perl_prototype name style;
5844         pr "\n\n";
5845         pr "%s\n\n" longdesc;
5846         if List.mem ProtocolLimitWarning flags then
5847           pr "%s\n\n" protocol_limit_warning;
5848         if List.mem DangerWillRobinson flags then
5849           pr "%s\n\n" danger_will_robinson
5850       )
5851   ) all_functions_sorted;
5852
5853   (* End of file. *)
5854   pr "\
5855 =cut
5856
5857 1;
5858
5859 =back
5860
5861 =head1 COPYRIGHT
5862
5863 Copyright (C) 2009 Red Hat Inc.
5864
5865 =head1 LICENSE
5866
5867 Please see the file COPYING.LIB for the full license.
5868
5869 =head1 SEE ALSO
5870
5871 L<guestfs(3)>, L<guestfish(1)>.
5872
5873 =cut
5874 "
5875
5876 and generate_perl_prototype name style =
5877   (match fst style with
5878    | RErr -> ()
5879    | RBool n
5880    | RInt n
5881    | RInt64 n
5882    | RConstString n
5883    | RString n -> pr "$%s = " n
5884    | RIntBool (n, m) -> pr "($%s, $%s) = " n m
5885    | RStringList n
5886    | RPVList n
5887    | RVGList n
5888    | RLVList n -> pr "@%s = " n
5889    | RStat n
5890    | RStatVFS n
5891    | RHashtable n -> pr "%%%s = " n
5892   );
5893   pr "$h->%s (" name;
5894   let comma = ref false in
5895   List.iter (
5896     fun arg ->
5897       if !comma then pr ", ";
5898       comma := true;
5899       match arg with
5900       | String n | OptString n | Bool n | Int n | FileIn n | FileOut n ->
5901           pr "$%s" n
5902       | StringList n ->
5903           pr "\\@%s" n
5904   ) (snd style);
5905   pr ");"
5906
5907 (* Generate Python C module. *)
5908 and generate_python_c () =
5909   generate_header CStyle LGPLv2;
5910
5911   pr "\
5912 #include <stdio.h>
5913 #include <stdlib.h>
5914 #include <assert.h>
5915
5916 #include <Python.h>
5917
5918 #include \"guestfs.h\"
5919
5920 typedef struct {
5921   PyObject_HEAD
5922   guestfs_h *g;
5923 } Pyguestfs_Object;
5924
5925 static guestfs_h *
5926 get_handle (PyObject *obj)
5927 {
5928   assert (obj);
5929   assert (obj != Py_None);
5930   return ((Pyguestfs_Object *) obj)->g;
5931 }
5932
5933 static PyObject *
5934 put_handle (guestfs_h *g)
5935 {
5936   assert (g);
5937   return
5938     PyCObject_FromVoidPtrAndDesc ((void *) g, (char *) \"guestfs_h\", NULL);
5939 }
5940
5941 /* This list should be freed (but not the strings) after use. */
5942 static const char **
5943 get_string_list (PyObject *obj)
5944 {
5945   int i, len;
5946   const char **r;
5947
5948   assert (obj);
5949
5950   if (!PyList_Check (obj)) {
5951     PyErr_SetString (PyExc_RuntimeError, \"expecting a list parameter\");
5952     return NULL;
5953   }
5954
5955   len = PyList_Size (obj);
5956   r = malloc (sizeof (char *) * (len+1));
5957   if (r == NULL) {
5958     PyErr_SetString (PyExc_RuntimeError, \"get_string_list: out of memory\");
5959     return NULL;
5960   }
5961
5962   for (i = 0; i < len; ++i)
5963     r[i] = PyString_AsString (PyList_GetItem (obj, i));
5964   r[len] = NULL;
5965
5966   return r;
5967 }
5968
5969 static PyObject *
5970 put_string_list (char * const * const argv)
5971 {
5972   PyObject *list;
5973   int argc, i;
5974
5975   for (argc = 0; argv[argc] != NULL; ++argc)
5976     ;
5977
5978   list = PyList_New (argc);
5979   for (i = 0; i < argc; ++i)
5980     PyList_SetItem (list, i, PyString_FromString (argv[i]));
5981
5982   return list;
5983 }
5984
5985 static PyObject *
5986 put_table (char * const * const argv)
5987 {
5988   PyObject *list, *item;
5989   int argc, i;
5990
5991   for (argc = 0; argv[argc] != NULL; ++argc)
5992     ;
5993
5994   list = PyList_New (argc >> 1);
5995   for (i = 0; i < argc; i += 2) {
5996     item = PyTuple_New (2);
5997     PyTuple_SetItem (item, 0, PyString_FromString (argv[i]));
5998     PyTuple_SetItem (item, 1, PyString_FromString (argv[i+1]));
5999     PyList_SetItem (list, i >> 1, item);
6000   }
6001
6002   return list;
6003 }
6004
6005 static void
6006 free_strings (char **argv)
6007 {
6008   int argc;
6009
6010   for (argc = 0; argv[argc] != NULL; ++argc)
6011     free (argv[argc]);
6012   free (argv);
6013 }
6014
6015 static PyObject *
6016 py_guestfs_create (PyObject *self, PyObject *args)
6017 {
6018   guestfs_h *g;
6019
6020   g = guestfs_create ();
6021   if (g == NULL) {
6022     PyErr_SetString (PyExc_RuntimeError,
6023                      \"guestfs.create: failed to allocate handle\");
6024     return NULL;
6025   }
6026   guestfs_set_error_handler (g, NULL, NULL);
6027   return put_handle (g);
6028 }
6029
6030 static PyObject *
6031 py_guestfs_close (PyObject *self, PyObject *args)
6032 {
6033   PyObject *py_g;
6034   guestfs_h *g;
6035
6036   if (!PyArg_ParseTuple (args, (char *) \"O:guestfs_close\", &py_g))
6037     return NULL;
6038   g = get_handle (py_g);
6039
6040   guestfs_close (g);
6041
6042   Py_INCREF (Py_None);
6043   return Py_None;
6044 }
6045
6046 ";
6047
6048   (* LVM structures, turned into Python dictionaries. *)
6049   List.iter (
6050     fun (typ, cols) ->
6051       pr "static PyObject *\n";
6052       pr "put_lvm_%s (struct guestfs_lvm_%s *%s)\n" typ typ typ;
6053       pr "{\n";
6054       pr "  PyObject *dict;\n";
6055       pr "\n";
6056       pr "  dict = PyDict_New ();\n";
6057       List.iter (
6058         function
6059         | name, `String ->
6060             pr "  PyDict_SetItemString (dict, \"%s\",\n" name;
6061             pr "                        PyString_FromString (%s->%s));\n"
6062               typ name
6063         | name, `UUID ->
6064             pr "  PyDict_SetItemString (dict, \"%s\",\n" name;
6065             pr "                        PyString_FromStringAndSize (%s->%s, 32));\n"
6066               typ name
6067         | name, `Bytes ->
6068             pr "  PyDict_SetItemString (dict, \"%s\",\n" name;
6069             pr "                        PyLong_FromUnsignedLongLong (%s->%s));\n"
6070               typ name
6071         | name, `Int ->
6072             pr "  PyDict_SetItemString (dict, \"%s\",\n" name;
6073             pr "                        PyLong_FromLongLong (%s->%s));\n"
6074               typ name
6075         | name, `OptPercent ->
6076             pr "  if (%s->%s >= 0)\n" typ name;
6077             pr "    PyDict_SetItemString (dict, \"%s\",\n" name;
6078             pr "                          PyFloat_FromDouble ((double) %s->%s));\n"
6079               typ name;
6080             pr "  else {\n";
6081             pr "    Py_INCREF (Py_None);\n";
6082             pr "    PyDict_SetItemString (dict, \"%s\", Py_None);" name;
6083             pr "  }\n"
6084       ) cols;
6085       pr "  return dict;\n";
6086       pr "};\n";
6087       pr "\n";
6088
6089       pr "static PyObject *\n";
6090       pr "put_lvm_%s_list (struct guestfs_lvm_%s_list *%ss)\n" typ typ typ;
6091       pr "{\n";
6092       pr "  PyObject *list;\n";
6093       pr "  int i;\n";
6094       pr "\n";
6095       pr "  list = PyList_New (%ss->len);\n" typ;
6096       pr "  for (i = 0; i < %ss->len; ++i)\n" typ;
6097       pr "    PyList_SetItem (list, i, put_lvm_%s (&%ss->val[i]));\n" typ typ;
6098       pr "  return list;\n";
6099       pr "};\n";
6100       pr "\n"
6101   ) ["pv", pv_cols; "vg", vg_cols; "lv", lv_cols];
6102
6103   (* Stat structures, turned into Python dictionaries. *)
6104   List.iter (
6105     fun (typ, cols) ->
6106       pr "static PyObject *\n";
6107       pr "put_%s (struct guestfs_%s *%s)\n" typ typ typ;
6108       pr "{\n";
6109       pr "  PyObject *dict;\n";
6110       pr "\n";
6111       pr "  dict = PyDict_New ();\n";
6112       List.iter (
6113         function
6114         | name, `Int ->
6115             pr "  PyDict_SetItemString (dict, \"%s\",\n" name;
6116             pr "                        PyLong_FromLongLong (%s->%s));\n"
6117               typ name
6118       ) cols;
6119       pr "  return dict;\n";
6120       pr "};\n";
6121       pr "\n";
6122   ) ["stat", stat_cols; "statvfs", statvfs_cols];
6123
6124   (* Python wrapper functions. *)
6125   List.iter (
6126     fun (name, style, _, _, _, _, _) ->
6127       pr "static PyObject *\n";
6128       pr "py_guestfs_%s (PyObject *self, PyObject *args)\n" name;
6129       pr "{\n";
6130
6131       pr "  PyObject *py_g;\n";
6132       pr "  guestfs_h *g;\n";
6133       pr "  PyObject *py_r;\n";
6134
6135       let error_code =
6136         match fst style with
6137         | RErr | RInt _ | RBool _ -> pr "  int r;\n"; "-1"
6138         | RInt64 _ -> pr "  int64_t r;\n"; "-1"
6139         | RConstString _ -> pr "  const char *r;\n"; "NULL"
6140         | RString _ -> pr "  char *r;\n"; "NULL"
6141         | RStringList _ | RHashtable _ -> pr "  char **r;\n"; "NULL"
6142         | RIntBool _ -> pr "  struct guestfs_int_bool *r;\n"; "NULL"
6143         | RPVList n -> pr "  struct guestfs_lvm_pv_list *r;\n"; "NULL"
6144         | RVGList n -> pr "  struct guestfs_lvm_vg_list *r;\n"; "NULL"
6145         | RLVList n -> pr "  struct guestfs_lvm_lv_list *r;\n"; "NULL"
6146         | RStat n -> pr "  struct guestfs_stat *r;\n"; "NULL"
6147         | RStatVFS n -> pr "  struct guestfs_statvfs *r;\n"; "NULL" in
6148
6149       List.iter (
6150         function
6151         | String n | FileIn n | FileOut n -> pr "  const char *%s;\n" n
6152         | OptString n -> pr "  const char *%s;\n" n
6153         | StringList n ->
6154             pr "  PyObject *py_%s;\n" n;
6155             pr "  const char **%s;\n" n
6156         | Bool n -> pr "  int %s;\n" n
6157         | Int n -> pr "  int %s;\n" n
6158       ) (snd style);
6159
6160       pr "\n";
6161
6162       (* Convert the parameters. *)
6163       pr "  if (!PyArg_ParseTuple (args, (char *) \"O";
6164       List.iter (
6165         function
6166         | String _ | FileIn _ | FileOut _ -> pr "s"
6167         | OptString _ -> pr "z"
6168         | StringList _ -> pr "O"
6169         | Bool _ -> pr "i" (* XXX Python has booleans? *)
6170         | Int _ -> pr "i"
6171       ) (snd style);
6172       pr ":guestfs_%s\",\n" name;
6173       pr "                         &py_g";
6174       List.iter (
6175         function
6176         | String n | FileIn n | FileOut n -> pr ", &%s" n
6177         | OptString n -> pr ", &%s" n
6178         | StringList n -> pr ", &py_%s" n
6179         | Bool n -> pr ", &%s" n
6180         | Int n -> pr ", &%s" n
6181       ) (snd style);
6182
6183       pr "))\n";
6184       pr "    return NULL;\n";
6185
6186       pr "  g = get_handle (py_g);\n";
6187       List.iter (
6188         function
6189         | String _ | FileIn _ | FileOut _ | OptString _ | Bool _ | Int _ -> ()
6190         | StringList n ->
6191             pr "  %s = get_string_list (py_%s);\n" n n;
6192             pr "  if (!%s) return NULL;\n" n
6193       ) (snd style);
6194
6195       pr "\n";
6196
6197       pr "  r = guestfs_%s " name;
6198       generate_call_args ~handle:"g" (snd style);
6199       pr ";\n";
6200
6201       List.iter (
6202         function
6203         | String _ | FileIn _ | FileOut _ | OptString _ | Bool _ | Int _ -> ()
6204         | StringList n ->
6205             pr "  free (%s);\n" n
6206       ) (snd style);
6207
6208       pr "  if (r == %s) {\n" error_code;
6209       pr "    PyErr_SetString (PyExc_RuntimeError, guestfs_last_error (g));\n";
6210       pr "    return NULL;\n";
6211       pr "  }\n";
6212       pr "\n";
6213
6214       (match fst style with
6215        | RErr ->
6216            pr "  Py_INCREF (Py_None);\n";
6217            pr "  py_r = Py_None;\n"
6218        | RInt _
6219        | RBool _ -> pr "  py_r = PyInt_FromLong ((long) r);\n"
6220        | RInt64 _ -> pr "  py_r = PyLong_FromLongLong (r);\n"
6221        | RConstString _ -> pr "  py_r = PyString_FromString (r);\n"
6222        | RString _ ->
6223            pr "  py_r = PyString_FromString (r);\n";
6224            pr "  free (r);\n"
6225        | RStringList _ ->
6226            pr "  py_r = put_string_list (r);\n";
6227            pr "  free_strings (r);\n"
6228        | RIntBool _ ->
6229            pr "  py_r = PyTuple_New (2);\n";
6230            pr "  PyTuple_SetItem (py_r, 0, PyInt_FromLong ((long) r->i));\n";
6231            pr "  PyTuple_SetItem (py_r, 1, PyInt_FromLong ((long) r->b));\n";
6232            pr "  guestfs_free_int_bool (r);\n"
6233        | RPVList n ->
6234            pr "  py_r = put_lvm_pv_list (r);\n";
6235            pr "  guestfs_free_lvm_pv_list (r);\n"
6236        | RVGList n ->
6237            pr "  py_r = put_lvm_vg_list (r);\n";
6238            pr "  guestfs_free_lvm_vg_list (r);\n"
6239        | RLVList n ->
6240            pr "  py_r = put_lvm_lv_list (r);\n";
6241            pr "  guestfs_free_lvm_lv_list (r);\n"
6242        | RStat n ->
6243            pr "  py_r = put_stat (r);\n";
6244            pr "  free (r);\n"
6245        | RStatVFS n ->
6246            pr "  py_r = put_statvfs (r);\n";
6247            pr "  free (r);\n"
6248        | RHashtable n ->
6249            pr "  py_r = put_table (r);\n";
6250            pr "  free_strings (r);\n"
6251       );
6252
6253       pr "  return py_r;\n";
6254       pr "}\n";
6255       pr "\n"
6256   ) all_functions;
6257
6258   (* Table of functions. *)
6259   pr "static PyMethodDef methods[] = {\n";
6260   pr "  { (char *) \"create\", py_guestfs_create, METH_VARARGS, NULL },\n";
6261   pr "  { (char *) \"close\", py_guestfs_close, METH_VARARGS, NULL },\n";
6262   List.iter (
6263     fun (name, _, _, _, _, _, _) ->
6264       pr "  { (char *) \"%s\", py_guestfs_%s, METH_VARARGS, NULL },\n"
6265         name name
6266   ) all_functions;
6267   pr "  { NULL, NULL, 0, NULL }\n";
6268   pr "};\n";
6269   pr "\n";
6270
6271   (* Init function. *)
6272   pr "\
6273 void
6274 initlibguestfsmod (void)
6275 {
6276   static int initialized = 0;
6277
6278   if (initialized) return;
6279   Py_InitModule ((char *) \"libguestfsmod\", methods);
6280   initialized = 1;
6281 }
6282 "
6283
6284 (* Generate Python module. *)
6285 and generate_python_py () =
6286   generate_header HashStyle LGPLv2;
6287
6288   pr "\
6289 u\"\"\"Python bindings for libguestfs
6290
6291 import guestfs
6292 g = guestfs.GuestFS ()
6293 g.add_drive (\"guest.img\")
6294 g.launch ()
6295 g.wait_ready ()
6296 parts = g.list_partitions ()
6297
6298 The guestfs module provides a Python binding to the libguestfs API
6299 for examining and modifying virtual machine disk images.
6300
6301 Amongst the things this is good for: making batch configuration
6302 changes to guests, getting disk used/free statistics (see also:
6303 virt-df), migrating between virtualization systems (see also:
6304 virt-p2v), performing partial backups, performing partial guest
6305 clones, cloning guests and changing registry/UUID/hostname info, and
6306 much else besides.
6307
6308 Libguestfs uses Linux kernel and qemu code, and can access any type of
6309 guest filesystem that Linux and qemu can, including but not limited
6310 to: ext2/3/4, btrfs, FAT and NTFS, LVM, many different disk partition
6311 schemes, qcow, qcow2, vmdk.
6312
6313 Libguestfs provides ways to enumerate guest storage (eg. partitions,
6314 LVs, what filesystem is in each LV, etc.).  It can also run commands
6315 in the context of the guest.  Also you can access filesystems over FTP.
6316
6317 Errors which happen while using the API are turned into Python
6318 RuntimeError exceptions.
6319
6320 To create a guestfs handle you usually have to perform the following
6321 sequence of calls:
6322
6323 # Create the handle, call add_drive at least once, and possibly
6324 # several times if the guest has multiple block devices:
6325 g = guestfs.GuestFS ()
6326 g.add_drive (\"guest.img\")
6327
6328 # Launch the qemu subprocess and wait for it to become ready:
6329 g.launch ()
6330 g.wait_ready ()
6331
6332 # Now you can issue commands, for example:
6333 logvols = g.lvs ()
6334
6335 \"\"\"
6336
6337 import libguestfsmod
6338
6339 class GuestFS:
6340     \"\"\"Instances of this class are libguestfs API handles.\"\"\"
6341
6342     def __init__ (self):
6343         \"\"\"Create a new libguestfs handle.\"\"\"
6344         self._o = libguestfsmod.create ()
6345
6346     def __del__ (self):
6347         libguestfsmod.close (self._o)
6348
6349 ";
6350
6351   List.iter (
6352     fun (name, style, _, flags, _, _, longdesc) ->
6353       pr "    def %s " name;
6354       generate_call_args ~handle:"self" (snd style);
6355       pr ":\n";
6356
6357       if not (List.mem NotInDocs flags) then (
6358         let doc = replace_str longdesc "C<guestfs_" "C<g." in
6359         let doc =
6360           match fst style with
6361           | RErr | RInt _ | RInt64 _ | RBool _ | RConstString _
6362           | RString _ -> doc
6363           | RStringList _ ->
6364               doc ^ "\n\nThis function returns a list of strings."
6365           | RIntBool _ ->
6366               doc ^ "\n\nThis function returns a tuple (int, bool).\n"
6367           | RPVList _ ->
6368               doc ^ "\n\nThis function returns a list of PVs.  Each PV is represented as a dictionary."
6369           | RVGList _ ->
6370               doc ^ "\n\nThis function returns a list of VGs.  Each VG is represented as a dictionary."
6371           | RLVList _ ->
6372               doc ^ "\n\nThis function returns a list of LVs.  Each LV is represented as a dictionary."
6373           | RStat _ ->
6374               doc ^ "\n\nThis function returns a dictionary, with keys matching the various fields in the stat structure."
6375           | RStatVFS _ ->
6376               doc ^ "\n\nThis function returns a dictionary, with keys matching the various fields in the statvfs structure."
6377           | RHashtable _ ->
6378               doc ^ "\n\nThis function returns a dictionary." in
6379         let doc =
6380           if List.mem ProtocolLimitWarning flags then
6381             doc ^ "\n\n" ^ protocol_limit_warning
6382           else doc in
6383         let doc =
6384           if List.mem DangerWillRobinson flags then
6385             doc ^ "\n\n" ^ danger_will_robinson
6386           else doc in
6387         let doc = pod2text ~width:60 name doc in
6388         let doc = List.map (fun line -> replace_str line "\\" "\\\\") doc in
6389         let doc = String.concat "\n        " doc in
6390         pr "        u\"\"\"%s\"\"\"\n" doc;
6391       );
6392       pr "        return libguestfsmod.%s " name;
6393       generate_call_args ~handle:"self._o" (snd style);
6394       pr "\n";
6395       pr "\n";
6396   ) all_functions
6397
6398 (* Useful if you need the longdesc POD text as plain text.  Returns a
6399  * list of lines.
6400  *
6401  * This is the slowest thing about autogeneration.
6402  *)
6403 and pod2text ~width name longdesc =
6404   let filename, chan = Filename.open_temp_file "gen" ".tmp" in
6405   fprintf chan "=head1 %s\n\n%s\n" name longdesc;
6406   close_out chan;
6407   let cmd = sprintf "pod2text -w %d %s" width (Filename.quote filename) in
6408   let chan = Unix.open_process_in cmd in
6409   let lines = ref [] in
6410   let rec loop i =
6411     let line = input_line chan in
6412     if i = 1 then               (* discard the first line of output *)
6413       loop (i+1)
6414     else (
6415       let line = triml line in
6416       lines := line :: !lines;
6417       loop (i+1)
6418     ) in
6419   let lines = try loop 1 with End_of_file -> List.rev !lines in
6420   Unix.unlink filename;
6421   match Unix.close_process_in chan with
6422   | Unix.WEXITED 0 -> lines
6423   | Unix.WEXITED i ->
6424       failwithf "pod2text: process exited with non-zero status (%d)" i
6425   | Unix.WSIGNALED i | Unix.WSTOPPED i ->
6426       failwithf "pod2text: process signalled or stopped by signal %d" i
6427
6428 (* Generate ruby bindings. *)
6429 and generate_ruby_c () =
6430   generate_header CStyle LGPLv2;
6431
6432   pr "\
6433 #include <stdio.h>
6434 #include <stdlib.h>
6435
6436 #include <ruby.h>
6437
6438 #include \"guestfs.h\"
6439
6440 #include \"extconf.h\"
6441
6442 /* For Ruby < 1.9 */
6443 #ifndef RARRAY_LEN
6444 #define RARRAY_LEN(r) (RARRAY((r))->len)
6445 #endif
6446
6447 static VALUE m_guestfs;                 /* guestfs module */
6448 static VALUE c_guestfs;                 /* guestfs_h handle */
6449 static VALUE e_Error;                   /* used for all errors */
6450
6451 static void ruby_guestfs_free (void *p)
6452 {
6453   if (!p) return;
6454   guestfs_close ((guestfs_h *) p);
6455 }
6456
6457 static VALUE ruby_guestfs_create (VALUE m)
6458 {
6459   guestfs_h *g;
6460
6461   g = guestfs_create ();
6462   if (!g)
6463     rb_raise (e_Error, \"failed to create guestfs handle\");
6464
6465   /* Don't print error messages to stderr by default. */
6466   guestfs_set_error_handler (g, NULL, NULL);
6467
6468   /* Wrap it, and make sure the close function is called when the
6469    * handle goes away.
6470    */
6471   return Data_Wrap_Struct (c_guestfs, NULL, ruby_guestfs_free, g);
6472 }
6473
6474 static VALUE ruby_guestfs_close (VALUE gv)
6475 {
6476   guestfs_h *g;
6477   Data_Get_Struct (gv, guestfs_h, g);
6478
6479   ruby_guestfs_free (g);
6480   DATA_PTR (gv) = NULL;
6481
6482   return Qnil;
6483 }
6484
6485 ";
6486
6487   List.iter (
6488     fun (name, style, _, _, _, _, _) ->
6489       pr "static VALUE ruby_guestfs_%s (VALUE gv" name;
6490       List.iter (fun arg -> pr ", VALUE %sv" (name_of_argt arg)) (snd style);
6491       pr ")\n";
6492       pr "{\n";
6493       pr "  guestfs_h *g;\n";
6494       pr "  Data_Get_Struct (gv, guestfs_h, g);\n";
6495       pr "  if (!g)\n";
6496       pr "    rb_raise (rb_eArgError, \"%%s: used handle after closing it\", \"%s\");\n"
6497         name;
6498       pr "\n";
6499
6500       List.iter (
6501         function
6502         | String n | FileIn n | FileOut n ->
6503             pr "  const char *%s = StringValueCStr (%sv);\n" n n;
6504             pr "  if (!%s)\n" n;
6505             pr "    rb_raise (rb_eTypeError, \"expected string for parameter %%s of %%s\",\n";
6506             pr "              \"%s\", \"%s\");\n" n name
6507         | OptString n ->
6508             pr "  const char *%s = !NIL_P (%sv) ? StringValueCStr (%sv) : NULL;\n" n n n
6509         | StringList n ->
6510             pr "  char **%s;\n" n;
6511             pr "  {\n";
6512             pr "    int i, len;\n";
6513             pr "    len = RARRAY_LEN (%sv);\n" n;
6514             pr "    %s = guestfs_safe_malloc (g, sizeof (char *) * (len+1));\n"
6515               n;
6516             pr "    for (i = 0; i < len; ++i) {\n";
6517             pr "      VALUE v = rb_ary_entry (%sv, i);\n" n;
6518             pr "      %s[i] = StringValueCStr (v);\n" n;
6519             pr "    }\n";
6520             pr "    %s[len] = NULL;\n" n;
6521             pr "  }\n";
6522         | Bool n ->
6523             pr "  int %s = RTEST (%sv);\n" n n
6524         | Int n ->
6525             pr "  int %s = NUM2INT (%sv);\n" n n
6526       ) (snd style);
6527       pr "\n";
6528
6529       let error_code =
6530         match fst style with
6531         | RErr | RInt _ | RBool _ -> pr "  int r;\n"; "-1"
6532         | RInt64 _ -> pr "  int64_t r;\n"; "-1"
6533         | RConstString _ -> pr "  const char *r;\n"; "NULL"
6534         | RString _ -> pr "  char *r;\n"; "NULL"
6535         | RStringList _ | RHashtable _ -> pr "  char **r;\n"; "NULL"
6536         | RIntBool _ -> pr "  struct guestfs_int_bool *r;\n"; "NULL"
6537         | RPVList n -> pr "  struct guestfs_lvm_pv_list *r;\n"; "NULL"
6538         | RVGList n -> pr "  struct guestfs_lvm_vg_list *r;\n"; "NULL"
6539         | RLVList n -> pr "  struct guestfs_lvm_lv_list *r;\n"; "NULL"
6540         | RStat n -> pr "  struct guestfs_stat *r;\n"; "NULL"
6541         | RStatVFS n -> pr "  struct guestfs_statvfs *r;\n"; "NULL" in
6542       pr "\n";
6543
6544       pr "  r = guestfs_%s " name;
6545       generate_call_args ~handle:"g" (snd style);
6546       pr ";\n";
6547
6548       List.iter (
6549         function
6550         | String _ | FileIn _ | FileOut _ | OptString _ | Bool _ | Int _ -> ()
6551         | StringList n ->
6552             pr "  free (%s);\n" n
6553       ) (snd style);
6554
6555       pr "  if (r == %s)\n" error_code;
6556       pr "    rb_raise (e_Error, \"%%s\", guestfs_last_error (g));\n";
6557       pr "\n";
6558
6559       (match fst style with
6560        | RErr ->
6561            pr "  return Qnil;\n"
6562        | RInt _ | RBool _ ->
6563            pr "  return INT2NUM (r);\n"
6564        | RInt64 _ ->
6565            pr "  return ULL2NUM (r);\n"
6566        | RConstString _ ->
6567            pr "  return rb_str_new2 (r);\n";
6568        | RString _ ->
6569            pr "  VALUE rv = rb_str_new2 (r);\n";
6570            pr "  free (r);\n";
6571            pr "  return rv;\n";
6572        | RStringList _ ->
6573            pr "  int i, len = 0;\n";
6574            pr "  for (i = 0; r[i] != NULL; ++i) len++;\n";
6575            pr "  VALUE rv = rb_ary_new2 (len);\n";
6576            pr "  for (i = 0; r[i] != NULL; ++i) {\n";
6577            pr "    rb_ary_push (rv, rb_str_new2 (r[i]));\n";
6578            pr "    free (r[i]);\n";
6579            pr "  }\n";
6580            pr "  free (r);\n";
6581            pr "  return rv;\n"
6582        | RIntBool _ ->
6583            pr "  VALUE rv = rb_ary_new2 (2);\n";
6584            pr "  rb_ary_push (rv, INT2NUM (r->i));\n";
6585            pr "  rb_ary_push (rv, INT2NUM (r->b));\n";
6586            pr "  guestfs_free_int_bool (r);\n";
6587            pr "  return rv;\n"
6588        | RPVList n ->
6589            generate_ruby_lvm_code "pv" pv_cols
6590        | RVGList n ->
6591            generate_ruby_lvm_code "vg" vg_cols
6592        | RLVList n ->
6593            generate_ruby_lvm_code "lv" lv_cols
6594        | RStat n ->
6595            pr "  VALUE rv = rb_hash_new ();\n";
6596            List.iter (
6597              function
6598              | name, `Int ->
6599                  pr "  rb_hash_aset (rv, rb_str_new2 (\"%s\"), ULL2NUM (r->%s));\n" name name
6600            ) stat_cols;
6601            pr "  free (r);\n";
6602            pr "  return rv;\n"
6603        | RStatVFS n ->
6604            pr "  VALUE rv = rb_hash_new ();\n";
6605            List.iter (
6606              function
6607              | name, `Int ->
6608                  pr "  rb_hash_aset (rv, rb_str_new2 (\"%s\"), ULL2NUM (r->%s));\n" name name
6609            ) statvfs_cols;
6610            pr "  free (r);\n";
6611            pr "  return rv;\n"
6612        | RHashtable _ ->
6613            pr "  VALUE rv = rb_hash_new ();\n";
6614            pr "  int i;\n";
6615            pr "  for (i = 0; r[i] != NULL; i+=2) {\n";
6616            pr "    rb_hash_aset (rv, rb_str_new2 (r[i]), rb_str_new2 (r[i+1]));\n";
6617            pr "    free (r[i]);\n";
6618            pr "    free (r[i+1]);\n";
6619            pr "  }\n";
6620            pr "  free (r);\n";
6621            pr "  return rv;\n"
6622       );
6623
6624       pr "}\n";
6625       pr "\n"
6626   ) all_functions;
6627
6628   pr "\
6629 /* Initialize the module. */
6630 void Init__guestfs ()
6631 {
6632   m_guestfs = rb_define_module (\"Guestfs\");
6633   c_guestfs = rb_define_class_under (m_guestfs, \"Guestfs\", rb_cObject);
6634   e_Error = rb_define_class_under (m_guestfs, \"Error\", rb_eStandardError);
6635
6636   rb_define_module_function (m_guestfs, \"create\", ruby_guestfs_create, 0);
6637   rb_define_method (c_guestfs, \"close\", ruby_guestfs_close, 0);
6638
6639 ";
6640   (* Define the rest of the methods. *)
6641   List.iter (
6642     fun (name, style, _, _, _, _, _) ->
6643       pr "  rb_define_method (c_guestfs, \"%s\",\n" name;
6644       pr "        ruby_guestfs_%s, %d);\n" name (List.length (snd style))
6645   ) all_functions;
6646
6647   pr "}\n"
6648
6649 (* Ruby code to return an LVM struct list. *)
6650 and generate_ruby_lvm_code typ cols =
6651   pr "  VALUE rv = rb_ary_new2 (r->len);\n";
6652   pr "  int i;\n";
6653   pr "  for (i = 0; i < r->len; ++i) {\n";
6654   pr "    VALUE hv = rb_hash_new ();\n";
6655   List.iter (
6656     function
6657     | name, `String ->
6658         pr "    rb_hash_aset (rv, rb_str_new2 (\"%s\"), rb_str_new2 (r->val[i].%s));\n" name name
6659     | name, `UUID ->
6660         pr "    rb_hash_aset (rv, rb_str_new2 (\"%s\"), rb_str_new (r->val[i].%s, 32));\n" name name
6661     | name, `Bytes
6662     | name, `Int ->
6663         pr "    rb_hash_aset (rv, rb_str_new2 (\"%s\"), ULL2NUM (r->val[i].%s));\n" name name
6664     | name, `OptPercent ->
6665         pr "    rb_hash_aset (rv, rb_str_new2 (\"%s\"), rb_dbl2big (r->val[i].%s));\n" name name
6666   ) cols;
6667   pr "    rb_ary_push (rv, hv);\n";
6668   pr "  }\n";
6669   pr "  guestfs_free_lvm_%s_list (r);\n" typ;
6670   pr "  return rv;\n"
6671
6672 (* Generate Java bindings GuestFS.java file. *)
6673 and generate_java_java () =
6674   generate_header CStyle LGPLv2;
6675
6676   pr "\
6677 package com.redhat.et.libguestfs;
6678
6679 import java.util.HashMap;
6680 import com.redhat.et.libguestfs.LibGuestFSException;
6681 import com.redhat.et.libguestfs.PV;
6682 import com.redhat.et.libguestfs.VG;
6683 import com.redhat.et.libguestfs.LV;
6684 import com.redhat.et.libguestfs.Stat;
6685 import com.redhat.et.libguestfs.StatVFS;
6686 import com.redhat.et.libguestfs.IntBool;
6687
6688 /**
6689  * The GuestFS object is a libguestfs handle.
6690  *
6691  * @author rjones
6692  */
6693 public class GuestFS {
6694   // Load the native code.
6695   static {
6696     System.loadLibrary (\"guestfs_jni\");
6697   }
6698
6699   /**
6700    * The native guestfs_h pointer.
6701    */
6702   long g;
6703
6704   /**
6705    * Create a libguestfs handle.
6706    *
6707    * @throws LibGuestFSException
6708    */
6709   public GuestFS () throws LibGuestFSException
6710   {
6711     g = _create ();
6712   }
6713   private native long _create () throws LibGuestFSException;
6714
6715   /**
6716    * Close a libguestfs handle.
6717    *
6718    * You can also leave handles to be collected by the garbage
6719    * collector, but this method ensures that the resources used
6720    * by the handle are freed up immediately.  If you call any
6721    * other methods after closing the handle, you will get an
6722    * exception.
6723    *
6724    * @throws LibGuestFSException
6725    */
6726   public void close () throws LibGuestFSException
6727   {
6728     if (g != 0)
6729       _close (g);
6730     g = 0;
6731   }
6732   private native void _close (long g) throws LibGuestFSException;
6733
6734   public void finalize () throws LibGuestFSException
6735   {
6736     close ();
6737   }
6738
6739 ";
6740
6741   List.iter (
6742     fun (name, style, _, flags, _, shortdesc, longdesc) ->
6743       if not (List.mem NotInDocs flags); then (
6744         let doc = replace_str longdesc "C<guestfs_" "C<g." in
6745         let doc =
6746           if List.mem ProtocolLimitWarning flags then
6747             doc ^ "\n\n" ^ protocol_limit_warning
6748           else doc in
6749         let doc =
6750           if List.mem DangerWillRobinson flags then
6751             doc ^ "\n\n" ^ danger_will_robinson
6752           else doc in
6753         let doc = pod2text ~width:60 name doc in
6754         let doc = List.map (            (* RHBZ#501883 *)
6755           function
6756           | "" -> "<p>"
6757           | nonempty -> nonempty
6758         ) doc in
6759         let doc = String.concat "\n   * " doc in
6760
6761         pr "  /**\n";
6762         pr "   * %s\n" shortdesc;
6763         pr "   * <p>\n";
6764         pr "   * %s\n" doc;
6765         pr "   * @throws LibGuestFSException\n";
6766         pr "   */\n";
6767         pr "  ";
6768       );
6769       generate_java_prototype ~public:true ~semicolon:false name style;
6770       pr "\n";
6771       pr "  {\n";
6772       pr "    if (g == 0)\n";
6773       pr "      throw new LibGuestFSException (\"%s: handle is closed\");\n"
6774         name;
6775       pr "    ";
6776       if fst style <> RErr then pr "return ";
6777       pr "_%s " name;
6778       generate_call_args ~handle:"g" (snd style);
6779       pr ";\n";
6780       pr "  }\n";
6781       pr "  ";
6782       generate_java_prototype ~privat:true ~native:true name style;
6783       pr "\n";
6784       pr "\n";
6785   ) all_functions;
6786
6787   pr "}\n"
6788
6789 and generate_java_prototype ?(public=false) ?(privat=false) ?(native=false)
6790     ?(semicolon=true) name style =
6791   if privat then pr "private ";
6792   if public then pr "public ";
6793   if native then pr "native ";
6794
6795   (* return type *)
6796   (match fst style with
6797    | RErr -> pr "void ";
6798    | RInt _ -> pr "int ";
6799    | RInt64 _ -> pr "long ";
6800    | RBool _ -> pr "boolean ";
6801    | RConstString _ | RString _ -> pr "String ";
6802    | RStringList _ -> pr "String[] ";
6803    | RIntBool _ -> pr "IntBool ";
6804    | RPVList _ -> pr "PV[] ";
6805    | RVGList _ -> pr "VG[] ";
6806    | RLVList _ -> pr "LV[] ";
6807    | RStat _ -> pr "Stat ";
6808    | RStatVFS _ -> pr "StatVFS ";
6809    | RHashtable _ -> pr "HashMap<String,String> ";
6810   );
6811
6812   if native then pr "_%s " name else pr "%s " name;
6813   pr "(";
6814   let needs_comma = ref false in
6815   if native then (
6816     pr "long g";
6817     needs_comma := true
6818   );
6819
6820   (* args *)
6821   List.iter (
6822     fun arg ->
6823       if !needs_comma then pr ", ";
6824       needs_comma := true;
6825
6826       match arg with
6827       | String n
6828       | OptString n
6829       | FileIn n
6830       | FileOut n ->
6831           pr "String %s" n
6832       | StringList n ->
6833           pr "String[] %s" n
6834       | Bool n ->
6835           pr "boolean %s" n
6836       | Int n ->
6837           pr "int %s" n
6838   ) (snd style);
6839
6840   pr ")\n";
6841   pr "    throws LibGuestFSException";
6842   if semicolon then pr ";"
6843
6844 and generate_java_struct typ cols =
6845   generate_header CStyle LGPLv2;
6846
6847   pr "\
6848 package com.redhat.et.libguestfs;
6849
6850 /**
6851  * Libguestfs %s structure.
6852  *
6853  * @author rjones
6854  * @see GuestFS
6855  */
6856 public class %s {
6857 " typ typ;
6858
6859   List.iter (
6860     function
6861     | name, `String
6862     | name, `UUID -> pr "  public String %s;\n" name
6863     | name, `Bytes
6864     | name, `Int -> pr "  public long %s;\n" name
6865     | name, `OptPercent ->
6866         pr "  /* The next field is [0..100] or -1 meaning 'not present': */\n";
6867         pr "  public float %s;\n" name
6868   ) cols;
6869
6870   pr "}\n"
6871
6872 and generate_java_c () =
6873   generate_header CStyle LGPLv2;
6874
6875   pr "\
6876 #include <stdio.h>
6877 #include <stdlib.h>
6878 #include <string.h>
6879
6880 #include \"com_redhat_et_libguestfs_GuestFS.h\"
6881 #include \"guestfs.h\"
6882
6883 /* Note that this function returns.  The exception is not thrown
6884  * until after the wrapper function returns.
6885  */
6886 static void
6887 throw_exception (JNIEnv *env, const char *msg)
6888 {
6889   jclass cl;
6890   cl = (*env)->FindClass (env,
6891                           \"com/redhat/et/libguestfs/LibGuestFSException\");
6892   (*env)->ThrowNew (env, cl, msg);
6893 }
6894
6895 JNIEXPORT jlong JNICALL
6896 Java_com_redhat_et_libguestfs_GuestFS__1create
6897   (JNIEnv *env, jobject obj)
6898 {
6899   guestfs_h *g;
6900
6901   g = guestfs_create ();
6902   if (g == NULL) {
6903     throw_exception (env, \"GuestFS.create: failed to allocate handle\");
6904     return 0;
6905   }
6906   guestfs_set_error_handler (g, NULL, NULL);
6907   return (jlong) (long) g;
6908 }
6909
6910 JNIEXPORT void JNICALL
6911 Java_com_redhat_et_libguestfs_GuestFS__1close
6912   (JNIEnv *env, jobject obj, jlong jg)
6913 {
6914   guestfs_h *g = (guestfs_h *) (long) jg;
6915   guestfs_close (g);
6916 }
6917
6918 ";
6919
6920   List.iter (
6921     fun (name, style, _, _, _, _, _) ->
6922       pr "JNIEXPORT ";
6923       (match fst style with
6924        | RErr -> pr "void ";
6925        | RInt _ -> pr "jint ";
6926        | RInt64 _ -> pr "jlong ";
6927        | RBool _ -> pr "jboolean ";
6928        | RConstString _ | RString _ -> pr "jstring ";
6929        | RIntBool _ | RStat _ | RStatVFS _ | RHashtable _ ->
6930            pr "jobject ";
6931        | RStringList _ | RPVList _ | RVGList _ | RLVList _ ->
6932            pr "jobjectArray ";
6933       );
6934       pr "JNICALL\n";
6935       pr "Java_com_redhat_et_libguestfs_GuestFS_";
6936       pr "%s" (replace_str ("_" ^ name) "_" "_1");
6937       pr "\n";
6938       pr "  (JNIEnv *env, jobject obj, jlong jg";
6939       List.iter (
6940         function
6941         | String n
6942         | OptString n
6943         | FileIn n
6944         | FileOut n ->
6945             pr ", jstring j%s" n
6946         | StringList n ->
6947             pr ", jobjectArray j%s" n
6948         | Bool n ->
6949             pr ", jboolean j%s" n
6950         | Int n ->
6951             pr ", jint j%s" n
6952       ) (snd style);
6953       pr ")\n";
6954       pr "{\n";
6955       pr "  guestfs_h *g = (guestfs_h *) (long) jg;\n";
6956       let error_code, no_ret =
6957         match fst style with
6958         | RErr -> pr "  int r;\n"; "-1", ""
6959         | RBool _
6960         | RInt _ -> pr "  int r;\n"; "-1", "0"
6961         | RInt64 _ -> pr "  int64_t r;\n"; "-1", "0"
6962         | RConstString _ -> pr "  const char *r;\n"; "NULL", "NULL"
6963         | RString _ ->
6964             pr "  jstring jr;\n";
6965             pr "  char *r;\n"; "NULL", "NULL"
6966         | RStringList _ ->
6967             pr "  jobjectArray jr;\n";
6968             pr "  int r_len;\n";
6969             pr "  jclass cl;\n";
6970             pr "  jstring jstr;\n";
6971             pr "  char **r;\n"; "NULL", "NULL"
6972         | RIntBool _ ->
6973             pr "  jobject jr;\n";
6974             pr "  jclass cl;\n";
6975             pr "  jfieldID fl;\n";
6976             pr "  struct guestfs_int_bool *r;\n"; "NULL", "NULL"
6977         | RStat _ ->
6978             pr "  jobject jr;\n";
6979             pr "  jclass cl;\n";
6980             pr "  jfieldID fl;\n";
6981             pr "  struct guestfs_stat *r;\n"; "NULL", "NULL"
6982         | RStatVFS _ ->
6983             pr "  jobject jr;\n";
6984             pr "  jclass cl;\n";
6985             pr "  jfieldID fl;\n";
6986             pr "  struct guestfs_statvfs *r;\n"; "NULL", "NULL"
6987         | RPVList _ ->
6988             pr "  jobjectArray jr;\n";
6989             pr "  jclass cl;\n";
6990             pr "  jfieldID fl;\n";
6991             pr "  jobject jfl;\n";
6992             pr "  struct guestfs_lvm_pv_list *r;\n"; "NULL", "NULL"
6993         | RVGList _ ->
6994             pr "  jobjectArray jr;\n";
6995             pr "  jclass cl;\n";
6996             pr "  jfieldID fl;\n";
6997             pr "  jobject jfl;\n";
6998             pr "  struct guestfs_lvm_vg_list *r;\n"; "NULL", "NULL"
6999         | RLVList _ ->
7000             pr "  jobjectArray jr;\n";
7001             pr "  jclass cl;\n";
7002             pr "  jfieldID fl;\n";
7003             pr "  jobject jfl;\n";
7004             pr "  struct guestfs_lvm_lv_list *r;\n"; "NULL", "NULL"
7005         | RHashtable _ -> pr "  char **r;\n"; "NULL", "NULL" in
7006       List.iter (
7007         function
7008         | String n
7009         | OptString n
7010         | FileIn n
7011         | FileOut n ->
7012             pr "  const char *%s;\n" n
7013         | StringList n ->
7014             pr "  int %s_len;\n" n;
7015             pr "  const char **%s;\n" n
7016         | Bool n
7017         | Int n ->
7018             pr "  int %s;\n" n
7019       ) (snd style);
7020
7021       let needs_i =
7022         (match fst style with
7023          | RStringList _ | RPVList _ | RVGList _ | RLVList _ -> true
7024          | RErr | RBool _ | RInt _ | RInt64 _ | RConstString _
7025          | RString _ | RIntBool _ | RStat _ | RStatVFS _
7026          | RHashtable _ -> false) ||
7027         List.exists (function StringList _ -> true | _ -> false) (snd style) in
7028       if needs_i then
7029         pr "  int i;\n";
7030
7031       pr "\n";
7032
7033       (* Get the parameters. *)
7034       List.iter (
7035         function
7036         | String n
7037         | FileIn n
7038         | FileOut n ->
7039             pr "  %s = (*env)->GetStringUTFChars (env, j%s, NULL);\n" n n
7040         | OptString n ->
7041             (* This is completely undocumented, but Java null becomes
7042              * a NULL parameter.
7043              *)
7044             pr "  %s = j%s ? (*env)->GetStringUTFChars (env, j%s, NULL) : NULL;\n" n n n
7045         | StringList n ->
7046             pr "  %s_len = (*env)->GetArrayLength (env, j%s);\n" n n;
7047             pr "  %s = guestfs_safe_malloc (g, sizeof (char *) * (%s_len+1));\n" n n;
7048             pr "  for (i = 0; i < %s_len; ++i) {\n" n;
7049             pr "    jobject o = (*env)->GetObjectArrayElement (env, j%s, i);\n"
7050               n;
7051             pr "    %s[i] = (*env)->GetStringUTFChars (env, o, NULL);\n" n;
7052             pr "  }\n";
7053             pr "  %s[%s_len] = NULL;\n" n n;
7054         | Bool n
7055         | Int n ->
7056             pr "  %s = j%s;\n" n n
7057       ) (snd style);
7058
7059       (* Make the call. *)
7060       pr "  r = guestfs_%s " name;
7061       generate_call_args ~handle:"g" (snd style);
7062       pr ";\n";
7063
7064       (* Release the parameters. *)
7065       List.iter (
7066         function
7067         | String n
7068         | FileIn n
7069         | FileOut n ->
7070             pr "  (*env)->ReleaseStringUTFChars (env, j%s, %s);\n" n n
7071         | OptString n ->
7072             pr "  if (j%s)\n" n;
7073             pr "    (*env)->ReleaseStringUTFChars (env, j%s, %s);\n" n n
7074         | StringList n ->
7075             pr "  for (i = 0; i < %s_len; ++i) {\n" n;
7076             pr "    jobject o = (*env)->GetObjectArrayElement (env, j%s, i);\n"
7077               n;
7078             pr "    (*env)->ReleaseStringUTFChars (env, o, %s[i]);\n" n;
7079             pr "  }\n";
7080             pr "  free (%s);\n" n
7081         | Bool n
7082         | Int n -> ()
7083       ) (snd style);
7084
7085       (* Check for errors. *)
7086       pr "  if (r == %s) {\n" error_code;
7087       pr "    throw_exception (env, guestfs_last_error (g));\n";
7088       pr "    return %s;\n" no_ret;
7089       pr "  }\n";
7090
7091       (* Return value. *)
7092       (match fst style with
7093        | RErr -> ()
7094        | RInt _ -> pr "  return (jint) r;\n"
7095        | RBool _ -> pr "  return (jboolean) r;\n"
7096        | RInt64 _ -> pr "  return (jlong) r;\n"
7097        | RConstString _ -> pr "  return (*env)->NewStringUTF (env, r);\n"
7098        | RString _ ->
7099            pr "  jr = (*env)->NewStringUTF (env, r);\n";
7100            pr "  free (r);\n";
7101            pr "  return jr;\n"
7102        | RStringList _ ->
7103            pr "  for (r_len = 0; r[r_len] != NULL; ++r_len) ;\n";
7104            pr "  cl = (*env)->FindClass (env, \"java/lang/String\");\n";
7105            pr "  jstr = (*env)->NewStringUTF (env, \"\");\n";
7106            pr "  jr = (*env)->NewObjectArray (env, r_len, cl, jstr);\n";
7107            pr "  for (i = 0; i < r_len; ++i) {\n";
7108            pr "    jstr = (*env)->NewStringUTF (env, r[i]);\n";
7109            pr "    (*env)->SetObjectArrayElement (env, jr, i, jstr);\n";
7110            pr "    free (r[i]);\n";
7111            pr "  }\n";
7112            pr "  free (r);\n";
7113            pr "  return jr;\n"
7114        | RIntBool _ ->
7115            pr "  cl = (*env)->FindClass (env, \"com/redhat/et/libguestfs/IntBool\");\n";
7116            pr "  jr = (*env)->AllocObject (env, cl);\n";
7117            pr "  fl = (*env)->GetFieldID (env, cl, \"i\", \"I\");\n";
7118            pr "  (*env)->SetIntField (env, jr, fl, r->i);\n";
7119            pr "  fl = (*env)->GetFieldID (env, cl, \"i\", \"Z\");\n";
7120            pr "  (*env)->SetBooleanField (env, jr, fl, r->b);\n";
7121            pr "  guestfs_free_int_bool (r);\n";
7122            pr "  return jr;\n"
7123        | RStat _ ->
7124            pr "  cl = (*env)->FindClass (env, \"com/redhat/et/libguestfs/Stat\");\n";
7125            pr "  jr = (*env)->AllocObject (env, cl);\n";
7126            List.iter (
7127              function
7128              | name, `Int ->
7129                  pr "  fl = (*env)->GetFieldID (env, cl, \"%s\", \"J\");\n"
7130                    name;
7131                  pr "  (*env)->SetLongField (env, jr, fl, r->%s);\n" name;
7132            ) stat_cols;
7133            pr "  free (r);\n";
7134            pr "  return jr;\n"
7135        | RStatVFS _ ->
7136            pr "  cl = (*env)->FindClass (env, \"com/redhat/et/libguestfs/StatVFS\");\n";
7137            pr "  jr = (*env)->AllocObject (env, cl);\n";
7138            List.iter (
7139              function
7140              | name, `Int ->
7141                  pr "  fl = (*env)->GetFieldID (env, cl, \"%s\", \"J\");\n"
7142                    name;
7143                  pr "  (*env)->SetLongField (env, jr, fl, r->%s);\n" name;
7144            ) statvfs_cols;
7145            pr "  free (r);\n";
7146            pr "  return jr;\n"
7147        | RPVList _ ->
7148            generate_java_lvm_return "pv" "PV" pv_cols
7149        | RVGList _ ->
7150            generate_java_lvm_return "vg" "VG" vg_cols
7151        | RLVList _ ->
7152            generate_java_lvm_return "lv" "LV" lv_cols
7153        | RHashtable _ ->
7154            (* XXX *)
7155            pr "  throw_exception (env, \"%s: internal error: please let us know how to make a Java HashMap from JNI bindings!\");\n" name;
7156            pr "  return NULL;\n"
7157       );
7158
7159       pr "}\n";
7160       pr "\n"
7161   ) all_functions
7162
7163 and generate_java_lvm_return typ jtyp cols =
7164   pr "  cl = (*env)->FindClass (env, \"com/redhat/et/libguestfs/%s\");\n" jtyp;
7165   pr "  jr = (*env)->NewObjectArray (env, r->len, cl, NULL);\n";
7166   pr "  for (i = 0; i < r->len; ++i) {\n";
7167   pr "    jfl = (*env)->AllocObject (env, cl);\n";
7168   List.iter (
7169     function
7170     | name, `String ->
7171         pr "    fl = (*env)->GetFieldID (env, cl, \"%s\", \"Ljava/lang/String;\");\n" name;
7172         pr "    (*env)->SetObjectField (env, jfl, fl, (*env)->NewStringUTF (env, r->val[i].%s));\n" name;
7173     | name, `UUID ->
7174         pr "    {\n";
7175         pr "      char s[33];\n";
7176         pr "      memcpy (s, r->val[i].%s, 32);\n" name;
7177         pr "      s[32] = 0;\n";
7178         pr "      fl = (*env)->GetFieldID (env, cl, \"%s\", \"Ljava/lang/String;\");\n" name;
7179         pr "      (*env)->SetObjectField (env, jfl, fl, (*env)->NewStringUTF (env, s));\n";
7180         pr "    }\n";
7181     | name, (`Bytes|`Int) ->
7182         pr "    fl = (*env)->GetFieldID (env, cl, \"%s\", \"J\");\n" name;
7183         pr "    (*env)->SetLongField (env, jfl, fl, r->val[i].%s);\n" name;
7184     | name, `OptPercent ->
7185         pr "    fl = (*env)->GetFieldID (env, cl, \"%s\", \"F\");\n" name;
7186         pr "    (*env)->SetFloatField (env, jfl, fl, r->val[i].%s);\n" name;
7187   ) cols;
7188   pr "    (*env)->SetObjectArrayElement (env, jfl, i, jfl);\n";
7189   pr "  }\n";
7190   pr "  guestfs_free_lvm_%s_list (r);\n" typ;
7191   pr "  return jr;\n"
7192
7193 and generate_haskell_hs () =
7194   generate_header HaskellStyle LGPLv2;
7195
7196   (* XXX We only know how to generate partial FFI for Haskell
7197    * at the moment.  Please help out!
7198    *)
7199   let can_generate style =
7200     let check_no_bad_args =
7201       List.for_all (function Bool _ | Int _ -> false | _ -> true)
7202     in
7203     match style with
7204     | RErr, args -> check_no_bad_args args
7205     | RBool _, _
7206     | RInt _, _
7207     | RInt64 _, _
7208     | RConstString _, _
7209     | RString _, _
7210     | RStringList _, _
7211     | RIntBool _, _
7212     | RPVList _, _
7213     | RVGList _, _
7214     | RLVList _, _
7215     | RStat _, _
7216     | RStatVFS _, _
7217     | RHashtable _, _ -> false in
7218
7219   pr "\
7220 {-# INCLUDE <guestfs.h> #-}
7221 {-# LANGUAGE ForeignFunctionInterface #-}
7222
7223 module Guestfs (
7224   create";
7225
7226   (* List out the names of the actions we want to export. *)
7227   List.iter (
7228     fun (name, style, _, _, _, _, _) ->
7229       if can_generate style then pr ",\n  %s" name
7230   ) all_functions;
7231
7232   pr "
7233   ) where
7234 import Foreign
7235 import Foreign.C
7236 import IO
7237 import Control.Exception
7238 import Data.Typeable
7239
7240 data GuestfsS = GuestfsS            -- represents the opaque C struct
7241 type GuestfsP = Ptr GuestfsS        -- guestfs_h *
7242 type GuestfsH = ForeignPtr GuestfsS -- guestfs_h * with attached finalizer
7243
7244 -- XXX define properly later XXX
7245 data PV = PV
7246 data VG = VG
7247 data LV = LV
7248 data IntBool = IntBool
7249 data Stat = Stat
7250 data StatVFS = StatVFS
7251 data Hashtable = Hashtable
7252
7253 foreign import ccall unsafe \"guestfs_create\" c_create
7254   :: IO GuestfsP
7255 foreign import ccall unsafe \"&guestfs_close\" c_close
7256   :: FunPtr (GuestfsP -> IO ())
7257 foreign import ccall unsafe \"guestfs_set_error_handler\" c_set_error_handler
7258   :: GuestfsP -> Ptr CInt -> Ptr CInt -> IO ()
7259
7260 create :: IO GuestfsH
7261 create = do
7262   p <- c_create
7263   c_set_error_handler p nullPtr nullPtr
7264   h <- newForeignPtr c_close p
7265   return h
7266
7267 foreign import ccall unsafe \"guestfs_last_error\" c_last_error
7268   :: GuestfsP -> IO CString
7269
7270 -- last_error :: GuestfsH -> IO (Maybe String)
7271 -- last_error h = do
7272 --   str <- withForeignPtr h (\\p -> c_last_error p)
7273 --   maybePeek peekCString str
7274
7275 last_error :: GuestfsH -> IO (String)
7276 last_error h = do
7277   str <- withForeignPtr h (\\p -> c_last_error p)
7278   if (str == nullPtr)
7279     then return \"no error\"
7280     else peekCString str
7281
7282 ";
7283
7284   (* Generate wrappers for each foreign function. *)
7285   List.iter (
7286     fun (name, style, _, _, _, _, _) ->
7287       if can_generate style then (
7288         pr "foreign import ccall unsafe \"guestfs_%s\" c_%s\n" name name;
7289         pr "  :: ";
7290         generate_haskell_prototype ~handle:"GuestfsP" style;
7291         pr "\n";
7292         pr "\n";
7293         pr "%s :: " name;
7294         generate_haskell_prototype ~handle:"GuestfsH" ~hs:true style;
7295         pr "\n";
7296         pr "%s %s = do\n" name
7297           (String.concat " " ("h" :: List.map name_of_argt (snd style)));
7298         pr "  r <- ";
7299         List.iter (
7300           function
7301           | FileIn n
7302           | FileOut n
7303           | String n -> pr "withCString %s $ \\%s -> " n n
7304           | OptString n -> pr "maybeWith withCString %s $ \\%s -> " n n
7305           | StringList n -> pr "withMany withCString %s $ \\%s -> withArray0 nullPtr %s $ \\%s -> " n n n n
7306           | Bool n ->
7307               (* XXX this doesn't work *)
7308               pr "      let\n";
7309               pr "        %s = case %s of\n" n n;
7310               pr "          False -> 0\n";
7311               pr "          True -> 1\n";
7312               pr "      in fromIntegral %s $ \\%s ->\n" n n
7313           | Int n -> pr "fromIntegral %s $ \\%s -> " n n
7314         ) (snd style);
7315         pr "withForeignPtr h (\\p -> c_%s %s)\n" name
7316           (String.concat " " ("p" :: List.map name_of_argt (snd style)));
7317         (match fst style with
7318          | RErr | RInt _ | RInt64 _ | RBool _ ->
7319              pr "  if (r == -1)\n";
7320              pr "    then do\n";
7321              pr "      err <- last_error h\n";
7322              pr "      fail err\n";
7323          | RConstString _ | RString _ | RStringList _ | RIntBool _
7324          | RPVList _ | RVGList _ | RLVList _ | RStat _ | RStatVFS _
7325          | RHashtable _ ->
7326              pr "  if (r == nullPtr)\n";
7327              pr "    then do\n";
7328              pr "      err <- last_error h\n";
7329              pr "      fail err\n";
7330         );
7331         (match fst style with
7332          | RErr ->
7333              pr "    else return ()\n"
7334          | RInt _ ->
7335              pr "    else return (fromIntegral r)\n"
7336          | RInt64 _ ->
7337              pr "    else return (fromIntegral r)\n"
7338          | RBool _ ->
7339              pr "    else return (toBool r)\n"
7340          | RConstString _
7341          | RString _
7342          | RStringList _
7343          | RIntBool _
7344          | RPVList _
7345          | RVGList _
7346          | RLVList _
7347          | RStat _
7348          | RStatVFS _
7349          | RHashtable _ ->
7350              pr "    else return ()\n" (* XXXXXXXXXXXXXXXXXXXX *)
7351         );
7352         pr "\n";
7353       )
7354   ) all_functions
7355
7356 and generate_haskell_prototype ~handle ?(hs = false) style =
7357   pr "%s -> " handle;
7358   let string = if hs then "String" else "CString" in
7359   let int = if hs then "Int" else "CInt" in
7360   let bool = if hs then "Bool" else "CInt" in
7361   let int64 = if hs then "Integer" else "Int64" in
7362   List.iter (
7363     fun arg ->
7364       (match arg with
7365        | String _ -> pr "%s" string
7366        | OptString _ -> if hs then pr "Maybe String" else pr "CString"
7367        | StringList _ -> if hs then pr "[String]" else pr "Ptr CString"
7368        | Bool _ -> pr "%s" bool
7369        | Int _ -> pr "%s" int
7370        | FileIn _ -> pr "%s" string
7371        | FileOut _ -> pr "%s" string
7372       );
7373       pr " -> ";
7374   ) (snd style);
7375   pr "IO (";
7376   (match fst style with
7377    | RErr -> if not hs then pr "CInt"
7378    | RInt _ -> pr "%s" int
7379    | RInt64 _ -> pr "%s" int64
7380    | RBool _ -> pr "%s" bool
7381    | RConstString _ -> pr "%s" string
7382    | RString _ -> pr "%s" string
7383    | RStringList _ -> pr "[%s]" string
7384    | RIntBool _ -> pr "IntBool"
7385    | RPVList _ -> pr "[PV]"
7386    | RVGList _ -> pr "[VG]"
7387    | RLVList _ -> pr "[LV]"
7388    | RStat _ -> pr "Stat"
7389    | RStatVFS _ -> pr "StatVFS"
7390    | RHashtable _ -> pr "Hashtable"
7391   );
7392   pr ")"
7393
7394 and generate_bindtests () =
7395   generate_header CStyle LGPLv2;
7396
7397   pr "\
7398 #include <stdio.h>
7399 #include <stdlib.h>
7400 #include <inttypes.h>
7401 #include <string.h>
7402
7403 #include \"guestfs.h\"
7404 #include \"guestfs_protocol.h\"
7405
7406 #define error guestfs_error
7407
7408 static void
7409 print_strings (char * const* const argv)
7410 {
7411   int argc;
7412
7413   printf (\"[\");
7414   for (argc = 0; argv[argc] != NULL; ++argc) {
7415     if (argc > 0) printf (\", \");
7416     printf (\"\\\"%%s\\\"\", argv[argc]);
7417   }
7418   printf (\"]\\n\");
7419 }
7420
7421 /* The test0 function prints its parameters to stdout. */
7422 ";
7423
7424   let test0, tests =
7425     match test_functions with
7426     | [] -> assert false
7427     | test0 :: tests -> test0, tests in
7428
7429   let () =
7430     let (name, style, _, _, _, _, _) = test0 in
7431     generate_prototype ~extern:false ~semicolon:false ~newline:true
7432       ~handle:"g" ~prefix:"guestfs_" name style;
7433     pr "{\n";
7434     List.iter (
7435       function
7436       | String n
7437       | FileIn n
7438       | FileOut n -> pr "  printf (\"%%s\\n\", %s);\n" n
7439       | OptString n -> pr "  printf (\"%%s\\n\", %s ? %s : \"null\");\n" n n
7440       | StringList n -> pr "  print_strings (%s);\n" n
7441       | Bool n -> pr "  printf (\"%%s\\n\", %s ? \"true\" : \"false\");\n" n
7442       | Int n -> pr "  printf (\"%%d\\n\", %s);\n" n
7443     ) (snd style);
7444     pr "  /* Java changes stdout line buffering so we need this: */\n";
7445     pr "  fflush (stdout);\n";
7446     pr "  return 0;\n";
7447     pr "}\n";
7448     pr "\n" in
7449
7450   List.iter (
7451     fun (name, style, _, _, _, _, _) ->
7452       if String.sub name (String.length name - 3) 3 <> "err" then (
7453         pr "/* Test normal return. */\n";
7454         generate_prototype ~extern:false ~semicolon:false ~newline:true
7455           ~handle:"g" ~prefix:"guestfs_" name style;
7456         pr "{\n";
7457         (match fst style with
7458          | RErr ->
7459              pr "  return 0;\n"
7460          | RInt _ ->
7461              pr "  int r;\n";
7462              pr "  sscanf (val, \"%%d\", &r);\n";
7463              pr "  return r;\n"
7464          | RInt64 _ ->
7465              pr "  int64_t r;\n";
7466              pr "  sscanf (val, \"%%\" SCNi64, &r);\n";
7467              pr "  return r;\n"
7468          | RBool _ ->
7469              pr "  return strcmp (val, \"true\") == 0;\n"
7470          | RConstString _ ->
7471              (* Can't return the input string here.  Return a static
7472               * string so we ensure we get a segfault if the caller
7473               * tries to free it.
7474               *)
7475              pr "  return \"static string\";\n"
7476          | RString _ ->
7477              pr "  return strdup (val);\n"
7478          | RStringList _ ->
7479              pr "  char **strs;\n";
7480              pr "  int n, i;\n";
7481              pr "  sscanf (val, \"%%d\", &n);\n";
7482              pr "  strs = malloc ((n+1) * sizeof (char *));\n";
7483              pr "  for (i = 0; i < n; ++i) {\n";
7484              pr "    strs[i] = malloc (16);\n";
7485              pr "    snprintf (strs[i], 16, \"%%d\", i);\n";
7486              pr "  }\n";
7487              pr "  strs[n] = NULL;\n";
7488              pr "  return strs;\n"
7489          | RIntBool _ ->
7490              pr "  struct guestfs_int_bool *r;\n";
7491              pr "  r = malloc (sizeof (struct guestfs_int_bool));\n";
7492              pr "  sscanf (val, \"%%\" SCNi32, &r->i);\n";
7493              pr "  r->b = 0;\n";
7494              pr "  return r;\n"
7495          | RPVList _ ->
7496              pr "  struct guestfs_lvm_pv_list *r;\n";
7497              pr "  int i;\n";
7498              pr "  r = malloc (sizeof (struct guestfs_lvm_pv_list));\n";
7499              pr "  sscanf (val, \"%%d\", &r->len);\n";
7500              pr "  r->val = calloc (r->len, sizeof (struct guestfs_lvm_pv));\n";
7501              pr "  for (i = 0; i < r->len; ++i) {\n";
7502              pr "    r->val[i].pv_name = malloc (16);\n";
7503              pr "    snprintf (r->val[i].pv_name, 16, \"%%d\", i);\n";
7504              pr "  }\n";
7505              pr "  return r;\n"
7506          | RVGList _ ->
7507              pr "  struct guestfs_lvm_vg_list *r;\n";
7508              pr "  int i;\n";
7509              pr "  r = malloc (sizeof (struct guestfs_lvm_vg_list));\n";
7510              pr "  sscanf (val, \"%%d\", &r->len);\n";
7511              pr "  r->val = calloc (r->len, sizeof (struct guestfs_lvm_vg));\n";
7512              pr "  for (i = 0; i < r->len; ++i) {\n";
7513              pr "    r->val[i].vg_name = malloc (16);\n";
7514              pr "    snprintf (r->val[i].vg_name, 16, \"%%d\", i);\n";
7515              pr "  }\n";
7516              pr "  return r;\n"
7517          | RLVList _ ->
7518              pr "  struct guestfs_lvm_lv_list *r;\n";
7519              pr "  int i;\n";
7520              pr "  r = malloc (sizeof (struct guestfs_lvm_lv_list));\n";
7521              pr "  sscanf (val, \"%%d\", &r->len);\n";
7522              pr "  r->val = calloc (r->len, sizeof (struct guestfs_lvm_lv));\n";
7523              pr "  for (i = 0; i < r->len; ++i) {\n";
7524              pr "    r->val[i].lv_name = malloc (16);\n";
7525              pr "    snprintf (r->val[i].lv_name, 16, \"%%d\", i);\n";
7526              pr "  }\n";
7527              pr "  return r;\n"
7528          | RStat _ ->
7529              pr "  struct guestfs_stat *r;\n";
7530              pr "  r = calloc (1, sizeof (*r));\n";
7531              pr "  sscanf (val, \"%%\" SCNi64, &r->dev);\n";
7532              pr "  return r;\n"
7533          | RStatVFS _ ->
7534              pr "  struct guestfs_statvfs *r;\n";
7535              pr "  r = calloc (1, sizeof (*r));\n";
7536              pr "  sscanf (val, \"%%\" SCNi64, &r->bsize);\n";
7537              pr "  return r;\n"
7538          | RHashtable _ ->
7539              pr "  char **strs;\n";
7540              pr "  int n, i;\n";
7541              pr "  sscanf (val, \"%%d\", &n);\n";
7542              pr "  strs = malloc ((n*2+1) * sizeof (char *));\n";
7543              pr "  for (i = 0; i < n; ++i) {\n";
7544              pr "    strs[i*2] = malloc (16);\n";
7545              pr "    strs[i*2+1] = malloc (16);\n";
7546              pr "    snprintf (strs[i*2], 16, \"%%d\", i);\n";
7547              pr "    snprintf (strs[i*2+1], 16, \"%%d\", i);\n";
7548              pr "  }\n";
7549              pr "  strs[n*2] = NULL;\n";
7550              pr "  return strs;\n"
7551         );
7552         pr "}\n";
7553         pr "\n"
7554       ) else (
7555         pr "/* Test error return. */\n";
7556         generate_prototype ~extern:false ~semicolon:false ~newline:true
7557           ~handle:"g" ~prefix:"guestfs_" name style;
7558         pr "{\n";
7559         pr "  error (g, \"error\");\n";
7560         (match fst style with
7561          | RErr | RInt _ | RInt64 _ | RBool _ ->
7562              pr "  return -1;\n"
7563          | RConstString _
7564          | RString _ | RStringList _ | RIntBool _
7565          | RPVList _ | RVGList _ | RLVList _ | RStat _ | RStatVFS _
7566          | RHashtable _ ->
7567              pr "  return NULL;\n"
7568         );
7569         pr "}\n";
7570         pr "\n"
7571       )
7572   ) tests
7573
7574 and generate_ocaml_bindtests () =
7575   generate_header OCamlStyle GPLv2;
7576
7577   pr "\
7578 let () =
7579   let g = Guestfs.create () in
7580 ";
7581
7582   let mkargs args =
7583     String.concat " " (
7584       List.map (
7585         function
7586         | CallString s -> "\"" ^ s ^ "\""
7587         | CallOptString None -> "None"
7588         | CallOptString (Some s) -> sprintf "(Some \"%s\")" s
7589         | CallStringList xs ->
7590             "[|" ^ String.concat ";" (List.map (sprintf "\"%s\"") xs) ^ "|]"
7591         | CallInt i when i >= 0 -> string_of_int i
7592         | CallInt i (* when i < 0 *) -> "(" ^ string_of_int i ^ ")"
7593         | CallBool b -> string_of_bool b
7594       ) args
7595     )
7596   in
7597
7598   generate_lang_bindtests (
7599     fun f args -> pr "  Guestfs.%s g %s;\n" f (mkargs args)
7600   );
7601
7602   pr "print_endline \"EOF\"\n"
7603
7604 and generate_perl_bindtests () =
7605   pr "#!/usr/bin/perl -w\n";
7606   generate_header HashStyle GPLv2;
7607
7608   pr "\
7609 use strict;
7610
7611 use Sys::Guestfs;
7612
7613 my $g = Sys::Guestfs->new ();
7614 ";
7615
7616   let mkargs args =
7617     String.concat ", " (
7618       List.map (
7619         function
7620         | CallString s -> "\"" ^ s ^ "\""
7621         | CallOptString None -> "undef"
7622         | CallOptString (Some s) -> sprintf "\"%s\"" s
7623         | CallStringList xs ->
7624             "[" ^ String.concat "," (List.map (sprintf "\"%s\"") xs) ^ "]"
7625         | CallInt i -> string_of_int i
7626         | CallBool b -> if b then "1" else "0"
7627       ) args
7628     )
7629   in
7630
7631   generate_lang_bindtests (
7632     fun f args -> pr "$g->%s (%s);\n" f (mkargs args)
7633   );
7634
7635   pr "print \"EOF\\n\"\n"
7636
7637 and generate_python_bindtests () =
7638   generate_header HashStyle GPLv2;
7639
7640   pr "\
7641 import guestfs
7642
7643 g = guestfs.GuestFS ()
7644 ";
7645
7646   let mkargs args =
7647     String.concat ", " (
7648       List.map (
7649         function
7650         | CallString s -> "\"" ^ s ^ "\""
7651         | CallOptString None -> "None"
7652         | CallOptString (Some s) -> sprintf "\"%s\"" s
7653         | CallStringList xs ->
7654             "[" ^ String.concat "," (List.map (sprintf "\"%s\"") xs) ^ "]"
7655         | CallInt i -> string_of_int i
7656         | CallBool b -> if b then "1" else "0"
7657       ) args
7658     )
7659   in
7660
7661   generate_lang_bindtests (
7662     fun f args -> pr "g.%s (%s)\n" f (mkargs args)
7663   );
7664
7665   pr "print \"EOF\"\n"
7666
7667 and generate_ruby_bindtests () =
7668   generate_header HashStyle GPLv2;
7669
7670   pr "\
7671 require 'guestfs'
7672
7673 g = Guestfs::create()
7674 ";
7675
7676   let mkargs args =
7677     String.concat ", " (
7678       List.map (
7679         function
7680         | CallString s -> "\"" ^ s ^ "\""
7681         | CallOptString None -> "nil"
7682         | CallOptString (Some s) -> sprintf "\"%s\"" s
7683         | CallStringList xs ->
7684             "[" ^ String.concat "," (List.map (sprintf "\"%s\"") xs) ^ "]"
7685         | CallInt i -> string_of_int i
7686         | CallBool b -> string_of_bool b
7687       ) args
7688     )
7689   in
7690
7691   generate_lang_bindtests (
7692     fun f args -> pr "g.%s(%s)\n" f (mkargs args)
7693   );
7694
7695   pr "print \"EOF\\n\"\n"
7696
7697 and generate_java_bindtests () =
7698   generate_header CStyle GPLv2;
7699
7700   pr "\
7701 import com.redhat.et.libguestfs.*;
7702
7703 public class Bindtests {
7704     public static void main (String[] argv)
7705     {
7706         try {
7707             GuestFS g = new GuestFS ();
7708 ";
7709
7710   let mkargs args =
7711     String.concat ", " (
7712       List.map (
7713         function
7714         | CallString s -> "\"" ^ s ^ "\""
7715         | CallOptString None -> "null"
7716         | CallOptString (Some s) -> sprintf "\"%s\"" s
7717         | CallStringList xs ->
7718             "new String[]{" ^
7719               String.concat "," (List.map (sprintf "\"%s\"") xs) ^ "}"
7720         | CallInt i -> string_of_int i
7721         | CallBool b -> string_of_bool b
7722       ) args
7723     )
7724   in
7725
7726   generate_lang_bindtests (
7727     fun f args -> pr "            g.%s (%s);\n" f (mkargs args)
7728   );
7729
7730   pr "
7731             System.out.println (\"EOF\");
7732         }
7733         catch (Exception exn) {
7734             System.err.println (exn);
7735             System.exit (1);
7736         }
7737     }
7738 }
7739 "
7740
7741 and generate_haskell_bindtests () =
7742   () (* XXX Haskell bindings need to be fleshed out. *)
7743
7744 (* Language-independent bindings tests - we do it this way to
7745  * ensure there is parity in testing bindings across all languages.
7746  *)
7747 and generate_lang_bindtests call =
7748   call "test0" [CallString "abc"; CallOptString (Some "def");
7749                 CallStringList []; CallBool false;
7750                 CallInt 0; CallString "123"; CallString "456"];
7751   call "test0" [CallString "abc"; CallOptString None;
7752                 CallStringList []; CallBool false;
7753                 CallInt 0; CallString "123"; CallString "456"];
7754   call "test0" [CallString ""; CallOptString (Some "def");
7755                 CallStringList []; CallBool false;
7756                 CallInt 0; CallString "123"; CallString "456"];
7757   call "test0" [CallString ""; CallOptString (Some "");
7758                 CallStringList []; CallBool false;
7759                 CallInt 0; CallString "123"; CallString "456"];
7760   call "test0" [CallString "abc"; CallOptString (Some "def");
7761                 CallStringList ["1"]; CallBool false;
7762                 CallInt 0; CallString "123"; CallString "456"];
7763   call "test0" [CallString "abc"; CallOptString (Some "def");
7764                 CallStringList ["1"; "2"]; CallBool false;
7765                 CallInt 0; CallString "123"; CallString "456"];
7766   call "test0" [CallString "abc"; CallOptString (Some "def");
7767                 CallStringList ["1"]; CallBool true;
7768                 CallInt 0; CallString "123"; CallString "456"];
7769   call "test0" [CallString "abc"; CallOptString (Some "def");
7770                 CallStringList ["1"]; CallBool false;
7771                 CallInt (-1); CallString "123"; CallString "456"];
7772   call "test0" [CallString "abc"; CallOptString (Some "def");
7773                 CallStringList ["1"]; CallBool false;
7774                 CallInt (-2); CallString "123"; CallString "456"];
7775   call "test0" [CallString "abc"; CallOptString (Some "def");
7776                 CallStringList ["1"]; CallBool false;
7777                 CallInt 1; CallString "123"; CallString "456"];
7778   call "test0" [CallString "abc"; CallOptString (Some "def");
7779                 CallStringList ["1"]; CallBool false;
7780                 CallInt 2; CallString "123"; CallString "456"];
7781   call "test0" [CallString "abc"; CallOptString (Some "def");
7782                 CallStringList ["1"]; CallBool false;
7783                 CallInt 4095; CallString "123"; CallString "456"];
7784   call "test0" [CallString "abc"; CallOptString (Some "def");
7785                 CallStringList ["1"]; CallBool false;
7786                 CallInt 0; CallString ""; CallString ""]
7787
7788   (* XXX Add here tests of the return and error functions. *)
7789
7790 let output_to filename =
7791   let filename_new = filename ^ ".new" in
7792   chan := open_out filename_new;
7793   let close () =
7794     close_out !chan;
7795     chan := stdout;
7796
7797     (* Is the new file different from the current file? *)
7798     if Sys.file_exists filename && files_equal filename filename_new then
7799       Unix.unlink filename_new          (* same, so skip it *)
7800     else (
7801       (* different, overwrite old one *)
7802       (try Unix.chmod filename 0o644 with Unix.Unix_error _ -> ());
7803       Unix.rename filename_new filename;
7804       Unix.chmod filename 0o444;
7805       printf "written %s\n%!" filename;
7806     )
7807   in
7808   close
7809
7810 (* Main program. *)
7811 let () =
7812   check_functions ();
7813
7814   if not (Sys.file_exists "configure.ac") then (
7815     eprintf "\
7816 You are probably running this from the wrong directory.
7817 Run it from the top source directory using the command
7818   src/generator.ml
7819 ";
7820     exit 1
7821   );
7822
7823   let close = output_to "src/guestfs_protocol.x" in
7824   generate_xdr ();
7825   close ();
7826
7827   let close = output_to "src/guestfs-structs.h" in
7828   generate_structs_h ();
7829   close ();
7830
7831   let close = output_to "src/guestfs-actions.h" in
7832   generate_actions_h ();
7833   close ();
7834
7835   let close = output_to "src/guestfs-actions.c" in
7836   generate_client_actions ();
7837   close ();
7838
7839   let close = output_to "daemon/actions.h" in
7840   generate_daemon_actions_h ();
7841   close ();
7842
7843   let close = output_to "daemon/stubs.c" in
7844   generate_daemon_actions ();
7845   close ();
7846
7847   let close = output_to "capitests/tests.c" in
7848   generate_tests ();
7849   close ();
7850
7851   let close = output_to "src/guestfs-bindtests.c" in
7852   generate_bindtests ();
7853   close ();
7854
7855   let close = output_to "fish/cmds.c" in
7856   generate_fish_cmds ();
7857   close ();
7858
7859   let close = output_to "fish/completion.c" in
7860   generate_fish_completion ();
7861   close ();
7862
7863   let close = output_to "guestfs-structs.pod" in
7864   generate_structs_pod ();
7865   close ();
7866
7867   let close = output_to "guestfs-actions.pod" in
7868   generate_actions_pod ();
7869   close ();
7870
7871   let close = output_to "guestfish-actions.pod" in
7872   generate_fish_actions_pod ();
7873   close ();
7874
7875   let close = output_to "ocaml/guestfs.mli" in
7876   generate_ocaml_mli ();
7877   close ();
7878
7879   let close = output_to "ocaml/guestfs.ml" in
7880   generate_ocaml_ml ();
7881   close ();
7882
7883   let close = output_to "ocaml/guestfs_c_actions.c" in
7884   generate_ocaml_c ();
7885   close ();
7886
7887   let close = output_to "ocaml/bindtests.ml" in
7888   generate_ocaml_bindtests ();
7889   close ();
7890
7891   let close = output_to "perl/Guestfs.xs" in
7892   generate_perl_xs ();
7893   close ();
7894
7895   let close = output_to "perl/lib/Sys/Guestfs.pm" in
7896   generate_perl_pm ();
7897   close ();
7898
7899   let close = output_to "perl/bindtests.pl" in
7900   generate_perl_bindtests ();
7901   close ();
7902
7903   let close = output_to "python/guestfs-py.c" in
7904   generate_python_c ();
7905   close ();
7906
7907   let close = output_to "python/guestfs.py" in
7908   generate_python_py ();
7909   close ();
7910
7911   let close = output_to "python/bindtests.py" in
7912   generate_python_bindtests ();
7913   close ();
7914
7915   let close = output_to "ruby/ext/guestfs/_guestfs.c" in
7916   generate_ruby_c ();
7917   close ();
7918
7919   let close = output_to "ruby/bindtests.rb" in
7920   generate_ruby_bindtests ();
7921   close ();
7922
7923   let close = output_to "java/com/redhat/et/libguestfs/GuestFS.java" in
7924   generate_java_java ();
7925   close ();
7926
7927   let close = output_to "java/com/redhat/et/libguestfs/PV.java" in
7928   generate_java_struct "PV" pv_cols;
7929   close ();
7930
7931   let close = output_to "java/com/redhat/et/libguestfs/VG.java" in
7932   generate_java_struct "VG" vg_cols;
7933   close ();
7934
7935   let close = output_to "java/com/redhat/et/libguestfs/LV.java" in
7936   generate_java_struct "LV" lv_cols;
7937   close ();
7938
7939   let close = output_to "java/com/redhat/et/libguestfs/Stat.java" in
7940   generate_java_struct "Stat" stat_cols;
7941   close ();
7942
7943   let close = output_to "java/com/redhat/et/libguestfs/StatVFS.java" in
7944   generate_java_struct "StatVFS" statvfs_cols;
7945   close ();
7946
7947   let close = output_to "java/com_redhat_et_libguestfs_GuestFS.c" in
7948   generate_java_c ();
7949   close ();
7950
7951   let close = output_to "java/Bindtests.java" in
7952   generate_java_bindtests ();
7953   close ();
7954
7955   let close = output_to "haskell/Guestfs.hs" in
7956   generate_haskell_hs ();
7957   close ();
7958
7959   let close = output_to "haskell/bindtests.hs" in
7960   generate_haskell_bindtests ();
7961   close ();