diff options
Diffstat (limited to 'tools/runner/java/vogar/HostMonitor.java')
| -rw-r--r-- | tools/runner/java/vogar/HostMonitor.java | 200 |
1 files changed, 200 insertions, 0 deletions
diff --git a/tools/runner/java/vogar/HostMonitor.java b/tools/runner/java/vogar/HostMonitor.java new file mode 100644 index 0000000000..6c30d8d2f6 --- /dev/null +++ b/tools/runner/java/vogar/HostMonitor.java @@ -0,0 +1,200 @@ +/* + * Copyright (C) 2010 The Android Open Source Project + * + * 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. + */ + +package vogar; + +import org.xml.sax.Attributes; +import org.xml.sax.InputSource; +import org.xml.sax.SAXException; +import org.xml.sax.helpers.DefaultHandler; + +import javax.xml.parsers.ParserConfigurationException; +import javax.xml.parsers.SAXParser; +import javax.xml.parsers.SAXParserFactory; +import java.io.BufferedInputStream; +import java.io.IOException; +import java.io.InputStream; +import java.net.Socket; +import java.util.Arrays; +import java.util.Collections; +import java.util.logging.Level; +import java.util.logging.Logger; + +/** + * Connects to a target process to monitor its action. + */ +class HostMonitor { + + private static final Logger logger = Logger.getLogger(HostMonitor.class.getName()); + + private final int MAX_CONNECT_ATTEMPTS = 10; + private final int CONNECTION_ATTEMPT_DELAY_MILLIS = 1000; + + /** + * Connect to the target process on the given port, read all of its + * outcomes into {@code handler}, and disconnect. + */ + public boolean monitor(int port, Handler handler) { + Socket socket; + InputStream in; + try { + int attempt = 0; + do { + socket = new Socket("localhost", port); + in = new BufferedInputStream(socket.getInputStream()); + if (checkStream(in)) { + logger.fine("action monitor connected to " + socket.getRemoteSocketAddress()); + break; + } + + if (attempt++ == MAX_CONNECT_ATTEMPTS) { + throw new IOException("Exceeded max connection attempts!"); + } + logger.fine("connection " + attempt + " to localhost:" + port + " is dead; retrying..."); + in.close(); + socket.close(); + try { + Thread.sleep(CONNECTION_ATTEMPT_DELAY_MILLIS); + } catch (InterruptedException e) { + } + } while (true); + } catch (IOException e) { + logger.log(Level.WARNING, "Failed to connect to localhost:" + port, e); + return false; + } + + try { + SAXParser parser = SAXParserFactory.newInstance().newSAXParser(); + InputSource inputSource = new InputSource(in); + parser.parse(inputSource, new ClientXmlHandler(handler)); + } catch (ParserConfigurationException e) { + throw new RuntimeException(e); + } catch (IOException e) { + logger.log(Level.WARNING, "Connection error from localhost:" + port, e); + return false; + } catch (SAXException e) { + logger.log(Level.WARNING, "Received bad XML from localhost:" + port, e); + return false; + } + + try { + socket.close(); + } catch (IOException ignored) { + } + + return true; + } + + /** + * Somewhere between the host and client process, broken socket connections + * are being accepted. Before we try to do any work on such a connection, + * check it to make sure it's not dead! + * + * TODO: file a bug (against adb?) for this + */ + private boolean checkStream(InputStream in) throws IOException { + in.mark(1); + if (in.read() == -1) { + return false; + } else { + in.reset(); + return true; + } + } + + /** + * Handles updates on the outcomes of a target process. + */ + public interface Handler { + + /** + * Receive a completed outcome. + */ + void outcome(Outcome outcome); + + /** + * Receive partial output from an action being executed. + */ + void output(String outcomeName, String output); + } + + class ClientXmlHandler extends DefaultHandler { + private final Handler handler; + + private String currentOutcomeName; + private String currentActionName; + private Result currentResult; + private StringBuilder output = new StringBuilder(); + + ClientXmlHandler(Handler handler) { + this.handler = handler; + } + + /* + * Our XML wire format looks like this: + * + * <?xml version='1.0' encoding='UTF-8' ?> + * <vogar> + * <outcome name="java.util.FormatterTest" action="java.util.FormatterTest"> + * test output + * more test output + * <result value="SUCCESS" /> + * </outcome> + * </vogar> + */ + + @Override public void startElement(String uri, String localName, + String qName, Attributes attributes) throws SAXException { + if (qName.equals("outcome")) { + if (currentOutcomeName != null) { + throw new IllegalStateException(); + } + + currentOutcomeName = attributes.getValue("name"); + currentActionName = attributes.getValue("action"); + return; + + } else if (qName.equals("result")) { + currentResult = Result.valueOf(attributes.getValue("value")); + return; + + } else if (!qName.equals("vogar")) { + throw new IllegalArgumentException("Unrecognized: " + qName); + } + } + + @Override public void characters(char[] ch, int start, int length) + throws SAXException { + if (currentOutcomeName != null) { + String text = new String(ch, start, length); + output.append(text); + handler.output(currentOutcomeName, text); + } + } + + @Override public void endElement(String uri, String localName, String qName) + throws SAXException { + if (qName.equals("outcome")) { + handler.outcome(new Outcome(currentOutcomeName, currentActionName, + currentResult, Collections.singletonList(output.toString()))); + currentOutcomeName = null; + currentActionName = null; + currentResult = null; + output.delete(0, output.length()); + } + } + } +} |