/*
* This file is part of libbluray
* Copyright (C) 2014 Petri Hintukainen <phintuka@users.sourceforge.net>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library. If not, see
* <http://www.gnu.org/licenses/>.
*/
package org.videolan;
import java.lang.reflect.Method;
import java.util.Iterator;
import java.util.LinkedList;
import java.security.AccessController;
import java.security.PrivilegedActionException;
import java.security.PrivilegedExceptionAction;
import java.net.Socket;
import java.net.SocketImpl;
class BDJSockets {
protected BDJSockets() {
}
protected synchronized void add(Object obj) {
if (!(obj instanceof SocketImpl)) {
logger.error("expected SocketImpl, got " + obj);
throw new Error("expected SocketImpl");
}
if (closed) {
logger.error("Terminated Xlet tried to create socket at " + Logger.dumpStack());
throw new Error("Terminated Xlet can not create sockets");
}
/* drop closed sockets */
for (Iterator it = sockets.iterator(); it.hasNext(); ) {
SocketImpl socketImpl = (SocketImpl)it.next();
Socket socket = getSocket(socketImpl);
if (socket != null && socket.isClosed()) {
it.remove();
}
}
sockets.addLast(obj);
}
protected synchronized void closeAll() {
closed = true;
while (!sockets.isEmpty()) {
SocketImpl socketImpl = (SocketImpl)sockets.removeFirst();
Socket socket = getSocket(socketImpl);
if (socket != null && !socket.isClosed()) {
logger.warning("Closing " + socket);
try {
socket.close();
} catch (Exception e) {
logger.error("Failed to close socket: " + e);
}
}
}
}
private Socket getSocket(SocketImpl socketImpl) {
try {
final SocketImpl si = socketImpl;
return (Socket)AccessController.doPrivileged(
new PrivilegedExceptionAction() {
public Object run() throws Exception {
Method getSocket = SocketImpl.class.getDeclaredMethod("getSocket", new Class[0]);
getSocket.setAccessible(true);
return getSocket.invoke(si, new Object[0]);
}
});
} catch (PrivilegedActionException e) {
logger.error("Failed to get Socket: " + e.getException() + " at " + Logger.dumpStack());
return null;
}
}
private LinkedList sockets = new LinkedList();
private boolean closed = false;
private static final Logger logger = Logger.getLogger(BDJSockets.class.getName());
}