mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-22 17:46:37 +03:00
new file with getch() method which we'll use for good samaritan feature
This commit is contained in:
parent
4bef12a2b4
commit
a6ab42c873
94
lib/utils/getch.py
Normal file
94
lib/utils/getch.py
Normal file
|
@ -0,0 +1,94 @@
|
|||
#!/usr/bin/env python
|
||||
|
||||
"""
|
||||
$Id$
|
||||
|
||||
This file is part of the sqlmap project, http://sqlmap.sourceforge.net.
|
||||
|
||||
sqlmap 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 version 2 of the License.
|
||||
|
||||
sqlmap 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 sqlmap; if not, write to the Free Software Foundation, Inc., 51
|
||||
Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
||||
"""
|
||||
|
||||
### Reference: http://code.activestate.com/recipes/134892/
|
||||
class _Getch:
|
||||
"""
|
||||
Gets a single character from standard input. Does not echo to
|
||||
the screen.
|
||||
"""
|
||||
def __init__(self):
|
||||
try:
|
||||
self.impl = _GetchWindows()
|
||||
except ImportError:
|
||||
try:
|
||||
self.impl = _GetchMacCarbon()
|
||||
except(AttributeError, ImportError):
|
||||
self.impl = _GetchUnix()
|
||||
|
||||
def __call__(self): return self.impl()
|
||||
|
||||
|
||||
class _GetchUnix:
|
||||
def __init__(self):
|
||||
import tty, sys
|
||||
|
||||
def __call__(self):
|
||||
import sys, tty, termios
|
||||
fd = sys.stdin.fileno()
|
||||
old_settings = termios.tcgetattr(fd)
|
||||
try:
|
||||
tty.setraw(sys.stdin.fileno())
|
||||
ch = sys.stdin.read(1)
|
||||
finally:
|
||||
termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
|
||||
return ch
|
||||
|
||||
|
||||
class _GetchWindows:
|
||||
def __init__(self):
|
||||
import msvcrt
|
||||
|
||||
def __call__(self):
|
||||
import msvcrt
|
||||
return msvcrt.getch()
|
||||
|
||||
|
||||
class _GetchMacCarbon:
|
||||
"""
|
||||
A function which returns the current ASCII key that is down;
|
||||
if no ASCII key is down, the null string is returned. The
|
||||
page http://www.mactech.com/macintosh-c/chap02-1.html was
|
||||
very helpful in figuring out how to do this.
|
||||
"""
|
||||
def __init__(self):
|
||||
import Carbon
|
||||
Carbon.Evt #see if it has this (in Unix, it doesn't)
|
||||
|
||||
def __call__(self):
|
||||
import Carbon
|
||||
if Carbon.Evt.EventAvail(0x0008)[0]==0: # 0x0008 is the keyDownMask
|
||||
return ''
|
||||
else:
|
||||
#
|
||||
# The event contains the following info:
|
||||
# (what,msg,when,where,mod)=Carbon.Evt.GetNextEvent(0x0008)[1]
|
||||
#
|
||||
# The message (msg) contains the ASCII char which is
|
||||
# extracted with the 0x000000FF charCodeMask; this
|
||||
# number is converted to an ASCII character with chr() and
|
||||
# returned
|
||||
#
|
||||
(what,msg,when,where,mod)=Carbon.Evt.GetNextEvent(0x0008)[1]
|
||||
return chr(msg & 0x000000FF)
|
||||
|
||||
|
||||
getch = _Getch()
|
Loading…
Reference in New Issue
Block a user