X-Git-Url: https://wimlib.net/git/?p=wimlib;a=blobdiff_plain;f=src%2Fcapture_common.c;h=e544b7c126794f0e5013b684d9c3d83f455058de;hp=4905e35160b3d9d0e64b5806d1915b66dfa0252b;hb=5fce6aa92826ec69fdeeea13e28292fa1fbb43a8;hpb=3071e89c11d1be71cf45b694432e5908e0c4ded9

diff --git a/src/capture_common.c b/src/capture_common.c
index 4905e351..e544b7c1 100644
--- a/src/capture_common.c
+++ b/src/capture_common.c
@@ -23,17 +23,17 @@
 #  include "config.h"
 #endif
 
+#include <string.h>
+
+#include "wimlib/blob_table.h"
 #include "wimlib/capture.h"
 #include "wimlib/dentry.h"
 #include "wimlib/error.h"
-#include "wimlib/lookup_table.h"
 #include "wimlib/paths.h"
 #include "wimlib/progress.h"
 #include "wimlib/textfile.h"
 #include "wimlib/wildcard.h"
 
-#include <string.h>
-
 /*
  * Tally a file (or directory) that has been scanned for a capture operation,
  * and possibly call the progress function provided by the library user.
@@ -50,7 +50,7 @@
  *	via additional hard links, inode->i_nlink will be greater than 1.
  */
 int
-do_capture_progress(struct add_image_params *params, int status,
+do_capture_progress(struct capture_params *params, int status,
 		    const struct wim_inode *inode)
 {
 	switch (status) {
@@ -71,12 +71,12 @@ do_capture_progress(struct add_image_params *params, int status,
 
 		/* Successful scan, and visiting inode for the first time  */
 
-		/* Tally size of all data streams.  */
-		const struct wim_lookup_table_entry *lte;
-		for (unsigned i = 0; i <= inode->i_num_ads; i++) {
-			lte = inode_stream_lte_resolved(inode, i);
-			if (lte)
-				params->progress.scan.num_bytes_scanned += lte->size;
+		/* Tally size of all streams.  */
+		for (unsigned i = 0; i < inode->i_num_streams; i++) {
+			const struct blob_descriptor *blob =
+				stream_blob_resolved(&inode->i_streams[i]);
+			if (blob)
+				params->progress.scan.num_bytes_scanned += blob->size;
 		}
 
 		/* Tally the file itself.  */
@@ -203,8 +203,25 @@ read_capture_config(const tchar *config_file, const void *buf,
 	ret = do_load_text_file(config_file, buf, bufsize, &mem,
 				sections, ARRAY_LEN(sections),
 				LOAD_TEXT_FILE_REMOVE_QUOTES, mangle_pat);
-	if (ret)
+	if (ret) {
+		ERROR("Failed to load capture configuration file \"%"TS"\"",
+		      config_file);
+		switch (ret) {
+		case WIMLIB_ERR_INVALID_UTF8_STRING:
+		case WIMLIB_ERR_INVALID_UTF16_STRING:
+			ERROR("Note: the capture configuration file must be "
+			      "valid UTF-8 or UTF-16LE");
+			ret = WIMLIB_ERR_INVALID_CAPTURE_CONFIG;
+			break;
+		case WIMLIB_ERR_OPEN:
+		case WIMLIB_ERR_STAT:
+		case WIMLIB_ERR_NOMEM:
+		case WIMLIB_ERR_READ:
+			ret = WIMLIB_ERR_UNABLE_TO_READ_CAPTURE_CONFIG;
+			break;
+		}
 		return ret;
+	}
 
 	FREE(prepopulate_pats.strings);
 	FREE(compression_exclusion_pats.strings);
@@ -298,7 +315,7 @@ should_exclude_path(const tchar *path, size_t path_nchars,
  */
 int
 try_exclude(const tchar *full_path, size_t full_path_nchars,
-	    const struct add_image_params *params)
+	    const struct capture_params *params)
 {
 	int ret;