home *** CD-ROM | disk | FTP | other *** search
/ linuxmafia.com 2016 / linuxmafia.com.tar / linuxmafia.com / pub / palmos / pippy-0.6beta-src.tar.gz / pippy-0.6beta-src.tar / pippy-0.6beta-src / src / Lib / plat-win / socket.py
Text File  |  2000-12-21  |  4KB  |  169 lines
  1. "Socket wrapper for Windows, which does not support dup()."
  2.  
  3. # (And hence, fromfd() and makefile() are unimplemented in C....)
  4.  
  5. # XXX Living dangerously here -- close() is implemented by deleting a
  6. # reference.  Thus we rely on the real _socket module to close on
  7. # deallocation, and also hope that nobody keeps a reference to our _sock
  8. # member.
  9.  
  10.  
  11.  
  12. try:
  13.     from _socket import *
  14. except ImportError:
  15.     from socket import *
  16.  
  17. _realsocketcall = socket
  18.  
  19.  
  20. def socket(family, type, proto=0):
  21.     return _socketobject(_realsocketcall(family, type, proto))
  22.  
  23.  
  24. class _socketobject:
  25.  
  26.     def __init__(self, sock):
  27.         self._sock = sock
  28.  
  29.     def close(self):
  30.         self._sock = 0
  31.  
  32.     def __del__(self):
  33.         self.close()
  34.  
  35.     def accept(self):
  36.         sock, addr = self._sock.accept()
  37.         return _socketobject(sock), addr
  38.  
  39.     def dup(self):
  40.         return _socketobject(self._sock)
  41.  
  42.     def makefile(self, mode='r', bufsize=-1):
  43.         return _fileobject(self._sock, mode, bufsize)
  44.  
  45.     _s = "def %s(self, *args): return apply(self._sock.%s, args)\n\n"
  46.     for _m in ('bind', 'connect', 'connect_ex', 'fileno', 'listen',
  47.            'getpeername', 'getsockname',
  48.            'getsockopt', 'setsockopt',
  49.            'recv', 'recvfrom', 'send', 'sendto',
  50.            'setblocking',
  51.            'shutdown'):
  52.         exec _s % (_m, _m)
  53.  
  54.  
  55. class _fileobject:
  56.  
  57.     def __init__(self, sock, mode, bufsize):
  58.         self._sock = sock
  59.         self._mode = mode
  60.         if bufsize < 0:
  61.             bufsize = 512
  62.         self._rbufsize = max(1, bufsize)
  63.         self._wbufsize = bufsize
  64.         self._wbuf = self._rbuf = ""
  65.  
  66.     def close(self):
  67.         try:
  68.             if self._sock:
  69.                 self.flush()
  70.         finally:
  71.             self._sock = 0
  72.  
  73.     def __del__(self):
  74.         self.close()
  75.  
  76.     def flush(self):
  77.         if self._wbuf:
  78.             self._sock.send(self._wbuf)
  79.             self._wbuf = ""
  80.  
  81.     def fileno(self):
  82.         return self._sock.fileno()
  83.  
  84.     def write(self, data):
  85.         self._wbuf = self._wbuf + data
  86.         if self._wbufsize == 1:
  87.             if '\n' in data:
  88.                 self.flush()
  89.         else:
  90.             if len(self._wbuf) >= self._wbufsize:
  91.                 self.flush()
  92.  
  93.     def writelines(self, list):
  94.         filter(self._sock.send, list)
  95.         self.flush()
  96.  
  97.     def read(self, n=-1):
  98.         if n >= 0:
  99.             k = len(self._rbuf)
  100.             if n <= k:
  101.                 data = self._rbuf[:n]
  102.                 self._rbuf = self._rbuf[n:]
  103.                 return data
  104.             n = n - k
  105.             l = [self._rbuf]
  106.             self._rbuf = ""
  107.             while n > 0:
  108.                 new = self._sock.recv(max(n, self._rbufsize))
  109.                 if not new: break
  110.                 k = len(new)
  111.                 if k > n:
  112.                     l.append(new[:n])
  113.                     self._rbuf = new[n:]
  114.                     break
  115.                 l.append(new)
  116.                 n = n - k
  117.             return "".join(l)
  118.         k = max(512, self._rbufsize)
  119.         l = [self._rbuf]
  120.         self._rbuf = ""
  121.         while 1:
  122.             new = self._sock.recv(k)
  123.             if not new: break
  124.             l.append(new)
  125.             k = min(k*2, 1024**2)
  126.         return "".join(l)
  127.  
  128.     def readline(self, limit=-1):
  129.         data = ""
  130.         i = self._rbuf.find('\n')
  131.         while i < 0 and not (0 < limit <= len(self._rbuf)):
  132.             new = self._sock.recv(self._rbufsize)
  133.             if not new: break
  134.             i = new.find('\n')
  135.             if i >= 0: i = i + len(self._rbuf)
  136.             self._rbuf = self._rbuf + new
  137.         if i < 0: i = len(self._rbuf)
  138.         else: i = i+1
  139.         if 0 <= limit < len(self._rbuf): i = limit
  140.         data, self._rbuf = self._rbuf[:i], self._rbuf[i:]
  141.         return data
  142.  
  143.     def readlines(self):
  144.         list = []
  145.         while 1:
  146.             line = self.readline()
  147.             if not line: break
  148.             list.append(line)
  149.         return list
  150.  
  151.  
  152. # WSA error codes
  153. errorTab = {}
  154. errorTab[10004] = "The operation was interrupted."
  155. errorTab[10009] = "A bad file handle was passed."
  156. errorTab[10013] = "Permission denied."
  157. errorTab[10014] = "A fault occurred on the network??" # WSAEFAULT
  158. errorTab[10022] = "An invalid operation was attempted."
  159. errorTab[10035] = "The socket operation would block"
  160. errorTab[10036] = "A blocking operation is already in progress."
  161. errorTab[10048] = "The network address is in use."
  162. errorTab[10054] = "The connection has been reset."
  163. errorTab[10058] = "The network has been shut down."
  164. errorTab[10060] = "The operation timed out."
  165. errorTab[10061] = "Connection refused."
  166. errorTab[10063] = "The name is too long."
  167. errorTab[10064] = "The host is down."
  168. errorTab[10065] = "The host is unreachable."
  169.