X-Git-Url: https://wimlib.net/git/?p=wimlib;a=blobdiff_plain;f=include%2Fwimlib%2Fapply.h;h=2f3b1cb941ab3b61a8821357f2cf2091d50a4f72;hp=becef7acfd23cfbf9ed97e7801ecf2d5423e61d7;hb=4c532e2a9f1a40e1c1a6ed44a50025cfbfab6e4f;hpb=b6f6a919c8291da9cf2a9ea72ec7f8e47fbd79cf diff --git a/include/wimlib/apply.h b/include/wimlib/apply.h index becef7ac..2f3b1cb9 100644 --- a/include/wimlib/apply.h +++ b/include/wimlib/apply.h @@ -1,102 +1,116 @@ #ifndef _WIMLIB_APPLY_H #define _WIMLIB_APPLY_H -#include "wimlib.h" +#include "wimlib/file_io.h" +#include "wimlib/list.h" +#include "wimlib/progress.h" #include "wimlib/types.h" +#include "wimlib.h" -#ifdef WITH_NTFS_3G -struct _ntfs_volume; -#endif +/* These can be treated as counts (for required_features) or booleans (for + * supported_features). */ +struct wim_features { + unsigned long archive_files; + unsigned long hidden_files; + unsigned long system_files; + unsigned long compressed_files; + unsigned long encrypted_files; + unsigned long encrypted_directories; + unsigned long not_context_indexed_files; + unsigned long sparse_files; + unsigned long named_data_streams; + unsigned long hard_links; + unsigned long reparse_points; + unsigned long symlink_reparse_points; + unsigned long other_reparse_points; + unsigned long security_descriptors; + unsigned long short_names; + unsigned long unix_data; + unsigned long timestamps; + unsigned long case_sensitive_filenames; +}; -struct apply_args { - WIMStruct *w; +struct wim_lookup_table_entry; +struct read_stream_list_callbacks; - /* Directory to which we're extracting the WIM image or directory tree, - * in user-specified form (may be slightly altered) */ - const tchar *target; - unsigned target_nchars; +struct apply_ctx { + /* The WIMStruct from which files are being extracted from the currently + * selected image. */ + WIMStruct *wim; -#ifdef __WIN32__ - /* \\?\-prefixed full path to the above directory; needed to work around - * lack of default support for long paths on Windoze. */ - tchar *target_lowlevel_path; - unsigned target_lowlevel_path_nchars; -#endif + /* The target of the extraction, usually the path to a directory. */ + const tchar *target; - /* Absolute path to the above directory; on UNIX this is simply a path - * beginning with /, while on Windoze this will be a path beginning with - * a drive letter followed by a backslash, but not with \\?\. */ - tchar *target_realpath; - unsigned target_realpath_len; + /* Length of @target in tchars. */ + size_t target_nchars; - struct wim_dentry *extract_root; - unsigned long invalid_sequence; + /* Extraction flags (WIMLIB_EXTRACT_FLAG_*) */ int extract_flags; + + /* User-provided progress function, or NULL if not specified. */ + wimlib_progress_func_t progfunc; + void *progctx; + + /* Progress data buffer, with progress.extract initialized. */ union wimlib_progress_info progress; - wimlib_progress_func_t progress_func; - int (*apply_dentry)(struct wim_dentry *, void *); - union { - #ifdef WITH_NTFS_3G - struct { - /* NTFS apply only */ - struct _ntfs_volume *vol; - }; - #endif - #ifdef __WIN32__ - struct { - /* Normal apply only (Win32) */ - unsigned long num_set_sacl_priv_notheld; - unsigned long num_set_sd_access_denied; - unsigned vol_flags; - unsigned long num_hard_links_failed; - unsigned long num_soft_links_failed; - unsigned long num_long_paths; - bool have_vol_flags; - }; - #else - struct { - /* Normal apply only (UNIX) */ - unsigned long num_utime_warnings; - }; - #endif - }; + + /* Features required to extract the files (with counts) */ + struct wim_features required_features; + + /* Features supported by the extraction mode (with booleans) */ + struct wim_features supported_features; + + /* The members below should not be used outside of extract.c */ + u64 next_progress; + unsigned long invalid_sequence; + unsigned long num_streams_remaining; + struct list_head stream_list; + const struct read_stream_list_callbacks *saved_cbs; + struct wim_lookup_table_entry *cur_stream; + u64 cur_stream_offset; + struct filedes tmpfile_fd; + tchar *tmpfile_name; }; -#ifdef WITH_NTFS_3G -extern int -apply_dentry_ntfs(struct wim_dentry *dentry, void *arg); +/* Maximum number of UNIX file descriptors, NTFS attributes, or Windows file + * handles that can be opened simultaneously to extract a single-instance + * stream to multiple destinations. */ +#define MAX_OPEN_STREAMS 512 -extern int -apply_dentry_timestamps_ntfs(struct wim_dentry *dentry, void *arg); -#endif +static inline int +extract_progress(struct apply_ctx *ctx, enum wimlib_progress_msg msg) +{ + return call_progress(ctx->progfunc, msg, &ctx->progress, ctx->progctx); +} -#ifdef __WIN32__ -extern int -win32_do_apply_dentry(const tchar *output_path, - size_t output_path_nbytes, - struct wim_dentry *dentry, - struct apply_args *args); +/* Returns any of the aliases of an inode that are being extracted. */ +#define inode_first_extraction_dentry(inode) \ + list_first_entry(&(inode)->i_extraction_aliases, \ + struct wim_dentry, d_extraction_alias_node) extern int -win32_do_apply_dentry_timestamps(const tchar *output_path, - size_t output_path_nbytes, - struct wim_dentry *dentry, - struct apply_args *args); -#else /* __WIN32__ */ -extern int -unix_do_apply_dentry(const tchar *output_path, size_t output_path_nbytes, - struct wim_dentry *dentry, struct apply_args *args); -extern int -unix_do_apply_dentry_timestamps(const tchar *output_path, - size_t output_path_nbytes, - struct wim_dentry *dentry, - struct apply_args *args); -#endif /* !__WIN32__ */ +extract_stream_list(struct apply_ctx *ctx, + const struct read_stream_list_callbacks *cbs); -/* Internal use only */ -#define WIMLIB_EXTRACT_FLAG_MULTI_IMAGE 0x80000000 -#define WIMLIB_EXTRACT_FLAG_NO_STREAMS 0x40000000 -#define WIMLIB_EXTRACT_MASK_PUBLIC 0x3fffffff +struct apply_operations { + const char *name; + int (*get_supported_features)(const tchar *target, + struct wim_features *supported_features); + int (*extract)(struct list_head *dentry_list, struct apply_ctx *ctx); + + size_t context_size; + bool single_tree_only; +}; + +#ifdef __WIN32__ + extern const struct apply_operations win32_apply_ops; +#else + extern const struct apply_operations unix_apply_ops; +#endif + +#ifdef WITH_NTFS_3G + extern const struct apply_operations ntfs_3g_apply_ops; +#endif #endif /* _WIMLIB_APPLY_H */