Merge branch 'vendor/OPENPAM'
[dragonfly.git] / sbin / hammer / cmd_reblock.c
1 /*
2  * Copyright (c) 2008 The DragonFly Project.  All rights reserved.
3  *
4  * This code is derived from software contributed to The DragonFly Project
5  * by Matthew Dillon <dillon@backplane.com>
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  *
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in
15  *    the documentation and/or other materials provided with the
16  *    distribution.
17  * 3. Neither the name of The DragonFly Project nor the names of its
18  *    contributors may be used to endorse or promote products derived
19  *    from this software without specific, prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
22  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
23  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
24  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
25  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
26  * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
27  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
28  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
29  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
30  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
31  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32  * SUCH DAMAGE.
33  *
34  * $DragonFly: src/sbin/hammer/cmd_reblock.c,v 1.11 2008/07/16 00:53:48 thomas Exp $
35  */
36
37 #include "hammer.h"
38
39 static void reblock_usage(int exit_code);
40
41 /*
42  * reblock <filesystem> [fill_percentage] (default 100%)
43  */
44 void
45 hammer_cmd_reblock(char **av, int ac, int flags)
46 {
47         struct hammer_ioc_reblock reblock;
48         const char *filesystem;
49         int fd;
50         int perc;
51
52         if (TimeoutOpt > 0)
53                 alarm(TimeoutOpt);
54
55         bzero(&reblock, sizeof(reblock));
56
57         reblock.key_beg.localization = HAMMER_MIN_LOCALIZATION;
58         reblock.key_beg.obj_id = HAMMER_MIN_OBJID;
59         hammer_get_cycle(&reblock.key_beg, NULL);
60
61         reblock.key_end.localization = HAMMER_MAX_LOCALIZATION;
62         reblock.key_end.obj_id = HAMMER_MAX_OBJID;
63
64         reblock.head.flags = flags & HAMMER_IOC_DO_FLAGS;
65
66         /*
67          * Restrict the localization domain if asked to do inodes or data,
68          * but not both.
69          */
70         switch(flags & (HAMMER_IOC_DO_INODES|HAMMER_IOC_DO_DATA|HAMMER_IOC_DO_DIRS)) {
71         case HAMMER_IOC_DO_INODES:
72                 reblock.key_beg.localization = HAMMER_LOCALIZE_INODE;
73                 reblock.key_end.localization = HAMMER_LOCALIZE_INODE;
74                 break;
75         case HAMMER_IOC_DO_DIRS:
76         case HAMMER_IOC_DO_DATA:
77                 reblock.key_beg.localization = HAMMER_LOCALIZE_MISC;
78                 reblock.key_end.localization = HAMMER_LOCALIZE_MISC;
79                 break;
80         }
81
82         if (ac == 0)
83                 reblock_usage(1);
84         filesystem = av[0];
85         if (ac == 1) {
86                 perc = 100;
87         } else {
88                 perc = strtol(av[1], NULL, 0);
89                 if (perc < 0 || perc > 100)
90                         reblock_usage(1);
91         }
92         reblock.free_level = (int)((int64_t)perc *
93                                    HAMMER_LARGEBLOCK_SIZE / 100);
94         reblock.free_level = HAMMER_LARGEBLOCK_SIZE - reblock.free_level;
95         if (reblock.free_level < 0)
96                 reblock.free_level = 0;
97         printf("reblock start %016jx:%04x free level %d\n",
98                 (uintmax_t)reblock.key_beg.obj_id,
99                 reblock.key_beg.localization,
100                 reblock.free_level);
101
102         fd = open(filesystem, O_RDONLY);
103         if (fd < 0)
104                 err(1, "Unable to open %s", filesystem);
105         RunningIoctl = 1;
106         if (ioctl(fd, HAMMERIOC_REBLOCK, &reblock) < 0) {
107                 printf("Reblock %s failed: %s\n", filesystem, strerror(errno));
108         } else if (reblock.head.flags & HAMMER_IOC_HEAD_INTR) {
109                 printf("Reblock %s interrupted by timer at %016jx:%04x\n",
110                         filesystem,
111                         (uintmax_t)reblock.key_cur.obj_id,
112                         reblock.key_cur.localization);
113                 if (CyclePath) {
114                         hammer_set_cycle(&reblock.key_cur, 0);
115                 }
116         } else {
117                 if (CyclePath)
118                         hammer_reset_cycle();
119                 printf("Reblock %s succeeded\n", filesystem);
120         }
121         RunningIoctl = 0;
122         close(fd);
123         printf("Reblocked:\n"
124                "    %jd/%jd btree nodes\n"
125                "    %jd/%jd data elements\n"
126                "    %jd/%jd data bytes\n",
127                (intmax_t)reblock.btree_moves, (intmax_t)reblock.btree_count,
128                (intmax_t)reblock.data_moves, (intmax_t)reblock.data_count,
129                (intmax_t)reblock.data_byte_moves,
130                (intmax_t)reblock.data_byte_count
131         );
132 }
133
134 static
135 void
136 reblock_usage(int exit_code)
137 {
138         fprintf(stderr, "hammer reblock <filesystem> [fill_percentage]\n");
139         fprintf(stderr, "hammer reblock-btree <filesystem> [fill_percentage]\n");
140         fprintf(stderr, "hammer reblock-inodes <filesystem> [fill_percentage]\n");
141         fprintf(stderr, "hammer reblock-dirs <filesystem> [fill_percentage]\n");
142         fprintf(stderr, "hammer reblock-data <filesystem> [fill_percentage]\n");
143         fprintf(stderr, "By default 100%% is used.\n");
144         exit(exit_code);
145 }
146