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