configure: Add --with-extra flag to allow setting the extra version string.
[libguestfs.git] / daemon / zero.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 <inttypes.h>
24 #include <string.h>
25 #include <fcntl.h>
26 #include <unistd.h>
27
28 #include "daemon.h"
29 #include "actions.h"
30
31 int
32 do_zero (const char *device)
33 {
34   int fd, i;
35   char buf[4096];
36
37   fd = open (device, O_WRONLY);
38   if (fd == -1) {
39     reply_with_perror ("%s", device);
40     return -1;
41   }
42
43   memset (buf, 0, sizeof buf);
44
45   for (i = 0; i < 32; ++i) {
46     if (write (fd, buf, sizeof buf) != sizeof buf) {
47       reply_with_perror ("write: %s", device);
48       close (fd);
49       return -1;
50     }
51     notify_progress ((uint64_t) i, 32);
52   }
53
54   if (close (fd) == -1) {
55     reply_with_perror ("close: %s", device);
56     return -1;
57   }
58
59   return 0;
60 }
61
62 int
63 do_zero_device (const char *device)
64 {
65   int64_t ssize = do_blockdev_getsize64 (device);
66   if (ssize == -1)
67     return -1;
68   uint64_t size = (uint64_t) ssize;
69
70   int fd = open (device, O_WRONLY);
71   if (fd == -1) {
72     reply_with_perror ("%s", device);
73     return -1;
74   }
75
76   char buf[1024*1024];
77   memset (buf, 0, sizeof buf);
78
79   uint64_t pos = 0;
80
81   while (pos < size) {
82     uint64_t n64 = size - pos;
83     size_t n;
84     if (n64 > sizeof buf)
85       n = sizeof buf;
86     else
87       n = (size_t) n64; /* safe because of if condition */
88
89     ssize_t r = write (fd, buf, n);
90     if (r == -1) {
91       reply_with_perror ("write: %s (with %" PRId64 " bytes left to write)",
92                          device, size);
93       close (fd);
94       return -1;
95     }
96
97     pos += r;
98     notify_progress (pos, size);
99   }
100
101   if (close (fd) == -1) {
102     reply_with_perror ("close: %s", device);
103     return -1;
104   }
105
106   return 0;
107 }