[分享] ext2文件系统中inode属性修改

[复制链接]

该用户从未签到

472

主题

532

回帖

1万

积分

[INTOHARD]旅长

Rank: 9Rank: 9Rank: 9

积分
16542
发表于 2015-1-21 15:20:24 | 显示全部楼层 |阅读模式
  1. #include <stdio.h>
  2. #include <unistd.h>
  3. #include <stdlib.h>
  4. #include <ctype.h>
  5. #include <string.h>
  6. #include <time.h>
  7. #include <getopt.h>
  8. #include <errno.h>
  9. #include <fcntl.h>
  10. #include <sys/types.h>
  11. #include <sys/stat.h>
  12. #include <ext2fs/ext2fs.h>



  13. /*
  14. * Flags used by the common argument processing functions
  15. */
  16. #define CHECK_FS_RW                0x0001
  17. #define CHECK_FS_BITMAPS        0x0002
  18. #define CHECK_FS_NOTOPEN        0x0004


  19. #ifdef DLL_EXPORT
  20. #define DLLEXPORT __declspec(dllexport)
  21. #else
  22. #define DLLEXPORT
  23. #endif

  24. ext2_filsys        current_fs = NULL;
  25. ext2_ino_t        root, cwd;

  26. DLLEXPORT
  27. int open_filesystem(char *device, int open_flags, blk_t superblock,
  28.                             blk_t blocksize, int catastrophic)
  29. {
  30.         int        retval;

  31.         if (superblock != 0 && blocksize == 0) {
  32.                 com_err(device, 0, "if you specify the superblock, you must also specify the block size");
  33.                 current_fs = NULL;
  34.                 return;
  35.         }

  36.         if (catastrophic && (open_flags & EXT2_FLAG_RW)) {
  37.                 com_err(device, 0,
  38.                         "opening read-only because of catastrophic mode");
  39.                 open_flags &= ~EXT2_FLAG_RW;
  40.         }
  41.        
  42.         retval = ext2fs_open(device, open_flags, superblock, blocksize,
  43.                              nt_io_manager(), ¤t_fs);
  44.         if (retval) {
  45.                 com_err(device, retval, "while opening filesystem");
  46.                 current_fs = NULL;
  47.                 return;
  48.         }

  49.         if (catastrophic)
  50.                 com_err(device, 0, "catastrophic mode - not reading inode or group bitmaps");
  51.         else {
  52.                 retval = ext2fs_read_inode_bitmap(current_fs);
  53.                 if (retval) {
  54.                         com_err(device, retval, "while reading inode bitmap");
  55.                         goto errout;
  56.                 }
  57.                 retval = ext2fs_read_block_bitmap(current_fs);
  58.                 if (retval) {
  59.                         com_err(device, retval, "while reading block bitmap");
  60.                         goto errout;
  61.                 }
  62.         }
  63.         root = cwd = EXT2_ROOT_INO;
  64.         return 0;

  65. errout:
  66.         retval = ext2fs_close(current_fs);
  67.         if (retval)
  68.                 com_err(device, retval, "while trying to close filesystem");
  69.         current_fs = NULL;
  70.         return -1;
  71. }

  72. DLLEXPORT
  73. void close_filesystem(void)
  74. {
  75.         int        retval;
  76.        
  77.         if (current_fs->flags & EXT2_FLAG_IB_DIRTY) {
  78.                 retval = ext2fs_write_inode_bitmap(current_fs);
  79.                 if (retval)
  80.                         com_err("ext2fs_write_inode_bitmap", retval, "");
  81.         }
  82.         if (current_fs->flags & EXT2_FLAG_BB_DIRTY) {
  83.                 retval = ext2fs_write_block_bitmap(current_fs);
  84.                 if (retval)
  85.                         com_err("ext2fs_write_block_bitmap", retval, "");
  86.         }
  87.         retval = ext2fs_close(current_fs);
  88.         if (retval)
  89.                 com_err("ext2fs_close", retval, "");
  90.         current_fs = NULL;
  91.         return;
  92. }


  93. /*
  94. * This routine is used whenever a command needs to turn a string into
  95. * an inode.
  96. */
  97. ext2_ino_t string_to_inode(char *str)
  98. {
  99.         ext2_ino_t        ino;
  100.         int                len = strlen(str);
  101.         char                *end;
  102.         int                retval;

  103.         /*
  104.          * If the string is of the form <ino>, then treat it as an
  105.          * inode number.
  106.          */
  107.         if ((len > 2) && (str[0] == '<') && (str[len-1] == '>')) {
  108.                 ino = strtoul(str+1, &end, 0);
  109.                 if (*end=='>')
  110.                         return ino;
  111.         }

  112.         retval = ext2fs_namei(current_fs, root, cwd, str, &ino);
  113.         if (retval) {
  114.                 com_err(str, retval, "");
  115.                 return 0;
  116.         }
  117.         return ino;
  118. }

  119. /*
  120. * This routine returns 1 if the filesystem is not open, and prints an
  121. * error message to that effect.
  122. */
  123. int check_fs_open(char *name)
  124. {
  125.         if (!current_fs) {
  126.                 com_err(name, 0, "Filesystem not open");
  127.                 return 1;
  128.         }
  129.         return 0;
  130. }

  131. /*
  132. * This routine returns 1 if a filesystem is open, and prints an
  133. * error message to that effect.
  134. */
  135. int check_fs_not_open(char *name)
  136. {
  137.         if (current_fs) {
  138.                 com_err(name, 0,
  139.                         "Filesystem %s is still open.  Close it first.\n",
  140.                         current_fs->device_name);
  141.                 return 1;
  142.         }
  143.         return 0;
  144. }

  145. /*
  146. * This routine returns 1 if a filesystem is not opened read/write,
  147. * and prints an error message to that effect.
  148. */
  149. int check_fs_read_write(char *name)
  150. {
  151.         if (!(current_fs->flags & EXT2_FLAG_RW)) {
  152.                 com_err(name, 0, "Filesystem opened read/only");
  153.                 return 1;
  154.         }
  155.         return 0;
  156. }
  157. /*
  158. * This routine returns 1 if a filesystem is doesn't have its inode
  159. * and block bitmaps loaded, and prints an error message to that
  160. * effect.
  161. */
  162. int check_fs_bitmaps(char *name)
  163. {
  164.         if (!current_fs->block_map || !current_fs->inode_map) {
  165.                 com_err(name, 0, "Filesystem bitmaps not loaded");
  166.                 return 1;
  167.         }
  168.         return 0;
  169. }

  170. /*
  171. * This function takes a __u32 time value and converts it to a string,
  172. * using ctime
  173. */
  174. char *time_to_string(__u32 cl)
  175. {
  176.         time_t        t = (time_t) cl;

  177.         return ctime(&t);
  178. }


  179. int debugfs_read_inode(ext2_ino_t ino, struct ext2_inode * inode,
  180.                         const char *cmd)
  181. {
  182.         int retval;

  183.         retval = ext2fs_read_inode(current_fs, ino, inode);
  184.         if (retval) {
  185.                 com_err(cmd, retval, "while reading inode %u", ino);
  186.                 return 1;
  187.         }
  188.         return 0;
  189. }

  190. int debugfs_write_inode(ext2_ino_t ino, struct ext2_inode * inode,
  191.                         const char *cmd)
  192. {
  193.         int retval;

  194.         retval = ext2fs_write_inode(current_fs, ino, inode);
  195.         if (retval) {
  196.                 com_err(cmd, retval, "while writing inode %u", ino);
  197.                 return 1;
  198.         }
  199.         return 0;
  200. }

  201. /*
  202. * This is a common helper function used by the command processing
  203. * routines
  204. */
  205. int common_args_process(int argc, char *argv[], int min_argc, int max_argc,
  206.                         const char *cmd, const char *usage, int flags)
  207. {
  208.         if (argc < min_argc || argc > max_argc) {
  209.                 com_err(argv[0], 0, "Usage: %s %s", cmd, usage);
  210.                 return 1;
  211.         }
  212.         if (flags & CHECK_FS_NOTOPEN) {
  213.                 if (check_fs_not_open(argv[0]))
  214.                         return 1;
  215.         } else {
  216.                 if (check_fs_open(argv[0]))
  217.                         return 1;
  218.         }
  219.         if ((flags & CHECK_FS_RW) && check_fs_read_write(argv[0]))
  220.                 return 1;
  221.         if ((flags & CHECK_FS_BITMAPS) && check_fs_bitmaps(argv[0]))
  222.                 return 1;
  223.         return 0;
  224. }


  225. /*
  226. * This is a helper function used by do_stat, do_freei, do_seti, and
  227. * do_testi, etc.  Basically, any command which takes a single
  228. * argument which is a file/inode number specifier.
  229. */
  230. int common_inode_args_process(int argc, char *argv[],
  231.                               ext2_ino_t *inode, int flags)
  232. {
  233.         if (common_args_process(argc, argv, 2, 2, argv[0], "<file>", flags))
  234.                 return 1;
  235.        
  236.         *inode = string_to_inode(argv[1]);
  237.         if (!*inode)
  238.                 return 1;
  239.         return 0;
  240. }

  241. static void modify_u16(char *com, const char *prompt,
  242.                        const char *format, __u16 *val)
  243. {
  244.         char buf[200];
  245.         unsigned long v;
  246.         char *tmp;

  247.         sprintf(buf, format, *val);
  248.         printf("%30s    [%s] ", prompt, buf);
  249.         fgets(buf, sizeof(buf), stdin);
  250.         if (buf[strlen (buf) - 1] == '\n')
  251.                 buf[strlen (buf) - 1] = '\0';
  252.         if (!buf[0])
  253.                 return;
  254.         v = strtoul(buf, &tmp, 0);
  255.         if (*tmp)
  256.                 com_err(com, 0, "Bad value - %s", buf);
  257.         else
  258.                 *val = v;
  259. }

  260. static void modify_u32(char *com, const char *prompt,
  261.                        const char *format, __u32 *val)
  262. {
  263.         char buf[200];
  264.         unsigned long v;
  265.         char *tmp;

  266.         sprintf(buf, format, *val);
  267.         printf("%30s    [%s] ", prompt, buf);
  268.         fgets(buf, sizeof(buf), stdin);
  269.         if (buf[strlen (buf) - 1] == '\n')
  270.                 buf[strlen (buf) - 1] = '\0';
  271.         if (!buf[0])
  272.                 return;
  273.         v = strtoul(buf, &tmp, 0);
  274.         if (*tmp)
  275.                 com_err(com, 0, "Bad value - %s", buf);
  276.         else
  277.                 *val = v;
  278. }

  279. #define COMMAND_MI "mi"
  280. DLLEXPORT
  281. void do_modify_inode(int readonly, char *filename, unsigned long mode)
  282. {
  283.         struct ext2_inode inode;
  284.         ext2_ino_t        inode_num;
  285.         int                 i;
  286.         unsigned char        *frag, *fsize;
  287.         char                buf[80];
  288.         int                 os;
  289.         const char        *hex_format = "0x%x";
  290.         const char        *octal_format = "0%o";
  291.         const char        *decimal_format = "%d";
  292.        
  293.         inode_num = string_to_inode(filename);
  294.         if (!inode_num)
  295.                 return ;
  296. /*               
  297.         if (common_inode_args_process(argc, argv, &inode_num, CHECK_FS_RW))
  298.                 return;
  299. */
  300.         os = current_fs->super->s_creator_os;

  301.         if (debugfs_read_inode(inode_num, &inode, filename))
  302.                 return;
  303.        
  304.         printf("Current Mode   [0%o] \n", inode.i_mode);
  305.         inode.i_mode = mode;
  306.                
  307. #if 0                       
  308.         modify_u16(COMMAND_MI, "Mode", octal_format, &inode.i_mode);

  309.         modify_u16(COMMAND_MI, "User ID", decimal_format, &inode.i_uid);
  310.         modify_u16(COMMAND_MI, "Group ID", decimal_format, &inode.i_gid);

  311.         modify_u32(COMMAND_MI, "Size", decimal_format, &inode.i_size);
  312.         modify_u32(COMMAND_MI, "Creation time", decimal_format, &inode.i_ctime);
  313.         modify_u32(COMMAND_MI, "Modification time", decimal_format, &inode.i_mtime);
  314.         modify_u32(COMMAND_MI, "Access time", decimal_format, &inode.i_atime);
  315.         modify_u32(COMMAND_MI, "Deletion time", decimal_format, &inode.i_dtime);
  316.         modify_u16(COMMAND_MI, "Link count", decimal_format, &inode.i_links_count);
  317.         modify_u32(COMMAND_MI, "Block count", decimal_format, &inode.i_blocks);
  318.         modify_u32(COMMAND_MI, "File flags", hex_format, &inode.i_flags);
  319.         modify_u32(COMMAND_MI, "Generation", hex_format, &inode.i_generation);
  320. #if 0         
  321.         modify_u32(COMMAND_MI, "Reserved1", decimal_format, &inode.i_reserved1);
  322. #endif
  323.         modify_u32(COMMAND_MI, "File acl", decimal_format, &inode.i_file_acl);
  324.         if (LINUX_S_ISDIR(inode.i_mode))
  325.                 modify_u32(argv[0], "Directory acl", decimal_format, &inode.i_dir_acl);
  326.         else
  327.                 modify_u32(argv[0], "High 32bits of size", decimal_format, &inode.i_size_high);

  328.         if (current_fs->super->s_creator_os == EXT2_OS_HURD)
  329.                 modify_u32(argv[0], "Translator Block",
  330.                             decimal_format, &inode.osd1.hurd1.h_i_translator);
  331.        
  332.         modify_u32(argv[0], "Fragment address", decimal_format, &inode.i_faddr);
  333.         switch (os) {
  334.             case EXT2_OS_LINUX:
  335.                 frag = &inode.osd2.linux2.l_i_frag;
  336.                 fsize = &inode.osd2.linux2.l_i_fsize;
  337.                 break;
  338.             case EXT2_OS_HURD:
  339.                 frag = &inode.osd2.hurd2.h_i_frag;
  340.                 fsize = &inode.osd2.hurd2.h_i_fsize;
  341.                 break;
  342.             case EXT2_OS_MASIX:
  343.                 frag = &inode.osd2.masix2.m_i_frag;
  344.                 fsize = &inode.osd2.masix2.m_i_fsize;
  345.                 break;
  346.             default:
  347.                 frag = fsize = 0;
  348.         }
  349.         if (frag)
  350.                 modify_u8(argv[0], "Fragment number", decimal_format, frag);
  351.         if (fsize)
  352.                 modify_u8(argv[0], "Fragment size", decimal_format, fsize);

  353.         for (i=0;  i < EXT2_NDIR_BLOCKS; i++) {
  354.                 sprintf(buf, "Direct Block #%d", i);
  355.                 modify_u32(argv[0], buf, decimal_format, &inode.i_block[i]);
  356.         }
  357.         modify_u32(argv[0], "Indirect Block", decimal_format,
  358.                     &inode.i_block[EXT2_IND_BLOCK]);   
  359.         modify_u32(argv[0], "Double Indirect Block", decimal_format,
  360.                     &inode.i_block[EXT2_DIND_BLOCK]);
  361.         modify_u32(argv[0], "Triple Indirect Block", decimal_format,
  362.                     &inode.i_block[EXT2_TIND_BLOCK]);
  363. #endif

  364.         if (readonly)
  365.                 return;
  366.                
  367.         if (debugfs_write_inode(inode_num, &inode, filename))
  368.                 return;
  369. }


  370. /*
  371. * This function will convert a string to an unsigned long, printing
  372. * an error message if it fails, and returning success or failure in err.
  373. */
  374. unsigned long parse_ulong(const char *str, const char *cmd,
  375.                           const char *descr, int *err)
  376. {
  377.         char                *tmp;
  378.         unsigned long        ret;
  379.        
  380.         ret = strtoul(str, &tmp, 0);
  381.         if (*tmp == 0) {
  382.                 if (err)
  383.                         *err = 0;
  384.                 return ret;
  385.         }
  386.         com_err(cmd, 0, "Bad %s - %s", descr, str);
  387.         if (err)
  388.                 *err = 1;
  389.         else
  390.                 exit(1);
  391.         return 0;
  392. }

  393. /*
  394. * This function will convert a string to a block number.  It returns
  395. * 0 on success, 1 on failure.
  396. */
  397. int strtoblk(const char *cmd, const char *str, blk_t *ret)
  398. {
  399.         blk_t        blk;
  400.         int        err;

  401.         blk = parse_ulong(str, cmd, "block number", &err);
  402.         *ret = blk;
  403.         if (err == 0 && blk == 0) {
  404.                 com_err(cmd, 0, "Invalid block number 0");
  405.                 err = 1;
  406.         }
  407.         return err;
  408. }

  409. #ifdef MOD_EXE
  410. int main(int argc, char **argv)
  411. {
  412.         int        catastrophic = 0;
  413.         blk_t        superblock = 0;
  414.         blk_t        blocksize = 0;
  415.         int open_flags = 0;
  416.         char *file_path = NULL;
  417.         char *dev_path = NULL;
  418.         int c = 0;
  419.         const char        *usage = "Usage: chmod [-b blocksize] [-s superblock] [-f file] [-F] [-V] [-c] [-i] [-r] [-o device]";
  420.         unsigned long mode;
  421.         char *tmp = NULL;
  422.         int readonly = 0;
  423.        
  424.         open_flags |= EXT2_FLAG_RW;
  425.        
  426.         while ((c = getopt (argc, argv, "icRF:f:b:s:o:Vr")) != EOF) {
  427.                 switch (c) {
  428.                 case 'F':
  429.                         open_flags |= EXT2_FLAG_FORCE;
  430.                         break;                       
  431.                 case 'f':
  432.                         file_path = optarg;
  433.                         break;
  434.                 case 'i':
  435.                         open_flags |= EXT2_FLAG_IMAGE_FILE;
  436.                         break;
  437.                 case 'o':
  438.                         dev_path = optarg;
  439.                         break;                       
  440.                 case 'b':
  441.                         blocksize = parse_ulong(optarg, argv[0],
  442.                                                 "block size", 0);
  443.                         break;
  444.                 case 's':
  445.                         superblock = parse_ulong(optarg, argv[0],
  446.                                                  "superblock number", 0);
  447.                         break;
  448.                 case 'c':
  449.                         catastrophic = 1;
  450.                         break;
  451.                        
  452.                 case 'r':
  453.                         readonly = 1;
  454.                         break;
  455.                        
  456.                 case 'V':
  457.                         /* Print version number and exit */
  458.                         fprintf(stderr, "\tUsing %s\n",
  459.                                 error_message(EXT2_ET_BASE));
  460.                         exit(0);
  461.                 default:
  462.                         com_err(argv[0], 0, usage);
  463.                         return 1;
  464.                 }
  465.         }
  466.         if (!file_path || !dev_path || (!readonly && !argv[optind])){
  467.                 com_err(argv[0], 0, usage);
  468.                 return 1;
  469.         }

  470.         if (!readonly) {
  471.                 printf("Change Mode   [%s] \n", argv[optind]);
  472.                 mode = strtoul(argv[optind], &tmp, 0);
  473.                 if (*tmp){
  474.                         com_err(argv[0], 0, "Bad Mode value - %s", argv[optind]);
  475.                         return 1;
  476.                 }
  477.         }
  478.         open_filesystem(dev_path, open_flags,
  479.                         superblock, blocksize, catastrophic);
  480.                        
  481.         if (!current_fs){
  482.                 return 1;
  483.         }
  484.         do_modify_inode(readonly, file_path, mode);
  485.        
  486.         close_filesystem();
  487.         return 0;
  488. }


  489. #endif
复制代码


参考:
e2fsprogs
MinGW
Msys
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表