update all NEED_ROOT uses
[libguestfs.git] / daemon / scrub.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 <fcntl.h>
25 #include <unistd.h>
26
27 #include "daemon.h"
28 #include "actions.h"
29
30 int
31 do_scrub_device (char *device)
32 {
33   char *err;
34   int r;
35
36   r = command (NULL, &err, "scrub", device, NULL);
37   if (r == -1) {
38     reply_with_error ("scrub_device: %s: %s", device, err);
39     free (err);
40     return -1;
41   }
42
43   free (err);
44
45   return 0;
46 }
47
48 int
49 do_scrub_file (char *file)
50 {
51   char *buf;
52   char *err;
53   int r;
54
55   NEED_ROOT (return -1);
56   ABS_PATH (file, return -1);
57
58   /* Make the path relative to /sysroot. */
59   buf = sysroot_path (file);
60   if (!buf) {
61     reply_with_perror ("malloc");
62     return -1;
63   }
64
65   r = command (NULL, &err, "scrub", "-r", buf, NULL);
66   free (buf);
67   if (r == -1) {
68     reply_with_error ("scrub_file: %s: %s", file, err);
69     free (err);
70     return -1;
71   }
72
73   free (err);
74
75   return 0;
76 }
77
78 int
79 do_scrub_freespace (char *dir)
80 {
81   char *buf;
82   char *err;
83   int r;
84
85   NEED_ROOT (return -1);
86   ABS_PATH (dir, return -1);
87
88   /* Make the path relative to /sysroot. */
89   buf = sysroot_path (dir);
90   if (!buf) {
91     reply_with_perror ("malloc");
92     return -1;
93   }
94
95   r = command (NULL, &err, "scrub", "-X", buf, NULL);
96   free (buf);
97   if (r == -1) {
98     reply_with_error ("scrub_freespace: %s: %s", dir, err);
99     free (err);
100     return -1;
101   }
102
103   free (err);
104
105   return 0;
106 }