wimlib.net Git - wimlib/blob - src/iterate_dir.c

   1 /*
   2  * iterate_dir.c
   3  *
   4  * Iterate through files in a WIM image.
   5  * This is the stable API; internal code can just use for_dentry_in_tree().
   6  */
   7
   8 /*
   9  * Copyright (C) 2013 Eric Biggers
  10  *
  11  * This file is part of wimlib, a library for working with WIM files.
  12  *
  13  * wimlib is free software; you can redistribute it and/or modify it under the
  14  * terms of the GNU General Public License as published by the Free
  15  * Software Foundation; either version 3 of the License, or (at your option)
  16  * any later version.
  17  *
  18  * wimlib is distributed in the hope that it will be useful, but WITHOUT ANY
  19  * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
  20  * A PARTICULAR PURPOSE. See the GNU General Public License for more
  21  * details.
  22  *
  23  * You should have received a copy of the GNU General Public License
  24  * along with wimlib; if not, see http://www.gnu.org/licenses/.
  25  */
  26
  27 #ifdef HAVE_CONFIG_H
  28 #  include "config.h"
  29 #endif
  30
  31 #include "wimlib.h"
  32 #include "wimlib/dentry.h"
  33 #include "wimlib/encoding.h"
  34 #include "wimlib/lookup_table.h"
  35 #include "wimlib/metadata.h"
  36 #include "wimlib/paths.h"
  37 #include "wimlib/security.h"
  38 #include "wimlib/timestamp.h"
  39 #include "wimlib/unix_data.h"
  40 #include "wimlib/util.h"
  41 #include "wimlib/wim.h"
  42
  43 static int
  44 init_wimlib_dentry(struct wimlib_dir_entry *wdentry, struct wim_dentry *dentry,
  45                    WIMStruct *wim, int flags)
  46 {
  47         int ret;
  48         size_t dummy;
  49         const struct wim_inode *inode = dentry->d_inode;
  50         struct wim_lookup_table_entry *lte;
  51         const u8 *hash;
  52         struct wimlib_unix_data unix_data;
  53
  54         ret = utf16le_get_tstr(dentry->file_name, dentry->file_name_nbytes,
  55                                &wdentry->filename, &dummy);
  56         if (ret)
  57                 return ret;
  58
  59         ret = utf16le_get_tstr(dentry->short_name, dentry->short_name_nbytes,
  60                                &wdentry->dos_name, &dummy);
  61         if (ret)
  62                 return ret;
  63
  64         ret = calculate_dentry_full_path(dentry);
  65         if (ret)
  66                 return ret;
  67         wdentry->full_path = dentry->_full_path;
  68
  69         for (struct wim_dentry *d = dentry; !dentry_is_root(d); d = d->parent)
  70                 wdentry->depth++;
  71
  72         if (inode->i_security_id >= 0) {
  73                 struct wim_security_data *sd;
  74
  75                 sd = wim_get_current_security_data(wim);
  76                 wdentry->security_descriptor = sd->descriptors[inode->i_security_id];
  77                 wdentry->security_descriptor_size = sd->sizes[inode->i_security_id];
  78         }
  79         wdentry->reparse_tag = inode->i_reparse_tag;
  80         wdentry->num_links = inode->i_nlink;
  81         wdentry->attributes = inode->i_attributes;
  82         wdentry->hard_link_group_id = inode->i_ino;
  83         wdentry->creation_time = wim_timestamp_to_timespec(inode->i_creation_time);
  84         wdentry->last_write_time = wim_timestamp_to_timespec(inode->i_last_write_time);
  85         wdentry->last_access_time = wim_timestamp_to_timespec(inode->i_last_access_time);
  86         if (inode_get_unix_data(inode, &unix_data)) {
  87                 wdentry->unix_uid = unix_data.uid;
  88                 wdentry->unix_gid = unix_data.gid;
  89                 wdentry->unix_mode = unix_data.mode;
  90         }
  91
  92         lte = inode_unnamed_lte(inode, wim->lookup_table);
  93         if (lte) {
  94                 lte_to_wimlib_resource_entry(lte, &wdentry->streams[0].resource);
  95         } else if (!is_zero_hash(hash = inode_unnamed_stream_hash(inode))) {
  96                 if (flags & WIMLIB_ITERATE_DIR_TREE_FLAG_RESOURCES_NEEDED)
  97                         return stream_not_found_error(inode, hash);
  98                 copy_hash(wdentry->streams[0].resource.sha1_hash, hash);
  99                 wdentry->streams[0].resource.is_missing = 1;
 100         }
 101
 102         for (unsigned i = 0; i < inode->i_num_ads; i++) {
 103                 if (!ads_entry_is_named_stream(&inode->i_ads_entries[i]))
 104                         continue;
 105                 lte = inode_stream_lte(inode, i + 1, wim->lookup_table);
 106                 wdentry->num_named_streams++;
 107                 if (lte) {
 108                         lte_to_wimlib_resource_entry(lte, &wdentry->streams[
 109                                                                 wdentry->num_named_streams].resource);
 110                 } else if (!is_zero_hash(hash = inode_stream_hash(inode, i + 1))) {
 111                         if (flags & WIMLIB_ITERATE_DIR_TREE_FLAG_RESOURCES_NEEDED)
 112                                 return stream_not_found_error(inode, hash);
 113                         copy_hash(wdentry->streams[
 114                                   wdentry->num_named_streams].resource.sha1_hash, hash);
 115                         wdentry->streams[
 116                                 wdentry->num_named_streams].resource.is_missing = 1;
 117                 }
 118
 119                 size_t dummy;
 120
 121                 ret = utf16le_get_tstr(inode->i_ads_entries[i].stream_name,
 122                                        inode->i_ads_entries[i].stream_name_nbytes,
 123                                        &wdentry->streams[
 124                                                wdentry->num_named_streams].stream_name,
 125                                        &dummy);
 126                 if (ret)
 127                         return ret;
 128         }
 129         return 0;
 130 }
 131
 132 static void
 133 free_wimlib_dentry(struct wimlib_dir_entry *wdentry)
 134 {
 135         utf16le_put_tstr(wdentry->filename);
 136         utf16le_put_tstr(wdentry->dos_name);
 137         for (unsigned i = 1; i <= wdentry->num_named_streams; i++)
 138                 utf16le_put_tstr(wdentry->streams[i].stream_name);
 139         FREE(wdentry);
 140 }
 141
 142 static int
 143 do_iterate_dir_tree(WIMStruct *wim,
 144                     struct wim_dentry *dentry, int flags,
 145                     wimlib_iterate_dir_tree_callback_t cb,
 146                     void *user_ctx)
 147 {
 148         struct wimlib_dir_entry *wdentry;
 149         int ret = WIMLIB_ERR_NOMEM;
 150
 151
 152         wdentry = CALLOC(1, sizeof(struct wimlib_dir_entry) +
 153                                   (1 + dentry->d_inode->i_num_ads) *
 154                                         sizeof(struct wimlib_stream_entry));
 155         if (wdentry == NULL)
 156                 goto out;
 157
 158         ret = init_wimlib_dentry(wdentry, dentry, wim, flags);
 159         if (ret)
 160                 goto out_free_wimlib_dentry;
 161
 162         if (!(flags & WIMLIB_ITERATE_DIR_TREE_FLAG_CHILDREN)) {
 163                 ret = (*cb)(wdentry, user_ctx);
 164                 if (ret)
 165                         goto out_free_wimlib_dentry;
 166         }
 167
 168         if (flags & (WIMLIB_ITERATE_DIR_TREE_FLAG_RECURSIVE |
 169                      WIMLIB_ITERATE_DIR_TREE_FLAG_CHILDREN))
 170         {
 171                 struct wim_dentry *child;
 172
 173                 ret = 0;
 174                 for_dentry_child(child, dentry) {
 175                         ret = do_iterate_dir_tree(wim, child,
 176                                                   flags & ~WIMLIB_ITERATE_DIR_TREE_FLAG_CHILDREN,
 177                                                   cb, user_ctx);
 178                         if (ret)
 179                                 break;
 180                 }
 181         }
 182 out_free_wimlib_dentry:
 183         free_wimlib_dentry(wdentry);
 184 out:
 185         return ret;
 186 }
 187
 188 struct image_iterate_dir_tree_ctx {
 189         const tchar *path;
 190         int flags;
 191         wimlib_iterate_dir_tree_callback_t cb;
 192         void *user_ctx;
 193 };
 194
 195
 196 static int
 197 image_do_iterate_dir_tree(WIMStruct *wim)
 198 {
 199         struct image_iterate_dir_tree_ctx *ctx = wim->private;
 200         struct wim_dentry *dentry;
 201
 202         dentry = get_dentry(wim, ctx->path, WIMLIB_CASE_PLATFORM_DEFAULT);
 203         if (dentry == NULL)
 204                 return WIMLIB_ERR_PATH_DOES_NOT_EXIST;
 205         return do_iterate_dir_tree(wim, dentry, ctx->flags, ctx->cb, ctx->user_ctx);
 206 }
 207
 208 /* API function documented in wimlib.h  */
 209 WIMLIBAPI int
 210 wimlib_iterate_dir_tree(WIMStruct *wim, int image, const tchar *_path,
 211                         int flags,
 212                         wimlib_iterate_dir_tree_callback_t cb, void *user_ctx)
 213 {
 214         tchar *path;
 215         int ret;
 216
 217         if (flags & ~(WIMLIB_ITERATE_DIR_TREE_FLAG_RECURSIVE |
 218                       WIMLIB_ITERATE_DIR_TREE_FLAG_CHILDREN |
 219                       WIMLIB_ITERATE_DIR_TREE_FLAG_RESOURCES_NEEDED))
 220                 return WIMLIB_ERR_INVALID_PARAM;
 221
 222         path = canonicalize_wim_path(_path);
 223         if (path == NULL)
 224                 return WIMLIB_ERR_NOMEM;
 225         struct image_iterate_dir_tree_ctx ctx = {
 226                 .path = path,
 227                 .flags = flags,
 228                 .cb = cb,
 229                 .user_ctx = user_ctx,
 230         };
 231         wim->private = &ctx;
 232         ret = for_image(wim, image, image_do_iterate_dir_tree);
 233         FREE(path);
 234         return ret;
 235 }