/**
* Reads the peer's address. First a cookie has to be sent which has to match my own cookie,
* otherwise the connection will be refused
*/
protected Address readPeerAddress(Socket client_sock) throws Exception {
int timeout = client_sock.getSoTimeout();
client_sock.setSoTimeout(server.peerAddressReadTimeout());
try {
// read the cookie first
byte[] input_cookie = new byte[cookie.length];
in.readFully(input_cookie, 0, input_cookie.length);
if (!Arrays.equals(cookie, input_cookie))
throw new SocketException(
"BaseServer.TcpConnection.readPeerAddress(): cookie read by "
+ server.localAddress()
+ " does not match own cookie; terminating connection");
// then read the version
short version = in.readShort();
if (!Version.isBinaryCompatible(version))
throw new IOException(
"packet from "
+ client_sock.getInetAddress()
+ ":"
+ client_sock.getPort()
+ " has different version ("
+ Version.print(version)
+ ") from ours ("
+ Version.printVersion()
+ "); discarding it");
short addr_len = in.readShort(); // only needed by NioConnection
Address client_peer_addr = new IpAddress();
client_peer_addr.readFrom(in);
updateLastAccessed();
return client_peer_addr;
} finally {
client_sock.setSoTimeout(timeout);
}
}
/**
* Reads the peer's address. First a cookie has to be sent which has to match my own cookie,
* otherwise the connection will be refused
*/
private Address readPeerAddress(Socket client_sock) throws Exception {
int timeout = client_sock.getSoTimeout();
client_sock.setSoTimeout(peer_addr_read_timeout);
try {
// read the cookie first
byte[] input_cookie = new byte[cookie.length];
in.readFully(input_cookie, 0, input_cookie.length);
if (!matchCookie(input_cookie))
throw new SocketException(
"ConnectionMap.Connection.readPeerAddress(): cookie read by "
+ getLocalAddress()
+ " does not match own cookie; terminating connection");
// then read the version
short version = in.readShort();
if (!Version.isBinaryCompatible(version)) {
if (log.isWarnEnabled())
log.warn(
new StringBuilder("packet from ")
.append(client_sock.getInetAddress())
.append(':')
.append(client_sock.getPort())
.append(" has different version (")
.append(Version.print(version))
.append(") from ours (")
.append(Version.printVersion())
.append("). This may cause problems")
.toString());
}
Address client_peer_addr = new IpAddress();
client_peer_addr.readFrom(in);
updateLastAccessed();
return client_peer_addr;
} finally {
client_sock.setSoTimeout(timeout);
}
}
public short readShort() throws IOException {
return in.readShort();
}
/** Internal method. Get and check response packet from searchd. */
private byte[] _GetResponse(Socket sock) {
/* connect */
DataInputStream sIn = null;
InputStream SockInput = null;
try {
SockInput = sock.getInputStream();
sIn = new DataInputStream(SockInput);
} catch (IOException e) {
_error = "getInputStream() failed: " + e;
return null;
}
/* read response */
byte[] response = null;
short status = 0, ver = 0;
int len = 0;
try {
/* read status fields */
status = sIn.readShort();
ver = sIn.readShort();
len = sIn.readInt();
/* read response if non-empty */
if (len <= 0) {
_error = "invalid response packet size (len=" + len + ")";
return null;
}
response = new byte[len];
sIn.readFully(response, 0, len);
/* check status */
if (status == SEARCHD_WARNING) {
DataInputStream in = new DataInputStream(new ByteArrayInputStream(response));
int iWarnLen = in.readInt();
_warning = new String(response, 4, iWarnLen);
System.arraycopy(response, 4 + iWarnLen, response, 0, response.length - 4 - iWarnLen);
} else if (status == SEARCHD_ERROR) {
_error = "searchd error: " + new String(response, 4, response.length - 4);
return null;
} else if (status == SEARCHD_RETRY) {
_error = "temporary searchd error: " + new String(response, 4, response.length - 4);
return null;
} else if (status != SEARCHD_OK) {
_error = "searched returned unknown status, code=" + status;
return null;
}
} catch (IOException e) {
if (len != 0) {
/* get trace, to provide even more failure details */
PrintWriter ew = new PrintWriter(new StringWriter());
e.printStackTrace(ew);
ew.flush();
ew.close();
String sTrace = ew.toString();
/* build error message */
_error =
"failed to read searchd response (status="
+ status
+ ", ver="
+ ver
+ ", len="
+ len
+ ", trace="
+ sTrace
+ ")";
} else {
_error = "received zero-sized searchd response (searchd crashed?): " + e.getMessage();
}
return null;
} finally {
if (_socket == null) {
try {
if (sIn != null) sIn.close();
if (sock != null && !sock.isConnected()) sock.close();
} catch (IOException e) {
/* silently ignore close failures; nothing could be done anyway */
}
}
}
return response;
}
