Subversion Repositories HelenOS

Rev

Rev 3107 | Rev 3403 | Go to most recent revision | Show entire file | Ignore whitespace | Details | Blame | Last modification | View Log | RSS feed

Rev 3107 Rev 3149
Line 40... Line 40...
40 
 
 40 
 
41 
def usage(prname):
 41 
def usage(prname):
42 
    "Print usage syntax"
 42 
    "Print usage syntax"
43 
    print prname + " <ALIGNMENT> <PATH> <IMAGE>"
 43 
    print prname + " <ALIGNMENT> <PATH> <IMAGE>"
44 
 
 44 
 
- 
 
 45 
def recursion(root, outf):
- 
 
 46 
    "Recursive directory walk"
- 
 
 47 
   
- 
 
 48 
    payload_size = 0
- 
 
 49 
   
- 
 
 50 
    for name in os.listdir(root):
- 
 
 51 
        canon = os.path.join(root, name)
- 
 
 52 
       
- 
 
 53 
        if (os.path.isfile(canon)):
- 
 
 54 
            outf.write(struct.pack("<BL" + ("%d" % len(name)) + "s", 1, len(name), name))
- 
 
 55 
            payload_size += 5 + len(name)
- 
 
 56 
            size = os.path.getsize(canon)
- 
 
 57 
            rd = 0;
- 
 
 58 
            outf.write(struct.pack("<L", size))
- 
 
 59 
            payload_size += 4
- 
 
 60 
           
- 
 
 61 
            inf = file(canon, "r")
- 
 
 62 
            while (rd < size):
- 
 
 63 
                data = inf.read(4096);
- 
 
 64 
                outf.write(data)
- 
 
 65 
                payload_size += len(data)
- 
 
 66 
                rd += len(data)
- 
 
 67 
            inf.close()
- 
 
 68 
       
- 
 
 69 
        if (os.path.isdir(canon)):
- 
 
 70 
            outf.write(struct.pack("<BL" + ("%d" % len(name)) + "s", 2, len(name), name))
- 
 
 71 
            payload_size += 5 + len(name)
- 
 
 72 
            payload_size += recursion(canon, outf)
- 
 
 73 
            outf.write(struct.pack("<BL", 0, 0))
- 
 
 74 
            payload_size += 5
- 
 
 75 
   
- 
 
 76 
    return payload_size
- 
 
 77 
 
45 
def main():
 78 
def main():
46 
    if (len(sys.argv) < 4):
 79 
    if (len(sys.argv) < 4):
47 
        usage(sys.argv[0])
 80 
        usage(sys.argv[0])
48 
        return
 81 
        return
49 
   
 82 
   
Line 56... Line 89...
56 
    if (not os.path.isdir(path)):
 89 
    if (not os.path.isdir(path)):
57 
        print "<PATH> must be a directory"
 90 
        print "<PATH> must be a directory"
58 
        return
 91 
        return
59 
   
 92 
   
60 
    header_size = align_up(18, align)
 93 
    header_size = align_up(18, align)
61 
    payload_size = 0
 - 
 
62 
    outf = file(sys.argv[3], "w")
 94 
    outf = file(sys.argv[3], "w")
63 
    outf.write(struct.pack("<" + ("%d" % header_size) + "x"))
 95 
    outf.write(struct.pack("<" + ("%d" % header_size) + "x"))
64 
   
 96 
   
65 
    for root, dirs, files in os.walk(path):
 - 
 
66 
        relpath = root[len(path):len(root)]
 - 
 
67 
        for name in files:
 - 
 
68 
            canon = os.path.join(relpath, name)
 - 
 
69 
            outf.write(struct.pack("<BL" + ("%d" % len(canon)) + "s", 1, len(canon), canon))
 - 
 
70 
            payload_size += 5 + len(canon)
 - 
 
71 
           
 - 
 
72 
            fn = os.path.join(root, name)
 - 
 
73 
            size = os.path.getsize(fn)
 - 
 
74 
            rd = 0;
 - 
 
75 
            outf.write(struct.pack("<L", size))
 97 
    outf.write(struct.pack("<5s", "TMPFS"))
76 
            payload_size += 4
 98 
    payload_size = 5
77 
           
 99 
   
78 
            inf = file(fn, "r")
 - 
 
79 
            while (rd < size):
 - 
 
80 
                data = inf.read(4096);
 - 
 
81 
                outf.write(data)
 - 
 
82 
                payload_size += len(data)
 100 
    payload_size += recursion(path, outf)
83 
                rd += len(data)
 - 
 
84 
            inf.close()
 - 
 
85 
       
 101 
       
86 
        for name in dirs:
 - 
 
87 
            canon = os.path.join(relpath, name)
 - 
 
88 
            outf.write(struct.pack("<BL" + ("%d" % len(canon)) + "s", 2, len(canon), canon))
 102 
    outf.write(struct.pack("<BL", 0, 0))
89 
            payload_size += 5 + len(canon)
 103 
    payload_size += 5
90 
   
 104 
   
91 
    aligned_size = align_up(payload_size, align)
 105 
    aligned_size = align_up(payload_size, align)
92 
   
 106 
   
93 
    if (aligned_size - payload_size > 0):
 107 
    if (aligned_size - payload_size > 0):
94 
        outf.write(struct.pack("<" + ("%d" % (aligned_size - payload_size)) + "x"))
 108 
        outf.write(struct.pack("<" + ("%d" % (aligned_size - payload_size)) + "x"))
95 
   
 109 
       
96 
    outf.seek(0)
 110 
    outf.seek(0)
97 
    outf.write(struct.pack("<4sBBLQ", "HORD", 1, 1, header_size, aligned_size))
 111 
    outf.write(struct.pack("<4sBBLQ", "HORD", 1, 1, header_size, aligned_size))
98 
    outf.close()
 112 
    outf.close()
99 
 
 113 
 
100 
if __name__ == '__main__':
 114 
if __name__ == '__main__':