forked from singe/mainframe_brute
-
Notifications
You must be signed in to change notification settings - Fork 17
/
py3270wrapper.py
executable file
·88 lines (78 loc) · 2.97 KB
/
py3270wrapper.py
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
#!/usr/bin/env python
from py3270 import Emulator,CommandError,FieldTruncateError,TerminatedError,WaitError,KeyboardStateError,FieldTruncateError,x3270App,s3270App
import platform
from time import sleep
# Override some behaviour of py3270 library
class EmulatorIntermediate(Emulator):
def __init__(self, visible=True, delay=0):
Emulator.__init__(self, visible)
self.delay = delay
def send_enter(self): # Allow a delay to be configured
self.exec_command('Enter')
if self.delay > 0:
sleep(self.delay)
def screen_get(self):
response = self.exec_command('Ascii()')
return response.data
# Send text without triggering field protection
def safe_send(self, text):
for i in xrange(0,len(text)):
self.send_string(text[i])
if self.status.field_protection == 'P':
return False # We triggered field protection, stop
return True # Safe
# Fill fields in carefully, checking for triggering field protections
def safe_fieldfill(self, ypos, xpos, tosend, length):
if length - len(tosend) < 0:
raise FieldTruncateError('length limit %d, but got "%s"' % (length, tosend))
if xpos is not None and ypos is not None:
self.move_to(ypos, xpos)
try:
self.delete_field()
if safe_send(self, tosend):
return True # Hah, we win, take that mainframe
else:
return False # we entered what we could, bailing
except CommandError, e:
# We hit an error, get mad
return False
# if str(e) == 'Keyboard locked':
# Search the screen for text when we don't know exactly where it is, checking for read errors
def find_response(self, response):
for rows in xrange(1,int(self.status.row_number)+1):
for cols in xrange(1,int(self.status.col_number)+1-len(response)):
try:
if self.string_found(rows, cols, response):
return True
except CommandError, e:
# We hit a read error, usually because the screen hasn't returned
# increasing the delay works
sleep(self.delay)
self.delay += 1
whine('Read error encountered, assuming host is slow, increasing delay by 1s to: ' + str(self.delay),kind='warn')
return False
return False
# Get the current x3270 cursor position
def get_pos(self):
results = self.exec_command('Query(Cursor)')
row = int(results.data[0].split(' ')[0])
col = int(results.data[0].split(' ')[1])
return (row,col)
def get_hostinfo(self):
return self.exec_command('Query(Host)').data[0].split(' ')
# Set the emulator intelligently based on your platform
if platform.system() == 'Darwin':
class WrappedEmulator(EmulatorIntermediate):
x3270App.executable = 'x3270'
s3270App.executable = 's3270'
elif platform.system() == 'Linux':
class WrappedEmulator(EmulatorIntermediate):
x3270App.executable = 'x3270'
s3270App.executable = 's3270'
elif platform.system() == 'Windows':
class WrappedEmulator(EmulatorIntermediate):
#x3270_executable = 'Windows_Binaries/wc3270.exe'
x3270App.executable = 'wc3270.exe'
else:
logger('Your Platform:', platform.system(), 'is not supported at this time.',kind='err')
sys.exit(1)