1 # -*- coding: utf-8 -*-
4 # Copyright P. Christeas <p_christ@hol.gr> 2008,2009
5 # Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>).
7 # This program is free software: you can redistribute it and/or modify
8 # it under the terms of the GNU Affero General Public License as
9 # published by the Free Software Foundation, either version 3 of the
10 # License, or (at your option) any later version.
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 # GNU Affero General Public License for more details.
17 # You should have received a copy of the GNU Affero General Public License
18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
20 ##############################################################################
22 """ This file contains instance of the net-rpc server
34 class TinySocketClientThread(threading.Thread, netsvc.OpenERPDispatcher):
35 def __init__(self, sock, threads):
36 threading.Thread.__init__(self)
38 # Only at the server side, use a big timeout: close the
39 # clients connection when they're idle for 20min.
40 self.sock.settimeout(1200)
41 self.threads = threads
47 getattr(socket, 'SHUT_RDWR', 2))
49 # That should garbage-collect and close it, too
56 ts = tiny_socket.mysocket(self.sock)
58 self.threads.remove(self)
65 self.threads.remove(self)
69 result = self.dispatch(msg[0], msg[1], msg[2:])
71 except netsvc.OpenERPDispatcherException, e:
73 new_e = Exception(tools.exception_to_unicode(e.exception)) # avoid problems of pickeling
74 ts.mysend(new_e, exception=True, traceback=e.traceback)
79 # this code should not be reachable, therefore we warn
80 netsvc.Logger().notifyChannel("net-rpc", netsvc.LOG_WARNING, "exception: %s" % str(e))
83 self.threads.remove(self)
91 class TinySocketServerThread(threading.Thread,netsvc.Server):
92 def __init__(self, interface, port, secure=False):
93 threading.Thread.__init__(self, name="Net-RPC socket")
94 netsvc.Server.__init__(self)
96 self.__interface = interface
97 self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
98 self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
99 self.socket.bind((self.__interface, self.__port))
100 self.socket.listen(5)
102 netsvc.Logger().notifyChannel("web-services", netsvc.LOG_INFO,
103 "starting NET-RPC service at %s port %d" % (interface or '0.0.0.0', port,))
110 (clientsocket, address) = self.socket.accept()
111 ct = TinySocketClientThread(clientsocket, self.threads)
113 self.threads.append(ct)
115 lt = len(self.threads)
116 if (lt > 10) and (lt % 10 == 0):
117 # Not many threads should be serving at the same time, so log
119 netsvc.Logger().notifyChannel("web-services", netsvc.LOG_DEBUG,
120 "Netrpc: %d threads" % len(self.threads))
124 logging.getLogger('web-services').warning("Netrpc: closing because of exception %s" % str(e))
130 for t in self.threads:
133 self.socket.shutdown(
134 getattr(socket, 'SHUT_RDWR', 2))
140 res = "Net-RPC: " + ( (self.running and "running") or "stopped")
142 for t in self.threads:
144 res += "\nNet-RPC #%d: %s " % (i, t.name)
157 if tools.config.get_misc('netrpcd','enable', True):
158 netrpcd = TinySocketServerThread(tools.config.get_misc('netrpcd','interface', ''), \
159 int(tools.config.get_misc('netrpcd','port', tools.config.get('netport', 8070))))