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
|
#
# Copyright OpenEmbedded Contributors
#
# SPDX-License-Identifier: MIT
#
# The empty wic plugin is used to create unformatted empty partitions for wic
# images.
# To use it you must pass "empty" as argument for the "--source" parameter in
# the wks file. For example:
# part foo --source empty --ondisk sda --size="1024" --align 1024
#
# The plugin supports writing zeros to the start of the
# partition. This is useful to overwrite old content like
# filesystem signatures which may be re-recognized otherwise.
# This feature can be enabled with
# '--sourceparams="[fill|size=<N>[S|s|K|k|M|G]][,][bs=<N>[S|s|K|k|M|G]]"'
# Conflicting or missing options throw errors.
import logging
import os
from wic import WicError
from wic.ksparser import sizetype
from wic.pluginbase import SourcePlugin
logger = logging.getLogger('wic')
class EmptyPartitionPlugin(SourcePlugin):
"""
Populate unformatted empty partition.
The following sourceparams are supported:
- fill
Fill the entire partition with zeros. Requires '--fixed-size' option
to be set.
- size=<N>[S|s|K|k|M|G]
Set the first N bytes of the partition to zero. Default unit is 'K'.
- bs=<N>[S|s|K|k|M|G]
Write at most N bytes at a time during source file creation.
Defaults to '1M'. Default unit is 'K'.
"""
name = 'empty'
@classmethod
def do_prepare_partition(cls, part, source_params, cr, cr_workdir,
oe_builddir, bootimg_dir, kernel_dir,
rootfs_dir, native_sysroot):
"""
Called to do the actual content population for a partition i.e. it
'prepares' the partition to be incorporated into the image.
"""
get_byte_count = sizetype('K', True)
size = 0
if 'fill' in source_params and 'size' in source_params:
raise WicError("Conflicting source parameters 'fill' and 'size' specified, exiting.")
# Set the size of the zeros to be written to the partition
if 'fill' in source_params:
if part.fixed_size == 0:
raise WicError("Source parameter 'fill' only works with the '--fixed-size' option, exiting.")
size = get_byte_count(part.fixed_size)
elif 'size' in source_params:
size = get_byte_count(source_params['size'])
if size == 0:
# Nothing to do, create empty partition
return
if 'bs' in source_params:
bs = get_byte_count(source_params['bs'])
else:
bs = get_byte_count('1M')
# Create a binary file of the requested size filled with zeros
source_file = os.path.join(cr_workdir, 'empty-plugin-zeros%s.bin' % part.lineno)
if not os.path.exists(os.path.dirname(source_file)):
os.makedirs(os.path.dirname(source_file))
quotient, remainder = divmod(size, bs)
with open(source_file, 'wb') as file:
for _ in range(quotient):
file.write(bytearray(bs))
file.write(bytearray(remainder))
part.size = (size + 1024 - 1) // 1024 # size in KB rounded up
part.source_file = source_file
|
