| 1 | """codefragments.py -- wrapper to modify code fragments."""
|
|---|
| 2 |
|
|---|
| 3 | # (c) 1998, Just van Rossum, Letterror
|
|---|
| 4 |
|
|---|
| 5 | __version__ = "0.8b3"
|
|---|
| 6 | __author__ = "jvr"
|
|---|
| 7 |
|
|---|
| 8 | import Carbon.File
|
|---|
| 9 | import struct
|
|---|
| 10 | from Carbon import Res
|
|---|
| 11 | import os
|
|---|
| 12 | import sys
|
|---|
| 13 |
|
|---|
| 14 | DEBUG = 0
|
|---|
| 15 |
|
|---|
| 16 | error = "cfm.error"
|
|---|
| 17 |
|
|---|
| 18 | BUFSIZE = 0x80000
|
|---|
| 19 |
|
|---|
| 20 | def mergecfmfiles(srclist, dst, architecture = 'fat'):
|
|---|
| 21 | """Merge all files in srclist into a new file dst.
|
|---|
| 22 |
|
|---|
| 23 | If architecture is given, only code fragments of that type will be used:
|
|---|
| 24 | "pwpc" for PPC, "m68k" for cfm68k. This does not work for "classic"
|
|---|
| 25 | 68k code, since it does not use code fragments to begin with.
|
|---|
| 26 | If architecture is None, all fragments will be used, enabling FAT binaries.
|
|---|
| 27 | """
|
|---|
| 28 |
|
|---|
| 29 | srclist = list(srclist)
|
|---|
| 30 | for i in range(len(srclist)):
|
|---|
| 31 | srclist[i] = Carbon.File.pathname(srclist[i])
|
|---|
| 32 | dst = Carbon.File.pathname(dst)
|
|---|
| 33 |
|
|---|
| 34 | dstfile = open(dst, "wb")
|
|---|
| 35 | rf = Res.FSpOpenResFile(dst, 3)
|
|---|
| 36 | try:
|
|---|
| 37 | dstcfrg = CfrgResource()
|
|---|
| 38 | for src in srclist:
|
|---|
| 39 | srccfrg = CfrgResource(src)
|
|---|
| 40 | for frag in srccfrg.fragments:
|
|---|
| 41 | if frag.architecture == 'pwpc' and architecture == 'm68k':
|
|---|
| 42 | continue
|
|---|
| 43 | if frag.architecture == 'm68k' and architecture == 'pwpc':
|
|---|
| 44 | continue
|
|---|
| 45 | dstcfrg.append(frag)
|
|---|
| 46 |
|
|---|
| 47 | frag.copydata(dstfile)
|
|---|
| 48 |
|
|---|
| 49 | cfrgres = Res.Resource(dstcfrg.build())
|
|---|
| 50 | Res.UseResFile(rf)
|
|---|
| 51 | cfrgres.AddResource('cfrg', 0, "")
|
|---|
| 52 | finally:
|
|---|
| 53 | dstfile.close()
|
|---|
| 54 | rf = Res.CloseResFile(rf)
|
|---|
| 55 |
|
|---|
| 56 |
|
|---|
| 57 | class CfrgResource:
|
|---|
| 58 |
|
|---|
| 59 | def __init__(self, path = None):
|
|---|
| 60 | self.version = 1
|
|---|
| 61 | self.fragments = []
|
|---|
| 62 | self.path = path
|
|---|
| 63 | if path is not None and os.path.exists(path):
|
|---|
| 64 | currentresref = Res.CurResFile()
|
|---|
| 65 | resref = Res.FSpOpenResFile(path, 1)
|
|---|
| 66 | Res.UseResFile(resref)
|
|---|
| 67 | try:
|
|---|
| 68 | try:
|
|---|
| 69 | data = Res.Get1Resource('cfrg', 0).data
|
|---|
| 70 | except Res.Error:
|
|---|
| 71 | raise Res.Error, "no 'cfrg' resource found", sys.exc_traceback
|
|---|
| 72 | finally:
|
|---|
| 73 | Res.CloseResFile(resref)
|
|---|
| 74 | Res.UseResFile(currentresref)
|
|---|
| 75 | self.parse(data)
|
|---|
| 76 | if self.version <> 1:
|
|---|
| 77 | raise error, "unknown 'cfrg' resource format"
|
|---|
| 78 |
|
|---|
| 79 | def parse(self, data):
|
|---|
| 80 | (res1, res2, self.version,
|
|---|
| 81 | res3, res4, res5, res6,
|
|---|
| 82 | self.memberCount) = struct.unpack("8l", data[:32])
|
|---|
| 83 | data = data[32:]
|
|---|
| 84 | while data:
|
|---|
| 85 | frag = FragmentDescriptor(self.path, data)
|
|---|
| 86 | data = data[frag.memberSize:]
|
|---|
| 87 | self.fragments.append(frag)
|
|---|
| 88 |
|
|---|
| 89 | def build(self):
|
|---|
| 90 | self.memberCount = len(self.fragments)
|
|---|
| 91 | data = struct.pack("8l", 0, 0, self.version, 0, 0, 0, 0, self.memberCount)
|
|---|
| 92 | for frag in self.fragments:
|
|---|
| 93 | data = data + frag.build()
|
|---|
| 94 | return data
|
|---|
| 95 |
|
|---|
| 96 | def append(self, frag):
|
|---|
| 97 | self.fragments.append(frag)
|
|---|
| 98 |
|
|---|
| 99 |
|
|---|
| 100 | class FragmentDescriptor:
|
|---|
| 101 |
|
|---|
| 102 | def __init__(self, path, data = None):
|
|---|
| 103 | self.path = path
|
|---|
| 104 | if data is not None:
|
|---|
| 105 | self.parse(data)
|
|---|
| 106 |
|
|---|
| 107 | def parse(self, data):
|
|---|
| 108 | self.architecture = data[:4]
|
|---|
| 109 | ( self.updatelevel,
|
|---|
| 110 | self.currentVersion,
|
|---|
| 111 | self.oldDefVersion,
|
|---|
| 112 | self.stacksize,
|
|---|
| 113 | self.applibdir,
|
|---|
| 114 | self.fragtype,
|
|---|
| 115 | self.where,
|
|---|
| 116 | self.offset,
|
|---|
| 117 | self.length,
|
|---|
|
|---|
