home *** CD-ROM | disk | FTP | other *** search
/ MacHack 2000 / MacHack 2000.toast / pc / The Hacks / MacHacksBug / Python 1.5.2c1 / Lib / binhex.pyc (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2000-06-23  |  19.6 KB  |  549 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 1.5)
  3.  
  4. '''binhex - Macintosh binhex compression/decompression
  5. easy interface:
  6. binhex(inputfilename, outputfilename)
  7. hexbin(inputfilename, outputfilename)
  8. '''
  9. import sys
  10. import os
  11. import struct
  12. import string
  13. import binascii
  14. Error = 'binhex.Error'
  15. (_DID_HEADER, _DID_DATA, _DID_RSRC) = range(3)
  16. REASONABLY_LARGE = 32768
  17. LINELEN = 64
  18. RUNCHAR = chr(144)
  19. if os.name == 'mac':
  20.     import macfs
  21.     import MacOS
  22.     
  23.     try:
  24.         openrf = MacOS.openrf
  25.     except AttributeError:
  26.         openrf = open
  27.  
  28.     
  29.     def FInfo():
  30.         return macfs.FInfo()
  31.  
  32.     
  33.     def getfileinfo(name):
  34.         finfo = macfs.FSSpec(name).GetFInfo()
  35.         (dir, file) = os.path.split(name)
  36.         fp = open(name, 'rb')
  37.         fp.seek(0, 2)
  38.         dlen = fp.tell()
  39.         fp = openrf(name, '*rb')
  40.         fp.seek(0, 2)
  41.         rlen = fp.tell()
  42.         return (file, finfo, dlen, rlen)
  43.  
  44.     
  45.     def openrsrc(name, *mode):
  46.         if not mode:
  47.             mode = '*rb'
  48.         else:
  49.             mode = '*' + mode[0]
  50.         return openrf(name, mode)
  51.  
  52. else:
  53.     
  54.     class FInfo:
  55.         
  56.         def __init__(self):
  57.             self.Type = '????'
  58.             self.Creator = '????'
  59.             self.Flags = 0
  60.  
  61.  
  62.     
  63.     def getfileinfo(name):
  64.         finfo = FInfo()
  65.         fp = open(name)
  66.         data = open(name).read(256)
  67.         for c in data:
  68.             pass
  69.         else:
  70.             finfo.Type = 'TEXT'
  71.         fp.seek(0, 2)
  72.         dsize = fp.tell()
  73.         fp.close()
  74.         (dir, file) = os.path.split(name)
  75.         file = string.replace(file, ':', '-', 1)
  76.         return (file, finfo, dsize, 0)
  77.  
  78.     
  79.     class openrsrc:
  80.         
  81.         def __init__(self, *args):
  82.             pass
  83.  
  84.         
  85.         def read(self, *args):
  86.             return ''
  87.  
  88.         
  89.         def write(self, *args):
  90.             pass
  91.  
  92.         
  93.         def close(self):
  94.             pass
  95.  
  96.  
  97.  
  98. class _Hqxcoderengine:
  99.     '''Write data to the coder in 3-byte chunks'''
  100.     
  101.     def __init__(self, ofp):
  102.         self.ofp = ofp
  103.         self.data = ''
  104.         self.hqxdata = ''
  105.         self.linelen = LINELEN - 1
  106.  
  107.     
  108.     def write(self, data):
  109.         self.data = self.data + data
  110.         datalen = len(self.data)
  111.         todo = (datalen / 3) * 3
  112.         data = self.data[:todo]
  113.         self.data = self.data[todo:]
  114.         if not data:
  115.             return None
  116.         
  117.         self.hqxdata = self.hqxdata + binascii.b2a_hqx(data)
  118.         self._flush(0)
  119.  
  120.     
  121.     def _flush(self, force):
  122.         first = 0
  123.         while first <= len(self.hqxdata) - self.linelen:
  124.             last = first + self.linelen
  125.             self.ofp.write(self.hqxdata[first:last] + '\n')
  126.             self.linelen = LINELEN
  127.             first = last
  128.         self.hqxdata = self.hqxdata[first:]
  129.         if force:
  130.             self.ofp.write(self.hqxdata + ':\n')
  131.         
  132.  
  133.     
  134.     def close(self):
  135.         if self.data:
  136.             self.hqxdata = self.hqxdata + binascii.b2a_hqx(self.data)
  137.         
  138.         self._flush(1)
  139.         self.ofp.close()
  140.         del self.ofp
  141.  
  142.  
  143.  
  144. class _Rlecoderengine:
  145.     '''Write data to the RLE-coder in suitably large chunks'''
  146.     
  147.     def __init__(self, ofp):
  148.         self.ofp = ofp
  149.         self.data = ''
  150.  
  151.     
  152.     def write(self, data):
  153.         self.data = self.data + data
  154.         if len(self.data) < REASONABLY_LARGE:
  155.             return None
  156.         
  157.         rledata = binascii.rlecode_hqx(self.data)
  158.         self.ofp.write(rledata)
  159.         self.data = ''
  160.  
  161.     
  162.     def close(self):
  163.         if self.data:
  164.             rledata = binascii.rlecode_hqx(self.data)
  165.             self.ofp.write(rledata)
  166.         
  167.         self.ofp.close()
  168.         del self.ofp
  169.  
  170.  
  171.  
  172. class BinHex:
  173.     
  174.     def __init__(self, .2, ofp):
  175.         (name, finfo, dlen, rlen) = .2
  176.         if type(ofp) == type(''):
  177.             ofname = ofp
  178.             ofp = open(ofname, 'w')
  179.             if os.name == 'mac':
  180.                 fss = macfs.FSSpec(ofname)
  181.                 fss.SetCreatorType('BnHq', 'TEXT')
  182.             
  183.         
  184.         ofp.write('(This file must be converted with BinHex 4.0)\n\n:')
  185.         hqxer = _Hqxcoderengine(ofp)
  186.         self.ofp = _Rlecoderengine(hqxer)
  187.         self.crc = 0
  188.         if finfo == None:
  189.             finfo = FInfo()
  190.         
  191.         self.dlen = dlen
  192.         self.rlen = rlen
  193.         self._writeinfo(name, finfo)
  194.         self.state = _DID_HEADER
  195.  
  196.     
  197.     def _writeinfo(self, name, finfo):
  198.         name = name
  199.         nl = len(name)
  200.         if nl > 63:
  201.             raise Error, 'Filename too long'
  202.         
  203.         d = chr(nl) + name + '\x00'
  204.         d2 = finfo.Type + finfo.Creator
  205.         d3 = struct.pack('>h', finfo.Flags)
  206.         d4 = struct.pack('>ii', self.dlen, self.rlen)
  207.         info = d + d2 + d3 + d4
  208.         self._write(info)
  209.         self._writecrc()
  210.  
  211.     
  212.     def _write(self, data):
  213.         self.crc = binascii.crc_hqx(data, self.crc)
  214.         self.ofp.write(data)
  215.  
  216.     
  217.     def _writecrc(self):
  218.         self.ofp.write(struct.pack('>h', self.crc))
  219.         self.crc = 0
  220.  
  221.     
  222.     def write(self, data):
  223.         if self.state != _DID_HEADER:
  224.             raise Error, 'Writing data at the wrong time'
  225.         
  226.         self.dlen = self.dlen - len(data)
  227.         self._write(data)
  228.  
  229.     
  230.     def close_data(self):
  231.         if self.dlen != 0:
  232.             raise Error, 'Incorrect data size, diff=' + `self.rlen`
  233.         
  234.         self._writecrc()
  235.         self.state = _DID_DATA
  236.  
  237.     
  238.     def write_rsrc(self, data):
  239.         if self.state < _DID_DATA:
  240.             self.close_data()
  241.         
  242.         if self.state != _DID_DATA:
  243.             raise Error, 'Writing resource data at the wrong time'
  244.         
  245.         self.rlen = self.rlen - len(data)
  246.         self._write(data)
  247.  
  248.     
  249.     def close(self):
  250.         if self.state < _DID_DATA:
  251.             self.close_data()
  252.         
  253.         if self.state != _DID_DATA:
  254.             raise Error, 'Close at the wrong time'
  255.         
  256.         if self.rlen != 0:
  257.             raise Error, 'Incorrect resource-datasize, diff=' + `self.rlen`
  258.         
  259.         self._writecrc()
  260.         self.ofp.close()
  261.         self.state = None
  262.         del self.ofp
  263.  
  264.  
  265.  
  266. def binhex(inp, out):
  267.     '''(infilename, outfilename) - Create binhex-encoded copy of a file'''
  268.     finfo = getfileinfo(inp)
  269.     ofp = BinHex(finfo, out)
  270.     ifp = open(inp, 'rb')
  271.     while 1:
  272.         d = ifp.read(128000)
  273.         if not d:
  274.             break
  275.         
  276.         ofp.write(d)
  277.     ofp.close_data()
  278.     ifp.close()
  279.     ifp = openrsrc(inp, 'rb')
  280.     while 1:
  281.         d = ifp.read(128000)
  282.         if not d:
  283.             break
  284.         
  285.         ofp.write_rsrc(d)
  286.     ofp.close()
  287.     ifp.close()
  288.  
  289.  
  290. class _Hqxdecoderengine:
  291.     '''Read data via the decoder in 4-byte chunks'''
  292.     
  293.     def __init__(self, ifp):
  294.         self.ifp = ifp
  295.         self.eof = 0
  296.  
  297.     
  298.     def read(self, totalwtd):
  299.         '''Read at least wtd bytes (or until EOF)'''
  300.         decdata = ''
  301.         wtd = totalwtd
  302.         while wtd > 0:
  303.             if self.eof:
  304.                 return decdata
  305.             
  306.             wtd = ((wtd + 2) / 3) * 4
  307.             data = self.ifp.read(wtd)
  308.             while 1:
  309.                 
  310.                 try:
  311.                     (decdatacur, self.eof) = binascii.a2b_hqx(data)
  312.                 except binascii.Incomplete:
  313.                     pass
  314.  
  315.                 newdata = self.ifp.read(1)
  316.                 if not newdata:
  317.                     raise Error, 'Premature EOF on binhex file'
  318.                 
  319.                 data = data + newdata
  320.             decdata = decdata + decdatacur
  321.             wtd = totalwtd - len(decdata)
  322.             if not decdata and not (self.eof):
  323.                 raise Error, 'Premature EOF on binhex file'
  324.             
  325.         return decdata
  326.  
  327.     
  328.     def close(self):
  329.         self.ifp.close()
  330.  
  331.  
  332.  
  333. class _Rledecoderengine:
  334.     '''Read data via the RLE-coder'''
  335.     
  336.     def __init__(self, ifp):
  337.         self.ifp = ifp
  338.         self.pre_buffer = ''
  339.         self.post_buffer = ''
  340.         self.eof = 0
  341.  
  342.     
  343.     def read(self, wtd):
  344.         if wtd > len(self.post_buffer):
  345.             self._fill(wtd - len(self.post_buffer))
  346.         
  347.         rv = self.post_buffer[:wtd]
  348.         self.post_buffer = self.post_buffer[wtd:]
  349.         return rv
  350.  
  351.     
  352.     def _fill(self, wtd):
  353.         self.pre_buffer = self.pre_buffer + self.ifp.read(wtd + 4)
  354.         if self.ifp.eof:
  355.             self.post_buffer = self.post_buffer + binascii.rledecode_hqx(self.pre_buffer)
  356.             self.pre_buffer = ''
  357.             return None
  358.         
  359.         mark = len(self.pre_buffer)
  360.         if self.pre_buffer[-3:] == RUNCHAR + '\x00' + RUNCHAR:
  361.             mark = mark - 3
  362.         elif self.pre_buffer[-1] == RUNCHAR:
  363.             mark = mark - 2
  364.         elif self.pre_buffer[-2:] == RUNCHAR + '\x00':
  365.             mark = mark - 2
  366.         elif self.pre_buffer[-2] == RUNCHAR:
  367.             pass
  368.         else:
  369.             mark = mark - 1
  370.         self.post_buffer = self.post_buffer + binascii.rledecode_hqx(self.pre_buffer[:mark])
  371.         self.pre_buffer = self.pre_buffer[mark:]
  372.  
  373.     
  374.     def close(self):
  375.         self.ifp.close()
  376.  
  377.  
  378.  
  379. class HexBin:
  380.     
  381.     def __init__(self, ifp):
  382.         if type(ifp) == type(''):
  383.             ifp = open(ifp)
  384.         
  385.         while 1:
  386.             ch = ifp.read(1)
  387.             if not ch:
  388.                 raise Error, 'No binhex data found'
  389.             
  390.             if ch == '\r':
  391.                 continue
  392.             
  393.             if ch == ':':
  394.                 break
  395.             
  396.             if ch != '\n':
  397.                 dummy = ifp.readline()
  398.             
  399.         hqxifp = _Hqxdecoderengine(ifp)
  400.         self.ifp = _Rledecoderengine(hqxifp)
  401.         self.crc = 0
  402.         self._readheader()
  403.  
  404.     
  405.     def _read(self, len):
  406.         data = self.ifp.read(len)
  407.         self.crc = binascii.crc_hqx(data, self.crc)
  408.         return data
  409.  
  410.     
  411.     def _checkcrc(self):
  412.         filecrc = struct.unpack('>h', self.ifp.read(2))[0] & 65535
  413.         self.crc = self.crc & 65535
  414.         if filecrc != self.crc:
  415.             raise Error, 'CRC error, computed %x, read %x' % (self.crc, filecrc)
  416.         
  417.         self.crc = 0
  418.  
  419.     
  420.     def _readheader(self):
  421.         len = self._read(1)
  422.         fname = self._read(ord(len))
  423.         rest = self._read(1 + 4 + 4 + 2 + 4 + 4)
  424.         self._checkcrc()
  425.         type = rest[1:5]
  426.         creator = rest[5:9]
  427.         flags = struct.unpack('>h', rest[9:11])[0]
  428.         self.dlen = struct.unpack('>l', rest[11:15])[0]
  429.         self.rlen = struct.unpack('>l', rest[15:19])[0]
  430.         self.FName = fname
  431.         self.FInfo = FInfo()
  432.         self.FInfo.Creator = creator
  433.         self.FInfo.Type = type
  434.         self.FInfo.Flags = flags
  435.         self.state = _DID_HEADER
  436.  
  437.     
  438.     def read(self, *n):
  439.         if self.state != _DID_HEADER:
  440.             raise Error, 'Read data at wrong time'
  441.         
  442.         if n:
  443.             n = n[0]
  444.             n = min(n, self.dlen)
  445.         else:
  446.             n = self.dlen
  447.         rv = ''
  448.         while len(rv) < n:
  449.             rv = rv + self._read(n - len(rv))
  450.         self.dlen = self.dlen - n
  451.         return rv
  452.  
  453.     
  454.     def close_data(self):
  455.         if self.state != _DID_HEADER:
  456.             raise Error, 'close_data at wrong time'
  457.         
  458.         if self.dlen:
  459.             dummy = self._read(self.dlen)
  460.         
  461.         self._checkcrc()
  462.         self.state = _DID_DATA
  463.  
  464.     
  465.     def read_rsrc(self, *n):
  466.         if self.state == _DID_HEADER:
  467.             self.close_data()
  468.         
  469.         if self.state != _DID_DATA:
  470.             raise Error, 'Read resource data at wrong time'
  471.         
  472.         if n:
  473.             n = n[0]
  474.             n = min(n, self.rlen)
  475.         else:
  476.             n = self.rlen
  477.         self.rlen = self.rlen - n
  478.         return self._read(n)
  479.  
  480.     
  481.     def close(self):
  482.         if self.rlen:
  483.             dummy = self.read_rsrc(self.rlen)
  484.         
  485.         self._checkcrc()
  486.         self.state = _DID_RSRC
  487.         self.ifp.close()
  488.  
  489.  
  490.  
  491. def hexbin(inp, out):
  492.     '''(infilename, outfilename) - Decode binhexed file'''
  493.     ifp = HexBin(inp)
  494.     finfo = ifp.FInfo
  495.     if not out:
  496.         out = ifp.FName
  497.     
  498.     if os.name == 'mac':
  499.         ofss = macfs.FSSpec(out)
  500.         out = ofss.as_pathname()
  501.     
  502.     ofp = open(out, 'wb')
  503.     while 1:
  504.         d = ifp.read(128000)
  505.         if not d:
  506.             break
  507.         
  508.         ofp.write(d)
  509.     ofp.close()
  510.     ifp.close_data()
  511.     d = ifp.read_rsrc(128000)
  512.     if d:
  513.         ofp = openrsrc(out, 'wb')
  514.         ofp.write(d)
  515.         while 1:
  516.             d = ifp.read_rsrc(128000)
  517.             if not d:
  518.                 break
  519.             
  520.             ofp.write(d)
  521.         ofp.close()
  522.     
  523.     if os.name == 'mac':
  524.         nfinfo = ofss.GetFInfo()
  525.         nfinfo.Creator = finfo.Creator
  526.         nfinfo.Type = finfo.Type
  527.         nfinfo.Flags = finfo.Flags
  528.         ofss.SetFInfo(nfinfo)
  529.     
  530.     ifp.close()
  531.  
  532.  
  533. def _test():
  534.     if os.name == 'mac':
  535.         (fss, ok) = macfs.PromptGetFile('File to convert:')
  536.         if not ok:
  537.             sys.exit(0)
  538.         
  539.         fname = fss.as_pathname()
  540.     else:
  541.         fname = sys.argv[1]
  542.     binhex(fname, fname + '.hqx')
  543.     hexbin(fname + '.hqx', fname + '.viahqx')
  544.     sys.exit(1)
  545.  
  546. if __name__ == '__main__':
  547.     _test()
  548.  
  549.