external/VirtScreen
1
0
Fork 0
mirror of https://github.com/kbumsik/VirtScreen.git synced 2025-03-09 15:40:18 +00:00
Code Issues Releases Wiki Activity

Password: Pass password to pipe, to hide password in command line

Browse source
This commit is contained in:
Bumsik Kim 2018-05-13 22:19:17 -04:00
parent dc7199e47d
commit 09cf4cb72f
No known key found for this signature in database
GPG key ID: E31041C8EC5B01C6
2 changed files with 18 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

24
virtscreen.py
View file

@ -51,8 +51,12 @@ class SubprocessWrapper:
with open(os.devnull, "w") as f:
subprocess.check_call(arg.split(), stdout=f, stderr=f)
def run(self, arg: str) -> str:
return subprocess.run(arg.split(), stdout=subprocess.PIPE).stdout.decode('utf-8')
def run(self, arg: str, input: str = None) -> str:
if input:
input = input.encode('utf-8')
with open(os.devnull, "w") as f:
return subprocess.run(arg.split(), input=input, stdout=subprocess.PIPE,
stderr=f).stdout.decode('utf-8')
#-------------------------------------------------------------------------------
# Twisted class
@ -157,7 +161,6 @@ class Display(object):
ret += f" not active {self.width}x{self.height}"
return ret
class DisplayProperty(QObject):
_display: Display
@ -538,11 +541,12 @@ class Backend(QObject):
@pyqtSlot(str)
def createVNCPassword(self, password):
if password:
password += '\n' + password + '\n\n' # verify + confirm
p = SubprocessWrapper()
try:
p.run(f"x11vnc -storepasswd {password} {X11VNC_PASSWORD_PATH}")
except:
print("Failed creating password")
p.run(f"x11vnc -storepasswd {X11VNC_PASSWORD_PATH}", input=password)
except Exception as e:
print("Failed creating password", e)
return
self.vncUsePassword = True
else:
@ -566,16 +570,20 @@ class Backend(QObject):
print("VNC Server is already running.")
return
# regex used in callbacks
re_connection = re.compile(r"^.*Got connection from client.*$", re.M)
patter_connected = re.compile(r"^.*Got connection from client.*$", re.M)
patter_disconnected = re.compile(r"^.*client_count: 0*$", re.M)
# define callbacks
def _onConnected():
print("VNC started.")
self.vncState = self.VNCState.WAITING
def _onReceived(data):
data = data.decode("utf-8")
if (self._vncState is not self.VNCState.CONNECTED) and re_connection.search(data):
if (self._vncState is not self.VNCState.CONNECTED) and patter_connected.search(data):
print("VNC connected.")
self.vncState = self.VNCState.CONNECTED
if (self._vncState is self.VNCState.CONNECTED) and patter_disconnected.search(data):
print("VNC disconnected.")
self.vncState = self.VNCState.WAITING
def _onEnded(exitCode):
print("VNC Exited.")
self.vncState = self.VNCState.OFF