Added 'du' command.
[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,cache=off>.
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 See also: C<guestfs_scrub_device>.");
1988
1989   ("grub_install", (RErr, [String "root"; String "device"]), 86, [],
1990    [InitBasicFS, Always, TestOutputTrue (
1991       [["grub_install"; "/"; "/dev/sda1"];
1992        ["is_dir"; "/boot"]])],
1993    "install GRUB",
1994    "\
1995 This command installs GRUB (the Grand Unified Bootloader) on
1996 C<device>, with the root directory being C<root>.");
1997
1998   ("cp", (RErr, [String "src"; String "dest"]), 87, [],
1999    [InitBasicFS, Always, TestOutput (
2000       [["write_file"; "/old"; "file content"; "0"];
2001        ["cp"; "/old"; "/new"];
2002        ["cat"; "/new"]], "file content");
2003     InitBasicFS, Always, TestOutputTrue (
2004       [["write_file"; "/old"; "file content"; "0"];
2005        ["cp"; "/old"; "/new"];
2006        ["is_file"; "/old"]]);
2007     InitBasicFS, Always, TestOutput (
2008       [["write_file"; "/old"; "file content"; "0"];
2009        ["mkdir"; "/dir"];
2010        ["cp"; "/old"; "/dir/new"];
2011        ["cat"; "/dir/new"]], "file content")],
2012    "copy a file",
2013    "\
2014 This copies a file from C<src> to C<dest> where C<dest> is
2015 either a destination filename or destination directory.");
2016
2017   ("cp_a", (RErr, [String "src"; String "dest"]), 88, [],
2018    [InitBasicFS, Always, TestOutput (
2019       [["mkdir"; "/olddir"];
2020        ["mkdir"; "/newdir"];
2021        ["write_file"; "/olddir/file"; "file content"; "0"];
2022        ["cp_a"; "/olddir"; "/newdir"];
2023        ["cat"; "/newdir/olddir/file"]], "file content")],
2024    "copy a file or directory recursively",
2025    "\
2026 This copies a file or directory from C<src> to C<dest>
2027 recursively using the C<cp -a> command.");
2028
2029   ("mv", (RErr, [String "src"; String "dest"]), 89, [],
2030    [InitBasicFS, Always, TestOutput (
2031       [["write_file"; "/old"; "file content"; "0"];
2032        ["mv"; "/old"; "/new"];
2033        ["cat"; "/new"]], "file content");
2034     InitBasicFS, Always, TestOutputFalse (
2035       [["write_file"; "/old"; "file content"; "0"];
2036        ["mv"; "/old"; "/new"];
2037        ["is_file"; "/old"]])],
2038    "move a file",
2039    "\
2040 This moves a file from C<src> to C<dest> where C<dest> is
2041 either a destination filename or destination directory.");
2042
2043   ("drop_caches", (RErr, [Int "whattodrop"]), 90, [],
2044    [InitEmpty, Always, TestRun (
2045       [["drop_caches"; "3"]])],
2046    "drop kernel page cache, dentries and inodes",
2047    "\
2048 This instructs the guest kernel to drop its page cache,
2049 and/or dentries and inode caches.  The parameter C<whattodrop>
2050 tells the kernel what precisely to drop, see
2051 L<http://linux-mm.org/Drop_Caches>
2052
2053 Setting C<whattodrop> to 3 should drop everything.
2054
2055 This automatically calls L<sync(2)> before the operation,
2056 so that the maximum guest memory is freed.");
2057
2058   ("dmesg", (RString "kmsgs", []), 91, [],
2059    [InitEmpty, Always, TestRun (
2060       [["dmesg"]])],
2061    "return kernel messages",
2062    "\
2063 This returns the kernel messages (C<dmesg> output) from
2064 the guest kernel.  This is sometimes useful for extended
2065 debugging of problems.
2066
2067 Another way to get the same information is to enable
2068 verbose messages with C<guestfs_set_verbose> or by setting
2069 the environment variable C<LIBGUESTFS_DEBUG=1> before
2070 running the program.");
2071
2072   ("ping_daemon", (RErr, []), 92, [],
2073    [InitEmpty, Always, TestRun (
2074       [["ping_daemon"]])],
2075    "ping the guest daemon",
2076    "\
2077 This is a test probe into the guestfs daemon running inside
2078 the qemu subprocess.  Calling this function checks that the
2079 daemon responds to the ping message, without affecting the daemon
2080 or attached block device(s) in any other way.");
2081
2082   ("equal", (RBool "equality", [String "file1"; String "file2"]), 93, [],
2083    [InitBasicFS, Always, TestOutputTrue (
2084       [["write_file"; "/file1"; "contents of a file"; "0"];
2085        ["cp"; "/file1"; "/file2"];
2086        ["equal"; "/file1"; "/file2"]]);
2087     InitBasicFS, Always, TestOutputFalse (
2088       [["write_file"; "/file1"; "contents of a file"; "0"];
2089        ["write_file"; "/file2"; "contents of another file"; "0"];
2090        ["equal"; "/file1"; "/file2"]]);
2091     InitBasicFS, Always, TestLastFail (
2092       [["equal"; "/file1"; "/file2"]])],
2093    "test if two files have equal contents",
2094    "\
2095 This compares the two files C<file1> and C<file2> and returns
2096 true if their content is exactly equal, or false otherwise.
2097
2098 The external L<cmp(1)> program is used for the comparison.");
2099
2100   ("strings", (RStringList "stringsout", [String "path"]), 94, [ProtocolLimitWarning],
2101    [InitBasicFS, Always, TestOutputList (
2102       [["write_file"; "/new"; "hello\nworld\n"; "0"];
2103        ["strings"; "/new"]], ["hello"; "world"]);
2104     InitBasicFS, Always, TestOutputList (
2105       [["touch"; "/new"];
2106        ["strings"; "/new"]], [])],
2107    "print the printable strings in a file",
2108    "\
2109 This runs the L<strings(1)> command on a file and returns
2110 the list of printable strings found.");
2111
2112   ("strings_e", (RStringList "stringsout", [String "encoding"; String "path"]), 95, [ProtocolLimitWarning],
2113    [InitBasicFS, Always, TestOutputList (
2114       [["write_file"; "/new"; "hello\nworld\n"; "0"];
2115        ["strings_e"; "b"; "/new"]], []);
2116     InitBasicFS, Disabled, TestOutputList (
2117       [["write_file"; "/new"; "\000h\000e\000l\000l\000o\000\n\000w\000o\000r\000l\000d\000\n"; "24"];
2118        ["strings_e"; "b"; "/new"]], ["hello"; "world"])],
2119    "print the printable strings in a file",
2120    "\
2121 This is like the C<guestfs_strings> command, but allows you to
2122 specify the encoding.
2123
2124 See the L<strings(1)> manpage for the full list of encodings.
2125
2126 Commonly useful encodings are C<l> (lower case L) which will
2127 show strings inside Windows/x86 files.
2128
2129 The returned strings are transcoded to UTF-8.");
2130
2131   ("hexdump", (RString "dump", [String "path"]), 96, [ProtocolLimitWarning],
2132    [InitBasicFS, Always, TestOutput (
2133       [["write_file"; "/new"; "hello\nworld\n"; "12"];
2134        ["hexdump"; "/new"]], "00000000  68 65 6c 6c 6f 0a 77 6f  72 6c 64 0a              |hello.world.|\n0000000c\n");
2135     (* Test for RHBZ#501888c2 regression which caused large hexdump
2136      * commands to segfault.
2137      *)
2138     InitBasicFS, Always, TestRun (
2139       [["mount_vfs"; "ro"; "squashfs"; "/dev/sdd"; "/"];
2140        ["hexdump"; "/100krandom"]])],
2141    "dump a file in hexadecimal",
2142    "\
2143 This runs C<hexdump -C> on the given C<path>.  The result is
2144 the human-readable, canonical hex dump of the file.");
2145
2146   ("zerofree", (RErr, [String "device"]), 97, [],
2147    [InitNone, Always, TestOutput (
2148       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
2149        ["mkfs"; "ext3"; "/dev/sda1"];
2150        ["mount"; "/dev/sda1"; "/"];
2151        ["write_file"; "/new"; "test file"; "0"];
2152        ["umount"; "/dev/sda1"];
2153        ["zerofree"; "/dev/sda1"];
2154        ["mount"; "/dev/sda1"; "/"];
2155        ["cat"; "/new"]], "test file")],
2156    "zero unused inodes and disk blocks on ext2/3 filesystem",
2157    "\
2158 This runs the I<zerofree> program on C<device>.  This program
2159 claims to zero unused inodes and disk blocks on an ext2/3
2160 filesystem, thus making it possible to compress the filesystem
2161 more effectively.
2162
2163 You should B<not> run this program if the filesystem is
2164 mounted.
2165
2166 It is possible that using this program can damage the filesystem
2167 or data on the filesystem.");
2168
2169   ("pvresize", (RErr, [String "device"]), 98, [],
2170    [],
2171    "resize an LVM physical volume",
2172    "\
2173 This resizes (expands or shrinks) an existing LVM physical
2174 volume to match the new size of the underlying device.");
2175
2176   ("sfdisk_N", (RErr, [String "device"; Int "partnum";
2177                        Int "cyls"; Int "heads"; Int "sectors";
2178                        String "line"]), 99, [DangerWillRobinson],
2179    [],
2180    "modify a single partition on a block device",
2181    "\
2182 This runs L<sfdisk(8)> option to modify just the single
2183 partition C<n> (note: C<n> counts from 1).
2184
2185 For other parameters, see C<guestfs_sfdisk>.  You should usually
2186 pass C<0> for the cyls/heads/sectors parameters.");
2187
2188   ("sfdisk_l", (RString "partitions", [String "device"]), 100, [],
2189    [],
2190    "display the partition table",
2191    "\
2192 This displays the partition table on C<device>, in the
2193 human-readable output of the L<sfdisk(8)> command.  It is
2194 not intended to be parsed.");
2195
2196   ("sfdisk_kernel_geometry", (RString "partitions", [String "device"]), 101, [],
2197    [],
2198    "display the kernel geometry",
2199    "\
2200 This displays the kernel's idea of the geometry of C<device>.
2201
2202 The result is in human-readable format, and not designed to
2203 be parsed.");
2204
2205   ("sfdisk_disk_geometry", (RString "partitions", [String "device"]), 102, [],
2206    [],
2207    "display the disk geometry from the partition table",
2208    "\
2209 This displays the disk geometry of C<device> read from the
2210 partition table.  Especially in the case where the underlying
2211 block device has been resized, this can be different from the
2212 kernel's idea of the geometry (see C<guestfs_sfdisk_kernel_geometry>).
2213
2214 The result is in human-readable format, and not designed to
2215 be parsed.");
2216
2217   ("vg_activate_all", (RErr, [Bool "activate"]), 103, [],
2218    [],
2219    "activate or deactivate all volume groups",
2220    "\
2221 This command activates or (if C<activate> is false) deactivates
2222 all logical volumes in all volume groups.
2223 If activated, then they are made known to the
2224 kernel, ie. they appear as C</dev/mapper> devices.  If deactivated,
2225 then those devices disappear.
2226
2227 This command is the same as running C<vgchange -a y|n>");
2228
2229   ("vg_activate", (RErr, [Bool "activate"; StringList "volgroups"]), 104, [],
2230    [],
2231    "activate or deactivate some volume groups",
2232    "\
2233 This command activates or (if C<activate> is false) deactivates
2234 all logical volumes in the listed volume groups C<volgroups>.
2235 If activated, then they are made known to the
2236 kernel, ie. they appear as C</dev/mapper> devices.  If deactivated,
2237 then those devices disappear.
2238
2239 This command is the same as running C<vgchange -a y|n volgroups...>
2240
2241 Note that if C<volgroups> is an empty list then B<all> volume groups
2242 are activated or deactivated.");
2243
2244   ("lvresize", (RErr, [String "device"; Int "mbytes"]), 105, [],
2245    [InitNone, Always, TestOutput (
2246     [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
2247      ["pvcreate"; "/dev/sda1"];
2248      ["vgcreate"; "VG"; "/dev/sda1"];
2249      ["lvcreate"; "LV"; "VG"; "10"];
2250      ["mkfs"; "ext2"; "/dev/VG/LV"];
2251      ["mount"; "/dev/VG/LV"; "/"];
2252      ["write_file"; "/new"; "test content"; "0"];
2253      ["umount"; "/"];
2254      ["lvresize"; "/dev/VG/LV"; "20"];
2255      ["e2fsck_f"; "/dev/VG/LV"];
2256      ["resize2fs"; "/dev/VG/LV"];
2257      ["mount"; "/dev/VG/LV"; "/"];
2258      ["cat"; "/new"]], "test content")],
2259    "resize an LVM logical volume",
2260    "\
2261 This resizes (expands or shrinks) an existing LVM logical
2262 volume to C<mbytes>.  When reducing, data in the reduced part
2263 is lost.");
2264
2265   ("resize2fs", (RErr, [String "device"]), 106, [],
2266    [], (* lvresize tests this *)
2267    "resize an ext2/ext3 filesystem",
2268    "\
2269 This resizes an ext2 or ext3 filesystem to match the size of
2270 the underlying device.
2271
2272 I<Note:> It is sometimes required that you run C<guestfs_e2fsck_f>
2273 on the C<device> before calling this command.  For unknown reasons
2274 C<resize2fs> sometimes gives an error about this and sometimes not.
2275 In any case, it is always safe to call C<guestfs_e2fsck_f> before
2276 calling this function.");
2277
2278   ("find", (RStringList "names", [String "directory"]), 107, [],
2279    [InitBasicFS, Always, TestOutputList (
2280       [["find"; "/"]], ["lost+found"]);
2281     InitBasicFS, Always, TestOutputList (
2282       [["touch"; "/a"];
2283        ["mkdir"; "/b"];
2284        ["touch"; "/b/c"];
2285        ["find"; "/"]], ["a"; "b"; "b/c"; "lost+found"]);
2286     InitBasicFS, Always, TestOutputList (
2287       [["mkdir_p"; "/a/b/c"];
2288        ["touch"; "/a/b/c/d"];
2289        ["find"; "/a/b/"]], ["c"; "c/d"])],
2290    "find all files and directories",
2291    "\
2292 This command lists out all files and directories, recursively,
2293 starting at C<directory>.  It is essentially equivalent to
2294 running the shell command C<find directory -print> but some
2295 post-processing happens on the output, described below.
2296
2297 This returns a list of strings I<without any prefix>.  Thus
2298 if the directory structure was:
2299
2300  /tmp/a
2301  /tmp/b
2302  /tmp/c/d
2303
2304 then the returned list from C<guestfs_find> C</tmp> would be
2305 4 elements:
2306
2307  a
2308  b
2309  c
2310  c/d
2311
2312 If C<directory> is not a directory, then this command returns
2313 an error.
2314
2315 The returned list is sorted.");
2316
2317   ("e2fsck_f", (RErr, [String "device"]), 108, [],
2318    [], (* lvresize tests this *)
2319    "check an ext2/ext3 filesystem",
2320    "\
2321 This runs C<e2fsck -p -f device>, ie. runs the ext2/ext3
2322 filesystem checker on C<device>, noninteractively (C<-p>),
2323 even if the filesystem appears to be clean (C<-f>).
2324
2325 This command is only needed because of C<guestfs_resize2fs>
2326 (q.v.).  Normally you should use C<guestfs_fsck>.");
2327
2328   ("sleep", (RErr, [Int "secs"]), 109, [],
2329    [InitNone, Always, TestRun (
2330     [["sleep"; "1"]])],
2331    "sleep for some seconds",
2332    "\
2333 Sleep for C<secs> seconds.");
2334
2335   ("ntfs_3g_probe", (RInt "status", [Bool "rw"; String "device"]), 110, [],
2336    [InitNone, Always, TestOutputInt (
2337       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
2338        ["mkfs"; "ntfs"; "/dev/sda1"];
2339        ["ntfs_3g_probe"; "true"; "/dev/sda1"]], 0);
2340     InitNone, Always, TestOutputInt (
2341       [["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
2342        ["mkfs"; "ext2"; "/dev/sda1"];
2343        ["ntfs_3g_probe"; "true"; "/dev/sda1"]], 12)],
2344    "probe NTFS volume",
2345    "\
2346 This command runs the L<ntfs-3g.probe(8)> command which probes
2347 an NTFS C<device> for mountability.  (Not all NTFS volumes can
2348 be mounted read-write, and some cannot be mounted at all).
2349
2350 C<rw> is a boolean flag.  Set it to true if you want to test
2351 if the volume can be mounted read-write.  Set it to false if
2352 you want to test if the volume can be mounted read-only.
2353
2354 The return value is an integer which C<0> if the operation
2355 would succeed, or some non-zero value documented in the
2356 L<ntfs-3g.probe(8)> manual page.");
2357
2358   ("sh", (RString "output", [String "command"]), 111, [],
2359    [], (* XXX needs tests *)
2360    "run a command via the shell",
2361    "\
2362 This call runs a command from the guest filesystem via the
2363 guest's C</bin/sh>.
2364
2365 This is like C<guestfs_command>, but passes the command to:
2366
2367  /bin/sh -c \"command\"
2368
2369 Depending on the guest's shell, this usually results in
2370 wildcards being expanded, shell expressions being interpolated
2371 and so on.
2372
2373 All the provisos about C<guestfs_command> apply to this call.");
2374
2375   ("sh_lines", (RStringList "lines", [String "command"]), 112, [],
2376    [], (* XXX needs tests *)
2377    "run a command via the shell returning lines",
2378    "\
2379 This is the same as C<guestfs_sh>, but splits the result
2380 into a list of lines.
2381
2382 See also: C<guestfs_command_lines>");
2383
2384   ("glob_expand", (RStringList "paths", [String "pattern"]), 113, [],
2385    [InitBasicFS, Always, TestOutputList (
2386       [["mkdir_p"; "/a/b/c"];
2387        ["touch"; "/a/b/c/d"];
2388        ["touch"; "/a/b/c/e"];
2389        ["glob_expand"; "/a/b/c/*"]], ["/a/b/c/d"; "/a/b/c/e"]);
2390     InitBasicFS, Always, TestOutputList (
2391       [["mkdir_p"; "/a/b/c"];
2392        ["touch"; "/a/b/c/d"];
2393        ["touch"; "/a/b/c/e"];
2394        ["glob_expand"; "/a/*/c/*"]], ["/a/b/c/d"; "/a/b/c/e"]);
2395     InitBasicFS, Always, TestOutputList (
2396       [["mkdir_p"; "/a/b/c"];
2397        ["touch"; "/a/b/c/d"];
2398        ["touch"; "/a/b/c/e"];
2399        ["glob_expand"; "/a/*/x/*"]], [])],
2400    "expand a wildcard path",
2401    "\
2402 This command searches for all the pathnames matching
2403 C<pattern> according to the wildcard expansion rules
2404 used by the shell.
2405
2406 If no paths match, then this returns an empty list
2407 (note: not an error).
2408
2409 It is just a wrapper around the C L<glob(3)> function
2410 with flags C<GLOB_MARK|GLOB_BRACE>.
2411 See that manual page for more details.");
2412
2413   ("scrub_device", (RErr, [String "device"]), 114, [DangerWillRobinson],
2414    [InitNone, Always, TestRun ( (* use /dev/sdc because it's smaller *)
2415       [["scrub_device"; "/dev/sdc"]])],
2416    "scrub (securely wipe) a device",
2417    "\
2418 This command writes patterns over C<device> to make data retrieval
2419 more difficult.
2420
2421 It is an interface to the L<scrub(1)> program.  See that
2422 manual page for more details.");
2423
2424   ("scrub_file", (RErr, [String "file"]), 115, [],
2425    [InitBasicFS, Always, TestRun (
2426       [["write_file"; "/file"; "content"; "0"];
2427        ["scrub_file"; "/file"]])],
2428    "scrub (securely wipe) a file",
2429    "\
2430 This command writes patterns over a file to make data retrieval
2431 more difficult.
2432
2433 The file is I<removed> after scrubbing.
2434
2435 It is an interface to the L<scrub(1)> program.  See that
2436 manual page for more details.");
2437
2438   ("scrub_freespace", (RErr, [String "dir"]), 116, [],
2439    [], (* XXX needs testing *)
2440    "scrub (securely wipe) free space",
2441    "\
2442 This command creates the directory C<dir> and then fills it
2443 with files until the filesystem is full, and scrubs the files
2444 as for C<guestfs_scrub_file>, and deletes them.
2445 The intention is to scrub any free space on the partition
2446 containing C<dir>.
2447
2448 It is an interface to the L<scrub(1)> program.  See that
2449 manual page for more details.");
2450
2451   ("mkdtemp", (RString "dir", [String "template"]), 117, [],
2452    [InitBasicFS, Always, TestRun (
2453       [["mkdir"; "/tmp"];
2454        ["mkdtemp"; "/tmp/tmpXXXXXX"]])],
2455    "create a temporary directory",
2456    "\
2457 This command creates a temporary directory.  The
2458 C<template> parameter should be a full pathname for the
2459 temporary directory name with the final six characters being
2460 \"XXXXXX\".
2461
2462 For example: \"/tmp/myprogXXXXXX\" or \"/Temp/myprogXXXXXX\",
2463 the second one being suitable for Windows filesystems.
2464
2465 The name of the temporary directory that was created
2466 is returned.
2467
2468 The temporary directory is created with mode 0700
2469 and is owned by root.
2470
2471 The caller is responsible for deleting the temporary
2472 directory and its contents after use.
2473
2474 See also: L<mkdtemp(3)>");
2475
2476   ("wc_l", (RInt "lines", [String "path"]), 118, [],
2477    [InitBasicFS, Always, TestOutputInt (
2478       [["mount_vfs"; "ro"; "squashfs"; "/dev/sdd"; "/"];
2479        ["wc_l"; "/10klines"]], 10000)],
2480    "count lines in a file",
2481    "\
2482 This command counts the lines in a file, using the
2483 C<wc -l> external command.");
2484
2485   ("wc_w", (RInt "words", [String "path"]), 119, [],
2486    [InitBasicFS, Always, TestOutputInt (
2487       [["mount_vfs"; "ro"; "squashfs"; "/dev/sdd"; "/"];
2488        ["wc_w"; "/10klines"]], 10000)],
2489    "count words in a file",
2490    "\
2491 This command counts the words in a file, using the
2492 C<wc -w> external command.");
2493
2494   ("wc_c", (RInt "chars", [String "path"]), 120, [],
2495    [InitBasicFS, Always, TestOutputInt (
2496       [["mount_vfs"; "ro"; "squashfs"; "/dev/sdd"; "/"];
2497        ["wc_c"; "/100kallspaces"]], 102400)],
2498    "count characters in a file",
2499    "\
2500 This command counts the characters in a file, using the
2501 C<wc -c> external command.");
2502
2503   ("head", (RStringList "lines", [String "path"]), 121, [ProtocolLimitWarning],
2504    [InitBasicFS, Always, TestOutputList (
2505       [["mount_vfs"; "ro"; "squashfs"; "/dev/sdd"; "/"];
2506        ["head"; "/10klines"]], ["0abcdefghijklmnopqrstuvwxyz";"1abcdefghijklmnopqrstuvwxyz";"2abcdefghijklmnopqrstuvwxyz";"3abcdefghijklmnopqrstuvwxyz";"4abcdefghijklmnopqrstuvwxyz";"5abcdefghijklmnopqrstuvwxyz";"6abcdefghijklmnopqrstuvwxyz";"7abcdefghijklmnopqrstuvwxyz";"8abcdefghijklmnopqrstuvwxyz";"9abcdefghijklmnopqrstuvwxyz"])],
2507    "return first 10 lines of a file",
2508    "\
2509 This command returns up to the first 10 lines of a file as
2510 a list of strings.");
2511
2512   ("head_n", (RStringList "lines", [Int "nrlines"; String "path"]), 122, [ProtocolLimitWarning],
2513    [InitBasicFS, Always, TestOutputList (
2514       [["mount_vfs"; "ro"; "squashfs"; "/dev/sdd"; "/"];
2515        ["head_n"; "3"; "/10klines"]], ["0abcdefghijklmnopqrstuvwxyz";"1abcdefghijklmnopqrstuvwxyz";"2abcdefghijklmnopqrstuvwxyz"]);
2516     InitBasicFS, Always, TestOutputList (
2517       [["mount_vfs"; "ro"; "squashfs"; "/dev/sdd"; "/"];
2518        ["head_n"; "-9997"; "/10klines"]], ["0abcdefghijklmnopqrstuvwxyz";"1abcdefghijklmnopqrstuvwxyz";"2abcdefghijklmnopqrstuvwxyz"]);
2519     InitBasicFS, Always, TestOutputList (
2520       [["mount_vfs"; "ro"; "squashfs"; "/dev/sdd"; "/"];
2521        ["head_n"; "0"; "/10klines"]], [])],
2522    "return first N lines of a file",
2523    "\
2524 If the parameter C<nrlines> is a positive number, this returns the first
2525 C<nrlines> lines of the file C<path>.
2526
2527 If the parameter C<nrlines> is a negative number, this returns lines
2528 from the file C<path>, excluding the last C<nrlines> lines.
2529
2530 If the parameter C<nrlines> is zero, this returns an empty list.");
2531
2532   ("tail", (RStringList "lines", [String "path"]), 123, [ProtocolLimitWarning],
2533    [InitBasicFS, Always, TestOutputList (
2534       [["mount_vfs"; "ro"; "squashfs"; "/dev/sdd"; "/"];
2535        ["tail"; "/10klines"]], ["9990abcdefghijklmnopqrstuvwxyz";"9991abcdefghijklmnopqrstuvwxyz";"9992abcdefghijklmnopqrstuvwxyz";"9993abcdefghijklmnopqrstuvwxyz";"9994abcdefghijklmnopqrstuvwxyz";"9995abcdefghijklmnopqrstuvwxyz";"9996abcdefghijklmnopqrstuvwxyz";"9997abcdefghijklmnopqrstuvwxyz";"9998abcdefghijklmnopqrstuvwxyz";"9999abcdefghijklmnopqrstuvwxyz"])],
2536    "return last 10 lines of a file",
2537    "\
2538 This command returns up to the last 10 lines of a file as
2539 a list of strings.");
2540
2541   ("tail_n", (RStringList "lines", [Int "nrlines"; String "path"]), 124, [ProtocolLimitWarning],
2542    [InitBasicFS, Always, TestOutputList (
2543       [["mount_vfs"; "ro"; "squashfs"; "/dev/sdd"; "/"];
2544        ["tail_n"; "3"; "/10klines"]], ["9997abcdefghijklmnopqrstuvwxyz";"9998abcdefghijklmnopqrstuvwxyz";"9999abcdefghijklmnopqrstuvwxyz"]);
2545     InitBasicFS, Always, TestOutputList (
2546       [["mount_vfs"; "ro"; "squashfs"; "/dev/sdd"; "/"];
2547        ["tail_n"; "-9998"; "/10klines"]], ["9997abcdefghijklmnopqrstuvwxyz";"9998abcdefghijklmnopqrstuvwxyz";"9999abcdefghijklmnopqrstuvwxyz"]);
2548     InitBasicFS, Always, TestOutputList (
2549       [["mount_vfs"; "ro"; "squashfs"; "/dev/sdd"; "/"];
2550        ["tail_n"; "0"; "/10klines"]], [])],
2551    "return last N lines of a file",
2552    "\
2553 If the parameter C<nrlines> is a positive number, this returns the last
2554 C<nrlines> lines of the file C<path>.
2555
2556 If the parameter C<nrlines> is a negative number, this returns lines
2557 from the file C<path>, starting with the C<-nrlines>th line.
2558
2559 If the parameter C<nrlines> is zero, this returns an empty list.");
2560
2561   ("df", (RString "output", []), 125, [],
2562    [], (* XXX Tricky to test because it depends on the exact format
2563         * of the 'df' command and other imponderables.
2564         *)
2565    "report file system disk space usage",
2566    "\
2567 This command runs the C<df> command to report disk space used.
2568
2569 This command is mostly useful for interactive sessions.  It
2570 is I<not> intended that you try to parse the output string.
2571 Use C<statvfs> from programs.");
2572
2573   ("df_h", (RString "output", []), 126, [],
2574    [], (* XXX Tricky to test because it depends on the exact format
2575         * of the 'df' command and other imponderables.
2576         *)
2577    "report file system disk space usage (human readable)",
2578    "\
2579 This command runs the C<df -h> command to report disk space used
2580 in human-readable format.
2581
2582 This command is mostly useful for interactive sessions.  It
2583 is I<not> intended that you try to parse the output string.
2584 Use C<statvfs> from programs.");
2585
2586   ("du", (RInt64 "sizekb", [String "path"]), 127, [],
2587    [InitBasicFS, Always, TestOutputInt (
2588       [["mkdir"; "/p"];
2589        ["du"; "/p"]], 1 (* ie. 1 block, so depends on ext3 blocksize *))],
2590    "estimate file space usage",
2591    "\
2592 This command runs the C<du -s> command to estimate file space
2593 usage for C<path>.
2594
2595 C<path> can be a file or a directory.  If C<path> is a directory
2596 then the estimate includes the contents of the directory and all
2597 subdirectories (recursively).
2598
2599 The result is the estimated size in I<kilobytes>
2600 (ie. units of 1024 bytes).");
2601
2602 ]
2603
2604 let all_functions = non_daemon_functions @ daemon_functions
2605
2606 (* In some places we want the functions to be displayed sorted
2607  * alphabetically, so this is useful:
2608  *)
2609 let all_functions_sorted =
2610   List.sort (fun (n1,_,_,_,_,_,_) (n2,_,_,_,_,_,_) ->
2611                compare n1 n2) all_functions
2612
2613 (* Column names and types from LVM PVs/VGs/LVs. *)
2614 let pv_cols = [
2615   "pv_name", `String;
2616   "pv_uuid", `UUID;
2617   "pv_fmt", `String;
2618   "pv_size", `Bytes;
2619   "dev_size", `Bytes;
2620   "pv_free", `Bytes;
2621   "pv_used", `Bytes;
2622   "pv_attr", `String (* XXX *);
2623   "pv_pe_count", `Int;
2624   "pv_pe_alloc_count", `Int;
2625   "pv_tags", `String;
2626   "pe_start", `Bytes;
2627   "pv_mda_count", `Int;
2628   "pv_mda_free", `Bytes;
2629 (* Not in Fedora 10:
2630   "pv_mda_size", `Bytes;
2631 *)
2632 ]
2633 let vg_cols = [
2634   "vg_name", `String;
2635   "vg_uuid", `UUID;
2636   "vg_fmt", `String;
2637   "vg_attr", `String (* XXX *);
2638   "vg_size", `Bytes;
2639   "vg_free", `Bytes;
2640   "vg_sysid", `String;
2641   "vg_extent_size", `Bytes;
2642   "vg_extent_count", `Int;
2643   "vg_free_count", `Int;
2644   "max_lv", `Int;
2645   "max_pv", `Int;
2646   "pv_count", `Int;
2647   "lv_count", `Int;
2648   "snap_count", `Int;
2649   "vg_seqno", `Int;
2650   "vg_tags", `String;
2651   "vg_mda_count", `Int;
2652   "vg_mda_free", `Bytes;
2653 (* Not in Fedora 10:
2654   "vg_mda_size", `Bytes;
2655 *)
2656 ]
2657 let lv_cols = [
2658   "lv_name", `String;
2659   "lv_uuid", `UUID;
2660   "lv_attr", `String (* XXX *);
2661   "lv_major", `Int;
2662   "lv_minor", `Int;
2663   "lv_kernel_major", `Int;
2664   "lv_kernel_minor", `Int;
2665   "lv_size", `Bytes;
2666   "seg_count", `Int;
2667   "origin", `String;
2668   "snap_percent", `OptPercent;
2669   "copy_percent", `OptPercent;
2670   "move_pv", `String;
2671   "lv_tags", `String;
2672   "mirror_log", `String;
2673   "modules", `String;
2674 ]
2675
2676 (* Column names and types from stat structures.
2677  * NB. Can't use things like 'st_atime' because glibc header files
2678  * define some of these as macros.  Ugh.
2679  *)
2680 let stat_cols = [
2681   "dev", `Int;
2682   "ino", `Int;
2683   "mode", `Int;
2684   "nlink", `Int;
2685   "uid", `Int;
2686   "gid", `Int;
2687   "rdev", `Int;
2688   "size", `Int;
2689   "blksize", `Int;
2690   "blocks", `Int;
2691   "atime", `Int;
2692   "mtime", `Int;
2693   "ctime", `Int;
2694 ]
2695 let statvfs_cols = [
2696   "bsize", `Int;
2697   "frsize", `Int;
2698   "blocks", `Int;
2699   "bfree", `Int;
2700   "bavail", `Int;
2701   "files", `Int;
2702   "ffree", `Int;
2703   "favail", `Int;
2704   "fsid", `Int;
2705   "flag", `Int;
2706   "namemax", `Int;
2707 ]
2708
2709 (* Used for testing language bindings. *)
2710 type callt =
2711   | CallString of string
2712   | CallOptString of string option
2713   | CallStringList of string list
2714   | CallInt of int
2715   | CallBool of bool
2716
2717 (* Useful functions.
2718  * Note we don't want to use any external OCaml libraries which
2719  * makes this a bit harder than it should be.
2720  *)
2721 let failwithf fs = ksprintf failwith fs
2722
2723 let replace_char s c1 c2 =
2724   let s2 = String.copy s in
2725   let r = ref false in
2726   for i = 0 to String.length s2 - 1 do
2727     if String.unsafe_get s2 i = c1 then (
2728       String.unsafe_set s2 i c2;
2729       r := true
2730     )
2731   done;
2732   if not !r then s else s2
2733
2734 let isspace c =
2735   c = ' '
2736   (* || c = '\f' *) || c = '\n' || c = '\r' || c = '\t' (* || c = '\v' *)
2737
2738 let triml ?(test = isspace) str =
2739   let i = ref 0 in
2740   let n = ref (String.length str) in
2741   while !n > 0 && test str.[!i]; do
2742     decr n;
2743     incr i
2744   done;
2745   if !i = 0 then str
2746   else String.sub str !i !n
2747
2748 let trimr ?(test = isspace) str =
2749   let n = ref (String.length str) in
2750   while !n > 0 && test str.[!n-1]; do
2751     decr n
2752   done;
2753   if !n = String.length str then str
2754   else String.sub str 0 !n
2755
2756 let trim ?(test = isspace) str =
2757   trimr ~test (triml ~test str)
2758
2759 let rec find s sub =
2760   let len = String.length s in
2761   let sublen = String.length sub in
2762   let rec loop i =
2763     if i <= len-sublen then (
2764       let rec loop2 j =
2765         if j < sublen then (
2766           if s.[i+j] = sub.[j] then loop2 (j+1)
2767           else -1
2768         ) else
2769           i (* found *)
2770       in
2771       let r = loop2 0 in
2772       if r = -1 then loop (i+1) else r
2773     ) else
2774       -1 (* not found *)
2775   in
2776   loop 0
2777
2778 let rec replace_str s s1 s2 =
2779   let len = String.length s in
2780   let sublen = String.length s1 in
2781   let i = find s s1 in
2782   if i = -1 then s
2783   else (
2784     let s' = String.sub s 0 i in
2785     let s'' = String.sub s (i+sublen) (len-i-sublen) in
2786     s' ^ s2 ^ replace_str s'' s1 s2
2787   )
2788
2789 let rec string_split sep str =
2790   let len = String.length str in
2791   let seplen = String.length sep in
2792   let i = find str sep in
2793   if i = -1 then [str]
2794   else (
2795     let s' = String.sub str 0 i in
2796     let s'' = String.sub str (i+seplen) (len-i-seplen) in
2797     s' :: string_split sep s''
2798   )
2799
2800 let files_equal n1 n2 =
2801   let cmd = sprintf "cmp -s %s %s" (Filename.quote n1) (Filename.quote n2) in
2802   match Sys.command cmd with
2803   | 0 -> true
2804   | 1 -> false
2805   | i -> failwithf "%s: failed with error code %d" cmd i
2806
2807 let rec find_map f = function
2808   | [] -> raise Not_found
2809   | x :: xs ->
2810       match f x with
2811       | Some y -> y
2812       | None -> find_map f xs
2813
2814 let iteri f xs =
2815   let rec loop i = function
2816     | [] -> ()
2817     | x :: xs -> f i x; loop (i+1) xs
2818   in
2819   loop 0 xs
2820
2821 let mapi f xs =
2822   let rec loop i = function
2823     | [] -> []
2824     | x :: xs -> let r = f i x in r :: loop (i+1) xs
2825   in
2826   loop 0 xs
2827
2828 let name_of_argt = function
2829   | String n | OptString n | StringList n | Bool n | Int n
2830   | FileIn n | FileOut n -> n
2831
2832 let seq_of_test = function
2833   | TestRun s | TestOutput (s, _) | TestOutputList (s, _)
2834   | TestOutputListOfDevices (s, _)
2835   | TestOutputInt (s, _) | TestOutputTrue s | TestOutputFalse s
2836   | TestOutputLength (s, _) | TestOutputStruct (s, _)
2837   | TestLastFail s -> s
2838
2839 (* Check function names etc. for consistency. *)
2840 let check_functions () =
2841   let contains_uppercase str =
2842     let len = String.length str in
2843     let rec loop i =
2844       if i >= len then false
2845       else (
2846         let c = str.[i] in
2847         if c >= 'A' && c <= 'Z' then true
2848         else loop (i+1)
2849       )
2850     in
2851     loop 0
2852   in
2853
2854   (* Check function names. *)
2855   List.iter (
2856     fun (name, _, _, _, _, _, _) ->
2857       if String.length name >= 7 && String.sub name 0 7 = "guestfs" then
2858         failwithf "function name %s does not need 'guestfs' prefix" name;
2859       if name = "" then
2860         failwithf "function name is empty";
2861       if name.[0] < 'a' || name.[0] > 'z' then
2862         failwithf "function name %s must start with lowercase a-z" name;
2863       if String.contains name '-' then
2864         failwithf "function name %s should not contain '-', use '_' instead."
2865           name
2866   ) all_functions;
2867
2868   (* Check function parameter/return names. *)
2869   List.iter (
2870     fun (name, style, _, _, _, _, _) ->
2871       let check_arg_ret_name n =
2872         if contains_uppercase n then
2873           failwithf "%s param/ret %s should not contain uppercase chars"
2874             name n;
2875         if String.contains n '-' || String.contains n '_' then
2876           failwithf "%s param/ret %s should not contain '-' or '_'"
2877             name n;
2878         if n = "value" then
2879           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;
2880         if n = "int" || n = "char" || n = "short" || n = "long" then
2881           failwithf "%s has a param/ret which conflicts with a C type (eg. 'int', 'char' etc.)" name;
2882         if n = "i" || n = "n" then
2883           failwithf "%s has a param/ret called 'i' or 'n', which will cause some conflicts in the generated code" name;
2884         if n = "argv" || n = "args" then
2885           failwithf "%s has a param/ret called 'argv' or 'args', which will cause some conflicts in the generated code" name
2886       in
2887
2888       (match fst style with
2889        | RErr -> ()
2890        | RInt n | RInt64 n | RBool n | RConstString n | RString n
2891        | RStringList n | RPVList n | RVGList n | RLVList n
2892        | RStat n | RStatVFS n
2893        | RHashtable n ->
2894            check_arg_ret_name n
2895        | RIntBool (n,m) ->
2896            check_arg_ret_name n;
2897            check_arg_ret_name m
2898       );
2899       List.iter (fun arg -> check_arg_ret_name (name_of_argt arg)) (snd style)
2900   ) all_functions;
2901
2902   (* Check short descriptions. *)
2903   List.iter (
2904     fun (name, _, _, _, _, shortdesc, _) ->
2905       if shortdesc.[0] <> Char.lowercase shortdesc.[0] then
2906         failwithf "short description of %s should begin with lowercase." name;
2907       let c = shortdesc.[String.length shortdesc-1] in
2908       if c = '\n' || c = '.' then
2909         failwithf "short description of %s should not end with . or \\n." name
2910   ) all_functions;
2911
2912   (* Check long dscriptions. *)
2913   List.iter (
2914     fun (name, _, _, _, _, _, longdesc) ->
2915       if longdesc.[String.length longdesc-1] = '\n' then
2916         failwithf "long description of %s should not end with \\n." name
2917   ) all_functions;
2918
2919   (* Check proc_nrs. *)
2920   List.iter (
2921     fun (name, _, proc_nr, _, _, _, _) ->
2922       if proc_nr <= 0 then
2923         failwithf "daemon function %s should have proc_nr > 0" name
2924   ) daemon_functions;
2925
2926   List.iter (
2927     fun (name, _, proc_nr, _, _, _, _) ->
2928       if proc_nr <> -1 then
2929         failwithf "non-daemon function %s should have proc_nr -1" name
2930   ) non_daemon_functions;
2931
2932   let proc_nrs =
2933     List.map (fun (name, _, proc_nr, _, _, _, _) -> name, proc_nr)
2934       daemon_functions in
2935   let proc_nrs =
2936     List.sort (fun (_,nr1) (_,nr2) -> compare nr1 nr2) proc_nrs in
2937   let rec loop = function
2938     | [] -> ()
2939     | [_] -> ()
2940     | (name1,nr1) :: ((name2,nr2) :: _ as rest) when nr1 < nr2 ->
2941         loop rest
2942     | (name1,nr1) :: (name2,nr2) :: _ ->
2943         failwithf "%s and %s have conflicting procedure numbers (%d, %d)"
2944           name1 name2 nr1 nr2
2945   in
2946   loop proc_nrs;
2947
2948   (* Check tests. *)
2949   List.iter (
2950     function
2951       (* Ignore functions that have no tests.  We generate a
2952        * warning when the user does 'make check' instead.
2953        *)
2954     | name, _, _, _, [], _, _ -> ()
2955     | name, _, _, _, tests, _, _ ->
2956         let funcs =
2957           List.map (
2958             fun (_, _, test) ->
2959               match seq_of_test test with
2960               | [] ->
2961                   failwithf "%s has a test containing an empty sequence" name
2962               | cmds -> List.map List.hd cmds
2963           ) tests in
2964         let funcs = List.flatten funcs in
2965
2966         let tested = List.mem name funcs in
2967
2968         if not tested then
2969           failwithf "function %s has tests but does not test itself" name
2970   ) all_functions
2971
2972 (* 'pr' prints to the current output file. *)
2973 let chan = ref stdout
2974 let pr fs = ksprintf (output_string !chan) fs
2975
2976 (* Generate a header block in a number of standard styles. *)
2977 type comment_style = CStyle | HashStyle | OCamlStyle | HaskellStyle
2978 type license = GPLv2 | LGPLv2
2979
2980 let generate_header comment license =
2981   let c = match comment with
2982     | CStyle ->     pr "/* "; " *"
2983     | HashStyle ->  pr "# ";  "#"
2984     | OCamlStyle -> pr "(* "; " *"
2985     | HaskellStyle -> pr "{- "; "  " in
2986   pr "libguestfs generated file\n";
2987   pr "%s WARNING: THIS FILE IS GENERATED BY 'src/generator.ml'.\n" c;
2988   pr "%s ANY CHANGES YOU MAKE TO THIS FILE WILL BE LOST.\n" c;
2989   pr "%s\n" c;
2990   pr "%s Copyright (C) 2009 Red Hat Inc.\n" c;
2991   pr "%s\n" c;
2992   (match license with
2993    | GPLv2 ->
2994        pr "%s This program is free software; you can redistribute it and/or modify\n" c;
2995        pr "%s it under the terms of the GNU General Public License as published by\n" c;
2996        pr "%s the Free Software Foundation; either version 2 of the License, or\n" c;
2997        pr "%s (at your option) any later version.\n" c;
2998        pr "%s\n" c;
2999        pr "%s This program is distributed in the hope that it will be useful,\n" c;
3000        pr "%s but WITHOUT ANY WARRANTY; without even the implied warranty of\n" c;
3001        pr "%s MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\n" c;
3002        pr "%s GNU General Public License for more details.\n" c;
3003        pr "%s\n" c;
3004        pr "%s You should have received a copy of the GNU General Public License along\n" c;
3005        pr "%s with this program; if not, write to the Free Software Foundation, Inc.,\n" c;
3006        pr "%s 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.\n" c;
3007
3008    | LGPLv2 ->
3009        pr "%s This library is free software; you can redistribute it and/or\n" c;
3010        pr "%s modify it under the terms of the GNU Lesser General Public\n" c;
3011        pr "%s License as published by the Free Software Foundation; either\n" c;
3012        pr "%s version 2 of the License, or (at your option) any later version.\n" c;
3013        pr "%s\n" c;
3014        pr "%s This library is distributed in the hope that it will be useful,\n" c;
3015        pr "%s but WITHOUT ANY WARRANTY; without even the implied warranty of\n" c;
3016        pr "%s MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU\n" c;
3017        pr "%s Lesser General Public License for more details.\n" c;
3018        pr "%s\n" c;
3019        pr "%s You should have received a copy of the GNU Lesser General Public\n" c;
3020        pr "%s License along with this library; if not, write to the Free Software\n" c;
3021        pr "%s Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA\n" c;
3022   );
3023   (match comment with
3024    | CStyle -> pr " */\n"
3025    | HashStyle -> ()
3026    | OCamlStyle -> pr " *)\n"
3027    | HaskellStyle -> pr "-}\n"
3028   );
3029   pr "\n"
3030
3031 (* Start of main code generation functions below this line. *)
3032
3033 (* Generate the pod documentation for the C API. *)
3034 let rec generate_actions_pod () =
3035   List.iter (
3036     fun (shortname, style, _, flags, _, _, longdesc) ->
3037       if not (List.mem NotInDocs flags) then (
3038         let name = "guestfs_" ^ shortname in
3039         pr "=head2 %s\n\n" name;
3040         pr " ";
3041         generate_prototype ~extern:false ~handle:"handle" name style;
3042         pr "\n\n";
3043         pr "%s\n\n" longdesc;
3044         (match fst style with
3045          | RErr ->
3046              pr "This function returns 0 on success or -1 on error.\n\n"
3047          | RInt _ ->
3048              pr "On error this function returns -1.\n\n"
3049          | RInt64 _ ->
3050              pr "On error this function returns -1.\n\n"
3051          | RBool _ ->
3052              pr "This function returns a C truth value on success or -1 on error.\n\n"
3053          | RConstString _ ->
3054              pr "This function returns a string, or NULL on error.
3055 The string is owned by the guest handle and must I<not> be freed.\n\n"
3056          | RString _ ->
3057              pr "This function returns a string, or NULL on error.
3058 I<The caller must free the returned string after use>.\n\n"
3059          | RStringList _ ->
3060              pr "This function returns a NULL-terminated array of strings
3061 (like L<environ(3)>), or NULL if there was an error.
3062 I<The caller must free the strings and the array after use>.\n\n"
3063          | RIntBool _ ->
3064              pr "This function returns a C<struct guestfs_int_bool *>,
3065 or NULL if there was an error.
3066 I<The caller must call C<guestfs_free_int_bool> after use>.\n\n"
3067          | RPVList _ ->
3068              pr "This function returns a C<struct guestfs_lvm_pv_list *>
3069 (see E<lt>guestfs-structs.hE<gt>),
3070 or NULL if there was an error.
3071 I<The caller must call C<guestfs_free_lvm_pv_list> after use>.\n\n"
3072          | RVGList _ ->
3073              pr "This function returns a C<struct guestfs_lvm_vg_list *>
3074 (see E<lt>guestfs-structs.hE<gt>),
3075 or NULL if there was an error.
3076 I<The caller must call C<guestfs_free_lvm_vg_list> after use>.\n\n"
3077          | RLVList _ ->
3078              pr "This function returns a C<struct guestfs_lvm_lv_list *>
3079 (see E<lt>guestfs-structs.hE<gt>),
3080 or NULL if there was an error.
3081 I<The caller must call C<guestfs_free_lvm_lv_list> after use>.\n\n"
3082          | RStat _ ->
3083              pr "This function returns a C<struct guestfs_stat *>
3084 (see L<stat(2)> and E<lt>guestfs-structs.hE<gt>),
3085 or NULL if there was an error.
3086 I<The caller must call C<free> after use>.\n\n"
3087          | RStatVFS _ ->
3088              pr "This function returns a C<struct guestfs_statvfs *>
3089 (see L<statvfs(2)> and E<lt>guestfs-structs.hE<gt>),
3090 or NULL if there was an error.
3091 I<The caller must call C<free> after use>.\n\n"
3092          | RHashtable _ ->
3093              pr "This function returns a NULL-terminated array of
3094 strings, or NULL if there was an error.
3095 The array of strings will always have length C<2n+1>, where
3096 C<n> keys and values alternate, followed by the trailing NULL entry.
3097 I<The caller must free the strings and the array after use>.\n\n"
3098         );
3099         if List.mem ProtocolLimitWarning flags then
3100           pr "%s\n\n" protocol_limit_warning;
3101         if List.mem DangerWillRobinson flags then
3102           pr "%s\n\n" danger_will_robinson
3103       )
3104   ) all_functions_sorted
3105
3106 and generate_structs_pod () =
3107   (* LVM structs documentation. *)
3108   List.iter (
3109     fun (typ, cols) ->
3110       pr "=head2 guestfs_lvm_%s\n" typ;
3111       pr "\n";
3112       pr " struct guestfs_lvm_%s {\n" typ;
3113       List.iter (
3114         function
3115         | name, `String -> pr "  char *%s;\n" name
3116         | name, `UUID ->
3117             pr "  /* The next field is NOT nul-terminated, be careful when printing it: */\n";
3118             pr "  char %s[32];\n" name
3119         | name, `Bytes -> pr "  uint64_t %s;\n" name
3120         | name, `Int -> pr "  int64_t %s;\n" name
3121         | name, `OptPercent ->
3122             pr "  /* The next field is [0..100] or -1 meaning 'not present': */\n";
3123             pr "  float %s;\n" name
3124       ) cols;
3125       pr " \n";
3126       pr " struct guestfs_lvm_%s_list {\n" typ;
3127       pr "   uint32_t len; /* Number of elements in list. */\n";
3128       pr "   struct guestfs_lvm_%s *val; /* Elements. */\n" typ;
3129       pr " };\n";
3130       pr " \n";
3131       pr " void guestfs_free_lvm_%s_list (struct guestfs_free_lvm_%s_list *);\n"
3132         typ typ;
3133       pr "\n"
3134   ) ["pv", pv_cols; "vg", vg_cols; "lv", lv_cols]
3135
3136 (* Generate the protocol (XDR) file, 'guestfs_protocol.x' and
3137  * indirectly 'guestfs_protocol.h' and 'guestfs_protocol.c'.
3138  *
3139  * We have to use an underscore instead of a dash because otherwise
3140  * rpcgen generates incorrect code.
3141  *
3142  * This header is NOT exported to clients, but see also generate_structs_h.
3143  *)
3144 and generate_xdr () =
3145   generate_header CStyle LGPLv2;
3146
3147   (* This has to be defined to get around a limitation in Sun's rpcgen. *)
3148   pr "typedef string str<>;\n";
3149   pr "\n";
3150
3151   (* LVM internal structures. *)
3152   List.iter (
3153     function
3154     | typ, cols ->
3155         pr "struct guestfs_lvm_int_%s {\n" typ;
3156         List.iter (function
3157                    | name, `String -> pr "  string %s<>;\n" name
3158                    | name, `UUID -> pr "  opaque %s[32];\n" name
3159                    | name, `Bytes -> pr "  hyper %s;\n" name
3160                    | name, `Int -> pr "  hyper %s;\n" name
3161                    | name, `OptPercent -> pr "  float %s;\n" name
3162                   ) cols;
3163         pr "};\n";
3164         pr "\n";
3165         pr "typedef struct guestfs_lvm_int_%s guestfs_lvm_int_%s_list<>;\n" typ typ;
3166         pr "\n";
3167   ) ["pv", pv_cols; "vg", vg_cols; "lv", lv_cols];
3168
3169   (* Stat internal structures. *)
3170   List.iter (
3171     function
3172     | typ, cols ->
3173         pr "struct guestfs_int_%s {\n" typ;
3174         List.iter (function
3175                    | name, `Int -> pr "  hyper %s;\n" name
3176                   ) cols;
3177         pr "};\n";
3178         pr "\n";
3179   ) ["stat", stat_cols; "statvfs", statvfs_cols];
3180
3181   List.iter (
3182     fun (shortname, style, _, _, _, _, _) ->
3183       let name = "guestfs_" ^ shortname in
3184
3185       (match snd style with
3186        | [] -> ()
3187        | args ->
3188            pr "struct %s_args {\n" name;
3189            List.iter (
3190              function
3191              | String n -> pr "  string %s<>;\n" n
3192              | OptString n -> pr "  str *%s;\n" n
3193              | StringList n -> pr "  str %s<>;\n" n
3194              | Bool n -> pr "  bool %s;\n" n
3195              | Int n -> pr "  int %s;\n" n
3196              | FileIn _ | FileOut _ -> ()
3197            ) args;
3198            pr "};\n\n"
3199       );
3200       (match fst style with
3201        | RErr -> ()
3202        | RInt n ->
3203            pr "struct %s_ret {\n" name;
3204            pr "  int %s;\n" n;
3205            pr "};\n\n"
3206        | RInt64 n ->
3207            pr "struct %s_ret {\n" name;
3208            pr "  hyper %s;\n" n;
3209            pr "};\n\n"
3210        | RBool n ->
3211            pr "struct %s_ret {\n" name;
3212            pr "  bool %s;\n" n;
3213            pr "};\n\n"
3214        | RConstString _ ->
3215            failwithf "RConstString cannot be returned from a daemon function"
3216        | RString n ->
3217            pr "struct %s_ret {\n" name;
3218            pr "  string %s<>;\n" n;
3219            pr "};\n\n"
3220        | RStringList n ->
3221            pr "struct %s_ret {\n" name;
3222            pr "  str %s<>;\n" n;
3223            pr "};\n\n"
3224        | RIntBool (n,m) ->
3225            pr "struct %s_ret {\n" name;
3226            pr "  int %s;\n" n;
3227            pr "  bool %s;\n" m;
3228            pr "};\n\n"
3229        | RPVList n ->
3230            pr "struct %s_ret {\n" name;
3231            pr "  guestfs_lvm_int_pv_list %s;\n" n;
3232            pr "};\n\n"
3233        | RVGList n ->
3234            pr "struct %s_ret {\n" name;
3235            pr "  guestfs_lvm_int_vg_list %s;\n" n;
3236            pr "};\n\n"
3237        | RLVList n ->
3238            pr "struct %s_ret {\n" name;
3239            pr "  guestfs_lvm_int_lv_list %s;\n" n;
3240            pr "};\n\n"
3241        | RStat n ->
3242            pr "struct %s_ret {\n" name;
3243            pr "  guestfs_int_stat %s;\n" n;
3244            pr "};\n\n"
3245        | RStatVFS n ->
3246            pr "struct %s_ret {\n" name;
3247            pr "  guestfs_int_statvfs %s;\n" n;
3248            pr "};\n\n"
3249        | RHashtable n ->
3250            pr "struct %s_ret {\n" name;
3251            pr "  str %s<>;\n" n;
3252            pr "};\n\n"
3253       );
3254   ) daemon_functions;
3255
3256   (* Table of procedure numbers. *)
3257   pr "enum guestfs_procedure {\n";
3258   List.iter (
3259     fun (shortname, _, proc_nr, _, _, _, _) ->
3260       pr "  GUESTFS_PROC_%s = %d,\n" (String.uppercase shortname) proc_nr
3261   ) daemon_functions;
3262   pr "  GUESTFS_PROC_NR_PROCS\n";
3263   pr "};\n";
3264   pr "\n";
3265
3266   (* Having to choose a maximum message size is annoying for several
3267    * reasons (it limits what we can do in the API), but it (a) makes
3268    * the protocol a lot simpler, and (b) provides a bound on the size
3269    * of the daemon which operates in limited memory space.  For large
3270    * file transfers you should use FTP.
3271    *)
3272   pr "const GUESTFS_MESSAGE_MAX = %d;\n" (4 * 1024 * 1024);
3273   pr "\n";
3274
3275   (* Message header, etc. *)
3276   pr "\
3277 /* The communication protocol is now documented in the guestfs(3)
3278  * manpage.
3279  */
3280
3281 const GUESTFS_PROGRAM = 0x2000F5F5;
3282 const GUESTFS_PROTOCOL_VERSION = 1;
3283
3284 /* These constants must be larger than any possible message length. */
3285 const GUESTFS_LAUNCH_FLAG = 0xf5f55ff5;
3286 const GUESTFS_CANCEL_FLAG = 0xffffeeee;
3287
3288 enum guestfs_message_direction {
3289   GUESTFS_DIRECTION_CALL = 0,        /* client -> daemon */
3290   GUESTFS_DIRECTION_REPLY = 1        /* daemon -> client */
3291 };
3292
3293 enum guestfs_message_status {
3294   GUESTFS_STATUS_OK = 0,
3295   GUESTFS_STATUS_ERROR = 1
3296 };
3297
3298 const GUESTFS_ERROR_LEN = 256;
3299
3300 struct guestfs_message_error {
3301   string error_message<GUESTFS_ERROR_LEN>;
3302 };
3303
3304 struct guestfs_message_header {
3305   unsigned prog;                     /* GUESTFS_PROGRAM */
3306   unsigned vers;                     /* GUESTFS_PROTOCOL_VERSION */
3307   guestfs_procedure proc;            /* GUESTFS_PROC_x */
3308   guestfs_message_direction direction;
3309   unsigned serial;                   /* message serial number */
3310   guestfs_message_status status;
3311 };
3312
3313 const GUESTFS_MAX_CHUNK_SIZE = 8192;
3314
3315 struct guestfs_chunk {
3316   int cancel;                        /* if non-zero, transfer is cancelled */
3317   /* data size is 0 bytes if the transfer has finished successfully */
3318   opaque data<GUESTFS_MAX_CHUNK_SIZE>;
3319 };
3320 "
3321
3322 (* Generate the guestfs-structs.h file. *)
3323 and generate_structs_h () =
3324   generate_header CStyle LGPLv2;
3325
3326   (* This is a public exported header file containing various
3327    * structures.  The structures are carefully written to have
3328    * exactly the same in-memory format as the XDR structures that
3329    * we use on the wire to the daemon.  The reason for creating
3330    * copies of these structures here is just so we don't have to
3331    * export the whole of guestfs_protocol.h (which includes much
3332    * unrelated and XDR-dependent stuff that we don't want to be
3333    * public, or required by clients).
3334    *
3335    * To reiterate, we will pass these structures to and from the
3336    * client with a simple assignment or memcpy, so the format
3337    * must be identical to what rpcgen / the RFC defines.
3338    *)
3339
3340   (* guestfs_int_bool structure. *)
3341   pr "struct guestfs_int_bool {\n";
3342   pr "  int32_t i;\n";
3343   pr "  int32_t b;\n";
3344   pr "};\n";
3345   pr "\n";
3346
3347   (* LVM public structures. *)
3348   List.iter (
3349     function
3350     | typ, cols ->
3351         pr "struct guestfs_lvm_%s {\n" typ;
3352         List.iter (
3353           function
3354           | name, `String -> pr "  char *%s;\n" name
3355           | name, `UUID -> pr "  char %s[32]; /* this is NOT nul-terminated, be careful when printing */\n" name
3356           | name, `Bytes -> pr "  uint64_t %s;\n" name
3357           | name, `Int -> pr "  int64_t %s;\n" name
3358           | name, `OptPercent -> pr "  float %s; /* [0..100] or -1 */\n" name
3359         ) cols;
3360         pr "};\n";
3361         pr "\n";
3362         pr "struct guestfs_lvm_%s_list {\n" typ;
3363         pr "  uint32_t len;\n";
3364         pr "  struct guestfs_lvm_%s *val;\n" typ;
3365         pr "};\n";
3366         pr "\n"
3367   ) ["pv", pv_cols; "vg", vg_cols; "lv", lv_cols];
3368
3369   (* Stat structures. *)
3370   List.iter (
3371     function
3372     | typ, cols ->
3373         pr "struct guestfs_%s {\n" typ;
3374         List.iter (
3375           function
3376           | name, `Int -> pr "  int64_t %s;\n" name
3377         ) cols;
3378         pr "};\n";
3379         pr "\n"
3380   ) ["stat", stat_cols; "statvfs", statvfs_cols]
3381
3382 (* Generate the guestfs-actions.h file. *)
3383 and generate_actions_h () =
3384   generate_header CStyle LGPLv2;
3385   List.iter (
3386     fun (shortname, style, _, _, _, _, _) ->
3387       let name = "guestfs_" ^ shortname in
3388       generate_prototype ~single_line:true ~newline:true ~handle:"handle"
3389         name style
3390   ) all_functions
3391
3392 (* Generate the client-side dispatch stubs. *)
3393 and generate_client_actions () =
3394   generate_header CStyle LGPLv2;
3395
3396   pr "\
3397 #include <stdio.h>
3398 #include <stdlib.h>
3399
3400 #include \"guestfs.h\"
3401 #include \"guestfs_protocol.h\"
3402
3403 #define error guestfs_error
3404 #define perrorf guestfs_perrorf
3405 #define safe_malloc guestfs_safe_malloc
3406 #define safe_realloc guestfs_safe_realloc
3407 #define safe_strdup guestfs_safe_strdup
3408 #define safe_memdup guestfs_safe_memdup
3409
3410 /* Check the return message from a call for validity. */
3411 static int
3412 check_reply_header (guestfs_h *g,
3413                     const struct guestfs_message_header *hdr,
3414                     int proc_nr, int serial)
3415 {
3416   if (hdr->prog != GUESTFS_PROGRAM) {
3417     error (g, \"wrong program (%%d/%%d)\", hdr->prog, GUESTFS_PROGRAM);
3418     return -1;
3419   }
3420   if (hdr->vers != GUESTFS_PROTOCOL_VERSION) {
3421     error (g, \"wrong protocol version (%%d/%%d)\",
3422            hdr->vers, GUESTFS_PROTOCOL_VERSION);
3423     return -1;
3424   }
3425   if (hdr->direction != GUESTFS_DIRECTION_REPLY) {
3426     error (g, \"unexpected message direction (%%d/%%d)\",
3427            hdr->direction, GUESTFS_DIRECTION_REPLY);
3428     return -1;
3429   }
3430   if (hdr->proc != proc_nr) {
3431     error (g, \"unexpected procedure number (%%d/%%d)\", hdr->proc, proc_nr);
3432     return -1;
3433   }
3434   if (hdr->serial != serial) {
3435     error (g, \"unexpected serial (%%d/%%d)\", hdr->serial, serial);
3436     return -1;
3437   }
3438
3439   return 0;
3440 }
3441
3442 /* Check we are in the right state to run a high-level action. */
3443 static int
3444 check_state (guestfs_h *g, const char *caller)
3445 {
3446   if (!guestfs_is_ready (g)) {
3447     if (guestfs_is_config (g))
3448       error (g, \"%%s: call launch() before using this function\",
3449         caller);
3450     else if (guestfs_is_launching (g))
3451       error (g, \"%%s: call wait_ready() before using this function\",
3452         caller);
3453     else
3454       error (g, \"%%s called from the wrong state, %%d != READY\",
3455         caller, guestfs_get_state (g));
3456     return -1;
3457   }
3458   return 0;
3459 }
3460
3461 ";
3462
3463   (* Client-side stubs for each function. *)
3464   List.iter (
3465     fun (shortname, style, _, _, _, _, _) ->
3466       let name = "guestfs_" ^ shortname in
3467
3468       (* Generate the context struct which stores the high-level
3469        * state between callback functions.
3470        *)
3471       pr "struct %s_ctx {\n" shortname;
3472       pr "  /* This flag is set by the callbacks, so we know we've done\n";
3473       pr "   * the callbacks as expected, and in the right sequence.\n";
3474       pr "   * 0 = not called, 1 = reply_cb called.\n";
3475       pr "   */\n";
3476       pr "  int cb_sequence;\n";
3477       pr "  struct guestfs_message_header hdr;\n";
3478       pr "  struct guestfs_message_error err;\n";
3479       (match fst style with
3480        | RErr -> ()
3481        | RConstString _ ->
3482            failwithf "RConstString cannot be returned from a daemon function"
3483        | RInt _ | RInt64 _
3484        | RBool _ | RString _ | RStringList _
3485        | RIntBool _
3486        | RPVList _ | RVGList _ | RLVList _
3487        | RStat _ | RStatVFS _
3488        | RHashtable _ ->
3489            pr "  struct %s_ret ret;\n" name
3490       );
3491       pr "};\n";
3492       pr "\n";
3493
3494       (* Generate the reply callback function. *)
3495       pr "static void %s_reply_cb (guestfs_h *g, void *data, XDR *xdr)\n" shortname;
3496       pr "{\n";
3497       pr "  guestfs_main_loop *ml = guestfs_get_main_loop (g);\n";
3498       pr "  struct %s_ctx *ctx = (struct %s_ctx *) data;\n" shortname shortname;
3499       pr "\n";
3500       pr "  /* This should definitely not happen. */\n";
3501       pr "  if (ctx->cb_sequence != 0) {\n";
3502       pr "    ctx->cb_sequence = 9999;\n";
3503       pr "    error (g, \"%%s: internal error: reply callback called twice\", \"%s\");\n" name;
3504       pr "    return;\n";
3505       pr "  }\n";
3506       pr "\n";
3507       pr "  ml->main_loop_quit (ml, g);\n";
3508       pr "\n";
3509       pr "  if (!xdr_guestfs_message_header (xdr, &ctx->hdr)) {\n";
3510       pr "    error (g, \"%%s: failed to parse reply header\", \"%s\");\n" name;
3511       pr "    return;\n";
3512       pr "  }\n";
3513       pr "  if (ctx->hdr.status == GUESTFS_STATUS_ERROR) {\n";
3514       pr "    if (!xdr_guestfs_message_error (xdr, &ctx->err)) {\n";
3515       pr "      error (g, \"%%s: failed to parse reply error\", \"%s\");\n"
3516         name;
3517       pr "      return;\n";
3518       pr "    }\n";
3519       pr "    goto done;\n";
3520       pr "  }\n";
3521
3522       (match fst style with
3523        | RErr -> ()
3524        | RConstString _ ->
3525            failwithf "RConstString cannot be returned from a daemon function"
3526        | RInt _ | RInt64 _
3527        | RBool _ | RString _ | RStringList _
3528        | RIntBool _
3529        | RPVList _ | RVGList _ | RLVList _
3530        | RStat _ | RStatVFS _
3531        | RHashtable _ ->
3532             pr "  if (!xdr_%s_ret (xdr, &ctx->ret)) {\n" name;
3533             pr "    error (g, \"%%s: failed to parse reply\", \"%s\");\n" name;
3534             pr "    return;\n";
3535             pr "  }\n";
3536       );
3537
3538       pr " done:\n";
3539       pr "  ctx->cb_sequence = 1;\n";
3540       pr "}\n\n";
3541
3542       (* Generate the action stub. *)
3543       generate_prototype ~extern:false ~semicolon:false ~newline:true
3544         ~handle:"g" name style;
3545
3546       let error_code =
3547         match fst style with
3548         | RErr | RInt _ | RInt64 _ | RBool _ -> "-1"
3549         | RConstString _ ->
3550             failwithf "RConstString cannot be returned from a daemon function"
3551         | RString _ | RStringList _ | RIntBool _
3552         | RPVList _ | RVGList _ | RLVList _
3553         | RStat _ | RStatVFS _
3554         | RHashtable _ ->
3555             "NULL" in
3556
3557       pr "{\n";
3558
3559       (match snd style with
3560        | [] -> ()
3561        | _ -> pr "  struct %s_args args;\n" name
3562       );
3563
3564       pr "  struct %s_ctx ctx;\n" shortname;
3565       pr "  guestfs_main_loop *ml = guestfs_get_main_loop (g);\n";
3566       pr "  int serial;\n";
3567       pr "\n";
3568       pr "  if (check_state (g, \"%s\") == -1) return %s;\n" name error_code;
3569       pr "  guestfs_set_busy (g);\n";
3570       pr "\n";
3571       pr "  memset (&ctx, 0, sizeof ctx);\n";
3572       pr "\n";
3573
3574       (* Send the main header and arguments. *)
3575       (match snd style with
3576        | [] ->
3577            pr "  serial = guestfs__send_sync (g, GUESTFS_PROC_%s, NULL, NULL);\n"
3578              (String.uppercase shortname)
3579        | args ->
3580            List.iter (
3581              function
3582              | String n ->
3583                  pr "  args.%s = (char *) %s;\n" n n
3584              | OptString n ->
3585                  pr "  args.%s = %s ? (char **) &%s : NULL;\n" n n n
3586              | StringList n ->
3587                  pr "  args.%s.%s_val = (char **) %s;\n" n n n;
3588                  pr "  for (args.%s.%s_len = 0; %s[args.%s.%s_len]; args.%s.%s_len++) ;\n" n n n n n n n;
3589              | Bool n ->
3590                  pr "  args.%s = %s;\n" n n
3591              | Int n ->
3592                  pr "  args.%s = %s;\n" n n
3593              | FileIn _ | FileOut _ -> ()
3594            ) args;
3595            pr "  serial = guestfs__send_sync (g, GUESTFS_PROC_%s,\n"
3596              (String.uppercase shortname);
3597            pr "        (xdrproc_t) xdr_%s_args, (char *) &args);\n"
3598              name;
3599       );
3600       pr "  if (serial == -1) {\n";
3601       pr "    guestfs_end_busy (g);\n";
3602       pr "    return %s;\n" error_code;
3603       pr "  }\n";
3604       pr "\n";
3605
3606       (* Send any additional files (FileIn) requested. *)
3607       let need_read_reply_label = ref false in
3608       List.iter (
3609         function
3610         | FileIn n ->
3611             pr "  {\n";
3612             pr "    int r;\n";
3613             pr "\n";
3614             pr "    r = guestfs__send_file_sync (g, %s);\n" n;
3615             pr "    if (r == -1) {\n";
3616             pr "      guestfs_end_busy (g);\n";
3617             pr "      return %s;\n" error_code;
3618             pr "    }\n";
3619             pr "    if (r == -2) /* daemon cancelled */\n";
3620             pr "      goto read_reply;\n";
3621             need_read_reply_label := true;
3622             pr "  }\n";
3623             pr "\n";
3624         | _ -> ()
3625       ) (snd style);
3626
3627       (* Wait for the reply from the remote end. *)
3628       if !need_read_reply_label then pr " read_reply:\n";
3629       pr "  guestfs__switch_to_receiving (g);\n";
3630       pr "  ctx.cb_sequence = 0;\n";
3631       pr "  guestfs_set_reply_callback (g, %s_reply_cb, &ctx);\n" shortname;
3632       pr "  (void) ml->main_loop_run (ml, g);\n";
3633       pr "  guestfs_set_reply_callback (g, NULL, NULL);\n";
3634       pr "  if (ctx.cb_sequence != 1) {\n";
3635       pr "    error (g, \"%%s reply failed, see earlier error messages\", \"%s\");\n" name;
3636       pr "    guestfs_end_busy (g);\n";
3637       pr "    return %s;\n" error_code;
3638       pr "  }\n";
3639       pr "\n";
3640
3641       pr "  if (check_reply_header (g, &ctx.hdr, GUESTFS_PROC_%s, serial) == -1) {\n"
3642         (String.uppercase shortname);
3643       pr "    guestfs_end_busy (g);\n";
3644       pr "    return %s;\n" error_code;
3645       pr "  }\n";
3646       pr "\n";
3647
3648       pr "  if (ctx.hdr.status == GUESTFS_STATUS_ERROR) {\n";
3649       pr "    error (g, \"%%s\", ctx.err.error_message);\n";
3650       pr "    free (ctx.err.error_message);\n";
3651       pr "    guestfs_end_busy (g);\n";
3652       pr "    return %s;\n" error_code;
3653       pr "  }\n";
3654       pr "\n";
3655
3656       (* Expecting to receive further files (FileOut)? *)
3657       List.iter (
3658         function
3659         | FileOut n ->
3660             pr "  if (guestfs__receive_file_sync (g, %s) == -1) {\n" n;
3661             pr "    guestfs_end_busy (g);\n";
3662             pr "    return %s;\n" error_code;
3663             pr "  }\n";
3664             pr "\n";
3665         | _ -> ()
3666       ) (snd style);
3667
3668       pr "  guestfs_end_busy (g);\n";
3669
3670       (match fst style with
3671        | RErr -> pr "  return 0;\n"
3672        | RInt n | RInt64 n | RBool n ->
3673            pr "  return ctx.ret.%s;\n" n
3674        | RConstString _ ->
3675            failwithf "RConstString cannot be returned from a daemon function"
3676        | RString n ->
3677            pr "  return ctx.ret.%s; /* caller will free */\n" n
3678        | RStringList n | RHashtable n ->
3679            pr "  /* caller will free this, but we need to add a NULL entry */\n";
3680            pr "  ctx.ret.%s.%s_val =\n" n n;
3681            pr "    safe_realloc (g, ctx.ret.%s.%s_val,\n" n n;
3682            pr "                  sizeof (char *) * (ctx.ret.%s.%s_len + 1));\n"
3683              n n;
3684            pr "  ctx.ret.%s.%s_val[ctx.ret.%s.%s_len] = NULL;\n" n n n n;
3685            pr "  return ctx.ret.%s.%s_val;\n" n n
3686        | RIntBool _ ->
3687            pr "  /* caller with free this */\n";
3688            pr "  return safe_memdup (g, &ctx.ret, sizeof (ctx.ret));\n"
3689        | RPVList n | RVGList n | RLVList n
3690        | RStat n | RStatVFS n ->
3691            pr "  /* caller will free this */\n";
3692            pr "  return safe_memdup (g, &ctx.ret.%s, sizeof (ctx.ret.%s));\n" n n
3693       );
3694
3695       pr "}\n\n"
3696   ) daemon_functions
3697
3698 (* Generate daemon/actions.h. *)
3699 and generate_daemon_actions_h () =
3700   generate_header CStyle GPLv2;
3701
3702   pr "#include \"../src/guestfs_protocol.h\"\n";
3703   pr "\n";
3704
3705   List.iter (
3706     fun (name, style, _, _, _, _, _) ->
3707         generate_prototype
3708           ~single_line:true ~newline:true ~in_daemon:true ~prefix:"do_"
3709           name style;
3710   ) daemon_functions
3711
3712 (* Generate the server-side stubs. *)
3713 and generate_daemon_actions () =
3714   generate_header CStyle GPLv2;
3715
3716   pr "#include <config.h>\n";
3717   pr "\n";
3718   pr "#include <stdio.h>\n";
3719   pr "#include <stdlib.h>\n";
3720   pr "#include <string.h>\n";
3721   pr "#include <inttypes.h>\n";
3722   pr "#include <ctype.h>\n";
3723   pr "#include <rpc/types.h>\n";
3724   pr "#include <rpc/xdr.h>\n";
3725   pr "\n";
3726   pr "#include \"daemon.h\"\n";
3727   pr "#include \"../src/guestfs_protocol.h\"\n";
3728   pr "#include \"actions.h\"\n";
3729   pr "\n";
3730
3731   List.iter (
3732     fun (name, style, _, _, _, _, _) ->
3733       (* Generate server-side stubs. *)
3734       pr "static void %s_stub (XDR *xdr_in)\n" name;
3735       pr "{\n";
3736       let error_code =
3737         match fst style with
3738         | RErr | RInt _ -> pr "  int r;\n"; "-1"
3739         | RInt64 _ -> pr "  int64_t r;\n"; "-1"
3740         | RBool _ -> pr "  int r;\n"; "-1"
3741         | RConstString _ ->
3742             failwithf "RConstString cannot be returned from a daemon function"
3743         | RString _ -> pr "  char *r;\n"; "NULL"
3744         | RStringList _ | RHashtable _ -> pr "  char **r;\n"; "NULL"
3745         | RIntBool _ -> pr "  guestfs_%s_ret *r;\n" name; "NULL"
3746         | RPVList _ -> pr "  guestfs_lvm_int_pv_list *r;\n"; "NULL"
3747         | RVGList _ -> pr "  guestfs_lvm_int_vg_list *r;\n"; "NULL"
3748         | RLVList _ -> pr "  guestfs_lvm_int_lv_list *r;\n"; "NULL"
3749         | RStat _ -> pr "  guestfs_int_stat *r;\n"; "NULL"
3750         | RStatVFS _ -> pr "  guestfs_int_statvfs *r;\n"; "NULL" in
3751
3752       (match snd style with
3753        | [] -> ()
3754        | args ->
3755            pr "  struct guestfs_%s_args args;\n" name;
3756            List.iter (
3757              function
3758                (* Note we allow the string to be writable, in order to
3759                 * allow device name translation.  This is safe because
3760                 * we can modify the string (passed from RPC).
3761                 *)
3762              | String n
3763              | OptString n -> pr "  char *%s;\n" n
3764              | StringList n -> pr "  char **%s;\n" n
3765              | Bool n -> pr "  int %s;\n" n
3766              | Int n -> pr "  int %s;\n" n
3767              | FileIn _ | FileOut _ -> ()
3768            ) args
3769       );
3770       pr "\n";
3771
3772       (match snd style with
3773        | [] -> ()
3774        | args ->
3775            pr "  memset (&args, 0, sizeof args);\n";
3776            pr "\n";
3777            pr "  if (!xdr_guestfs_%s_args (xdr_in, &args)) {\n" name;
3778            pr "    reply_with_error (\"%%s: daemon failed to decode procedure arguments\", \"%s\");\n" name;
3779            pr "    return;\n";
3780            pr "  }\n";
3781            List.iter (
3782              function
3783              | String n -> pr "  %s = args.%s;\n" n n
3784              | OptString n -> pr "  %s = args.%s ? *args.%s : NULL;\n" n n n
3785              | StringList n ->
3786                  pr "  %s = realloc (args.%s.%s_val,\n" n n n;
3787                  pr "                sizeof (char *) * (args.%s.%s_len+1));\n" n n;
3788                  pr "  if (%s == NULL) {\n" n;
3789                  pr "    reply_with_perror (\"realloc\");\n";
3790                  pr "    goto done;\n";
3791                  pr "  }\n";
3792                  pr "  %s[args.%s.%s_len] = NULL;\n" n n n;
3793                  pr "  args.%s.%s_val = %s;\n" n n n;
3794              | Bool n -> pr "  %s = args.%s;\n" n n
3795              | Int n -> pr "  %s = args.%s;\n" n n
3796              | FileIn _ | FileOut _ -> ()
3797            ) args;
3798            pr "\n"
3799       );
3800
3801       (* Don't want to call the impl with any FileIn or FileOut
3802        * parameters, since these go "outside" the RPC protocol.
3803        *)
3804       let argsnofile =
3805         List.filter (function FileIn _ | FileOut _ -> false | _ -> true)
3806           (snd style) in
3807       pr "  r = do_%s " name;
3808       generate_call_args argsnofile;
3809       pr ";\n";
3810
3811       pr "  if (r == %s)\n" error_code;
3812       pr "    /* do_%s has already called reply_with_error */\n" name;
3813       pr "    goto done;\n";
3814       pr "\n";
3815
3816       (* If there are any FileOut parameters, then the impl must
3817        * send its own reply.
3818        *)
3819       let no_reply =
3820         List.exists (function FileOut _ -> true | _ -> false) (snd style) in
3821       if no_reply then
3822         pr "  /* do_%s has already sent a reply */\n" name
3823       else (
3824         match fst style with
3825         | RErr -> pr "  reply (NULL, NULL);\n"
3826         | RInt n | RInt64 n | RBool n ->
3827             pr "  struct guestfs_%s_ret ret;\n" name;
3828             pr "  ret.%s = r;\n" n;
3829             pr "  reply ((xdrproc_t) &xdr_guestfs_%s_ret, (char *) &ret);\n"
3830               name
3831         | RConstString _ ->
3832             failwithf "RConstString cannot be returned from a daemon function"
3833         | RString n ->
3834             pr "  struct guestfs_%s_ret ret;\n" name;
3835             pr "  ret.%s = r;\n" n;
3836             pr "  reply ((xdrproc_t) &xdr_guestfs_%s_ret, (char *) &ret);\n"
3837               name;
3838             pr "  free (r);\n"
3839         | RStringList n | RHashtable n ->
3840             pr "  struct guestfs_%s_ret ret;\n" name;
3841             pr "  ret.%s.%s_len = count_strings (r);\n" n n;
3842             pr "  ret.%s.%s_val = r;\n" n n;
3843             pr "  reply ((xdrproc_t) &xdr_guestfs_%s_ret, (char *) &ret);\n"
3844               name;
3845             pr "  free_strings (r);\n"
3846         | RIntBool _ ->
3847             pr "  reply ((xdrproc_t) xdr_guestfs_%s_ret, (char *) r);\n"
3848               name;
3849             pr "  xdr_free ((xdrproc_t) xdr_guestfs_%s_ret, (char *) r);\n" name
3850         | RPVList n | RVGList n | RLVList n
3851         | RStat n | RStatVFS n ->
3852             pr "  struct guestfs_%s_ret ret;\n" name;
3853             pr "  ret.%s = *r;\n" n;
3854             pr "  reply ((xdrproc_t) xdr_guestfs_%s_ret, (char *) &ret);\n"
3855               name;
3856             pr "  xdr_free ((xdrproc_t) xdr_guestfs_%s_ret, (char *) &ret);\n"
3857               name
3858       );
3859
3860       (* Free the args. *)
3861       (match snd style with
3862        | [] ->
3863            pr "done: ;\n";
3864        | _ ->
3865            pr "done:\n";
3866            pr "  xdr_free ((xdrproc_t) xdr_guestfs_%s_args, (char *) &args);\n"
3867              name
3868       );
3869
3870       pr "}\n\n";
3871   ) daemon_functions;
3872
3873   (* Dispatch function. *)
3874   pr "void dispatch_incoming_message (XDR *xdr_in)\n";
3875   pr "{\n";
3876   pr "  switch (proc_nr) {\n";
3877
3878   List.iter (
3879     fun (name, style, _, _, _, _, _) ->
3880         pr "    case GUESTFS_PROC_%s:\n" (String.uppercase name);
3881         pr "      %s_stub (xdr_in);\n" name;
3882         pr "      break;\n"
3883   ) daemon_functions;
3884
3885   pr "    default:\n";
3886   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";
3887   pr "  }\n";
3888   pr "}\n";
3889   pr "\n";
3890
3891   (* LVM columns and tokenization functions. *)
3892   (* XXX This generates crap code.  We should rethink how we
3893    * do this parsing.
3894    *)
3895   List.iter (
3896     function
3897     | typ, cols ->
3898         pr "static const char *lvm_%s_cols = \"%s\";\n"
3899           typ (String.concat "," (List.map fst cols));
3900         pr "\n";
3901
3902         pr "static int lvm_tokenize_%s (char *str, struct guestfs_lvm_int_%s *r)\n" typ typ;
3903         pr "{\n";
3904         pr "  char *tok, *p, *next;\n";
3905         pr "  int i, j;\n";
3906         pr "\n";
3907         (*
3908         pr "  fprintf (stderr, \"%%s: <<%%s>>\\n\", __func__, str);\n";
3909         pr "\n";
3910         *)
3911         pr "  if (!str) {\n";
3912         pr "    fprintf (stderr, \"%%s: failed: passed a NULL string\\n\", __func__);\n";
3913         pr "    return -1;\n";
3914         pr "  }\n";
3915         pr "  if (!*str || isspace (*str)) {\n";
3916         pr "    fprintf (stderr, \"%%s: failed: passed a empty string or one beginning with whitespace\\n\", __func__);\n";
3917         pr "    return -1;\n";
3918         pr "  }\n";
3919         pr "  tok = str;\n";
3920         List.iter (
3921           fun (name, coltype) ->
3922             pr "  if (!tok) {\n";
3923             pr "    fprintf (stderr, \"%%s: failed: string finished early, around token %%s\\n\", __func__, \"%s\");\n" name;
3924             pr "    return -1;\n";
3925             pr "  }\n";
3926             pr "  p = strchrnul (tok, ',');\n";
3927             pr "  if (*p) next = p+1; else next = NULL;\n";
3928             pr "  *p = '\\0';\n";
3929             (match coltype with
3930              | `String ->
3931                  pr "  r->%s = strdup (tok);\n" name;
3932                  pr "  if (r->%s == NULL) {\n" name;
3933                  pr "    perror (\"strdup\");\n";
3934                  pr "    return -1;\n";
3935                  pr "  }\n"
3936              | `UUID ->
3937                  pr "  for (i = j = 0; i < 32; ++j) {\n";
3938                  pr "    if (tok[j] == '\\0') {\n";
3939                  pr "      fprintf (stderr, \"%%s: failed to parse UUID from '%%s'\\n\", __func__, tok);\n";
3940                  pr "      return -1;\n";
3941                  pr "    } else if (tok[j] != '-')\n";
3942                  pr "      r->%s[i++] = tok[j];\n" name;
3943                  pr "  }\n";
3944              | `Bytes ->
3945                  pr "  if (sscanf (tok, \"%%\"SCNu64, &r->%s) != 1) {\n" name;
3946                  pr "    fprintf (stderr, \"%%s: failed to parse size '%%s' from token %%s\\n\", __func__, tok, \"%s\");\n" name;
3947                  pr "    return -1;\n";
3948                  pr "  }\n";
3949              | `Int ->
3950                  pr "  if (sscanf (tok, \"%%\"SCNi64, &r->%s) != 1) {\n" name;
3951                  pr "    fprintf (stderr, \"%%s: failed to parse int '%%s' from token %%s\\n\", __func__, tok, \"%s\");\n" name;
3952                  pr "    return -1;\n";
3953                  pr "  }\n";
3954              | `OptPercent ->
3955                  pr "  if (tok[0] == '\\0')\n";
3956                  pr "    r->%s = -1;\n" name;
3957                  pr "  else if (sscanf (tok, \"%%f\", &r->%s) != 1) {\n" name;
3958                  pr "    fprintf (stderr, \"%%s: failed to parse float '%%s' from token %%s\\n\", __func__, tok, \"%s\");\n" name;
3959                  pr "    return -1;\n";
3960                  pr "  }\n";
3961             );
3962             pr "  tok = next;\n";
3963         ) cols;
3964
3965         pr "  if (tok != NULL) {\n";
3966         pr "    fprintf (stderr, \"%%s: failed: extra tokens at end of string\\n\", __func__);\n";
3967         pr "    return -1;\n";
3968         pr "  }\n";
3969         pr "  return 0;\n";
3970         pr "}\n";
3971         pr "\n";
3972
3973         pr "guestfs_lvm_int_%s_list *\n" typ;
3974         pr "parse_command_line_%ss (void)\n" typ;
3975         pr "{\n";
3976         pr "  char *out, *err;\n";
3977         pr "  char *p, *pend;\n";
3978         pr "  int r, i;\n";
3979         pr "  guestfs_lvm_int_%s_list *ret;\n" typ;
3980         pr "  void *newp;\n";
3981         pr "\n";
3982         pr "  ret = malloc (sizeof *ret);\n";
3983         pr "  if (!ret) {\n";
3984         pr "    reply_with_perror (\"malloc\");\n";
3985         pr "    return NULL;\n";
3986         pr "  }\n";
3987         pr "\n";
3988         pr "  ret->guestfs_lvm_int_%s_list_len = 0;\n" typ;
3989         pr "  ret->guestfs_lvm_int_%s_list_val = NULL;\n" typ;
3990         pr "\n";
3991         pr "  r = command (&out, &err,\n";
3992         pr "           \"/sbin/lvm\", \"%ss\",\n" typ;
3993         pr "           \"-o\", lvm_%s_cols, \"--unbuffered\", \"--noheadings\",\n" typ;
3994         pr "           \"--nosuffix\", \"--separator\", \",\", \"--units\", \"b\", NULL);\n";
3995         pr "  if (r == -1) {\n";
3996         pr "    reply_with_error (\"%%s\", err);\n";
3997         pr "    free (out);\n";
3998         pr "    free (err);\n";
3999         pr "    free (ret);\n";
4000         pr "    return NULL;\n";
4001         pr "  }\n";
4002         pr "\n";
4003         pr "  free (err);\n";
4004         pr "\n";
4005         pr "  /* Tokenize each line of the output. */\n";
4006         pr "  p = out;\n";
4007         pr "  i = 0;\n";
4008         pr "  while (p) {\n";
4009         pr "    pend = strchr (p, '\\n');       /* Get the next line of output. */\n";
4010         pr "    if (pend) {\n";
4011         pr "      *pend = '\\0';\n";
4012         pr "      pend++;\n";
4013         pr "    }\n";
4014         pr "\n";
4015         pr "    while (*p && isspace (*p))      /* Skip any leading whitespace. */\n";
4016         pr "      p++;\n";
4017         pr "\n";
4018         pr "    if (!*p) {                      /* Empty line?  Skip it. */\n";
4019         pr "      p = pend;\n";
4020         pr "      continue;\n";
4021         pr "    }\n";
4022         pr "\n";
4023         pr "    /* Allocate some space to store this next entry. */\n";
4024         pr "    newp = realloc (ret->guestfs_lvm_int_%s_list_val,\n" typ;
4025         pr "                sizeof (guestfs_lvm_int_%s) * (i+1));\n" typ;
4026         pr "    if (newp == NULL) {\n";
4027         pr "      reply_with_perror (\"realloc\");\n";
4028         pr "      free (ret->guestfs_lvm_int_%s_list_val);\n" typ;
4029         pr "      free (ret);\n";
4030         pr "      free (out);\n";
4031         pr "      return NULL;\n";
4032         pr "    }\n";
4033         pr "    ret->guestfs_lvm_int_%s_list_val = newp;\n" typ;
4034         pr "\n";
4035         pr "    /* Tokenize the next entry. */\n";
4036         pr "    r = lvm_tokenize_%s (p, &ret->guestfs_lvm_int_%s_list_val[i]);\n" typ typ;
4037         pr "    if (r == -1) {\n";
4038         pr "      reply_with_error (\"failed to parse output of '%ss' command\");\n" typ;
4039         pr "      free (ret->guestfs_lvm_int_%s_list_val);\n" typ;
4040         pr "      free (ret);\n";
4041         pr "      free (out);\n";
4042         pr "      return NULL;\n";
4043         pr "    }\n";
4044         pr "\n";
4045         pr "    ++i;\n";
4046         pr "    p = pend;\n";
4047         pr "  }\n";
4048         pr "\n";
4049         pr "  ret->guestfs_lvm_int_%s_list_len = i;\n" typ;
4050         pr "\n";
4051         pr "  free (out);\n";
4052         pr "  return ret;\n";
4053         pr "}\n"
4054
4055   ) ["pv", pv_cols; "vg", vg_cols; "lv", lv_cols]
4056
4057 (* Generate the tests. *)
4058 and generate_tests () =
4059   generate_header CStyle GPLv2;
4060
4061   pr "\
4062 #include <stdio.h>
4063 #include <stdlib.h>
4064 #include <string.h>
4065 #include <unistd.h>
4066 #include <sys/types.h>
4067 #include <fcntl.h>
4068
4069 #include \"guestfs.h\"
4070
4071 static guestfs_h *g;
4072 static int suppress_error = 0;
4073
4074 static void print_error (guestfs_h *g, void *data, const char *msg)
4075 {
4076   if (!suppress_error)
4077     fprintf (stderr, \"%%s\\n\", msg);
4078 }
4079
4080 static void print_strings (char * const * const argv)
4081 {
4082   int argc;
4083
4084   for (argc = 0; argv[argc] != NULL; ++argc)
4085     printf (\"\\t%%s\\n\", argv[argc]);
4086 }
4087
4088 /*
4089 static void print_table (char * const * const argv)
4090 {
4091   int i;
4092
4093   for (i = 0; argv[i] != NULL; i += 2)
4094     printf (\"%%s: %%s\\n\", argv[i], argv[i+1]);
4095 }
4096 */
4097
4098 static void no_test_warnings (void)
4099 {
4100 ";
4101
4102   List.iter (
4103     function
4104     | name, _, _, _, [], _, _ ->
4105         pr "  fprintf (stderr, \"warning: \\\"guestfs_%s\\\" has no tests\\n\");\n" name
4106     | name, _, _, _, tests, _, _ -> ()
4107   ) all_functions;
4108
4109   pr "}\n";
4110   pr "\n";
4111
4112   (* Generate the actual tests.  Note that we generate the tests
4113    * in reverse order, deliberately, so that (in general) the
4114    * newest tests run first.  This makes it quicker and easier to
4115    * debug them.
4116    *)
4117   let test_names =
4118     List.map (
4119       fun (name, _, _, _, tests, _, _) ->
4120         mapi (generate_one_test name) tests
4121     ) (List.rev all_functions) in
4122   let test_names = List.concat test_names in
4123   let nr_tests = List.length test_names in
4124
4125   pr "\
4126 int main (int argc, char *argv[])
4127 {
4128   char c = 0;
4129   int failed = 0;
4130   const char *filename;
4131   int fd;
4132   int nr_tests, test_num = 0;
4133
4134   no_test_warnings ();
4135
4136   g = guestfs_create ();
4137   if (g == NULL) {
4138     printf (\"guestfs_create FAILED\\n\");
4139     exit (1);
4140   }
4141
4142   guestfs_set_error_handler (g, print_error, NULL);
4143
4144   guestfs_set_path (g, \"../appliance\");
4145
4146   filename = \"test1.img\";
4147   fd = open (filename, O_WRONLY|O_CREAT|O_NOCTTY|O_NONBLOCK|O_TRUNC, 0666);
4148   if (fd == -1) {
4149     perror (filename);
4150     exit (1);
4151   }
4152   if (lseek (fd, %d, SEEK_SET) == -1) {
4153     perror (\"lseek\");
4154     close (fd);
4155     unlink (filename);
4156     exit (1);
4157   }
4158   if (write (fd, &c, 1) == -1) {
4159     perror (\"write\");
4160     close (fd);
4161     unlink (filename);
4162     exit (1);
4163   }
4164   if (close (fd) == -1) {
4165     perror (filename);
4166     unlink (filename);
4167     exit (1);
4168   }
4169   if (guestfs_add_drive (g, filename) == -1) {
4170     printf (\"guestfs_add_drive %%s FAILED\\n\", filename);
4171     exit (1);
4172   }
4173
4174   filename = \"test2.img\";
4175   fd = open (filename, O_WRONLY|O_CREAT|O_NOCTTY|O_NONBLOCK|O_TRUNC, 0666);
4176   if (fd == -1) {
4177     perror (filename);
4178     exit (1);
4179   }
4180   if (lseek (fd, %d, SEEK_SET) == -1) {
4181     perror (\"lseek\");
4182     close (fd);
4183     unlink (filename);
4184     exit (1);
4185   }
4186   if (write (fd, &c, 1) == -1) {
4187     perror (\"write\");
4188     close (fd);
4189     unlink (filename);
4190     exit (1);
4191   }
4192   if (close (fd) == -1) {
4193     perror (filename);
4194     unlink (filename);
4195     exit (1);
4196   }
4197   if (guestfs_add_drive (g, filename) == -1) {
4198     printf (\"guestfs_add_drive %%s FAILED\\n\", filename);
4199     exit (1);
4200   }
4201
4202   filename = \"test3.img\";
4203   fd = open (filename, O_WRONLY|O_CREAT|O_NOCTTY|O_NONBLOCK|O_TRUNC, 0666);
4204   if (fd == -1) {
4205     perror (filename);
4206     exit (1);
4207   }
4208   if (lseek (fd, %d, SEEK_SET) == -1) {
4209     perror (\"lseek\");
4210     close (fd);
4211     unlink (filename);
4212     exit (1);
4213   }
4214   if (write (fd, &c, 1) == -1) {
4215     perror (\"write\");
4216     close (fd);
4217     unlink (filename);
4218     exit (1);
4219   }
4220   if (close (fd) == -1) {
4221     perror (filename);
4222     unlink (filename);
4223     exit (1);
4224   }
4225   if (guestfs_add_drive (g, filename) == -1) {
4226     printf (\"guestfs_add_drive %%s FAILED\\n\", filename);
4227     exit (1);
4228   }
4229
4230   if (guestfs_add_drive_ro (g, \"../images/test.sqsh\") == -1) {
4231     printf (\"guestfs_add_drive_ro ../images/test.sqsh FAILED\\n\");
4232     exit (1);
4233   }
4234
4235   if (guestfs_launch (g) == -1) {
4236     printf (\"guestfs_launch FAILED\\n\");
4237     exit (1);
4238   }
4239
4240   /* Set a timeout in case qemu hangs during launch (RHBZ#505329). */
4241   alarm (600);
4242
4243   if (guestfs_wait_ready (g) == -1) {
4244     printf (\"guestfs_wait_ready FAILED\\n\");
4245     exit (1);
4246   }
4247
4248   /* Cancel previous alarm. */
4249   alarm (0);
4250
4251   nr_tests = %d;
4252
4253 " (500 * 1024 * 1024) (50 * 1024 * 1024) (10 * 1024 * 1024) nr_tests;
4254
4255   iteri (
4256     fun i test_name ->
4257       pr "  test_num++;\n";
4258       pr "  printf (\"%%3d/%%3d %s\\n\", test_num, nr_tests);\n" test_name;
4259       pr "  if (%s () == -1) {\n" test_name;
4260       pr "    printf (\"%s FAILED\\n\");\n" test_name;
4261       pr "    failed++;\n";
4262       pr "  }\n";
4263   ) test_names;
4264   pr "\n";
4265
4266   pr "  guestfs_close (g);\n";
4267   pr "  unlink (\"test1.img\");\n";
4268   pr "  unlink (\"test2.img\");\n";
4269   pr "  unlink (\"test3.img\");\n";
4270   pr "\n";
4271
4272   pr "  if (failed > 0) {\n";
4273   pr "    printf (\"***** %%d / %%d tests FAILED *****\\n\", failed, nr_tests);\n";
4274   pr "    exit (1);\n";
4275   pr "  }\n";
4276   pr "\n";
4277
4278   pr "  exit (0);\n";
4279   pr "}\n"
4280
4281 and generate_one_test name i (init, prereq, test) =
4282   let test_name = sprintf "test_%s_%d" name i in
4283
4284   pr "\
4285 static int %s_skip (void)
4286 {
4287   const char *str;
4288
4289   str = getenv (\"TEST_ONLY\");
4290   if (str)
4291     return strstr (str, \"%s\") == NULL;
4292   str = getenv (\"SKIP_%s\");
4293   if (str && strcmp (str, \"1\") == 0) return 1;
4294   str = getenv (\"SKIP_TEST_%s\");
4295   if (str && strcmp (str, \"1\") == 0) return 1;
4296   return 0;
4297 }
4298
4299 " test_name name (String.uppercase test_name) (String.uppercase name);
4300
4301   (match prereq with
4302    | Disabled | Always -> ()
4303    | If code | Unless code ->
4304        pr "static int %s_prereq (void)\n" test_name;
4305        pr "{\n";
4306        pr "  %s\n" code;
4307        pr "}\n";
4308        pr "\n";
4309   );
4310
4311   pr "\
4312 static int %s (void)
4313 {
4314   if (%s_skip ()) {
4315     printf (\"%%s skipped (reason: environment variable set)\\n\", \"%s\");
4316     return 0;
4317   }
4318
4319 " test_name test_name test_name;
4320
4321   (match prereq with
4322    | Disabled ->
4323        pr "  printf (\"%%s skipped (reason: test disabled in generator)\\n\", \"%s\");\n" test_name
4324    | If _ ->
4325        pr "  if (! %s_prereq ()) {\n" test_name;
4326        pr "    printf (\"%%s skipped (reason: test prerequisite)\\n\", \"%s\");\n" test_name;
4327        pr "    return 0;\n";
4328        pr "  }\n";
4329        pr "\n";
4330        generate_one_test_body name i test_name init test;
4331    | Unless _ ->
4332        pr "  if (%s_prereq ()) {\n" test_name;
4333        pr "    printf (\"%%s skipped (reason: test prerequisite)\\n\", \"%s\");\n" test_name;
4334        pr "    return 0;\n";
4335        pr "  }\n";
4336        pr "\n";
4337        generate_one_test_body name i test_name init test;
4338    | Always ->
4339        generate_one_test_body name i test_name init test
4340   );
4341
4342   pr "  return 0;\n";
4343   pr "}\n";
4344   pr "\n";
4345   test_name
4346
4347 and generate_one_test_body name i test_name init test =
4348   (match init with
4349    | InitNone
4350    | InitEmpty ->
4351        pr "  /* InitNone|InitEmpty for %s */\n" test_name;
4352        List.iter (generate_test_command_call test_name)
4353          [["blockdev_setrw"; "/dev/sda"];
4354           ["umount_all"];
4355           ["lvm_remove_all"]]
4356    | InitBasicFS ->
4357        pr "  /* InitBasicFS for %s: create ext2 on /dev/sda1 */\n" test_name;
4358        List.iter (generate_test_command_call test_name)
4359          [["blockdev_setrw"; "/dev/sda"];
4360           ["umount_all"];
4361           ["lvm_remove_all"];
4362           ["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
4363           ["mkfs"; "ext2"; "/dev/sda1"];
4364           ["mount"; "/dev/sda1"; "/"]]
4365    | InitBasicFSonLVM ->
4366        pr "  /* InitBasicFSonLVM for %s: create ext2 on /dev/VG/LV */\n"
4367          test_name;
4368        List.iter (generate_test_command_call test_name)
4369          [["blockdev_setrw"; "/dev/sda"];
4370           ["umount_all"];
4371           ["lvm_remove_all"];
4372           ["sfdisk"; "/dev/sda"; "0"; "0"; "0"; ","];
4373           ["pvcreate"; "/dev/sda1"];
4374           ["vgcreate"; "VG"; "/dev/sda1"];
4375           ["lvcreate"; "LV"; "VG"; "8"];
4376           ["mkfs"; "ext2"; "/dev/VG/LV"];
4377           ["mount"; "/dev/VG/LV"; "/"]]
4378   );
4379
4380   let get_seq_last = function
4381     | [] ->
4382         failwithf "%s: you cannot use [] (empty list) when expecting a command"
4383           test_name
4384     | seq ->
4385         let seq = List.rev seq in
4386         List.rev (List.tl seq), List.hd seq
4387   in
4388
4389   match test with
4390   | TestRun seq ->
4391       pr "  /* TestRun for %s (%d) */\n" name i;
4392       List.iter (generate_test_command_call test_name) seq
4393   | TestOutput (seq, expected) ->
4394       pr "  /* TestOutput for %s (%d) */\n" name i;
4395       pr "  char expected[] = \"%s\";\n" (c_quote expected);
4396       let seq, last = get_seq_last seq in
4397       let test () =
4398         pr "    if (strcmp (r, expected) != 0) {\n";
4399         pr "      fprintf (stderr, \"%s: expected \\\"%%s\\\" but got \\\"%%s\\\"\\n\", expected, r);\n" test_name;
4400         pr "      return -1;\n";
4401         pr "    }\n"
4402       in
4403       List.iter (generate_test_command_call test_name) seq;
4404       generate_test_command_call ~test test_name last
4405   | TestOutputList (seq, expected) ->
4406       pr "  /* TestOutputList for %s (%d) */\n" name i;
4407       let seq, last = get_seq_last seq in
4408       let test () =
4409         iteri (
4410           fun i str ->
4411             pr "    if (!r[%d]) {\n" i;
4412             pr "      fprintf (stderr, \"%s: short list returned from command\\n\");\n" test_name;
4413             pr "      print_strings (r);\n";
4414             pr "      return -1;\n";
4415             pr "    }\n";
4416             pr "    {\n";
4417             pr "      char expected[] = \"%s\";\n" (c_quote str);
4418             pr "      if (strcmp (r[%d], expected) != 0) {\n" i;
4419             pr "        fprintf (stderr, \"%s: expected \\\"%%s\\\" but got \\\"%%s\\\"\\n\", expected, r[%d]);\n" test_name i;
4420             pr "        return -1;\n";
4421             pr "      }\n";
4422             pr "    }\n"
4423         ) expected;
4424         pr "    if (r[%d] != NULL) {\n" (List.length expected);
4425         pr "      fprintf (stderr, \"%s: extra elements returned from command\\n\");\n"
4426           test_name;
4427         pr "      print_strings (r);\n";
4428         pr "      return -1;\n";
4429         pr "    }\n"
4430       in
4431       List.iter (generate_test_command_call test_name) seq;
4432       generate_test_command_call ~test test_name last
4433   | TestOutputListOfDevices (seq, expected) ->
4434       pr "  /* TestOutputListOfDevices for %s (%d) */\n" name i;
4435       let seq, last = get_seq_last seq in
4436       let test () =
4437         iteri (
4438           fun i str ->
4439             pr "    if (!r[%d]) {\n" i;
4440             pr "      fprintf (stderr, \"%s: short list returned from command\\n\");\n" test_name;
4441             pr "      print_strings (r);\n";
4442             pr "      return -1;\n";
4443             pr "    }\n";
4444             pr "    {\n";
4445             pr "      char expected[] = \"%s\";\n" (c_quote str);
4446             pr "      r[%d][5] = 's';\n" i;
4447             pr "      if (strcmp (r[%d], expected) != 0) {\n" i;
4448             pr "        fprintf (stderr, \"%s: expected \\\"%%s\\\" but got \\\"%%s\\\"\\n\", expected, r[%d]);\n" test_name i;
4449             pr "        return -1;\n";
4450             pr "      }\n";
4451             pr "    }\n"
4452         ) expected;
4453         pr "    if (r[%d] != NULL) {\n" (List.length expected);
4454         pr "      fprintf (stderr, \"%s: extra elements returned from command\\n\");\n"
4455           test_name;
4456         pr "      print_strings (r);\n";
4457         pr "      return -1;\n";
4458         pr "    }\n"
4459       in
4460       List.iter (generate_test_command_call test_name) seq;
4461       generate_test_command_call ~test test_name last
4462   | TestOutputInt (seq, expected) ->
4463       pr "  /* TestOutputInt for %s (%d) */\n" name i;
4464       let seq, last = get_seq_last seq in
4465       let test () =
4466         pr "    if (r != %d) {\n" expected;
4467         pr "      fprintf (stderr, \"%s: expected %d but got %%d\\n\","
4468           test_name expected;
4469         pr "               (int) r);\n";
4470         pr "      return -1;\n";
4471         pr "    }\n"
4472       in
4473       List.iter (generate_test_command_call test_name) seq;
4474       generate_test_command_call ~test test_name last
4475   | TestOutputTrue seq ->
4476       pr "  /* TestOutputTrue for %s (%d) */\n" name i;
4477       let seq, last = get_seq_last seq in
4478       let test () =
4479         pr "    if (!r) {\n";
4480         pr "      fprintf (stderr, \"%s: expected true, got false\\n\");\n"
4481           test_name;
4482         pr "      return -1;\n";
4483         pr "    }\n"
4484       in
4485       List.iter (generate_test_command_call test_name) seq;
4486       generate_test_command_call ~test test_name last
4487   | TestOutputFalse seq ->
4488       pr "  /* TestOutputFalse for %s (%d) */\n" name i;
4489       let seq, last = get_seq_last seq in
4490       let test () =
4491         pr "    if (r) {\n";
4492         pr "      fprintf (stderr, \"%s: expected false, got true\\n\");\n"
4493           test_name;
4494         pr "      return -1;\n";
4495         pr "    }\n"
4496       in
4497       List.iter (generate_test_command_call test_name) seq;
4498       generate_test_command_call ~test test_name last
4499   | TestOutputLength (seq, expected) ->
4500       pr "  /* TestOutputLength for %s (%d) */\n" name i;
4501       let seq, last = get_seq_last seq in
4502       let test () =
4503         pr "    int j;\n";
4504         pr "    for (j = 0; j < %d; ++j)\n" expected;
4505         pr "      if (r[j] == NULL) {\n";
4506         pr "        fprintf (stderr, \"%s: short list returned\\n\");\n"
4507           test_name;
4508         pr "        print_strings (r);\n";
4509         pr "        return -1;\n";
4510         pr "      }\n";
4511         pr "    if (r[j] != NULL) {\n";
4512         pr "      fprintf (stderr, \"%s: long list returned\\n\");\n"
4513           test_name;
4514         pr "      print_strings (r);\n";
4515         pr "      return -1;\n";
4516         pr "    }\n"
4517       in
4518       List.iter (generate_test_command_call test_name) seq;
4519       generate_test_command_call ~test test_name last
4520   | TestOutputStruct (seq, checks) ->
4521       pr "  /* TestOutputStruct for %s (%d) */\n" name i;
4522       let seq, last = get_seq_last seq in
4523       let test () =
4524         List.iter (
4525           function
4526           | CompareWithInt (field, expected) ->
4527               pr "    if (r->%s != %d) {\n" field expected;
4528               pr "      fprintf (stderr, \"%s: %s was %%d, expected %d\\n\",\n"
4529                 test_name field expected;
4530               pr "               (int) r->%s);\n" field;
4531               pr "      return -1;\n";
4532               pr "    }\n"
4533           | CompareWithString (field, expected) ->
4534               pr "    if (strcmp (r->%s, \"%s\") != 0) {\n" field expected;
4535               pr "      fprintf (stderr, \"%s: %s was \"%%s\", expected \"%s\"\\n\",\n"
4536                 test_name field expected;
4537               pr "               r->%s);\n" field;
4538               pr "      return -1;\n";
4539               pr "    }\n"
4540           | CompareFieldsIntEq (field1, field2) ->
4541               pr "    if (r->%s != r->%s) {\n" field1 field2;
4542               pr "      fprintf (stderr, \"%s: %s (%%d) <> %s (%%d)\\n\",\n"
4543                 test_name field1 field2;
4544               pr "               (int) r->%s, (int) r->%s);\n" field1 field2;
4545               pr "      return -1;\n";
4546               pr "    }\n"
4547           | CompareFieldsStrEq (field1, field2) ->
4548               pr "    if (strcmp (r->%s, r->%s) != 0) {\n" field1 field2;
4549               pr "      fprintf (stderr, \"%s: %s (\"%%s\") <> %s (\"%%s\")\\n\",\n"
4550                 test_name field1 field2;
4551               pr "               r->%s, r->%s);\n" field1 field2;
4552               pr "      return -1;\n";
4553               pr "    }\n"
4554         ) checks
4555       in
4556       List.iter (generate_test_command_call test_name) seq;
4557       generate_test_command_call ~test test_name last
4558   | TestLastFail seq ->
4559       pr "  /* TestLastFail for %s (%d) */\n" name i;
4560       let seq, last = get_seq_last seq in
4561       List.iter (generate_test_command_call test_name) seq;
4562       generate_test_command_call test_name ~expect_error:true last
4563
4564 (* Generate the code to run a command, leaving the result in 'r'.
4565  * If you expect to get an error then you should set expect_error:true.
4566  *)
4567 and generate_test_command_call ?(expect_error = false) ?test test_name cmd =
4568   match cmd with
4569   | [] -> assert false
4570   | name :: args ->
4571       (* Look up the command to find out what args/ret it has. *)
4572       let style =
4573         try
4574           let _, style, _, _, _, _, _ =
4575             List.find (fun (n, _, _, _, _, _, _) -> n = name) all_functions in
4576           style
4577         with Not_found ->
4578           failwithf "%s: in test, command %s was not found" test_name name in
4579
4580       if List.length (snd style) <> List.length args then
4581         failwithf "%s: in test, wrong number of args given to %s"
4582           test_name name;
4583
4584       pr "  {\n";
4585
4586       List.iter (
4587         function
4588         | OptString n, "NULL" -> ()
4589         | String n, arg
4590         | OptString n, arg ->
4591             pr "    char %s[] = \"%s\";\n" n (c_quote arg);
4592         | Int _, _
4593         | Bool _, _
4594         | FileIn _, _ | FileOut _, _ -> ()
4595         | StringList n, arg ->
4596             let strs = string_split " " arg in
4597             iteri (
4598               fun i str ->
4599                 pr "    char %s_%d[] = \"%s\";\n" n i (c_quote str);
4600             ) strs;
4601             pr "    char *%s[] = {\n" n;
4602             iteri (
4603               fun i _ -> pr "      %s_%d,\n" n i
4604             ) strs;
4605             pr "      NULL\n";
4606             pr "    };\n";
4607       ) (List.combine (snd style) args);
4608
4609       let error_code =
4610         match fst style with
4611         | RErr | RInt _ | RBool _ -> pr "    int r;\n"; "-1"
4612         | RInt64 _ -> pr "    int64_t r;\n"; "-1"
4613         | RConstString _ -> pr "    const char *r;\n"; "NULL"
4614         | RString _ -> pr "    char *r;\n"; "NULL"
4615         | RStringList _ | RHashtable _ ->
4616             pr "    char **r;\n";
4617             pr "    int i;\n";
4618             "NULL"
4619         | RIntBool _ ->
4620             pr "    struct guestfs_int_bool *r;\n"; "NULL"
4621         | RPVList _ ->
4622             pr "    struct guestfs_lvm_pv_list *r;\n"; "NULL"
4623         | RVGList _ ->
4624             pr "    struct guestfs_lvm_vg_list *r;\n"; "NULL"
4625         | RLVList _ ->
4626             pr "    struct guestfs_lvm_lv_list *r;\n"; "NULL"
4627         | RStat _ ->
4628             pr "    struct guestfs_stat *r;\n"; "NULL"
4629         | RStatVFS _ ->
4630             pr "    struct guestfs_statvfs *r;\n"; "NULL" in
4631
4632       pr "    suppress_error = %d;\n" (if expect_error then 1 else 0);
4633       pr "    r = guestfs_%s (g" name;
4634
4635       (* Generate the parameters. *)
4636       List.iter (
4637         function
4638         | OptString _, "NULL" -> pr ", NULL"
4639         | String n, _
4640         | OptString n, _ ->
4641             pr ", %s" n
4642         | FileIn _, arg | FileOut _, arg ->
4643             pr ", \"%s\"" (c_quote arg)
4644         | StringList n, _ ->
4645             pr ", %s" n
4646         | Int _, arg ->
4647             let i =
4648               try int_of_string arg
4649               with Failure "int_of_string" ->
4650                 failwithf "%s: expecting an int, but got '%s'" test_name arg in
4651             pr ", %d" i
4652         | Bool _, arg ->
4653             let b = bool_of_string arg in pr ", %d" (if b then 1 else 0)
4654       ) (List.combine (snd style) args);
4655
4656       pr ");\n";
4657       if not expect_error then
4658         pr "    if (r == %s)\n" error_code
4659       else
4660         pr "    if (r != %s)\n" error_code;
4661       pr "      return -1;\n";
4662
4663       (* Insert the test code. *)
4664       (match test with
4665        | None -> ()
4666        | Some f -> f ()
4667       );
4668
4669       (match fst style with
4670        | RErr | RInt _ | RInt64 _ | RBool _ | RConstString _ -> ()
4671        | RString _ -> pr "    free (r);\n"
4672        | RStringList _ | RHashtable _ ->
4673            pr "    for (i = 0; r[i] != NULL; ++i)\n";
4674            pr "      free (r[i]);\n";
4675            pr "    free (r);\n"
4676        | RIntBool _ ->
4677            pr "    guestfs_free_int_bool (r);\n"
4678        | RPVList _ ->
4679            pr "    guestfs_free_lvm_pv_list (r);\n"
4680        | RVGList _ ->
4681            pr "    guestfs_free_lvm_vg_list (r);\n"
4682        | RLVList _ ->
4683            pr "    guestfs_free_lvm_lv_list (r);\n"
4684        | RStat _ | RStatVFS _ ->
4685            pr "    free (r);\n"
4686       );
4687
4688       pr "  }\n"
4689
4690 and c_quote str =
4691   let str = replace_str str "\r" "\\r" in
4692   let str = replace_str str "\n" "\\n" in
4693   let str = replace_str str "\t" "\\t" in
4694   let str = replace_str str "\000" "\\0" in
4695   str
4696
4697 (* Generate a lot of different functions for guestfish. *)
4698 and generate_fish_cmds () =
4699   generate_header CStyle GPLv2;
4700
4701   let all_functions =
4702     List.filter (
4703       fun (_, _, _, flags, _, _, _) -> not (List.mem NotInFish flags)
4704     ) all_functions in
4705   let all_functions_sorted =
4706     List.filter (
4707       fun (_, _, _, flags, _, _, _) -> not (List.mem NotInFish flags)
4708     ) all_functions_sorted in
4709
4710   pr "#include <stdio.h>\n";
4711   pr "#include <stdlib.h>\n";
4712   pr "#include <string.h>\n";
4713   pr "#include <inttypes.h>\n";
4714   pr "\n";
4715   pr "#include <guestfs.h>\n";
4716   pr "#include \"fish.h\"\n";
4717   pr "\n";
4718
4719   (* list_commands function, which implements guestfish -h *)
4720   pr "void list_commands (void)\n";
4721   pr "{\n";
4722   pr "  printf (\"    %%-16s     %%s\\n\", \"Command\", \"Description\");\n";
4723   pr "  list_builtin_commands ();\n";
4724   List.iter (
4725     fun (name, _, _, flags, _, shortdesc, _) ->
4726       let name = replace_char name '_' '-' in
4727       pr "  printf (\"%%-20s %%s\\n\", \"%s\", \"%s\");\n"
4728         name shortdesc
4729   ) all_functions_sorted;
4730   pr "  printf (\"    Use -h <cmd> / help <cmd> to show detailed help for a command.\\n\");\n";
4731   pr "}\n";
4732   pr "\n";
4733
4734   (* display_command function, which implements guestfish -h cmd *)
4735   pr "void display_command (const char *cmd)\n";
4736   pr "{\n";
4737   List.iter (
4738     fun (name, style, _, flags, _, shortdesc, longdesc) ->
4739       let name2 = replace_char name '_' '-' in
4740       let alias =
4741         try find_map (function FishAlias n -> Some n | _ -> None) flags
4742         with Not_found -> name in
4743       let longdesc = replace_str longdesc "C<guestfs_" "C<" in
4744       let synopsis =
4745         match snd style with
4746         | [] -> name2
4747         | args ->
4748             sprintf "%s <%s>"
4749               name2 (String.concat "> <" (List.map name_of_argt args)) in
4750
4751       let warnings =
4752         if List.mem ProtocolLimitWarning flags then
4753           ("\n\n" ^ protocol_limit_warning)
4754         else "" in
4755
4756       (* For DangerWillRobinson commands, we should probably have
4757        * guestfish prompt before allowing you to use them (especially
4758        * in interactive mode). XXX
4759        *)
4760       let warnings =
4761         warnings ^
4762           if List.mem DangerWillRobinson flags then
4763             ("\n\n" ^ danger_will_robinson)
4764           else "" in
4765
4766       let describe_alias =
4767         if name <> alias then
4768           sprintf "\n\nYou can use '%s' as an alias for this command." alias
4769         else "" in
4770
4771       pr "  if (";
4772       pr "strcasecmp (cmd, \"%s\") == 0" name;
4773       if name <> name2 then
4774         pr " || strcasecmp (cmd, \"%s\") == 0" name2;
4775       if name <> alias then
4776         pr " || strcasecmp (cmd, \"%s\") == 0" alias;
4777       pr ")\n";
4778       pr "    pod2text (\"%s - %s\", %S);\n"
4779         name2 shortdesc
4780         (" " ^ synopsis ^ "\n\n" ^ longdesc ^ warnings ^ describe_alias);
4781       pr "  else\n"
4782   ) all_functions;
4783   pr "    display_builtin_command (cmd);\n";
4784   pr "}\n";
4785   pr "\n";
4786
4787   (* print_{pv,vg,lv}_list functions *)
4788   List.iter (
4789     function
4790     | typ, cols ->
4791         pr "static void print_%s (struct guestfs_lvm_%s *%s)\n" typ typ typ;
4792         pr "{\n";
4793         pr "  int i;\n";
4794         pr "\n";
4795         List.iter (
4796           function
4797           | name, `String ->
4798               pr "  printf (\"%s: %%s\\n\", %s->%s);\n" name typ name
4799           | name, `UUID ->
4800               pr "  printf (\"%s: \");\n" name;
4801               pr "  for (i = 0; i < 32; ++i)\n";
4802               pr "    printf (\"%%c\", %s->%s[i]);\n" typ name;
4803               pr "  printf (\"\\n\");\n"
4804           | name, `Bytes ->
4805               pr "  printf (\"%s: %%\" PRIu64 \"\\n\", %s->%s);\n" name typ name
4806           | name, `Int ->
4807               pr "  printf (\"%s: %%\" PRIi64 \"\\n\", %s->%s);\n" name typ name
4808           | name, `OptPercent ->
4809               pr "  if (%s->%s >= 0) printf (\"%s: %%g %%%%\\n\", %s->%s);\n"
4810                 typ name name typ name;
4811               pr "  else printf (\"%s: \\n\");\n" name
4812         ) cols;
4813         pr "}\n";
4814         pr "\n";
4815         pr "static void print_%s_list (struct guestfs_lvm_%s_list *%ss)\n"
4816           typ typ typ;
4817         pr "{\n";
4818         pr "  int i;\n";
4819         pr "\n";
4820         pr "  for (i = 0; i < %ss->len; ++i)\n" typ;
4821         pr "    print_%s (&%ss->val[i]);\n" typ typ;
4822         pr "}\n";
4823         pr "\n";
4824   ) ["pv", pv_cols; "vg", vg_cols; "lv", lv_cols];
4825
4826   (* print_{stat,statvfs} functions *)
4827   List.iter (
4828     function
4829     | typ, cols ->
4830         pr "static void print_%s (struct guestfs_%s *%s)\n" typ typ typ;
4831         pr "{\n";
4832         List.iter (
4833           function
4834           | name, `Int ->
4835               pr "  printf (\"%s: %%\" PRIi64 \"\\n\", %s->%s);\n" name typ name
4836         ) cols;
4837         pr "}\n";
4838         pr "\n";
4839   ) ["stat", stat_cols; "statvfs", statvfs_cols];
4840
4841   (* run_<action> actions *)
4842   List.iter (
4843     fun (name, style, _, flags, _, _, _) ->
4844       pr "static int run_%s (const char *cmd, int argc, char *argv[])\n" name;
4845       pr "{\n";
4846       (match fst style with
4847        | RErr
4848        | RInt _
4849        | RBool _ -> pr "  int r;\n"
4850        | RInt64 _ -> pr "  int64_t r;\n"
4851        | RConstString _ -> pr "  const char *r;\n"
4852        | RString _ -> pr "  char *r;\n"
4853        | RStringList _ | RHashtable _ -> pr "  char **r;\n"
4854        | RIntBool _ -> pr "  struct guestfs_int_bool *r;\n"
4855        | RPVList _ -> pr "  struct guestfs_lvm_pv_list *r;\n"
4856        | RVGList _ -> pr "  struct guestfs_lvm_vg_list *r;\n"
4857        | RLVList _ -> pr "  struct guestfs_lvm_lv_list *r;\n"
4858        | RStat _ -> pr "  struct guestfs_stat *r;\n"
4859        | RStatVFS _ -> pr "  struct guestfs_statvfs *r;\n"
4860       );
4861       List.iter (
4862         function
4863         | String n
4864         | OptString n
4865         | FileIn n
4866         | FileOut n -> pr "  const char *%s;\n" n
4867         | StringList n -> pr "  char **%s;\n" n
4868         | Bool n -> pr "  int %s;\n" n
4869         | Int n -> pr "  int %s;\n" n
4870       ) (snd style);
4871
4872       (* Check and convert parameters. *)
4873       let argc_expected = List.length (snd style) in
4874       pr "  if (argc != %d) {\n" argc_expected;
4875       pr "    fprintf (stderr, \"%%s should have %d parameter(s)\\n\", cmd);\n"
4876         argc_expected;
4877       pr "    fprintf (stderr, \"type 'help %%s' for help on %%s\\n\", cmd, cmd);\n";
4878       pr "    return -1;\n";
4879       pr "  }\n";
4880       iteri (
4881         fun i ->
4882           function
4883           | String name -> pr "  %s = argv[%d];\n" name i
4884           | OptString name ->
4885               pr "  %s = strcmp (argv[%d], \"\") != 0 ? argv[%d] : NULL;\n"
4886                 name i i
4887           | FileIn name ->
4888               pr "  %s = strcmp (argv[%d], \"-\") != 0 ? argv[%d] : \"/dev/stdin\";\n"
4889                 name i i
4890           | FileOut name ->
4891               pr "  %s = strcmp (argv[%d], \"-\") != 0 ? argv[%d] : \"/dev/stdout\";\n"
4892                 name i i
4893           | StringList name ->
4894               pr "  %s = parse_string_list (argv[%d]);\n" name i
4895           | Bool name ->
4896               pr "  %s = is_true (argv[%d]) ? 1 : 0;\n" name i
4897           | Int name ->
4898               pr "  %s = atoi (argv[%d]);\n" name i
4899       ) (snd style);
4900
4901       (* Call C API function. *)
4902       let fn =
4903         try find_map (function FishAction n -> Some n | _ -> None) flags
4904         with Not_found -> sprintf "guestfs_%s" name in
4905       pr "  r = %s " fn;
4906       generate_call_args ~handle:"g" (snd style);
4907       pr ";\n";
4908
4909       (* Check return value for errors and display command results. *)
4910       (match fst style with
4911        | RErr -> pr "  return r;\n"
4912        | RInt _ ->
4913            pr "  if (r == -1) return -1;\n";
4914            pr "  printf (\"%%d\\n\", r);\n";
4915            pr "  return 0;\n"
4916        | RInt64 _ ->
4917            pr "  if (r == -1) return -1;\n";
4918            pr "  printf (\"%%\" PRIi64 \"\\n\", r);\n";
4919            pr "  return 0;\n"
4920        | RBool _ ->
4921            pr "  if (r == -1) return -1;\n";
4922            pr "  if (r) printf (\"true\\n\"); else printf (\"false\\n\");\n";
4923            pr "  return 0;\n"
4924        | RConstString _ ->
4925            pr "  if (r == NULL) return -1;\n";
4926            pr "  printf (\"%%s\\n\", r);\n";
4927            pr "  return 0;\n"
4928        | RString _ ->
4929            pr "  if (r == NULL) return -1;\n";
4930            pr "  printf (\"%%s\\n\", r);\n";
4931            pr "  free (r);\n";
4932            pr "  return 0;\n"
4933        | RStringList _ ->
4934            pr "  if (r == NULL) return -1;\n";
4935            pr "  print_strings (r);\n";
4936            pr "  free_strings (r);\n";
4937            pr "  return 0;\n"
4938        | RIntBool _ ->
4939            pr "  if (r == NULL) return -1;\n";
4940            pr "  printf (\"%%d, %%s\\n\", r->i,\n";
4941            pr "    r->b ? \"true\" : \"false\");\n";
4942            pr "  guestfs_free_int_bool (r);\n";
4943            pr "  return 0;\n"
4944        | RPVList _ ->
4945            pr "  if (r == NULL) return -1;\n";
4946            pr "  print_pv_list (r);\n";
4947            pr "  guestfs_free_lvm_pv_list (r);\n";
4948            pr "  return 0;\n"
4949        | RVGList _ ->
4950            pr "  if (r == NULL) return -1;\n";
4951            pr "  print_vg_list (r);\n";
4952            pr "  guestfs_free_lvm_vg_list (r);\n";
4953            pr "  return 0;\n"
4954        | RLVList _ ->
4955            pr "  if (r == NULL) return -1;\n";
4956            pr "  print_lv_list (r);\n";
4957            pr "  guestfs_free_lvm_lv_list (r);\n";
4958            pr "  return 0;\n"
4959        | RStat _ ->
4960            pr "  if (r == NULL) return -1;\n";
4961            pr "  print_stat (r);\n";
4962            pr "  free (r);\n";
4963            pr "  return 0;\n"
4964        | RStatVFS _ ->
4965            pr "  if (r == NULL) return -1;\n";
4966            pr "  print_statvfs (r);\n";
4967            pr "  free (r);\n";
4968            pr "  return 0;\n"
4969        | RHashtable _ ->
4970            pr "  if (r == NULL) return -1;\n";
4971            pr "  print_table (r);\n";
4972            pr "  free_strings (r);\n";
4973            pr "  return 0;\n"
4974       );
4975       pr "}\n";
4976       pr "\n"
4977   ) all_functions;
4978
4979   (* run_action function *)
4980   pr "int run_action (const char *cmd, int argc, char *argv[])\n";
4981   pr "{\n";
4982   List.iter (
4983     fun (name, _, _, flags, _, _, _) ->
4984       let name2 = replace_char name '_' '-' in
4985       let alias =
4986         try find_map (function FishAlias n -> Some n | _ -> None) flags
4987         with Not_found -> name in
4988       pr "  if (";
4989       pr "strcasecmp (cmd, \"%s\") == 0" name;
4990       if name <> name2 then
4991         pr " || strcasecmp (cmd, \"%s\") == 0" name2;
4992       if name <> alias then
4993         pr " || strcasecmp (cmd, \"%s\") == 0" alias;
4994       pr ")\n";
4995       pr "    return run_%s (cmd, argc, argv);\n" name;
4996       pr "  else\n";
4997   ) all_functions;
4998   pr "    {\n";
4999   pr "      fprintf (stderr, \"%%s: unknown command\\n\", cmd);\n";
5000   pr "      return -1;\n";
5001   pr "    }\n";
5002   pr "  return 0;\n";
5003   pr "}\n";
5004   pr "\n"
5005
5006 (* Readline completion for guestfish. *)
5007 and generate_fish_completion () =
5008   generate_header CStyle GPLv2;
5009
5010   let all_functions =
5011     List.filter (
5012       fun (_, _, _, flags, _, _, _) -> not (List.mem NotInFish flags)
5013     ) all_functions in
5014
5015   pr "\
5016 #include <config.h>
5017
5018 #include <stdio.h>
5019 #include <stdlib.h>
5020 #include <string.h>
5021
5022 #ifdef HAVE_LIBREADLINE
5023 #include <readline/readline.h>
5024 #endif
5025
5026 #include \"fish.h\"
5027
5028 #ifdef HAVE_LIBREADLINE
5029
5030 static const char *const commands[] = {
5031   BUILTIN_COMMANDS_FOR_COMPLETION,
5032 ";
5033
5034   (* Get the commands, including the aliases.  They don't need to be
5035    * sorted - the generator() function just does a dumb linear search.
5036    *)
5037   let commands =
5038     List.map (
5039       fun (name, _, _, flags, _, _, _) ->
5040         let name2 = replace_char name '_' '-' in
5041         let alias =
5042           try find_map (function FishAlias n -> Some n | _ -> None) flags
5043           with Not_found -> name in
5044
5045         if name <> alias then [name2; alias] else [name2]
5046     ) all_functions in
5047   let commands = List.flatten commands in
5048
5049   List.iter (pr "  \"%s\",\n") commands;
5050
5051   pr "  NULL
5052 };
5053
5054 static char *
5055 generator (const char *text, int state)
5056 {
5057   static int index, len;
5058   const char *name;
5059
5060   if (!state) {
5061     index = 0;
5062     len = strlen (text);
5063   }
5064
5065   rl_attempted_completion_over = 1;
5066
5067   while ((name = commands[index]) != NULL) {
5068     index++;
5069     if (strncasecmp (name, text, len) == 0)
5070       return strdup (name);
5071   }
5072
5073   return NULL;
5074 }
5075
5076 #endif /* HAVE_LIBREADLINE */
5077
5078 char **do_completion (const char *text, int start, int end)
5079 {
5080   char **matches = NULL;
5081
5082 #ifdef HAVE_LIBREADLINE
5083   rl_completion_append_character = ' ';
5084
5085   if (start == 0)
5086     matches = rl_completion_matches (text, generator);
5087   else if (complete_dest_paths)
5088     matches = rl_completion_matches (text, complete_dest_paths_generator);
5089 #endif
5090
5091   return matches;
5092 }
5093 ";
5094
5095 (* Generate the POD documentation for guestfish. *)
5096 and generate_fish_actions_pod () =
5097   let all_functions_sorted =
5098     List.filter (
5099       fun (_, _, _, flags, _, _, _) ->
5100         not (List.mem NotInFish flags || List.mem NotInDocs flags)
5101     ) all_functions_sorted in
5102
5103   let rex = Str.regexp "C<guestfs_\\([^>]+\\)>" in
5104
5105   List.iter (
5106     fun (name, style, _, flags, _, _, longdesc) ->
5107       let longdesc =
5108         Str.global_substitute rex (
5109           fun s ->
5110             let sub =
5111               try Str.matched_group 1 s
5112               with Not_found ->
5113                 failwithf "error substituting C<guestfs_...> in longdesc of function %s" name in
5114             "C<" ^ replace_char sub '_' '-' ^ ">"
5115         ) longdesc in
5116       let name = replace_char name '_' '-' in
5117       let alias =
5118         try find_map (function FishAlias n -> Some n | _ -> None) flags
5119         with Not_found -> name in
5120
5121       pr "=head2 %s" name;
5122       if name <> alias then
5123         pr " | %s" alias;
5124       pr "\n";
5125       pr "\n";
5126       pr " %s" name;
5127       List.iter (
5128         function
5129         | String n -> pr " %s" n
5130         | OptString n -> pr " %s" n
5131         | StringList n -> pr " '%s ...'" n
5132         | Bool _ -> pr " true|false"
5133         | Int n -> pr " %s" n
5134         | FileIn n | FileOut n -> pr " (%s|-)" n
5135       ) (snd style);
5136       pr "\n";
5137       pr "\n";
5138       pr "%s\n\n" longdesc;
5139
5140       if List.exists (function FileIn _ | FileOut _ -> true
5141                       | _ -> false) (snd style) then
5142         pr "Use C<-> instead of a filename to read/write from stdin/stdout.\n\n";
5143
5144       if List.mem ProtocolLimitWarning flags then
5145         pr "%s\n\n" protocol_limit_warning;
5146
5147       if List.mem DangerWillRobinson flags then
5148         pr "%s\n\n" danger_will_robinson
5149   ) all_functions_sorted
5150
5151 (* Generate a C function prototype. *)
5152 and generate_prototype ?(extern = true) ?(static = false) ?(semicolon = true)
5153     ?(single_line = false) ?(newline = false) ?(in_daemon = false)
5154     ?(prefix = "")
5155     ?handle name style =
5156   if extern then pr "extern ";
5157   if static then pr "static ";
5158   (match fst style with
5159    | RErr -> pr "int "
5160    | RInt _ -> pr "int "
5161    | RInt64 _ -> pr "int64_t "
5162    | RBool _ -> pr "int "
5163    | RConstString _ -> pr "const char *"
5164    | RString _ -> pr "char *"
5165    | RStringList _ | RHashtable _ -> pr "char **"
5166    | RIntBool _ ->
5167        if not in_daemon then pr "struct guestfs_int_bool *"
5168        else pr "guestfs_%s_ret *" name
5169    | RPVList _ ->
5170        if not in_daemon then pr "struct guestfs_lvm_pv_list *"
5171        else pr "guestfs_lvm_int_pv_list *"
5172    | RVGList _ ->
5173        if not in_daemon then pr "struct guestfs_lvm_vg_list *"
5174        else pr "guestfs_lvm_int_vg_list *"
5175    | RLVList _ ->
5176        if not in_daemon then pr "struct guestfs_lvm_lv_list *"
5177        else pr "guestfs_lvm_int_lv_list *"
5178    | RStat _ ->
5179        if not in_daemon then pr "struct guestfs_stat *"
5180        else pr "guestfs_int_stat *"
5181    | RStatVFS _ ->
5182        if not in_daemon then pr "struct guestfs_statvfs *"
5183        else pr "guestfs_int_statvfs *"
5184   );
5185   pr "%s%s (" prefix name;
5186   if handle = None && List.length (snd style) = 0 then
5187     pr "void"
5188   else (
5189     let comma = ref false in
5190     (match handle with
5191      | None -> ()
5192      | Some handle -> pr "guestfs_h *%s" handle; comma := true
5193     );
5194     let next () =
5195       if !comma then (
5196         if single_line then pr ", " else pr ",\n\t\t"
5197       );
5198       comma := true
5199     in
5200     List.iter (
5201       function
5202       | String n
5203       | OptString n ->
5204           next ();
5205           if not in_daemon then pr "const char *%s" n
5206           else pr "char *%s" n
5207       | StringList n ->
5208           next ();
5209           if not in_daemon then pr "char * const* const %s" n
5210           else pr "char **%s" n
5211       | Bool n -> next (); pr "int %s" n
5212       | Int n -> next (); pr "int %s" n
5213       | FileIn n
5214       | FileOut n ->
5215           if not in_daemon then (next (); pr "const char *%s" n)
5216     ) (snd style);
5217   );
5218   pr ")";
5219   if semicolon then pr ";";
5220   if newline then pr "\n"
5221
5222 (* Generate C call arguments, eg "(handle, foo, bar)" *)
5223 and generate_call_args ?handle args =
5224   pr "(";
5225   let comma = ref false in
5226   (match handle with
5227    | None -> ()
5228    | Some handle -> pr "%s" handle; comma := true
5229   );
5230   List.iter (
5231     fun arg ->
5232       if !comma then pr ", ";
5233       comma := true;
5234       pr "%s" (name_of_argt arg)
5235   ) args;
5236   pr ")"
5237
5238 (* Generate the OCaml bindings interface. *)
5239 and generate_ocaml_mli () =
5240   generate_header OCamlStyle LGPLv2;
5241
5242   pr "\
5243 (** For API documentation you should refer to the C API
5244     in the guestfs(3) manual page.  The OCaml API uses almost
5245     exactly the same calls. *)
5246
5247 type t
5248 (** A [guestfs_h] handle. *)
5249
5250 exception Error of string
5251 (** This exception is raised when there is an error. *)
5252
5253 val create : unit -> t
5254
5255 val close : t -> unit
5256 (** Handles are closed by the garbage collector when they become
5257     unreferenced, but callers can also call this in order to
5258     provide predictable cleanup. *)
5259
5260 ";
5261   generate_ocaml_lvm_structure_decls ();
5262
5263   generate_ocaml_stat_structure_decls ();
5264
5265   (* The actions. *)
5266   List.iter (
5267     fun (name, style, _, _, _, shortdesc, _) ->
5268       generate_ocaml_prototype name style;
5269       pr "(** %s *)\n" shortdesc;
5270       pr "\n"
5271   ) all_functions
5272
5273 (* Generate the OCaml bindings implementation. *)
5274 and generate_ocaml_ml () =
5275   generate_header OCamlStyle LGPLv2;
5276
5277   pr "\
5278 type t
5279 exception Error of string
5280 external create : unit -> t = \"ocaml_guestfs_create\"
5281 external close : t -> unit = \"ocaml_guestfs_close\"
5282
5283 let () =
5284   Callback.register_exception \"ocaml_guestfs_error\" (Error \"\")
5285
5286 ";
5287
5288   generate_ocaml_lvm_structure_decls ();
5289
5290   generate_ocaml_stat_structure_decls ();
5291
5292   (* The actions. *)
5293   List.iter (
5294     fun (name, style, _, _, _, shortdesc, _) ->
5295       generate_ocaml_prototype ~is_external:true name style;
5296   ) all_functions
5297
5298 (* Generate the OCaml bindings C implementation. *)
5299 and generate_ocaml_c () =
5300   generate_header CStyle LGPLv2;
5301
5302   pr "\
5303 #include <stdio.h>
5304 #include <stdlib.h>
5305 #include <string.h>
5306
5307 #include <caml/config.h>
5308 #include <caml/alloc.h>
5309 #include <caml/callback.h>
5310 #include <caml/fail.h>
5311 #include <caml/memory.h>
5312 #include <caml/mlvalues.h>
5313 #include <caml/signals.h>
5314
5315 #include <guestfs.h>
5316
5317 #include \"guestfs_c.h\"
5318
5319 /* Copy a hashtable of string pairs into an assoc-list.  We return
5320  * the list in reverse order, but hashtables aren't supposed to be
5321  * ordered anyway.
5322  */
5323 static CAMLprim value
5324 copy_table (char * const * argv)
5325 {
5326   CAMLparam0 ();
5327   CAMLlocal5 (rv, pairv, kv, vv, cons);
5328   int i;
5329
5330   rv = Val_int (0);
5331   for (i = 0; argv[i] != NULL; i += 2) {
5332     kv = caml_copy_string (argv[i]);
5333     vv = caml_copy_string (argv[i+1]);
5334     pairv = caml_alloc (2, 0);
5335     Store_field (pairv, 0, kv);
5336     Store_field (pairv, 1, vv);
5337     cons = caml_alloc (2, 0);
5338     Store_field (cons, 1, rv);
5339     rv = cons;
5340     Store_field (cons, 0, pairv);
5341   }
5342
5343   CAMLreturn (rv);
5344 }
5345
5346 ";
5347
5348   (* LVM struct copy functions. *)
5349   List.iter (
5350     fun (typ, cols) ->
5351       let has_optpercent_col =
5352         List.exists (function (_, `OptPercent) -> true | _ -> false) cols in
5353
5354       pr "static CAMLprim value\n";
5355       pr "copy_lvm_%s (const struct guestfs_lvm_%s *%s)\n" typ typ typ;
5356       pr "{\n";
5357       pr "  CAMLparam0 ();\n";
5358       if has_optpercent_col then
5359         pr "  CAMLlocal3 (rv, v, v2);\n"
5360       else
5361         pr "  CAMLlocal2 (rv, v);\n";
5362       pr "\n";
5363       pr "  rv = caml_alloc (%d, 0);\n" (List.length cols);
5364       iteri (
5365         fun i col ->
5366           (match col with
5367            | name, `String ->
5368                pr "  v = caml_copy_string (%s->%s);\n" typ name
5369            | name, `UUID ->
5370                pr "  v = caml_alloc_string (32);\n";
5371                pr "  memcpy (String_val (v), %s->%s, 32);\n" typ name
5372            | name, `Bytes
5373            | name, `Int ->
5374                pr "  v = caml_copy_int64 (%s->%s);\n" typ name
5375            | name, `OptPercent ->
5376                pr "  if (%s->%s >= 0) { /* Some %s */\n" typ name name;
5377                pr "    v2 = caml_copy_double (%s->%s);\n" typ name;
5378                pr "    v = caml_alloc (1, 0);\n";
5379                pr "    Store_field (v, 0, v2);\n";
5380                pr "  } else /* None */\n";
5381                pr "    v = Val_int (0);\n";
5382           );
5383           pr "  Store_field (rv, %d, v);\n" i
5384       ) cols;
5385       pr "  CAMLreturn (rv);\n";
5386       pr "}\n";
5387       pr "\n";
5388
5389       pr "static CAMLprim value\n";
5390       pr "copy_lvm_%s_list (const struct guestfs_lvm_%s_list *%ss)\n"
5391         typ typ typ;
5392       pr "{\n";
5393       pr "  CAMLparam0 ();\n";
5394       pr "  CAMLlocal2 (rv, v);\n";
5395       pr "  int i;\n";
5396       pr "\n";
5397       pr "  if (%ss->len == 0)\n" typ;
5398       pr "    CAMLreturn (Atom (0));\n";
5399       pr "  else {\n";
5400       pr "    rv = caml_alloc (%ss->len, 0);\n" typ;
5401       pr "    for (i = 0; i < %ss->len; ++i) {\n" typ;
5402       pr "      v = copy_lvm_%s (&%ss->val[i]);\n" typ typ;
5403       pr "      caml_modify (&Field (rv, i), v);\n";
5404       pr "    }\n";
5405       pr "    CAMLreturn (rv);\n";
5406       pr "  }\n";
5407       pr "}\n";
5408       pr "\n";
5409   ) ["pv", pv_cols; "vg", vg_cols; "lv", lv_cols];
5410
5411   (* Stat copy functions. *)
5412   List.iter (
5413     fun (typ, cols) ->
5414       pr "static CAMLprim value\n";
5415       pr "copy_%s (const struct guestfs_%s *%s)\n" typ typ typ;
5416       pr "{\n";
5417       pr "  CAMLparam0 ();\n";
5418       pr "  CAMLlocal2 (rv, v);\n";
5419       pr "\n";
5420       pr "  rv = caml_alloc (%d, 0);\n" (List.length cols);
5421       iteri (
5422         fun i col ->
5423           (match col with
5424            | name, `Int ->
5425                pr "  v = caml_copy_int64 (%s->%s);\n" typ name
5426           );
5427           pr "  Store_field (rv, %d, v);\n" i
5428       ) cols;
5429       pr "  CAMLreturn (rv);\n";
5430       pr "}\n";
5431       pr "\n";
5432   ) ["stat", stat_cols; "statvfs", statvfs_cols];
5433
5434   (* The wrappers. *)
5435   List.iter (
5436     fun (name, style, _, _, _, _, _) ->
5437       let params =
5438         "gv" :: List.map (fun arg -> name_of_argt arg ^ "v") (snd style) in
5439
5440       pr "CAMLprim value\n";
5441       pr "ocaml_guestfs_%s (value %s" name (List.hd params);
5442       List.iter (pr ", value %s") (List.tl params);
5443       pr ")\n";
5444       pr "{\n";
5445
5446       (match params with
5447        | [p1; p2; p3; p4; p5] ->
5448            pr "  CAMLparam5 (%s);\n" (String.concat ", " params)
5449        | p1 :: p2 :: p3 :: p4 :: p5 :: rest ->
5450            pr "  CAMLparam5 (%s);\n" (String.concat ", " [p1; p2; p3; p4; p5]);
5451            pr "  CAMLxparam%d (%s);\n"
5452              (List.length rest) (String.concat ", " rest)
5453        | ps ->
5454            pr "  CAMLparam%d (%s);\n" (List.length ps) (String.concat ", " ps)
5455       );
5456       pr "  CAMLlocal1 (rv);\n";
5457       pr "\n";
5458
5459       pr "  guestfs_h *g = Guestfs_val (gv);\n";
5460       pr "  if (g == NULL)\n";
5461       pr "    caml_failwith (\"%s: used handle after closing it\");\n" name;
5462       pr "\n";
5463
5464       List.iter (
5465         function
5466         | String n
5467         | FileIn n
5468         | FileOut n ->
5469             pr "  const char *%s = String_val (%sv);\n" n n
5470         | OptString n ->
5471             pr "  const char *%s =\n" n;
5472             pr "    %sv != Val_int (0) ? String_val (Field (%sv, 0)) : NULL;\n"
5473               n n
5474         | StringList n ->
5475             pr "  char **%s = ocaml_guestfs_strings_val (g, %sv);\n" n n
5476         | Bool n ->
5477             pr "  int %s = Bool_val (%sv);\n" n n
5478         | Int n ->
5479             pr "  int %s = Int_val (%sv);\n" n n
5480       ) (snd style);
5481       let error_code =
5482         match fst style with
5483         | RErr -> pr "  int r;\n"; "-1"
5484         | RInt _ -> pr "  int r;\n"; "-1"
5485         | RInt64 _ -> pr "  int64_t r;\n"; "-1"
5486         | RBool _ -> pr "  int r;\n"; "-1"
5487         | RConstString _ -> pr "  const char *r;\n"; "NULL"
5488         | RString _ -> pr "  char *r;\n"; "NULL"
5489         | RStringList _ ->
5490             pr "  int i;\n";
5491             pr "  char **r;\n";
5492             "NULL"
5493         | RIntBool _ ->
5494             pr "  struct guestfs_int_bool *r;\n"; "NULL"
5495         | RPVList _ ->
5496             pr "  struct guestfs_lvm_pv_list *r;\n"; "NULL"
5497         | RVGList _ ->
5498             pr "  struct guestfs_lvm_vg_list *r;\n"; "NULL"
5499         | RLVList _ ->
5500             pr "  struct guestfs_lvm_lv_list *r;\n"; "NULL"
5501         | RStat _ ->
5502             pr "  struct guestfs_stat *r;\n"; "NULL"
5503         | RStatVFS _ ->
5504             pr "  struct guestfs_statvfs *r;\n"; "NULL"
5505         | RHashtable _ ->
5506             pr "  int i;\n";
5507             pr "  char **r;\n";
5508             "NULL" in
5509       pr "\n";
5510
5511       pr "  caml_enter_blocking_section ();\n";
5512       pr "  r = guestfs_%s " name;
5513       generate_call_args ~handle:"g" (snd style);
5514       pr ";\n";
5515       pr "  caml_leave_blocking_section ();\n";
5516
5517       List.iter (
5518         function
5519         | StringList n ->
5520             pr "  ocaml_guestfs_free_strings (%s);\n" n;
5521         | String _ | OptString _ | Bool _ | Int _ | FileIn _ | FileOut _ -> ()
5522       ) (snd style);
5523
5524       pr "  if (r == %s)\n" error_code;
5525       pr "    ocaml_guestfs_raise_error (g, \"%s\");\n" name;
5526       pr "\n";
5527
5528       (match fst style with
5529        | RErr -> pr "  rv = Val_unit;\n"
5530        | RInt _ -> pr "  rv = Val_int (r);\n"
5531        | RInt64 _ ->
5532            pr "  rv = caml_copy_int64 (r);\n"
5533        | RBool _ -> pr "  rv = Val_bool (r);\n"
5534        | RConstString _ -> pr "  rv = caml_copy_string (r);\n"
5535        | RString _ ->
5536            pr "  rv = caml_copy_string (r);\n";
5537            pr "  free (r);\n"
5538        | RStringList _ ->
5539            pr "  rv = caml_copy_string_array ((const char **) r);\n";
5540            pr "  for (i = 0; r[i] != NULL; ++i) free (r[i]);\n";
5541            pr "  free (r);\n"
5542        | RIntBool _ ->
5543            pr "  rv = caml_alloc (2, 0);\n";
5544            pr "  Store_field (rv, 0, Val_int (r->i));\n";
5545            pr "  Store_field (rv, 1, Val_bool (r->b));\n";
5546            pr "  guestfs_free_int_bool (r);\n";
5547        | RPVList _ ->
5548            pr "  rv = copy_lvm_pv_list (r);\n";
5549            pr "  guestfs_free_lvm_pv_list (r);\n";
5550        | RVGList _ ->
5551            pr "  rv = copy_lvm_vg_list (r);\n";
5552            pr "  guestfs_free_lvm_vg_list (r);\n";
5553        | RLVList _ ->
5554            pr "  rv = copy_lvm_lv_list (r);\n";
5555            pr "  guestfs_free_lvm_lv_list (r);\n";
5556        | RStat _ ->
5557            pr "  rv = copy_stat (r);\n";
5558            pr "  free (r);\n";
5559        | RStatVFS _ ->
5560            pr "  rv = copy_statvfs (r);\n";
5561            pr "  free (r);\n";
5562        | RHashtable _ ->
5563            pr "  rv = copy_table (r);\n";
5564            pr "  for (i = 0; r[i] != NULL; ++i) free (r[i]);\n";
5565            pr "  free (r);\n";
5566       );
5567
5568       pr "  CAMLreturn (rv);\n";
5569       pr "}\n";
5570       pr "\n";
5571
5572       if List.length params > 5 then (
5573         pr "CAMLprim value\n";
5574         pr "ocaml_guestfs_%s_byte (value *argv, int argn)\n" name;
5575         pr "{\n";
5576         pr "  return ocaml_guestfs_%s (argv[0]" name;
5577         iteri (fun i _ -> pr ", argv[%d]" i) (List.tl params);
5578         pr ");\n";
5579         pr "}\n";
5580         pr "\n"
5581       )
5582   ) all_functions
5583
5584 and generate_ocaml_lvm_structure_decls () =
5585   List.iter (
5586     fun (typ, cols) ->
5587       pr "type lvm_%s = {\n" typ;
5588       List.iter (
5589         function
5590         | name, `String -> pr "  %s : string;\n" name
5591         | name, `UUID -> pr "  %s : string;\n" name
5592         | name, `Bytes -> pr "  %s : int64;\n" name
5593         | name, `Int -> pr "  %s : int64;\n" name
5594         | name, `OptPercent -> pr "  %s : float option;\n" name
5595       ) cols;
5596       pr "}\n";
5597       pr "\n"
5598   ) ["pv", pv_cols; "vg", vg_cols; "lv", lv_cols]
5599
5600 and generate_ocaml_stat_structure_decls () =
5601   List.iter (
5602     fun (typ, cols) ->
5603       pr "type %s = {\n" typ;
5604       List.iter (
5605         function
5606         | name, `Int -> pr "  %s : int64;\n" name
5607       ) cols;
5608       pr "}\n";
5609       pr "\n"
5610   ) ["stat", stat_cols; "statvfs", statvfs_cols]
5611
5612 and generate_ocaml_prototype ?(is_external = false) name style =
5613   if is_external then pr "external " else pr "val ";
5614   pr "%s : t -> " name;
5615   List.iter (
5616     function
5617     | String _ | FileIn _ | FileOut _ -> pr "string -> "
5618     | OptString _ -> pr "string option -> "
5619     | StringList _ -> pr "string array -> "
5620     | Bool _ -> pr "bool -> "
5621     | Int _ -> pr "int -> "
5622   ) (snd style);
5623   (match fst style with
5624    | RErr -> pr "unit" (* all errors are turned into exceptions *)
5625    | RInt _ -> pr "int"
5626    | RInt64 _ -> pr "int64"
5627    | RBool _ -> pr "bool"
5628    | RConstString _ -> pr "string"
5629    | RString _ -> pr "string"
5630    | RStringList _ -> pr "string array"
5631    | RIntBool _ -> pr "int * bool"
5632    | RPVList _ -> pr "lvm_pv array"
5633    | RVGList _ -> pr "lvm_vg array"
5634    | RLVList _ -> pr "lvm_lv array"
5635    | RStat _ -> pr "stat"
5636    | RStatVFS _ -> pr "statvfs"
5637    | RHashtable _ -> pr "(string * string) list"
5638   );
5639   if is_external then (
5640     pr " = ";
5641     if List.length (snd style) + 1 > 5 then
5642       pr "\"ocaml_guestfs_%s_byte\" " name;
5643     pr "\"ocaml_guestfs_%s\"" name
5644   );
5645   pr "\n"
5646
5647 (* Generate Perl xs code, a sort of crazy variation of C with macros. *)
5648 and generate_perl_xs () =
5649   generate_header CStyle LGPLv2;
5650
5651   pr "\
5652 #include \"EXTERN.h\"
5653 #include \"perl.h\"
5654 #include \"XSUB.h\"
5655
5656 #include <guestfs.h>
5657
5658 #ifndef PRId64
5659 #define PRId64 \"lld\"
5660 #endif
5661
5662 static SV *
5663 my_newSVll(long long val) {
5664 #ifdef USE_64_BIT_ALL
5665   return newSViv(val);
5666 #else
5667   char buf[100];
5668   int len;
5669   len = snprintf(buf, 100, \"%%\" PRId64, val);
5670   return newSVpv(buf, len);
5671 #endif
5672 }
5673
5674 #ifndef PRIu64
5675 #define PRIu64 \"llu\"
5676 #endif
5677
5678 static SV *
5679 my_newSVull(unsigned long long val) {
5680 #ifdef USE_64_BIT_ALL
5681   return newSVuv(val);
5682 #else
5683   char buf[100];
5684   int len;
5685   len = snprintf(buf, 100, \"%%\" PRIu64, val);
5686   return newSVpv(buf, len);
5687 #endif
5688 }
5689
5690 /* http://www.perlmonks.org/?node_id=680842 */
5691 static char **
5692 XS_unpack_charPtrPtr (SV *arg) {
5693   char **ret;
5694   AV *av;
5695   I32 i;
5696
5697   if (!arg || !SvOK (arg) || !SvROK (arg) || SvTYPE (SvRV (arg)) != SVt_PVAV)
5698     croak (\"array reference expected\");
5699
5700   av = (AV *)SvRV (arg);
5701   ret = malloc ((av_len (av) + 1 + 1) * sizeof (char *));
5702   if (!ret)
5703     croak (\"malloc failed\");
5704
5705   for (i = 0; i <= av_len (av); i++) {
5706     SV **elem = av_fetch (av, i, 0);
5707
5708     if (!elem || !*elem)
5709       croak (\"missing element in list\");
5710
5711     ret[i] = SvPV_nolen (*elem);
5712   }
5713
5714   ret[i] = NULL;
5715
5716   return ret;
5717 }
5718
5719 MODULE = Sys::Guestfs  PACKAGE = Sys::Guestfs
5720
5721 PROTOTYPES: ENABLE
5722
5723 guestfs_h *
5724 _create ()
5725    CODE:
5726       RETVAL = guestfs_create ();
5727       if (!RETVAL)
5728         croak (\"could not create guestfs handle\");
5729       guestfs_set_error_handler (RETVAL, NULL, NULL);
5730  OUTPUT:
5731       RETVAL
5732
5733 void
5734 DESTROY (g)
5735       guestfs_h *g;
5736  PPCODE:
5737       guestfs_close (g);
5738
5739 ";
5740
5741   List.iter (
5742     fun (name, style, _, _, _, _, _) ->
5743       (match fst style with
5744        | RErr -> pr "void\n"
5745        | RInt _ -> pr "SV *\n"
5746        | RInt64 _ -> pr "SV *\n"
5747        | RBool _ -> pr "SV *\n"
5748        | RConstString _ -> pr "SV *\n"
5749        | RString _ -> pr "SV *\n"
5750        | RStringList _
5751        | RIntBool _
5752        | RPVList _ | RVGList _ | RLVList _
5753        | RStat _ | RStatVFS _
5754        | RHashtable _ ->
5755            pr "void\n" (* all lists returned implictly on the stack *)
5756       );
5757       (* Call and arguments. *)
5758       pr "%s " name;
5759       generate_call_args ~handle:"g" (snd style);
5760       pr "\n";
5761       pr "      guestfs_h *g;\n";
5762       iteri (
5763         fun i ->
5764           function
5765           | String n | FileIn n | FileOut n -> pr "      char *%s;\n" n
5766           | OptString n ->
5767               (* http://www.perlmonks.org/?node_id=554277
5768                * Note that the implicit handle argument means we have
5769                * to add 1 to the ST(x) operator.
5770                *)
5771               pr "      char *%s = SvOK(ST(%d)) ? SvPV_nolen(ST(%d)) : NULL;\n" n (i+1) (i+1)
5772           | StringList n -> pr "      char **%s;\n" n
5773           | Bool n -> pr "      int %s;\n" n
5774           | Int n -> pr "      int %s;\n" n
5775       ) (snd style);
5776
5777       let do_cleanups () =
5778         List.iter (
5779           function
5780           | String _ | OptString _ | Bool _ | Int _
5781           | FileIn _ | FileOut _ -> ()
5782           | StringList n -> pr "      free (%s);\n" n
5783         ) (snd style)
5784       in
5785
5786       (* Code. *)
5787       (match fst style with
5788        | RErr ->
5789            pr "PREINIT:\n";
5790            pr "      int r;\n";
5791            pr " PPCODE:\n";
5792            pr "      r = guestfs_%s " name;
5793            generate_call_args ~handle:"g" (snd style);
5794            pr ";\n";
5795            do_cleanups ();
5796            pr "      if (r == -1)\n";
5797            pr "        croak (\"%s: %%s\", guestfs_last_error (g));\n" name;
5798        | RInt n
5799        | RBool n ->
5800            pr "PREINIT:\n";
5801            pr "      int %s;\n" n;
5802            pr "   CODE:\n";
5803            pr "      %s = guestfs_%s " n name;
5804            generate_call_args ~handle:"g" (snd style);
5805            pr ";\n";
5806            do_cleanups ();
5807            pr "      if (%s == -1)\n" n;
5808            pr "        croak (\"%s: %%s\", guestfs_last_error (g));\n" name;
5809            pr "      RETVAL = newSViv (%s);\n" n;
5810            pr " OUTPUT:\n";
5811            pr "      RETVAL\n"
5812        | RInt64 n ->
5813            pr "PREINIT:\n";
5814            pr "      int64_t %s;\n" n;
5815            pr "   CODE:\n";
5816            pr "      %s = guestfs_%s " n name;
5817            generate_call_args ~handle:"g" (snd style);
5818            pr ";\n";
5819            do_cleanups ();
5820            pr "      if (%s == -1)\n" n;
5821            pr "        croak (\"%s: %%s\", guestfs_last_error (g));\n" name;
5822            pr "      RETVAL = my_newSVll (%s);\n" n;
5823            pr " OUTPUT:\n";
5824            pr "      RETVAL\n"
5825        | RConstString n ->
5826            pr "PREINIT:\n";
5827            pr "      const char *%s;\n" n;
5828            pr "   CODE:\n";
5829            pr "      %s = guestfs_%s " n name;
5830            generate_call_args ~handle:"g" (snd style);
5831            pr ";\n";
5832            do_cleanups ();
5833            pr "      if (%s == NULL)\n" n;
5834            pr "        croak (\"%s: %%s\", guestfs_last_error (g));\n" name;
5835            pr "      RETVAL = newSVpv (%s, 0);\n" n;
5836            pr " OUTPUT:\n";
5837            pr "      RETVAL\n"
5838        | RString n ->
5839            pr "PREINIT:\n";
5840            pr "      char *%s;\n" n;
5841            pr "   CODE:\n";
5842            pr "      %s = guestfs_%s " n name;
5843            generate_call_args ~handle:"g" (snd style);
5844            pr ";\n";
5845            do_cleanups ();
5846            pr "      if (%s == NULL)\n" n;
5847            pr "        croak (\"%s: %%s\", guestfs_last_error (g));\n" name;
5848            pr "      RETVAL = newSVpv (%s, 0);\n" n;
5849            pr "      free (%s);\n" n;
5850            pr " OUTPUT:\n";
5851            pr "      RETVAL\n"
5852        | RStringList n | RHashtable n ->
5853            pr "PREINIT:\n";
5854            pr "      char **%s;\n" n;
5855            pr "      int i, n;\n";
5856            pr " PPCODE:\n";
5857            pr "      %s = guestfs_%s " n name;
5858            generate_call_args ~handle:"g" (snd style);
5859            pr ";\n";
5860            do_cleanups ();
5861            pr "      if (%s == NULL)\n" n;
5862            pr "        croak (\"%s: %%s\", guestfs_last_error (g));\n" name;
5863            pr "      for (n = 0; %s[n] != NULL; ++n) /**/;\n" n;
5864            pr "      EXTEND (SP, n);\n";
5865            pr "      for (i = 0; i < n; ++i) {\n";
5866            pr "        PUSHs (sv_2mortal (newSVpv (%s[i], 0)));\n" n;
5867            pr "        free (%s[i]);\n" n;
5868            pr "      }\n";
5869            pr "      free (%s);\n" n;
5870        | RIntBool _ ->
5871            pr "PREINIT:\n";
5872            pr "      struct guestfs_int_bool *r;\n";
5873            pr " PPCODE:\n";
5874            pr "      r = guestfs_%s " name;
5875            generate_call_args ~handle:"g" (snd style);
5876            pr ";\n";
5877            do_cleanups ();
5878            pr "      if (r == NULL)\n";
5879            pr "        croak (\"%s: %%s\", guestfs_last_error (g));\n" name;
5880            pr "      EXTEND (SP, 2);\n";
5881            pr "      PUSHs (sv_2mortal (newSViv (r->i)));\n";
5882            pr "      PUSHs (sv_2mortal (newSViv (r->b)));\n";
5883            pr "      guestfs_free_int_bool (r);\n";
5884        | RPVList n ->
5885            generate_perl_lvm_code "pv" pv_cols name style n do_cleanups
5886        | RVGList n ->
5887            generate_perl_lvm_code "vg" vg_cols name style n do_cleanups
5888        | RLVList n ->
5889            generate_perl_lvm_code "lv" lv_cols name style n do_cleanups
5890        | RStat n ->
5891            generate_perl_stat_code "stat" stat_cols name style n do_cleanups
5892        | RStatVFS n ->
5893            generate_perl_stat_code
5894              "statvfs" statvfs_cols name style n do_cleanups
5895       );
5896
5897       pr "\n"
5898   ) all_functions
5899
5900 and generate_perl_lvm_code typ cols name style n do_cleanups =
5901   pr "PREINIT:\n";
5902   pr "      struct guestfs_lvm_%s_list *%s;\n" typ n;
5903   pr "      int i;\n";
5904   pr "      HV *hv;\n";
5905   pr " PPCODE:\n";
5906   pr "      %s = guestfs_%s " n name;
5907   generate_call_args ~handle:"g" (snd style);
5908   pr ";\n";
5909   do_cleanups ();
5910   pr "      if (%s == NULL)\n" n;
5911   pr "        croak (\"%s: %%s\", guestfs_last_error (g));\n" name;
5912   pr "      EXTEND (SP, %s->len);\n" n;
5913   pr "      for (i = 0; i < %s->len; ++i) {\n" n;
5914   pr "        hv = newHV ();\n";
5915   List.iter (
5916     function
5917     | name, `String ->
5918         pr "        (void) hv_store (hv, \"%s\", %d, newSVpv (%s->val[i].%s, 0), 0);\n"
5919           name (String.length name) n name
5920     | name, `UUID ->
5921         pr "        (void) hv_store (hv, \"%s\", %d, newSVpv (%s->val[i].%s, 32), 0);\n"
5922           name (String.length name) n name
5923     | name, `Bytes ->
5924         pr "        (void) hv_store (hv, \"%s\", %d, my_newSVull (%s->val[i].%s), 0);\n"
5925           name (String.length name) n name
5926     | name, `Int ->
5927         pr "        (void) hv_store (hv, \"%s\", %d, my_newSVll (%s->val[i].%s), 0);\n"
5928           name (String.length name) n name
5929     | name, `OptPercent ->
5930         pr "        (void) hv_store (hv, \"%s\", %d, newSVnv (%s->val[i].%s), 0);\n"
5931           name (String.length name) n name
5932   ) cols;
5933   pr "        PUSHs (sv_2mortal ((SV *) hv));\n";
5934   pr "      }\n";
5935   pr "      guestfs_free_lvm_%s_list (%s);\n" typ n
5936
5937 and generate_perl_stat_code typ cols name style n do_cleanups =
5938   pr "PREINIT:\n";
5939   pr "      struct guestfs_%s *%s;\n" typ n;
5940   pr " PPCODE:\n";
5941   pr "      %s = guestfs_%s " n name;
5942   generate_call_args ~handle:"g" (snd style);
5943   pr ";\n";
5944   do_cleanups ();
5945   pr "      if (%s == NULL)\n" n;
5946   pr "        croak (\"%s: %%s\", guestfs_last_error (g));\n" name;
5947   pr "      EXTEND (SP, %d);\n" (List.length cols);
5948   List.iter (
5949     function
5950     | name, `Int ->
5951         pr "      PUSHs (sv_2mortal (my_newSVll (%s->%s)));\n" n name
5952   ) cols;
5953   pr "      free (%s);\n" n
5954
5955 (* Generate Sys/Guestfs.pm. *)
5956 and generate_perl_pm () =
5957   generate_header HashStyle LGPLv2;
5958
5959   pr "\
5960 =pod
5961
5962 =head1 NAME
5963
5964 Sys::Guestfs - Perl bindings for libguestfs
5965
5966 =head1 SYNOPSIS
5967
5968  use Sys::Guestfs;
5969  
5970  my $h = Sys::Guestfs->new ();
5971  $h->add_drive ('guest.img');
5972  $h->launch ();
5973  $h->wait_ready ();
5974  $h->mount ('/dev/sda1', '/');
5975  $h->touch ('/hello');
5976  $h->sync ();
5977
5978 =head1 DESCRIPTION
5979
5980 The C<Sys::Guestfs> module provides a Perl XS binding to the
5981 libguestfs API for examining and modifying virtual machine
5982 disk images.
5983
5984 Amongst the things this is good for: making batch configuration
5985 changes to guests, getting disk used/free statistics (see also:
5986 virt-df), migrating between virtualization systems (see also:
5987 virt-p2v), performing partial backups, performing partial guest
5988 clones, cloning guests and changing registry/UUID/hostname info, and
5989 much else besides.
5990
5991 Libguestfs uses Linux kernel and qemu code, and can access any type of
5992 guest filesystem that Linux and qemu can, including but not limited
5993 to: ext2/3/4, btrfs, FAT and NTFS, LVM, many different disk partition
5994 schemes, qcow, qcow2, vmdk.
5995
5996 Libguestfs provides ways to enumerate guest storage (eg. partitions,
5997 LVs, what filesystem is in each LV, etc.).  It can also run commands
5998 in the context of the guest.  Also you can access filesystems over FTP.
5999
6000 =head1 ERRORS
6001
6002 All errors turn into calls to C<croak> (see L<Carp(3)>).
6003
6004 =head1 METHODS
6005
6006 =over 4
6007
6008 =cut
6009
6010 package Sys::Guestfs;
6011
6012 use strict;
6013 use warnings;
6014
6015 require XSLoader;
6016 XSLoader::load ('Sys::Guestfs');
6017
6018 =item $h = Sys::Guestfs->new ();
6019
6020 Create a new guestfs handle.
6021
6022 =cut
6023
6024 sub new {
6025   my $proto = shift;
6026   my $class = ref ($proto) || $proto;
6027
6028   my $self = Sys::Guestfs::_create ();
6029   bless $self, $class;
6030   return $self;
6031 }
6032
6033 ";
6034
6035   (* Actions.  We only need to print documentation for these as
6036    * they are pulled in from the XS code automatically.
6037    *)
6038   List.iter (
6039     fun (name, style, _, flags, _, _, longdesc) ->
6040       if not (List.mem NotInDocs flags) then (
6041         let longdesc = replace_str longdesc "C<guestfs_" "C<$h-E<gt>" in
6042         pr "=item ";
6043         generate_perl_prototype name style;
6044         pr "\n\n";
6045         pr "%s\n\n" longdesc;
6046         if List.mem ProtocolLimitWarning flags then
6047           pr "%s\n\n" protocol_limit_warning;
6048         if List.mem DangerWillRobinson flags then
6049           pr "%s\n\n" danger_will_robinson
6050       )
6051   ) all_functions_sorted;
6052
6053   (* End of file. *)
6054   pr "\
6055 =cut
6056
6057 1;
6058
6059 =back
6060
6061 =head1 COPYRIGHT
6062
6063 Copyright (C) 2009 Red Hat Inc.
6064
6065 =head1 LICENSE
6066
6067 Please see the file COPYING.LIB for the full license.
6068
6069 =head1 SEE ALSO
6070
6071 L<guestfs(3)>, L<guestfish(1)>.
6072
6073 =cut
6074 "
6075
6076 and generate_perl_prototype name style =
6077   (match fst style with
6078    | RErr -> ()
6079    | RBool n
6080    | RInt n
6081    | RInt64 n
6082    | RConstString n
6083    | RString n -> pr "$%s = " n
6084    | RIntBool (n, m) -> pr "($%s, $%s) = " n m
6085    | RStringList n
6086    | RPVList n
6087    | RVGList n
6088    | RLVList n -> pr "@%s = " n
6089    | RStat n
6090    | RStatVFS n
6091    | RHashtable n -> pr "%%%s = " n
6092   );
6093   pr "$h->%s (" name;
6094   let comma = ref false in
6095   List.iter (
6096     fun arg ->
6097       if !comma then pr ", ";
6098       comma := true;
6099       match arg with
6100       | String n | OptString n | Bool n | Int n | FileIn n | FileOut n ->
6101           pr "$%s" n
6102       | StringList n ->
6103           pr "\\@%s" n
6104   ) (snd style);
6105   pr ");"
6106
6107 (* Generate Python C module. *)
6108 and generate_python_c () =
6109   generate_header CStyle LGPLv2;
6110
6111   pr "\
6112 #include <stdio.h>
6113 #include <stdlib.h>
6114 #include <assert.h>
6115
6116 #include <Python.h>
6117
6118 #include \"guestfs.h\"
6119
6120 typedef struct {
6121   PyObject_HEAD
6122   guestfs_h *g;
6123 } Pyguestfs_Object;
6124
6125 static guestfs_h *
6126 get_handle (PyObject *obj)
6127 {
6128   assert (obj);
6129   assert (obj != Py_None);
6130   return ((Pyguestfs_Object *) obj)->g;
6131 }
6132
6133 static PyObject *
6134 put_handle (guestfs_h *g)
6135 {
6136   assert (g);
6137   return
6138     PyCObject_FromVoidPtrAndDesc ((void *) g, (char *) \"guestfs_h\", NULL);
6139 }
6140
6141 /* This list should be freed (but not the strings) after use. */
6142 static const char **
6143 get_string_list (PyObject *obj)
6144 {
6145   int i, len;
6146   const char **r;
6147
6148   assert (obj);
6149
6150   if (!PyList_Check (obj)) {
6151     PyErr_SetString (PyExc_RuntimeError, \"expecting a list parameter\");
6152     return NULL;
6153   }
6154
6155   len = PyList_Size (obj);
6156   r = malloc (sizeof (char *) * (len+1));
6157   if (r == NULL) {
6158     PyErr_SetString (PyExc_RuntimeError, \"get_string_list: out of memory\");
6159     return NULL;
6160   }
6161
6162   for (i = 0; i < len; ++i)
6163     r[i] = PyString_AsString (PyList_GetItem (obj, i));
6164   r[len] = NULL;
6165
6166   return r;
6167 }
6168
6169 static PyObject *
6170 put_string_list (char * const * const argv)
6171 {
6172   PyObject *list;
6173   int argc, i;
6174
6175   for (argc = 0; argv[argc] != NULL; ++argc)
6176     ;
6177
6178   list = PyList_New (argc);
6179   for (i = 0; i < argc; ++i)
6180     PyList_SetItem (list, i, PyString_FromString (argv[i]));
6181
6182   return list;
6183 }
6184
6185 static PyObject *
6186 put_table (char * const * const argv)
6187 {
6188   PyObject *list, *item;
6189   int argc, i;
6190
6191   for (argc = 0; argv[argc] != NULL; ++argc)
6192     ;
6193
6194   list = PyList_New (argc >> 1);
6195   for (i = 0; i < argc; i += 2) {
6196     item = PyTuple_New (2);
6197     PyTuple_SetItem (item, 0, PyString_FromString (argv[i]));
6198     PyTuple_SetItem (item, 1, PyString_FromString (argv[i+1]));
6199     PyList_SetItem (list, i >> 1, item);
6200   }
6201
6202   return list;
6203 }
6204
6205 static void
6206 free_strings (char **argv)
6207 {
6208   int argc;
6209
6210   for (argc = 0; argv[argc] != NULL; ++argc)
6211     free (argv[argc]);
6212   free (argv);
6213 }
6214
6215 static PyObject *
6216 py_guestfs_create (PyObject *self, PyObject *args)
6217 {
6218   guestfs_h *g;
6219
6220   g = guestfs_create ();
6221   if (g == NULL) {
6222     PyErr_SetString (PyExc_RuntimeError,
6223                      \"guestfs.create: failed to allocate handle\");
6224     return NULL;
6225   }
6226   guestfs_set_error_handler (g, NULL, NULL);
6227   return put_handle (g);
6228 }
6229
6230 static PyObject *
6231 py_guestfs_close (PyObject *self, PyObject *args)
6232 {
6233   PyObject *py_g;
6234   guestfs_h *g;
6235
6236   if (!PyArg_ParseTuple (args, (char *) \"O:guestfs_close\", &py_g))
6237     return NULL;
6238   g = get_handle (py_g);
6239
6240   guestfs_close (g);
6241
6242   Py_INCREF (Py_None);
6243   return Py_None;
6244 }
6245
6246 ";
6247
6248   (* LVM structures, turned into Python dictionaries. *)
6249   List.iter (
6250     fun (typ, cols) ->
6251       pr "static PyObject *\n";
6252       pr "put_lvm_%s (struct guestfs_lvm_%s *%s)\n" typ typ typ;
6253       pr "{\n";
6254       pr "  PyObject *dict;\n";
6255       pr "\n";
6256       pr "  dict = PyDict_New ();\n";
6257       List.iter (
6258         function
6259         | name, `String ->
6260             pr "  PyDict_SetItemString (dict, \"%s\",\n" name;
6261             pr "                        PyString_FromString (%s->%s));\n"
6262               typ name
6263         | name, `UUID ->
6264             pr "  PyDict_SetItemString (dict, \"%s\",\n" name;
6265             pr "                        PyString_FromStringAndSize (%s->%s, 32));\n"
6266               typ name
6267         | name, `Bytes ->
6268             pr "  PyDict_SetItemString (dict, \"%s\",\n" name;
6269             pr "                        PyLong_FromUnsignedLongLong (%s->%s));\n"
6270               typ name
6271         | name, `Int ->
6272             pr "  PyDict_SetItemString (dict, \"%s\",\n" name;
6273             pr "                        PyLong_FromLongLong (%s->%s));\n"
6274               typ name
6275         | name, `OptPercent ->
6276             pr "  if (%s->%s >= 0)\n" typ name;
6277             pr "    PyDict_SetItemString (dict, \"%s\",\n" name;
6278             pr "                          PyFloat_FromDouble ((double) %s->%s));\n"
6279               typ name;
6280             pr "  else {\n";
6281             pr "    Py_INCREF (Py_None);\n";
6282             pr "    PyDict_SetItemString (dict, \"%s\", Py_None);" name;
6283             pr "  }\n"
6284       ) cols;
6285       pr "  return dict;\n";
6286       pr "};\n";
6287       pr "\n";
6288
6289       pr "static PyObject *\n";
6290       pr "put_lvm_%s_list (struct guestfs_lvm_%s_list *%ss)\n" typ typ typ;
6291       pr "{\n";
6292       pr "  PyObject *list;\n";
6293       pr "  int i;\n";
6294       pr "\n";
6295       pr "  list = PyList_New (%ss->len);\n" typ;
6296       pr "  for (i = 0; i < %ss->len; ++i)\n" typ;
6297       pr "    PyList_SetItem (list, i, put_lvm_%s (&%ss->val[i]));\n" typ typ;
6298       pr "  return list;\n";
6299       pr "};\n";
6300       pr "\n"
6301   ) ["pv", pv_cols; "vg", vg_cols; "lv", lv_cols];
6302
6303   (* Stat structures, turned into Python dictionaries. *)
6304   List.iter (
6305     fun (typ, cols) ->
6306       pr "static PyObject *\n";
6307       pr "put_%s (struct guestfs_%s *%s)\n" typ typ typ;
6308       pr "{\n";
6309       pr "  PyObject *dict;\n";
6310       pr "\n";
6311       pr "  dict = PyDict_New ();\n";
6312       List.iter (
6313         function
6314         | name, `Int ->
6315             pr "  PyDict_SetItemString (dict, \"%s\",\n" name;
6316             pr "                        PyLong_FromLongLong (%s->%s));\n"
6317               typ name
6318       ) cols;
6319       pr "  return dict;\n";
6320       pr "};\n";
6321       pr "\n";
6322   ) ["stat", stat_cols; "statvfs", statvfs_cols];
6323
6324   (* Python wrapper functions. *)
6325   List.iter (
6326     fun (name, style, _, _, _, _, _) ->
6327       pr "static PyObject *\n";
6328       pr "py_guestfs_%s (PyObject *self, PyObject *args)\n" name;
6329       pr "{\n";
6330
6331       pr "  PyObject *py_g;\n";
6332       pr "  guestfs_h *g;\n";
6333       pr "  PyObject *py_r;\n";
6334
6335       let error_code =
6336         match fst style with
6337         | RErr | RInt _ | RBool _ -> pr "  int r;\n"; "-1"
6338         | RInt64 _ -> pr "  int64_t r;\n"; "-1"
6339         | RConstString _ -> pr "  const char *r;\n"; "NULL"
6340         | RString _ -> pr "  char *r;\n"; "NULL"
6341         | RStringList _ | RHashtable _ -> pr "  char **r;\n"; "NULL"
6342         | RIntBool _ -> pr "  struct guestfs_int_bool *r;\n"; "NULL"
6343         | RPVList n -> pr "  struct guestfs_lvm_pv_list *r;\n"; "NULL"
6344         | RVGList n -> pr "  struct guestfs_lvm_vg_list *r;\n"; "NULL"
6345         | RLVList n -> pr "  struct guestfs_lvm_lv_list *r;\n"; "NULL"
6346         | RStat n -> pr "  struct guestfs_stat *r;\n"; "NULL"
6347         | RStatVFS n -> pr "  struct guestfs_statvfs *r;\n"; "NULL" in
6348
6349       List.iter (
6350         function
6351         | String n | FileIn n | FileOut n -> pr "  const char *%s;\n" n
6352         | OptString n -> pr "  const char *%s;\n" n
6353         | StringList n ->
6354             pr "  PyObject *py_%s;\n" n;
6355             pr "  const char **%s;\n" n
6356         | Bool n -> pr "  int %s;\n" n
6357         | Int n -> pr "  int %s;\n" n
6358       ) (snd style);
6359
6360       pr "\n";
6361
6362       (* Convert the parameters. *)
6363       pr "  if (!PyArg_ParseTuple (args, (char *) \"O";
6364       List.iter (
6365         function
6366         | String _ | FileIn _ | FileOut _ -> pr "s"
6367         | OptString _ -> pr "z"
6368         | StringList _ -> pr "O"
6369         | Bool _ -> pr "i" (* XXX Python has booleans? *)
6370         | Int _ -> pr "i"
6371       ) (snd style);
6372       pr ":guestfs_%s\",\n" name;
6373       pr "                         &py_g";
6374       List.iter (
6375         function
6376         | String n | FileIn n | FileOut n -> pr ", &%s" n
6377         | OptString n -> pr ", &%s" n
6378         | StringList n -> pr ", &py_%s" n
6379         | Bool n -> pr ", &%s" n
6380         | Int n -> pr ", &%s" n
6381       ) (snd style);
6382
6383       pr "))\n";
6384       pr "    return NULL;\n";
6385
6386       pr "  g = get_handle (py_g);\n";
6387       List.iter (
6388         function
6389         | String _ | FileIn _ | FileOut _ | OptString _ | Bool _ | Int _ -> ()
6390         | StringList n ->
6391             pr "  %s = get_string_list (py_%s);\n" n n;
6392             pr "  if (!%s) return NULL;\n" n
6393       ) (snd style);
6394
6395       pr "\n";
6396
6397       pr "  r = guestfs_%s " name;
6398       generate_call_args ~handle:"g" (snd style);
6399       pr ";\n";
6400
6401       List.iter (
6402         function
6403         | String _ | FileIn _ | FileOut _ | OptString _ | Bool _ | Int _ -> ()
6404         | StringList n ->
6405             pr "  free (%s);\n" n
6406       ) (snd style);
6407
6408       pr "  if (r == %s) {\n" error_code;
6409       pr "    PyErr_SetString (PyExc_RuntimeError, guestfs_last_error (g));\n";
6410       pr "    return NULL;\n";
6411       pr "  }\n";
6412       pr "\n";
6413
6414       (match fst style with
6415        | RErr ->
6416            pr "  Py_INCREF (Py_None);\n";
6417            pr "  py_r = Py_None;\n"
6418        | RInt _
6419        | RBool _ -> pr "  py_r = PyInt_FromLong ((long) r);\n"
6420        | RInt64 _ -> pr "  py_r = PyLong_FromLongLong (r);\n"
6421        | RConstString _ -> pr "  py_r = PyString_FromString (r);\n"
6422        | RString _ ->
6423            pr "  py_r = PyString_FromString (r);\n";
6424            pr "  free (r);\n"
6425        | RStringList _ ->
6426            pr "  py_r = put_string_list (r);\n";
6427            pr "  free_strings (r);\n"
6428        | RIntBool _ ->
6429            pr "  py_r = PyTuple_New (2);\n";
6430            pr "  PyTuple_SetItem (py_r, 0, PyInt_FromLong ((long) r->i));\n";
6431            pr "  PyTuple_SetItem (py_r, 1, PyInt_FromLong ((long) r->b));\n";
6432            pr "  guestfs_free_int_bool (r);\n"
6433        | RPVList n ->
6434            pr "  py_r = put_lvm_pv_list (r);\n";
6435            pr "  guestfs_free_lvm_pv_list (r);\n"
6436        | RVGList n ->
6437            pr "  py_r = put_lvm_vg_list (r);\n";
6438            pr "  guestfs_free_lvm_vg_list (r);\n"
6439        | RLVList n ->
6440            pr "  py_r = put_lvm_lv_list (r);\n";
6441            pr "  guestfs_free_lvm_lv_list (r);\n"
6442        | RStat n ->
6443            pr "  py_r = put_stat (r);\n";
6444            pr "  free (r);\n"
6445        | RStatVFS n ->
6446            pr "  py_r = put_statvfs (r);\n";
6447            pr "  free (r);\n"
6448        | RHashtable n ->
6449            pr "  py_r = put_table (r);\n";
6450            pr "  free_strings (r);\n"
6451       );
6452
6453       pr "  return py_r;\n";
6454       pr "}\n";
6455       pr "\n"
6456   ) all_functions;
6457
6458   (* Table of functions. *)
6459   pr "static PyMethodDef methods[] = {\n";
6460   pr "  { (char *) \"create\", py_guestfs_create, METH_VARARGS, NULL },\n";
6461   pr "  { (char *) \"close\", py_guestfs_close, METH_VARARGS, NULL },\n";
6462   List.iter (
6463     fun (name, _, _, _, _, _, _) ->
6464       pr "  { (char *) \"%s\", py_guestfs_%s, METH_VARARGS, NULL },\n"
6465         name name
6466   ) all_functions;
6467   pr "  { NULL, NULL, 0, NULL }\n";
6468   pr "};\n";
6469   pr "\n";
6470
6471   (* Init function. *)
6472   pr "\
6473 void
6474 initlibguestfsmod (void)
6475 {
6476   static int initialized = 0;
6477
6478   if (initialized) return;
6479   Py_InitModule ((char *) \"libguestfsmod\", methods);
6480   initialized = 1;
6481 }
6482 "
6483
6484 (* Generate Python module. *)
6485 and generate_python_py () =
6486   generate_header HashStyle LGPLv2;
6487
6488   pr "\
6489 u\"\"\"Python bindings for libguestfs
6490
6491 import guestfs
6492 g = guestfs.GuestFS ()
6493 g.add_drive (\"guest.img\")
6494 g.launch ()
6495 g.wait_ready ()
6496 parts = g.list_partitions ()
6497
6498 The guestfs module provides a Python binding to the libguestfs API
6499 for examining and modifying virtual machine disk images.
6500
6501 Amongst the things this is good for: making batch configuration
6502 changes to guests, getting disk used/free statistics (see also:
6503 virt-df), migrating between virtualization systems (see also:
6504 virt-p2v), performing partial backups, performing partial guest
6505 clones, cloning guests and changing registry/UUID/hostname info, and
6506 much else besides.
6507
6508 Libguestfs uses Linux kernel and qemu code, and can access any type of
6509 guest filesystem that Linux and qemu can, including but not limited
6510 to: ext2/3/4, btrfs, FAT and NTFS, LVM, many different disk partition
6511 schemes, qcow, qcow2, vmdk.
6512
6513 Libguestfs provides ways to enumerate guest storage (eg. partitions,
6514 LVs, what filesystem is in each LV, etc.).  It can also run commands
6515 in the context of the guest.  Also you can access filesystems over FTP.
6516
6517 Errors which happen while using the API are turned into Python
6518 RuntimeError exceptions.
6519
6520 To create a guestfs handle you usually have to perform the following
6521 sequence of calls:
6522
6523 # Create the handle, call add_drive at least once, and possibly
6524 # several times if the guest has multiple block devices:
6525 g = guestfs.GuestFS ()
6526 g.add_drive (\"guest.img\")
6527
6528 # Launch the qemu subprocess and wait for it to become ready:
6529 g.launch ()
6530 g.wait_ready ()
6531
6532 # Now you can issue commands, for example:
6533 logvols = g.lvs ()
6534
6535 \"\"\"
6536
6537 import libguestfsmod
6538
6539 class GuestFS:
6540     \"\"\"Instances of this class are libguestfs API handles.\"\"\"
6541
6542     def __init__ (self):
6543         \"\"\"Create a new libguestfs handle.\"\"\"
6544         self._o = libguestfsmod.create ()
6545
6546     def __del__ (self):
6547         libguestfsmod.close (self._o)
6548
6549 ";
6550
6551   List.iter (
6552     fun (name, style, _, flags, _, _, longdesc) ->
6553       pr "    def %s " name;
6554       generate_call_args ~handle:"self" (snd style);
6555       pr ":\n";
6556
6557       if not (List.mem NotInDocs flags) then (
6558         let doc = replace_str longdesc "C<guestfs_" "C<g." in
6559         let doc =
6560           match fst style with
6561           | RErr | RInt _ | RInt64 _ | RBool _ | RConstString _
6562           | RString _ -> doc
6563           | RStringList _ ->
6564               doc ^ "\n\nThis function returns a list of strings."
6565           | RIntBool _ ->
6566               doc ^ "\n\nThis function returns a tuple (int, bool).\n"
6567           | RPVList _ ->
6568               doc ^ "\n\nThis function returns a list of PVs.  Each PV is represented as a dictionary."
6569           | RVGList _ ->
6570               doc ^ "\n\nThis function returns a list of VGs.  Each VG is represented as a dictionary."
6571           | RLVList _ ->
6572               doc ^ "\n\nThis function returns a list of LVs.  Each LV is represented as a dictionary."
6573           | RStat _ ->
6574               doc ^ "\n\nThis function returns a dictionary, with keys matching the various fields in the stat structure."
6575           | RStatVFS _ ->
6576               doc ^ "\n\nThis function returns a dictionary, with keys matching the various fields in the statvfs structure."
6577           | RHashtable _ ->
6578               doc ^ "\n\nThis function returns a dictionary." in
6579         let doc =
6580           if List.mem ProtocolLimitWarning flags then
6581             doc ^ "\n\n" ^ protocol_limit_warning
6582           else doc in
6583         let doc =
6584           if List.mem DangerWillRobinson flags then
6585             doc ^ "\n\n" ^ danger_will_robinson
6586           else doc in
6587         let doc = pod2text ~width:60 name doc in
6588         let doc = List.map (fun line -> replace_str line "\\" "\\\\") doc in
6589         let doc = String.concat "\n        " doc in
6590         pr "        u\"\"\"%s\"\"\"\n" doc;
6591       );
6592       pr "        return libguestfsmod.%s " name;
6593       generate_call_args ~handle:"self._o" (snd style);
6594       pr "\n";
6595       pr "\n";
6596   ) all_functions
6597
6598 (* Useful if you need the longdesc POD text as plain text.  Returns a
6599  * list of lines.
6600  *
6601  * This is the slowest thing about autogeneration.
6602  *)
6603 and pod2text ~width name longdesc =
6604   let filename, chan = Filename.open_temp_file "gen" ".tmp" in
6605   fprintf chan "=head1 %s\n\n%s\n" name longdesc;
6606   close_out chan;
6607   let cmd = sprintf "pod2text -w %d %s" width (Filename.quote filename) in
6608   let chan = Unix.open_process_in cmd in
6609   let lines = ref [] in
6610   let rec loop i =
6611     let line = input_line chan in
6612     if i = 1 then               (* discard the first line of output *)
6613       loop (i+1)
6614     else (
6615       let line = triml line in
6616       lines := line :: !lines;
6617       loop (i+1)
6618     ) in
6619   let lines = try loop 1 with End_of_file -> List.rev !lines in
6620   Unix.unlink filename;
6621   match Unix.close_process_in chan with
6622   | Unix.WEXITED 0 -> lines
6623   | Unix.WEXITED i ->
6624       failwithf "pod2text: process exited with non-zero status (%d)" i
6625   | Unix.WSIGNALED i | Unix.WSTOPPED i ->
6626       failwithf "pod2text: process signalled or stopped by signal %d" i
6627
6628 (* Generate ruby bindings. *)
6629 and generate_ruby_c () =
6630   generate_header CStyle LGPLv2;
6631
6632   pr "\
6633 #include <stdio.h>
6634 #include <stdlib.h>
6635
6636 #include <ruby.h>
6637
6638 #include \"guestfs.h\"
6639
6640 #include \"extconf.h\"
6641
6642 /* For Ruby < 1.9 */
6643 #ifndef RARRAY_LEN
6644 #define RARRAY_LEN(r) (RARRAY((r))->len)
6645 #endif
6646
6647 static VALUE m_guestfs;                 /* guestfs module */
6648 static VALUE c_guestfs;                 /* guestfs_h handle */
6649 static VALUE e_Error;                   /* used for all errors */
6650
6651 static void ruby_guestfs_free (void *p)
6652 {
6653   if (!p) return;
6654   guestfs_close ((guestfs_h *) p);
6655 }
6656
6657 static VALUE ruby_guestfs_create (VALUE m)
6658 {
6659   guestfs_h *g;
6660
6661   g = guestfs_create ();
6662   if (!g)
6663     rb_raise (e_Error, \"failed to create guestfs handle\");
6664
6665   /* Don't print error messages to stderr by default. */
6666   guestfs_set_error_handler (g, NULL, NULL);
6667
6668   /* Wrap it, and make sure the close function is called when the
6669    * handle goes away.
6670    */
6671   return Data_Wrap_Struct (c_guestfs, NULL, ruby_guestfs_free, g);
6672 }
6673
6674 static VALUE ruby_guestfs_close (VALUE gv)
6675 {
6676   guestfs_h *g;
6677   Data_Get_Struct (gv, guestfs_h, g);
6678
6679   ruby_guestfs_free (g);
6680   DATA_PTR (gv) = NULL;
6681
6682   return Qnil;
6683 }
6684
6685 ";
6686
6687   List.iter (
6688     fun (name, style, _, _, _, _, _) ->
6689       pr "static VALUE ruby_guestfs_%s (VALUE gv" name;
6690       List.iter (fun arg -> pr ", VALUE %sv" (name_of_argt arg)) (snd style);
6691       pr ")\n";
6692       pr "{\n";
6693       pr "  guestfs_h *g;\n";
6694       pr "  Data_Get_Struct (gv, guestfs_h, g);\n";
6695       pr "  if (!g)\n";
6696       pr "    rb_raise (rb_eArgError, \"%%s: used handle after closing it\", \"%s\");\n"
6697         name;
6698       pr "\n";
6699
6700       List.iter (
6701         function
6702         | String n | FileIn n | FileOut n ->
6703             pr "  Check_Type (%sv, T_STRING);\n" n;
6704             pr "  const char *%s = StringValueCStr (%sv);\n" n n;
6705             pr "  if (!%s)\n" n;
6706             pr "    rb_raise (rb_eTypeError, \"expected string for parameter %%s of %%s\",\n";
6707             pr "              \"%s\", \"%s\");\n" n name
6708         | OptString n ->
6709             pr "  const char *%s = !NIL_P (%sv) ? StringValueCStr (%sv) : NULL;\n" n n n
6710         | StringList n ->
6711             pr "  char **%s;\n" n;
6712             pr "  Check_Type (%sv, T_ARRAY);\n" n;
6713             pr "  {\n";
6714             pr "    int i, len;\n";
6715             pr "    len = RARRAY_LEN (%sv);\n" n;
6716             pr "    %s = guestfs_safe_malloc (g, sizeof (char *) * (len+1));\n"
6717               n;
6718             pr "    for (i = 0; i < len; ++i) {\n";
6719             pr "      VALUE v = rb_ary_entry (%sv, i);\n" n;
6720             pr "      %s[i] = StringValueCStr (v);\n" n;
6721             pr "    }\n";
6722             pr "    %s[len] = NULL;\n" n;
6723             pr "  }\n";
6724         | Bool n ->
6725             pr "  int %s = RTEST (%sv);\n" n n
6726         | Int n ->
6727             pr "  int %s = NUM2INT (%sv);\n" n n
6728       ) (snd style);
6729       pr "\n";
6730
6731       let error_code =
6732         match fst style with
6733         | RErr | RInt _ | RBool _ -> pr "  int r;\n"; "-1"
6734         | RInt64 _ -> pr "  int64_t r;\n"; "-1"
6735         | RConstString _ -> pr "  const char *r;\n"; "NULL"
6736         | RString _ -> pr "  char *r;\n"; "NULL"
6737         | RStringList _ | RHashtable _ -> pr "  char **r;\n"; "NULL"
6738         | RIntBool _ -> pr "  struct guestfs_int_bool *r;\n"; "NULL"
6739         | RPVList n -> pr "  struct guestfs_lvm_pv_list *r;\n"; "NULL"
6740         | RVGList n -> pr "  struct guestfs_lvm_vg_list *r;\n"; "NULL"
6741         | RLVList n -> pr "  struct guestfs_lvm_lv_list *r;\n"; "NULL"
6742         | RStat n -> pr "  struct guestfs_stat *r;\n"; "NULL"
6743         | RStatVFS n -> pr "  struct guestfs_statvfs *r;\n"; "NULL" in
6744       pr "\n";
6745
6746       pr "  r = guestfs_%s " name;
6747       generate_call_args ~handle:"g" (snd style);
6748       pr ";\n";
6749
6750       List.iter (
6751         function
6752         | String _ | FileIn _ | FileOut _ | OptString _ | Bool _ | Int _ -> ()
6753         | StringList n ->
6754             pr "  free (%s);\n" n
6755       ) (snd style);
6756
6757       pr "  if (r == %s)\n" error_code;
6758       pr "    rb_raise (e_Error, \"%%s\", guestfs_last_error (g));\n";
6759       pr "\n";
6760
6761       (match fst style with
6762        | RErr ->
6763            pr "  return Qnil;\n"
6764        | RInt _ | RBool _ ->
6765            pr "  return INT2NUM (r);\n"
6766        | RInt64 _ ->
6767            pr "  return ULL2NUM (r);\n"
6768        | RConstString _ ->
6769            pr "  return rb_str_new2 (r);\n";
6770        | RString _ ->
6771            pr "  VALUE rv = rb_str_new2 (r);\n";
6772            pr "  free (r);\n";
6773            pr "  return rv;\n";
6774        | RStringList _ ->
6775            pr "  int i, len = 0;\n";
6776            pr "  for (i = 0; r[i] != NULL; ++i) len++;\n";
6777            pr "  VALUE rv = rb_ary_new2 (len);\n";
6778            pr "  for (i = 0; r[i] != NULL; ++i) {\n";
6779            pr "    rb_ary_push (rv, rb_str_new2 (r[i]));\n";
6780            pr "    free (r[i]);\n";
6781            pr "  }\n";
6782            pr "  free (r);\n";
6783            pr "  return rv;\n"
6784        | RIntBool _ ->
6785            pr "  VALUE rv = rb_ary_new2 (2);\n";
6786            pr "  rb_ary_push (rv, INT2NUM (r->i));\n";
6787            pr "  rb_ary_push (rv, INT2NUM (r->b));\n";
6788            pr "  guestfs_free_int_bool (r);\n";
6789            pr "  return rv;\n"
6790        | RPVList n ->
6791            generate_ruby_lvm_code "pv" pv_cols
6792        | RVGList n ->
6793            generate_ruby_lvm_code "vg" vg_cols
6794        | RLVList n ->
6795            generate_ruby_lvm_code "lv" lv_cols
6796        | RStat n ->
6797            pr "  VALUE rv = rb_hash_new ();\n";
6798            List.iter (
6799              function
6800              | name, `Int ->
6801                  pr "  rb_hash_aset (rv, rb_str_new2 (\"%s\"), ULL2NUM (r->%s));\n" name name
6802            ) stat_cols;
6803            pr "  free (r);\n";
6804            pr "  return rv;\n"
6805        | RStatVFS n ->
6806            pr "  VALUE rv = rb_hash_new ();\n";
6807            List.iter (
6808              function
6809              | name, `Int ->
6810                  pr "  rb_hash_aset (rv, rb_str_new2 (\"%s\"), ULL2NUM (r->%s));\n" name name
6811            ) statvfs_cols;
6812            pr "  free (r);\n";
6813            pr "  return rv;\n"
6814        | RHashtable _ ->
6815            pr "  VALUE rv = rb_hash_new ();\n";
6816            pr "  int i;\n";
6817            pr "  for (i = 0; r[i] != NULL; i+=2) {\n";
6818            pr "    rb_hash_aset (rv, rb_str_new2 (r[i]), rb_str_new2 (r[i+1]));\n";
6819            pr "    free (r[i]);\n";
6820            pr "    free (r[i+1]);\n";
6821            pr "  }\n";
6822            pr "  free (r);\n";
6823            pr "  return rv;\n"
6824       );
6825
6826       pr "}\n";
6827       pr "\n"
6828   ) all_functions;
6829
6830   pr "\
6831 /* Initialize the module. */
6832 void Init__guestfs ()
6833 {
6834   m_guestfs = rb_define_module (\"Guestfs\");
6835   c_guestfs = rb_define_class_under (m_guestfs, \"Guestfs\", rb_cObject);
6836   e_Error = rb_define_class_under (m_guestfs, \"Error\", rb_eStandardError);
6837
6838   rb_define_module_function (m_guestfs, \"create\", ruby_guestfs_create, 0);
6839   rb_define_method (c_guestfs, \"close\", ruby_guestfs_close, 0);
6840
6841 ";
6842   (* Define the rest of the methods. *)
6843   List.iter (
6844     fun (name, style, _, _, _, _, _) ->
6845       pr "  rb_define_method (c_guestfs, \"%s\",\n" name;
6846       pr "        ruby_guestfs_%s, %d);\n" name (List.length (snd style))
6847   ) all_functions;
6848
6849   pr "}\n"
6850
6851 (* Ruby code to return an LVM struct list. *)
6852 and generate_ruby_lvm_code typ cols =
6853   pr "  VALUE rv = rb_ary_new2 (r->len);\n";
6854   pr "  int i;\n";
6855   pr "  for (i = 0; i < r->len; ++i) {\n";
6856   pr "    VALUE hv = rb_hash_new ();\n";
6857   List.iter (
6858     function
6859     | name, `String ->
6860         pr "    rb_hash_aset (rv, rb_str_new2 (\"%s\"), rb_str_new2 (r->val[i].%s));\n" name name
6861     | name, `UUID ->
6862         pr "    rb_hash_aset (rv, rb_str_new2 (\"%s\"), rb_str_new (r->val[i].%s, 32));\n" name name
6863     | name, `Bytes
6864     | name, `Int ->
6865         pr "    rb_hash_aset (rv, rb_str_new2 (\"%s\"), ULL2NUM (r->val[i].%s));\n" name name
6866     | name, `OptPercent ->
6867         pr "    rb_hash_aset (rv, rb_str_new2 (\"%s\"), rb_dbl2big (r->val[i].%s));\n" name name
6868   ) cols;
6869   pr "    rb_ary_push (rv, hv);\n";
6870   pr "  }\n";
6871   pr "  guestfs_free_lvm_%s_list (r);\n" typ;
6872   pr "  return rv;\n"
6873
6874 (* Generate Java bindings GuestFS.java file. *)
6875 and generate_java_java () =
6876   generate_header CStyle LGPLv2;
6877
6878   pr "\
6879 package com.redhat.et.libguestfs;
6880
6881 import java.util.HashMap;
6882 import com.redhat.et.libguestfs.LibGuestFSException;
6883 import com.redhat.et.libguestfs.PV;
6884 import com.redhat.et.libguestfs.VG;
6885 import com.redhat.et.libguestfs.LV;
6886 import com.redhat.et.libguestfs.Stat;
6887 import com.redhat.et.libguestfs.StatVFS;
6888 import com.redhat.et.libguestfs.IntBool;
6889
6890 /**
6891  * The GuestFS object is a libguestfs handle.
6892  *
6893  * @author rjones
6894  */
6895 public class GuestFS {
6896   // Load the native code.
6897   static {
6898     System.loadLibrary (\"guestfs_jni\");
6899   }
6900
6901   /**
6902    * The native guestfs_h pointer.
6903    */
6904   long g;
6905
6906   /**
6907    * Create a libguestfs handle.
6908    *
6909    * @throws LibGuestFSException
6910    */
6911   public GuestFS () throws LibGuestFSException
6912   {
6913     g = _create ();
6914   }
6915   private native long _create () throws LibGuestFSException;
6916
6917   /**
6918    * Close a libguestfs handle.
6919    *
6920    * You can also leave handles to be collected by the garbage
6921    * collector, but this method ensures that the resources used
6922    * by the handle are freed up immediately.  If you call any
6923    * other methods after closing the handle, you will get an
6924    * exception.
6925    *
6926    * @throws LibGuestFSException
6927    */
6928   public void close () throws LibGuestFSException
6929   {
6930     if (g != 0)
6931       _close (g);
6932     g = 0;
6933   }
6934   private native void _close (long g) throws LibGuestFSException;
6935
6936   public void finalize () throws LibGuestFSException
6937   {
6938     close ();
6939   }
6940
6941 ";
6942
6943   List.iter (
6944     fun (name, style, _, flags, _, shortdesc, longdesc) ->
6945       if not (List.mem NotInDocs flags); then (
6946         let doc = replace_str longdesc "C<guestfs_" "C<g." in
6947         let doc =
6948           if List.mem ProtocolLimitWarning flags then
6949             doc ^ "\n\n" ^ protocol_limit_warning
6950           else doc in
6951         let doc =
6952           if List.mem DangerWillRobinson flags then
6953             doc ^ "\n\n" ^ danger_will_robinson
6954           else doc in
6955         let doc = pod2text ~width:60 name doc in
6956         let doc = List.map (            (* RHBZ#501883 *)
6957           function
6958           | "" -> "<p>"
6959           | nonempty -> nonempty
6960         ) doc in
6961         let doc = String.concat "\n   * " doc in
6962
6963         pr "  /**\n";
6964         pr "   * %s\n" shortdesc;
6965         pr "   * <p>\n";
6966         pr "   * %s\n" doc;
6967         pr "   * @throws LibGuestFSException\n";
6968         pr "   */\n";
6969         pr "  ";
6970       );
6971       generate_java_prototype ~public:true ~semicolon:false name style;
6972       pr "\n";
6973       pr "  {\n";
6974       pr "    if (g == 0)\n";
6975       pr "      throw new LibGuestFSException (\"%s: handle is closed\");\n"
6976         name;
6977       pr "    ";
6978       if fst style <> RErr then pr "return ";
6979       pr "_%s " name;
6980       generate_call_args ~handle:"g" (snd style);
6981       pr ";\n";
6982       pr "  }\n";
6983       pr "  ";
6984       generate_java_prototype ~privat:true ~native:true name style;
6985       pr "\n";
6986       pr "\n";
6987   ) all_functions;
6988
6989   pr "}\n"
6990
6991 and generate_java_prototype ?(public=false) ?(privat=false) ?(native=false)
6992     ?(semicolon=true) name style =
6993   if privat then pr "private ";
6994   if public then pr "public ";
6995   if native then pr "native ";
6996
6997   (* return type *)
6998   (match fst style with
6999    | RErr -> pr "void ";
7000    | RInt _ -> pr "int ";
7001    | RInt64 _ -> pr "long ";
7002    | RBool _ -> pr "boolean ";
7003    | RConstString _ | RString _ -> pr "String ";
7004    | RStringList _ -> pr "String[] ";
7005    | RIntBool _ -> pr "IntBool ";
7006    | RPVList _ -> pr "PV[] ";
7007    | RVGList _ -> pr "VG[] ";
7008    | RLVList _ -> pr "LV[] ";
7009    | RStat _ -> pr "Stat ";
7010    | RStatVFS _ -> pr "StatVFS ";
7011    | RHashtable _ -> pr "HashMap<String,String> ";
7012   );
7013
7014   if native then pr "_%s " name else pr "%s " name;
7015   pr "(";
7016   let needs_comma = ref false in
7017   if native then (
7018     pr "long g";
7019     needs_comma := true
7020   );
7021
7022   (* args *)
7023   List.iter (
7024     fun arg ->
7025       if !needs_comma then pr ", ";
7026       needs_comma := true;
7027
7028       match arg with
7029       | String n
7030       | OptString n
7031       | FileIn n
7032       | FileOut n ->
7033           pr "String %s" n
7034       | StringList n ->
7035           pr "String[] %s" n
7036       | Bool n ->
7037           pr "boolean %s" n
7038       | Int n ->
7039           pr "int %s" n
7040   ) (snd style);
7041
7042   pr ")\n";
7043   pr "    throws LibGuestFSException";
7044   if semicolon then pr ";"
7045
7046 and generate_java_struct typ cols =
7047   generate_header CStyle LGPLv2;
7048
7049   pr "\
7050 package com.redhat.et.libguestfs;
7051
7052 /**
7053  * Libguestfs %s structure.
7054  *
7055  * @author rjones
7056  * @see GuestFS
7057  */
7058 public class %s {
7059 " typ typ;
7060
7061   List.iter (
7062     function
7063     | name, `String
7064     | name, `UUID -> pr "  public String %s;\n" name
7065     | name, `Bytes
7066     | name, `Int -> pr "  public long %s;\n" name
7067     | name, `OptPercent ->
7068         pr "  /* The next field is [0..100] or -1 meaning 'not present': */\n";
7069         pr "  public float %s;\n" name
7070   ) cols;
7071
7072   pr "}\n"
7073
7074 and generate_java_c () =
7075   generate_header CStyle LGPLv2;
7076
7077   pr "\
7078 #include <stdio.h>
7079 #include <stdlib.h>
7080 #include <string.h>
7081
7082 #include \"com_redhat_et_libguestfs_GuestFS.h\"
7083 #include \"guestfs.h\"
7084
7085 /* Note that this function returns.  The exception is not thrown
7086  * until after the wrapper function returns.
7087  */
7088 static void
7089 throw_exception (JNIEnv *env, const char *msg)
7090 {
7091   jclass cl;
7092   cl = (*env)->FindClass (env,
7093                           \"com/redhat/et/libguestfs/LibGuestFSException\");
7094   (*env)->ThrowNew (env, cl, msg);
7095 }
7096
7097 JNIEXPORT jlong JNICALL
7098 Java_com_redhat_et_libguestfs_GuestFS__1create
7099   (JNIEnv *env, jobject obj)
7100 {
7101   guestfs_h *g;
7102
7103   g = guestfs_create ();
7104   if (g == NULL) {
7105     throw_exception (env, \"GuestFS.create: failed to allocate handle\");
7106     return 0;
7107   }
7108   guestfs_set_error_handler (g, NULL, NULL);
7109   return (jlong) (long) g;
7110 }
7111
7112 JNIEXPORT void JNICALL
7113 Java_com_redhat_et_libguestfs_GuestFS__1close
7114   (JNIEnv *env, jobject obj, jlong jg)
7115 {
7116   guestfs_h *g = (guestfs_h *) (long) jg;
7117   guestfs_close (g);
7118 }
7119
7120 ";
7121
7122   List.iter (
7123     fun (name, style, _, _, _, _, _) ->
7124       pr "JNIEXPORT ";
7125       (match fst style with
7126        | RErr -> pr "void ";
7127        | RInt _ -> pr "jint ";
7128        | RInt64 _ -> pr "jlong ";
7129        | RBool _ -> pr "jboolean ";
7130        | RConstString _ | RString _ -> pr "jstring ";
7131        | RIntBool _ | RStat _ | RStatVFS _ | RHashtable _ ->
7132            pr "jobject ";
7133        | RStringList _ | RPVList _ | RVGList _ | RLVList _ ->
7134            pr "jobjectArray ";
7135       );
7136       pr "JNICALL\n";
7137       pr "Java_com_redhat_et_libguestfs_GuestFS_";
7138       pr "%s" (replace_str ("_" ^ name) "_" "_1");
7139       pr "\n";
7140       pr "  (JNIEnv *env, jobject obj, jlong jg";
7141       List.iter (
7142         function
7143         | String n
7144         | OptString n
7145         | FileIn n
7146         | FileOut n ->
7147             pr ", jstring j%s" n
7148         | StringList n ->
7149             pr ", jobjectArray j%s" n
7150         | Bool n ->
7151             pr ", jboolean j%s" n
7152         | Int n ->
7153             pr ", jint j%s" n
7154       ) (snd style);
7155       pr ")\n";
7156       pr "{\n";
7157       pr "  guestfs_h *g = (guestfs_h *) (long) jg;\n";
7158       let error_code, no_ret =
7159         match fst style with
7160         | RErr -> pr "  int r;\n"; "-1", ""
7161         | RBool _
7162         | RInt _ -> pr "  int r;\n"; "-1", "0"
7163         | RInt64 _ -> pr "  int64_t r;\n"; "-1", "0"
7164         | RConstString _ -> pr "  const char *r;\n"; "NULL", "NULL"
7165         | RString _ ->
7166             pr "  jstring jr;\n";
7167             pr "  char *r;\n"; "NULL", "NULL"
7168         | RStringList _ ->
7169             pr "  jobjectArray jr;\n";
7170             pr "  int r_len;\n";
7171             pr "  jclass cl;\n";
7172             pr "  jstring jstr;\n";
7173             pr "  char **r;\n"; "NULL", "NULL"
7174         | RIntBool _ ->
7175             pr "  jobject jr;\n";
7176             pr "  jclass cl;\n";
7177             pr "  jfieldID fl;\n";
7178             pr "  struct guestfs_int_bool *r;\n"; "NULL", "NULL"
7179         | RStat _ ->
7180             pr "  jobject jr;\n";
7181             pr "  jclass cl;\n";
7182             pr "  jfieldID fl;\n";
7183             pr "  struct guestfs_stat *r;\n"; "NULL", "NULL"
7184         | RStatVFS _ ->
7185             pr "  jobject jr;\n";
7186             pr "  jclass cl;\n";
7187             pr "  jfieldID fl;\n";
7188             pr "  struct guestfs_statvfs *r;\n"; "NULL", "NULL"
7189         | RPVList _ ->
7190             pr "  jobjectArray jr;\n";
7191             pr "  jclass cl;\n";
7192             pr "  jfieldID fl;\n";
7193             pr "  jobject jfl;\n";
7194             pr "  struct guestfs_lvm_pv_list *r;\n"; "NULL", "NULL"
7195         | RVGList _ ->
7196             pr "  jobjectArray jr;\n";
7197             pr "  jclass cl;\n";
7198             pr "  jfieldID fl;\n";
7199             pr "  jobject jfl;\n";
7200             pr "  struct guestfs_lvm_vg_list *r;\n"; "NULL", "NULL"
7201         | RLVList _ ->
7202             pr "  jobjectArray jr;\n";
7203             pr "  jclass cl;\n";
7204             pr "  jfieldID fl;\n";
7205             pr "  jobject jfl;\n";
7206             pr "  struct guestfs_lvm_lv_list *r;\n"; "NULL", "NULL"
7207         | RHashtable _ -> pr "  char **r;\n"; "NULL", "NULL" in
7208       List.iter (
7209         function
7210         | String n
7211         | OptString n
7212         | FileIn n
7213         | FileOut n ->
7214             pr "  const char *%s;\n" n
7215         | StringList n ->
7216             pr "  int %s_len;\n" n;
7217             pr "  const char **%s;\n" n
7218         | Bool n
7219         | Int n ->
7220             pr "  int %s;\n" n
7221       ) (snd style);
7222
7223       let needs_i =
7224         (match fst style with
7225          | RStringList _ | RPVList _ | RVGList _ | RLVList _ -> true
7226          | RErr | RBool _ | RInt _ | RInt64 _ | RConstString _
7227          | RString _ | RIntBool _ | RStat _ | RStatVFS _
7228          | RHashtable _ -> false) ||
7229         List.exists (function StringList _ -> true | _ -> false) (snd style) in
7230       if needs_i then
7231         pr "  int i;\n";
7232
7233       pr "\n";
7234
7235       (* Get the parameters. *)
7236       List.iter (
7237         function
7238         | String n
7239         | FileIn n
7240         | FileOut n ->
7241             pr "  %s = (*env)->GetStringUTFChars (env, j%s, NULL);\n" n n
7242         | OptString n ->
7243             (* This is completely undocumented, but Java null becomes
7244              * a NULL parameter.
7245              *)
7246             pr "  %s = j%s ? (*env)->GetStringUTFChars (env, j%s, NULL) : NULL;\n" n n n
7247         | StringList n ->
7248             pr "  %s_len = (*env)->GetArrayLength (env, j%s);\n" n n;
7249             pr "  %s = guestfs_safe_malloc (g, sizeof (char *) * (%s_len+1));\n" n n;
7250             pr "  for (i = 0; i < %s_len; ++i) {\n" n;
7251             pr "    jobject o = (*env)->GetObjectArrayElement (env, j%s, i);\n"
7252               n;
7253             pr "    %s[i] = (*env)->GetStringUTFChars (env, o, NULL);\n" n;
7254             pr "  }\n";
7255             pr "  %s[%s_len] = NULL;\n" n n;
7256         | Bool n
7257         | Int n ->
7258             pr "  %s = j%s;\n" n n
7259       ) (snd style);
7260
7261       (* Make the call. *)
7262       pr "  r = guestfs_%s " name;
7263       generate_call_args ~handle:"g" (snd style);
7264       pr ";\n";
7265
7266       (* Release the parameters. *)
7267       List.iter (
7268         function
7269         | String n
7270         | FileIn n
7271         | FileOut n ->
7272             pr "  (*env)->ReleaseStringUTFChars (env, j%s, %s);\n" n n
7273         | OptString n ->
7274             pr "  if (j%s)\n" n;
7275             pr "    (*env)->ReleaseStringUTFChars (env, j%s, %s);\n" n n
7276         | StringList n ->
7277             pr "  for (i = 0; i < %s_len; ++i) {\n" n;
7278             pr "    jobject o = (*env)->GetObjectArrayElement (env, j%s, i);\n"
7279               n;
7280             pr "    (*env)->ReleaseStringUTFChars (env, o, %s[i]);\n" n;
7281             pr "  }\n";
7282             pr "  free (%s);\n" n
7283         | Bool n
7284         | Int n -> ()
7285       ) (snd style);
7286
7287       (* Check for errors. *)
7288       pr "  if (r == %s) {\n" error_code;
7289       pr "    throw_exception (env, guestfs_last_error (g));\n";
7290       pr "    return %s;\n" no_ret;
7291       pr "  }\n";
7292
7293       (* Return value. *)
7294       (match fst style with
7295        | RErr -> ()
7296        | RInt _ -> pr "  return (jint) r;\n"
7297        | RBool _ -> pr "  return (jboolean) r;\n"
7298        | RInt64 _ -> pr "  return (jlong) r;\n"
7299        | RConstString _ -> pr "  return (*env)->NewStringUTF (env, r);\n"
7300        | RString _ ->
7301            pr "  jr = (*env)->NewStringUTF (env, r);\n";
7302            pr "  free (r);\n";
7303            pr "  return jr;\n"
7304        | RStringList _ ->
7305            pr "  for (r_len = 0; r[r_len] != NULL; ++r_len) ;\n";
7306            pr "  cl = (*env)->FindClass (env, \"java/lang/String\");\n";
7307            pr "  jstr = (*env)->NewStringUTF (env, \"\");\n";
7308            pr "  jr = (*env)->NewObjectArray (env, r_len, cl, jstr);\n";
7309            pr "  for (i = 0; i < r_len; ++i) {\n";
7310            pr "    jstr = (*env)->NewStringUTF (env, r[i]);\n";
7311            pr "    (*env)->SetObjectArrayElement (env, jr, i, jstr);\n";
7312            pr "    free (r[i]);\n";
7313            pr "  }\n";
7314            pr "  free (r);\n";
7315            pr "  return jr;\n"
7316        | RIntBool _ ->
7317            pr "  cl = (*env)->FindClass (env, \"com/redhat/et/libguestfs/IntBool\");\n";
7318            pr "  jr = (*env)->AllocObject (env, cl);\n";
7319            pr "  fl = (*env)->GetFieldID (env, cl, \"i\", \"I\");\n";
7320            pr "  (*env)->SetIntField (env, jr, fl, r->i);\n";
7321            pr "  fl = (*env)->GetFieldID (env, cl, \"i\", \"Z\");\n";
7322            pr "  (*env)->SetBooleanField (env, jr, fl, r->b);\n";
7323            pr "  guestfs_free_int_bool (r);\n";
7324            pr "  return jr;\n"
7325        | RStat _ ->
7326            pr "  cl = (*env)->FindClass (env, \"com/redhat/et/libguestfs/Stat\");\n";
7327            pr "  jr = (*env)->AllocObject (env, cl);\n";
7328            List.iter (
7329              function
7330              | name, `Int ->
7331                  pr "  fl = (*env)->GetFieldID (env, cl, \"%s\", \"J\");\n"
7332                    name;
7333                  pr "  (*env)->SetLongField (env, jr, fl, r->%s);\n" name;
7334            ) stat_cols;
7335            pr "  free (r);\n";
7336            pr "  return jr;\n"
7337        | RStatVFS _ ->
7338            pr "  cl = (*env)->FindClass (env, \"com/redhat/et/libguestfs/StatVFS\");\n";
7339            pr "  jr = (*env)->AllocObject (env, cl);\n";
7340            List.iter (
7341              function
7342              | name, `Int ->
7343                  pr "  fl = (*env)->GetFieldID (env, cl, \"%s\", \"J\");\n"
7344                    name;
7345                  pr "  (*env)->SetLongField (env, jr, fl, r->%s);\n" name;
7346            ) statvfs_cols;
7347            pr "  free (r);\n";
7348            pr "  return jr;\n"
7349        | RPVList _ ->
7350            generate_java_lvm_return "pv" "PV" pv_cols
7351        | RVGList _ ->
7352            generate_java_lvm_return "vg" "VG" vg_cols
7353        | RLVList _ ->
7354            generate_java_lvm_return "lv" "LV" lv_cols
7355        | RHashtable _ ->
7356            (* XXX *)
7357            pr "  throw_exception (env, \"%s: internal error: please let us know how to make a Java HashMap from JNI bindings!\");\n" name;
7358            pr "  return NULL;\n"
7359       );
7360
7361       pr "}\n";
7362       pr "\n"
7363   ) all_functions
7364
7365 and generate_java_lvm_return typ jtyp cols =
7366   pr "  cl = (*env)->FindClass (env, \"com/redhat/et/libguestfs/%s\");\n" jtyp;
7367   pr "  jr = (*env)->NewObjectArray (env, r->len, cl, NULL);\n";
7368   pr "  for (i = 0; i < r->len; ++i) {\n";
7369   pr "    jfl = (*env)->AllocObject (env, cl);\n";
7370   List.iter (
7371     function
7372     | name, `String ->
7373         pr "    fl = (*env)->GetFieldID (env, cl, \"%s\", \"Ljava/lang/String;\");\n" name;
7374         pr "    (*env)->SetObjectField (env, jfl, fl, (*env)->NewStringUTF (env, r->val[i].%s));\n" name;
7375     | name, `UUID ->
7376         pr "    {\n";
7377         pr "      char s[33];\n";
7378         pr "      memcpy (s, r->val[i].%s, 32);\n" name;
7379         pr "      s[32] = 0;\n";
7380         pr "      fl = (*env)->GetFieldID (env, cl, \"%s\", \"Ljava/lang/String;\");\n" name;
7381         pr "      (*env)->SetObjectField (env, jfl, fl, (*env)->NewStringUTF (env, s));\n";
7382         pr "    }\n";
7383     | name, (`Bytes|`Int) ->
7384         pr "    fl = (*env)->GetFieldID (env, cl, \"%s\", \"J\");\n" name;
7385         pr "    (*env)->SetLongField (env, jfl, fl, r->val[i].%s);\n" name;
7386     | name, `OptPercent ->
7387         pr "    fl = (*env)->GetFieldID (env, cl, \"%s\", \"F\");\n" name;
7388         pr "    (*env)->SetFloatField (env, jfl, fl, r->val[i].%s);\n" name;
7389   ) cols;
7390   pr "    (*env)->SetObjectArrayElement (env, jfl, i, jfl);\n";
7391   pr "  }\n";
7392   pr "  guestfs_free_lvm_%s_list (r);\n" typ;
7393   pr "  return jr;\n"
7394
7395 and generate_haskell_hs () =
7396   generate_header HaskellStyle LGPLv2;
7397
7398   (* XXX We only know how to generate partial FFI for Haskell
7399    * at the moment.  Please help out!
7400    *)
7401   let can_generate style =
7402     match style with
7403     | RErr, _
7404     | RInt _, _
7405     | RInt64 _, _ -> true
7406     | RBool _, _
7407     | RConstString _, _
7408     | RString _, _
7409     | RStringList _, _
7410     | RIntBool _, _
7411     | RPVList _, _
7412     | RVGList _, _
7413     | RLVList _, _
7414     | RStat _, _
7415     | RStatVFS _, _
7416     | RHashtable _, _ -> false in
7417
7418   pr "\
7419 {-# INCLUDE <guestfs.h> #-}
7420 {-# LANGUAGE ForeignFunctionInterface #-}
7421
7422 module Guestfs (
7423   create";
7424
7425   (* List out the names of the actions we want to export. *)
7426   List.iter (
7427     fun (name, style, _, _, _, _, _) ->
7428       if can_generate style then pr ",\n  %s" name
7429   ) all_functions;
7430
7431   pr "
7432   ) where
7433 import Foreign
7434 import Foreign.C
7435 import Foreign.C.Types
7436 import IO
7437 import Control.Exception
7438 import Data.Typeable
7439
7440 data GuestfsS = GuestfsS            -- represents the opaque C struct
7441 type GuestfsP = Ptr GuestfsS        -- guestfs_h *
7442 type GuestfsH = ForeignPtr GuestfsS -- guestfs_h * with attached finalizer
7443
7444 -- XXX define properly later XXX
7445 data PV = PV
7446 data VG = VG
7447 data LV = LV
7448 data IntBool = IntBool
7449 data Stat = Stat
7450 data StatVFS = StatVFS
7451 data Hashtable = Hashtable
7452
7453 foreign import ccall unsafe \"guestfs_create\" c_create
7454   :: IO GuestfsP
7455 foreign import ccall unsafe \"&guestfs_close\" c_close
7456   :: FunPtr (GuestfsP -> IO ())
7457 foreign import ccall unsafe \"guestfs_set_error_handler\" c_set_error_handler
7458   :: GuestfsP -> Ptr CInt -> Ptr CInt -> IO ()
7459
7460 create :: IO GuestfsH
7461 create = do
7462   p <- c_create
7463   c_set_error_handler p nullPtr nullPtr
7464   h <- newForeignPtr c_close p
7465   return h
7466
7467 foreign import ccall unsafe \"guestfs_last_error\" c_last_error
7468   :: GuestfsP -> IO CString
7469
7470 -- last_error :: GuestfsH -> IO (Maybe String)
7471 -- last_error h = do
7472 --   str <- withForeignPtr h (\\p -> c_last_error p)
7473 --   maybePeek peekCString str
7474
7475 last_error :: GuestfsH -> IO (String)
7476 last_error h = do
7477   str <- withForeignPtr h (\\p -> c_last_error p)
7478   if (str == nullPtr)
7479     then return \"no error\"
7480     else peekCString str
7481
7482 ";
7483
7484   (* Generate wrappers for each foreign function. *)
7485   List.iter (
7486     fun (name, style, _, _, _, _, _) ->
7487       if can_generate style then (
7488         pr "foreign import ccall unsafe \"guestfs_%s\" c_%s\n" name name;
7489         pr "  :: ";
7490         generate_haskell_prototype ~handle:"GuestfsP" style;
7491         pr "\n";
7492         pr "\n";
7493         pr "%s :: " name;
7494         generate_haskell_prototype ~handle:"GuestfsH" ~hs:true style;
7495         pr "\n";
7496         pr "%s %s = do\n" name
7497           (String.concat " " ("h" :: List.map name_of_argt (snd style)));
7498         pr "  r <- ";
7499         (* Convert pointer arguments using with* functions. *)
7500         List.iter (
7501           function
7502           | FileIn n
7503           | FileOut n
7504           | String n -> pr "withCString %s $ \\%s -> " n n
7505           | OptString n -> pr "maybeWith withCString %s $ \\%s -> " n n
7506           | StringList n -> pr "withMany withCString %s $ \\%s -> withArray0 nullPtr %s $ \\%s -> " n n n n
7507           | Bool _ | Int _ -> ()
7508         ) (snd style);
7509         (* Convert integer arguments. *)
7510         let args =
7511           List.map (
7512             function
7513             | Bool n -> sprintf "(fromBool %s)" n
7514             | Int n -> sprintf "(fromIntegral %s)" n
7515             | FileIn n | FileOut n | String n | OptString n | StringList n -> n
7516           ) (snd style) in
7517         pr "withForeignPtr h (\\p -> c_%s %s)\n" name
7518           (String.concat " " ("p" :: args));
7519         (match fst style with
7520          | RErr | RInt _ | RInt64 _ | RBool _ ->
7521              pr "  if (r == -1)\n";
7522              pr "    then do\n";
7523              pr "      err <- last_error h\n";
7524              pr "      fail err\n";
7525          | RConstString _ | RString _ | RStringList _ | RIntBool _
7526          | RPVList _ | RVGList _ | RLVList _ | RStat _ | RStatVFS _
7527          | RHashtable _ ->
7528              pr "  if (r == nullPtr)\n";
7529              pr "    then do\n";
7530              pr "      err <- last_error h\n";
7531              pr "      fail err\n";
7532         );
7533         (match fst style with
7534          | RErr ->
7535              pr "    else return ()\n"
7536          | RInt _ ->
7537              pr "    else return (fromIntegral r)\n"
7538          | RInt64 _ ->
7539              pr "    else return (fromIntegral r)\n"
7540          | RBool _ ->
7541              pr "    else return (toBool r)\n"
7542          | RConstString _
7543          | RString _
7544          | RStringList _
7545          | RIntBool _
7546          | RPVList _
7547          | RVGList _
7548          | RLVList _
7549          | RStat _
7550          | RStatVFS _
7551          | RHashtable _ ->
7552              pr "    else return ()\n" (* XXXXXXXXXXXXXXXXXXXX *)
7553         );
7554         pr "\n";
7555       )
7556   ) all_functions
7557
7558 and generate_haskell_prototype ~handle ?(hs = false) style =
7559   pr "%s -> " handle;
7560   let string = if hs then "String" else "CString" in
7561   let int = if hs then "Int" else "CInt" in
7562   let bool = if hs then "Bool" else "CInt" in
7563   let int64 = if hs then "Integer" else "Int64" in
7564   List.iter (
7565     fun arg ->
7566       (match arg with
7567        | String _ -> pr "%s" string
7568        | OptString _ -> if hs then pr "Maybe String" else pr "CString"
7569        | StringList _ -> if hs then pr "[String]" else pr "Ptr CString"
7570        | Bool _ -> pr "%s" bool
7571        | Int _ -> pr "%s" int
7572        | FileIn _ -> pr "%s" string
7573        | FileOut _ -> pr "%s" string
7574       );
7575       pr " -> ";
7576   ) (snd style);
7577   pr "IO (";
7578   (match fst style with
7579    | RErr -> if not hs then pr "CInt"
7580    | RInt _ -> pr "%s" int
7581    | RInt64 _ -> pr "%s" int64
7582    | RBool _ -> pr "%s" bool
7583    | RConstString _ -> pr "%s" string
7584    | RString _ -> pr "%s" string
7585    | RStringList _ -> pr "[%s]" string
7586    | RIntBool _ -> pr "IntBool"
7587    | RPVList _ -> pr "[PV]"
7588    | RVGList _ -> pr "[VG]"
7589    | RLVList _ -> pr "[LV]"
7590    | RStat _ -> pr "Stat"
7591    | RStatVFS _ -> pr "StatVFS"
7592    | RHashtable _ -> pr "Hashtable"
7593   );
7594   pr ")"
7595
7596 and generate_bindtests () =
7597   generate_header CStyle LGPLv2;
7598
7599   pr "\
7600 #include <stdio.h>
7601 #include <stdlib.h>
7602 #include <inttypes.h>
7603 #include <string.h>
7604
7605 #include \"guestfs.h\"
7606 #include \"guestfs_protocol.h\"
7607
7608 #define error guestfs_error
7609
7610 static void
7611 print_strings (char * const* const argv)
7612 {
7613   int argc;
7614
7615   printf (\"[\");
7616   for (argc = 0; argv[argc] != NULL; ++argc) {
7617     if (argc > 0) printf (\", \");
7618     printf (\"\\\"%%s\\\"\", argv[argc]);
7619   }
7620   printf (\"]\\n\");
7621 }
7622
7623 /* The test0 function prints its parameters to stdout. */
7624 ";
7625
7626   let test0, tests =
7627     match test_functions with
7628     | [] -> assert false
7629     | test0 :: tests -> test0, tests in
7630
7631   let () =
7632     let (name, style, _, _, _, _, _) = test0 in
7633     generate_prototype ~extern:false ~semicolon:false ~newline:true
7634       ~handle:"g" ~prefix:"guestfs_" name style;
7635     pr "{\n";
7636     List.iter (
7637       function
7638       | String n
7639       | FileIn n
7640       | FileOut n -> pr "  printf (\"%%s\\n\", %s);\n" n
7641       | OptString n -> pr "  printf (\"%%s\\n\", %s ? %s : \"null\");\n" n n
7642       | StringList n -> pr "  print_strings (%s);\n" n
7643       | Bool n -> pr "  printf (\"%%s\\n\", %s ? \"true\" : \"false\");\n" n
7644       | Int n -> pr "  printf (\"%%d\\n\", %s);\n" n
7645     ) (snd style);
7646     pr "  /* Java changes stdout line buffering so we need this: */\n";
7647     pr "  fflush (stdout);\n";
7648     pr "  return 0;\n";
7649     pr "}\n";
7650     pr "\n" in
7651
7652   List.iter (
7653     fun (name, style, _, _, _, _, _) ->
7654       if String.sub name (String.length name - 3) 3 <> "err" then (
7655         pr "/* Test normal return. */\n";
7656         generate_prototype ~extern:false ~semicolon:false ~newline:true
7657           ~handle:"g" ~prefix:"guestfs_" name style;
7658         pr "{\n";
7659         (match fst style with
7660          | RErr ->
7661              pr "  return 0;\n"
7662          | RInt _ ->
7663              pr "  int r;\n";
7664              pr "  sscanf (val, \"%%d\", &r);\n";
7665              pr "  return r;\n"
7666          | RInt64 _ ->
7667              pr "  int64_t r;\n";
7668              pr "  sscanf (val, \"%%\" SCNi64, &r);\n";
7669              pr "  return r;\n"
7670          | RBool _ ->
7671              pr "  return strcmp (val, \"true\") == 0;\n"
7672          | RConstString _ ->
7673              (* Can't return the input string here.  Return a static
7674               * string so we ensure we get a segfault if the caller
7675               * tries to free it.
7676               *)
7677              pr "  return \"static string\";\n"
7678          | RString _ ->
7679              pr "  return strdup (val);\n"
7680          | RStringList _ ->
7681              pr "  char **strs;\n";
7682              pr "  int n, i;\n";
7683              pr "  sscanf (val, \"%%d\", &n);\n";
7684              pr "  strs = malloc ((n+1) * sizeof (char *));\n";
7685              pr "  for (i = 0; i < n; ++i) {\n";
7686              pr "    strs[i] = malloc (16);\n";
7687              pr "    snprintf (strs[i], 16, \"%%d\", i);\n";
7688              pr "  }\n";
7689              pr "  strs[n] = NULL;\n";
7690              pr "  return strs;\n"
7691          | RIntBool _ ->
7692              pr "  struct guestfs_int_bool *r;\n";
7693              pr "  r = malloc (sizeof (struct guestfs_int_bool));\n";
7694              pr "  sscanf (val, \"%%\" SCNi32, &r->i);\n";
7695              pr "  r->b = 0;\n";
7696              pr "  return r;\n"
7697          | RPVList _ ->
7698              pr "  struct guestfs_lvm_pv_list *r;\n";
7699              pr "  int i;\n";
7700              pr "  r = malloc (sizeof (struct guestfs_lvm_pv_list));\n";
7701              pr "  sscanf (val, \"%%d\", &r->len);\n";
7702              pr "  r->val = calloc (r->len, sizeof (struct guestfs_lvm_pv));\n";
7703              pr "  for (i = 0; i < r->len; ++i) {\n";
7704              pr "    r->val[i].pv_name = malloc (16);\n";
7705              pr "    snprintf (r->val[i].pv_name, 16, \"%%d\", i);\n";
7706              pr "  }\n";
7707              pr "  return r;\n"
7708          | RVGList _ ->
7709              pr "  struct guestfs_lvm_vg_list *r;\n";
7710              pr "  int i;\n";
7711              pr "  r = malloc (sizeof (struct guestfs_lvm_vg_list));\n";
7712              pr "  sscanf (val, \"%%d\", &r->len);\n";
7713              pr "  r->val = calloc (r->len, sizeof (struct guestfs_lvm_vg));\n";
7714              pr "  for (i = 0; i < r->len; ++i) {\n";
7715              pr "    r->val[i].vg_name = malloc (16);\n";
7716              pr "    snprintf (r->val[i].vg_name, 16, \"%%d\", i);\n";
7717              pr "  }\n";
7718              pr "  return r;\n"
7719          | RLVList _ ->
7720              pr "  struct guestfs_lvm_lv_list *r;\n";
7721              pr "  int i;\n";
7722              pr "  r = malloc (sizeof (struct guestfs_lvm_lv_list));\n";
7723              pr "  sscanf (val, \"%%d\", &r->len);\n";
7724              pr "  r->val = calloc (r->len, sizeof (struct guestfs_lvm_lv));\n";
7725              pr "  for (i = 0; i < r->len; ++i) {\n";
7726              pr "    r->val[i].lv_name = malloc (16);\n";
7727              pr "    snprintf (r->val[i].lv_name, 16, \"%%d\", i);\n";
7728              pr "  }\n";
7729              pr "  return r;\n"
7730          | RStat _ ->
7731              pr "  struct guestfs_stat *r;\n";
7732              pr "  r = calloc (1, sizeof (*r));\n";
7733              pr "  sscanf (val, \"%%\" SCNi64, &r->dev);\n";
7734              pr "  return r;\n"
7735          | RStatVFS _ ->
7736              pr "  struct guestfs_statvfs *r;\n";
7737              pr "  r = calloc (1, sizeof (*r));\n";
7738              pr "  sscanf (val, \"%%\" SCNi64, &r->bsize);\n";
7739              pr "  return r;\n"
7740          | RHashtable _ ->
7741              pr "  char **strs;\n";
7742              pr "  int n, i;\n";
7743              pr "  sscanf (val, \"%%d\", &n);\n";
7744              pr "  strs = malloc ((n*2+1) * sizeof (char *));\n";
7745              pr "  for (i = 0; i < n; ++i) {\n";
7746              pr "    strs[i*2] = malloc (16);\n";
7747              pr "    strs[i*2+1] = malloc (16);\n";
7748              pr "    snprintf (strs[i*2], 16, \"%%d\", i);\n";
7749              pr "    snprintf (strs[i*2+1], 16, \"%%d\", i);\n";
7750              pr "  }\n";
7751              pr "  strs[n*2] = NULL;\n";
7752              pr "  return strs;\n"
7753         );
7754         pr "}\n";
7755         pr "\n"
7756       ) else (
7757         pr "/* Test error return. */\n";
7758         generate_prototype ~extern:false ~semicolon:false ~newline:true
7759           ~handle:"g" ~prefix:"guestfs_" name style;
7760         pr "{\n";
7761         pr "  error (g, \"error\");\n";
7762         (match fst style with
7763          | RErr | RInt _ | RInt64 _ | RBool _ ->
7764              pr "  return -1;\n"
7765          | RConstString _
7766          | RString _ | RStringList _ | RIntBool _
7767          | RPVList _ | RVGList _ | RLVList _ | RStat _ | RStatVFS _
7768          | RHashtable _ ->
7769              pr "  return NULL;\n"
7770         );
7771         pr "}\n";
7772         pr "\n"
7773       )
7774   ) tests
7775
7776 and generate_ocaml_bindtests () =
7777   generate_header OCamlStyle GPLv2;
7778
7779   pr "\
7780 let () =
7781   let g = Guestfs.create () in
7782 ";
7783
7784   let mkargs args =
7785     String.concat " " (
7786       List.map (
7787         function
7788         | CallString s -> "\"" ^ s ^ "\""
7789         | CallOptString None -> "None"
7790         | CallOptString (Some s) -> sprintf "(Some \"%s\")" s
7791         | CallStringList xs ->
7792             "[|" ^ String.concat ";" (List.map (sprintf "\"%s\"") xs) ^ "|]"
7793         | CallInt i when i >= 0 -> string_of_int i
7794         | CallInt i (* when i < 0 *) -> "(" ^ string_of_int i ^ ")"
7795         | CallBool b -> string_of_bool b
7796       ) args
7797     )
7798   in
7799
7800   generate_lang_bindtests (
7801     fun f args -> pr "  Guestfs.%s g %s;\n" f (mkargs args)
7802   );
7803
7804   pr "print_endline \"EOF\"\n"
7805
7806 and generate_perl_bindtests () =
7807   pr "#!/usr/bin/perl -w\n";
7808   generate_header HashStyle GPLv2;
7809
7810   pr "\
7811 use strict;
7812
7813 use Sys::Guestfs;
7814
7815 my $g = Sys::Guestfs->new ();
7816 ";
7817
7818   let mkargs args =
7819     String.concat ", " (
7820       List.map (
7821         function
7822         | CallString s -> "\"" ^ s ^ "\""
7823         | CallOptString None -> "undef"
7824         | CallOptString (Some s) -> sprintf "\"%s\"" s
7825         | CallStringList xs ->
7826             "[" ^ String.concat "," (List.map (sprintf "\"%s\"") xs) ^ "]"
7827         | CallInt i -> string_of_int i
7828         | CallBool b -> if b then "1" else "0"
7829       ) args
7830     )
7831   in
7832
7833   generate_lang_bindtests (
7834     fun f args -> pr "$g->%s (%s);\n" f (mkargs args)
7835   );
7836
7837   pr "print \"EOF\\n\"\n"
7838
7839 and generate_python_bindtests () =
7840   generate_header HashStyle GPLv2;
7841
7842   pr "\
7843 import guestfs
7844
7845 g = guestfs.GuestFS ()
7846 ";
7847
7848   let mkargs args =
7849     String.concat ", " (
7850       List.map (
7851         function
7852         | CallString s -> "\"" ^ s ^ "\""
7853         | CallOptString None -> "None"
7854         | CallOptString (Some s) -> sprintf "\"%s\"" s
7855         | CallStringList xs ->
7856             "[" ^ String.concat "," (List.map (sprintf "\"%s\"") xs) ^ "]"
7857         | CallInt i -> string_of_int i
7858         | CallBool b -> if b then "1" else "0"
7859       ) args
7860     )
7861   in
7862
7863   generate_lang_bindtests (
7864     fun f args -> pr "g.%s (%s)\n" f (mkargs args)
7865   );
7866
7867   pr "print \"EOF\"\n"
7868
7869 and generate_ruby_bindtests () =
7870   generate_header HashStyle GPLv2;
7871
7872   pr "\
7873 require 'guestfs'
7874
7875 g = Guestfs::create()
7876 ";
7877
7878   let mkargs args =
7879     String.concat ", " (
7880       List.map (
7881         function
7882         | CallString s -> "\"" ^ s ^ "\""
7883         | CallOptString None -> "nil"
7884         | CallOptString (Some s) -> sprintf "\"%s\"" s
7885         | CallStringList xs ->
7886             "[" ^ String.concat "," (List.map (sprintf "\"%s\"") xs) ^ "]"
7887         | CallInt i -> string_of_int i
7888         | CallBool b -> string_of_bool b
7889       ) args
7890     )
7891   in
7892
7893   generate_lang_bindtests (
7894     fun f args -> pr "g.%s(%s)\n" f (mkargs args)
7895   );
7896
7897   pr "print \"EOF\\n\"\n"
7898
7899 and generate_java_bindtests () =
7900   generate_header CStyle GPLv2;
7901
7902   pr "\
7903 import com.redhat.et.libguestfs.*;
7904
7905 public class Bindtests {
7906     public static void main (String[] argv)
7907     {
7908         try {
7909             GuestFS g = new GuestFS ();
7910 ";
7911
7912   let mkargs args =
7913     String.concat ", " (
7914       List.map (
7915         function
7916         | CallString s -> "\"" ^ s ^ "\""
7917         | CallOptString None -> "null"
7918         | CallOptString (Some s) -> sprintf "\"%s\"" s
7919         | CallStringList xs ->
7920             "new String[]{" ^
7921               String.concat "," (List.map (sprintf "\"%s\"") xs) ^ "}"
7922         | CallInt i -> string_of_int i
7923         | CallBool b -> string_of_bool b
7924       ) args
7925     )
7926   in
7927
7928   generate_lang_bindtests (
7929     fun f args -> pr "            g.%s (%s);\n" f (mkargs args)
7930   );
7931
7932   pr "
7933             System.out.println (\"EOF\");
7934         }
7935         catch (Exception exn) {
7936             System.err.println (exn);
7937             System.exit (1);
7938         }
7939     }
7940 }
7941 "
7942
7943 and generate_haskell_bindtests () =
7944   generate_header HaskellStyle GPLv2;
7945
7946   pr "\
7947 module Bindtests where
7948 import qualified Guestfs
7949
7950 main = do
7951   g <- Guestfs.create
7952 ";
7953
7954   let mkargs args =
7955     String.concat " " (
7956       List.map (
7957         function
7958         | CallString s -> "\"" ^ s ^ "\""
7959         | CallOptString None -> "Nothing"
7960         | CallOptString (Some s) -> sprintf "(Just \"%s\")" s
7961         | CallStringList xs ->
7962             "[" ^ String.concat "," (List.map (sprintf "\"%s\"") xs) ^ "]"
7963         | CallInt i when i < 0 -> "(" ^ string_of_int i ^ ")"
7964         | CallInt i -> string_of_int i
7965         | CallBool true -> "True"
7966         | CallBool false -> "False"
7967       ) args
7968     )
7969   in
7970
7971   generate_lang_bindtests (
7972     fun f args -> pr "  Guestfs.%s g %s\n" f (mkargs args)
7973   );
7974
7975   pr "  putStrLn \"EOF\"\n"
7976
7977 (* Language-independent bindings tests - we do it this way to
7978  * ensure there is parity in testing bindings across all languages.
7979  *)
7980 and generate_lang_bindtests call =
7981   call "test0" [CallString "abc"; CallOptString (Some "def");
7982                 CallStringList []; CallBool false;
7983                 CallInt 0; CallString "123"; CallString "456"];
7984   call "test0" [CallString "abc"; CallOptString None;
7985                 CallStringList []; CallBool false;
7986                 CallInt 0; CallString "123"; CallString "456"];
7987   call "test0" [CallString ""; CallOptString (Some "def");
7988                 CallStringList []; CallBool false;
7989                 CallInt 0; CallString "123"; CallString "456"];
7990   call "test0" [CallString ""; CallOptString (Some "");
7991                 CallStringList []; CallBool false;
7992                 CallInt 0; CallString "123"; CallString "456"];
7993   call "test0" [CallString "abc"; CallOptString (Some "def");
7994                 CallStringList ["1"]; CallBool false;
7995                 CallInt 0; CallString "123"; CallString "456"];
7996   call "test0" [CallString "abc"; CallOptString (Some "def");
7997                 CallStringList ["1"; "2"]; CallBool false;
7998                 CallInt 0; CallString "123"; CallString "456"];
7999   call "test0" [CallString "abc"; CallOptString (Some "def");
8000                 CallStringList ["1"]; CallBool true;
8001                 CallInt 0; CallString "123"; CallString "456"];
8002   call "test0" [CallString "abc"; CallOptString (Some "def");
8003                 CallStringList ["1"]; CallBool false;
8004                 CallInt (-1); CallString "123"; CallString "456"];
8005   call "test0" [CallString "abc"; CallOptString (Some "def");
8006                 CallStringList ["1"]; CallBool false;
8007                 CallInt (-2); CallString "123"; CallString "456"];
8008   call "test0" [CallString "abc"; CallOptString (Some "def");
8009                 CallStringList ["1"]; CallBool false;
8010                 CallInt 1; CallString "123"; CallString "456"];
8011   call "test0" [CallString "abc"; CallOptString (Some "def");
8012                 CallStringList ["1"]; CallBool false;
8013                 CallInt 2; CallString "123"; CallString "456"];
8014   call "test0" [CallString "abc"; CallOptString (Some "def");
8015                 CallStringList ["1"]; CallBool false;
8016                 CallInt 4095; CallString "123"; CallString "456"];
8017   call "test0" [CallString "abc"; CallOptString (Some "def");
8018                 CallStringList ["1"]; CallBool false;
8019                 CallInt 0; CallString ""; CallString ""]
8020
8021   (* XXX Add here tests of the return and error functions. *)
8022
8023 (* This is used to generate the src/MAX_PROC_NR file which
8024  * contains the maximum procedure number, a surrogate for the
8025  * ABI version number.  See src/Makefile.am for the details.
8026  *)
8027 and generate_max_proc_nr () =
8028   let proc_nrs = List.map (
8029     fun (_, _, proc_nr, _, _, _, _) -> proc_nr
8030   ) daemon_functions in
8031
8032   let max_proc_nr = List.fold_left max 0 proc_nrs in
8033
8034   pr "%d\n" max_proc_nr
8035
8036 let output_to filename =
8037   let filename_new = filename ^ ".new" in
8038   chan := open_out filename_new;
8039   let close () =
8040     close_out !chan;
8041     chan := stdout;
8042
8043     (* Is the new file different from the current file? *)
8044     if Sys.file_exists filename && files_equal filename filename_new then
8045       Unix.unlink filename_new          (* same, so skip it *)
8046     else (
8047       (* different, overwrite old one *)
8048       (try Unix.chmod filename 0o644 with Unix.Unix_error _ -> ());
8049       Unix.rename filename_new filename;
8050       Unix.chmod filename 0o444;
8051       printf "written %s\n%!" filename;
8052     )
8053   in
8054   close
8055
8056 (* Main program. *)
8057 let () =
8058   check_functions ();
8059
8060   if not (Sys.file_exists "configure.ac") then (
8061     eprintf "\
8062 You are probably running this from the wrong directory.
8063 Run it from the top source directory using the command
8064   src/generator.ml
8065 ";
8066     exit 1
8067   );
8068
8069   let close = output_to "src/guestfs_protocol.x" in
8070   generate_xdr ();
8071   close ();
8072
8073   let close = output_to "src/guestfs-structs.h" in
8074   generate_structs_h ();
8075   close ();
8076
8077   let close = output_to "src/guestfs-actions.h" in
8078   generate_actions_h ();
8079   close ();
8080
8081   let close = output_to "src/guestfs-actions.c" in
8082   generate_client_actions ();
8083   close ();
8084
8085   let close = output_to "daemon/actions.h" in
8086   generate_daemon_actions_h ();
8087   close ();
8088
8089   let close = output_to "daemon/stubs.c" in
8090   generate_daemon_actions ();
8091   close ();
8092
8093   let close = output_to "capitests/tests.c" in
8094   generate_tests ();
8095   close ();
8096
8097   let close = output_to "src/guestfs-bindtests.c" in
8098   generate_bindtests ();
8099   close ();
8100
8101   let close = output_to "fish/cmds.c" in
8102   generate_fish_cmds ();
8103   close ();
8104
8105   let close = output_to "fish/completion.c" in
8106   generate_fish_completion ();
8107   close ();
8108
8109   let close = output_to "guestfs-structs.pod" in
8110   generate_structs_pod ();
8111   close ();
8112
8113   let close = output_to "guestfs-actions.pod" in
8114   generate_actions_pod ();
8115   close ();
8116
8117   let close = output_to "guestfish-actions.pod" in
8118   generate_fish_actions_pod ();
8119   close ();
8120
8121   let close = output_to "ocaml/guestfs.mli" in
8122   generate_ocaml_mli ();
8123   close ();
8124
8125   let close = output_to "ocaml/guestfs.ml" in
8126   generate_ocaml_ml ();
8127   close ();
8128
8129   let close = output_to "ocaml/guestfs_c_actions.c" in
8130   generate_ocaml_c ();
8131   close ();
8132
8133   let close = output_to "ocaml/bindtests.ml" in
8134   generate_ocaml_bindtests ();
8135   close ();
8136
8137   let close = output_to "perl/Guestfs.xs" in
8138   generate_perl_xs ();
8139   close ();
8140
8141   let close = output_to "perl/lib/Sys/Guestfs.pm" in
8142   generate_perl_pm ();
8143   close ();
8144
8145   let close = output_to "perl/bindtests.pl" in
8146   generate_perl_bindtests ();
8147   close ();
8148
8149   let close = output_to "python/guestfs-py.c" in
8150   generate_python_c ();
8151   close ();
8152
8153   let close = output_to "python/guestfs.py" in
8154   generate_python_py ();
8155   close ();
8156
8157   let close = output_to "python/bindtests.py" in
8158   generate_python_bindtests ();
8159   close ();
8160
8161   let close = output_to "ruby/ext/guestfs/_guestfs.c" in
8162   generate_ruby_c ();
8163   close ();
8164
8165   let close = output_to "ruby/bindtests.rb" in
8166   generate_ruby_bindtests ();
8167   close ();
8168
8169   let close = output_to "java/com/redhat/et/libguestfs/GuestFS.java" in
8170   generate_java_java ();
8171   close ();
8172
8173   let close = output_to "java/com/redhat/et/libguestfs/PV.java" in
8174   generate_java_struct "PV" pv_cols;
8175   close ();
8176
8177   let close = output_to "java/com/redhat/et/libguestfs/VG.java" in
8178   generate_java_struct "VG" vg_cols;
8179   close ();
8180
8181   let close = output_to "java/com/redhat/et/libguestfs/LV.java" in
8182   generate_java_struct "LV" lv_cols;
8183   close ();
8184
8185   let close = output_to "java/com/redhat/et/libguestfs/Stat.java" in
8186   generate_java_struct "Stat" stat_cols;
8187   close ();
8188
8189   let close = output_to "java/com/redhat/et/libguestfs/StatVFS.java" in
8190   generate_java_struct "StatVFS" statvfs_cols;
8191   close ();
8192
8193   let close = output_to "java/com_redhat_et_libguestfs_GuestFS.c" in
8194   generate_java_c ();
8195   close ();
8196
8197   let close = output_to "java/Bindtests.java" in
8198   generate_java_bindtests ();
8199   close ();
8200
8201   let close = output_to "haskell/Guestfs.hs" in
8202   generate_haskell_hs ();
8203   close ();
8204
8205   let close = output_to "haskell/Bindtests.hs" in
8206   generate_haskell_bindtests ();
8207   close ();
8208
8209   let close = output_to "src/MAX_PROC_NR" in
8210   generate_max_proc_nr ();
8211   close ();