internal: Use size_t instead of int for command line size.
[libguestfs.git] / daemon / fill.c
1 /* libguestfs - the guestfsd daemon
2  * Copyright (C) 2009 Red Hat Inc.
3  *
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.
8  *
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.
13  *
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.
17  */
18
19 #include <config.h>
20
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <unistd.h>
25 #include <fcntl.h>
26
27 #include "guestfs_protocol.h"
28 #include "daemon.h"
29 #include "actions.h"
30
31 int
32 do_fill (int c, int len, const char *path)
33 {
34   int fd;
35   ssize_t r;
36   size_t len_sz;
37   size_t n;
38   char buf[BUFSIZ];
39
40   if (c < 0 || c > 255) {
41     reply_with_error ("%d: byte number must be in range 0..255", c);
42     return -1;
43   }
44   memset (buf, c, BUFSIZ);
45   if (len < 0) {
46     reply_with_error ("%d: length is < 0", len);
47     return -1;
48   }
49   len_sz = (size_t) len;
50
51   CHROOT_IN;
52   fd = open (path, O_WRONLY | O_CREAT | O_NOCTTY, 0666);
53   CHROOT_OUT;
54
55   if (fd == -1) {
56     reply_with_perror ("open: %s", path);
57     return -1;
58   }
59
60   n = 0;
61   while (n < len_sz) {
62     r = write (fd, buf, len_sz - n < BUFSIZ ? len_sz - n : BUFSIZ);
63     if (r == -1) {
64       reply_with_perror ("write: %s", path);
65       close (fd);
66       return -1;
67     }
68     n += r;
69     notify_progress ((uint64_t) n, (uint64_t) len_sz);
70   }
71
72   if (close (fd) == -1) {
73     reply_with_perror ("close: %s", path);
74     return -1;
75   }
76
77   return 0;
78 }
79
80 int
81 do_fill_pattern (const char *pattern, int len, const char *path)
82 {
83   size_t patlen = strlen (pattern);
84
85   if (patlen < 1) {
86     reply_with_error ("pattern string must be non-empty");
87     return -1;
88   }
89
90   if (len < 0) {
91     reply_with_error ("%d: length is < 0", len);
92     return -1;
93   }
94   size_t len_sz = (size_t) len;
95
96   int fd;
97   CHROOT_IN;
98   fd = open (path, O_WRONLY | O_CREAT | O_NOCTTY, 0666);
99   CHROOT_OUT;
100
101   if (fd == -1) {
102     reply_with_perror ("open: %s", path);
103     return -1;
104   }
105
106   /* XXX This implementation won't be very efficient for large files. */
107   size_t n = 0;
108   while (n < len_sz) {
109     size_t wrlen = len_sz - n < patlen ? len_sz - n : patlen;
110     if (xwrite (fd, pattern, wrlen) == -1) {
111       reply_with_perror ("write: %s", path);
112       close (fd);
113       return -1;
114     }
115     n += wrlen;
116     notify_progress ((uint64_t) n, (uint64_t) len_sz);
117   }
118
119   if (close (fd) == -1) {
120     reply_with_perror ("close: %s", path);
121     return -1;
122   }
123
124   return 0;
125 }