mirror of
https://github.com/xcat2/confluent.git
synced 2026-09-05 12:37:56 +00:00
Markus Hilger
b996a30a44
The Console was never connected, so main_loop ran against a session that had never been established. Input arrived on a thread that called send_data and dropped the coroutine; the thread is gone and the loop watches stdin with add_reader instead. The output handler was a plain function that Console._print_data awaits.
99 lines
2.9 KiB
Python
Executable File
99 lines
2.9 KiB
Python
Executable File
# Copyright 2013 IBM Corporation
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
""" A simple little script to exemplify/test ipmi.console module """
|
|
|
|
import asyncio
|
|
import errno
|
|
import fcntl
|
|
import os
|
|
import sys
|
|
import termios
|
|
import tty
|
|
|
|
|
|
from aiohmi.ipmi import console
|
|
|
|
|
|
# Tasks are kept alive here: the event loop only holds weak references to
|
|
# them, so a task that nothing else refers to can be collected mid flight.
|
|
_pending = set()
|
|
|
|
|
|
def _got_input(sol):
|
|
"""Called by the event loop whenever stdin has something to read"""
|
|
|
|
try:
|
|
data = sys.stdin.read()
|
|
except (IOError, OSError) as e:
|
|
if e.errno == errno.EAGAIN:
|
|
return
|
|
raise
|
|
if not data:
|
|
return
|
|
task = asyncio.get_running_loop().create_task(sol.send_data(data))
|
|
_pending.add(task)
|
|
task.add_done_callback(_pending.discard)
|
|
|
|
|
|
async def _print(data):
|
|
bailout = False
|
|
if not isinstance(data, str):
|
|
bailout = True
|
|
data = repr(data)
|
|
sys.stdout.write(data)
|
|
sys.stdout.flush()
|
|
if bailout:
|
|
raise Exception(data)
|
|
|
|
|
|
async def main():
|
|
tcattr = termios.tcgetattr(sys.stdin)
|
|
newtcattr = tcattr
|
|
# TODO(jbjohnso): add our exit handler
|
|
newtcattr[-1][termios.VINTR] = 0
|
|
newtcattr[-1][termios.VSUSP] = 0
|
|
termios.tcsetattr(sys.stdin, termios.TCSADRAIN, newtcattr)
|
|
|
|
tty.setraw(sys.stdin.fileno())
|
|
currfl = fcntl.fcntl(sys.stdin.fileno(), fcntl.F_GETFL)
|
|
fcntl.fcntl(sys.stdin.fileno(), fcntl.F_SETFL, currfl | os.O_NONBLOCK)
|
|
|
|
try:
|
|
if sys.argv[3] is None:
|
|
passwd = os.environ['IPMIPASSWORD']
|
|
else:
|
|
passwd_file = sys.argv[3]
|
|
with open(passwd_file, "r") as f:
|
|
passwd = f.read()
|
|
|
|
sol = console.Console(bmc=sys.argv[1], userid=sys.argv[2],
|
|
password=passwd, iohandler=_print, force=True)
|
|
await sol.connect()
|
|
asyncio.get_running_loop().add_reader(sys.stdin, _got_input, sol)
|
|
try:
|
|
await sol.main_loop()
|
|
finally:
|
|
asyncio.get_running_loop().remove_reader(sys.stdin)
|
|
|
|
except Exception:
|
|
currfl = fcntl.fcntl(sys.stdin.fileno(), fcntl.F_GETFL)
|
|
fcntl.fcntl(sys.stdin.fileno(), fcntl.F_SETFL, currfl ^ os.O_NONBLOCK)
|
|
termios.tcsetattr(sys.stdin, termios.TCSANOW, tcattr)
|
|
return 0
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(asyncio.run(main()))
|