1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
|
# This file is Copyright 2004 Darryl A. Dixon <esrever_otua@pythonhacker.is-a-geek.net>
# and is part of 'NTLM Authorization Proxy Server',
# Copyright 2001 Dmitry A. Rozmanov <dima@xenon.spb.ru>
#
# NTLM APS is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# NTLM APS is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with the sofware; see the file COPYING. If not, write to the
# Free Software Foundation, Inc.,
# 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
#
import signal, httplib, time, socket, thread, os
#--------------------------------------------------------------
class monitorThread:
#--------------------------------------------------------------
def __init__(self, config, die_sig=signal.SIGINT):
self.alive = 1
self.config = config
self.die_sig = die_sig
self.threadsToKill = []
self.timeoutSeconds = self.config['GENERAL']['PARENT_PROXY_TIMEOUT']
#--------------------------------------------------------------
def run(self):
while self.alive:
self.alarmThread = timerThread(self.timeoutSeconds, self.alive, self.die_sig)
thread.start_new_thread(self.alarmThread.run, ())
# We poll the current proxy for responsiveness...
# TODO: add logger entries for all these exceptions
self.httpTest()
self.alarmThread.alive = 0
time.sleep(self.timeoutSeconds+1)
# Maximum timeout before hitting bottom of this loop is therefore
# at most self.timeoutSeconds*2+1 and at least self.timeoutSeconds+1.
# Hopefully this is a reasonable tradeoff between noticeable
# service outage for user and creaming the proxy with stacks of
# our spurious requests... :)
def die(self):
try:
self.alarmThread.alive = 0
except AttributeError:
pass # self.alarmThread is already dead
self.alive = 0
die_func = signal.getsignal(self.die_sig)
die_func(self.die_sig)
def httpTest(self):
"""
Although the httplib.HTTP class is retained in Python 2.x for backwards
compatability with Python 1.5 it is not recommended for new code, and as
most users will now be using the Python 2.x stream I have abstracted away
two code paths in here. The maintenance will be slightly higher, but in
the long term it will make it easier to fork off a Python 1.5.2 maintenance
version of ntlmaps and just keep the pure Python 2.x optimised version.
"""
if 'HTTPConnection' in dir(httplib):
# Python 2.x
try:
conn = httplib.HTTPConnection(self.config['GENERAL']['PARENT_PROXY'], self.config['GENERAL']['PARENT_PROXY_PORT'])
try:
conn.request("GET", "/")
try:
data = conn.getresponse()
try:
if not data.read(): # Got a b0rked response?
self.die()
except AssertionError: # Yup, got a wacky response
self.die()
conn.close()
except (AttributeError, httplib.BadStatusLine): # Didn't somehow connect?
self.die()
except socket.error: # Service not running/listening on specified port?
self.die()
except socket.gaierror: # Name resolution error for this proxy?
self.die()
else:
# Python 1.5.2
try:
# This call fails far more regularly than it should. Best to move to
# using Python 2.x
conn = httplib.HTTP(self.config['GENERAL']['PARENT_PROXY'], self.config['GENERAL']['PARENT_PROXY_PORT'])
try:
conn.putrequest('GET', '/')
conn.endheaders()
try:
errcode, errmsg, headers = conn.getreply()
try:
data = conn.getfile()
if not data.read():
self.die()
data.close()
except:
self.die()
except:
self.die()
except:
self.die()
except:
self.die()
return
#--------------------------------------------------------------
class timerThread:
"""
Used in place of SIGALRM as Windows doesn't support it.
"""
#--------------------------------------------------------------
def __init__(self, seconds, parentAlive, die_sig=signal.SIGINT):
self.seconds = seconds
self.parentAlive = parentAlive
self.die_sig = die_sig
self.alive = 1
#--------------------------------------------------------------
def run(self):
time.sleep(self.seconds)
if self.alive and self.parentAlive:
die_func = signal.getsignal(self.die_sig)
die_func(self.die_sig)
|
