← Back to branch summary

~vojtech-horky/helenos/numa

« back to all changes in this revision

Viewing changes to tools/mkhord.py

  • Committer: Martin Decky
  • Date: 2009-08-04 11:19:19 UTC
  • Revision ID: martin@uranus.dsrg.hide.ms.mff.cuni.cz-20090804111919-evyclddlr3v5lhmp
Initial import

Show diffs side-by-side

added added

removed removed

Lines of Context:
tools/mkhord.py
 
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
 
35
import xstruct
 
36
 
 
37
HEADER = """little:
 
38
        char tag[4]            /* 'HORD' */
 
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
"""
 
44
 
 
45
HORD_LSB = 1
 
46
 
 
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)
 
65
        if (align <= 0):
 
66
                print "<ALIGNMENT> must be positive"
 
67
                return
 
68
        
 
69
        fs_image = os.path.abspath(sys.argv[2])
 
70
        if (not os.path.isfile(fs_image)):
 
71
                print "<FS_IMAGE> must be a file"
 
72
                return
 
73
        
 
74
        inf = file(fs_image, "rb")
 
75
        outf = file(sys.argv[3], "wb")
 
76
        
 
77
        header = xstruct.create(HEADER)
 
78
        
 
79
        header_size = header.size()
 
80
        payload_size = os.path.getsize(fs_image)
 
81
        
 
82
        header_size_aligned = align_up(header_size, align)
 
83
        payload_size_aligned = align_up(payload_size, align)
 
84
        
 
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
 
90
        
 
91
        outf.write(header.pack())
 
92
        outf.write(xstruct.create("little: padding[%d]" % (header_size_aligned - header_size)).pack())
 
93
        
 
94
        outf.write(inf.read())
 
95
        
 
96
        padding = payload_size_aligned - payload_size
 
97
        if (padding > 0):
 
98
                outf.write(xstruct.create("little: padding[%d]" % padding).pack())
 
99
        
 
100
        inf.close()
 
101
        outf.close()
 
102
 
 
103
if __name__ == '__main__':
 
104
        main()