Revert "add_drive_ro adds readonly=on option if available." (RHBZ#617200).
[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 "../src/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   }
70
71   if (close (fd) == -1) {
72     reply_with_perror ("close: %s", path);
73     return -1;
74   }
75
76   return 0;
77 }
78
79 int
80 do_fill_pattern (const char *pattern, int len, const char *path)
81 {
82   size_t patlen = strlen (pattern);
83
84   if (patlen < 1) {
85     reply_with_error ("pattern string must be non-empty");
86     return -1;
87   }
88
89   if (len < 0) {
90     reply_with_error ("%d: length is < 0", len);
91     return -1;
92   }
93   size_t len_sz = (size_t) len;
94
95   int fd;
96   CHROOT_IN;
97   fd = open (path, O_WRONLY | O_CREAT | O_NOCTTY, 0666);
98   CHROOT_OUT;
99
100   if (fd == -1) {
101     reply_with_perror ("open: %s", path);
102     return -1;
103   }
104
105   /* XXX This implementation won't be very efficient for large files. */
106   size_t n = 0;
107   while (n < len_sz) {
108     size_t wrlen = len_sz - n < patlen ? len_sz - n : patlen;
109     if (xwrite (fd, pattern, wrlen) == -1) {
110       reply_with_perror ("write: %s", path);
111       close (fd);
112       return -1;
113     }
114     n += wrlen;
115   }
116
117   if (close (fd) == -1) {
118     reply_with_perror ("close: %s", path);
119     return -1;
120   }
121
122   return 0;
123 }