- goto err_free_imd;
-
- metadata_lte->resource_entry.flags = WIM_RESHDR_FLAG_METADATA;
- random_hash(metadata_lte->hash);
- lookup_table_insert(w->lookup_table, metadata_lte);
-
- new_imd = &imd[w->hdr.image_count];
-
- new_imd->root_dentry = root_dentry;
- new_imd->metadata_lte = metadata_lte;
- new_imd->security_data = sd;
- new_imd->modified = 1;
-
- FREE(w->image_metadata);
- w->image_metadata = imd;
- w->hdr.image_count++;
- return 0;
-err_free_imd:
- FREE(imd);
-err:
- return WIMLIB_ERR_NOMEM;
-
-}
-
-#if defined(__CYGWIN__) || defined(__WIN32__)
-static u64 FILETIME_to_u64(const FILETIME *ft)
-{
- return ((u64)ft->dwHighDateTime << 32) | (u64)ft->dwLowDateTime;
-}
-
-#ifdef ENABLE_ERROR_MESSAGES
-static void win32_error(DWORD err_code)
-{
- char *buffer;
- DWORD nchars;
- nchars = FormatMessageA(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER,
- NULL, err_code, 0,
- (char*)&buffer, 0, NULL);
- if (nchars == 0) {
- ERROR("Error printing error message! "
- "Computer will self-destruct in 3 seconds.");
- } else {
- ERROR("Win32 error: %s", buffer);
- LocalFree(buffer);
- }
-}
-#else
-#define win32_error(err_code)
-#endif
-
-static HANDLE win32_open_file(const wchar_t *path)
-{
- return CreateFileW(path,
- GENERIC_READ | READ_CONTROL,
- FILE_SHARE_READ,
- NULL, /* lpSecurityAttributes */
- OPEN_EXISTING,
- FILE_FLAG_BACKUP_SEMANTICS |
- FILE_FLAG_OPEN_REPARSE_POINT,
- NULL /* hTemplateFile */);
-}
-
-int win32_read_file(const char *filename,
- void *handle, u64 offset, size_t size, u8 *buf)
-{
- HANDLE h = handle;
- DWORD err;
- DWORD bytesRead;
- LARGE_INTEGER liOffset = {.QuadPart = offset};
-
- wimlib_assert(size <= 0xffffffff);
-
- if (SetFilePointerEx(h, liOffset, NULL, FILE_BEGIN))
- if (ReadFile(h, buf, size, &bytesRead, NULL) && bytesRead == size)
- return 0;
- err = GetLastError();
- ERROR("Error reading \"%s\"", filename);
- win32_error(err);
- return WIMLIB_ERR_READ;
-}
-
-void win32_close_handle(void *handle)
-{
- CloseHandle((HANDLE)handle);
-}
-
-void *win32_open_handle(const char *path_utf16)
-{
- return (void*)win32_open_file((const wchar_t*)path_utf16);
-}
-
-static int build_dentry_tree(struct wim_dentry **root_ret,
- const char *root_disk_path,
- struct wim_lookup_table *lookup_table,
- struct wim_security_data *sd,
- const struct capture_config *config,
- int add_image_flags,
- wimlib_progress_func_t progress_func,
- void *extra_arg);
-
-static int win32_get_short_name(struct wim_dentry *dentry,
- const wchar_t *path_utf16)
-{
- WIN32_FIND_DATAW dat;
- if (FindFirstFileW(path_utf16, &dat) &&
- dat.cAlternateFileName[0] != L'\0')
- {
- size_t short_name_len = wcslen(dat.cAlternateFileName) * 2;
- size_t n = short_name_len + sizeof(wchar_t);
- dentry->short_name = MALLOC(n);
- if (!dentry->short_name)
- return WIMLIB_ERR_NOMEM;
- memcpy(dentry->short_name, dat.cAlternateFileName, n);
- dentry->short_name_len = short_name_len;
- }
- return 0;
-}
-
-static int win32_get_security_descriptor(struct wim_dentry *dentry,
- struct sd_set *sd_set,
- const wchar_t *path_utf16,
- const char *path)
-{
- SECURITY_INFORMATION requestedInformation;
- DWORD lenNeeded = 0;
- BOOL status;
- DWORD err;
-
-#ifdef BACKUP_SECURITY_INFORMATION
- requestedInformation = BACKUP_SECURITY_INFORMATION;
-#else
- requestedInformation = DACL_SECURITY_INFORMATION |
- SACL_SECURITY_INFORMATION |
- OWNER_SECURITY_INFORMATION |
- GROUP_SECURITY_INFORMATION;
-#endif
- /* Request length of security descriptor */
- status = GetFileSecurityW(path_utf16, requestedInformation,
- NULL, 0, &lenNeeded);
- err = GetLastError();
-
- /* Error code appears to be ERROR_INSUFFICIENT_BUFFER but
- * GetFileSecurity is poorly documented... */
- if (err == ERROR_INSUFFICIENT_BUFFER || err == NO_ERROR) {
- DWORD len = lenNeeded;
- char buf[len];
- if (GetFileSecurityW(path_utf16, requestedInformation,
- buf, len, &lenNeeded))
- {
- int security_id = sd_set_add_sd(sd_set, buf, len);
- if (security_id < 0)
- return WIMLIB_ERR_NOMEM;
- else {
- dentry->d_inode->i_security_id = security_id;
- return 0;
- }
- } else {
- err = GetLastError();
- }
- }
- ERROR("Win32 API: Failed to read security descriptor of \"%s\"",
- path);
- win32_error(err);
- return WIMLIB_ERR_READ;
-}
-
-
-static int win32_recurse_directory(struct wim_dentry *root,
- const char *root_disk_path,
- struct wim_lookup_table *lookup_table,
- struct wim_security_data *sd,
- const struct capture_config *config,
- int add_image_flags,
- wimlib_progress_func_t progress_func,
- struct sd_set *sd_set,
- const wchar_t *path_utf16,
- size_t path_utf16_nchars)
-{
- WIN32_FIND_DATAW dat;
- HANDLE hFind;
- DWORD err;
- int ret;
-
- {
- wchar_t pattern_buf[path_utf16_nchars + 3];
- memcpy(pattern_buf, path_utf16,
- path_utf16_nchars * sizeof(wchar_t));
- pattern_buf[path_utf16_nchars] = L'/';
- pattern_buf[path_utf16_nchars + 1] = L'*';
- pattern_buf[path_utf16_nchars + 2] = L'\0';
- hFind = FindFirstFileW(pattern_buf, &dat);
- }
- if (hFind == INVALID_HANDLE_VALUE) {
- err = GetLastError();
- if (err == ERROR_FILE_NOT_FOUND) {
- return 0;
- } else {
- ERROR("Win32 API: Failed to read directory \"%s\"",
- root_disk_path);
- win32_error(err);
- return WIMLIB_ERR_READ;
- }
- }
- ret = 0;
- do {
- if (!(dat.cFileName[0] == L'.' &&
- (dat.cFileName[1] == L'\0' ||
- (dat.cFileName[1] == L'.' && dat.cFileName[2] == L'\0'))))
- {
- struct wim_dentry *child;
-
- char *utf8_name;
- size_t utf8_name_nbytes;
- ret = utf16_to_utf8((const char*)dat.cFileName,
- wcslen(dat.cFileName) * sizeof(wchar_t),
- &utf8_name,
- &utf8_name_nbytes);
- if (ret)
- goto out_find_close;
-
- char name[strlen(root_disk_path) + utf8_name_nbytes + 1];
- sprintf(name, "%s/%s", root_disk_path, utf8_name);
- FREE(utf8_name);
- ret = build_dentry_tree(&child, name, lookup_table,
- sd, config, add_image_flags,
- progress_func, sd_set);
- if (ret)
- goto out_find_close;
- if (child)
- dentry_add_child(root, child);
- }
- } while (FindNextFileW(hFind, &dat));
- err = GetLastError();
- if (err != ERROR_NO_MORE_FILES) {
- ERROR("Win32 API: Failed to read directory \"%s\"", root_disk_path);
- win32_error(err);
- if (ret == 0)
- ret = WIMLIB_ERR_READ;
- }
-out_find_close:
- FindClose(hFind);
- return ret;
-}
-
-static int win32_capture_reparse_point(const char *path,
- HANDLE hFile,
- struct wim_inode *inode,
- struct wim_lookup_table *lookup_table)
-{
- /* "Reparse point data, including the tag and optional GUID,
- * cannot exceed 16 kilobytes." - MSDN */
- char reparse_point_buf[16 * 1024];
- DWORD bytesReturned;
- const REPARSE_DATA_BUFFER *buf;
-
- if (!DeviceIoControl(hFile, FSCTL_GET_REPARSE_POINT,
- NULL, 0, reparse_point_buf,
- sizeof(reparse_point_buf), &bytesReturned, NULL))
- {
- ERROR("Win32 API: Failed to get reparse data of \"%s\"", path);
- return WIMLIB_ERR_READ;
- }
- buf = (const REPARSE_DATA_BUFFER*)reparse_point_buf;
- inode->i_reparse_tag = buf->ReparseTag;
- return inode_add_ads_with_data(inode, "", (const u8*)buf + 8,
- bytesReturned - 8, lookup_table);
-}
-
-static int win32_sha1sum(const wchar_t *path, u8 hash[SHA1_HASH_SIZE])
-{
- HANDLE hFile;
- SHA_CTX ctx;
- u8 buf[32768];
- DWORD bytesRead;
- int ret;
-
- hFile = win32_open_file(path);
- if (hFile == INVALID_HANDLE_VALUE)
- return WIMLIB_ERR_OPEN;
-
- sha1_init(&ctx);
- for (;;) {
- if (!ReadFile(hFile, buf, sizeof(buf), &bytesRead, NULL)) {
- ret = WIMLIB_ERR_READ;
- goto out_close_handle;
- }
- if (bytesRead == 0)
- break;
- sha1_update(&ctx, buf, bytesRead);
- }
- ret = 0;
- sha1_final(hash, &ctx);
-out_close_handle:
- CloseHandle(hFile);
- return ret;
-}
-
-static int win32_capture_stream(const char *path,
- const wchar_t *path_utf16,
- size_t path_utf16_nchars,
- struct wim_inode *inode,
- struct wim_lookup_table *lookup_table,
- WIN32_FIND_STREAM_DATA *dat)
-{
- struct wim_ads_entry *ads_entry;
- u8 hash[SHA1_HASH_SIZE];
- struct wim_lookup_table_entry *lte;
- int ret;
- wchar_t *p, *colon;
- bool is_named_stream;
- wchar_t *spath;
- size_t spath_nchars;
- DWORD err;
-
- p = dat->cStreamName;
- wimlib_assert(*p == L':');
- p += 1;
- colon = wcschr(p, L':');
- wimlib_assert(colon != NULL);
-
- if (wcscmp(colon + 1, L"$DATA")) {
- /* Not a DATA stream */
- ret = 0;