diff --git a/meta/lib/oeqa/utils/oeqemuconsole.py b/meta/lib/oeqa/utils/oeqemuconsole.py new file mode 100644 index 0000000000..95a21332de --- /dev/null +++ b/meta/lib/oeqa/utils/oeqemuconsole.py @@ -0,0 +1,45 @@ +import socket +import time +import re +from telnetlib import Telnet + +class oeQemuConsole(Telnet): + + """ + Override Telnet class to use unix domain sockets, + Telnet uses AF_INET for socket, we don't want that. + Also, provide a read_all variant with timeout, that + returns whatever output there is. + """ + + def __init__(self, stream, logfile): + + Telnet.__init__(self, host=None) + self.stream = stream + self.logfile = logfile + self.eof = 0 + self.sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) + self.sock.connect(stream) + + def log(self, msg): + if self.logfile: + with open(self.logfile, "a") as f: + f.write("%s\n" % msg) + + + def read_all_timeout(self, match, timeout=200): + """Read until EOF or until timeout or until match. + """ + ret = False + self.process_rawq() + endtime = time.time() + timeout + while not self.eof and time.time() < endtime: + self.fill_rawq() + self.process_rawq() + if re.search(match, self.cookedq): + ret = True + break + buf = self.cookedq + self.cookedq = '' + self.log(buf) + return (ret, buf)