Merge from vendor branch LIBARCHIVE:
[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.7 2008/06/14 01:44:11 dillon Exp $
35  */
36
37 #include "hammer.h"
38
39 static void reblock_usage(int exit_code);
40
41 /*
42  * reblock <filesystem> [compaction_precentage] (default 90%)
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         bzero(&reblock, sizeof(reblock));
53
54         reblock.beg_localization = HAMMER_MIN_LOCALIZATION;
55         reblock.beg_obj_id = HAMMER_MIN_OBJID;
56         hammer_get_cycle(&reblock.beg_obj_id, &reblock.beg_localization);
57
58         reblock.end_localization = HAMMER_MAX_LOCALIZATION;
59         reblock.end_obj_id = HAMMER_MAX_OBJID;
60
61         reblock.head.flags = flags & HAMMER_IOC_DO_FLAGS;
62
63         /*
64          * Restrict the localization domain if asked to do inodes or data,
65          * but not both.
66          */
67         switch(flags & (HAMMER_IOC_DO_INODES|HAMMER_IOC_DO_DATA|HAMMER_IOC_DO_DIRS)) {
68         case HAMMER_IOC_DO_INODES:
69                 reblock.beg_localization = HAMMER_LOCALIZE_INODE;
70                 reblock.end_localization = HAMMER_LOCALIZE_INODE;
71                 break;
72         case HAMMER_IOC_DO_DIRS:
73         case HAMMER_IOC_DO_DATA:
74                 reblock.beg_localization = HAMMER_LOCALIZE_MISC;
75                 reblock.end_localization = HAMMER_LOCALIZE_MISC;
76                 break;
77         }
78
79         if (ac == 0)
80                 reblock_usage(1);
81         filesystem = av[0];
82         if (ac == 1) {
83                 perc = 100;
84         } else {
85                 perc = strtol(av[1], NULL, 0);
86                 if (perc < 0 || perc > 100)
87                         reblock_usage(1);
88         }
89         reblock.free_level = (int)((int64_t)perc *
90                                    HAMMER_LARGEBLOCK_SIZE / 100);
91         reblock.free_level = HAMMER_LARGEBLOCK_SIZE - reblock.free_level;
92         if (reblock.free_level < 0)
93                 reblock.free_level = 0;
94         printf("reblock free level %d\n", reblock.free_level);
95
96         fd = open(filesystem, O_RDONLY);
97         if (fd < 0)
98                 err(1, "Unable to open %s", filesystem);
99         if (ioctl(fd, HAMMERIOC_REBLOCK, &reblock) < 0) {
100                 printf("Reblock %s failed: %s\n", filesystem, strerror(errno));
101         } else if (reblock.head.flags & HAMMER_IOC_HEAD_INTR) {
102                 printf("Reblock %s interrupted by timer at %016llx %04x\n",
103                         filesystem,
104                         reblock.cur_obj_id, reblock.cur_localization);
105                 if (CyclePath) {
106                         hammer_set_cycle(reblock.cur_obj_id,
107                                          reblock.cur_localization);
108                 }
109         } else {
110                 if (CyclePath)
111                         hammer_reset_cycle();
112                 printf("Reblock %s succeeded\n", filesystem);
113         }
114         close(fd);
115         printf("Reblocked:\n"
116                "    %lld/%lld btree nodes\n"
117                "    %lld/%lld data elements\n"
118                "    %lld/%lld data bytes\n",
119                reblock.btree_moves, reblock.btree_count,
120                reblock.data_moves, reblock.data_count,
121                reblock.data_byte_moves, reblock.data_byte_count
122         );
123 }
124
125 static
126 void
127 reblock_usage(int exit_code)
128 {
129         fprintf(stderr, "hammer reblock <filesystem> [percentage]\n");
130         fprintf(stderr, "hammer reblock-btree <filesystem> [percentage]\n");
131         fprintf(stderr, "hammer reblock-inodes <filesystem> [percentage]\n");
132         fprintf(stderr, "hammer reblock-dirs <filesystem> [percentage]\n");
133         fprintf(stderr, "hammer reblock-data <filesystem> [percentage]\n");
134         fprintf(stderr, "By default 90%% is used.  Use 100%% to defragment\n");
135         exit(exit_code);
136 }
137