2be46622fde15d7ddf879e6b12b2ac57c298c3b0
[dragonfly.git] / sbin / hammer2 / main.c
1 /*
2  * Copyright (c) 2011-2012 The DragonFly Project.  All rights reserved.
3  *
4  * This code is derived from software contributed to The DragonFly Project
5  * by Matthew Dillon <dillon@dragonflybsd.org>
6  * by Venkatesh Srinivas <vsrinivas@dragonflybsd.org>
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  *
12  * 1. Redistributions of source code must retain the above copyright
13  *    notice, this list of conditions and the following disclaimer.
14  * 2. Redistributions in binary form must reproduce the above copyright
15  *    notice, this list of conditions and the following disclaimer in
16  *    the documentation and/or other materials provided with the
17  *    distribution.
18  * 3. Neither the name of The DragonFly Project nor the names of its
19  *    contributors may be used to endorse or promote products derived
20  *    from this software without specific, prior written permission.
21  *
22  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
23  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
24  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
25  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
26  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
27  * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
28  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
29  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
30  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
31  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
32  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33  * SUCH DAMAGE.
34  */
35
36 #include "hammer2.h"
37
38 static void usage(int code);
39
40 int DebugOpt;
41 int NormalExit = 1;     /* if set to 0 main() has to pthread_exit() */
42
43 int
44 main(int ac, char **av)
45 {
46         const char *sel_path = NULL;
47         const char *uuid_str = NULL;
48         const char *arg;
49         int pfs_type = HAMMER2_PFSTYPE_NONE;
50         int quick_opt = 0;
51         int all_opt = 0;
52         int ecode = 0;
53         int ch;
54
55         srandomdev();
56
57         /*
58          * Core options
59          */
60         while ((ch = getopt(ac, av, "adqs:t:u:")) != -1) {
61                 switch(ch) {
62                 case 'a':
63                         all_opt = 1;
64                         break;
65                 case 'q':
66                         /*
67                          * Quick mode - do not block verifying certain
68                          * operations such as (connect).
69                          */
70                         quick_opt = 1;
71                         break;
72                 case 's':
73                         sel_path = optarg;
74                         break;
75                 case 't':
76                         /*
77                          * set node type for mkpfs
78                          */
79                         if (strcasecmp(optarg, "ADMIN") == 0) {
80                                 pfs_type = HAMMER2_PFSTYPE_ADMIN;
81                         } else if (strcasecmp(optarg, "CACHE") == 0) {
82                                 pfs_type = HAMMER2_PFSTYPE_CACHE;
83                         } else if (strcasecmp(optarg, "COPY") == 0) {
84                                 pfs_type = HAMMER2_PFSTYPE_COPY;
85                         } else if (strcasecmp(optarg, "SLAVE") == 0) {
86                                 pfs_type = HAMMER2_PFSTYPE_SLAVE;
87                         } else if (strcasecmp(optarg, "SOFT_SLAVE") == 0) {
88                                 pfs_type = HAMMER2_PFSTYPE_SOFT_SLAVE;
89                         } else if (strcasecmp(optarg, "SOFT_MASTER") == 0) {
90                                 pfs_type = HAMMER2_PFSTYPE_SOFT_MASTER;
91                         } else if (strcasecmp(optarg, "MASTER") == 0) {
92                                 pfs_type = HAMMER2_PFSTYPE_MASTER;
93                         } else {
94                                 fprintf(stderr, "-t: Unrecognized node type\n");
95                                 usage(1);
96                         }
97                         break;
98                 case 'u':
99                         /*
100                          * set uuid for mkpfs, else one will be generated
101                          * (required for all except the MASTER node_type)
102                          */
103                         uuid_str = optarg;
104                         break;
105                 case 'd':
106                         DebugOpt = 1;
107                         break;
108                 default:
109                         fprintf(stderr, "Unknown option: %c\n", ch);
110                         usage(1);
111                         /* not reached */
112                         break;
113                 }
114         }
115
116         /*
117          * Adjust, then process the command
118          */
119         ac -= optind;
120         av += optind;
121         if (ac < 1) {
122                 fprintf(stderr, "Missing command\n");
123                 usage(1);
124                 /* not reached */
125         }
126
127         if (strcmp(av[0], "connect") == 0) {
128                 /*
129                  * Add cluster connection
130                  */
131                 if (ac < 2) {
132                         fprintf(stderr, "connect: missing argument\n");
133                         usage(1);
134                 }
135                 ecode = cmd_remote_connect(sel_path, av[1]);
136         } else if (strcmp(av[0], "disconnect") == 0) {
137                 /*
138                  * Remove cluster connection
139                  */
140                 if (ac < 2) {
141                         fprintf(stderr, "disconnect: missing argument\n");
142                         usage(1);
143                 }
144                 ecode = cmd_remote_disconnect(sel_path, av[1]);
145         } else if (strcmp(av[0], "status") == 0) {
146                 /*
147                  * Get status of PFS and its connections (-a for all PFSs)
148                  */
149                 ecode = cmd_remote_status(sel_path, all_opt);
150         } else if (strcmp(av[0], "pfs_list") == 0) {
151                 /*
152                  * List all PFSs
153                  */
154                 ecode = cmd_pfs_list(sel_path);
155         } else if (strcmp(av[0], "pfs_create") == 0) {
156                 /*
157                  * Create new PFS using pfs_type
158                  */
159                 if (ac < 2) {
160                         fprintf(stderr, "pfs_create: requires name\n");
161                         usage(1);
162                 }
163                 ecode = cmd_pfs_create(sel_path, av[1], pfs_type, uuid_str);
164         } else if (strcmp(av[0], "pfs_delete") == 0) {
165                 /*
166                  * Delete a PFS by name
167                  */
168                 if (ac < 2) {
169                         fprintf(stderr, "pfs_delete: requires name\n");
170                         usage(1);
171                 }
172                 ecode = cmd_pfs_delete(sel_path, av[1]);
173         } else if (strcmp(av[0], "snapshot") == 0) {
174                 /*
175                  * Create snapshot with optional pfs_type and optional
176                  * label override.
177                  */
178         } else if (strcmp(av[0], "service") == 0) {
179                 /*
180                  * Start the service daemon.  This daemon accepts
181                  * connections from local and remote clients, handles
182                  * the security handshake, and manages the core messaging
183                  * protocol.
184                  */
185                 ecode = cmd_service();
186         } else if (strcmp(av[0], "leaf") == 0) {
187                 /*
188                  * Start the management daemon for a specific PFS.
189                  *
190                  * This will typically connect to the local master node
191                  * daemon, register the PFS, and then pass its side of
192                  * the socket descriptor to the kernel HAMMER2 VFS via an
193                  * ioctl().  The process and/or thread context remains in the
194                  * kernel until the PFS is unmounted or the connection is
195                  * lost, then returns from the ioctl.
196                  *
197                  * It is possible to connect directly to a remote master node
198                  * instead of the local master node in situations where
199                  * encryption is not desired or no local master node is
200                  * desired.  This is not recommended because it represents
201                  * a single point of failure for the PFS's communications.
202                  *
203                  * Direct kernel<->kernel communication between HAMMER2 VFSs
204                  * is theoretically possible for directly-connected
205                  * registrations (i.e. where the spanning tree is degenerate),
206                  * but not recommended.  We specifically try to reduce the
207                  * complexity of the HAMMER2 VFS kernel code.
208                  */
209                 ecode = cmd_leaf(sel_path);
210         } else if (strcmp(av[0], "debug") == 0) {
211                 /*
212                  * Connect to the command line monitor in the hammer2 master
213                  * node for the machine using HAMMER2_DBG_SHELL messages.
214                  */
215                 ecode = cmd_debug((ac < 2) ? NULL : av[1]);
216         } else if (strcmp(av[0], "rsainit") == 0) {
217                 /*
218                  * Initialize a RSA keypair.  If no target directory is
219                  * specified we default to "/etc/hammer2".
220                  */
221                 arg = (ac < 2) ? HAMMER2_DEFAULT_DIR : av[1];
222                 ecode = cmd_rsainit(arg);
223         } else if (strcmp(av[0], "rsaenc") == 0) {
224                 /*
225                  * Encrypt the input symmetrically by running it through
226                  * the specified public and/or private key files.
227                  *
228                  * If no key files are specified data is encoded using
229                  * "/etc/hammer2/rsa.pub".
230                  *
231                  * WARNING: no padding is added, data stream must contain
232                  *          random padding for this to be secure.
233                  *
234                  * Used for debugging only
235                  */
236                 if (ac == 1) {
237                         const char *rsapath = HAMMER2_DEFAULT_DIR "/rsa.pub";
238                         ecode = cmd_rsaenc(&rsapath, 1);
239                 } else {
240                         ecode = cmd_rsaenc((const char **)&av[1], ac - 1);
241                 }
242         } else if (strcmp(av[0], "rsadec") == 0) {
243                 /*
244                  * Decrypt the input symmetrically by running it through
245                  * the specified public and/or private key files.
246                  *
247                  * If no key files are specified data is decoded using
248                  * "/etc/hammer2/rsa.prv".
249                  *
250                  * WARNING: no padding is added, data stream must contain
251                  *          random padding for this to be secure.
252                  *
253                  * Used for debugging only
254                  */
255                 if (ac == 1) {
256                         const char *rsapath = HAMMER2_DEFAULT_DIR "/rsa.prv";
257                         ecode = cmd_rsadec(&rsapath, 1);
258                 } else {
259                         ecode = cmd_rsadec((const char **)&av[1], ac - 1);
260                 }
261         } else {
262                 fprintf(stderr, "Unrecognized command: %s\n", av[0]);
263                 usage(1);
264         }
265
266         /*
267          * In DebugMode we may wind up starting several pthreads in the
268          * original process, in which case we have to let them run and
269          * not actually exit.
270          */
271         if (NormalExit) {
272                 return (ecode);
273         } else {
274                 pthread_exit(NULL);
275                 _exit(2);       /* NOT REACHED */
276         }
277 }
278
279 static
280 void
281 usage(int code)
282 {
283         fprintf(stderr,
284                 "hammer2 [-s path] command...\n"
285                 "    -s path            Select filesystem\n"
286         );
287         exit(code);
288 }