1 /* 2 * util.c --- utilities for the debugfs program 3 * 4 * Copyright (C) 1993, 1994 Theodore Ts'o. This file may be 5 * redistributed under the terms of the GNU Public License. 6 * 7 */ 8 9 #define _XOPEN_SOURCE 600 /* needed for strptime */ 10 11 #include "config.h" 12 #include <stdio.h> 13 #include <unistd.h> 14 #include <stdlib.h> 15 #include <ctype.h> 16 #include <string.h> 17 #include <time.h> 18 #include <signal.h> 19 #ifdef HAVE_GETOPT_H 20 #include <getopt.h> 21 #else 22 extern int optind; 23 extern char *optarg; 24 #endif 25 #ifdef HAVE_OPTRESET 26 extern int optreset; /* defined by BSD, but not others */ 27 #endif 28 29 #include "ss/ss.h" 30 #include "debugfs.h" 31 32 /* 33 * This function resets the libc getopt() function, which keeps 34 * internal state. Bad design! Stupid libc API designers! No 35 * biscuit! 36 * 37 * BSD-derived getopt() functions require that optind be reset to 1 in 38 * order to reset getopt() state. This used to be generally accepted 39 * way of resetting getopt(). However, glibc's getopt() 40 * has additional getopt() state beyond optind, and requires that 41 * optind be set zero to reset its state. So the unfortunate state of 42 * affairs is that BSD-derived versions of getopt() misbehave if 43 * optind is set to 0 in order to reset getopt(), and glibc's getopt() 44 * will core dump if optind is set 1 in order to reset getopt(). 45 * 46 * More modern versions of BSD require that optreset be set to 1 in 47 * order to reset getopt(). Sigh. Standards, anyone? 48 * 49 * We hide the hair here. 50 */ 51 void reset_getopt(void) 52 { 53 #if defined(__GLIBC__) || defined(__linux__) 54 optind = 0; 55 #else 56 optind = 1; 57 #endif 58 #ifdef HAVE_OPTRESET 59 optreset = 1; /* Makes BSD getopt happy */ 60 #endif 61 } 62 63 static const char *pager_search_list[] = { "pager", "more", "less", 0 }; 64 static const char *pager_dir_list[] = { "/usr/bin", "/bin", 0 }; 65 66 static const char *find_pager(char *buf) 67 { 68 const char **i, **j; 69 70 for (i = pager_search_list; *i; i++) { 71 for (j = pager_dir_list; *j; j++) { 72 sprintf(buf, "%s/%s", *j, *i); 73 if (access(buf, X_OK) == 0) 74 return(buf); 75 } 76 } 77 return 0; 78 } 79 80 FILE *open_pager(void) 81 { 82 FILE *outfile = 0; 83 const char *pager = ss_safe_getenv("DEBUGFS_PAGER"); 84 char buf[80]; 85 86 signal(SIGPIPE, SIG_IGN); 87 if (!isatty(1)) 88 return stdout; 89 if (!pager) 90 pager = ss_safe_getenv("PAGER"); 91 if (!pager) 92 pager = find_pager(buf); 93 if (!pager || 94 (strcmp(pager, "__none__") == 0) || 95 ((outfile = popen(pager, "w")) == 0)) 96 return stdout; 97 return outfile; 98 } 99 100 void close_pager(FILE *stream) 101 { 102 if (stream && stream != stdout) pclose(stream); 103 } 104 105 /* 106 * This routine is used whenever a command needs to turn a string into 107 * an inode. 108 */ 109 ext2_ino_t string_to_inode(char *str) 110 { 111 ext2_ino_t ino; 112 int len = strlen(str); 113 char *end; 114 int retval; 115 116 /* 117 * If the string is of the form <ino>, then treat it as an 118 * inode number. 119 */ 120 if ((len > 2) && (str[0] == '<') && (str[len-1] == '>')) { 121 ino = strtoul(str+1, &end, 0); 122 if (*end=='>') 123 return ino; 124 } 125 126 retval = ext2fs_namei(current_fs, root, cwd, str, &ino); 127 if (retval) { 128 com_err(str, retval, 0); 129 return 0; 130 } 131 return ino; 132 } 133 134 /* 135 * This routine returns 1 if the filesystem is not open, and prints an 136 * error message to that effect. 137 */ 138 int check_fs_open(char *name) 139 { 140 if (!current_fs) { 141 com_err(name, 0, "Filesystem not open"); 142 return 1; 143 } 144 return 0; 145 } 146 147 /* 148 * This routine returns 1 if a filesystem is open, and prints an 149 * error message to that effect. 150 */ 151 int check_fs_not_open(char *name) 152 { 153 if (current_fs) { 154 com_err(name, 0, 155 "Filesystem %s is still open. Close it first.\n", 156 current_fs->device_name); 157 return 1; 158 } 159 return 0; 160 } 161 162 /* 163 * This routine returns 1 if a filesystem is not opened read/write, 164 * and prints an error message to that effect. 165 */ 166 int check_fs_read_write(char *name) 167 { 168 if (!(current_fs->flags & EXT2_FLAG_RW)) { 169 com_err(name, 0, "Filesystem opened read/only"); 170 return 1; 171 } 172 return 0; 173 } 174 175 /* 176 * This routine returns 1 if a filesystem is doesn't have its inode 177 * and block bitmaps loaded, and prints an error message to that 178 * effect. 179 */ 180 int check_fs_bitmaps(char *name) 181 { 182 if (!current_fs->block_map || !current_fs->inode_map) { 183 com_err(name, 0, "Filesystem bitmaps not loaded"); 184 return 1; 185 } 186 return 0; 187 } 188 189 char *inode_time_to_string(__u32 xtime, __u32 xtime_extra) 190 { 191 __s64 t = (__s32) xtime; 192 193 t += (__s64) (xtime_extra & EXT4_EPOCH_MASK) << 32; 194 return time_to_string(t); 195 } 196 197 /* 198 * This function takes a __s64 time value and converts it to a string, 199 * using ctime 200 */ 201 char *time_to_string(__s64 cl) 202 { 203 static int do_gmt = -1; 204 time_t t = (time_t) cl; 205 const char *tz; 206 207 if (do_gmt == -1) { 208 /* The diet libc doesn't respect the TZ environemnt variable */ 209 tz = ss_safe_getenv("TZ"); 210 if (!tz) 211 tz = ""; 212 do_gmt = !strcmp(tz, "GMT") || !strcmp(tz, "GMT0"); 213 } 214 215 return asctime((do_gmt) ? gmtime(&t) : localtime(&t)); 216 } 217 218 /* 219 * Parse a string as a time. Return ((time_t)-1) if the string 220 * doesn't appear to be a sane time. 221 */ 222 extern __s64 string_to_time(const char *arg) 223 { 224 struct tm ts; 225 __s64 ret; 226 char *tmp; 227 228 if (strcmp(arg, "now") == 0) { 229 return time(0); 230 } 231 if (arg[0] == '@') { 232 /* interpret it as an integer */ 233 arg++; 234 fallback: 235 ret = strtoll(arg, &tmp, 0); 236 if (*tmp) 237 return -1; 238 return ret; 239 } 240 memset(&ts, 0, sizeof(ts)); 241 #ifdef HAVE_STRPTIME 242 tmp = strptime(arg, "%Y%m%d%H%M%S", &ts); 243 if (tmp == NULL) 244 tmp = strptime(arg, "%Y%m%d%H%M", &ts); 245 if (tmp == NULL) 246 tmp = strptime(arg, "%Y%m%d", &ts); 247 if (tmp == NULL) 248 goto fallback; 249 #else 250 sscanf(arg, "%4d%2d%2d%2d%2d%2d", &ts.tm_year, &ts.tm_mon, 251 &ts.tm_mday, &ts.tm_hour, &ts.tm_min, &ts.tm_sec); 252 ts.tm_year -= 1900; 253 ts.tm_mon -= 1; 254 if (ts.tm_year < 0 || ts.tm_mon < 0 || ts.tm_mon > 11 || 255 ts.tm_mday <= 0 || ts.tm_mday > 31 || ts.tm_hour > 23 || 256 ts.tm_min > 59 || ts.tm_sec > 61) 257 goto fallback; 258 #endif 259 ts.tm_isdst = -1; 260 /* strptime() may only update the specified fields, which does not 261 * necessarily include ts.tm_yday (%j). Calculate this if unset: 262 * 263 * Jan Feb Mar Apr May Jun Jul Aug Sep Oct Nov Dec 264 * 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 265 * 266 * Start with 31 days per month. Even months have only 30 days, but 267 * reverse in August, subtract one day for those months. February has 268 * only 28 days, not 30, subtract two days. Add day of month, minus 269 * one, since day is not finished yet. Leap years handled afterward. */ 270 if (ts.tm_yday == 0) 271 ts.tm_yday = (ts.tm_mon * 31) - 272 ((ts.tm_mon - (ts.tm_mon > 7)) / 2) - 273 2 * (ts.tm_mon > 1) + ts.tm_mday - 1; 274 ret = ts.tm_sec + ts.tm_min*60 + ts.tm_hour*3600 + ts.tm_yday*86400 + 275 ((__s64) ts.tm_year-70)*31536000 + 276 (((__s64) ts.tm_year-69)/4)*86400 - 277 (((__s64) ts.tm_year-1)/100)*86400 + 278 (((__s64) ts.tm_year+299)/400)*86400; 279 return ret; 280 } 281 282 /* 283 * This function will convert a string to an unsigned long, printing 284 * an error message if it fails, and returning success or failure in err. 285 */ 286 unsigned long parse_ulong(const char *str, const char *cmd, 287 const char *descr, int *err) 288 { 289 char *tmp; 290 unsigned long ret; 291 292 ret = strtoul(str, &tmp, 0); 293 if (*tmp == 0) { 294 if (err) 295 *err = 0; 296 return ret; 297 } 298 com_err(cmd, 0, "Bad %s - %s", descr, str); 299 if (err) 300 *err = 1; 301 else 302 exit(1); 303 return 0; 304 } 305 306 /* 307 * This function will convert a string to an unsigned long long, printing 308 * an error message if it fails, and returning success or failure in err. 309 */ 310 unsigned long long parse_ulonglong(const char *str, const char *cmd, 311 const char *descr, int *err) 312 { 313 char *tmp; 314 unsigned long long ret; 315 316 ret = strtoull(str, &tmp, 0); 317 if (*tmp == 0) { 318 if (err) 319 *err = 0; 320 return ret; 321 } 322 com_err(cmd, 0, "Bad %s - %s", descr, str); 323 if (err) 324 *err = 1; 325 else 326 exit(1); 327 return 0; 328 } 329 330 /* 331 * This function will convert a string to a block number. It returns 332 * 0 on success, 1 on failure. On failure, it outputs either an optionally 333 * specified error message or a default. 334 */ 335 int strtoblk(const char *cmd, const char *str, const char *errmsg, 336 blk64_t *ret) 337 { 338 blk64_t blk; 339 int err; 340 341 if (errmsg == NULL) 342 blk = parse_ulonglong(str, cmd, "block number", &err); 343 else 344 blk = parse_ulonglong(str, cmd, errmsg, &err); 345 *ret = blk; 346 return err; 347 } 348 349 /* 350 * This is a common helper function used by the command processing 351 * routines 352 */ 353 int common_args_process(int argc, char *argv[], int min_argc, int max_argc, 354 const char *cmd, const char *usage, int flags) 355 { 356 if (argc < min_argc || argc > max_argc) { 357 com_err(argv[0], 0, "Usage: %s %s", cmd, usage); 358 return 1; 359 } 360 if (flags & CHECK_FS_NOTOPEN) { 361 if (check_fs_not_open(argv[0])) 362 return 1; 363 } else { 364 if (check_fs_open(argv[0])) 365 return 1; 366 } 367 if ((flags & CHECK_FS_RW) && check_fs_read_write(argv[0])) 368 return 1; 369 if ((flags & CHECK_FS_BITMAPS) && check_fs_bitmaps(argv[0])) 370 return 1; 371 return 0; 372 } 373 374 /* 375 * This is a helper function used by do_stat, do_freei, do_seti, and 376 * do_testi, etc. Basically, any command which takes a single 377 * argument which is a file/inode number specifier. 378 */ 379 int common_inode_args_process(int argc, char *argv[], 380 ext2_ino_t *inode, int flags) 381 { 382 if (common_args_process(argc, argv, 2, 2, argv[0], "<file>", flags)) 383 return 1; 384 385 *inode = string_to_inode(argv[1]); 386 if (!*inode) 387 return 1; 388 return 0; 389 } 390 391 /* 392 * This is a helper function used by do_freeb, do_setb, and do_testb 393 */ 394 int common_block_args_process(int argc, char *argv[], 395 blk64_t *block, blk64_t *count) 396 { 397 int err; 398 399 if (common_args_process(argc, argv, 2, 3, argv[0], 400 "<block> [count]", CHECK_FS_BITMAPS)) 401 return 1; 402 403 if (strtoblk(argv[0], argv[1], NULL, block)) 404 return 1; 405 if (*block == 0) { 406 com_err(argv[0], 0, "Invalid block number 0"); 407 return 1; 408 } 409 410 if (argc > 2) { 411 err = strtoblk(argv[0], argv[2], "count", count); 412 if (err) 413 return 1; 414 } 415 return 0; 416 } 417 418 int debugfs_read_inode_full(ext2_ino_t ino, struct ext2_inode * inode, 419 const char *cmd, int bufsize) 420 { 421 int retval; 422 423 retval = ext2fs_read_inode_full(current_fs, ino, inode, bufsize); 424 if (retval) { 425 com_err(cmd, retval, "while reading inode %u", ino); 426 return 1; 427 } 428 return 0; 429 } 430 431 int debugfs_read_inode(ext2_ino_t ino, struct ext2_inode * inode, 432 const char *cmd) 433 { 434 int retval; 435 436 retval = ext2fs_read_inode(current_fs, ino, inode); 437 if (retval) { 438 com_err(cmd, retval, "while reading inode %u", ino); 439 return 1; 440 } 441 return 0; 442 } 443 444 int debugfs_write_inode_full(ext2_ino_t ino, 445 struct ext2_inode *inode, 446 const char *cmd, 447 int bufsize) 448 { 449 int retval; 450 451 retval = ext2fs_write_inode_full(current_fs, ino, 452 inode, bufsize); 453 if (retval) { 454 com_err(cmd, retval, "while writing inode %u", ino); 455 return 1; 456 } 457 return 0; 458 } 459 460 int debugfs_write_inode(ext2_ino_t ino, struct ext2_inode * inode, 461 const char *cmd) 462 { 463 int retval; 464 465 retval = ext2fs_write_inode(current_fs, ino, inode); 466 if (retval) { 467 com_err(cmd, retval, "while writing inode %u", ino); 468 return 1; 469 } 470 return 0; 471 } 472 473 int debugfs_write_new_inode(ext2_ino_t ino, struct ext2_inode * inode, 474 const char *cmd) 475 { 476 int retval; 477 478 retval = ext2fs_write_new_inode(current_fs, ino, inode); 479 if (retval) { 480 com_err(cmd, retval, "while creating inode %u", ino); 481 return 1; 482 } 483 return 0; 484 } 485 486 /* 487 * Given a mode, return the ext2 file type 488 */ 489 int ext2_file_type(unsigned int mode) 490 { 491 if (LINUX_S_ISREG(mode)) 492 return EXT2_FT_REG_FILE; 493 494 if (LINUX_S_ISDIR(mode)) 495 return EXT2_FT_DIR; 496 497 if (LINUX_S_ISCHR(mode)) 498 return EXT2_FT_CHRDEV; 499 500 if (LINUX_S_ISBLK(mode)) 501 return EXT2_FT_BLKDEV; 502 503 if (LINUX_S_ISLNK(mode)) 504 return EXT2_FT_SYMLINK; 505 506 if (LINUX_S_ISFIFO(mode)) 507 return EXT2_FT_FIFO; 508 509 if (LINUX_S_ISSOCK(mode)) 510 return EXT2_FT_SOCK; 511 512 return 0; 513 } 514 515 errcode_t read_list(char *str, blk64_t **list, size_t *len) 516 { 517 blk64_t *lst = *list; 518 size_t ln = *len; 519 char *tok, *p = str; 520 errcode_t retval; 521 522 while ((tok = strtok(p, ","))) { 523 blk64_t *l; 524 blk64_t x, y; 525 char *e; 526 527 errno = 0; 528 y = x = strtoull(tok, &e, 0); 529 if (errno) 530 return errno; 531 if (*e == '-') { 532 y = strtoull(e + 1, NULL, 0); 533 if (errno) 534 return errno; 535 } else if (*e != 0) { 536 retval = EINVAL; 537 goto err; 538 } 539 if (y < x) { 540 retval = EINVAL; 541 goto err; 542 } 543 l = realloc(lst, sizeof(blk64_t) * (ln + y - x + 1)); 544 if (l == NULL) { 545 retval = ENOMEM; 546 goto err; 547 } 548 lst = l; 549 for (; x <= y; x++) 550 lst[ln++] = x; 551 p = NULL; 552 } 553 554 *list = lst; 555 *len = ln; 556 return 0; 557 err: 558 free(lst); 559 return retval; 560 } 561