1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
/*
* capturewim.c - A program to capture a directory tree into a WIM file.
*
* Copyright 2022 Eric Biggers
*
* Permission is hereby granted, free of charge, to any person
* obtaining a copy of this software and associated documentation
* files (the "Software"), to deal in the Software without
* restriction, including without limitation the rights to use,
* copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following
* conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
* OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
* HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
* WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
* OTHER DEALINGS IN THE SOFTWARE.
*/
#include <wimlib.h>
#include <stdio.h>
/*
* Windows compatibility defines for string encoding. Applications using wimlib
* that need to run on both UNIX and Windows will need to do something similar
* to this, whereas applications that only need to run on one or the other can
* just use their platform's convention directly.
*/
#ifdef _WIN32
# define main wmain
typedef wchar_t tchar;
# define TS "ls"
#else
typedef char tchar;
# define TS "s"
#endif
#define TO_PERCENT(numerator, denominator) \
((float)(((denominator) == 0) ? 0 : ((numerator) * 100 / (float)(denominator))))
static enum wimlib_progress_status
write_progress(enum wimlib_progress_msg msg,
union wimlib_progress_info *info, void *progctx)
{
switch (msg) {
case WIMLIB_PROGRESS_MSG_WRITE_STREAMS:
printf("Writing WIM: %.2f%% complete\n",
TO_PERCENT(info->write_streams.completed_bytes,
info->write_streams.total_bytes));
break;
default:
break;
}
return WIMLIB_PROGRESS_STATUS_CONTINUE;
}
int main(int argc, tchar **argv)
{
int ret;
WIMStruct *wim = NULL;
const tchar *srcdir;
const tchar *wimpath;
/* Check for the correct number of arguments. */
if (argc != 3) {
fprintf(stderr, "Usage: capturewim DIR WIM\n");
return 2;
}
srcdir = argv[1];
wimpath = argv[2];
/* Create a WIMStruct for a WIM. */
ret = wimlib_create_new_wim(WIMLIB_COMPRESSION_TYPE_LZX, &wim);
if (ret != 0) /* Always should check the error codes. */
goto out;
/* Register our progress function. */
wimlib_register_progress_function(wim, write_progress, NULL);
/* Add the directory tree to the WIMStruct as an image. */
ret = wimlib_add_image(wim, /* WIMStruct to which to add the image */
srcdir, /* Directory from which to add the image */
NULL, /* Name to give the image (NULL means none) */
NULL, /* Capture configuration structure (NULL means none) */
0); /* WIMLIB_ADD_FLAG_* flags (0 means all defaults) */
if (ret != 0)
goto out;
/* Write the WIM file. */
ret = wimlib_write(wim, /* WIMStruct from which to write a WIM */
wimpath, /* Path to write the WIM to */
WIMLIB_ALL_IMAGES, /* Image(s) in the WIM to write */
0, /* WIMLIB_WRITE_FLAG_* flags (0 means all defaults) */
0); /* Number of compressor threads (0 means default) */
out:
/* Free the WIMStruct. Has no effect if the pointer to it is NULL. */
wimlib_free(wim);
/* Check for error status. */
if (ret != 0) {
fprintf(stderr, "wimlib error %d: %" TS"\n",
ret, wimlib_get_error_string((enum wimlib_error_code)ret));
}
/* Free global memory (optional). */
wimlib_global_cleanup();
return ret;
}
|
