--- /dev/null
+"""cfmfile - Interface to code fragments on file"""
+import struct
+import Res
+import macfs
+import string
+
+Error = 'cfmfile.Error'
+
+READ = 1
+WRITE = 2
+smAllScripts = -3
+BUFSIZE = 0x100000
+
+class FragmentInfo:
+ """Information on a single fragment"""
+ def __init__(self):
+ self.arch = 'pwpc'
+ self.current_version = 0
+ self.oldest_version = 0
+ self.stacksize = 0
+ self.libdir = 0
+ self.fragtype = 1
+ self.location = 1
+ self.offset = 0
+ self.length = 0
+ self.res_0 = 0
+ self.res_1 = 0
+ self.name = ''
+ self.ifp = None
+
+ def load(self, data):
+ if len(data) < 43:
+ raise Error, 'Not enough data in cfrg resource'
+ self.arch = data[:4]
+ self.update, self.current_version, self.oldest_version, \
+ self.stacksize, self.libdir, self.fragtype, self.location, \
+ self.offset, self.length, self.res_0, self.res_1, length = \
+ struct.unpack("llllhbbllllh", data[4:42])
+ namelen = ord(data[42])
+ self.name = data[43:43+namelen]
+ if len(self.name) != namelen:
+ raise Error, 'Not enough data in cfrg resource'
+ return length
+
+ def save(self):
+ length = (43+len(self.name)+3) & ~3
+ data = self.arch + struct.pack("llllhbbllllh", self.update, \
+ self.current_version, self.oldest_version, self.stacksize, \
+ self.libdir, self.fragtype, self.location, self.offset, \
+ self.length, self.res_0, self.res_1, length)
+ data = data + chr(len(self.name)) + self.name
+ data = data + ('\0'*(length-len(data)))
+ return data
+
+ def copydata(self, ofp):
+ """Copy fragment data to a new file, updating self.offset"""
+ if self.location != 1:
+ raise Error, 'Can only copy kOnDiskFlat (data fork) fragments'
+ if not self.ifp:
+ raise Error, 'No source file for fragment'
+ # Find out real length (if zero)
+ if self.length == 0:
+ self.ifp.seek(0, 2)
+ self.length = self.ifp.tell()
+ # Position input file and record new offset from output file
+ self.ifp.seek(self.offset)
+ self.offset = ofp.tell()
+ l = self.length
+ while l:
+ if l > BUFSIZE:
+ ofp.write(self.ifp.read(BUFSIZE))
+ l = l - BUFSIZE
+ else:
+ ofp.write(self.ifp.read(l))
+ l = 0
+ self.ifp = ofp
+
+ def setfile(self, ifp):
+ self.ifp = ifp
+
+class FragmentResource:
+
+ def __init__(self, data):
+ self.load(data)
+
+ def load(self, data):
+ r0, r1, version, r3, r4, r5, r6, nfrag = struct.unpack("llllllll", data[:32])
+ if version != 1:
+ raise Error, 'Unsupported cfrg version number %d'%version
+ data = data[32:]
+ self.fragments = []
+ for i in range(nfrag):
+ f = FragmentInfo()
+ len = f.load(data)
+ data = data[len:]
+ self.fragments.append(f)
+ if data:
+ raise Error, 'Spurious data after fragment descriptions'
+
+ def save(self):
+ data = struct.pack("llllllll", 0, 0, 1, 0, 0, 0, 0, len(self.fragments))
+ for f in self.fragments:
+ data = data+f.save()
+ return data
+
+ def setfile(self, ifp):
+ for f in self.fragments:
+ f.setfile(ifp)
+
+ def copydata(self, ofp):
+ for f in self.fragments:
+ f.copydata(ofp)
+
+ def getfragments(self):
+ return self.fragments
+
+ def addfragments(self, fragments):
+ self.fragments = self.fragments + fragments
+
+class ResourceCollection:
+ def __init__(self, fhandle):
+ self.reslist = []
+ self.fhandle = fhandle
+ oldresfile = Res.CurResFile()
+ Res.UseResFile(fhandle)
+ Res.SetResLoad(0)
+ ntypes = Res.Count1Types()
+ for itype in range(1, 1+ntypes):
+ type = Res.Get1IndType(itype)
+ nresources = Res.Count1Resources(type)
+ for ires in range(1, 1+nresources):
+ res = Res.Get1IndResource(type, ires)
+ id, type, name = res.GetResInfo()
+ self.reslist.append((type, id))
+ Res.SetResLoad(1)
+ Res.UseResFile(oldresfile)
+
+ def contains(self, type, id):
+ return (type, id) in self.reslist
+
+ def getresource(self, type, id):
+ oldresfile = Res.CurResFile()
+ Res.UseResFile(self.fhandle)
+ Res.SetResLoad(1)
+ resource = Res.Get1Resource(type, id)
+ Res.UseResFile(oldresfile)
+ return resource
+
+ def saveresto(self, type, id, fhandle):
+ oldresfile = Res.CurResFile()
+ resource = self.getresource(type, id)
+ id, type, name = resource.GetResInfo()
+ resource.DetachResource()
+ Res.UseResFile(fhandle)
+ resource.AddResource(type, id, name)
+ Res.UseResFile(oldresfile)
+
+ def getreslist(self):
+ return self.reslist
+
+class CfmFile(ResourceCollection, FragmentResource):
+
+ def __init__(self, fsspec):
+ rfork = Res.FSpOpenResFile(fsspec, READ)
+ dfork = open(fsspec.as_pathname(), 'rb')
+ ResourceCollection.__init__(self, rfork)
+ cfrg_resource = self.getresource('cfrg', 0)
+ FragmentResource.__init__(self, cfrg_resource.data)
+ self.setfile(dfork)
+
+def mergecfmfiles(inputs, output):
+ # Convert inputs/outputs to fsspecs
+ for i in range(len(inputs)):
+ if type(inputs[i]) == type(''):
+ inputs[i] = macfs.FSSpec(inputs[i])
+ if type(output) == type(''):
+ output = macfs.FSSpec(output)
+
+ input_list = []
+ for i in inputs:
+ input_list.append(CfmFile(i))
+
+ # Create output file, if needed
+ creator, tp = inputs[0].GetCreatorType()
+ try:
+ Res.FSpCreateResFile(output, creator, tp, smAllScripts)
+ except Res.Error:
+ pass
+
+ # Copy fragments
+ dfork = open(output.as_pathname(), 'wb')
+ for i in input_list:
+ i.copydata(dfork)
+ dfork.close()
+
+ # Merge cfrg's
+ for i in input_list[1:]:
+ input_list[0].addfragments(i.getfragments())
+
+ old_res_file = Res.CurResFile()
+ rfork = Res.FSpOpenResFile(output, WRITE)
+ Res.UseResFile(rfork)
+
+ # Write cfrg
+ data = input_list[0].save()
+ cfrg_resource = Res.Resource(data)
+ cfrg_resource.AddResource('cfrg', 0, '')
+ resources_done = [('cfrg', 0)]
+
+ # Write other resources
+ for i in input_list:
+ todo = i.getreslist()
+ for tp, id in todo:
+ if (tp, id) in resources_done:
+ continue
+ i.saveresto(tp, id, rfork)
+ resources_done.append(tp, id)
+
+def main():
+ list = []
+ while 1:
+ fss, ok = macfs.PromptGetFile("Next input file:", "shlb", "APPL")
+ if not ok: break
+ list.append(fss)
+ if not list:
+ sys.exit(0)
+ output, ok = macfs.StandardPutFile("Output file:")
+ if not ok:
+ sys.exit(0)
+ mergecfmfiles(list, output)
+
+if __name__ == '__main__':
+ main()
+
+
projects / python / commitdiff