1 /* libguestfs - the guestfsd daemon
2 * Copyright (C) 2009 Red Hat Inc.
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
28 #include "../src/guestfs_protocol.h"
33 input_to_nul (FILE *fp, char *buf, int maxlen)
46 reply_with_error ("input_to_nul: input string too long");
51 do_find (const char *dir)
54 int r, len, sysrootdirlen;
58 int size = 0, alloc = 0;
62 sysrootdir = sysroot_path (dir);
64 reply_with_perror ("malloc");
68 r = stat (sysrootdir, &statbuf);
70 reply_with_perror ("%s", dir);
74 if (!S_ISDIR (statbuf.st_mode)) {
75 reply_with_error ("%s: not a directory", dir);
80 sysrootdirlen = strlen (sysrootdir);
82 /* Assemble the external find command. */
83 if (asprintf_nowarn (&cmd, "find %Q -print0", sysrootdir) == -1) {
84 reply_with_perror ("malloc");
90 fprintf (stderr, "%s\n", cmd);
92 fp = popen (cmd, "r");
94 reply_with_perror ("%s", cmd);
100 while ((r = input_to_nul (fp, str, PATH_MAX)) > 0) {
102 printf ("find string: %s\n", str);
105 if (len <= sysrootdirlen)
108 /* Remove the directory part of the path when adding it. */
109 if (add_string (&res, &size, &alloc, str + sysrootdirlen) == -1) {
114 if (pclose (fp) != 0) {
115 reply_with_perror ("pclose: find");
116 free_stringslen (res, size);
121 free_stringslen (res, size);
125 if (add_string (&res, &size, &alloc, NULL) == -1)
128 sort_strings (res, size-1);
130 return res; /* caller frees */
133 /* The code below assumes each path returned can fit into a protocol
134 * chunk. If this turns out not to be true at some point in the
135 * future then we'll need to modify the code a bit to handle it.
137 #if PATH_MAX > GUESTFS_MAX_CHUNK_SIZE
138 #error "PATH_MAX > GUESTFS_MAX_CHUNK_SIZE"
141 /* Has one FileOut parameter. */
143 do_find0 (const char *dir)
150 size_t sysrootdirlen, len;
151 char str[GUESTFS_MAX_CHUNK_SIZE];
153 sysrootdir = sysroot_path (dir);
155 reply_with_perror ("malloc");
159 r = stat (sysrootdir, &statbuf);
161 reply_with_perror ("%s", dir);
165 if (!S_ISDIR (statbuf.st_mode)) {
166 reply_with_error ("%s: not a directory", dir);
171 sysrootdirlen = strlen (sysrootdir);
173 if (asprintf_nowarn (&cmd, "find %Q -print0", sysrootdir) == -1) {
174 reply_with_perror ("asprintf");
181 fprintf (stderr, "%s\n", cmd);
183 fp = popen (cmd, "r");
185 reply_with_perror ("%s", cmd);
191 /* Now we must send the reply message, before the file contents. After
192 * this there is no opportunity in the protocol to send any error
193 * message back. Instead we can only cancel the transfer.
197 while ((r = input_to_nul (fp, str, GUESTFS_MAX_CHUNK_SIZE)) > 0) {
199 printf ("find0 string: %s\n", str);
202 if (len <= sysrootdirlen)
205 /* Remove the directory part of the path before sending it. */
206 if (send_file_write (str + sysrootdirlen, r - sysrootdirlen) < 0) {
214 send_file_end (1); /* Cancel. */
219 if (pclose (fp) != 0) {
221 send_file_end (1); /* Cancel. */
225 if (send_file_end (0)) /* Normal end of file. */