/*
* Copyright 2012-2013 eBay Software Foundation and ios-driver committers
*
* 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 org.uiautomation.ios.wkrdp.internal;
import org.openqa.selenium.SessionNotCreatedException;
import org.openqa.selenium.WebDriverException;
import org.uiautomation.ios.utils.PlistManager;
import org.uiautomation.ios.wkrdp.ResponseFinder;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;
import java.net.SocketException;
import java.util.Arrays;
import java.util.List;
import java.util.logging.Logger;
/**
* WKRDP implementation for the simulator.
*/
public class SimulatorProtocolImpl extends WebKitRemoteDebugProtocol {
private static final Logger log = Logger.getLogger(SimulatorProtocolImpl.class.getName());
private static Socket socket;
private ByteArrayOutputStream buf = new ByteArrayOutputStream();
private final String LOCALHOST_IPV6 = "::1";
private final int port = 27753;
public SimulatorProtocolImpl(List<ResponseFinder> finders) {
super(finders);
}
public void start() {
try {
if (socket != null && (socket.isConnected() || !socket.isClosed())) {
socket.close();
}
socket = new Socket(LOCALHOST_IPV6, port);
} catch (IOException e) {
throw new SessionNotCreatedException(
"Cannot connect to the simulator socket on port 27753. Check that you don't have a firewall blocking it.",
e);
}
startListenerThread();
//sendCommand(PlistManager.SET_CONNECTION_KEY);
//sendCommand(PlistManager.CONNECT_TO_APP);
//sendCommand(PlistManager.SET_SENDER_KEY);
}
/**
* sends the message to the AUT.
*/
protected void sendMessage(String xml) {
byte[] bytes = null;
try {
bytes = xml.getBytes("UTF-8");
writeBytes(bytes);
} catch (SocketException se) {
// when we get a socket exception, let's try to re-establish
stop();
start();
try {
writeBytes(bytes);
} catch (IOException e) {
throw new WebDriverException(e);
}
} catch (IOException e) {
throw new WebDriverException(e);
}
}
private void writeBytes(byte[] bytes) throws IOException {
OutputStream os = socket.getOutputStream();
os.write((byte) ((bytes.length >> 24) & 0xFF));
os.write((byte) ((bytes.length >> 16) & 0xFF));
os.write((byte) ((bytes.length >> 8) & 0xFF));
os.write((byte) (bytes.length & 0xFF));
os.write(bytes);
}
/**
* reads the messages from the AUT.
*/
private void pushInput(byte[] inputBytes) throws Exception {
buf.write(inputBytes);
while (buf.size() >= 4) {
byte[] bytes = buf.toByteArray();
int size = 0;
size = (size << 8) + byteToInt(bytes[0]);
size = (size << 8) + byteToInt(bytes[1]);
size = (size << 8) + byteToInt(bytes[2]);
size = (size << 8) + byteToInt(bytes[3]);
if (bytes.length >= 4 + size) {
String
message =
new PlistManager().plistBinaryToXml(Arrays.copyOfRange(bytes, 4, size + 4));
handler.handle(message);
buf = new ByteArrayOutputStream();
buf.write(bytes, 4 + size, bytes.length - size - 4);
} else {
break;
}
}
}
/**
* listen for a complete message.
*/
protected void read() throws Exception {
InputStream is = socket.getInputStream();
while (is.available() > 0) {
byte[] bytes = new byte[1024 * 1024];
int read = is.read(bytes);
byte[] actuallyRead = new byte[read];
System.arraycopy(bytes, 0, actuallyRead, 0, read);
pushInput(actuallyRead);
}
}
private int byteToInt(byte b) {
int i = (int) b;
return i >= 0 ? i : i + 256;
}
public void stop() {
stopListenerThread();
try {
if (socket != null) {
socket.close();
}
} catch (Exception e) {
log.warning("couldn't stop the wkrdp properly." + e.getMessage());
}
}
}