docs: Routine refresh of the documentation for guestfs(3) and guestfish(1).
[libguestfs.git] / daemon / sfdisk.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 #include <dirent.h>
27 #include <sys/stat.h>
28
29 #include "daemon.h"
30 #include "actions.h"
31
32 static int
33 sfdisk (const char *device, int n, int cyls, int heads, int sectors,
34         const char *extra_flag,
35         char *const *lines)
36 {
37   FILE *fp;
38   char buf[256];
39   int i;
40
41   strcpy (buf, "sfdisk");
42
43   if (n > 0)
44     sprintf (buf + strlen (buf), " -N %d", n);
45   if (cyls)
46     sprintf (buf + strlen (buf), " -C %d", cyls);
47   if (heads)
48     sprintf (buf + strlen (buf), " -H %d", heads);
49   if (sectors)
50     sprintf (buf + strlen (buf), " -S %d", sectors);
51
52   /* The above are all guaranteed to fit in the fixed-size buffer.
53      However, extra_flag and device have no restrictions,
54      so we must check.  */
55
56   if (extra_flag) {
57     if (strlen (buf) + 1 + strlen (extra_flag) >= sizeof buf) {
58       reply_with_error ("internal buffer overflow: sfdisk extra_flag too long");
59       return -1;
60     }
61     sprintf (buf + strlen (buf), " %s", extra_flag);
62   }
63
64   if (strlen (buf) + 1 + strlen (device) >= sizeof buf) {
65     reply_with_error ("internal buffer overflow: sfdisk device name too long");
66     return -1;
67   }
68   sprintf (buf + strlen (buf), " %s", device);
69
70   if (verbose)
71     printf ("%s\n", buf);
72
73   fp = popen (buf, "w");
74   if (fp == NULL) {
75     reply_with_perror ("failed to open pipe: %s", buf);
76     return -1;
77   }
78
79   for (i = 0; lines[i] != NULL; ++i) {
80     if (fprintf (fp, "%s\n", lines[i]) < 0) {
81       reply_with_perror ("failed to write to pipe: %s", buf);
82       pclose (fp);
83       return -1;
84     }
85   }
86
87   if (pclose (fp) != 0) {
88     reply_with_error ("%s: external command failed", buf);
89     return -1;
90   }
91
92   udev_settle ();
93
94   return 0;
95 }
96
97 int
98 do_sfdisk (const char *device, int cyls, int heads, int sectors,
99            char *const *lines)
100 {
101   return sfdisk (device, 0, cyls, heads, sectors, NULL, lines);
102 }
103
104 int
105 do_sfdisk_N (const char *device, int n, int cyls, int heads, int sectors,
106              const char *line)
107 {
108   char const *const lines[2] = { line, NULL };
109
110   return sfdisk (device, n, cyls, heads, sectors, NULL, (void *) lines);
111 }
112
113 int
114 do_sfdiskM (const char *device, char *const *lines)
115 {
116   return sfdisk (device, 0, 0, 0, 0, "-uM", lines);
117 }
118
119 static char *
120 sfdisk_flag (const char *device, const char *flag)
121 {
122   char *out, *err;
123   int r;
124
125   r = command (&out, &err, "sfdisk", flag, device, NULL);
126   if (r == -1) {
127     reply_with_error ("%s: %s", device, err);
128     free (out);
129     free (err);
130     return NULL;
131   }
132
133   free (err);
134
135   udev_settle ();
136
137   return out;                   /* caller frees */
138 }
139
140 char *
141 do_sfdisk_l (const char *device)
142 {
143   return sfdisk_flag (device, "-l");
144 }
145
146 char *
147 do_sfdisk_kernel_geometry (const char *device)
148 {
149   return sfdisk_flag (device, "-g");
150 }
151
152 char *
153 do_sfdisk_disk_geometry (const char *device)
154 {
155   return sfdisk_flag (device, "-G");
156 }