Python code coverage for Lib/plat-mac/cfmfile.py

#	count	content
1	n/a	"""codefragments.py -- wrapper to modify code fragments."""
2	n/a
3	n/a	# (c) 1998, Just van Rossum, Letterror
4	n/a
5	n/a	__version__ = "0.8b3"
6	n/a	__author__ = "jvr"
7	n/a
8	n/a	import warnings
9	n/a	warnings.warnpy3k("the cfmfile module is deprecated and is removed in 3,0",
10	n/a	stacklevel=2)
11	n/a
12	n/a	import Carbon.File
13	n/a	import struct
14	n/a	from Carbon import Res
15	n/a	import os
16	n/a	import sys
17	n/a
18	n/a	DEBUG = 0
19	n/a
20	n/a	error = "cfm.error"
21	n/a
22	n/a	BUFSIZE = 0x80000
23	n/a
24	n/a	def mergecfmfiles(srclist, dst, architecture = 'fat'):
25	n/a	"""Merge all files in srclist into a new file dst.
26	n/a
27	n/a	If architecture is given, only code fragments of that type will be used:
28	n/a	"pwpc" for PPC, "m68k" for cfm68k. This does not work for "classic"
29	n/a	68k code, since it does not use code fragments to begin with.
30	n/a	If architecture is None, all fragments will be used, enabling FAT binaries.
31	n/a	"""
32	n/a
33	n/a	srclist = list(srclist)
34	n/a	for i in range(len(srclist)):
35	n/a	srclist[i] = Carbon.File.pathname(srclist[i])
36	n/a	dst = Carbon.File.pathname(dst)
37	n/a
38	n/a	dstfile = open(dst, "wb")
39	n/a	rf = Res.FSpOpenResFile(dst, 3)
40	n/a	try:
41	n/a	dstcfrg = CfrgResource()
42	n/a	for src in srclist:
43	n/a	srccfrg = CfrgResource(src)
44	n/a	for frag in srccfrg.fragments:
45	n/a	if frag.architecture == 'pwpc' and architecture == 'm68k':
46	n/a	continue
47	n/a	if frag.architecture == 'm68k' and architecture == 'pwpc':
48	n/a	continue
49	n/a	dstcfrg.append(frag)
50	n/a
51	n/a	frag.copydata(dstfile)
52	n/a
53	n/a	cfrgres = Res.Resource(dstcfrg.build())
54	n/a	Res.UseResFile(rf)
55	n/a	cfrgres.AddResource('cfrg', 0, "")
56	n/a	finally:
57	n/a	dstfile.close()
58	n/a	rf = Res.CloseResFile(rf)
59	n/a
60	n/a
61	n/a	class CfrgResource:
62	n/a
63	n/a	def __init__(self, path = None):
64	n/a	self.version = 1
65	n/a	self.fragments = []
66	n/a	self.path = path
67	n/a	if path is not None and os.path.exists(path):
68	n/a	currentresref = Res.CurResFile()
69	n/a	resref = Res.FSpOpenResFile(path, 1)
70	n/a	Res.UseResFile(resref)
71	n/a	try:
72	n/a	try:
73	n/a	data = Res.Get1Resource('cfrg', 0).data
74	n/a	except Res.Error:
75	n/a	raise Res.Error, "no 'cfrg' resource found", sys.exc_traceback
76	n/a	finally:
77	n/a	Res.CloseResFile(resref)
78	n/a	Res.UseResFile(currentresref)
79	n/a	self.parse(data)
80	n/a	if self.version != 1:
81	n/a	raise error, "unknown 'cfrg' resource format"
82	n/a
83	n/a	def parse(self, data):
84	n/a	(res1, res2, self.version,
85	n/a	res3, res4, res5, res6,
86	n/a	self.memberCount) = struct.unpack("8l", data[:32])
87	n/a	data = data[32:]
88	n/a	while data:
89	n/a	frag = FragmentDescriptor(self.path, data)
90	n/a	data = data[frag.memberSize:]
91	n/a	self.fragments.append(frag)
92	n/a
93	n/a	def build(self):
94	n/a	self.memberCount = len(self.fragments)
95	n/a	data = struct.pack("8l", 0, 0, self.version, 0, 0, 0, 0, self.memberCount)
96	n/a	for frag in self.fragments:
97	n/a	data = data + frag.build()
98	n/a	return data
99	n/a
100	n/a	def append(self, frag):
101	n/a	self.fragments.append(frag)
102	n/a
103	n/a
104	n/a	class FragmentDescriptor:
105	n/a
106	n/a	def __init__(self, path, data = None):
107	n/a	self.path = path
108	n/a	if data is not None:
109	n/a	self.parse(data)
110	n/a
111	n/a	def parse(self, data):
112	n/a	self.architecture = data[:4]
113	n/a	( self.updatelevel,
114	n/a	self.currentVersion,
115	n/a	self.oldDefVersion,
116	n/a	self.stacksize,
117	n/a	self.applibdir,
118	n/a	self.fragtype,
119	n/a	self.where,
120	n/a	self.offset,
121	n/a	self.length,
122	n/a	self.res1, self.res2,
123	n/a	self.memberSize,) = struct.unpack("4lhBB4lh", data[4:42])
124	n/a	pname = data[42:self.memberSize]
125	n/a	self.name = pname[1:1+ord(pname[0])]
126	n/a
127	n/a	def build(self):
128	n/a	data = self.architecture
129	n/a	data = data + struct.pack("4lhBB4l",
130	n/a	self.updatelevel,
131	n/a	self.currentVersion,
132	n/a	self.oldDefVersion,
133	n/a	self.stacksize,
134	n/a	self.applibdir,
135	n/a	self.fragtype,
136	n/a	self.where,
137	n/a	self.offset,
138	n/a	self.length,
139	n/a	self.res1, self.res2)
140	n/a	self.memberSize = len(data) + 2 + 1 + len(self.name)
141	n/a	# pad to 4 byte boundaries
142	n/a	if self.memberSize % 4:
143	n/a	self.memberSize = self.memberSize + 4 - (self.memberSize % 4)
144	n/a	data = data + struct.pack("hb", self.memberSize, len(self.name))
145	n/a	data = data + self.name
146	n/a	data = data + '\000' * (self.memberSize - len(data))
147	n/a	return data
148	n/a
149	n/a	def getfragment(self):
150	n/a	if self.where != 1:
151	n/a	raise error, "can't read fragment, unsupported location"
152	n/a	f = open(self.path, "rb")
153	n/a	f.seek(self.offset)
154	n/a	if self.length:
155	n/a	frag = f.read(self.length)
156	n/a	else:
157	n/a	frag = f.read()
158	n/a	f.close()
159	n/a	return frag
160	n/a
161	n/a	def copydata(self, outfile):
162	n/a	if self.where != 1:
163	n/a	raise error, "can't read fragment, unsupported location"
164	n/a	infile = open(self.path, "rb")
165	n/a	if self.length == 0:
166	n/a	infile.seek(0, 2)
167	n/a	self.length = infile.tell()
168	n/a
169	n/a	# Position input file and record new offset from output file
170	n/a	infile.seek(self.offset)
171	n/a
172	n/a	# pad to 16 byte boundaries
173	n/a	offset = outfile.tell()
174	n/a	if offset % 16:
175	n/a	offset = offset + 16 - (offset % 16)
176	n/a	outfile.seek(offset)
177	n/a	self.offset = offset
178	n/a
179	n/a	l = self.length
180	n/a	while l:
181	n/a	if l > BUFSIZE:
182	n/a	outfile.write(infile.read(BUFSIZE))
183	n/a	l = l - BUFSIZE
184	n/a	else:
185	n/a	outfile.write(infile.read(l))
186	n/a	l = 0
187	n/a	infile.close()