X-Git-Url: https://wimlib.net/git/?a=blobdiff_plain;f=src%2Fwrite.c;h=ea3f583c77442ec960ec4517b515caee91a3e8f5;hb=2a33c303e30fd740f740e21632fd06b9e414b0c7;hp=427da767a9977a85ad5ae95aeb34d12a4cd68d9d;hpb=9e56d04309e3e6a896319225288f0c86bd36d34e;p=wimlib

diff --git a/src/write.c b/src/write.c
index 427da767..ea3f583c 100644
--- a/src/write.c
+++ b/src/write.c
@@ -32,6 +32,13 @@
 #include <sys/file.h>
 #endif
 
+#ifdef __WIN32__
+#	include <windows.h>
+#	ifdef ERROR
+#		undef ERROR
+#	endif
+#endif
+
 #include "list.h"
 #include "wimlib_internal.h"
 #include "buffer_io.h"
@@ -61,6 +68,13 @@
 #include <stdlib.h>
 #endif
 
+#ifdef __WIN32__
+#	ifdef fsync
+#		undef fsync
+#	endif
+#	define fsync(fd) 0
+#endif
+
 static int fflush_and_ftruncate(FILE *fp, off_t size)
 {
 	int ret;
@@ -257,43 +271,55 @@ static int prepare_resource_for_read(struct wim_lookup_table_entry *lte
 					#endif
 		)
 {
-	if (lte->resource_location == RESOURCE_IN_FILE_ON_DISK
-	     && !lte->file_on_disk_fp)
-	{
-		wimlib_assert(lte->file_on_disk);
-		lte->file_on_disk_fp = fopen(lte->file_on_disk, "rb");
+	switch (lte->resource_location) {
+	case RESOURCE_IN_FILE_ON_DISK:
 		if (!lte->file_on_disk_fp) {
-			ERROR_WITH_ERRNO("Failed to open the file `%s' for "
-					 "reading", lte->file_on_disk);
-			return WIMLIB_ERR_OPEN;
+			lte->file_on_disk_fp = fopen(lte->file_on_disk, "rb");
+			if (!lte->file_on_disk_fp) {
+				ERROR_WITH_ERRNO("Failed to open the file "
+						 "`%s'", lte->file_on_disk);
+				return WIMLIB_ERR_OPEN;
+			}
 		}
-	}
+		break;
 #ifdef WITH_NTFS_3G
-	else if (lte->resource_location == RESOURCE_IN_NTFS_VOLUME
-		  && !lte->attr)
-	{
-		struct ntfs_location *loc = lte->ntfs_loc;
-		ntfs_inode *ni;
-		wimlib_assert(loc);
-		ni = ntfs_pathname_to_inode(*loc->ntfs_vol_p, NULL, loc->path_utf8);
-		if (!ni) {
-			ERROR_WITH_ERRNO("Failed to open inode `%s' in NTFS "
-					 "volume", loc->path_utf8);
-			return WIMLIB_ERR_NTFS_3G;
-		}
-		lte->attr = ntfs_attr_open(ni,
-					   loc->is_reparse_point ? AT_REPARSE_POINT : AT_DATA,
-					   (ntfschar*)loc->stream_name_utf16,
-					   loc->stream_name_utf16_num_chars);
+	case RESOURCE_IN_NTFS_VOLUME:
 		if (!lte->attr) {
-			ERROR_WITH_ERRNO("Failed to open attribute of `%s' in "
-					 "NTFS volume", loc->path_utf8);
-			ntfs_inode_close(ni);
-			return WIMLIB_ERR_NTFS_3G;
+			struct ntfs_location *loc = lte->ntfs_loc;
+			ntfs_inode *ni;
+			wimlib_assert(loc);
+			ni = ntfs_pathname_to_inode(*loc->ntfs_vol_p, NULL, loc->path_utf8);
+			if (!ni) {
+				ERROR_WITH_ERRNO("Failed to open inode `%s' in NTFS "
+						 "volume", loc->path_utf8);
+				return WIMLIB_ERR_NTFS_3G;
+			}
+			lte->attr = ntfs_attr_open(ni,
+						   loc->is_reparse_point ? AT_REPARSE_POINT : AT_DATA,
+						   (ntfschar*)loc->stream_name_utf16,
+						   loc->stream_name_utf16_num_chars);
+			if (!lte->attr) {
+				ERROR_WITH_ERRNO("Failed to open attribute of `%s' in "
+						 "NTFS volume", loc->path_utf8);
+				ntfs_inode_close(ni);
+				return WIMLIB_ERR_NTFS_3G;
+			}
+			*ni_ret = ni;
 		}
-		*ni_ret = ni;
-	}
+		break;
 #endif
+#if defined(__CYGWIN__) || defined(__WIN32__)
+	case RESOURCE_WIN32:
+		if (!lte->file_on_disk_fp) {
+			lte->file_on_disk_fp = win32_open_file_readonly(lte->file_on_disk);
+			if (!lte->file_on_disk_fp)
+				return WIMLIB_ERR_OPEN;
+		}
+		break;
+#endif
+	default:
+		break;
+	}
 	return 0;
 }
 
@@ -306,7 +332,8 @@ static void end_wim_resource_read(struct wim_lookup_table_entry *lte
 					)
 {
 	if (lte->resource_location == RESOURCE_IN_FILE_ON_DISK
-	    && lte->file_on_disk_fp) {
+	    && lte->file_on_disk_fp)
+	{
 		fclose(lte->file_on_disk_fp);
 		lte->file_on_disk_fp = NULL;
 	}
@@ -320,6 +347,14 @@ static void end_wim_resource_read(struct wim_lookup_table_entry *lte
 			ntfs_inode_close(ni);
 	}
 #endif
+#if defined(__CYGWIN__) || defined(__WIN32__)
+	else if (lte->resource_location == RESOURCE_WIN32
+		 && lte->file_on_disk_fp)
+	{
+		win32_close_file(lte->file_on_disk_fp);
+		lte->file_on_disk_fp = NULL;
+	}
+#endif
 }
 
 static int
@@ -835,10 +870,11 @@ static int main_writer_thread_proc(struct list_head *stream_list,
 		for (size_t j = 0; j < MAX_CHUNKS_PER_MSG; j++) {
 			msgs[i].compressed_chunks[j] = MALLOC(WIM_CHUNK_SIZE);
 
-			// The extra 8 bytes is because longest_match() in lz.c
-			// may read a little bit off the end of the uncompressed
-			// data.  It doesn't need to be initialized--- we really
-			// just need to avoid accessing an unmapped page.
+			// The extra 8 bytes is because longest_match() in
+			// lz77.c may read a little bit off the end of the
+			// uncompressed data.  It doesn't need to be
+			// initialized--- we really just need to avoid accessing
+			// an unmapped page.
 			msgs[i].uncompressed_chunks[j] = MALLOC(WIM_CHUNK_SIZE + 8);
 			if (msgs[i].compressed_chunks[j] == NULL ||
 			    msgs[i].uncompressed_chunks[j] == NULL)
@@ -1183,6 +1219,16 @@ out:
 	return ret;
 }
 
+static long get_default_num_threads()
+{
+#ifdef __WIN32__
+	SYSTEM_INFO sysinfo;
+	GetSystemInfo(&sysinfo);
+	return sysinfo.dwNumberOfProcessors;
+#else
+	return sysconf(_SC_NPROCESSORS_ONLN);
+#endif
+}
 
 static int write_stream_list_parallel(struct list_head *stream_list,
 				      FILE *out_fp,
@@ -1198,8 +1244,8 @@ static int write_stream_list_parallel(struct list_head *stream_list,
 	pthread_t *compressor_threads = NULL;
 
 	if (num_threads == 0) {
-		long nthreads = sysconf(_SC_NPROCESSORS_ONLN);
-		if (nthreads < 1) {
+		long nthreads = get_default_num_threads();
+		if (nthreads < 1 || nthreads > UINT_MAX) {
 			WARNING("Could not determine number of processors! Assuming 1");
 			goto out_serial;
 		} else {
@@ -1703,6 +1749,7 @@ WIMLIBAPI int wimlib_write(WIMStruct *w, const char *path,
 	ret = finish_write(w, image, write_flags, progress_func);
 out:
 	close_wim_writable(w);
+	DEBUG("wimlib_write(path=%s) = %d", path, ret);
 	return ret;
 }