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