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