Lib/plat-win/socket.py

   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."