Subversion Repositories HelenOS

Rev

Rev 3262 | Details | Compare with Previous | Last modification | View Log | RSS feed

Rev Author Line No. Line
3244 jermar 1
#!/usr/bin/env python
2
#
3
# Copyright (c) 2008 Martin Decky
4
# All rights reserved.
5
#
6
# Redistribution and use in source and binary forms, with or without
7
# modification, are permitted provided that the following conditions
8
# are met:
9
#
10
# - Redistributions of source code must retain the above copyright
11
#   notice, this list of conditions and the following disclaimer.
12
# - Redistributions in binary form must reproduce the above copyright
13
#   notice, this list of conditions and the following disclaimer in the
14
#   documentation and/or other materials provided with the distribution.
15
# - The name of the author may not be used to endorse or promote products
16
#   derived from this software without specific prior written permission.
17
#
18
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
19
# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
20
# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
21
# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
22
# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
23
# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
27
# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28
#
29
"""
30
HORD encapsulator
31
"""
32
 
33
import sys
34
import os
3260 decky 35
import xstruct
3244 jermar 36
 
3262 decky 37
HEADER = """little:
3263 decky 38
    char tag[4]            /* 'HORD' */
3262 decky 39
    uint8_t version        /* version */
40
    uint8_t encoding       /* encoding */
41
    uint32_t header_size   /* header size */
42
    uint64_t payload_size  /* payload size */
43
"""
3260 decky 44
 
45
HORD_LSB = 1
46
 
3244 jermar 47
def align_up(size, alignment):
48
    "Align upwards to a given alignment"
49
    return (((size) + ((alignment) - 1)) & ~((alignment) - 1))
50
 
51
def usage(prname):
52
    "Print usage syntax"
53
    print prname + " <ALIGNMENT> <FS_IMAGE> <HORD_IMAGE>"
54
 
55
def main():
56
    if (len(sys.argv) < 4):
57
        usage(sys.argv[0])
58
        return
59
 
60
    if (not sys.argv[1].isdigit()):
61
        print "<ALIGNMENT> must be a number"
62
        return
63
 
64
    align = int(sys.argv[1], 0)
3260 decky 65
    if (align <= 0):
66
        print "<ALIGNMENT> must be positive"
67
        return
68
 
3244 jermar 69
    fs_image = os.path.abspath(sys.argv[2])
70
    if (not os.path.isfile(fs_image)):
3245 decky 71
        print "<FS_IMAGE> must be a file"
3244 jermar 72
        return
73
 
74
    inf = file(fs_image, "rb")
75
    outf = file(sys.argv[3], "wb")
3259 decky 76
 
3262 decky 77
    header = xstruct.create(HEADER)
78
 
79
    header_size = header.size()
3260 decky 80
    payload_size = os.path.getsize(fs_image)
3244 jermar 81
 
3260 decky 82
    header_size_aligned = align_up(header_size, align)
83
    payload_size_aligned = align_up(payload_size, align)
3244 jermar 84
 
3262 decky 85
    header.tag = "HORD"
86
    header.version = 1
87
    header.encoding = HORD_LSB
88
    header.header_size = header_size_aligned
89
    header.payload_size = payload_size_aligned
3260 decky 90
 
3262 decky 91
    outf.write(header.pack())
92
    outf.write(xstruct.create("little: padding[%d]" % (header_size_aligned - header_size)).pack())
93
 
3244 jermar 94
    outf.write(inf.read())
95
 
3260 decky 96
    padding = payload_size_aligned - payload_size
3244 jermar 97
    if (padding > 0):
3262 decky 98
        outf.write(xstruct.create("little: padding[%d]" % padding).pack())
3244 jermar 99
 
100
    inf.close()
101
    outf.close()
102
 
103
if __name__ == '__main__':
104
    main()